summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_itertools.py
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2014-12-15 12:02:43 (GMT)
committerSerhiy Storchaka <storchaka@gmail.com>2014-12-15 12:02:43 (GMT)
commit655720e2758a15c85effb340ff65f4b93e7cc7a7 (patch)
treed582731bc3e14a86222093e806d2ec88b2640f8f /Lib/test/test_itertools.py
parentc3741a067b3443cdac7739b21e0cd1bacf155b02 (diff)
downloadcpython-655720e2758a15c85effb340ff65f4b93e7cc7a7.zip
cpython-655720e2758a15c85effb340ff65f4b93e7cc7a7.tar.gz
cpython-655720e2758a15c85effb340ff65f4b93e7cc7a7.tar.bz2
Issue #22777: Test pickling with all protocols.
Diffstat (limited to 'Lib/test/test_itertools.py')
-rw-r--r--Lib/test/test_itertools.py3
1 files changed, 2 insertions, 1 deletions
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 @@ class TestBasicOps(unittest.TestCase):
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)])