From aaef34483c294c79dfc4e96f3168eea0f6b0e3f8 Mon Sep 17 00:00:00 2001 From: Ezio Melotti Date: Mon, 4 Mar 2013 15:17:56 +0200 Subject: [PATCH] #17346: make sure pickle tests are run against all protocols. Initial patch by Marius Gedminas. --- Lib/test/pickletester.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/Lib/test/pickletester.py b/Lib/test/pickletester.py index 4d491b0173c..1ce01c3d079 100644 --- a/Lib/test/pickletester.py +++ b/Lib/test/pickletester.py @@ -577,10 +577,10 @@ class AbstractPickleTests(unittest.TestCase): i = C() i.attr = i for proto in protocols: - s = self.dumps(i, 2) + s = self.dumps(i, proto) x = self.loads(s) self.assertEqual(dir(x), dir(i)) - self.assertTrue(x.attr is x) + self.assertIs(x.attr, x) def test_recursive_multi(self): l = [] @@ -637,13 +637,13 @@ class AbstractPickleTests(unittest.TestCase): def test_bytes(self): for proto in protocols: for s in b'', b'xyz', b'xyz'*100: - p = self.dumps(s) + p = self.dumps(s, proto) self.assertEqual(self.loads(p), s) for s in [bytes([i]) for i in range(256)]: - p = self.dumps(s) + p = self.dumps(s, proto) self.assertEqual(self.loads(p), s) for s in [bytes([i, i]) for i in range(256)]: - p = self.dumps(s) + p = self.dumps(s, proto) self.assertEqual(self.loads(p), s) def test_ints(self):