吕诚-科技事业部

helloworld.py

@@ -1,2 +1,13 @@
 def get_greetings():
     return 'Hello World!'
+
+
+def fizzbuzz(i):
+    if i % 3 == 0 and i % 5 == 0:
+        return 'fizzbuzz'
+    elif i % 3 == 0:
+        return 'fizz'
+    elif i % 5 == 0:
+        return 'buzz'
+    else:
+        return str(i)

helloworld_test.py

@@ -1,5 +1,10 @@
 from helloworld import get_greetings
+from helloworld import fizzbuzz
 
 
 def test_get_helloworld():
     assert get_greetings() == 'Hello World!'
+    assert fizzbuzz(1) == '1'
+    assert fizzbuzz(3) == 'fizz'
+    assert fizzbuzz(5) == 'buzz'
+    assert fizzbuzz(15) == 'fizzbuzz'