commit 655720e2758a15c85effb340ff65f4b93e7cc7a7
author Serhiy Storchaka <storchaka@gmail.com> Mon Dec 15 14:02:43 2014 +0200
committer Serhiy Storchaka <storchaka@gmail.com> Mon Dec 15 14:02:43 2014 +0200
tree d582731bc3e14a86222093e806d2ec88b2640f8f
parent c3741a067b3443cdac7739b21e0cd1bacf155b02

Issue #22777: Test pickling with all protocols.

Lib/test/test_set.py

diff --git a/Lib/test/test_set.py b/Lib/test/test_set.py
index 610be7c..dd65202 100644
--- a/Lib/test/test_set.py
+++ b/Lib/test/test_set.py
@@ -233,7 +233,7 @@
             self.assertEqual(self.s, dup, "%s != %s" % (self.s, dup))
             if type(self.s) not in (set, frozenset):
                 self.s.x = 10
-                p = pickle.dumps(self.s)
+                p = pickle.dumps(self.s, i)
                 dup = pickle.loads(p)
                 self.assertEqual(self.s.x, dup.x)
 
@@ -786,10 +786,11 @@
         self.assertEqual(setiter.__length_hint__(), len(self.set))
 
     def test_pickling(self):
-        p = pickle.dumps(self.set)
-        copy = pickle.loads(p)
-        self.assertEqual(self.set, copy,
-                         "%s != %s" % (self.set, copy))
+        for proto in range(pickle.HIGHEST_PROTOCOL + 1):
+            p = pickle.dumps(self.set, proto)
+            copy = pickle.loads(p)
+            self.assertEqual(self.set, copy,
+                             "%s != %s" % (self.set, copy))
 
 #------------------------------------------------------------------------------