mirrors
/
cpython
mirror of https://github.com/python/cpython.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bpo-44353: Improve tests covering typing.NewType pickling (GH-27302)

This commit is contained in:
Yurii Karabas 2021-07-24 12:53:40 +03:00 committed by GitHub
parent 8072a1181d
commit a22b05da87
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Lib/test/test_typing.py
View File

@ -3755,16 +3755,17 @@ def test_repr(self):
def test_pickle(self): def test_pickle(self):
UserAge = self.module.NewType('UserAge', float) UserAge = self.module.NewType('UserAge', float)
for proto in range(pickle.HIGHEST_PROTOCOL + 1): for proto in range(pickle.HIGHEST_PROTOCOL + 1):
pickled = pickle.dumps(UserId, proto) with self.subTest(proto=proto):
loaded = pickle.loads(pickled) pickled = pickle.dumps(UserId, proto)
self.assertIs(loaded, UserId) loaded = pickle.loads(pickled)
self.assertIs(loaded, UserId)
pickled = pickle.dumps(self.UserName, proto) pickled = pickle.dumps(self.UserName, proto)
loaded = pickle.loads(pickled) loaded = pickle.loads(pickled)
self.assertIs(loaded, self.UserName) self.assertIs(loaded, self.UserName)
with self.assertRaises(pickle.PicklingError): with self.assertRaises(pickle.PicklingError):
pickle.dumps(UserAge, proto) pickle.dumps(UserAge, proto)
class NewTypePythonTests(NewTypeTests, BaseTestCase): class NewTypePythonTests(NewTypeTests, BaseTestCase):