micropython/tests/basics/gen_yield_from.py

56 lines
949 B
Python

# Case of terminating subgen using return with value
def gen():
yield 1
yield 2
return 3
def gen2():
print("here1")
print((yield from gen()))
print("here2")
g = gen2()
print(list(g))
# Like above, but terminate subgen using StopIteration
def gen3():
yield 1
yield 2
raise StopIteration
def gen4():
print("here1")
print((yield from gen3()))
print("here2")
g = gen4()
print(list(g))
# Like above, but terminate subgen using StopIteration with value
def gen5():
yield 1
yield 2
raise StopIteration(123)
def gen6():
print("here1")
print((yield from gen5()))
print("here2")
g = gen6()
print(list(g))
# StopIteration from within a Python function, within a native iterator (map), within a yield from
def gen7(x):
if x < 3:
return x
else:
raise StopIteration(444)
def gen8():
print((yield from map(gen7, range(100))))
g = gen8()
print(list(g))