summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_array.py
diff options
context:
space:
mode:
authorBrett Cannon <brett@python.org>2012-04-06 16:54:57 (GMT)
committerBrett Cannon <brett@python.org>2012-04-06 16:54:57 (GMT)
commit83c02ee8e8d62dbdd1576f2d02a7925b4381cc02 (patch)
tree3e79ba531c5dce1aa910fd033f7652e7c86270ca /Lib/test/test_array.py
parent8ac95ee62af237d21c695578cb19f54ddc1234a8 (diff)
parentb2e58185e5ca329b16177ad2095b35adaaafdd91 (diff)
downloadcpython-83c02ee8e8d62dbdd1576f2d02a7925b4381cc02.zip
cpython-83c02ee8e8d62dbdd1576f2d02a7925b4381cc02.tar.gz
cpython-83c02ee8e8d62dbdd1576f2d02a7925b4381cc02.tar.bz2
merge
Diffstat (limited to 'Lib/test/test_array.py')
-rwxr-xr-xLib/test/test_array.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/Lib/test/test_array.py b/Lib/test/test_array.py
index 434e495..9385339 100755
--- a/Lib/test/test_array.py
+++ b/Lib/test/test_array.py
@@ -285,6 +285,20 @@ class BaseTest(unittest.TestCase):
self.assertEqual(a.x, b.x)
self.assertEqual(type(a), type(b))
+ def test_iterator_pickle(self):
+ data = array.array(self.typecode, self.example)
+ orgit = iter(data)
+ d = pickle.dumps(orgit)
+ it = pickle.loads(d)
+ self.assertEqual(type(orgit), type(it))
+ self.assertEqual(list(it), list(data))
+
+ if len(data):
+ it = pickle.loads(d)
+ next(it)
+ d = pickle.dumps(it)
+ self.assertEqual(list(it), list(data)[1:])
+
def test_insert(self):
a = array.array(self.typecode, self.example)
a.insert(0, self.example[0])