From 761d2f67415ba43f64c7b04b6ea72fd4b65afc50 Mon Sep 17 00:00:00 2001
From: Damien George <damien@micropython.org>
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 <damien@micropython.org>
---
 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