summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/generator_exc.py
blob: 09aca5d9a0f4be9b60b0e3b0dc4985cd4824ee06 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# Test proper handling of exceptions within generator across yield
def gen():
    try:
        yield 1
        raise ValueError
    except ValueError:
        print("Caught")
    yield 2

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")


# Test throwing exception into generator
def gen3():
    yield 1
    try:
        yield 2
    except ValueError:
        print("ValueError received")
        yield 3
    yield 4
    yield 5

g = gen3()
print(next(g))
print(next(g))
print("out of throw:", g.throw(ValueError))
print(next(g))
try:
    print("out of throw2:", g.throw(ValueError))
except ValueError:
    print("Boomerang ValueError caught")