diff options
author | Noah <33094578+coolreader18@users.noreply.github.com> | 2021-08-02 21:17:18 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-08-02 19:17:18 -0700 |
commit | 83ca46b7784b7357d82ec47b33295e09ed7380cb (patch) | |
tree | 30c63b0b1f0a26e65dbe0954ad7065bdd795c566 /Lib/test/test_generators.py | |
parent | 54f185b6d321a6354aef2b2886c766677f487ecb (diff) | |
download | cpython-git-83ca46b7784b7357d82ec47b33295e09ed7380cb.tar.gz |
closes bpo-39091: Fix segfault when Exception constructor returns non-exception for gen.throw. (#17658)
Co-authored-by: Benjamin Peterson <benjamin@python.org>
Diffstat (limited to 'Lib/test/test_generators.py')
-rw-r--r-- | Lib/test/test_generators.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/Lib/test/test_generators.py b/Lib/test/test_generators.py index ebf8bb7e32..53d579e723 100644 --- a/Lib/test/test_generators.py +++ b/Lib/test/test_generators.py @@ -270,6 +270,32 @@ class ExceptionTest(unittest.TestCase): self.assertEqual(next(g), "done") self.assertEqual(sys.exc_info(), (None, None, None)) + def test_except_throw_bad_exception(self): + class E(Exception): + def __new__(cls, *args, **kwargs): + return cls + + def boring_generator(): + yield + + gen = boring_generator() + + err_msg = 'should have returned an instance of BaseException' + + with self.assertRaisesRegex(TypeError, err_msg): + gen.throw(E) + + self.assertRaises(StopIteration, next, gen) + + def generator(): + with self.assertRaisesRegex(TypeError, err_msg): + yield + + gen = generator() + next(gen) + with self.assertRaises(StopIteration): + gen.throw(E) + def test_stopiteration_error(self): # See also PEP 479. |