summaryrefslogtreecommitdiff
path: root/tests/micropython/native_gen.py
blob: 30c4c37beacd93c52b59aae3b61ae8deaaf385e7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# test for native generators

# simple generator with yield and return
@micropython.native
def gen1(x):
    yield x
    yield x + 1
    return x + 2
g = gen1(3)
print(next(g))
print(next(g))
try:
    next(g)
except StopIteration as e:
    print(e.args[0])

# using yield from
@micropython.native
def gen2(x):
    yield from range(x)
print(list(gen2(3)))