mirror of
https://github.com/python/cpython.git
synced 2025-10-09 08:31:26 +00:00
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:
parent
f40bd466bf
commit
02047265eb
3 changed files with 27 additions and 0 deletions
|
@ -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():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue