commit 31668b8f7a3efc7b17511bb08525b28e8ff5f23a
author Kristján Valur Jónsson <kristjan@ccpgames.com> Tue Apr 03 10:49:41 2012 +0000
committer Kristján Valur Jónsson <kristjan@ccpgames.com> Tue Apr 03 10:49:41 2012 +0000
tree acc8d778b0135e0f345e94c3334d7c1eeb1dca0a
parent 283b96b6bd974179b7f7f93f1c4631c3cc20c96d

Issue #14288: Serialization support for builtin iterators.

Lib/test/test_bytes.py

diff --git a/Lib/test/test_bytes.py b/Lib/test/test_bytes.py
index 203fc5c..8ce6c22 100644
--- a/Lib/test/test_bytes.py
+++ b/Lib/test/test_bytes.py
@@ -518,6 +518,24 @@
                 q = pickle.loads(ps)
                 self.assertEqual(b, q)
 
+    def test_iterator_pickling(self):
+        for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
+            it = itorg = iter(self.type2test(b))
+            data = list(self.type2test(b))
+            d = pickle.dumps(it)
+            it = pickle.loads(d)
+            self.assertEqual(type(itorg), type(it))
+            self.assertEqual(list(it), data)
+
+            it = pickle.loads(d)
+            try:
+                next(it)
+            except StopIteration:
+                continue
+            d = pickle.dumps(it)
+            it = pickle.loads(d)
+            self.assertEqual(list(it), data[1:])
+
     def test_strip(self):
         b = self.type2test(b'mississippi')
         self.assertEqual(b.strip(b'i'), b'mississipp')