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_itertools.py

diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index 5da4a6f..e5225f2 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -358,7 +358,8 @@
             c = count(value)
             self.assertEqual(next(copy.copy(c)), value)
             self.assertEqual(next(copy.deepcopy(c)), value)
-            self.assertEqual(next(pickle.loads(pickle.dumps(c))), value)
+            for proto in range(pickle.HIGHEST_PROTOCOL + 1):
+                self.assertEqual(next(pickle.loads(pickle.dumps(c, proto))), value)
 
     def test_count_with_stride(self):
         self.assertEqual(zip('abc',count(2,3)), [('a', 2), ('b', 5), ('c', 8)])