diff options
| -rw-r--r-- | Lib/test/test_typing.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py index 5e3fba3..8cbbe62 100644 --- a/Lib/test/test_typing.py +++ b/Lib/test/test_typing.py @@ -3730,16 +3730,17 @@ class NewTypeTests(BaseTestCase): def test_pickle(self): UserAge = NewType('UserAge', float) for proto in range(pickle.HIGHEST_PROTOCOL + 1): - pickled = pickle.dumps(UserId, proto) - loaded = pickle.loads(pickled) - self.assertIs(loaded, UserId) + with self.subTest(proto=proto): + pickled = pickle.dumps(UserId, proto) + loaded = pickle.loads(pickled) + self.assertIs(loaded, UserId) - pickled = pickle.dumps(self.UserName, proto) - loaded = pickle.loads(pickled) - self.assertIs(loaded, self.UserName) + pickled = pickle.dumps(self.UserName, proto) + loaded = pickle.loads(pickled) + self.assertIs(loaded, self.UserName) - with self.assertRaises(pickle.PicklingError): - pickle.dumps(UserAge, proto) + with self.assertRaises(pickle.PicklingError): + pickle.dumps(UserAge, proto) class NamedTupleTests(BaseTestCase): |