summaryrefslogtreecommitdiff
path: root/tests/micropython/heapalloc_yield_from.py
diff options
context:
space:
mode:
authorDavid Lechner <david@lechnology.com>2020-03-22 21:26:08 -0500
committerDamien George <damien.p.george@gmail.com>2020-03-30 13:21:58 +1100
commit3dc324d3f1312e40d3a8ed87e7244966bb756f26 (patch)
tree94ff44f8eabba0039582c245b901173597edd11e /tests/micropython/heapalloc_yield_from.py
parent488613bca6c460340ed2995ae5cafafe22d0bfff (diff)
tests: Format all Python code with black, except tests in basics subdir.
This adds the Python files in the tests/ directory to be formatted with ./tools/codeformat.py. The basics/ subdirectory is excluded for now so we aren't changing too much at once. In a few places `# fmt: off`/`# fmt: on` was used where the code had special formatting for readability or where the test was actually testing the specific formatting.
Diffstat (limited to 'tests/micropython/heapalloc_yield_from.py')
-rw-r--r--tests/micropython/heapalloc_yield_from.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/micropython/heapalloc_yield_from.py b/tests/micropython/heapalloc_yield_from.py
index 8443210f3..58788b1db 100644
--- a/tests/micropython/heapalloc_yield_from.py
+++ b/tests/micropython/heapalloc_yield_from.py
@@ -6,8 +6,12 @@ import micropython
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))
@@ -18,12 +22,16 @@ micropython.heap_unlock()
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))