From 93f5bac556460f869c6c1910ba6302e9b0cf8847 Mon Sep 17 00:00:00 2001 From: sobolevn Date: Mon, 9 Jan 2023 16:43:25 +0300 Subject: [PATCH] gh-100882: Improve `test_pickling` case in `test_ast.py` --- Lib/test/test_ast.py | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py index ab6a63faa59085..53a6418329e5d0 100644 --- a/Lib/test/test_ast.py +++ b/Lib/test/test_ast.py @@ -639,18 +639,11 @@ def test_no_fields(self): def test_pickling(self): import pickle - mods = [pickle] - try: - import cPickle - mods.append(cPickle) - except ImportError: - pass - protocols = [0, 1, 2] - for mod in mods: - for protocol in protocols: - for ast in (compile(i, "?", "exec", 0x400) for i in exec_tests): - ast2 = mod.loads(mod.dumps(ast, protocol)) - self.assertEqual(to_tuple(ast2), to_tuple(ast)) + + for protocol in range(pickle.HIGHEST_PROTOCOL + 1): + for ast in (compile(i, "?", "exec", 0x400) for i in exec_tests): + ast2 = pickle.loads(pickle.dumps(ast, protocol)) + self.assertEqual(to_tuple(ast2), to_tuple(ast)) def test_invalid_sum(self): pos = dict(lineno=2, col_offset=3)