diff options
Diffstat (limited to 'Lib/test')
| -rw-r--r-- | Lib/test/test_enum.py | 8 | ||||
| -rw-r--r-- | Lib/test/test_socket.py | 5 |
2 files changed, 13 insertions, 0 deletions
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index dccaa4f..5db4040 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -581,6 +581,14 @@ class TestEnum(unittest.TestCase): test_pickle_dump_load(self.assertIs, self.NestedEnum.twigs, protocol=(4, HIGHEST_PROTOCOL)) + def test_pickle_by_name(self): + class ReplaceGlobalInt(IntEnum): + ONE = 1 + TWO = 2 + ReplaceGlobalInt.__reduce_ex__ = enum._reduce_ex_by_name + for proto in range(HIGHEST_PROTOCOL): + self.assertEqual(ReplaceGlobalInt.TWO.__reduce_ex__(proto), 'TWO') + def test_exploding_pickle(self): BadPickle = Enum( 'BadPickle', 'dill sweet bread-n-butter', module=__name__) diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index b412386..0db760f 100644 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -1375,6 +1375,11 @@ class GeneralModuleTests(unittest.TestCase): with sock: for protocol in range(pickle.HIGHEST_PROTOCOL + 1): self.assertRaises(TypeError, pickle.dumps, sock, protocol) + for protocol in range(pickle.HIGHEST_PROTOCOL + 1): + family = pickle.loads(pickle.dumps(socket.AF_INET, protocol)) + self.assertEqual(family, socket.AF_INET) + type = pickle.loads(pickle.dumps(socket.SOCK_STREAM, protocol)) + self.assertEqual(type, socket.SOCK_STREAM) def test_listen_backlog(self): for backlog in 0, -1: |