From 68115234bec9f66c107951d667cc3011685663d1 Mon Sep 17 00:00:00 2001 From: Bruno Cauet Date: Wed, 11 Mar 2015 12:23:18 +0100 Subject: [PATCH] Add tests for issue 23192 --- Lib/test/test_generators.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/Lib/test/test_generators.py b/Lib/test/test_generators.py index 9e61013..25b1c4d 100644 --- a/Lib/test/test_generators.py +++ b/Lib/test/test_generators.py @@ -88,6 +88,26 @@ class GeneratorTest(unittest.TestCase): self.assertEqual(gen.__qualname__, "GeneratorTest.test_name..") + def test_lambda_generator(self): + # Test that a lambda returning a generator behaves like the equivalent + # function + f = lambda: (yield 1) + def g(): return (yield 1) + + # test 'yield from' + f2 = lambda: (yield from g()) + def g2(): return (yield from g()) + + f3 = lambda: (yield from f()) + def g3(): return (yield from f()) + + for gen_fun in (f, g, f2, g2, f3, g3): + gen = gen_fun() + self.assertEqual(next(gen), 1) + with self.assertRaises(StopIteration) as exc_context: + gen.send(2) + self.assertEqual(str(exc_context.exception), '2') + class ExceptionTest(unittest.TestCase): # Tests for the issue #23353: check that the currently handled exception -- 2.3.1