tests/micropython: Add test for native generators.
This commit is contained in:
parent
6647d03e42
commit
095f90f04e
21
tests/micropython/native_gen.py
Normal file
21
tests/micropython/native_gen.py
Normal file
@ -0,0 +1,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)))
|
4
tests/micropython/native_gen.py.exp
Normal file
4
tests/micropython/native_gen.py.exp
Normal file
@ -0,0 +1,4 @@
|
||||
3
|
||||
4
|
||||
5
|
||||
[0, 1, 2]
|
Loading…
Reference in New Issue
Block a user