summaryrefslogtreecommitdiff
path: root/tests/micropython/native_gen.py
blob: 7e1ee25bcf45252ade0a88e1bb9059b6c9ca77dc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# 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)))


# catching an exception from .throw()
@micropython.native
def gen3():
    try:
        yield 1
        yield 2
    except Exception as er:
        print("caught", repr(er))
        yield 3


g = gen3()
print(next(g))
print(g.throw(ValueError(42)))


# responding to .close()
@micropython.native
def gen4():
    try:
        yield 1
    except:
        print("raising GeneratorExit")
        raise GeneratorExit


g = gen4()
print(next(g))
print(g.close())