gh-106300: Improve assertRaises(Exception) usages in tests (GH-106302)

This commit is contained in:
Nikita Sobolev 2023-07-07 23:42:40 +03:00 committed by GitHub
parent 80b9b3a517
commit 6e6a4cd523
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 20 additions and 12 deletions

View file

@ -2822,14 +2822,15 @@ class TransformCodecTest(unittest.TestCase):
def test_custom_zlib_error_is_noted(self):
# Check zlib codec gives a good error for malformed input
msg = "decoding with 'zlib_codec' codec failed"
with self.assertRaises(Exception) as failure:
with self.assertRaises(zlib.error) as failure:
codecs.decode(b"hello", "zlib_codec")
self.assertEqual(msg, failure.exception.__notes__[0])
def test_custom_hex_error_is_noted(self):
# Check hex codec gives a good error for malformed input
import binascii
msg = "decoding with 'hex_codec' codec failed"
with self.assertRaises(Exception) as failure:
with self.assertRaises(binascii.Error) as failure:
codecs.decode(b"hello", "hex_codec")
self.assertEqual(msg, failure.exception.__notes__[0])