fizzbuzz.py

fizzbuzz.py

"""fizzbuzz
https://gist.github.com/kazuho/3300555 のようにPythonでif文を使わずにFizzBuzzを実装したもの

doctest実行するときは`python -m doctest fizzbuzz.py -v`
"""

from itertools import cycle
from collections.abc import Iterator

fizz = cycle(["", "", "Fizz"])
buzz = cycle(["", "", "", "", "Buzz"])


def fizzbuzz(n) -> Iterator[str]:
    """関数が呼び出された回数に応じてfizzbuzzとなる文字列を返す

    :return: fizzbuzzを実現する文字列
    >>> fb = fizzbuzz(5)
    >>> next(fb)
    '1'
    >>> next(fb)
    '2'
    >>> next(fb)
    'Fizz'
    >>> next(fb)
    '4'
    >>> next(fb)
    'Buzz'
    """
    for i in range(1, n + 1):
        res = (next(fizz) + next(buzz)) or str(i)
        yield res


if __name__ == "__main__":
    for i in fizzbuzz(100):
        print(i)