顾文渊-大数据部

fizzbuzz.py

@@ -0,0 +1,15 @@
+# @File  : fizzbuzz.py
+# @Author: Wenyuan Gu
+# @Date  : 2020/1/15
+
+
+def fizzbuzz(n):
+    return 'Fizz'[n % 3 * 4:] + 'Buzz'[n % 5 * 4:] or n
+
+
+def main():
+    for i in range(1, 101):
+        print(fizzbuzz(i))
+
+if __name__ == '__main__':
+    main()

fizzbuzz_test.py

@@ -0,0 +1,20 @@
+# @File  : fizzbuzz_test.py
+# @Author: Wenyuan Gu
+# @Date  : 2020/1/15
+
+
+import pytest
+from fizzbuzz import fizzbuzz
+
+class Test_fizzbuzz():
+    def test2(self):
+        assert fizzbuzz(2) == 2
+    def test3(self):
+        assert fizzbuzz(3) == 'Fizz'
+    def test5(self):
+        assert fizzbuzz(5) == 'Buzz'
+    def test15(self):
+        assert fizzbuzz(15) == 'FizzBuzz'
+
+if __name__ == "__main__":
+    pytest.main()