64 lines
1.3 KiB
Python
64 lines
1.3 KiB
Python
|
# test StopIteration interaction with generators
|
||
|
|
|
||
|
|
try:
|
||
|
|
enumerate, exec
|
||
|
|
except:
|
||
|
|
print("SKIP")
|
||
|
|
raise SystemExit
|
||
|
|
|
||
|
|
|
||
|
|
def get_stop_iter_arg(msg, code):
|
||
|
|
try:
|
||
|
|
exec(code)
|
||
|
|
print("FAIL")
|
||
|
|
except StopIteration as er:
|
||
|
|
print(msg, er.args)
|
||
|
|
|
||
|
|
|
||
|
|
class A:
|
||
|
|
def __iter__(self):
|
||
|
|
return self
|
||
|
|
|
||
|
|
def __next__(self):
|
||
|
|
raise StopIteration(42)
|
||
|
|
|
||
|
|
|
||
|
|
class B:
|
||
|
|
def __getitem__(self, index):
|
||
|
|
# argument to StopIteration should get ignored
|
||
|
|
raise StopIteration(42)
|
||
|
|
|
||
|
|
|
||
|
|
def gen(x):
|
||
|
|
return x
|
||
|
|
yield
|
||
|
|
|
||
|
|
|
||
|
|
def gen2(x):
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
return x
|
||
|
|
|
||
|
|
|
||
|
|
get_stop_iter_arg("next", "next(A())")
|
||
|
|
get_stop_iter_arg("iter", "next(iter(B()))")
|
||
|
|
get_stop_iter_arg("enumerate", "next(enumerate(A()))")
|
||
|
|
get_stop_iter_arg("map", "next(map(lambda x:x, A()))")
|
||
|
|
get_stop_iter_arg("zip", "next(zip(A()))")
|
||
|
|
g = gen(None)
|
||
|
|
get_stop_iter_arg("generator0", "next(g)")
|
||
|
|
get_stop_iter_arg("generator1", "next(g)")
|
||
|
|
g = gen(42)
|
||
|
|
get_stop_iter_arg("generator0", "next(g)")
|
||
|
|
get_stop_iter_arg("generator1", "next(g)")
|
||
|
|
get_stop_iter_arg("send", "gen(None).send(None)")
|
||
|
|
get_stop_iter_arg("send", "gen(42).send(None)")
|
||
|
|
g = gen2(None)
|
||
|
|
next(g)
|
||
|
|
get_stop_iter_arg("throw", "g.throw(ValueError)")
|
||
|
|
g = gen2(42)
|
||
|
|
next(g)
|
||
|
|
get_stop_iter_arg("throw", "g.throw(ValueError)")
|