diff options
Diffstat (limited to 'tests/basics/generator-exc.py')
-rw-r--r-- | tests/basics/generator-exc.py | 20 |
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") |