diff options
| author | Serhiy Storchaka <storchaka@gmail.com> | 2014-12-15 12:03:42 (GMT) |
|---|---|---|
| committer | Serhiy Storchaka <storchaka@gmail.com> | 2014-12-15 12:03:42 (GMT) |
| commit | bad1257c96943dcb39584a41e1a178542479df97 (patch) | |
| tree | dbb4594291c7df3660da701eb062bec35446bcba /Lib/test/test_array.py | |
| parent | 79b81738ef052e9406adb8e98f0d292928053c01 (diff) | |
| download | cpython-bad1257c96943dcb39584a41e1a178542479df97.zip cpython-bad1257c96943dcb39584a41e1a178542479df97.tar.gz cpython-bad1257c96943dcb39584a41e1a178542479df97.tar.bz2 | |
Issue #22777: Test pickling with all protocols.
Diffstat (limited to 'Lib/test/test_array.py')
| -rw-r--r-- | Lib/test/test_array.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/Lib/test/test_array.py b/Lib/test/test_array.py index f8dbf06..07c9bf9 100644 --- a/Lib/test/test_array.py +++ b/Lib/test/test_array.py @@ -285,17 +285,18 @@ class BaseTest: 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): + for proto in range(pickle.HIGHEST_PROTOCOL + 1): + orgit = iter(data) + d = pickle.dumps(orgit, proto) it = pickle.loads(d) - next(it) - d = pickle.dumps(it) - self.assertEqual(list(it), list(data)[1:]) + 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, proto) + self.assertEqual(list(it), list(data)[1:]) def test_insert(self): a = array.array(self.typecode, self.example) |