Почему генератор возвращает неверное значение?

Question

[Kit Scribe]

Есть вот такой

код

def test_gen():
    value = yield
    end = value + 20

    while True:
        if value > end:
            break
        yield value
        value += 1

mygen = test_gen()
next(mygen)
mygen.send(10)

for item in mygen:
    print(item)

Я ожидаю, что итерация начнётся с 10 и закончится 30, но получаю следующий

вывод

11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

Почему происходит именно так?

Answer 1

[Vindicar]

Потому что .send() не только передаёт значение внутрь генератора, но и возвращает следующее yield'нутое значение.

mygen = test_gen()
next(mygen)
print('Send:', mygen.send(10))

выведет Send: 10.
Так что тебе придётся сделать ещё один промежуточный yield None сразу после получения value = yield.

Ну и да, зачем break когда можно сделать условие в while?

Comments

[Kit Scribe]

Большое спасибо за объяснение и предложенные правки к коду

[Kit Scribe]

Подскажи, пожалуйста, а если я реализую генератор на основе класса с реализацией интерфейса итератора, мне нужно yield'ить значения или использовать return?

[Vindicar]

Kit Scribe, интерфейс итератора - это __next__()?
Насколько я знаю, __next__() должен вернуть следующее значение.

[Kit Scribe]

Vindicar, __iter__ и __next__, да. Добро, спасибо, завтра попробую