[Python-checkins] bpo-44353: Improve tests covering typing.NewType pickling (GH-27302) (GH-27328)

[ambv]

https://github.com/python/cpython/commit/e8c01749c02a97a2f5b01086951bb564121c5113
commit: e8c01749c02a97a2f5b01086951bb564121c5113
branch: 3.10
author: Miss Islington (bot) <31488909+miss-islington at users.noreply.github.com>
committer: ambv <lukasz at langa.pl>
date: 2021-07-24T12:29:19+02:00
summary:

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

(cherry picked from commit a22b05da87bdfb081d6aaecfce136ae8dbb8680c)

Co-authored-by: Yurii Karabas <1998uriyyo at gmail.com>

files:
M Lib/test/test_typing.py

diff --git a/Lib/test/test_typing.py b/Lib/test/test_typing.py
index 5e3fba3a8687b..8cbbe622d5a49 100644
--- a/Lib/test/test_typing.py
+++ b/Lib/test/test_typing.py
@@ -3730,16 +3730,17 @@ def test_repr(self):
     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):