diff options
| -rwxr-xr-x | Lib/test/test_array.py | 6 | ||||
| -rw-r--r-- | Lib/test/test_deque.py | 4 | ||||
| -rw-r--r-- | Lib/test/test_set.py | 2 |
3 files changed, 6 insertions, 6 deletions
diff --git a/Lib/test/test_array.py b/Lib/test/test_array.py index 5a505de7ec..2b80a7d041 100755 --- a/Lib/test/test_array.py +++ b/Lib/test/test_array.py @@ -7,7 +7,7 @@ import unittest from test import support from weakref import proxy import array, io, math -from pickle import loads, dumps +from pickle import loads, dumps, HIGHEST_PROTOCOL import operator class ArraySubclass(array.array): @@ -98,7 +98,7 @@ class BaseTest(unittest.TestCase): self.assertEqual(a, b) def test_pickle(self): - for protocol in (0, 1, 2): + for protocol in range(HIGHEST_PROTOCOL + 1): a = array.array(self.typecode, self.example) b = loads(dumps(a, protocol)) self.assertNotEqual(id(a), id(b)) @@ -113,7 +113,7 @@ class BaseTest(unittest.TestCase): self.assertEqual(type(a), type(b)) def test_pickle_for_empty_array(self): - for protocol in (0, 1, 2): + for protocol in range(HIGHEST_PROTOCOL + 1): a = array.array(self.typecode) b = loads(dumps(a, protocol)) self.assertNotEqual(id(a), id(b)) diff --git a/Lib/test/test_deque.py b/Lib/test/test_deque.py index 1ac43dad27..7af0803e5d 100644 --- a/Lib/test/test_deque.py +++ b/Lib/test/test_deque.py @@ -375,7 +375,7 @@ class TestBasic(unittest.TestCase): def test_pickle(self): d = deque(range(200)) - for i in (0, 1, 2): + for i in range(pickle.HIGHEST_PROTOCOL + 1): s = pickle.dumps(d, i) e = pickle.loads(s) self.assertNotEqual(id(d), id(e)) @@ -384,7 +384,7 @@ class TestBasic(unittest.TestCase): ## def test_pickle_recursive(self): ## d = deque('abc') ## d.append(d) -## for i in (0, 1, 2): +## for i in range(pickle.HIGHEST_PROTOCOL + 1): ## e = pickle.loads(pickle.dumps(d, i)) ## self.assertNotEqual(id(d), id(e)) ## self.assertEqual(id(e), id(e[-1])) diff --git a/Lib/test/test_set.py b/Lib/test/test_set.py index 07319fc20d..da3ed0f53a 100644 --- a/Lib/test/test_set.py +++ b/Lib/test/test_set.py @@ -219,7 +219,7 @@ class TestJointOps(unittest.TestCase): self.failIf(set('cbs').issuperset('a')) def test_pickling(self): - for i in (0, 1, 2): + for i in range(pickle.HIGHEST_PROTOCOL + 1): p = pickle.dumps(self.s, i) dup = pickle.loads(p) self.assertEqual(self.s, dup, "%s != %s" % (self.s, dup)) |