From 761d2f67415ba43f64c7b04b6ea72fd4b65afc50 Mon Sep 17 00:00:00 2001 From: Damien George Date: Wed, 18 May 2022 15:22:22 +1000 Subject: [PATCH] tests/micropython: Add more test cases for native generators. Signed-off-by: Damien George --- tests/micropython/native_gen.py | 31 +++++++++++++++++++++++++++++ tests/micropython/native_gen.py.exp | 6 ++++++ 2 files changed, 37 insertions(+) diff --git a/tests/micropython/native_gen.py b/tests/micropython/native_gen.py index fb42f9e25e..62c324634d 100644 --- a/tests/micropython/native_gen.py +++ b/tests/micropython/native_gen.py @@ -23,3 +23,34 @@ def gen2(x): print(list(gen2(3))) + + +# catching an exception from .throw() +@micropython.native +def gen3(): + try: + yield 1 + yield 2 + except Exception as er: + print("caught", repr(er)) + yield 3 + + +g = gen3() +print(next(g)) +print(g.throw(ValueError(42))) + + +# responding to .close() +@micropython.native +def gen4(): + try: + yield 1 + except: + print("raising GeneratorExit") + raise GeneratorExit + + +g = gen4() +print(next(g)) +print(g.close()) diff --git a/tests/micropython/native_gen.py.exp b/tests/micropython/native_gen.py.exp index cc09e309f2..fba4e558cc 100644 --- a/tests/micropython/native_gen.py.exp +++ b/tests/micropython/native_gen.py.exp @@ -2,3 +2,9 @@ 4 5 [0, 1, 2] +1 +caught ValueError(42,) +3 +1 +raising GeneratorExit +None