blob: 3c091411849114c85af2c628f014bd249ab4ea06 [file] [log] [blame]
Guido van Rossum254348e2007-11-21 19:29:53 +00001"""Unit tests for the bytes and bytearray types.
Guido van Rossum98297ee2007-11-06 21:34:58 +00002
3XXX This is a mess. Common tests should be moved to buffer_tests.py,
4which itself ought to be unified with string_tests.py (and the latter
5should be modernized).
6"""
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00007
Guido van Rossumd624f182006-04-24 13:47:05 +00008import os
9import re
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000010import sys
Guido van Rossuma6c04be2007-11-03 00:24:24 +000011import copy
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000012import functools
Guido van Rossum0ad08122007-04-11 04:37:43 +000013import pickle
Guido van Rossumd624f182006-04-24 13:47:05 +000014import tempfile
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000015import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +000016import test.support
Georg Brandlc7885542007-03-06 19:16:20 +000017import test.string_tests
Gregory P. Smith60d241f2007-10-16 06:31:30 +000018import test.buffer_tests
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000019
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000020
21if sys.flags.bytes_warning:
22 def check_bytes_warnings(func):
23 @functools.wraps(func)
24 def wrapper(*args, **kw):
25 with test.support.check_warnings(('', BytesWarning)):
26 return func(*args, **kw)
27 return wrapper
28else:
29 # no-op
30 def check_bytes_warnings(func):
31 return func
32
33
Georg Brandl9a54d7c2008-07-16 23:15:30 +000034class Indexable:
35 def __init__(self, value=0):
36 self.value = value
37 def __index__(self):
38 return self.value
39
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000040
Ezio Melotti0dceb562013-01-10 07:43:26 +020041class BaseBytesTest:
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000042
43 def test_basics(self):
Christian Heimes510711d2008-01-30 11:57:58 +000044 b = self.type2test()
45 self.assertEqual(type(b), self.type2test)
46 self.assertEqual(b.__class__, self.type2test)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000047
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +000048 def test_copy(self):
49 a = self.type2test(b"abcd")
50 for copy_method in (copy.copy, copy.deepcopy):
51 b = copy_method(a)
52 self.assertEqual(a, b)
53 self.assertEqual(type(a), type(b))
54
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000055 def test_empty_sequence(self):
Christian Heimes510711d2008-01-30 11:57:58 +000056 b = self.type2test()
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000057 self.assertEqual(len(b), 0)
58 self.assertRaises(IndexError, lambda: b[0])
59 self.assertRaises(IndexError, lambda: b[1])
Christian Heimesa37d4c62007-12-04 23:02:19 +000060 self.assertRaises(IndexError, lambda: b[sys.maxsize])
61 self.assertRaises(IndexError, lambda: b[sys.maxsize+1])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000062 self.assertRaises(IndexError, lambda: b[10**100])
63 self.assertRaises(IndexError, lambda: b[-1])
64 self.assertRaises(IndexError, lambda: b[-2])
Christian Heimesa37d4c62007-12-04 23:02:19 +000065 self.assertRaises(IndexError, lambda: b[-sys.maxsize])
66 self.assertRaises(IndexError, lambda: b[-sys.maxsize-1])
67 self.assertRaises(IndexError, lambda: b[-sys.maxsize-2])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000068 self.assertRaises(IndexError, lambda: b[-10**100])
69
70 def test_from_list(self):
71 ints = list(range(256))
Christian Heimes510711d2008-01-30 11:57:58 +000072 b = self.type2test(i for i in ints)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000073 self.assertEqual(len(b), 256)
74 self.assertEqual(list(b), ints)
75
76 def test_from_index(self):
Georg Brandl9a54d7c2008-07-16 23:15:30 +000077 b = self.type2test([Indexable(), Indexable(1), Indexable(254),
78 Indexable(255)])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000079 self.assertEqual(list(b), [0, 1, 254, 255])
Benjamin Peterson4c045832010-04-16 22:35:32 +000080 self.assertRaises(ValueError, self.type2test, [Indexable(-1)])
81 self.assertRaises(ValueError, self.type2test, [Indexable(256)])
Guido van Rossum98297ee2007-11-06 21:34:58 +000082
83 def test_from_ssize(self):
Benjamin Peterson4c045832010-04-16 22:35:32 +000084 self.assertEqual(self.type2test(0), b'')
85 self.assertEqual(self.type2test(1), b'\x00')
86 self.assertEqual(self.type2test(5), b'\x00\x00\x00\x00\x00')
87 self.assertRaises(ValueError, self.type2test, -1)
Guido van Rossum98297ee2007-11-06 21:34:58 +000088
Benjamin Peterson4c045832010-04-16 22:35:32 +000089 self.assertEqual(self.type2test('0', 'ascii'), b'0')
90 self.assertEqual(self.type2test(b'0'), b'0')
Benjamin Peterson8380dd52010-04-16 22:51:37 +000091 self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000092
93 def test_constructor_type_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +000094 self.assertRaises(TypeError, self.type2test, 0.0)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000095 class C:
96 pass
Christian Heimes510711d2008-01-30 11:57:58 +000097 self.assertRaises(TypeError, self.type2test, ["0"])
98 self.assertRaises(TypeError, self.type2test, [0.0])
99 self.assertRaises(TypeError, self.type2test, [None])
100 self.assertRaises(TypeError, self.type2test, [C()])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000101
102 def test_constructor_value_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000103 self.assertRaises(ValueError, self.type2test, [-1])
104 self.assertRaises(ValueError, self.type2test, [-sys.maxsize])
105 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-1])
106 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-2])
107 self.assertRaises(ValueError, self.type2test, [-10**100])
108 self.assertRaises(ValueError, self.type2test, [256])
109 self.assertRaises(ValueError, self.type2test, [257])
110 self.assertRaises(ValueError, self.type2test, [sys.maxsize])
111 self.assertRaises(ValueError, self.type2test, [sys.maxsize+1])
112 self.assertRaises(ValueError, self.type2test, [10**100])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000113
114 def test_compare(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000115 b1 = self.type2test([1, 2, 3])
116 b2 = self.type2test([1, 2, 3])
117 b3 = self.type2test([1, 3])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000118
Guido van Rossume61fd5b2007-07-11 12:20:59 +0000119 self.assertEqual(b1, b2)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000120 self.assertTrue(b2 != b3)
121 self.assertTrue(b1 <= b2)
122 self.assertTrue(b1 <= b3)
123 self.assertTrue(b1 < b3)
124 self.assertTrue(b1 >= b2)
125 self.assertTrue(b3 >= b2)
126 self.assertTrue(b3 > b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000127
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000128 self.assertFalse(b1 != b2)
129 self.assertFalse(b2 == b3)
130 self.assertFalse(b1 > b2)
131 self.assertFalse(b1 > b3)
132 self.assertFalse(b1 >= b3)
133 self.assertFalse(b1 < b2)
134 self.assertFalse(b3 < b2)
135 self.assertFalse(b3 <= b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000136
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000137 @check_bytes_warnings
Guido van Rossum343e97f2007-04-09 00:43:24 +0000138 def test_compare_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000139 # Byte comparisons with unicode should always fail!
140 # Test this for all expected byte orders and Unicode character
141 # sizes.
142 self.assertEqual(self.type2test(b"\0a\0b\0c") == "abc", False)
143 self.assertEqual(self.type2test(b"\0\0\0a\0\0\0b\0\0\0c") == "abc",
144 False)
145 self.assertEqual(self.type2test(b"a\0b\0c\0") == "abc", False)
146 self.assertEqual(self.type2test(b"a\0\0\0b\0\0\0c\0\0\0") == "abc",
147 False)
148 self.assertEqual(self.type2test() == str(), False)
149 self.assertEqual(self.type2test() != str(), True)
Christian Heimes510711d2008-01-30 11:57:58 +0000150
151 def test_reversed(self):
152 input = list(map(ord, "Hello"))
153 b = self.type2test(input)
154 output = list(reversed(b))
155 input.reverse()
156 self.assertEqual(output, input)
157
158 def test_getslice(self):
159 def by(s):
160 return self.type2test(map(ord, s))
161 b = by("Hello, world")
162
163 self.assertEqual(b[:5], by("Hello"))
164 self.assertEqual(b[1:5], by("ello"))
165 self.assertEqual(b[5:7], by(", "))
166 self.assertEqual(b[7:], by("world"))
167 self.assertEqual(b[7:12], by("world"))
168 self.assertEqual(b[7:100], by("world"))
169
170 self.assertEqual(b[:-7], by("Hello"))
171 self.assertEqual(b[-11:-7], by("ello"))
172 self.assertEqual(b[-7:-5], by(", "))
173 self.assertEqual(b[-5:], by("world"))
174 self.assertEqual(b[-5:12], by("world"))
175 self.assertEqual(b[-5:100], by("world"))
176 self.assertEqual(b[-100:5], by("Hello"))
177
178 def test_extended_getslice(self):
179 # Test extended slicing by comparing with list slicing.
180 L = list(range(255))
181 b = self.type2test(L)
182 indices = (0, None, 1, 3, 19, 100, -1, -2, -31, -100)
183 for start in indices:
184 for stop in indices:
185 # Skip step 0 (invalid)
186 for step in indices[1:]:
187 self.assertEqual(b[start:stop:step], self.type2test(L[start:stop:step]))
188
189 def test_encoding(self):
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000190 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000191 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000192 b = self.type2test(sample, enc)
193 self.assertEqual(b, self.type2test(sample.encode(enc)))
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000194 self.assertRaises(UnicodeEncodeError, self.type2test, sample, "latin-1")
195 b = self.type2test(sample, "latin-1", "ignore")
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000196 self.assertEqual(b, self.type2test(sample[:-3], "utf-8"))
Christian Heimes510711d2008-01-30 11:57:58 +0000197
198 def test_decode(self):
199 sample = "Hello world\n\u1234\u5678\u9abc\def0\def0"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000200 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000201 b = self.type2test(sample, enc)
202 self.assertEqual(b.decode(enc), sample)
203 sample = "Hello world\n\x80\x81\xfe\xff"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000204 b = self.type2test(sample, "latin-1")
205 self.assertRaises(UnicodeDecodeError, b.decode, "utf-8")
206 self.assertEqual(b.decode("utf-8", "ignore"), "Hello world\n")
207 self.assertEqual(b.decode(errors="ignore", encoding="utf-8"),
Benjamin Peterson308d6372009-09-18 21:42:35 +0000208 "Hello world\n")
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000209 # Default encoding is utf-8
210 self.assertEqual(self.type2test(b'\xe2\x98\x83').decode(), '\u2603')
Christian Heimes510711d2008-01-30 11:57:58 +0000211
212 def test_from_int(self):
213 b = self.type2test(0)
214 self.assertEqual(b, self.type2test())
215 b = self.type2test(10)
216 self.assertEqual(b, self.type2test([0]*10))
217 b = self.type2test(10000)
218 self.assertEqual(b, self.type2test([0]*10000))
219
220 def test_concat(self):
221 b1 = self.type2test(b"abc")
222 b2 = self.type2test(b"def")
223 self.assertEqual(b1 + b2, b"abcdef")
224 self.assertEqual(b1 + bytes(b"def"), b"abcdef")
225 self.assertEqual(bytes(b"def") + b1, b"defabc")
226 self.assertRaises(TypeError, lambda: b1 + "def")
227 self.assertRaises(TypeError, lambda: "abc" + b2)
228
229 def test_repeat(self):
230 for b in b"abc", self.type2test(b"abc"):
231 self.assertEqual(b * 3, b"abcabcabc")
232 self.assertEqual(b * 0, b"")
233 self.assertEqual(b * -1, b"")
234 self.assertRaises(TypeError, lambda: b * 3.14)
235 self.assertRaises(TypeError, lambda: 3.14 * b)
236 # XXX Shouldn't bytes and bytearray agree on what to raise?
Mark Dickinsoncf940c72010-08-10 18:35:01 +0000237 with self.assertRaises((OverflowError, MemoryError)):
238 c = b * sys.maxsize
239 with self.assertRaises((OverflowError, MemoryError)):
240 b *= sys.maxsize
Christian Heimes510711d2008-01-30 11:57:58 +0000241
242 def test_repeat_1char(self):
243 self.assertEqual(self.type2test(b'x')*100, self.type2test([ord('x')]*100))
244
245 def test_contains(self):
246 b = self.type2test(b"abc")
Benjamin Peterson577473f2010-01-19 00:09:57 +0000247 self.assertIn(ord('a'), b)
248 self.assertIn(int(ord('a')), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000249 self.assertNotIn(200, b)
Christian Heimes510711d2008-01-30 11:57:58 +0000250 self.assertRaises(ValueError, lambda: 300 in b)
251 self.assertRaises(ValueError, lambda: -1 in b)
252 self.assertRaises(TypeError, lambda: None in b)
253 self.assertRaises(TypeError, lambda: float(ord('a')) in b)
254 self.assertRaises(TypeError, lambda: "a" in b)
255 for f in bytes, bytearray:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000256 self.assertIn(f(b""), b)
257 self.assertIn(f(b"a"), b)
258 self.assertIn(f(b"b"), b)
259 self.assertIn(f(b"c"), b)
260 self.assertIn(f(b"ab"), b)
261 self.assertIn(f(b"bc"), b)
262 self.assertIn(f(b"abc"), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000263 self.assertNotIn(f(b"ac"), b)
264 self.assertNotIn(f(b"d"), b)
265 self.assertNotIn(f(b"dab"), b)
266 self.assertNotIn(f(b"abd"), b)
Christian Heimes510711d2008-01-30 11:57:58 +0000267
268 def test_fromhex(self):
269 self.assertRaises(TypeError, self.type2test.fromhex)
270 self.assertRaises(TypeError, self.type2test.fromhex, 1)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000271 self.assertEqual(self.type2test.fromhex(''), self.type2test())
Christian Heimes510711d2008-01-30 11:57:58 +0000272 b = bytearray([0x1a, 0x2b, 0x30])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000273 self.assertEqual(self.type2test.fromhex('1a2B30'), b)
274 self.assertEqual(self.type2test.fromhex(' 1A 2B 30 '), b)
275 self.assertEqual(self.type2test.fromhex('0000'), b'\0\0')
Christian Heimes510711d2008-01-30 11:57:58 +0000276 self.assertRaises(TypeError, self.type2test.fromhex, b'1B')
277 self.assertRaises(ValueError, self.type2test.fromhex, 'a')
278 self.assertRaises(ValueError, self.type2test.fromhex, 'rt')
279 self.assertRaises(ValueError, self.type2test.fromhex, '1a b cd')
280 self.assertRaises(ValueError, self.type2test.fromhex, '\x00')
281 self.assertRaises(ValueError, self.type2test.fromhex, '12 \x00 34')
282
283 def test_join(self):
284 self.assertEqual(self.type2test(b"").join([]), b"")
285 self.assertEqual(self.type2test(b"").join([b""]), b"")
286 for lst in [[b"abc"], [b"a", b"bc"], [b"ab", b"c"], [b"a", b"b", b"c"]]:
287 lst = list(map(self.type2test, lst))
288 self.assertEqual(self.type2test(b"").join(lst), b"abc")
289 self.assertEqual(self.type2test(b"").join(tuple(lst)), b"abc")
290 self.assertEqual(self.type2test(b"").join(iter(lst)), b"abc")
291 self.assertEqual(self.type2test(b".").join([b"ab", b"cd"]), b"ab.cd")
292 # XXX more...
293
Christian Heimes510711d2008-01-30 11:57:58 +0000294 def test_count(self):
295 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200296 i = 105
297 p = 112
298 w = 119
299
Christian Heimes510711d2008-01-30 11:57:58 +0000300 self.assertEqual(b.count(b'i'), 4)
301 self.assertEqual(b.count(b'ss'), 2)
302 self.assertEqual(b.count(b'w'), 0)
303
Antoine Pitrouac65d962011-10-20 23:54:17 +0200304 self.assertEqual(b.count(i), 4)
305 self.assertEqual(b.count(w), 0)
306
307 self.assertEqual(b.count(b'i', 6), 2)
308 self.assertEqual(b.count(b'p', 6), 2)
309 self.assertEqual(b.count(b'i', 1, 3), 1)
310 self.assertEqual(b.count(b'p', 7, 9), 1)
311
312 self.assertEqual(b.count(i, 6), 2)
313 self.assertEqual(b.count(p, 6), 2)
314 self.assertEqual(b.count(i, 1, 3), 1)
315 self.assertEqual(b.count(p, 7, 9), 1)
316
Christian Heimes510711d2008-01-30 11:57:58 +0000317 def test_startswith(self):
318 b = self.type2test(b'hello')
319 self.assertFalse(self.type2test().startswith(b"anything"))
320 self.assertTrue(b.startswith(b"hello"))
321 self.assertTrue(b.startswith(b"hel"))
322 self.assertTrue(b.startswith(b"h"))
323 self.assertFalse(b.startswith(b"hellow"))
324 self.assertFalse(b.startswith(b"ha"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300325 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300326 b.startswith([b'h'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300327 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300328 self.assertIn('bytes', exc)
329 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000330
331 def test_endswith(self):
332 b = self.type2test(b'hello')
333 self.assertFalse(bytearray().endswith(b"anything"))
334 self.assertTrue(b.endswith(b"hello"))
335 self.assertTrue(b.endswith(b"llo"))
336 self.assertTrue(b.endswith(b"o"))
337 self.assertFalse(b.endswith(b"whello"))
338 self.assertFalse(b.endswith(b"no"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300339 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300340 b.endswith([b'o'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300341 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300342 self.assertIn('bytes', exc)
343 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000344
345 def test_find(self):
346 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200347 i = 105
348 w = 119
349
Christian Heimes510711d2008-01-30 11:57:58 +0000350 self.assertEqual(b.find(b'ss'), 2)
Christian Heimes510711d2008-01-30 11:57:58 +0000351 self.assertEqual(b.find(b'w'), -1)
352 self.assertEqual(b.find(b'mississippian'), -1)
353
Antoine Pitrouac65d962011-10-20 23:54:17 +0200354 self.assertEqual(b.find(i), 1)
355 self.assertEqual(b.find(w), -1)
356
357 self.assertEqual(b.find(b'ss', 3), 5)
358 self.assertEqual(b.find(b'ss', 1, 7), 2)
359 self.assertEqual(b.find(b'ss', 1, 3), -1)
360
361 self.assertEqual(b.find(i, 6), 7)
362 self.assertEqual(b.find(i, 1, 3), 1)
363 self.assertEqual(b.find(w, 1, 3), -1)
364
Victor Stinnerf8eac002011-12-18 01:17:41 +0100365 for index in (-1, 256, sys.maxsize + 1):
366 self.assertRaisesRegex(
367 ValueError, r'byte must be in range\(0, 256\)',
368 b.find, index)
369
Christian Heimes510711d2008-01-30 11:57:58 +0000370 def test_rfind(self):
371 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200372 i = 105
373 w = 119
374
Christian Heimes510711d2008-01-30 11:57:58 +0000375 self.assertEqual(b.rfind(b'ss'), 5)
Christian Heimes510711d2008-01-30 11:57:58 +0000376 self.assertEqual(b.rfind(b'w'), -1)
377 self.assertEqual(b.rfind(b'mississippian'), -1)
378
Antoine Pitrouac65d962011-10-20 23:54:17 +0200379 self.assertEqual(b.rfind(i), 10)
380 self.assertEqual(b.rfind(w), -1)
381
382 self.assertEqual(b.rfind(b'ss', 3), 5)
383 self.assertEqual(b.rfind(b'ss', 0, 6), 2)
384
385 self.assertEqual(b.rfind(i, 1, 3), 1)
386 self.assertEqual(b.rfind(i, 3, 9), 7)
387 self.assertEqual(b.rfind(w, 1, 3), -1)
388
Christian Heimes510711d2008-01-30 11:57:58 +0000389 def test_index(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200390 b = self.type2test(b'mississippi')
391 i = 105
392 w = 119
393
394 self.assertEqual(b.index(b'ss'), 2)
395 self.assertRaises(ValueError, b.index, b'w')
396 self.assertRaises(ValueError, b.index, b'mississippian')
397
398 self.assertEqual(b.index(i), 1)
399 self.assertRaises(ValueError, b.index, w)
400
401 self.assertEqual(b.index(b'ss', 3), 5)
402 self.assertEqual(b.index(b'ss', 1, 7), 2)
403 self.assertRaises(ValueError, b.index, b'ss', 1, 3)
404
405 self.assertEqual(b.index(i, 6), 7)
406 self.assertEqual(b.index(i, 1, 3), 1)
407 self.assertRaises(ValueError, b.index, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000408
409 def test_rindex(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200410 b = self.type2test(b'mississippi')
411 i = 105
412 w = 119
413
414 self.assertEqual(b.rindex(b'ss'), 5)
415 self.assertRaises(ValueError, b.rindex, b'w')
416 self.assertRaises(ValueError, b.rindex, b'mississippian')
417
418 self.assertEqual(b.rindex(i), 10)
419 self.assertRaises(ValueError, b.rindex, w)
420
421 self.assertEqual(b.rindex(b'ss', 3), 5)
422 self.assertEqual(b.rindex(b'ss', 0, 6), 2)
423
424 self.assertEqual(b.rindex(i, 1, 3), 1)
425 self.assertEqual(b.rindex(i, 3, 9), 7)
426 self.assertRaises(ValueError, b.rindex, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000427
428 def test_replace(self):
429 b = self.type2test(b'mississippi')
430 self.assertEqual(b.replace(b'i', b'a'), b'massassappa')
431 self.assertEqual(b.replace(b'ss', b'x'), b'mixixippi')
432
433 def test_split(self):
434 b = self.type2test(b'mississippi')
435 self.assertEqual(b.split(b'i'), [b'm', b'ss', b'ss', b'pp', b''])
436 self.assertEqual(b.split(b'ss'), [b'mi', b'i', b'ippi'])
437 self.assertEqual(b.split(b'w'), [b])
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200438 # with keyword args
439 b = self.type2test(b'a|b|c|d')
440 self.assertEqual(b.split(sep=b'|'), [b'a', b'b', b'c', b'd'])
441 self.assertEqual(b.split(b'|', maxsplit=1), [b'a', b'b|c|d'])
442 self.assertEqual(b.split(sep=b'|', maxsplit=1), [b'a', b'b|c|d'])
443 self.assertEqual(b.split(maxsplit=1, sep=b'|'), [b'a', b'b|c|d'])
444 b = self.type2test(b'a b c d')
445 self.assertEqual(b.split(maxsplit=1), [b'a', b'b c d'])
Christian Heimes510711d2008-01-30 11:57:58 +0000446
447 def test_split_whitespace(self):
448 for b in (b' arf barf ', b'arf\tbarf', b'arf\nbarf', b'arf\rbarf',
449 b'arf\fbarf', b'arf\vbarf'):
450 b = self.type2test(b)
451 self.assertEqual(b.split(), [b'arf', b'barf'])
452 self.assertEqual(b.split(None), [b'arf', b'barf'])
453 self.assertEqual(b.split(None, 2), [b'arf', b'barf'])
454 for b in (b'a\x1Cb', b'a\x1Db', b'a\x1Eb', b'a\x1Fb'):
455 b = self.type2test(b)
456 self.assertEqual(b.split(), [b])
457 self.assertEqual(self.type2test(b' a bb c ').split(None, 0), [b'a bb c '])
458 self.assertEqual(self.type2test(b' a bb c ').split(None, 1), [b'a', b'bb c '])
459 self.assertEqual(self.type2test(b' a bb c ').split(None, 2), [b'a', b'bb', b'c '])
460 self.assertEqual(self.type2test(b' a bb c ').split(None, 3), [b'a', b'bb', b'c'])
461
462 def test_split_string_error(self):
463 self.assertRaises(TypeError, self.type2test(b'a b').split, ' ')
464
Antoine Pitrouf8453022010-01-12 22:05:42 +0000465 def test_split_unicodewhitespace(self):
466 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
467 self.assertEqual(b.split(), [b'\x1c\x1d\x1e\x1f'])
468
Christian Heimes510711d2008-01-30 11:57:58 +0000469 def test_rsplit(self):
470 b = self.type2test(b'mississippi')
471 self.assertEqual(b.rsplit(b'i'), [b'm', b'ss', b'ss', b'pp', b''])
472 self.assertEqual(b.rsplit(b'ss'), [b'mi', b'i', b'ippi'])
473 self.assertEqual(b.rsplit(b'w'), [b])
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200474 # with keyword args
475 b = self.type2test(b'a|b|c|d')
476 self.assertEqual(b.rsplit(sep=b'|'), [b'a', b'b', b'c', b'd'])
477 self.assertEqual(b.rsplit(b'|', maxsplit=1), [b'a|b|c', b'd'])
478 self.assertEqual(b.rsplit(sep=b'|', maxsplit=1), [b'a|b|c', b'd'])
479 self.assertEqual(b.rsplit(maxsplit=1, sep=b'|'), [b'a|b|c', b'd'])
480 b = self.type2test(b'a b c d')
481 self.assertEqual(b.rsplit(maxsplit=1), [b'a b c', b'd'])
Christian Heimes510711d2008-01-30 11:57:58 +0000482
483 def test_rsplit_whitespace(self):
484 for b in (b' arf barf ', b'arf\tbarf', b'arf\nbarf', b'arf\rbarf',
485 b'arf\fbarf', b'arf\vbarf'):
486 b = self.type2test(b)
487 self.assertEqual(b.rsplit(), [b'arf', b'barf'])
488 self.assertEqual(b.rsplit(None), [b'arf', b'barf'])
489 self.assertEqual(b.rsplit(None, 2), [b'arf', b'barf'])
490 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 0), [b' a bb c'])
491 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 1), [b' a bb', b'c'])
492 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 2), [b' a', b'bb', b'c'])
493 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 3), [b'a', b'bb', b'c'])
494
495 def test_rsplit_string_error(self):
496 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, ' ')
497
498 def test_rsplit_unicodewhitespace(self):
499 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
Christian Heimes510711d2008-01-30 11:57:58 +0000500 self.assertEqual(b.rsplit(), [b'\x1c\x1d\x1e\x1f'])
501
502 def test_partition(self):
503 b = self.type2test(b'mississippi')
504 self.assertEqual(b.partition(b'ss'), (b'mi', b'ss', b'issippi'))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000505 self.assertEqual(b.partition(b'w'), (b'mississippi', b'', b''))
Christian Heimes510711d2008-01-30 11:57:58 +0000506
507 def test_rpartition(self):
508 b = self.type2test(b'mississippi')
509 self.assertEqual(b.rpartition(b'ss'), (b'missi', b'ss', b'ippi'))
510 self.assertEqual(b.rpartition(b'i'), (b'mississipp', b'i', b''))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000511 self.assertEqual(b.rpartition(b'w'), (b'', b'', b'mississippi'))
Christian Heimes510711d2008-01-30 11:57:58 +0000512
513 def test_pickling(self):
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +0000514 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Christian Heimes510711d2008-01-30 11:57:58 +0000515 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
516 b = self.type2test(b)
517 ps = pickle.dumps(b, proto)
518 q = pickle.loads(ps)
519 self.assertEqual(b, q)
520
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000521 def test_iterator_pickling(self):
522 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
523 it = itorg = iter(self.type2test(b))
524 data = list(self.type2test(b))
525 d = pickle.dumps(it)
526 it = pickle.loads(d)
527 self.assertEqual(type(itorg), type(it))
528 self.assertEqual(list(it), data)
529
530 it = pickle.loads(d)
531 try:
532 next(it)
533 except StopIteration:
534 continue
535 d = pickle.dumps(it)
536 it = pickle.loads(d)
537 self.assertEqual(list(it), data[1:])
538
Christian Heimes510711d2008-01-30 11:57:58 +0000539 def test_strip(self):
540 b = self.type2test(b'mississippi')
541 self.assertEqual(b.strip(b'i'), b'mississipp')
542 self.assertEqual(b.strip(b'm'), b'ississippi')
543 self.assertEqual(b.strip(b'pi'), b'mississ')
544 self.assertEqual(b.strip(b'im'), b'ssissipp')
545 self.assertEqual(b.strip(b'pim'), b'ssiss')
546 self.assertEqual(b.strip(b), b'')
547
548 def test_lstrip(self):
549 b = self.type2test(b'mississippi')
550 self.assertEqual(b.lstrip(b'i'), b'mississippi')
551 self.assertEqual(b.lstrip(b'm'), b'ississippi')
552 self.assertEqual(b.lstrip(b'pi'), b'mississippi')
553 self.assertEqual(b.lstrip(b'im'), b'ssissippi')
554 self.assertEqual(b.lstrip(b'pim'), b'ssissippi')
555
556 def test_rstrip(self):
557 b = self.type2test(b'mississippi')
558 self.assertEqual(b.rstrip(b'i'), b'mississipp')
559 self.assertEqual(b.rstrip(b'm'), b'mississippi')
560 self.assertEqual(b.rstrip(b'pi'), b'mississ')
561 self.assertEqual(b.rstrip(b'im'), b'mississipp')
562 self.assertEqual(b.rstrip(b'pim'), b'mississ')
563
564 def test_strip_whitespace(self):
565 b = self.type2test(b' \t\n\r\f\vabc \t\n\r\f\v')
566 self.assertEqual(b.strip(), b'abc')
567 self.assertEqual(b.lstrip(), b'abc \t\n\r\f\v')
568 self.assertEqual(b.rstrip(), b' \t\n\r\f\vabc')
569
570 def test_strip_bytearray(self):
571 self.assertEqual(self.type2test(b'abc').strip(memoryview(b'ac')), b'b')
572 self.assertEqual(self.type2test(b'abc').lstrip(memoryview(b'ac')), b'bc')
573 self.assertEqual(self.type2test(b'abc').rstrip(memoryview(b'ac')), b'ab')
574
575 def test_strip_string_error(self):
576 self.assertRaises(TypeError, self.type2test(b'abc').strip, 'b')
577 self.assertRaises(TypeError, self.type2test(b'abc').lstrip, 'b')
578 self.assertRaises(TypeError, self.type2test(b'abc').rstrip, 'b')
579
Eli Bendersky906b88f2011-07-29 07:05:08 +0300580 def test_center(self):
581 # Fill character can be either bytes or bytearray (issue 12380)
582 b = self.type2test(b'abc')
583 for fill_type in (bytes, bytearray):
584 self.assertEqual(b.center(7, fill_type(b'-')),
585 self.type2test(b'--abc--'))
586
587 def test_ljust(self):
588 # Fill character can be either bytes or bytearray (issue 12380)
589 b = self.type2test(b'abc')
590 for fill_type in (bytes, bytearray):
591 self.assertEqual(b.ljust(7, fill_type(b'-')),
592 self.type2test(b'abc----'))
593
594 def test_rjust(self):
595 # Fill character can be either bytes or bytearray (issue 12380)
596 b = self.type2test(b'abc')
597 for fill_type in (bytes, bytearray):
598 self.assertEqual(b.rjust(7, fill_type(b'-')),
599 self.type2test(b'----abc'))
600
Christian Heimes510711d2008-01-30 11:57:58 +0000601 def test_ord(self):
602 b = self.type2test(b'\0A\x7f\x80\xff')
603 self.assertEqual([ord(b[i:i+1]) for i in range(len(b))],
604 [0, 65, 127, 128, 255])
605
Georg Brandlabc38772009-04-12 15:51:51 +0000606 def test_maketrans(self):
607 transtable = b'\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037 !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`xyzdefghijklmnopqrstuvwxyz{|}~\177\200\201\202\203\204\205\206\207\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374\375\376\377'
Georg Brandlabc38772009-04-12 15:51:51 +0000608 self.assertEqual(self.type2test.maketrans(b'abc', b'xyz'), transtable)
Antoine Pitrou403ce782009-10-14 17:14:16 +0000609 transtable = b'\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037 !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\177\200\201\202\203\204\205\206\207\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374xyz'
610 self.assertEqual(self.type2test.maketrans(b'\375\376\377', b'xyz'), transtable)
Georg Brandlabc38772009-04-12 15:51:51 +0000611 self.assertRaises(ValueError, self.type2test.maketrans, b'abc', b'xyzq')
612 self.assertRaises(TypeError, self.type2test.maketrans, 'abc', 'def')
613
Jesus Ceaac451502011-04-20 17:09:23 +0200614 def test_none_arguments(self):
615 # issue 11828
616 b = self.type2test(b'hello')
617 l = self.type2test(b'l')
618 h = self.type2test(b'h')
619 x = self.type2test(b'x')
620 o = self.type2test(b'o')
621
622 self.assertEqual(2, b.find(l, None))
623 self.assertEqual(3, b.find(l, -2, None))
624 self.assertEqual(2, b.find(l, None, -2))
625 self.assertEqual(0, b.find(h, None, None))
626
627 self.assertEqual(3, b.rfind(l, None))
628 self.assertEqual(3, b.rfind(l, -2, None))
629 self.assertEqual(2, b.rfind(l, None, -2))
630 self.assertEqual(0, b.rfind(h, None, None))
631
632 self.assertEqual(2, b.index(l, None))
633 self.assertEqual(3, b.index(l, -2, None))
634 self.assertEqual(2, b.index(l, None, -2))
635 self.assertEqual(0, b.index(h, None, None))
636
637 self.assertEqual(3, b.rindex(l, None))
638 self.assertEqual(3, b.rindex(l, -2, None))
639 self.assertEqual(2, b.rindex(l, None, -2))
640 self.assertEqual(0, b.rindex(h, None, None))
641
642 self.assertEqual(2, b.count(l, None))
643 self.assertEqual(1, b.count(l, -2, None))
644 self.assertEqual(1, b.count(l, None, -2))
645 self.assertEqual(0, b.count(x, None, None))
646
647 self.assertEqual(True, b.endswith(o, None))
648 self.assertEqual(True, b.endswith(o, -2, None))
649 self.assertEqual(True, b.endswith(l, None, -2))
650 self.assertEqual(False, b.endswith(x, None, None))
651
652 self.assertEqual(True, b.startswith(h, None))
653 self.assertEqual(True, b.startswith(l, -2, None))
654 self.assertEqual(True, b.startswith(h, None, -2))
655 self.assertEqual(False, b.startswith(x, None, None))
656
Antoine Pitrouac65d962011-10-20 23:54:17 +0200657 def test_integer_arguments_out_of_byte_range(self):
658 b = self.type2test(b'hello')
659
660 for method in (b.count, b.find, b.index, b.rfind, b.rindex):
661 self.assertRaises(ValueError, method, -1)
662 self.assertRaises(ValueError, method, 256)
663 self.assertRaises(ValueError, method, 9999)
664
Jesus Ceaac451502011-04-20 17:09:23 +0200665 def test_find_etc_raise_correct_error_messages(self):
666 # issue 11828
667 b = self.type2test(b'hello')
668 x = self.type2test(b'x')
Ezio Melottiaf928422011-04-20 21:56:21 +0300669 self.assertRaisesRegex(TypeError, r'\bfind\b', b.find,
Jesus Ceaac451502011-04-20 17:09:23 +0200670 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300671 self.assertRaisesRegex(TypeError, r'\brfind\b', b.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +0200672 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300673 self.assertRaisesRegex(TypeError, r'\bindex\b', b.index,
Jesus Ceaac451502011-04-20 17:09:23 +0200674 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300675 self.assertRaisesRegex(TypeError, r'\brindex\b', b.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +0200676 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300677 self.assertRaisesRegex(TypeError, r'\bcount\b', b.count,
Jesus Ceaac451502011-04-20 17:09:23 +0200678 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300679 self.assertRaisesRegex(TypeError, r'\bstartswith\b', b.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200680 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300681 self.assertRaisesRegex(TypeError, r'\bendswith\b', b.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200682 x, None, None, None)
683
Christian Heimes510711d2008-01-30 11:57:58 +0000684
Ezio Melotti0dceb562013-01-10 07:43:26 +0200685class BytesTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +0000686 type2test = bytes
687
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000688 def test_buffer_is_readonly(self):
Antoine Pitrou47d305d2008-08-16 23:28:44 +0000689 fd = os.dup(sys.stdin.fileno())
690 with open(fd, "rb", buffering=0) as f:
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000691 self.assertRaises(TypeError, f.readinto, b"")
692
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000693 def test_custom(self):
694 class A:
695 def __bytes__(self):
696 return b'abc'
697 self.assertEqual(bytes(A()), b'abc')
698 class A: pass
699 self.assertRaises(TypeError, bytes, A())
700 class A:
701 def __bytes__(self):
702 return None
703 self.assertRaises(TypeError, bytes, A())
Benjamin Peterson5ff3f732012-12-19 15:27:41 -0600704 class A:
705 def __bytes__(self):
706 return b'a'
707 def __index__(self):
708 return 42
709 self.assertEqual(bytes(A()), b'a')
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000710
Victor Stinner29e762c2011-01-05 03:33:28 +0000711 # Test PyBytes_FromFormat()
712 def test_from_format(self):
713 test.support.import_module('ctypes')
714 from ctypes import pythonapi, py_object, c_int, c_char_p
715 PyBytes_FromFormat = pythonapi.PyBytes_FromFormat
716 PyBytes_FromFormat.restype = py_object
717
718 self.assertEqual(PyBytes_FromFormat(b'format'),
719 b'format')
720
721 self.assertEqual(PyBytes_FromFormat(b'%'), b'%')
722 self.assertEqual(PyBytes_FromFormat(b'%%'), b'%')
723 self.assertEqual(PyBytes_FromFormat(b'%%s'), b'%s')
724 self.assertEqual(PyBytes_FromFormat(b'[%%]'), b'[%]')
725 self.assertEqual(PyBytes_FromFormat(b'%%%c', c_int(ord('_'))), b'%_')
726
727 self.assertEqual(PyBytes_FromFormat(b'c:%c', c_int(255)),
728 b'c:\xff')
729 self.assertEqual(PyBytes_FromFormat(b's:%s', c_char_p(b'cstr')),
730 b's:cstr')
731
Victor Stinnerc9362cf2013-12-13 12:14:44 +0100732 # Issue #19969
733 self.assertRaises(OverflowError,
734 PyBytes_FromFormat, b'%c', c_int(-1))
735 self.assertRaises(OverflowError,
736 PyBytes_FromFormat, b'%c', c_int(256))
737
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000738
Ezio Melotti0dceb562013-01-10 07:43:26 +0200739class ByteArrayTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +0000740 type2test = bytearray
Guido van Rossumebea9be2007-04-09 00:49:13 +0000741
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000742 def test_nohash(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000743 self.assertRaises(TypeError, hash, bytearray())
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000744
Guido van Rossum254348e2007-11-21 19:29:53 +0000745 def test_bytearray_api(self):
Guido van Rossum75d6f1a2007-05-24 17:52:00 +0000746 short_sample = b"Hello world\n"
Guido van Rossum98297ee2007-11-06 21:34:58 +0000747 sample = short_sample + b"\0"*(20 - len(short_sample))
Guido van Rossumd624f182006-04-24 13:47:05 +0000748 tfn = tempfile.mktemp()
749 try:
750 # Prepare
751 with open(tfn, "wb") as f:
752 f.write(short_sample)
753 # Test readinto
754 with open(tfn, "rb") as f:
Guido van Rossum254348e2007-11-21 19:29:53 +0000755 b = bytearray(20)
Guido van Rossumd624f182006-04-24 13:47:05 +0000756 n = f.readinto(b)
757 self.assertEqual(n, len(short_sample))
Guido van Rossumc1f779c2007-07-03 08:25:58 +0000758 self.assertEqual(list(b), list(sample))
Guido van Rossumd624f182006-04-24 13:47:05 +0000759 # Test writing in binary mode
760 with open(tfn, "wb") as f:
761 f.write(b)
762 with open(tfn, "rb") as f:
763 self.assertEqual(f.read(), sample)
Guido van Rossum13633bb2007-04-13 18:42:35 +0000764 # Text mode is ambiguous; don't test
Guido van Rossumd624f182006-04-24 13:47:05 +0000765 finally:
766 try:
767 os.remove(tfn)
768 except os.error:
769 pass
770
Neal Norwitz6968b052007-02-27 19:02:19 +0000771 def test_reverse(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000772 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +0000773 self.assertEqual(b.reverse(), None)
774 self.assertEqual(b, b'olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +0000775 b = bytearray(b'hello1') # test even number of items
Neal Norwitz6968b052007-02-27 19:02:19 +0000776 b.reverse()
777 self.assertEqual(b, b'1olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +0000778 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +0000779 b.reverse()
780 self.assertFalse(b)
781
Eli Bendersky4db28d32011-03-03 18:21:02 +0000782 def test_clear(self):
783 b = bytearray(b'python')
784 b.clear()
785 self.assertEqual(b, b'')
786
787 b = bytearray(b'')
788 b.clear()
789 self.assertEqual(b, b'')
790
791 b = bytearray(b'')
792 b.append(ord('r'))
793 b.clear()
794 b.append(ord('p'))
795 self.assertEqual(b, b'p')
796
797 def test_copy(self):
798 b = bytearray(b'abc')
799 bb = b.copy()
800 self.assertEqual(bb, b'abc')
801
802 b = bytearray(b'')
803 bb = b.copy()
804 self.assertEqual(bb, b'')
805
806 # test that it's indeed a copy and not a reference
807 b = bytearray(b'abc')
808 bb = b.copy()
809 self.assertEqual(b, bb)
810 self.assertIsNot(b, bb)
811 bb.append(ord('d'))
812 self.assertEqual(bb, b'abcd')
813 self.assertEqual(b, b'abc')
814
Guido van Rossumd624f182006-04-24 13:47:05 +0000815 def test_regexps(self):
816 def by(s):
Guido van Rossum254348e2007-11-21 19:29:53 +0000817 return bytearray(map(ord, s))
Guido van Rossumd624f182006-04-24 13:47:05 +0000818 b = by("Hello, world")
Antoine Pitroufd036452008-08-19 17:56:33 +0000819 self.assertEqual(re.findall(br"\w+", b), [by("Hello"), by("world")])
Guido van Rossumd624f182006-04-24 13:47:05 +0000820
821 def test_setitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000822 b = bytearray([1, 2, 3])
Guido van Rossumd624f182006-04-24 13:47:05 +0000823 b[1] = 100
Guido van Rossum254348e2007-11-21 19:29:53 +0000824 self.assertEqual(b, bytearray([1, 100, 3]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000825 b[-1] = 200
Guido van Rossum254348e2007-11-21 19:29:53 +0000826 self.assertEqual(b, bytearray([1, 100, 200]))
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000827 b[0] = Indexable(10)
Guido van Rossum254348e2007-11-21 19:29:53 +0000828 self.assertEqual(b, bytearray([10, 100, 200]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000829 try:
830 b[3] = 0
831 self.fail("Didn't raise IndexError")
832 except IndexError:
833 pass
834 try:
835 b[-10] = 0
836 self.fail("Didn't raise IndexError")
837 except IndexError:
838 pass
839 try:
840 b[0] = 256
841 self.fail("Didn't raise ValueError")
842 except ValueError:
843 pass
844 try:
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000845 b[0] = Indexable(-1)
Guido van Rossumd624f182006-04-24 13:47:05 +0000846 self.fail("Didn't raise ValueError")
847 except ValueError:
848 pass
849 try:
850 b[0] = None
851 self.fail("Didn't raise TypeError")
852 except TypeError:
853 pass
854
855 def test_delitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000856 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +0000857 del b[0]
Guido van Rossum254348e2007-11-21 19:29:53 +0000858 self.assertEqual(b, bytearray(range(1, 10)))
Guido van Rossumd624f182006-04-24 13:47:05 +0000859 del b[-1]
Guido van Rossum254348e2007-11-21 19:29:53 +0000860 self.assertEqual(b, bytearray(range(1, 9)))
Guido van Rossumd624f182006-04-24 13:47:05 +0000861 del b[4]
Guido van Rossum254348e2007-11-21 19:29:53 +0000862 self.assertEqual(b, bytearray([1, 2, 3, 4, 6, 7, 8]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000863
864 def test_setslice(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000865 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +0000866 self.assertEqual(list(b), list(range(10)))
867
Guido van Rossum254348e2007-11-21 19:29:53 +0000868 b[0:5] = bytearray([1, 1, 1, 1, 1])
869 self.assertEqual(b, bytearray([1, 1, 1, 1, 1, 5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000870
871 del b[0:-5]
Guido van Rossum254348e2007-11-21 19:29:53 +0000872 self.assertEqual(b, bytearray([5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000873
Guido van Rossum254348e2007-11-21 19:29:53 +0000874 b[0:0] = bytearray([0, 1, 2, 3, 4])
875 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumd624f182006-04-24 13:47:05 +0000876
Guido van Rossum254348e2007-11-21 19:29:53 +0000877 b[-7:-3] = bytearray([100, 101])
878 self.assertEqual(b, bytearray([0, 1, 2, 100, 101, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000879
880 b[3:5] = [3, 4, 5, 6]
Guido van Rossum254348e2007-11-21 19:29:53 +0000881 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumad7d8d12007-04-13 01:39:34 +0000882
Thomas Wouters9a6e62b2006-08-23 23:20:29 +0000883 b[3:0] = [42, 42, 42]
Guido van Rossum254348e2007-11-21 19:29:53 +0000884 self.assertEqual(b, bytearray([0, 1, 2, 42, 42, 42, 3, 4, 5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +0000885
Ezio Melottic64bcbe2012-11-03 21:19:06 +0200886 b[3:] = b'foo'
887 self.assertEqual(b, bytearray([0, 1, 2, 102, 111, 111]))
888
889 b[:3] = memoryview(b'foo')
890 self.assertEqual(b, bytearray([102, 111, 111, 102, 111, 111]))
891
892 b[3:4] = []
893 self.assertEqual(b, bytearray([102, 111, 111, 111, 111]))
894
895 for elem in [5, -5, 0, int(10e20), 'str', 2.3,
896 ['a', 'b'], [b'a', b'b'], [[]]]:
897 with self.assertRaises(TypeError):
898 b[3:4] = elem
899
900 for elem in [[254, 255, 256], [-256, 9000]]:
901 with self.assertRaises(ValueError):
902 b[3:4] = elem
903
Thomas Wouters376446d2006-12-19 08:30:14 +0000904 def test_extended_set_del_slice(self):
Mark Dickinsonbc099642010-01-29 17:27:24 +0000905 indices = (0, None, 1, 3, 19, 300, 1<<333, -1, -2, -31, -300)
Thomas Wouters376446d2006-12-19 08:30:14 +0000906 for start in indices:
907 for stop in indices:
908 # Skip invalid step 0
909 for step in indices[1:]:
910 L = list(range(255))
Guido van Rossum254348e2007-11-21 19:29:53 +0000911 b = bytearray(L)
Thomas Wouters376446d2006-12-19 08:30:14 +0000912 # Make sure we have a slice of exactly the right length,
913 # but with different data.
914 data = L[start:stop:step]
915 data.reverse()
916 L[start:stop:step] = data
917 b[start:stop:step] = data
Ezio Melottib3aedd42010-11-20 19:04:17 +0000918 self.assertEqual(b, bytearray(L))
Guido van Rossumad7d8d12007-04-13 01:39:34 +0000919
Thomas Wouters376446d2006-12-19 08:30:14 +0000920 del L[start:stop:step]
921 del b[start:stop:step]
Ezio Melottib3aedd42010-11-20 19:04:17 +0000922 self.assertEqual(b, bytearray(L))
Thomas Wouters376446d2006-12-19 08:30:14 +0000923
Guido van Rossumd624f182006-04-24 13:47:05 +0000924 def test_setslice_trap(self):
925 # This test verifies that we correctly handle assigning self
926 # to a slice of self (the old Lambert Meertens trap).
Guido van Rossum254348e2007-11-21 19:29:53 +0000927 b = bytearray(range(256))
Guido van Rossumd624f182006-04-24 13:47:05 +0000928 b[8:] = b
Guido van Rossum254348e2007-11-21 19:29:53 +0000929 self.assertEqual(b, bytearray(list(range(8)) + list(range(256))))
Guido van Rossumd624f182006-04-24 13:47:05 +0000930
Guido van Rossum13e57212006-04-27 22:54:26 +0000931 def test_iconcat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000932 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +0000933 b1 = b
Guido van Rossum9c627722007-08-27 18:31:48 +0000934 b += b"def"
935 self.assertEqual(b, b"abcdef")
Guido van Rossum13e57212006-04-27 22:54:26 +0000936 self.assertEqual(b, b1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000937 self.assertTrue(b is b1)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000938 b += b"xyz"
Guido van Rossumad7d8d12007-04-13 01:39:34 +0000939 self.assertEqual(b, b"abcdefxyz")
940 try:
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000941 b += ""
Guido van Rossumad7d8d12007-04-13 01:39:34 +0000942 except TypeError:
943 pass
944 else:
945 self.fail("bytes += unicode didn't raise TypeError")
Guido van Rossum13e57212006-04-27 22:54:26 +0000946
947 def test_irepeat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000948 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +0000949 b1 = b
950 b *= 3
Guido van Rossum9c627722007-08-27 18:31:48 +0000951 self.assertEqual(b, b"abcabcabc")
Guido van Rossum13e57212006-04-27 22:54:26 +0000952 self.assertEqual(b, b1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000953 self.assertTrue(b is b1)
Guido van Rossum13e57212006-04-27 22:54:26 +0000954
955 def test_irepeat_1char(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000956 b = bytearray(b"x")
Guido van Rossum13e57212006-04-27 22:54:26 +0000957 b1 = b
958 b *= 100
Guido van Rossum98297ee2007-11-06 21:34:58 +0000959 self.assertEqual(b, b"x"*100)
Guido van Rossum13e57212006-04-27 22:54:26 +0000960 self.assertEqual(b, b1)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000961 self.assertTrue(b is b1)
Guido van Rossum13e57212006-04-27 22:54:26 +0000962
Guido van Rossum20188312006-05-05 15:15:40 +0000963 def test_alloc(self):
Guido van Rossum254348e2007-11-21 19:29:53 +0000964 b = bytearray()
Guido van Rossum20188312006-05-05 15:15:40 +0000965 alloc = b.__alloc__()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000966 self.assertTrue(alloc >= 0)
Guido van Rossum20188312006-05-05 15:15:40 +0000967 seq = [alloc]
968 for i in range(100):
Guido van Rossum9c627722007-08-27 18:31:48 +0000969 b += b"x"
Guido van Rossum20188312006-05-05 15:15:40 +0000970 alloc = b.__alloc__()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000971 self.assertTrue(alloc >= len(b))
Guido van Rossum20188312006-05-05 15:15:40 +0000972 if alloc not in seq:
973 seq.append(alloc)
Guido van Rossum20188312006-05-05 15:15:40 +0000974
Neal Norwitz6968b052007-02-27 19:02:19 +0000975 def test_extend(self):
976 orig = b'hello'
Guido van Rossum254348e2007-11-21 19:29:53 +0000977 a = bytearray(orig)
Neal Norwitz6968b052007-02-27 19:02:19 +0000978 a.extend(a)
979 self.assertEqual(a, orig + orig)
980 self.assertEqual(a[5:], orig)
Alexandre Vassalotti09121e82007-12-04 05:51:13 +0000981 a = bytearray(b'')
982 # Test iterators that don't have a __length_hint__
983 a.extend(map(int, orig * 25))
984 a.extend(int(x) for x in orig * 25)
985 self.assertEqual(a, orig * 50)
986 self.assertEqual(a[-5:], orig)
987 a = bytearray(b'')
988 a.extend(iter(map(int, orig * 50)))
989 self.assertEqual(a, orig * 50)
990 self.assertEqual(a[-5:], orig)
991 a = bytearray(b'')
992 a.extend(list(map(int, orig * 50)))
993 self.assertEqual(a, orig * 50)
994 self.assertEqual(a[-5:], orig)
995 a = bytearray(b'')
996 self.assertRaises(ValueError, a.extend, [0, 1, 2, 256])
997 self.assertRaises(ValueError, a.extend, [0, 1, 2, -1])
998 self.assertEqual(len(a), 0)
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000999 a = bytearray(b'')
1000 a.extend([Indexable(ord('a'))])
1001 self.assertEqual(a, b'a')
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001002
Neal Norwitz6968b052007-02-27 19:02:19 +00001003 def test_remove(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001004 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +00001005 b.remove(ord('l'))
1006 self.assertEqual(b, b'helo')
1007 b.remove(ord('l'))
1008 self.assertEqual(b, b'heo')
1009 self.assertRaises(ValueError, lambda: b.remove(ord('l')))
1010 self.assertRaises(ValueError, lambda: b.remove(400))
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001011 self.assertRaises(TypeError, lambda: b.remove('e'))
Neal Norwitz6968b052007-02-27 19:02:19 +00001012 # remove first and last
1013 b.remove(ord('o'))
1014 b.remove(ord('h'))
1015 self.assertEqual(b, b'e')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001016 self.assertRaises(TypeError, lambda: b.remove(b'e'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001017 b.remove(Indexable(ord('e')))
1018 self.assertEqual(b, b'')
Neal Norwitz6968b052007-02-27 19:02:19 +00001019
1020 def test_pop(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001021 b = bytearray(b'world')
Neal Norwitz6968b052007-02-27 19:02:19 +00001022 self.assertEqual(b.pop(), ord('d'))
1023 self.assertEqual(b.pop(0), ord('w'))
1024 self.assertEqual(b.pop(-2), ord('r'))
1025 self.assertRaises(IndexError, lambda: b.pop(10))
Eli Bendersky1bc4f192011-03-04 04:55:25 +00001026 self.assertRaises(IndexError, lambda: bytearray().pop())
Mark Dickinson54a3db92009-09-06 10:19:23 +00001027 # test for issue #6846
1028 self.assertEqual(bytearray(b'\xff').pop(), 0xff)
Neal Norwitz6968b052007-02-27 19:02:19 +00001029
1030 def test_nosort(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001031 self.assertRaises(AttributeError, lambda: bytearray().sort())
Neal Norwitz6968b052007-02-27 19:02:19 +00001032
Neal Norwitz6968b052007-02-27 19:02:19 +00001033 def test_append(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001034 b = bytearray(b'hell')
Neal Norwitz6968b052007-02-27 19:02:19 +00001035 b.append(ord('o'))
1036 self.assertEqual(b, b'hello')
1037 self.assertEqual(b.append(100), None)
Guido van Rossum254348e2007-11-21 19:29:53 +00001038 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +00001039 b.append(ord('A'))
1040 self.assertEqual(len(b), 1)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001041 self.assertRaises(TypeError, lambda: b.append(b'o'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001042 b = bytearray()
1043 b.append(Indexable(ord('A')))
1044 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001045
1046 def test_insert(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001047 b = bytearray(b'msssspp')
Neal Norwitz6968b052007-02-27 19:02:19 +00001048 b.insert(1, ord('i'))
1049 b.insert(4, ord('i'))
1050 b.insert(-2, ord('i'))
1051 b.insert(1000, ord('i'))
1052 self.assertEqual(b, b'mississippi')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001053 self.assertRaises(TypeError, lambda: b.insert(0, b'1'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001054 b = bytearray()
1055 b.insert(0, Indexable(ord('A')))
1056 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001057
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001058 def test_copied(self):
1059 # Issue 4348. Make sure that operations that don't mutate the array
1060 # copy the bytes.
1061 b = bytearray(b'abc')
Amaury Forgeot d'Arcb0c29162008-11-22 22:18:04 +00001062 self.assertFalse(b is b.replace(b'abc', b'cde', 0))
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001063
1064 t = bytearray([i for i in range(256)])
1065 x = bytearray(b'')
1066 self.assertFalse(x is x.translate(t))
1067
Guido van Rossum254348e2007-11-21 19:29:53 +00001068 def test_partition_bytearray_doesnt_share_nullstring(self):
1069 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001070 self.assertEqual(b, b"")
1071 self.assertEqual(c, b"")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001072 self.assertTrue(b is not c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001073 b += b"!"
1074 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001075 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001076 self.assertEqual(b, b"")
1077 self.assertEqual(c, b"")
1078 # Same for rpartition
Guido van Rossum254348e2007-11-21 19:29:53 +00001079 b, c, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001080 self.assertEqual(b, b"")
1081 self.assertEqual(c, b"")
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001082 self.assertTrue(b is not c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001083 b += b"!"
1084 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001085 c, b, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001086 self.assertEqual(b, b"")
1087 self.assertEqual(c, b"")
1088
Antoine Pitrou5504e892008-12-06 21:27:53 +00001089 def test_resize_forbidden(self):
1090 # #4509: can't resize a bytearray when there are buffer exports, even
1091 # if it wouldn't reallocate the underlying buffer.
1092 # Furthermore, no destructive changes to the buffer may be applied
1093 # before raising the error.
1094 b = bytearray(range(10))
1095 v = memoryview(b)
1096 def resize(n):
1097 b[1:-1] = range(n + 1, 2*n - 1)
1098 resize(10)
1099 orig = b[:]
1100 self.assertRaises(BufferError, resize, 11)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001101 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001102 self.assertRaises(BufferError, resize, 9)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001103 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001104 self.assertRaises(BufferError, resize, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001105 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001106 # Other operations implying resize
1107 self.assertRaises(BufferError, b.pop, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001108 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001109 self.assertRaises(BufferError, b.remove, b[1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001110 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001111 def delitem():
1112 del b[1]
1113 self.assertRaises(BufferError, delitem)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001114 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001115 # deleting a non-contiguous slice
1116 def delslice():
1117 b[1:-1:2] = b""
1118 self.assertRaises(BufferError, delslice)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001119 self.assertEqual(b, orig)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001120
Antoine Pitroufc8d6f42010-01-17 12:38:54 +00001121
Christian Heimes510711d2008-01-30 11:57:58 +00001122class AssortedBytesTest(unittest.TestCase):
1123 #
1124 # Test various combinations of bytes and bytearray
1125 #
1126
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001127 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001128 def test_repr_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001129 for f in str, repr:
1130 self.assertEqual(f(bytearray()), "bytearray(b'')")
1131 self.assertEqual(f(bytearray([0])), "bytearray(b'\\x00')")
1132 self.assertEqual(f(bytearray([0, 1, 254, 255])),
1133 "bytearray(b'\\x00\\x01\\xfe\\xff')")
1134 self.assertEqual(f(b"abc"), "b'abc'")
1135 self.assertEqual(f(b"'"), '''b"'"''') # '''
1136 self.assertEqual(f(b"'\""), r"""b'\'"'""") # '
Christian Heimes510711d2008-01-30 11:57:58 +00001137
1138 def test_compare_bytes_to_bytearray(self):
1139 self.assertEqual(b"abc" == bytes(b"abc"), True)
1140 self.assertEqual(b"ab" != bytes(b"abc"), True)
1141 self.assertEqual(b"ab" <= bytes(b"abc"), True)
1142 self.assertEqual(b"ab" < bytes(b"abc"), True)
1143 self.assertEqual(b"abc" >= bytes(b"ab"), True)
1144 self.assertEqual(b"abc" > bytes(b"ab"), True)
1145
1146 self.assertEqual(b"abc" != bytes(b"abc"), False)
1147 self.assertEqual(b"ab" == bytes(b"abc"), False)
1148 self.assertEqual(b"ab" > bytes(b"abc"), False)
1149 self.assertEqual(b"ab" >= bytes(b"abc"), False)
1150 self.assertEqual(b"abc" < bytes(b"ab"), False)
1151 self.assertEqual(b"abc" <= bytes(b"ab"), False)
1152
1153 self.assertEqual(bytes(b"abc") == b"abc", True)
1154 self.assertEqual(bytes(b"ab") != b"abc", True)
1155 self.assertEqual(bytes(b"ab") <= b"abc", True)
1156 self.assertEqual(bytes(b"ab") < b"abc", True)
1157 self.assertEqual(bytes(b"abc") >= b"ab", True)
1158 self.assertEqual(bytes(b"abc") > b"ab", True)
1159
1160 self.assertEqual(bytes(b"abc") != b"abc", False)
1161 self.assertEqual(bytes(b"ab") == b"abc", False)
1162 self.assertEqual(bytes(b"ab") > b"abc", False)
1163 self.assertEqual(bytes(b"ab") >= b"abc", False)
1164 self.assertEqual(bytes(b"abc") < b"ab", False)
1165 self.assertEqual(bytes(b"abc") <= b"ab", False)
1166
Stefan Krah6e572b82013-01-26 13:06:36 +01001167 @test.support.requires_docstrings
Christian Heimes510711d2008-01-30 11:57:58 +00001168 def test_doc(self):
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001169 self.assertIsNotNone(bytearray.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001170 self.assertTrue(bytearray.__doc__.startswith("bytearray("), bytearray.__doc__)
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001171 self.assertIsNotNone(bytes.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001172 self.assertTrue(bytes.__doc__.startswith("bytes("), bytes.__doc__)
Christian Heimes510711d2008-01-30 11:57:58 +00001173
1174 def test_from_bytearray(self):
1175 sample = bytes(b"Hello world\n\x80\x81\xfe\xff")
1176 buf = memoryview(sample)
1177 b = bytearray(buf)
1178 self.assertEqual(b, bytearray(sample))
1179
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001180 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001181 def test_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001182 self.assertEqual(str(b''), "b''")
1183 self.assertEqual(str(b'x'), "b'x'")
1184 self.assertEqual(str(b'\x80'), "b'\\x80'")
1185 self.assertEqual(str(bytearray(b'')), "bytearray(b'')")
1186 self.assertEqual(str(bytearray(b'x')), "bytearray(b'x')")
1187 self.assertEqual(str(bytearray(b'\x80')), "bytearray(b'\\x80')")
Christian Heimes510711d2008-01-30 11:57:58 +00001188
1189 def test_literal(self):
1190 tests = [
1191 (b"Wonderful spam", "Wonderful spam"),
1192 (br"Wonderful spam too", "Wonderful spam too"),
1193 (b"\xaa\x00\000\200", "\xaa\x00\000\200"),
1194 (br"\xaa\x00\000\200", r"\xaa\x00\000\200"),
1195 ]
1196 for b, s in tests:
1197 self.assertEqual(b, bytearray(s, 'latin-1'))
1198 for c in range(128, 256):
1199 self.assertRaises(SyntaxError, eval,
1200 'b"%s"' % chr(c))
1201
1202 def test_translate(self):
1203 b = b'hello'
Georg Brandlccc47b62008-12-28 11:44:14 +00001204 ba = bytearray(b)
Christian Heimes510711d2008-01-30 11:57:58 +00001205 rosetta = bytearray(range(0, 256))
1206 rosetta[ord('o')] = ord('e')
1207 c = b.translate(rosetta, b'l')
1208 self.assertEqual(b, b'hello')
1209 self.assertEqual(c, b'hee')
Georg Brandlccc47b62008-12-28 11:44:14 +00001210 c = ba.translate(rosetta, b'l')
1211 self.assertEqual(ba, b'hello')
1212 self.assertEqual(c, b'hee')
1213 c = b.translate(None, b'e')
1214 self.assertEqual(c, b'hllo')
1215 c = ba.translate(None, b'e')
1216 self.assertEqual(c, b'hllo')
1217 self.assertRaises(TypeError, b.translate, None, None)
1218 self.assertRaises(TypeError, ba.translate, None, None)
Christian Heimes510711d2008-01-30 11:57:58 +00001219
1220 def test_split_bytearray(self):
1221 self.assertEqual(b'a b'.split(memoryview(b' ')), [b'a', b'b'])
1222
1223 def test_rsplit_bytearray(self):
1224 self.assertEqual(b'a b'.rsplit(memoryview(b' ')), [b'a', b'b'])
1225
Georg Brandleb2c9642008-05-30 12:05:02 +00001226 def test_return_self(self):
1227 # bytearray.replace must always return a new bytearray
1228 b = bytearray()
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001229 self.assertFalse(b.replace(b'', b'') is b)
Georg Brandleb2c9642008-05-30 12:05:02 +00001230
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001231 def test_compare(self):
1232 if sys.flags.bytes_warning:
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001233 def bytes_warning():
1234 return test.support.check_warnings(('', BytesWarning))
1235 with bytes_warning():
1236 b'' == ''
1237 with bytes_warning():
1238 b'' != ''
1239 with bytes_warning():
1240 bytearray(b'') == ''
1241 with bytes_warning():
1242 bytearray(b'') != ''
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001243 else:
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001244 self.skipTest("BytesWarning is needed for this test: use -bb option")
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001245
Guido van Rossumd624f182006-04-24 13:47:05 +00001246 # Optimizations:
Guido van Rossume06b6b82006-04-23 07:43:54 +00001247 # __iter__? (optimization)
Guido van Rossumd624f182006-04-24 13:47:05 +00001248 # __reversed__? (optimization)
1249
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001250 # XXX More string methods? (Those that don't use character properties)
1251
Neal Norwitz6968b052007-02-27 19:02:19 +00001252 # There are tests in string_tests.py that are more
1253 # comprehensive for things like split, partition, etc.
1254 # Unfortunately they are all bundled with tests that
1255 # are not appropriate for bytes
Guido van Rossume06b6b82006-04-23 07:43:54 +00001256
Guido van Rossum254348e2007-11-21 19:29:53 +00001257 # I've started porting some of those into bytearray_tests.py, we should port
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001258 # the rest that make sense (the code can be cleaned up to use modern
1259 # unittest methods at the same time).
1260
Guido van Rossum254348e2007-11-21 19:29:53 +00001261class BytearrayPEP3137Test(unittest.TestCase,
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001262 test.buffer_tests.MixinBytesBufferCommonTests):
1263 def marshal(self, x):
Guido van Rossum254348e2007-11-21 19:29:53 +00001264 return bytearray(x)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001265
1266 def test_returns_new_copy(self):
1267 val = self.marshal(b'1234')
1268 # On immutable types these MAY return a reference to themselves
Guido van Rossum254348e2007-11-21 19:29:53 +00001269 # but on mutable types like bytearray they MUST return a new copy.
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001270 for methname in ('zfill', 'rjust', 'ljust', 'center'):
1271 method = getattr(val, methname)
1272 newval = method(3)
1273 self.assertEqual(val, newval)
1274 self.assertTrue(val is not newval,
1275 methname+' returned self on a mutable object')
Antoine Pitroucbaa62502010-01-13 15:03:39 +00001276 for expr in ('val.split()[0]', 'val.rsplit()[0]',
1277 'val.partition(b".")[0]', 'val.rpartition(b".")[2]',
1278 'val.splitlines()[0]', 'val.replace(b"", b"")'):
1279 newval = eval(expr)
1280 self.assertEqual(val, newval)
1281 self.assertTrue(val is not newval,
1282 expr+' returned val on a mutable object')
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001283
Christian Heimes510711d2008-01-30 11:57:58 +00001284class FixedStringTest(test.string_tests.BaseTest):
Georg Brandlc7885542007-03-06 19:16:20 +00001285
Guido van Rossum9c627722007-08-27 18:31:48 +00001286 def fixtype(self, obj):
1287 if isinstance(obj, str):
1288 return obj.encode("utf-8")
1289 return super().fixtype(obj)
1290
Georg Brandlc7885542007-03-06 19:16:20 +00001291 # Currently the bytes containment testing uses a single integer
1292 # value. This may not be the final design, but until then the
1293 # bytes section with in a bytes containment not valid
1294 def test_contains(self):
1295 pass
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001296 def test_expandtabs(self):
1297 pass
1298 def test_upper(self):
1299 pass
1300 def test_lower(self):
1301 pass
Georg Brandlc7885542007-03-06 19:16:20 +00001302
Ezio Melotti0dceb562013-01-10 07:43:26 +02001303class ByteArrayAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001304 type2test = bytearray
Antoine Pitrouac65d962011-10-20 23:54:17 +02001305 contains_bytes = True
Christian Heimes510711d2008-01-30 11:57:58 +00001306
Ezio Melotti0dceb562013-01-10 07:43:26 +02001307class BytesAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001308 type2test = bytes
Antoine Pitrouac65d962011-10-20 23:54:17 +02001309 contains_bytes = True
Christian Heimes510711d2008-01-30 11:57:58 +00001310
Georg Brandlc7885542007-03-06 19:16:20 +00001311
Ezio Melotti0dceb562013-01-10 07:43:26 +02001312class SubclassTest:
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001313
1314 def test_basic(self):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001315 self.assertTrue(issubclass(self.subclass2test, self.type2test))
Ezio Melottie9615932010-01-24 19:26:24 +00001316 self.assertIsInstance(self.subclass2test(), self.type2test)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001317
1318 a, b = b"abcd", b"efgh"
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001319 _a, _b = self.subclass2test(a), self.subclass2test(b)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001320
1321 # test comparison operators with subclass instances
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001322 self.assertTrue(_a == _a)
1323 self.assertTrue(_a != _b)
1324 self.assertTrue(_a < _b)
1325 self.assertTrue(_a <= _b)
1326 self.assertTrue(_b >= _a)
1327 self.assertTrue(_b > _a)
1328 self.assertTrue(_a is not a)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001329
1330 # test concat of subclass instances
1331 self.assertEqual(a + b, _a + _b)
1332 self.assertEqual(a + b, a + _b)
1333 self.assertEqual(a + b, _a + b)
1334
1335 # test repeat
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001336 self.assertTrue(a*5 == _a*5)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001337
1338 def test_join(self):
1339 # Make sure join returns a NEW object for single item sequences
1340 # involving a subclass.
1341 # Make sure that it is of the appropriate type.
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001342 s1 = self.subclass2test(b"abcd")
1343 s2 = self.type2test().join([s1])
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001344 self.assertTrue(s1 is not s2)
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001345 self.assertTrue(type(s2) is self.type2test, type(s2))
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001346
1347 # Test reverse, calling join on subclass
1348 s3 = s1.join([b"abcd"])
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001349 self.assertTrue(type(s3) is self.type2test)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001350
1351 def test_pickle(self):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001352 a = self.subclass2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001353 a.x = 10
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001354 a.y = self.subclass2test(b"efgh")
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +00001355 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001356 b = pickle.loads(pickle.dumps(a, proto))
1357 self.assertNotEqual(id(a), id(b))
1358 self.assertEqual(a, b)
1359 self.assertEqual(a.x, b.x)
1360 self.assertEqual(a.y, b.y)
1361 self.assertEqual(type(a), type(b))
1362 self.assertEqual(type(a.y), type(b.y))
1363
1364 def test_copy(self):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001365 a = self.subclass2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001366 a.x = 10
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001367 a.y = self.subclass2test(b"efgh")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001368 for copy_method in (copy.copy, copy.deepcopy):
1369 b = copy_method(a)
1370 self.assertNotEqual(id(a), id(b))
1371 self.assertEqual(a, b)
1372 self.assertEqual(a.x, b.x)
1373 self.assertEqual(a.y, b.y)
1374 self.assertEqual(type(a), type(b))
1375 self.assertEqual(type(a.y), type(b.y))
1376
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001377
1378class ByteArraySubclass(bytearray):
1379 pass
1380
1381class BytesSubclass(bytes):
1382 pass
1383
Ezio Melotti0dceb562013-01-10 07:43:26 +02001384class ByteArraySubclassTest(SubclassTest, unittest.TestCase):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001385 type2test = bytearray
1386 subclass2test = ByteArraySubclass
1387
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001388 def test_init_override(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001389 class subclass(bytearray):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001390 def __init__(me, newarg=1, *args, **kwargs):
1391 bytearray.__init__(me, *args, **kwargs)
1392 x = subclass(4, b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001393 x = subclass(4, source=b"abcd")
1394 self.assertEqual(x, b"abcd")
1395 x = subclass(newarg=4, source=b"abcd")
1396 self.assertEqual(x, b"abcd")
1397
1398
Ezio Melotti0dceb562013-01-10 07:43:26 +02001399class BytesSubclassTest(SubclassTest, unittest.TestCase):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001400 type2test = bytes
1401 subclass2test = BytesSubclass
1402
1403
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001404if __name__ == "__main__":
Ezio Melotti0dceb562013-01-10 07:43:26 +02001405 unittest.main()