bpo-29587: Update gen.throw() to chain exceptions (#19823)

Before this commit, if an exception was active inside a generator
when calling gen.throw(), that exception was lost (i.e. there was
no implicit exception chaining).  This commit fixes that by
setting exc.__context__ when calling gen.throw(exc).
This commit is contained in:
Chris Jerdonek 2020-05-01 18:14:19 -07:00 committed by GitHub
parent f40bd466bf
commit 02047265eb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 27 additions and 0 deletions

View file

@ -316,6 +316,23 @@ class ExceptionTest(unittest.TestCase):
self.assertEqual(cm.exception.value.value, 2)
class GeneratorThrowTest(unittest.TestCase):
def test_exception_context_set(self):
def f():
try:
raise KeyError('a')
except Exception:
yield
gen = f()
gen.send(None)
with self.assertRaises(ValueError) as cm:
gen.throw(ValueError)
context = cm.exception.__context__
self.assertEqual((type(context), context.args), (KeyError, ('a',)))
class YieldFromTests(unittest.TestCase):
def test_generator_gi_yieldfrom(self):
def a():