fizzbuzz: add working fizzbuzz implementation

fizzbuzz/fizzbuzz.py

@@ -2,8 +2,19 @@
 
 
 def fizzbuzz(max: int = 100) -> None:
+    words = {
+        3: "fizz",
+        5: "buzz",
+    }
     for i in range(1, max + 1):
-        print(i)
+        out = []
+        for div, word in words.items():
+            if i % div == 0:
+                out.append(word)
+        if len(out) > 0:
+            print("".join(out))
+        else:
+            print(i)
 
 
 if __name__ == "__main__":

fizzbuzz/test_fizzbuzz.py

@@ -1,7 +1,43 @@
 from fizzbuzz import fizzbuzz
 
 
-def test_fizzbuzz_counts_to_two(capsys):
+def list_output(max, capsys, expected_list):
-    fizzbuzz(2)
+    fizzbuzz(max)
     out, __ = capsys.readouterr()
-    assert out == "1\n2\n"
+    assert out == "\n".join(map(lambda x: str(x), expected_list)) + "\n"
+
+
+def test_fizzbuzz_counts_to_two(capsys):
+    list_output(2, capsys, [1, 2])
+
+
+def test_fizzbuzz_shows_fizz_on_three(capsys):
+    list_output(3, capsys, [1, 2, "fizz"])
+
+
+def test_fizzbuzz_shows_buzz_on_five(capsys):
+    list_output(5, capsys, [1, 2, "fizz", 4, "buzz"])
+
+
+def test_fizzbuzz_shows_fizzbuzz_on_fifteen(capsys):
+    list_output(
+        15,
+        capsys,
+        [
+            1,
+            2,
+            "fizz",
+            4,
+            "buzz",
+            "fizz",
+            7,
+            8,
+            "fizz",
+            "buzz",
+            11,
+            "fizz",
+            13,
+            14,
+            "fizzbuzz",
+        ],
+    )