2014-03-23 13:48:29 -06:00
|
|
|
def gen1():
|
|
|
|
yield 1
|
|
|
|
yield 2
|
|
|
|
|
|
|
|
# Test that it's possible to close just created gen
|
|
|
|
g = gen1()
|
|
|
|
print(g.close())
|
|
|
|
try:
|
|
|
|
next(g)
|
|
|
|
except StopIteration:
|
|
|
|
print("StopIteration")
|
|
|
|
|
|
|
|
# Test that it's possible to close gen in progress
|
|
|
|
g = gen1()
|
|
|
|
print(next(g))
|
|
|
|
print(g.close())
|
|
|
|
try:
|
|
|
|
next(g)
|
|
|
|
print("No StopIteration")
|
|
|
|
except StopIteration:
|
|
|
|
print("StopIteration")
|
|
|
|
|
|
|
|
# Test that it's possible to close finished gen
|
|
|
|
g = gen1()
|
|
|
|
print(list(g))
|
|
|
|
print(g.close())
|
|
|
|
try:
|
|
|
|
next(g)
|
|
|
|
print("No StopIteration")
|
|
|
|
except StopIteration:
|
|
|
|
print("StopIteration")
|
|
|
|
|
|
|
|
|
2018-09-13 08:44:06 -06:00
|
|
|
# Throwing GeneratorExit in response to close() is ok
|
2014-03-23 13:48:29 -06:00
|
|
|
def gen2():
|
|
|
|
try:
|
|
|
|
yield 1
|
|
|
|
yield 2
|
|
|
|
except:
|
2018-09-13 08:44:06 -06:00
|
|
|
print('raising GeneratorExit')
|
|
|
|
raise GeneratorExit
|
2014-03-23 13:48:29 -06:00
|
|
|
|
|
|
|
g = gen2()
|
|
|
|
next(g)
|
|
|
|
print(g.close())
|
|
|
|
|
|
|
|
# Any other exception is propagated to the .close() caller
|
|
|
|
def gen3():
|
|
|
|
try:
|
|
|
|
yield 1
|
|
|
|
yield 2
|
|
|
|
except:
|
|
|
|
raise ValueError
|
|
|
|
|
|
|
|
g = gen3()
|
|
|
|
next(g)
|
|
|
|
try:
|
|
|
|
print(g.close())
|
|
|
|
except ValueError:
|
|
|
|
print("ValueError")
|