乔萧雅-电子支付研究院

fizzbuzz.py

@@ -0,0 +1,17 @@
+def fizz_buzz(num):
+
+    if num % 3 == 0 and num % 5 == 0:
+        return 'FizzBuzz'
+
+    if num % 3 == 0:
+        return 'Fizz'
+
+    if num % 5 == 0:
+        return 'Buzz'
+
+    return str(num)
+
+
+if __name__ == '__main__':
+    for i in range(1, 101):
+        print(fizz_buzz(i))

fizzbuzz_test.py

@@ -0,0 +1,17 @@
+import unittest
+from fizzbuzz import fizz_buzz
+
+
+class test_fb(unittest.TestCase):
+
+    def test_default_case(self):
+        self.assertEqual(fizz_buzz(1), '1')
+
+    def test_fizz(self):
+        self.assertEqual(fizz_buzz(6), 'Fizz')
+
+    def test_buzz(self):
+        self.assertEqual(fizz_buzz(10), 'Buzz')
+
+    def test_fizzbuzz(self):
+        self.assertEqual(fizz_buzz(30), 'FizzBuzz')