From: Nikita Sobolev <mail@sobolevn.me>
Date: Mon, 9 Jan 2023 19:20:25 +0000 (+0300)
Subject: gh-100882: Improve `test_pickling` case in `test_ast.py` (#100883)
X-Git-Tag: v3.12.0a4~10
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=2e80c2a976c13dcb69a654b386164dca362295a3;p=thirdparty%2FPython%2Fcpython.git

gh-100882: Improve `test_pickling` case in `test_ast.py` (#100883)
---

diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py
index ab6a63faa590..53a6418329e5 100644
--- a/Lib/test/test_ast.py
+++ b/Lib/test/test_ast.py
@@ -639,18 +639,11 @@ class AST_Tests(unittest.TestCase):
 
     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)