]> granicus.if.org Git - python/commitdiff
Fix the picklevalue test; it broke due to the change in default protocol.
authorGuido van Rossum <guido@python.org>
Tue, 15 May 2007 19:52:52 +0000 (19:52 +0000)
committerGuido van Rossum <guido@python.org>
Tue, 15 May 2007 19:52:52 +0000 (19:52 +0000)
Lib/test/test_bool.py

index 5dc71cc1b449e3878465929e92858ea5f239cbea..3788f08776d5e411b711281602f03b290e46b4e4 100644 (file)
@@ -299,19 +299,25 @@ class BoolTest(unittest.TestCase):
     def test_picklevalues(self):
         # Test for specific backwards-compatible pickle values
         import pickle
-        self.assertEqual(pickle.dumps(True), "I01\n.")
-        self.assertEqual(pickle.dumps(False), "I00\n.")
-        self.assertEqual(pickle.dumps(True, True), "I01\n.")
-        self.assertEqual(pickle.dumps(False, True), "I00\n.")
-
+        self.assertEqual(pickle.dumps(True, protocol=0), b"I01\n.")
+        self.assertEqual(pickle.dumps(False, protocol=0), b"I00\n.")
+        self.assertEqual(pickle.dumps(True, protocol=1), b"I01\n.")
+        self.assertEqual(pickle.dumps(False, protocol=1), b"I00\n.")
+        self.assertEqual(pickle.dumps(True, protocol=2), b'\x80\x02\x88.')
+        self.assertEqual(pickle.dumps(False, protocol=2), b'\x80\x02\x89.')
+
+    def test_cpicklevalues(self):
+        # Test for specific backwards-compatible pickle values
         try:
             import cPickle
         except ImportError:
             return # Just ignore the rest if cPickle doesn't exist
-        self.assertEqual(cPickle.dumps(True), "I01\n.")
-        self.assertEqual(cPickle.dumps(False), "I00\n.")
-        self.assertEqual(cPickle.dumps(True, True), "I01\n.")
-        self.assertEqual(cPickle.dumps(False, True), "I00\n.")
+        self.assertEqual(cPickle.dumps(True, protocol=0), b"I01\n.")
+        self.assertEqual(cPickle.dumps(False, protocol=0), b"I00\n.")
+        self.assertEqual(cPickle.dumps(True, protocol=1), b"I01\n.")
+        self.assertEqual(cPickle.dumps(False, protocol=1), b"I00\n.")
+        self.assertEqual(cPickle.dumps(True, protocol=2), b'\x80\x02\x88.')
+        self.assertEqual(cPickle.dumps(False, protocol=2), b'\x80\x02\x89.')
 
     def test_convert_to_bool(self):
         # Verify that TypeError occurs when bad things are returned