blob: 9d4f8c6940f16e1e88b1a39b85828d09227b2405 (
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
|
# Check that yield-from can work without heap allocation
try:
import micropython
micropython.heap_lock
except (ImportError, AttributeError):
print("SKIP")
raise SystemExit
# Yielding from a function generator
def sub_gen(a):
for i in range(a):
yield i
def gen(g):
yield from g
g = gen(sub_gen(4))
micropython.heap_lock()
print(next(g))
print(next(g))
micropython.heap_unlock()
# Yielding from a user iterator
class G:
def __init__(self):
self.value = 0
def __iter__(self):
return self
def __next__(self):
v = self.value
self.value += 1
return v
g = gen(G())
micropython.heap_lock()
print(next(g))
print(next(g))
micropython.heap_unlock()
|
