Include the highest pickle protocol in a couple of tests. (GH-10735)

test_reduce_ex() in test_array.py and
test_reversevaluesiterator_pickling() in test_dict.py weren't using
the highest pickle protocol.
This commit is contained in:
Zackery Spytz 2018-11-26 22:40:49 -07:00 committed by Serhiy Storchaka
parent da324d53d4
commit d1cbc6f8a0
2 changed files with 2 additions and 2 deletions

View File

@ -248,7 +248,7 @@ class BaseTest:
a = array.array(self.typecode, self.example) a = array.array(self.typecode, self.example)
for protocol in range(3): for protocol in range(3):
self.assertIs(a.__reduce_ex__(protocol)[0], array.array) self.assertIs(a.__reduce_ex__(protocol)[0], array.array)
for protocol in range(3, pickle.HIGHEST_PROTOCOL): for protocol in range(3, pickle.HIGHEST_PROTOCOL + 1):
self.assertIs(a.__reduce_ex__(protocol)[0], array_reconstructor) self.assertIs(a.__reduce_ex__(protocol)[0], array_reconstructor)
def test_pickle(self): def test_pickle(self):

View File

@ -1112,7 +1112,7 @@ class DictTest(unittest.TestCase):
self.assertEqual(dict(it), data) self.assertEqual(dict(it), data)
def test_reversevaluesiterator_pickling(self): def test_reversevaluesiterator_pickling(self):
for proto in range(pickle.HIGHEST_PROTOCOL): for proto in range(pickle.HIGHEST_PROTOCOL + 1):
data = {1:"a", 2:"b", 3:"c"} data = {1:"a", 2:"b", 3:"c"}
# data.values() isn't picklable, only its iterator # data.values() isn't picklable, only its iterator
it = reversed(data.values()) it = reversed(data.values())