From 3c2b2acd8c21294d046f4d2ef9195a976a5ef999 Mon Sep 17 00:00:00 2001 From: Paul Sokolovsky Date: Wed, 26 Mar 2014 19:27:52 +0200 Subject: tests: Add testcases for yield from. --- tests/basics/gen-yield-from.py | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 tests/basics/gen-yield-from.py (limited to 'tests/basics/gen-yield-from.py') diff --git a/tests/basics/gen-yield-from.py b/tests/basics/gen-yield-from.py new file mode 100644 index 0000000000..5196b48d2b --- /dev/null +++ b/tests/basics/gen-yield-from.py @@ -0,0 +1,42 @@ +# Case of terminating subgen using return with value +def gen(): + yield 1 + yield 2 + return 3 + +def gen2(): + print("here1") + print((yield from gen())) + print("here2") + +g = gen2() +print(list(g)) + + +# Like above, but terminate subgen using StopIteration +def gen3(): + yield 1 + yield 2 + raise StopIteration + +def gen4(): + print("here1") + print((yield from gen3())) + print("here2") + +g = gen4() +print(list(g)) + +# Like above, but terminate subgen using StopIteration with value +def gen5(): + yield 1 + yield 2 + raise StopIteration(123) + +def gen6(): + print("here1") + print((yield from gen5())) + print("here2") + +g = gen6() +print(list(g)) -- cgit v1.2.3