tests/micropython/native_closure.py


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

# test native emitter can handle closures correctly

# basic closure
@micropython.native
def f():
    x = 1
    @micropython.native
    def g():
        nonlocal x
        return x
    return g
print(f()())

# closing over an argument
@micropython.native
def f(x):
    @micropython.native
    def g():
        nonlocal x
        return x
    return g
print(f(2)())

# closing over an argument and a normal local
@micropython.native
def f(x):
    y = 2 * x
    @micropython.native
    def g(z):
        return x + y + z
    return g
print(f(2)(3))