summaryrefslogtreecommitdiff
path: root/tests/micropython/native_closure.py
blob: 07014e90dac2d53a3540cbec8f9da63a5ed5167b (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
# 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))