南瑀-大数据部

FizzBuzz.py

@@ -0,0 +1,11 @@
+class Game(object):
+    def __init__(self, num):
+        self.raw_num = num
+
+    def number_off(self):
+        result = ""
+        if self.raw_num % 3 == 0:
+            result += 'Fizz'
+        if self.raw_num % 5 == 0:
+            result += 'Buzz'
+        return str(self.raw_num) if result == "" else result

test_FizzBuzz.py

@@ -0,0 +1,24 @@
+import unittest
+from FizzBuzz import Game
+
+
+class MyTestCase(unittest.TestCase):
+    def test_get_1_if_pass_1(self):
+        game = Game(1)
+        self.assertEqual('1', game.number_off())
+
+    def test_get_Fizz_if_pass_3(self):
+        game = Game(3)
+        self.assertEqual('Fizz', game.number_off())
+
+    def test_get_Buzz_if_pass_5(self):
+        game = Game(5)
+        self.assertEqual('Buzz', game.number_off())
+
+    def test_get_FizzBuzz_if_pass_15(self):
+        game = Game(15)
+        self.assertEqual('FizzBuzz', game.number_off())
+
+
+if __name__ == '__main__':
+    unittest.main()