Fizzbuzz

fizzbuzz.py

import itertools


def fizzbuzz():
    for i in itertools.count(1):
        fizz = i % 3 == 0
        buzz = i % 5 == 0
        if fizz and buzz:
            yield 'fizz buzz'
        elif fizz:
            yield 'fizz'
        elif buzz:
            yield 'buzz'
        else:
            yield i
            
            
if __name__ == '__main__':
    print(
        list(
            itertools.islice(fizzbuzz(), 20)
        )
    )