L07E01: Generator function

Vytvořte modul progression obsahující funkci (generátorovou) arithmetic_progression(begin, step, end=None). Uvažujme pouze rostoucí posloupnosti.

Funkce arithmetic_progression(begin, step, end)

Generátor generující prvky aritmetické posloupnosti počínaje prvkem begin, diferencí step a končicí prvkem end (posloupnost poslední prvek neobsahuje). Pokud end není uveden (má hodnotu None), aritmetická posloupnost je nekonečná.

for number in arithmetic_progression(5, 2, 8):
    print(number)

# Vypíše následující čísla, v každý moment se vypočítá a uloží pouze jedno, celá posloupnost není nikde uložena
5
7

progression = arithmetic_progression(5, 2)

assert next(progression) == 5
assert next(progression) == 7
assert next(progression) == 9

Lokální testování

Funkčnost řešení ověříte následujícím příkazem:

pytest tests.py