[pypy-commit] pypy py3.3: Additional checks related to pickling of zip iterators.

[kvas]

Author: Vasily Kuznetsov <kvas.it at gmail.com>
Branch: py3.3
Changeset: r72536:d26183fe9a2b
Date: 2014-07-26 16:43 +0200
http://bitbucket.org/pypy/pypy/changeset/d26183fe9a2b/

Log:	Additional checks related to pickling of zip iterators.

diff --git a/pypy/module/__builtin__/test/test_zip_pickle.py b/pypy/module/__builtin__/test/test_zip_pickle.py
--- a/pypy/module/__builtin__/test/test_zip_pickle.py
+++ b/pypy/module/__builtin__/test/test_zip_pickle.py
@@ -3,12 +3,16 @@
     def test_zip_pickle(self):
         import pickle
 
-        def pickle_unpickle(obj):
-            d = pickle.dumps(obj)
-            return pickle.loads(d)
+        def check_pickle_unpickle(itr):
+            d = pickle.dumps(itr)
+            itr_ = pickle.loads(d)
+            lst, lst_ = list(itr), list(itr_)
+            assert lst == lst_
 
-        z1 = zip([1, 2, 3], [4, 5, 6])
-        z1_ = pickle_unpickle(z1)
-        l1, l1_ = list(z1), list(z1_)
+        check_pickle_unpickle(zip([1, 2, 3], [4, 5, 6]))
+        check_pickle_unpickle(zip())
 
-        assert l1 == l1_
+        a = iter("abc")
+        b = iter("cdef")
+        next(b)
+        check_pickle_unpickle(zip(a, b))