Issue #4740: Use HIGHEST_PROTOCOL in pickle test. This enables test for protocol 3

(== HIGHEST_PROTOCOL in 3.x)
This commit is contained in:
Hirokazu Yamamoto 2008-12-27 04:21:44 +00:00
parent 81d90a220f
commit 801f9d3888
3 changed files with 6 additions and 6 deletions

View file

@ -7,7 +7,7 @@
from test import support from test import support
from weakref import proxy from weakref import proxy
import array, io, math import array, io, math
from pickle import loads, dumps from pickle import loads, dumps, HIGHEST_PROTOCOL
import operator import operator
class ArraySubclass(array.array): class ArraySubclass(array.array):
@ -98,7 +98,7 @@ def test_deepcopy(self):
self.assertEqual(a, b) self.assertEqual(a, b)
def test_pickle(self): def test_pickle(self):
for protocol in (0, 1, 2): for protocol in range(HIGHEST_PROTOCOL + 1):
a = array.array(self.typecode, self.example) a = array.array(self.typecode, self.example)
b = loads(dumps(a, protocol)) b = loads(dumps(a, protocol))
self.assertNotEqual(id(a), id(b)) self.assertNotEqual(id(a), id(b))
@ -113,7 +113,7 @@ def test_pickle(self):
self.assertEqual(type(a), type(b)) self.assertEqual(type(a), type(b))
def test_pickle_for_empty_array(self): 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) a = array.array(self.typecode)
b = loads(dumps(a, protocol)) b = loads(dumps(a, protocol))
self.assertNotEqual(id(a), id(b)) self.assertNotEqual(id(a), id(b))

View file

@ -375,7 +375,7 @@ def test_roundtrip_iter_init(self):
def test_pickle(self): def test_pickle(self):
d = deque(range(200)) d = deque(range(200))
for i in (0, 1, 2): for i in range(pickle.HIGHEST_PROTOCOL + 1):
s = pickle.dumps(d, i) s = pickle.dumps(d, i)
e = pickle.loads(s) e = pickle.loads(s)
self.assertNotEqual(id(d), id(e)) self.assertNotEqual(id(d), id(e))
@ -384,7 +384,7 @@ def test_pickle(self):
## def test_pickle_recursive(self): ## def test_pickle_recursive(self):
## d = deque('abc') ## d = deque('abc')
## d.append(d) ## d.append(d)
## for i in (0, 1, 2): ## for i in range(pickle.HIGHEST_PROTOCOL + 1):
## e = pickle.loads(pickle.dumps(d, i)) ## e = pickle.loads(pickle.dumps(d, i))
## self.assertNotEqual(id(d), id(e)) ## self.assertNotEqual(id(d), id(e))
## self.assertEqual(id(e), id(e[-1])) ## self.assertEqual(id(e), id(e[-1]))

View file

@ -219,7 +219,7 @@ def test_sub_and_super(self):
self.failIf(set('cbs').issuperset('a')) self.failIf(set('cbs').issuperset('a'))
def test_pickling(self): def test_pickling(self):
for i in (0, 1, 2): for i in range(pickle.HIGHEST_PROTOCOL + 1):
p = pickle.dumps(self.s, i) p = pickle.dumps(self.s, i)
dup = pickle.loads(p) dup = pickle.loads(p)
self.assertEqual(self.s, dup, "%s != %s" % (self.s, dup)) self.assertEqual(self.s, dup, "%s != %s" % (self.s, dup))