remove 'fizzbuzz_loop' function for simplicity

Author: AaronRobson

fizzbuzz.py

@@ -1,20 +1,13 @@
 #!/usr/bin/python3
 
-from typing import Iterable
-from itertools import count, islice
-
 
 def fizzbuzz(num: int) -> str:
     return ('Fizz' * int(num % 3 == 0) + 'Buzz' * int(num % 5 == 0)) or str(num)
 
 
-def fizzbuzz_loop() -> Iterable[str]:
-    return map(fizzbuzz, count(1))
-
-
 def main() -> None:
-    for value in islice(fizzbuzz_loop(), 100):
-        print(value)
+    for value in range(1, 100 + 1):
+        print(fizzbuzz(value))
 
 
 if __name__ == "__main__":

tests/test_fizzbuzz.py

@@ -2,7 +2,6 @@
 
 import unittest
 from unittest.mock import call, patch
-from itertools import islice
 
 import fizzbuzz
 
@@ -18,29 +17,6 @@ def test(self):
         self.assertEqual(f(15), 'FizzBuzz')
 
 
-class TestFizzBuzzLoop(unittest.TestCase):
-    def test(self):
-        expected = [
-            '1',
-            '2',
-            'Fizz',
-            '4',
-            'Buzz',
-            'Fizz',
-            '7',
-            '8',
-            'Fizz',
-            'Buzz',
-            '11',
-            'Fizz',
-            '13',
-            '14',
-            'FizzBuzz',
-        ]
-        actual = list(islice(fizzbuzz.fizzbuzz_loop(), len(expected)))
-        self.assertEqual(expected, actual)
-
-
 class TestMain(unittest.TestCase):
     @patch('fizzbuzz.print')
     def test(self, mock_patch):