summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_bytes.py
diff options
context:
space:
mode:
authorGeorg Brandl <georg@python.org>2008-12-28 11:44:14 (GMT)
committerGeorg Brandl <georg@python.org>2008-12-28 11:44:14 (GMT)
commitccc47b6eee267d6661c8521578e2fc9d53494029 (patch)
tree5bd70ee57571e66f907c68e57110430eefa8fda7 /Lib/test/test_bytes.py
parent15fafbe6f2c4e78037c9c4bd4e914ebbedc3b2bd (diff)
downloadcpython-ccc47b6eee267d6661c8521578e2fc9d53494029.zip
cpython-ccc47b6eee267d6661c8521578e2fc9d53494029.tar.gz
cpython-ccc47b6eee267d6661c8521578e2fc9d53494029.tar.bz2
#4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Also fix segfault for bytearray.translate(x, None) -- will backport this part to 3.0 and 2.6.
Diffstat (limited to 'Lib/test/test_bytes.py')
-rw-r--r--Lib/test/test_bytes.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/Lib/test/test_bytes.py b/Lib/test/test_bytes.py
index b1427b5..cb7fb46 100644
--- a/Lib/test/test_bytes.py
+++ b/Lib/test/test_bytes.py
@@ -888,11 +888,21 @@ class AssortedBytesTest(unittest.TestCase):
def test_translate(self):
b = b'hello'
+ ba = bytearray(b)
rosetta = bytearray(range(0, 256))
rosetta[ord('o')] = ord('e')
c = b.translate(rosetta, b'l')
self.assertEqual(b, b'hello')
self.assertEqual(c, b'hee')
+ c = ba.translate(rosetta, b'l')
+ self.assertEqual(ba, b'hello')
+ self.assertEqual(c, b'hee')
+ c = b.translate(None, b'e')
+ self.assertEqual(c, b'hllo')
+ c = ba.translate(None, b'e')
+ self.assertEqual(c, b'hllo')
+ self.assertRaises(TypeError, b.translate, None, None)
+ self.assertRaises(TypeError, ba.translate, None, None)
def test_split_bytearray(self):
self.assertEqual(b'a b'.split(memoryview(b' ')), [b'a', b'b'])