fizzbuzz: add fizzbuzzer custom function creator

fizzbuzz/fizzbuzz.py

@@ -1,11 +1,9 @@
 #! /usr/bin/env python
+from typing import Callable, Dict
 
 
-def fizzbuzz(max: int = 100) -> None:
-    words = {
-        3: "fizz",
-        5: "buzz",
-    }
+def fizzbuzzer(words: Dict[int, str]) -> Callable[[int], None]:
+    def _fun(max: int) -> None:
         for i in range(1, max + 1):
             out = []
             for div, word in words.items():
@@ -16,6 +14,17 @@ def fizzbuzz(max: int = 100) -> None:
             else:
                 print(i)
 
+    return _fun
+
+
+def fizzbuzz(max: int = 100) -> None:
+    words = {
+        3: "fizz",
+        5: "buzz",
+    }
+    f = fizzbuzzer(words)
+    f(max)
+
 
 if __name__ == "__main__":
     fizzbuzz()

fizzbuzz/test_fizzbuzz.py

@@ -1,26 +1,30 @@
-from fizzbuzz import fizzbuzz
+from fizzbuzz import fizzbuzz, fizzbuzzer
 
 
-def list_output(max, capsys, expected_list):
-    fizzbuzz(max)
+def list_output(max, capsys, expected_list, function):
+    function(max)
     out, __ = capsys.readouterr()
     assert out == "\n".join(map(lambda x: str(x), expected_list)) + "\n"
 
 
+def list_fizzbuzz_output(max, capsys, expected_list):
+    list_output(max, capsys, expected_list, fizzbuzz)
+
+
 def test_fizzbuzz_counts_to_two(capsys):
-    list_output(2, capsys, [1, 2])
+    list_fizzbuzz_output(2, capsys, [1, 2])
 
 
 def test_fizzbuzz_shows_fizz_on_three(capsys):
-    list_output(3, capsys, [1, 2, "fizz"])
+    list_fizzbuzz_output(3, capsys, [1, 2, "fizz"])
 
 
 def test_fizzbuzz_shows_buzz_on_five(capsys):
-    list_output(5, capsys, [1, 2, "fizz", 4, "buzz"])
+    list_fizzbuzz_output(5, capsys, [1, 2, "fizz", 4, "buzz"])
 
 
 def test_fizzbuzz_shows_fizzbuzz_on_fifteen(capsys):
-    list_output(
+    list_fizzbuzz_output(
         15,
         capsys,
         [
@@ -41,3 +45,26 @@ def test_fizzbuzz_shows_fizzbuzz_on_fifteen(capsys):
             "fizzbuzz",
         ],
     )
+
+
+def test_can_foobarbazz_customization(capsys):
+    foobarbazz = fizzbuzzer({2: "foo", 3: "bar", 4: "bazz"})
+    list_output(
+        function=foobarbazz,
+        max=12,
+        capsys=capsys,
+        expected_list=[
+            1,
+            "foo",
+            "bar",
+            "foobazz",
+            5,
+            "foobar",
+            7,
+            "foobazz",
+            "bar",
+            "foo",
+            11,
+            "foobarbazz",
+        ],
+    )