summaryrefslogtreecommitdiffstats
path: root/Lib/test
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2009-11-30 19:44:40 (GMT)
committerRaymond Hettinger <python@rcn.com>2009-11-30 19:44:40 (GMT)
commite09f45a2e35e741e627f0ef49e94b84a28a4e72a (patch)
tree224550aea5ac0aeacda9af044d2b04f1c08bdcb2 /Lib/test
parentd46430bd8147da24d1e17b5f13064e70acb8ee01 (diff)
downloadcpython-e09f45a2e35e741e627f0ef49e94b84a28a4e72a.zip
cpython-e09f45a2e35e741e627f0ef49e94b84a28a4e72a.tar.gz
cpython-e09f45a2e35e741e627f0ef49e94b84a28a4e72a.tar.bz2
Issue 7410: deepcopy of itertools.count resets the count
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_itertools.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/Lib/test/test_itertools.py b/Lib/test/test_itertools.py
index f8bdb8b..03f1d85 100644
--- a/Lib/test/test_itertools.py
+++ b/Lib/test/test_itertools.py
@@ -7,6 +7,8 @@ from fractions import Fraction
import sys
import operator
import random
+import copy
+import pickle
maxsize = test_support.MAX_Py_ssize_t
minsize = -maxsize-1
@@ -346,6 +348,13 @@ class TestBasicOps(unittest.TestCase):
r2 = 'count(%r)'.__mod__(i).replace('L', '')
self.assertEqual(r1, r2)
+ # check copy, deepcopy, pickle
+ for value in -3, 3, sys.maxint-5, sys.maxint+5:
+ 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)
+
def test_count_with_stride(self):
self.assertEqual(zip('abc',count(2,3)), [('a', 2), ('b', 5), ('c', 8)])
self.assertEqual(zip('abc',count(start=2,step=3)),