赵富-科技事业部

fizzbuzz.py

@@ -0,0 +1,9 @@
+def fb(N):
+    res = ""
+    if N % 3 == 0:
+        res += "Fizz"
+    if N % 5 == 0:
+        res += "Buzz"
+    if len(res) == 0:
+        res = str(N)
+    return res

test_FizzBuzz.py

@@ -0,0 +1,9 @@
+from fizzbuzz import fb
+
+inputs = [1, 3, 5, 15]
+results = ["1", "Fizz", "Buzz", "FizzBuzz"]
+
+
+def test_fb():
+    for i in range(len(inputs)):
+        assert fb(inputs[i]) == results[i]