shenfeng · August 29, 2015 14:00 · Apr 30, 2014 · Apr 30, 2014
diff --git a/tw.py b/tw.py
@@ -7,11 +7,7 @@ def transform(specials, mapping, n):
     if str(specials[0]) in str(n):
         return mapping[specials[0]]
 
-    r = ''.join(mapping[i] for i in specials if n % i == 0)
-    if r:
-        return r  # rule 3, 4
-
-    return str(n)
+    return ''.join(mapping[i] for i in specials if n % i == 0) or str(n)
 
 
 def test():

diff --git a/tw.py b/tw.py
@@ -0,0 +1,65 @@
+def transform(specials, mapping, n):
+    """
+    https://www.jinshuju.net/f/EGQL3D
+
+    """
+    # rule 5
+    if str(specials[0]) in str(n):
+        return mapping[specials[0]]
+
+    r = ''.join(mapping[i] for i in specials if n % i == 0)
+    if r:
+        return r  # rule 3, 4
+
+    return str(n)
+
+
+def test():
+    right_answers = """
+    1
+2
+Fizz
+4
+Buzz
+Fizz
+Whizz
+8
+Fizz
+Buzz
+11
+Fizz
+Fizz
+Whizz
+FizzBuzz
+16
+17
+Fizz
+19
+Buzz
+    """
+
+    specials = [3, 5, 7]
+    mapping = {3: 'Fizz', 5: 'Buzz', 7: 'Whizz'}
+
+    right_answers = [line.strip() for line in right_answers.split('\n') if line.strip()]
+    for i in range(1, len(right_answers)):
+        assert transform(specials, mapping, i) == right_answers[i - 1]
+
+    print 'test pass'
+
+
+def main():
+    specials = []
+    while len(specials) != 3:
+        raw = raw_input('input three less than 10 numbers: ')
+        specials = [int(n) for n in raw.split(' ') if n.isdigit() and int(n) < 10]
+
+    mapping = dict(zip(specials, ['Fizz', 'Buzz', 'Whizz']))
+
+    for i in range(1, 100):
+        print transform(specials, mapping, i)
+
+
+if __name__ == "__main__":
+    test()
+    main()
No results found