潘浩-银联科技事业部

fizzbuzz.py

@@ -0,0 +1,14 @@
+def process(number):
+    if number % 15 == 0:
+        return 'FizzBuzz'
+    elif number % 3 == 0:
+        return 'Fizz'
+    elif number % 5 == 0:
+        return 'Buzz'
+    else:
+        return number
+
+
+if __name__ == '__main__':
+    for i in range(1, 101):
+        print(process(i))

test_fizzbuzz.py

@@ -0,0 +1,21 @@
+import fizzbuzz
+import unittest
+
+
+class TestFizzBuzz(unittest.TestCase):
+
+    def test_div3(self):
+        self.assertEqual(fizzbuzz.process(3), 'Fizz')
+
+    def test_div5(self):
+        self.assertEqual(fizzbuzz.process(5), 'Buzz')
+
+    def test_div3and5(self):
+        self.assertEqual(fizzbuzz.process(15), 'FizzBuzz')
+
+    def test_divother(self):
+        self.assertEqual(fizzbuzz.process(2), 2)
+
+
+if __name__ == '__main__':
+    unittest.main()