diff --git a/Lib/test/test_urllib.py b/Lib/test/test_urllib.py index 23492fb3bb7..c68fe54b939 100644 --- a/Lib/test/test_urllib.py +++ b/Lib/test/test_urllib.py @@ -555,8 +555,8 @@ def test_unquoting(self): self.assertEqual(result.count('%'), 1, "using unquote(): not all characters escaped: " "%s" % result) - self.assertRaises(TypeError, urllib.parse.unquote, None) - self.assertRaises(TypeError, urllib.parse.unquote, ()) + self.assertRaises((TypeError, AttributeError), urllib.parse.unquote, None) + self.assertRaises((TypeError, AttributeError), urllib.parse.unquote, ()) def test_unquoting_badpercent(self): # Test unquoting on bad percent-escapes @@ -591,8 +591,8 @@ def test_unquoting_badpercent(self): result = urllib.parse.unquote_to_bytes(given) self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r" % (expect, result)) - self.assertRaises(TypeError, urllib.parse.unquote_to_bytes, None) - self.assertRaises(TypeError, urllib.parse.unquote_to_bytes, ()) + self.assertRaises((TypeError, AttributeError), urllib.parse.unquote_to_bytes, None) + self.assertRaises((TypeError, AttributeError), urllib.parse.unquote_to_bytes, ()) def test_unquoting_mixed_case(self): # Test unquoting on mixed-case hex digits in the percent-escapes