summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics
diff options
context:
space:
mode:
Diffstat (limited to 'tests/basics')
-rw-r--r--tests/basics/gen_yield_from_close.py14
-rw-r--r--tests/basics/gen_yield_from_throw.py11
-rw-r--r--tests/basics/gen_yield_from_throw2.py15
-rw-r--r--tests/basics/gen_yield_from_throw2.py.exp3
-rw-r--r--tests/basics/generator1.py3
5 files changed, 46 insertions, 0 deletions
diff --git a/tests/basics/gen_yield_from_close.py b/tests/basics/gen_yield_from_close.py
index 7982d5c990..674e117a08 100644
--- a/tests/basics/gen_yield_from_close.py
+++ b/tests/basics/gen_yield_from_close.py
@@ -85,3 +85,17 @@ try:
print(next(g))
except StopIteration:
print("StopIteration")
+
+# case where generator ignores the close request and yields instead
+def gen7():
+ try:
+ yield 123
+ except GeneratorExit:
+ yield 456
+
+g = gen7()
+print(next(g))
+try:
+ g.close()
+except RuntimeError:
+ print('RuntimeError')
diff --git a/tests/basics/gen_yield_from_throw.py b/tests/basics/gen_yield_from_throw.py
index 30960fb9c1..829bf0f3b4 100644
--- a/tests/basics/gen_yield_from_throw.py
+++ b/tests/basics/gen_yield_from_throw.py
@@ -17,3 +17,14 @@ try:
except TypeError:
print("got TypeError from downstream!")
+# case where generator doesn't intercept the thrown/injected exception
+def gen3():
+ yield 123
+ yield 456
+
+g3 = gen3()
+print(next(g3))
+try:
+ g3.throw(StopIteration)
+except StopIteration:
+ print('got StopIteration from downstream!')
diff --git a/tests/basics/gen_yield_from_throw2.py b/tests/basics/gen_yield_from_throw2.py
new file mode 100644
index 0000000000..2cff9e08ba
--- /dev/null
+++ b/tests/basics/gen_yield_from_throw2.py
@@ -0,0 +1,15 @@
+# uPy differs from CPython for this test
+# generator ignored GeneratorExit
+def gen():
+ try:
+ yield 123
+ except GeneratorExit:
+ print('GeneratorExit')
+ yield 456
+
+g = gen()
+print(next(g))
+try:
+ g.throw(GeneratorExit)
+except RuntimeError:
+ print('RuntimeError')
diff --git a/tests/basics/gen_yield_from_throw2.py.exp b/tests/basics/gen_yield_from_throw2.py.exp
new file mode 100644
index 0000000000..d5805b4947
--- /dev/null
+++ b/tests/basics/gen_yield_from_throw2.py.exp
@@ -0,0 +1,3 @@
+123
+GeneratorExit
+RuntimeError
diff --git a/tests/basics/generator1.py b/tests/basics/generator1.py
index d4028b0ce3..ea1d3b38bc 100644
--- a/tests/basics/generator1.py
+++ b/tests/basics/generator1.py
@@ -20,3 +20,6 @@ for val in f(3):
#print(gen.__next__())
#print(gen.__next__())
#print(gen.__next__())
+
+# test printing, but only the first chars that match CPython
+print(repr(f(0))[0:17])