tests/micropython: Add test for native generators.

pull/1/head
Damien George 2019-09-26 16:53:47 +10:00
parent 6647d03e42
commit 095f90f04e
2 changed files with 25 additions and 0 deletions

View 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)))

View File

@ -0,0 +1,4 @@
3
4
5
[0, 1, 2]