summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/generator-exc.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/basics/generator-exc.py')
-rw-r--r--tests/basics/generator-exc.py20
1 files changed, 20 insertions, 0 deletions
diff --git a/tests/basics/generator-exc.py b/tests/basics/generator-exc.py
index ebcd06e974..601fa2c00e 100644
--- a/tests/basics/generator-exc.py
+++ b/tests/basics/generator-exc.py
@@ -9,3 +9,23 @@ def gen():
for i in gen():
print(i)
+
+
+# Test throwing exceptions out of generator
+def gen2():
+ yield 1
+ raise ValueError
+ yield 2
+ yield 3
+
+g = gen2()
+print(next(g))
+try:
+ print(next(g))
+except ValueError:
+ print("ValueError")
+
+try:
+ print(next(g))
+except StopIteration:
+ print("StopIteration")