于祥祥-信息总中心

Fizz.py

@@ -0,0 +1,11 @@
+def FizzBuzz(i):
+    if i % 15 == 0:
+        return "FizzBuzz"
+
+    if i % 3 == 0:
+        return "Fizz"
+
+    if i % 5 == 0:
+        return "Buzz"
+
+    return i

test_Fizz.py

@@ -0,0 +1,14 @@
+import unittest
+from Fizz import FizzBuzz
+
+
+class test(unittest.TestCase):
+    def test_fizz(self):
+        self.assertEqual("FizzBuzz", FizzBuzz(30))
+        self.assertEqual("Fizz", FizzBuzz(3))
+        self.assertEqual("Buzz", FizzBuzz(10))
+        self.assertEqual(31, FizzBuzz(31))
+
+
+if __name__ == '__main__':
+    unittest.main()