blob: 05568f2c6797193d642563ec90e2a7250c1297c5 [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
Martin Panter152a19c2016-04-06 06:37:17 +00003XXX This is a mess. Common tests should be unified with string_tests.py (and
4the latter should be modernized).
Guido van Rossum98297ee2007-11-06 21:34:58 +00005"""
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00006
INADA Naokia634e232017-01-06 17:32:01 +09007import array
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
Victor Stinner22eb6892019-06-26 00:51:05 +020015import textwrap
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000016import unittest
Antoine Pitroucc231542014-11-02 18:40:09 +010017
Benjamin Petersonee8712c2008-05-20 21:35:26 +000018import test.support
Georg Brandlc7885542007-03-06 19:16:20 +000019import test.string_tests
Serhiy Storchaka8dc2ec12016-03-30 21:01:26 +030020import test.list_tests
Antoine Pitroucc231542014-11-02 18:40:09 +010021from test.support import bigaddrspacetest, MAX_Py_ssize_t
Victor Stinner22eb6892019-06-26 00:51:05 +020022from test.support.script_helper import assert_python_failure
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000023
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000024
25if sys.flags.bytes_warning:
26 def check_bytes_warnings(func):
27 @functools.wraps(func)
28 def wrapper(*args, **kw):
29 with test.support.check_warnings(('', BytesWarning)):
30 return func(*args, **kw)
31 return wrapper
32else:
33 # no-op
34 def check_bytes_warnings(func):
35 return func
36
37
Georg Brandl9a54d7c2008-07-16 23:15:30 +000038class Indexable:
39 def __init__(self, value=0):
40 self.value = value
41 def __index__(self):
42 return self.value
43
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000044
Ezio Melotti0dceb562013-01-10 07:43:26 +020045class BaseBytesTest:
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000046
47 def test_basics(self):
Christian Heimes510711d2008-01-30 11:57:58 +000048 b = self.type2test()
49 self.assertEqual(type(b), self.type2test)
50 self.assertEqual(b.__class__, self.type2test)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000051
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +000052 def test_copy(self):
53 a = self.type2test(b"abcd")
54 for copy_method in (copy.copy, copy.deepcopy):
55 b = copy_method(a)
56 self.assertEqual(a, b)
57 self.assertEqual(type(a), type(b))
58
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000059 def test_empty_sequence(self):
Christian Heimes510711d2008-01-30 11:57:58 +000060 b = self.type2test()
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000061 self.assertEqual(len(b), 0)
62 self.assertRaises(IndexError, lambda: b[0])
63 self.assertRaises(IndexError, lambda: b[1])
Christian Heimesa37d4c62007-12-04 23:02:19 +000064 self.assertRaises(IndexError, lambda: b[sys.maxsize])
65 self.assertRaises(IndexError, lambda: b[sys.maxsize+1])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000066 self.assertRaises(IndexError, lambda: b[10**100])
67 self.assertRaises(IndexError, lambda: b[-1])
68 self.assertRaises(IndexError, lambda: b[-2])
Christian Heimesa37d4c62007-12-04 23:02:19 +000069 self.assertRaises(IndexError, lambda: b[-sys.maxsize])
70 self.assertRaises(IndexError, lambda: b[-sys.maxsize-1])
71 self.assertRaises(IndexError, lambda: b[-sys.maxsize-2])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000072 self.assertRaises(IndexError, lambda: b[-10**100])
73
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030074 def test_from_iterable(self):
75 b = self.type2test(range(256))
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000076 self.assertEqual(len(b), 256)
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030077 self.assertEqual(list(b), list(range(256)))
78
79 # Non-sequence iterable.
80 b = self.type2test({42})
81 self.assertEqual(b, b"*")
82 b = self.type2test({43, 45})
83 self.assertIn(tuple(b), {(43, 45), (45, 43)})
84
85 # Iterator that has a __length_hint__.
86 b = self.type2test(iter(range(256)))
87 self.assertEqual(len(b), 256)
88 self.assertEqual(list(b), list(range(256)))
89
90 # Iterator that doesn't have a __length_hint__.
91 b = self.type2test(i for i in range(256) if i % 2)
92 self.assertEqual(len(b), 128)
93 self.assertEqual(list(b), list(range(256))[1::2])
94
95 # Sequence without __iter__.
96 class S:
97 def __getitem__(self, i):
98 return (1, 2, 3)[i]
99 b = self.type2test(S())
100 self.assertEqual(b, b"\x01\x02\x03")
101
102 def test_from_tuple(self):
103 # There is a special case for tuples.
104 b = self.type2test(tuple(range(256)))
105 self.assertEqual(len(b), 256)
106 self.assertEqual(list(b), list(range(256)))
107 b = self.type2test((1, 2, 3))
108 self.assertEqual(b, b"\x01\x02\x03")
109
110 def test_from_list(self):
111 # There is a special case for lists.
112 b = self.type2test(list(range(256)))
113 self.assertEqual(len(b), 256)
114 self.assertEqual(list(b), list(range(256)))
115 b = self.type2test([1, 2, 3])
116 self.assertEqual(b, b"\x01\x02\x03")
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000117
Serhiy Storchaka914f9a02018-10-21 15:25:53 +0300118 def test_from_mutating_list(self):
119 # Issue #34973: Crash in bytes constructor with mutating list.
120 class X:
121 def __index__(self):
122 a.clear()
123 return 42
124 a = [X(), X()]
125 self.assertEqual(bytes(a), b'*')
126
127 class Y:
128 def __index__(self):
129 if len(a) < 1000:
130 a.append(self)
131 return 42
132 a = [Y()]
133 self.assertEqual(bytes(a), b'*' * 1000) # should not crash
134
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000135 def test_from_index(self):
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000136 b = self.type2test([Indexable(), Indexable(1), Indexable(254),
137 Indexable(255)])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000138 self.assertEqual(list(b), [0, 1, 254, 255])
Benjamin Peterson4c045832010-04-16 22:35:32 +0000139 self.assertRaises(ValueError, self.type2test, [Indexable(-1)])
140 self.assertRaises(ValueError, self.type2test, [Indexable(256)])
Guido van Rossum98297ee2007-11-06 21:34:58 +0000141
INADA Naokia634e232017-01-06 17:32:01 +0900142 def test_from_buffer(self):
143 a = self.type2test(array.array('B', [1, 2, 3]))
144 self.assertEqual(a, b"\x01\x02\x03")
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +0300145 a = self.type2test(b"\x01\x02\x03")
146 self.assertEqual(a, b"\x01\x02\x03")
INADA Naokia634e232017-01-06 17:32:01 +0900147
Serhiy Storchakae8904212018-10-15 00:02:57 +0300148 # Issues #29159 and #34974.
149 # Fallback when __index__ raises a TypeError
INADA Naokia634e232017-01-06 17:32:01 +0900150 class B(bytes):
151 def __index__(self):
152 raise TypeError
153
154 self.assertEqual(self.type2test(B(b"foobar")), b"foobar")
155
Guido van Rossum98297ee2007-11-06 21:34:58 +0000156 def test_from_ssize(self):
Benjamin Peterson4c045832010-04-16 22:35:32 +0000157 self.assertEqual(self.type2test(0), b'')
158 self.assertEqual(self.type2test(1), b'\x00')
159 self.assertEqual(self.type2test(5), b'\x00\x00\x00\x00\x00')
160 self.assertRaises(ValueError, self.type2test, -1)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000161
Benjamin Peterson4c045832010-04-16 22:35:32 +0000162 self.assertEqual(self.type2test('0', 'ascii'), b'0')
163 self.assertEqual(self.type2test(b'0'), b'0')
Benjamin Peterson8380dd52010-04-16 22:51:37 +0000164 self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000165
166 def test_constructor_type_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000167 self.assertRaises(TypeError, self.type2test, 0.0)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000168 class C:
169 pass
Christian Heimes510711d2008-01-30 11:57:58 +0000170 self.assertRaises(TypeError, self.type2test, ["0"])
171 self.assertRaises(TypeError, self.type2test, [0.0])
172 self.assertRaises(TypeError, self.type2test, [None])
173 self.assertRaises(TypeError, self.type2test, [C()])
Serhiy Storchaka2c2044e2018-10-21 15:29:12 +0300174 self.assertRaises(TypeError, self.type2test, encoding='ascii')
175 self.assertRaises(TypeError, self.type2test, errors='ignore')
Serhiy Storchaka83cf99d2014-12-02 09:24:06 +0200176 self.assertRaises(TypeError, self.type2test, 0, 'ascii')
177 self.assertRaises(TypeError, self.type2test, b'', 'ascii')
178 self.assertRaises(TypeError, self.type2test, 0, errors='ignore')
179 self.assertRaises(TypeError, self.type2test, b'', errors='ignore')
180 self.assertRaises(TypeError, self.type2test, '')
181 self.assertRaises(TypeError, self.type2test, '', errors='ignore')
182 self.assertRaises(TypeError, self.type2test, '', b'ascii')
183 self.assertRaises(TypeError, self.type2test, '', 'ascii', b'ignore')
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000184
185 def test_constructor_value_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000186 self.assertRaises(ValueError, self.type2test, [-1])
187 self.assertRaises(ValueError, self.type2test, [-sys.maxsize])
188 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-1])
189 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-2])
190 self.assertRaises(ValueError, self.type2test, [-10**100])
191 self.assertRaises(ValueError, self.type2test, [256])
192 self.assertRaises(ValueError, self.type2test, [257])
193 self.assertRaises(ValueError, self.type2test, [sys.maxsize])
194 self.assertRaises(ValueError, self.type2test, [sys.maxsize+1])
195 self.assertRaises(ValueError, self.type2test, [10**100])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000196
Antoine Pitroucc231542014-11-02 18:40:09 +0100197 @bigaddrspacetest
198 def test_constructor_overflow(self):
199 size = MAX_Py_ssize_t
200 self.assertRaises((OverflowError, MemoryError), self.type2test, size)
201 try:
202 # Should either pass or raise an error (e.g. on debug builds with
203 # additional malloc() overhead), but shouldn't crash.
204 bytearray(size - 4)
205 except (OverflowError, MemoryError):
206 pass
207
Serhiy Storchakae8904212018-10-15 00:02:57 +0300208 def test_constructor_exceptions(self):
209 # Issue #34974: bytes and bytearray constructors replace unexpected
210 # exceptions.
211 class BadInt:
212 def __index__(self):
213 1/0
214 self.assertRaises(ZeroDivisionError, self.type2test, BadInt())
215 self.assertRaises(ZeroDivisionError, self.type2test, [BadInt()])
216
217 class BadIterable:
218 def __iter__(self):
219 1/0
220 self.assertRaises(ZeroDivisionError, self.type2test, BadIterable())
221
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000222 def test_compare(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000223 b1 = self.type2test([1, 2, 3])
224 b2 = self.type2test([1, 2, 3])
225 b3 = self.type2test([1, 3])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000226
Guido van Rossume61fd5b2007-07-11 12:20:59 +0000227 self.assertEqual(b1, b2)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000228 self.assertTrue(b2 != b3)
229 self.assertTrue(b1 <= b2)
230 self.assertTrue(b1 <= b3)
231 self.assertTrue(b1 < b3)
232 self.assertTrue(b1 >= b2)
233 self.assertTrue(b3 >= b2)
234 self.assertTrue(b3 > b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000235
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000236 self.assertFalse(b1 != b2)
237 self.assertFalse(b2 == b3)
238 self.assertFalse(b1 > b2)
239 self.assertFalse(b1 > b3)
240 self.assertFalse(b1 >= b3)
241 self.assertFalse(b1 < b2)
242 self.assertFalse(b3 < b2)
243 self.assertFalse(b3 <= b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000244
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000245 @check_bytes_warnings
Guido van Rossum343e97f2007-04-09 00:43:24 +0000246 def test_compare_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000247 # Byte comparisons with unicode should always fail!
248 # Test this for all expected byte orders and Unicode character
249 # sizes.
250 self.assertEqual(self.type2test(b"\0a\0b\0c") == "abc", False)
251 self.assertEqual(self.type2test(b"\0\0\0a\0\0\0b\0\0\0c") == "abc",
252 False)
253 self.assertEqual(self.type2test(b"a\0b\0c\0") == "abc", False)
254 self.assertEqual(self.type2test(b"a\0\0\0b\0\0\0c\0\0\0") == "abc",
255 False)
256 self.assertEqual(self.type2test() == str(), False)
257 self.assertEqual(self.type2test() != str(), True)
Christian Heimes510711d2008-01-30 11:57:58 +0000258
259 def test_reversed(self):
260 input = list(map(ord, "Hello"))
261 b = self.type2test(input)
262 output = list(reversed(b))
263 input.reverse()
264 self.assertEqual(output, input)
265
266 def test_getslice(self):
267 def by(s):
268 return self.type2test(map(ord, s))
269 b = by("Hello, world")
270
271 self.assertEqual(b[:5], by("Hello"))
272 self.assertEqual(b[1:5], by("ello"))
273 self.assertEqual(b[5:7], by(", "))
274 self.assertEqual(b[7:], by("world"))
275 self.assertEqual(b[7:12], by("world"))
276 self.assertEqual(b[7:100], by("world"))
277
278 self.assertEqual(b[:-7], by("Hello"))
279 self.assertEqual(b[-11:-7], by("ello"))
280 self.assertEqual(b[-7:-5], by(", "))
281 self.assertEqual(b[-5:], by("world"))
282 self.assertEqual(b[-5:12], by("world"))
283 self.assertEqual(b[-5:100], by("world"))
284 self.assertEqual(b[-100:5], by("Hello"))
285
286 def test_extended_getslice(self):
287 # Test extended slicing by comparing with list slicing.
288 L = list(range(255))
289 b = self.type2test(L)
Zackery Spytz14514d92019-05-17 01:13:03 -0600290 indices = (0, None, 1, 3, 19, 100, sys.maxsize, -1, -2, -31, -100)
Christian Heimes510711d2008-01-30 11:57:58 +0000291 for start in indices:
292 for stop in indices:
293 # Skip step 0 (invalid)
294 for step in indices[1:]:
295 self.assertEqual(b[start:stop:step], self.type2test(L[start:stop:step]))
296
297 def test_encoding(self):
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000298 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000299 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000300 b = self.type2test(sample, enc)
301 self.assertEqual(b, self.type2test(sample.encode(enc)))
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000302 self.assertRaises(UnicodeEncodeError, self.type2test, sample, "latin-1")
303 b = self.type2test(sample, "latin-1", "ignore")
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000304 self.assertEqual(b, self.type2test(sample[:-3], "utf-8"))
Christian Heimes510711d2008-01-30 11:57:58 +0000305
306 def test_decode(self):
Serhiy Storchakac49a5162016-09-08 15:47:27 +0300307 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000308 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000309 b = self.type2test(sample, enc)
310 self.assertEqual(b.decode(enc), sample)
311 sample = "Hello world\n\x80\x81\xfe\xff"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000312 b = self.type2test(sample, "latin-1")
313 self.assertRaises(UnicodeDecodeError, b.decode, "utf-8")
314 self.assertEqual(b.decode("utf-8", "ignore"), "Hello world\n")
315 self.assertEqual(b.decode(errors="ignore", encoding="utf-8"),
Benjamin Peterson308d6372009-09-18 21:42:35 +0000316 "Hello world\n")
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000317 # Default encoding is utf-8
318 self.assertEqual(self.type2test(b'\xe2\x98\x83').decode(), '\u2603')
Christian Heimes510711d2008-01-30 11:57:58 +0000319
Victor Stinner22eb6892019-06-26 00:51:05 +0200320 def test_check_encoding_errors(self):
321 # bpo-37388: bytes(str) and bytes.encode() must check encoding
322 # and errors arguments in dev mode
323 invalid = 'Boom, Shaka Laka, Boom!'
324 encodings = ('ascii', 'utf8', 'latin1')
325 code = textwrap.dedent(f'''
326 import sys
327 type2test = {self.type2test.__name__}
328 encodings = {encodings!r}
329
330 for data in ('', 'short string'):
331 try:
332 type2test(data, encoding={invalid!r})
333 except LookupError:
334 pass
335 else:
336 sys.exit(21)
337
338 for encoding in encodings:
339 try:
340 type2test(data, encoding=encoding, errors={invalid!r})
341 except LookupError:
342 pass
343 else:
344 sys.exit(22)
345
346 for data in (b'', b'short string'):
347 data = type2test(data)
348 print(repr(data))
349 try:
350 data.decode(encoding={invalid!r})
351 except LookupError:
352 sys.exit(10)
353 else:
354 sys.exit(23)
355
356 try:
357 data.decode(errors={invalid!r})
358 except LookupError:
359 pass
360 else:
361 sys.exit(24)
362
363 for encoding in encodings:
364 try:
365 data.decode(encoding=encoding, errors={invalid!r})
366 except LookupError:
367 pass
368 else:
369 sys.exit(25)
370
371 sys.exit(10)
372 ''')
373 proc = assert_python_failure('-X', 'dev', '-c', code)
374 self.assertEqual(proc.rc, 10, proc)
375
Christian Heimes510711d2008-01-30 11:57:58 +0000376 def test_from_int(self):
377 b = self.type2test(0)
378 self.assertEqual(b, self.type2test())
379 b = self.type2test(10)
380 self.assertEqual(b, self.type2test([0]*10))
381 b = self.type2test(10000)
382 self.assertEqual(b, self.type2test([0]*10000))
383
384 def test_concat(self):
385 b1 = self.type2test(b"abc")
386 b2 = self.type2test(b"def")
387 self.assertEqual(b1 + b2, b"abcdef")
388 self.assertEqual(b1 + bytes(b"def"), b"abcdef")
389 self.assertEqual(bytes(b"def") + b1, b"defabc")
390 self.assertRaises(TypeError, lambda: b1 + "def")
391 self.assertRaises(TypeError, lambda: "abc" + b2)
392
393 def test_repeat(self):
394 for b in b"abc", self.type2test(b"abc"):
395 self.assertEqual(b * 3, b"abcabcabc")
396 self.assertEqual(b * 0, b"")
397 self.assertEqual(b * -1, b"")
398 self.assertRaises(TypeError, lambda: b * 3.14)
399 self.assertRaises(TypeError, lambda: 3.14 * b)
400 # XXX Shouldn't bytes and bytearray agree on what to raise?
Mark Dickinsoncf940c72010-08-10 18:35:01 +0000401 with self.assertRaises((OverflowError, MemoryError)):
402 c = b * sys.maxsize
403 with self.assertRaises((OverflowError, MemoryError)):
404 b *= sys.maxsize
Christian Heimes510711d2008-01-30 11:57:58 +0000405
406 def test_repeat_1char(self):
407 self.assertEqual(self.type2test(b'x')*100, self.type2test([ord('x')]*100))
408
409 def test_contains(self):
410 b = self.type2test(b"abc")
Benjamin Peterson577473f2010-01-19 00:09:57 +0000411 self.assertIn(ord('a'), b)
412 self.assertIn(int(ord('a')), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000413 self.assertNotIn(200, b)
Christian Heimes510711d2008-01-30 11:57:58 +0000414 self.assertRaises(ValueError, lambda: 300 in b)
415 self.assertRaises(ValueError, lambda: -1 in b)
Serhiy Storchakaf9efb8b2016-07-10 12:37:30 +0300416 self.assertRaises(ValueError, lambda: sys.maxsize+1 in b)
Christian Heimes510711d2008-01-30 11:57:58 +0000417 self.assertRaises(TypeError, lambda: None in b)
418 self.assertRaises(TypeError, lambda: float(ord('a')) in b)
419 self.assertRaises(TypeError, lambda: "a" in b)
420 for f in bytes, bytearray:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000421 self.assertIn(f(b""), b)
422 self.assertIn(f(b"a"), b)
423 self.assertIn(f(b"b"), b)
424 self.assertIn(f(b"c"), b)
425 self.assertIn(f(b"ab"), b)
426 self.assertIn(f(b"bc"), b)
427 self.assertIn(f(b"abc"), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000428 self.assertNotIn(f(b"ac"), b)
429 self.assertNotIn(f(b"d"), b)
430 self.assertNotIn(f(b"dab"), b)
431 self.assertNotIn(f(b"abd"), b)
Christian Heimes510711d2008-01-30 11:57:58 +0000432
433 def test_fromhex(self):
434 self.assertRaises(TypeError, self.type2test.fromhex)
435 self.assertRaises(TypeError, self.type2test.fromhex, 1)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000436 self.assertEqual(self.type2test.fromhex(''), self.type2test())
Christian Heimes510711d2008-01-30 11:57:58 +0000437 b = bytearray([0x1a, 0x2b, 0x30])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000438 self.assertEqual(self.type2test.fromhex('1a2B30'), b)
439 self.assertEqual(self.type2test.fromhex(' 1A 2B 30 '), b)
Serhiy Storchakadd1da7f2016-12-19 18:51:37 +0200440
441 # check that ASCII whitespace is ignored
442 self.assertEqual(self.type2test.fromhex(' 1A\n2B\t30\v'), b)
443 for c in "\x09\x0A\x0B\x0C\x0D\x20":
444 self.assertEqual(self.type2test.fromhex(c), self.type2test())
445 for c in "\x1C\x1D\x1E\x1F\x85\xa0\u2000\u2002\u2028":
446 self.assertRaises(ValueError, self.type2test.fromhex, c)
447
Ezio Melottib3aedd42010-11-20 19:04:17 +0000448 self.assertEqual(self.type2test.fromhex('0000'), b'\0\0')
Christian Heimes510711d2008-01-30 11:57:58 +0000449 self.assertRaises(TypeError, self.type2test.fromhex, b'1B')
450 self.assertRaises(ValueError, self.type2test.fromhex, 'a')
451 self.assertRaises(ValueError, self.type2test.fromhex, 'rt')
452 self.assertRaises(ValueError, self.type2test.fromhex, '1a b cd')
453 self.assertRaises(ValueError, self.type2test.fromhex, '\x00')
454 self.assertRaises(ValueError, self.type2test.fromhex, '12 \x00 34')
455
Victor Stinner2bf89932015-10-14 11:25:33 +0200456 for data, pos in (
457 # invalid first hexadecimal character
458 ('12 x4 56', 3),
459 # invalid second hexadecimal character
460 ('12 3x 56', 4),
461 # two invalid hexadecimal characters
462 ('12 xy 56', 3),
463 # test non-ASCII string
464 ('12 3\xff 56', 4),
465 ):
466 with self.assertRaises(ValueError) as cm:
467 self.type2test.fromhex(data)
468 self.assertIn('at position %s' % pos, str(cm.exception))
469
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000470 def test_hex(self):
471 self.assertRaises(TypeError, self.type2test.hex)
472 self.assertRaises(TypeError, self.type2test.hex, 1)
Gregory P. Smith32d34bc2015-04-26 05:05:53 +0000473 self.assertEqual(self.type2test(b"").hex(), "")
474 self.assertEqual(bytearray([0x1a, 0x2b, 0x30]).hex(), '1a2b30')
475 self.assertEqual(self.type2test(b"\x1a\x2b\x30").hex(), '1a2b30')
476 self.assertEqual(memoryview(b"\x1a\x2b\x30").hex(), '1a2b30')
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000477
Gregory P. Smith0c2f9302019-05-29 11:46:58 -0700478 def test_hex_separator_basics(self):
479 three_bytes = self.type2test(b'\xb9\x01\xef')
480 self.assertEqual(three_bytes.hex(), 'b901ef')
481 with self.assertRaises(ValueError):
482 three_bytes.hex('')
483 with self.assertRaises(ValueError):
484 three_bytes.hex('xx')
485 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
486 with self.assertRaises(TypeError):
487 three_bytes.hex(None, 0)
488 with self.assertRaises(ValueError):
489 three_bytes.hex('\xff')
490 with self.assertRaises(ValueError):
491 three_bytes.hex(b'\xff')
492 with self.assertRaises(ValueError):
493 three_bytes.hex(b'\x80')
494 with self.assertRaises(ValueError):
495 three_bytes.hex(chr(0x100))
496 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
497 self.assertEqual(three_bytes.hex(b'\x00'), 'b9\x0001\x00ef')
498 self.assertEqual(three_bytes.hex('\x00'), 'b9\x0001\x00ef')
499 self.assertEqual(three_bytes.hex(b'\x7f'), 'b9\x7f01\x7fef')
500 self.assertEqual(three_bytes.hex('\x7f'), 'b9\x7f01\x7fef')
501 self.assertEqual(three_bytes.hex(':', 3), 'b901ef')
502 self.assertEqual(three_bytes.hex(':', 4), 'b901ef')
503 self.assertEqual(three_bytes.hex(':', -4), 'b901ef')
504 self.assertEqual(three_bytes.hex(':'), 'b9:01:ef')
505 self.assertEqual(three_bytes.hex(b'$'), 'b9$01$ef')
506 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
507 self.assertEqual(three_bytes.hex(':', -1), 'b9:01:ef')
508 self.assertEqual(three_bytes.hex(':', 2), 'b9:01ef')
509 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
510 self.assertEqual(three_bytes.hex('*', -2), 'b901*ef')
511
512 value = b'{s\005\000\000\000worldi\002\000\000\000s\005\000\000\000helloi\001\000\000\0000'
513 self.assertEqual(value.hex('.', 8), '7b7305000000776f.726c646902000000.730500000068656c.6c6f690100000030')
514
515 def test_hex_separator_five_bytes(self):
516 five_bytes = self.type2test(range(90,95))
517 self.assertEqual(five_bytes.hex(), '5a5b5c5d5e')
518
519 def test_hex_separator_six_bytes(self):
520 six_bytes = self.type2test(x*3 for x in range(1, 7))
521 self.assertEqual(six_bytes.hex(), '0306090c0f12')
522 self.assertEqual(six_bytes.hex('.', 1), '03.06.09.0c.0f.12')
523 self.assertEqual(six_bytes.hex(' ', 2), '0306 090c 0f12')
524 self.assertEqual(six_bytes.hex('-', 3), '030609-0c0f12')
525 self.assertEqual(six_bytes.hex(':', 4), '0306:090c0f12')
526 self.assertEqual(six_bytes.hex(':', 5), '03:06090c0f12')
527 self.assertEqual(six_bytes.hex(':', 6), '0306090c0f12')
528 self.assertEqual(six_bytes.hex(':', 95), '0306090c0f12')
529 self.assertEqual(six_bytes.hex('_', -3), '030609_0c0f12')
530 self.assertEqual(six_bytes.hex(':', -4), '0306090c:0f12')
531 self.assertEqual(six_bytes.hex(b'@', -5), '0306090c0f@12')
532 self.assertEqual(six_bytes.hex(':', -6), '0306090c0f12')
533 self.assertEqual(six_bytes.hex(' ', -95), '0306090c0f12')
534
Christian Heimes510711d2008-01-30 11:57:58 +0000535 def test_join(self):
536 self.assertEqual(self.type2test(b"").join([]), b"")
537 self.assertEqual(self.type2test(b"").join([b""]), b"")
538 for lst in [[b"abc"], [b"a", b"bc"], [b"ab", b"c"], [b"a", b"b", b"c"]]:
539 lst = list(map(self.type2test, lst))
540 self.assertEqual(self.type2test(b"").join(lst), b"abc")
541 self.assertEqual(self.type2test(b"").join(tuple(lst)), b"abc")
542 self.assertEqual(self.type2test(b"").join(iter(lst)), b"abc")
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200543 dot_join = self.type2test(b".:").join
544 self.assertEqual(dot_join([b"ab", b"cd"]), b"ab.:cd")
545 self.assertEqual(dot_join([memoryview(b"ab"), b"cd"]), b"ab.:cd")
546 self.assertEqual(dot_join([b"ab", memoryview(b"cd")]), b"ab.:cd")
547 self.assertEqual(dot_join([bytearray(b"ab"), b"cd"]), b"ab.:cd")
548 self.assertEqual(dot_join([b"ab", bytearray(b"cd")]), b"ab.:cd")
549 # Stress it with many items
Bruce Merryd07d9f42020-01-29 09:09:24 +0200550 seq = [b"abc"] * 100000
551 expected = b"abc" + b".:abc" * 99999
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200552 self.assertEqual(dot_join(seq), expected)
Bruce Merryd07d9f42020-01-29 09:09:24 +0200553 # Stress test with empty separator
554 seq = [b"abc"] * 100000
555 expected = b"abc" * 100000
556 self.assertEqual(self.type2test(b"").join(seq), expected)
Benjamin Petersonc31f12d2014-09-28 12:56:42 -0400557 self.assertRaises(TypeError, self.type2test(b" ").join, None)
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200558 # Error handling and cleanup when some item in the middle of the
559 # sequence has the wrong type.
560 with self.assertRaises(TypeError):
561 dot_join([bytearray(b"ab"), "cd", b"ef"])
562 with self.assertRaises(TypeError):
563 dot_join([memoryview(b"ab"), "cd", b"ef"])
Christian Heimes510711d2008-01-30 11:57:58 +0000564
Christian Heimes510711d2008-01-30 11:57:58 +0000565 def test_count(self):
566 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200567 i = 105
568 p = 112
569 w = 119
570
Christian Heimes510711d2008-01-30 11:57:58 +0000571 self.assertEqual(b.count(b'i'), 4)
572 self.assertEqual(b.count(b'ss'), 2)
573 self.assertEqual(b.count(b'w'), 0)
574
Antoine Pitrouac65d962011-10-20 23:54:17 +0200575 self.assertEqual(b.count(i), 4)
576 self.assertEqual(b.count(w), 0)
577
578 self.assertEqual(b.count(b'i', 6), 2)
579 self.assertEqual(b.count(b'p', 6), 2)
580 self.assertEqual(b.count(b'i', 1, 3), 1)
581 self.assertEqual(b.count(b'p', 7, 9), 1)
582
583 self.assertEqual(b.count(i, 6), 2)
584 self.assertEqual(b.count(p, 6), 2)
585 self.assertEqual(b.count(i, 1, 3), 1)
586 self.assertEqual(b.count(p, 7, 9), 1)
587
Christian Heimes510711d2008-01-30 11:57:58 +0000588 def test_startswith(self):
589 b = self.type2test(b'hello')
590 self.assertFalse(self.type2test().startswith(b"anything"))
591 self.assertTrue(b.startswith(b"hello"))
592 self.assertTrue(b.startswith(b"hel"))
593 self.assertTrue(b.startswith(b"h"))
594 self.assertFalse(b.startswith(b"hellow"))
595 self.assertFalse(b.startswith(b"ha"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300596 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300597 b.startswith([b'h'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300598 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300599 self.assertIn('bytes', exc)
600 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000601
602 def test_endswith(self):
603 b = self.type2test(b'hello')
604 self.assertFalse(bytearray().endswith(b"anything"))
605 self.assertTrue(b.endswith(b"hello"))
606 self.assertTrue(b.endswith(b"llo"))
607 self.assertTrue(b.endswith(b"o"))
608 self.assertFalse(b.endswith(b"whello"))
609 self.assertFalse(b.endswith(b"no"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300610 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300611 b.endswith([b'o'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300612 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300613 self.assertIn('bytes', exc)
614 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000615
616 def test_find(self):
617 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200618 i = 105
619 w = 119
620
Christian Heimes510711d2008-01-30 11:57:58 +0000621 self.assertEqual(b.find(b'ss'), 2)
Christian Heimes510711d2008-01-30 11:57:58 +0000622 self.assertEqual(b.find(b'w'), -1)
623 self.assertEqual(b.find(b'mississippian'), -1)
624
Antoine Pitrouac65d962011-10-20 23:54:17 +0200625 self.assertEqual(b.find(i), 1)
626 self.assertEqual(b.find(w), -1)
627
628 self.assertEqual(b.find(b'ss', 3), 5)
629 self.assertEqual(b.find(b'ss', 1, 7), 2)
630 self.assertEqual(b.find(b'ss', 1, 3), -1)
631
632 self.assertEqual(b.find(i, 6), 7)
633 self.assertEqual(b.find(i, 1, 3), 1)
634 self.assertEqual(b.find(w, 1, 3), -1)
635
Victor Stinnerf8eac002011-12-18 01:17:41 +0100636 for index in (-1, 256, sys.maxsize + 1):
637 self.assertRaisesRegex(
638 ValueError, r'byte must be in range\(0, 256\)',
639 b.find, index)
640
Christian Heimes510711d2008-01-30 11:57:58 +0000641 def test_rfind(self):
642 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200643 i = 105
644 w = 119
645
Christian Heimes510711d2008-01-30 11:57:58 +0000646 self.assertEqual(b.rfind(b'ss'), 5)
Christian Heimes510711d2008-01-30 11:57:58 +0000647 self.assertEqual(b.rfind(b'w'), -1)
648 self.assertEqual(b.rfind(b'mississippian'), -1)
649
Antoine Pitrouac65d962011-10-20 23:54:17 +0200650 self.assertEqual(b.rfind(i), 10)
651 self.assertEqual(b.rfind(w), -1)
652
653 self.assertEqual(b.rfind(b'ss', 3), 5)
654 self.assertEqual(b.rfind(b'ss', 0, 6), 2)
655
656 self.assertEqual(b.rfind(i, 1, 3), 1)
657 self.assertEqual(b.rfind(i, 3, 9), 7)
658 self.assertEqual(b.rfind(w, 1, 3), -1)
659
Christian Heimes510711d2008-01-30 11:57:58 +0000660 def test_index(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200661 b = self.type2test(b'mississippi')
662 i = 105
663 w = 119
664
665 self.assertEqual(b.index(b'ss'), 2)
666 self.assertRaises(ValueError, b.index, b'w')
667 self.assertRaises(ValueError, b.index, b'mississippian')
668
669 self.assertEqual(b.index(i), 1)
670 self.assertRaises(ValueError, b.index, w)
671
672 self.assertEqual(b.index(b'ss', 3), 5)
673 self.assertEqual(b.index(b'ss', 1, 7), 2)
674 self.assertRaises(ValueError, b.index, b'ss', 1, 3)
675
676 self.assertEqual(b.index(i, 6), 7)
677 self.assertEqual(b.index(i, 1, 3), 1)
678 self.assertRaises(ValueError, b.index, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000679
680 def test_rindex(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200681 b = self.type2test(b'mississippi')
682 i = 105
683 w = 119
684
685 self.assertEqual(b.rindex(b'ss'), 5)
686 self.assertRaises(ValueError, b.rindex, b'w')
687 self.assertRaises(ValueError, b.rindex, b'mississippian')
688
689 self.assertEqual(b.rindex(i), 10)
690 self.assertRaises(ValueError, b.rindex, w)
691
692 self.assertEqual(b.rindex(b'ss', 3), 5)
693 self.assertEqual(b.rindex(b'ss', 0, 6), 2)
694
695 self.assertEqual(b.rindex(i, 1, 3), 1)
696 self.assertEqual(b.rindex(i, 3, 9), 7)
697 self.assertRaises(ValueError, b.rindex, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000698
Ethan Furmanb95b5612015-01-23 20:05:18 -0800699 def test_mod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300700 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800701 orig = b
702 b = b % b'world'
703 self.assertEqual(b, b'hello, world!')
704 self.assertEqual(orig, b'hello, %b!')
705 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300706 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800707 a = b % (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200708 self.assertEqual(a, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200709 self.assertIs(type(a), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800710 # issue 29714
711 b = self.type2test(b'hello,\x00%b!')
712 b = b % b'world'
713 self.assertEqual(b, b'hello,\x00world!')
714 self.assertIs(type(b), self.type2test)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800715
716 def test_imod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300717 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800718 orig = b
719 b %= b'world'
720 self.assertEqual(b, b'hello, world!')
721 self.assertEqual(orig, b'hello, %b!')
722 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300723 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800724 b %= (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200725 self.assertEqual(b, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200726 self.assertIs(type(b), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800727 # issue 29714
728 b = self.type2test(b'hello,\x00%b!')
729 b %= b'world'
730 self.assertEqual(b, b'hello,\x00world!')
731 self.assertIs(type(b), self.type2test)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300732
733 def test_rmod(self):
734 with self.assertRaises(TypeError):
735 object() % self.type2test(b'abc')
736 self.assertIs(self.type2test(b'abc').__rmod__('%r'), NotImplemented)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800737
Christian Heimes510711d2008-01-30 11:57:58 +0000738 def test_replace(self):
739 b = self.type2test(b'mississippi')
740 self.assertEqual(b.replace(b'i', b'a'), b'massassappa')
741 self.assertEqual(b.replace(b'ss', b'x'), b'mixixippi')
742
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300743 def test_replace_int_error(self):
744 self.assertRaises(TypeError, self.type2test(b'a b').replace, 32, b'')
745
Christian Heimes510711d2008-01-30 11:57:58 +0000746 def test_split_string_error(self):
747 self.assertRaises(TypeError, self.type2test(b'a b').split, ' ')
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300748 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, ' ')
749
750 def test_split_int_error(self):
751 self.assertRaises(TypeError, self.type2test(b'a b').split, 32)
752 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000753
Antoine Pitrouf8453022010-01-12 22:05:42 +0000754 def test_split_unicodewhitespace(self):
Martin Panter0d0db6c2016-04-10 08:45:26 +0000755 for b in (b'a\x1Cb', b'a\x1Db', b'a\x1Eb', b'a\x1Fb'):
756 b = self.type2test(b)
757 self.assertEqual(b.split(), [b])
Antoine Pitrouf8453022010-01-12 22:05:42 +0000758 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
759 self.assertEqual(b.split(), [b'\x1c\x1d\x1e\x1f'])
760
Christian Heimes510711d2008-01-30 11:57:58 +0000761 def test_rsplit_unicodewhitespace(self):
762 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
Christian Heimes510711d2008-01-30 11:57:58 +0000763 self.assertEqual(b.rsplit(), [b'\x1c\x1d\x1e\x1f'])
764
765 def test_partition(self):
766 b = self.type2test(b'mississippi')
767 self.assertEqual(b.partition(b'ss'), (b'mi', b'ss', b'issippi'))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000768 self.assertEqual(b.partition(b'w'), (b'mississippi', b'', b''))
Christian Heimes510711d2008-01-30 11:57:58 +0000769
770 def test_rpartition(self):
771 b = self.type2test(b'mississippi')
772 self.assertEqual(b.rpartition(b'ss'), (b'missi', b'ss', b'ippi'))
773 self.assertEqual(b.rpartition(b'i'), (b'mississipp', b'i', b''))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000774 self.assertEqual(b.rpartition(b'w'), (b'', b'', b'mississippi'))
Christian Heimes510711d2008-01-30 11:57:58 +0000775
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300776 def test_partition_string_error(self):
777 self.assertRaises(TypeError, self.type2test(b'a b').partition, ' ')
778 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, ' ')
779
780 def test_partition_int_error(self):
781 self.assertRaises(TypeError, self.type2test(b'a b').partition, 32)
782 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, 32)
783
Christian Heimes510711d2008-01-30 11:57:58 +0000784 def test_pickling(self):
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +0000785 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Christian Heimes510711d2008-01-30 11:57:58 +0000786 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
787 b = self.type2test(b)
788 ps = pickle.dumps(b, proto)
789 q = pickle.loads(ps)
790 self.assertEqual(b, q)
791
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000792 def test_iterator_pickling(self):
Serhiy Storchakabad12572014-12-15 14:03:42 +0200793 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
794 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
795 it = itorg = iter(self.type2test(b))
796 data = list(self.type2test(b))
797 d = pickle.dumps(it, proto)
798 it = pickle.loads(d)
799 self.assertEqual(type(itorg), type(it))
800 self.assertEqual(list(it), data)
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000801
Serhiy Storchakabad12572014-12-15 14:03:42 +0200802 it = pickle.loads(d)
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200803 if not b:
Serhiy Storchakabad12572014-12-15 14:03:42 +0200804 continue
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200805 next(it)
Serhiy Storchakabad12572014-12-15 14:03:42 +0200806 d = pickle.dumps(it, proto)
807 it = pickle.loads(d)
808 self.assertEqual(list(it), data[1:])
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000809
Christian Heimes510711d2008-01-30 11:57:58 +0000810 def test_strip_bytearray(self):
811 self.assertEqual(self.type2test(b'abc').strip(memoryview(b'ac')), b'b')
812 self.assertEqual(self.type2test(b'abc').lstrip(memoryview(b'ac')), b'bc')
813 self.assertEqual(self.type2test(b'abc').rstrip(memoryview(b'ac')), b'ab')
814
815 def test_strip_string_error(self):
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300816 self.assertRaises(TypeError, self.type2test(b'abc').strip, 'ac')
817 self.assertRaises(TypeError, self.type2test(b'abc').lstrip, 'ac')
818 self.assertRaises(TypeError, self.type2test(b'abc').rstrip, 'ac')
819
820 def test_strip_int_error(self):
821 self.assertRaises(TypeError, self.type2test(b' abc ').strip, 32)
822 self.assertRaises(TypeError, self.type2test(b' abc ').lstrip, 32)
823 self.assertRaises(TypeError, self.type2test(b' abc ').rstrip, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000824
Eli Bendersky906b88f2011-07-29 07:05:08 +0300825 def test_center(self):
826 # Fill character can be either bytes or bytearray (issue 12380)
827 b = self.type2test(b'abc')
828 for fill_type in (bytes, bytearray):
829 self.assertEqual(b.center(7, fill_type(b'-')),
830 self.type2test(b'--abc--'))
831
832 def test_ljust(self):
833 # Fill character can be either bytes or bytearray (issue 12380)
834 b = self.type2test(b'abc')
835 for fill_type in (bytes, bytearray):
836 self.assertEqual(b.ljust(7, fill_type(b'-')),
837 self.type2test(b'abc----'))
838
839 def test_rjust(self):
840 # Fill character can be either bytes or bytearray (issue 12380)
841 b = self.type2test(b'abc')
842 for fill_type in (bytes, bytearray):
843 self.assertEqual(b.rjust(7, fill_type(b'-')),
844 self.type2test(b'----abc'))
845
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300846 def test_xjust_int_error(self):
847 self.assertRaises(TypeError, self.type2test(b'abc').center, 7, 32)
848 self.assertRaises(TypeError, self.type2test(b'abc').ljust, 7, 32)
849 self.assertRaises(TypeError, self.type2test(b'abc').rjust, 7, 32)
850
Christian Heimes510711d2008-01-30 11:57:58 +0000851 def test_ord(self):
852 b = self.type2test(b'\0A\x7f\x80\xff')
853 self.assertEqual([ord(b[i:i+1]) for i in range(len(b))],
854 [0, 65, 127, 128, 255])
855
Georg Brandlabc38772009-04-12 15:51:51 +0000856 def test_maketrans(self):
857 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 +0000858 self.assertEqual(self.type2test.maketrans(b'abc', b'xyz'), transtable)
Antoine Pitrou403ce782009-10-14 17:14:16 +0000859 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'
860 self.assertEqual(self.type2test.maketrans(b'\375\376\377', b'xyz'), transtable)
Georg Brandlabc38772009-04-12 15:51:51 +0000861 self.assertRaises(ValueError, self.type2test.maketrans, b'abc', b'xyzq')
862 self.assertRaises(TypeError, self.type2test.maketrans, 'abc', 'def')
863
Jesus Ceaac451502011-04-20 17:09:23 +0200864 def test_none_arguments(self):
865 # issue 11828
866 b = self.type2test(b'hello')
867 l = self.type2test(b'l')
868 h = self.type2test(b'h')
869 x = self.type2test(b'x')
870 o = self.type2test(b'o')
871
872 self.assertEqual(2, b.find(l, None))
873 self.assertEqual(3, b.find(l, -2, None))
874 self.assertEqual(2, b.find(l, None, -2))
875 self.assertEqual(0, b.find(h, None, None))
876
877 self.assertEqual(3, b.rfind(l, None))
878 self.assertEqual(3, b.rfind(l, -2, None))
879 self.assertEqual(2, b.rfind(l, None, -2))
880 self.assertEqual(0, b.rfind(h, None, None))
881
882 self.assertEqual(2, b.index(l, None))
883 self.assertEqual(3, b.index(l, -2, None))
884 self.assertEqual(2, b.index(l, None, -2))
885 self.assertEqual(0, b.index(h, None, None))
886
887 self.assertEqual(3, b.rindex(l, None))
888 self.assertEqual(3, b.rindex(l, -2, None))
889 self.assertEqual(2, b.rindex(l, None, -2))
890 self.assertEqual(0, b.rindex(h, None, None))
891
892 self.assertEqual(2, b.count(l, None))
893 self.assertEqual(1, b.count(l, -2, None))
894 self.assertEqual(1, b.count(l, None, -2))
895 self.assertEqual(0, b.count(x, None, None))
896
897 self.assertEqual(True, b.endswith(o, None))
898 self.assertEqual(True, b.endswith(o, -2, None))
899 self.assertEqual(True, b.endswith(l, None, -2))
900 self.assertEqual(False, b.endswith(x, None, None))
901
902 self.assertEqual(True, b.startswith(h, None))
903 self.assertEqual(True, b.startswith(l, -2, None))
904 self.assertEqual(True, b.startswith(h, None, -2))
905 self.assertEqual(False, b.startswith(x, None, None))
906
Antoine Pitrouac65d962011-10-20 23:54:17 +0200907 def test_integer_arguments_out_of_byte_range(self):
908 b = self.type2test(b'hello')
909
910 for method in (b.count, b.find, b.index, b.rfind, b.rindex):
911 self.assertRaises(ValueError, method, -1)
912 self.assertRaises(ValueError, method, 256)
913 self.assertRaises(ValueError, method, 9999)
914
Jesus Ceaac451502011-04-20 17:09:23 +0200915 def test_find_etc_raise_correct_error_messages(self):
916 # issue 11828
917 b = self.type2test(b'hello')
918 x = self.type2test(b'x')
Ezio Melottiaf928422011-04-20 21:56:21 +0300919 self.assertRaisesRegex(TypeError, r'\bfind\b', b.find,
Jesus Ceaac451502011-04-20 17:09:23 +0200920 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300921 self.assertRaisesRegex(TypeError, r'\brfind\b', b.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +0200922 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300923 self.assertRaisesRegex(TypeError, r'\bindex\b', b.index,
Jesus Ceaac451502011-04-20 17:09:23 +0200924 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300925 self.assertRaisesRegex(TypeError, r'\brindex\b', b.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +0200926 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300927 self.assertRaisesRegex(TypeError, r'\bcount\b', b.count,
Jesus Ceaac451502011-04-20 17:09:23 +0200928 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300929 self.assertRaisesRegex(TypeError, r'\bstartswith\b', b.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200930 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300931 self.assertRaisesRegex(TypeError, r'\bendswith\b', b.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200932 x, None, None, None)
933
Serhiy Storchakafbb1c5e2016-03-30 20:40:02 +0300934 def test_free_after_iterating(self):
935 test.support.check_free_after_iterating(self, iter, self.type2test)
936 test.support.check_free_after_iterating(self, reversed, self.type2test)
937
Martin Panter1b6c6da2016-08-27 08:35:02 +0000938 def test_translate(self):
939 b = self.type2test(b'hello')
940 rosetta = bytearray(range(256))
941 rosetta[ord('o')] = ord('e')
942
943 self.assertRaises(TypeError, b.translate)
944 self.assertRaises(TypeError, b.translate, None, None)
945 self.assertRaises(ValueError, b.translate, bytes(range(255)))
946
947 c = b.translate(rosetta, b'hello')
948 self.assertEqual(b, b'hello')
949 self.assertIsInstance(c, self.type2test)
950
951 c = b.translate(rosetta)
952 d = b.translate(rosetta, b'')
953 self.assertEqual(c, d)
954 self.assertEqual(c, b'helle')
955
956 c = b.translate(rosetta, b'l')
957 self.assertEqual(c, b'hee')
958 c = b.translate(None, b'e')
959 self.assertEqual(c, b'hllo')
960
961 # test delete as a keyword argument
962 c = b.translate(rosetta, delete=b'')
963 self.assertEqual(c, b'helle')
964 c = b.translate(rosetta, delete=b'l')
965 self.assertEqual(c, b'hee')
966 c = b.translate(None, delete=b'e')
967 self.assertEqual(c, b'hllo')
968
Sergey Fedoseev92709a22019-09-09 21:28:34 +0500969 def test_sq_item(self):
970 _testcapi = test.support.import_module('_testcapi')
971 obj = self.type2test((42,))
972 with self.assertRaises(IndexError):
973 _testcapi.sequence_getitem(obj, -2)
974 with self.assertRaises(IndexError):
975 _testcapi.sequence_getitem(obj, 1)
976 self.assertEqual(_testcapi.sequence_getitem(obj, 0), 42)
977
Christian Heimes510711d2008-01-30 11:57:58 +0000978
Ezio Melotti0dceb562013-01-10 07:43:26 +0200979class BytesTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +0000980 type2test = bytes
981
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400982 def test_getitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200983 b = b'python'
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400984 msg = "byte indices must be integers or slices"
985 with self.assertRaisesRegex(TypeError, msg):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200986 b['a']
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400987
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000988 def test_buffer_is_readonly(self):
Victor Stinner76119642014-07-28 22:07:07 +0200989 fd = os.open(__file__, os.O_RDONLY)
Antoine Pitrou47d305d2008-08-16 23:28:44 +0000990 with open(fd, "rb", buffering=0) as f:
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000991 self.assertRaises(TypeError, f.readinto, b"")
992
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000993 def test_custom(self):
994 class A:
995 def __bytes__(self):
996 return b'abc'
997 self.assertEqual(bytes(A()), b'abc')
998 class A: pass
999 self.assertRaises(TypeError, bytes, A())
1000 class A:
1001 def __bytes__(self):
1002 return None
1003 self.assertRaises(TypeError, bytes, A())
Benjamin Peterson5ff3f732012-12-19 15:27:41 -06001004 class A:
1005 def __bytes__(self):
1006 return b'a'
1007 def __index__(self):
1008 return 42
1009 self.assertEqual(bytes(A()), b'a')
Serhiy Storchaka5aac3ed2015-12-20 16:36:34 +02001010 # Issue #25766
1011 class A(str):
1012 def __bytes__(self):
1013 return b'abc'
1014 self.assertEqual(bytes(A('\u20ac')), b'abc')
1015 self.assertEqual(bytes(A('\u20ac'), 'iso8859-15'), b'\xa4')
Serhiy Storchaka15095802015-11-25 15:47:01 +02001016 # Issue #24731
1017 class A:
1018 def __bytes__(self):
1019 return OtherBytesSubclass(b'abc')
1020 self.assertEqual(bytes(A()), b'abc')
1021 self.assertIs(type(bytes(A())), OtherBytesSubclass)
1022 self.assertEqual(BytesSubclass(A()), b'abc')
1023 self.assertIs(type(BytesSubclass(A())), BytesSubclass)
Benjamin Petersonc15a0732008-08-26 16:46:47 +00001024
Victor Stinner29e762c2011-01-05 03:33:28 +00001025 # Test PyBytes_FromFormat()
1026 def test_from_format(self):
Victor Stinner7ab986d2015-10-14 02:55:12 +02001027 ctypes = test.support.import_module('ctypes')
Victor Stinner03dab782015-10-14 00:21:35 +02001028 _testcapi = test.support.import_module('_testcapi')
1029 from ctypes import pythonapi, py_object
1030 from ctypes import (
1031 c_int, c_uint,
1032 c_long, c_ulong,
1033 c_size_t, c_ssize_t,
1034 c_char_p)
1035
Victor Stinner29e762c2011-01-05 03:33:28 +00001036 PyBytes_FromFormat = pythonapi.PyBytes_FromFormat
Ronald Oussorene8b1c032020-11-22 11:18:40 +01001037 PyBytes_FromFormat.argtypes = (c_char_p,)
Victor Stinner29e762c2011-01-05 03:33:28 +00001038 PyBytes_FromFormat.restype = py_object
1039
Victor Stinner03dab782015-10-14 00:21:35 +02001040 # basic tests
Victor Stinner29e762c2011-01-05 03:33:28 +00001041 self.assertEqual(PyBytes_FromFormat(b'format'),
1042 b'format')
Victor Stinner03dab782015-10-14 00:21:35 +02001043 self.assertEqual(PyBytes_FromFormat(b'Hello %s !', b'world'),
1044 b'Hello world !')
Victor Stinner29e762c2011-01-05 03:33:28 +00001045
Victor Stinner03dab782015-10-14 00:21:35 +02001046 # test formatters
1047 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(0)),
1048 b'c=\0')
1049 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(ord('@'))),
1050 b'c=@')
1051 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(255)),
1052 b'c=\xff')
1053 self.assertEqual(PyBytes_FromFormat(b'd=%d ld=%ld zd=%zd',
1054 c_int(1), c_long(2),
1055 c_size_t(3)),
1056 b'd=1 ld=2 zd=3')
1057 self.assertEqual(PyBytes_FromFormat(b'd=%d ld=%ld zd=%zd',
1058 c_int(-1), c_long(-2),
1059 c_size_t(-3)),
1060 b'd=-1 ld=-2 zd=-3')
1061 self.assertEqual(PyBytes_FromFormat(b'u=%u lu=%lu zu=%zu',
1062 c_uint(123), c_ulong(456),
1063 c_size_t(789)),
1064 b'u=123 lu=456 zu=789')
1065 self.assertEqual(PyBytes_FromFormat(b'i=%i', c_int(123)),
1066 b'i=123')
1067 self.assertEqual(PyBytes_FromFormat(b'i=%i', c_int(-123)),
1068 b'i=-123')
1069 self.assertEqual(PyBytes_FromFormat(b'x=%x', c_int(0xabc)),
1070 b'x=abc')
Victor Stinner83ff8a62015-10-14 15:28:59 +02001071
1072 sizeof_ptr = ctypes.sizeof(c_char_p)
1073
1074 if os.name == 'nt':
1075 # Windows (MSCRT)
1076 ptr_format = '0x%0{}X'.format(2 * sizeof_ptr)
1077 def ptr_formatter(ptr):
1078 return (ptr_format % ptr)
1079 else:
1080 # UNIX (glibc)
1081 def ptr_formatter(ptr):
1082 return '%#x' % ptr
1083
Victor Stinner7ab986d2015-10-14 02:55:12 +02001084 ptr = 0xabcdef
Victor Stinner83ff8a62015-10-14 15:28:59 +02001085 self.assertEqual(PyBytes_FromFormat(b'ptr=%p', c_char_p(ptr)),
1086 ('ptr=' + ptr_formatter(ptr)).encode('ascii'))
Victor Stinner03dab782015-10-14 00:21:35 +02001087 self.assertEqual(PyBytes_FromFormat(b's=%s', c_char_p(b'cstr')),
1088 b's=cstr')
1089
1090 # test minimum and maximum integer values
1091 size_max = c_size_t(-1).value
1092 for formatstr, ctypes_type, value, py_formatter in (
1093 (b'%d', c_int, _testcapi.INT_MIN, str),
1094 (b'%d', c_int, _testcapi.INT_MAX, str),
1095 (b'%ld', c_long, _testcapi.LONG_MIN, str),
1096 (b'%ld', c_long, _testcapi.LONG_MAX, str),
1097 (b'%lu', c_ulong, _testcapi.ULONG_MAX, str),
1098 (b'%zd', c_ssize_t, _testcapi.PY_SSIZE_T_MIN, str),
1099 (b'%zd', c_ssize_t, _testcapi.PY_SSIZE_T_MAX, str),
1100 (b'%zu', c_size_t, size_max, str),
Victor Stinner83ff8a62015-10-14 15:28:59 +02001101 (b'%p', c_char_p, size_max, ptr_formatter),
Victor Stinner03dab782015-10-14 00:21:35 +02001102 ):
1103 self.assertEqual(PyBytes_FromFormat(formatstr, ctypes_type(value)),
1104 py_formatter(value).encode('ascii')),
1105
1106 # width and precision (width is currently ignored)
1107 self.assertEqual(PyBytes_FromFormat(b'%5s', b'a'),
1108 b'a')
1109 self.assertEqual(PyBytes_FromFormat(b'%.3s', b'abcdef'),
1110 b'abc')
1111
1112 # '%%' formatter
1113 self.assertEqual(PyBytes_FromFormat(b'%%'),
1114 b'%')
1115 self.assertEqual(PyBytes_FromFormat(b'[%%]'),
1116 b'[%]')
1117 self.assertEqual(PyBytes_FromFormat(b'%%%c', c_int(ord('_'))),
1118 b'%_')
1119 self.assertEqual(PyBytes_FromFormat(b'%%s'),
1120 b'%s')
1121
1122 # Invalid formats and partial formatting
Victor Stinner29e762c2011-01-05 03:33:28 +00001123 self.assertEqual(PyBytes_FromFormat(b'%'), b'%')
Victor Stinner03dab782015-10-14 00:21:35 +02001124 self.assertEqual(PyBytes_FromFormat(b'x=%i y=%', c_int(2), c_int(3)),
1125 b'x=2 y=%')
Victor Stinner29e762c2011-01-05 03:33:28 +00001126
Victor Stinner03dab782015-10-14 00:21:35 +02001127 # Issue #19969: %c must raise OverflowError for values
1128 # not in the range [0; 255]
Victor Stinnerc9362cf2013-12-13 12:14:44 +01001129 self.assertRaises(OverflowError,
1130 PyBytes_FromFormat, b'%c', c_int(-1))
1131 self.assertRaises(OverflowError,
1132 PyBytes_FromFormat, b'%c', c_int(256))
1133
Serhiy Storchaka44cc4822019-01-12 09:22:29 +02001134 # Issue #33817: empty strings
1135 self.assertEqual(PyBytes_FromFormat(b''),
1136 b'')
1137 self.assertEqual(PyBytes_FromFormat(b'%s', b''),
1138 b'')
1139
Guido van Rossum97c1adf2016-08-18 09:22:23 -07001140 def test_bytes_blocking(self):
1141 class IterationBlocked(list):
1142 __bytes__ = None
1143 i = [0, 1, 2, 3]
1144 self.assertEqual(bytes(i), b'\x00\x01\x02\x03')
1145 self.assertRaises(TypeError, bytes, IterationBlocked(i))
1146
1147 # At least in CPython, because bytes.__new__ and the C API
1148 # PyBytes_FromObject have different fallback rules, integer
1149 # fallback is handled specially, so test separately.
1150 class IntBlocked(int):
1151 __bytes__ = None
1152 self.assertEqual(bytes(3), b'\0\0\0')
1153 self.assertRaises(TypeError, bytes, IntBlocked(3))
1154
1155 # While there is no separately-defined rule for handling bytes
1156 # subclasses differently from other buffer-interface classes,
1157 # an implementation may well special-case them (as CPython 2.x
1158 # str did), so test them separately.
1159 class BytesSubclassBlocked(bytes):
1160 __bytes__ = None
1161 self.assertEqual(bytes(b'ab'), b'ab')
1162 self.assertRaises(TypeError, bytes, BytesSubclassBlocked(b'ab'))
1163
1164 class BufferBlocked(bytearray):
1165 __bytes__ = None
1166 ba, bb = bytearray(b'ab'), BufferBlocked(b'ab')
1167 self.assertEqual(bytes(ba), b'ab')
1168 self.assertRaises(TypeError, bytes, bb)
1169
Antoine Pitrou2f89aa62008-08-02 21:02:48 +00001170
Ezio Melotti0dceb562013-01-10 07:43:26 +02001171class ByteArrayTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001172 type2test = bytearray
Guido van Rossumebea9be2007-04-09 00:49:13 +00001173
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001174 def test_getitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001175 b = bytearray(b'python')
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001176 msg = "bytearray indices must be integers or slices"
1177 with self.assertRaisesRegex(TypeError, msg):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001178 b['a']
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001179
1180 def test_setitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001181 b = bytearray(b'python')
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001182 msg = "bytearray indices must be integers or slices"
1183 with self.assertRaisesRegex(TypeError, msg):
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001184 b['a'] = "python"
1185
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001186 def test_nohash(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001187 self.assertRaises(TypeError, hash, bytearray())
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001188
Guido van Rossum254348e2007-11-21 19:29:53 +00001189 def test_bytearray_api(self):
Guido van Rossum75d6f1a2007-05-24 17:52:00 +00001190 short_sample = b"Hello world\n"
Guido van Rossum98297ee2007-11-06 21:34:58 +00001191 sample = short_sample + b"\0"*(20 - len(short_sample))
Guido van Rossumd624f182006-04-24 13:47:05 +00001192 tfn = tempfile.mktemp()
1193 try:
1194 # Prepare
1195 with open(tfn, "wb") as f:
1196 f.write(short_sample)
1197 # Test readinto
1198 with open(tfn, "rb") as f:
Guido van Rossum254348e2007-11-21 19:29:53 +00001199 b = bytearray(20)
Guido van Rossumd624f182006-04-24 13:47:05 +00001200 n = f.readinto(b)
1201 self.assertEqual(n, len(short_sample))
Guido van Rossumc1f779c2007-07-03 08:25:58 +00001202 self.assertEqual(list(b), list(sample))
Guido van Rossumd624f182006-04-24 13:47:05 +00001203 # Test writing in binary mode
1204 with open(tfn, "wb") as f:
1205 f.write(b)
1206 with open(tfn, "rb") as f:
1207 self.assertEqual(f.read(), sample)
Guido van Rossum13633bb2007-04-13 18:42:35 +00001208 # Text mode is ambiguous; don't test
Guido van Rossumd624f182006-04-24 13:47:05 +00001209 finally:
1210 try:
1211 os.remove(tfn)
Andrew Svetlov8b33dd82012-12-24 19:58:48 +02001212 except OSError:
Guido van Rossumd624f182006-04-24 13:47:05 +00001213 pass
1214
Neal Norwitz6968b052007-02-27 19:02:19 +00001215 def test_reverse(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001216 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +00001217 self.assertEqual(b.reverse(), None)
1218 self.assertEqual(b, b'olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +00001219 b = bytearray(b'hello1') # test even number of items
Neal Norwitz6968b052007-02-27 19:02:19 +00001220 b.reverse()
1221 self.assertEqual(b, b'1olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +00001222 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +00001223 b.reverse()
1224 self.assertFalse(b)
1225
Eli Bendersky4db28d32011-03-03 18:21:02 +00001226 def test_clear(self):
1227 b = bytearray(b'python')
1228 b.clear()
1229 self.assertEqual(b, b'')
1230
1231 b = bytearray(b'')
1232 b.clear()
1233 self.assertEqual(b, b'')
1234
1235 b = bytearray(b'')
1236 b.append(ord('r'))
1237 b.clear()
1238 b.append(ord('p'))
1239 self.assertEqual(b, b'p')
1240
1241 def test_copy(self):
1242 b = bytearray(b'abc')
1243 bb = b.copy()
1244 self.assertEqual(bb, b'abc')
1245
1246 b = bytearray(b'')
1247 bb = b.copy()
1248 self.assertEqual(bb, b'')
1249
1250 # test that it's indeed a copy and not a reference
1251 b = bytearray(b'abc')
1252 bb = b.copy()
1253 self.assertEqual(b, bb)
1254 self.assertIsNot(b, bb)
1255 bb.append(ord('d'))
1256 self.assertEqual(bb, b'abcd')
1257 self.assertEqual(b, b'abc')
1258
Guido van Rossumd624f182006-04-24 13:47:05 +00001259 def test_regexps(self):
1260 def by(s):
Guido van Rossum254348e2007-11-21 19:29:53 +00001261 return bytearray(map(ord, s))
Guido van Rossumd624f182006-04-24 13:47:05 +00001262 b = by("Hello, world")
Antoine Pitroufd036452008-08-19 17:56:33 +00001263 self.assertEqual(re.findall(br"\w+", b), [by("Hello"), by("world")])
Guido van Rossumd624f182006-04-24 13:47:05 +00001264
1265 def test_setitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001266 b = bytearray([1, 2, 3])
Guido van Rossumd624f182006-04-24 13:47:05 +00001267 b[1] = 100
Guido van Rossum254348e2007-11-21 19:29:53 +00001268 self.assertEqual(b, bytearray([1, 100, 3]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001269 b[-1] = 200
Guido van Rossum254348e2007-11-21 19:29:53 +00001270 self.assertEqual(b, bytearray([1, 100, 200]))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001271 b[0] = Indexable(10)
Guido van Rossum254348e2007-11-21 19:29:53 +00001272 self.assertEqual(b, bytearray([10, 100, 200]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001273 try:
1274 b[3] = 0
1275 self.fail("Didn't raise IndexError")
1276 except IndexError:
1277 pass
1278 try:
1279 b[-10] = 0
1280 self.fail("Didn't raise IndexError")
1281 except IndexError:
1282 pass
1283 try:
1284 b[0] = 256
1285 self.fail("Didn't raise ValueError")
1286 except ValueError:
1287 pass
1288 try:
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001289 b[0] = Indexable(-1)
Guido van Rossumd624f182006-04-24 13:47:05 +00001290 self.fail("Didn't raise ValueError")
1291 except ValueError:
1292 pass
1293 try:
1294 b[0] = None
1295 self.fail("Didn't raise TypeError")
1296 except TypeError:
1297 pass
1298
1299 def test_delitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001300 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +00001301 del b[0]
Guido van Rossum254348e2007-11-21 19:29:53 +00001302 self.assertEqual(b, bytearray(range(1, 10)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001303 del b[-1]
Guido van Rossum254348e2007-11-21 19:29:53 +00001304 self.assertEqual(b, bytearray(range(1, 9)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001305 del b[4]
Guido van Rossum254348e2007-11-21 19:29:53 +00001306 self.assertEqual(b, bytearray([1, 2, 3, 4, 6, 7, 8]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001307
1308 def test_setslice(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001309 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +00001310 self.assertEqual(list(b), list(range(10)))
1311
Guido van Rossum254348e2007-11-21 19:29:53 +00001312 b[0:5] = bytearray([1, 1, 1, 1, 1])
1313 self.assertEqual(b, bytearray([1, 1, 1, 1, 1, 5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001314
1315 del b[0:-5]
Guido van Rossum254348e2007-11-21 19:29:53 +00001316 self.assertEqual(b, bytearray([5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001317
Guido van Rossum254348e2007-11-21 19:29:53 +00001318 b[0:0] = bytearray([0, 1, 2, 3, 4])
1319 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001320
Guido van Rossum254348e2007-11-21 19:29:53 +00001321 b[-7:-3] = bytearray([100, 101])
1322 self.assertEqual(b, bytearray([0, 1, 2, 100, 101, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001323
1324 b[3:5] = [3, 4, 5, 6]
Guido van Rossum254348e2007-11-21 19:29:53 +00001325 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001326
Thomas Wouters9a6e62b2006-08-23 23:20:29 +00001327 b[3:0] = [42, 42, 42]
Guido van Rossum254348e2007-11-21 19:29:53 +00001328 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 +00001329
Ezio Melottic64bcbe2012-11-03 21:19:06 +02001330 b[3:] = b'foo'
1331 self.assertEqual(b, bytearray([0, 1, 2, 102, 111, 111]))
1332
1333 b[:3] = memoryview(b'foo')
1334 self.assertEqual(b, bytearray([102, 111, 111, 102, 111, 111]))
1335
1336 b[3:4] = []
1337 self.assertEqual(b, bytearray([102, 111, 111, 111, 111]))
1338
1339 for elem in [5, -5, 0, int(10e20), 'str', 2.3,
1340 ['a', 'b'], [b'a', b'b'], [[]]]:
1341 with self.assertRaises(TypeError):
1342 b[3:4] = elem
1343
1344 for elem in [[254, 255, 256], [-256, 9000]]:
1345 with self.assertRaises(ValueError):
1346 b[3:4] = elem
1347
Antoine Pitrou5df8a8a2013-10-05 21:12:18 +02001348 def test_setslice_extend(self):
1349 # Exercise the resizing logic (see issue #19087)
1350 b = bytearray(range(100))
1351 self.assertEqual(list(b), list(range(100)))
1352 del b[:10]
1353 self.assertEqual(list(b), list(range(10, 100)))
1354 b.extend(range(100, 110))
1355 self.assertEqual(list(b), list(range(10, 110)))
1356
Antoine Pitrou25454112015-05-19 20:52:27 +02001357 def test_fifo_overrun(self):
1358 # Test for issue #23985, a buffer overrun when implementing a FIFO
1359 # Build Python in pydebug mode for best results.
1360 b = bytearray(10)
1361 b.pop() # Defeat expanding buffer off-by-one quirk
1362 del b[:1] # Advance start pointer without reallocating
1363 b += bytes(2) # Append exactly the number of deleted bytes
1364 del b # Free memory buffer, allowing pydebug verification
1365
1366 def test_del_expand(self):
1367 # Reducing the size should not expand the buffer (issue #23985)
1368 b = bytearray(10)
1369 size = sys.getsizeof(b)
1370 del b[:1]
1371 self.assertLessEqual(sys.getsizeof(b), size)
1372
Thomas Wouters376446d2006-12-19 08:30:14 +00001373 def test_extended_set_del_slice(self):
Zackery Spytz14514d92019-05-17 01:13:03 -06001374 indices = (0, None, 1, 3, 19, 300, 1<<333, sys.maxsize,
1375 -1, -2, -31, -300)
Thomas Wouters376446d2006-12-19 08:30:14 +00001376 for start in indices:
1377 for stop in indices:
1378 # Skip invalid step 0
1379 for step in indices[1:]:
1380 L = list(range(255))
Guido van Rossum254348e2007-11-21 19:29:53 +00001381 b = bytearray(L)
Thomas Wouters376446d2006-12-19 08:30:14 +00001382 # Make sure we have a slice of exactly the right length,
1383 # but with different data.
1384 data = L[start:stop:step]
1385 data.reverse()
1386 L[start:stop:step] = data
1387 b[start:stop:step] = data
Ezio Melottib3aedd42010-11-20 19:04:17 +00001388 self.assertEqual(b, bytearray(L))
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001389
Thomas Wouters376446d2006-12-19 08:30:14 +00001390 del L[start:stop:step]
1391 del b[start:stop:step]
Ezio Melottib3aedd42010-11-20 19:04:17 +00001392 self.assertEqual(b, bytearray(L))
Thomas Wouters376446d2006-12-19 08:30:14 +00001393
Guido van Rossumd624f182006-04-24 13:47:05 +00001394 def test_setslice_trap(self):
1395 # This test verifies that we correctly handle assigning self
1396 # to a slice of self (the old Lambert Meertens trap).
Guido van Rossum254348e2007-11-21 19:29:53 +00001397 b = bytearray(range(256))
Guido van Rossumd624f182006-04-24 13:47:05 +00001398 b[8:] = b
Guido van Rossum254348e2007-11-21 19:29:53 +00001399 self.assertEqual(b, bytearray(list(range(8)) + list(range(256))))
Guido van Rossumd624f182006-04-24 13:47:05 +00001400
Guido van Rossum13e57212006-04-27 22:54:26 +00001401 def test_iconcat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001402 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001403 b1 = b
Guido van Rossum9c627722007-08-27 18:31:48 +00001404 b += b"def"
1405 self.assertEqual(b, b"abcdef")
Guido van Rossum13e57212006-04-27 22:54:26 +00001406 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001407 self.assertIs(b, b1)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001408 b += b"xyz"
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001409 self.assertEqual(b, b"abcdefxyz")
1410 try:
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001411 b += ""
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001412 except TypeError:
1413 pass
1414 else:
1415 self.fail("bytes += unicode didn't raise TypeError")
Guido van Rossum13e57212006-04-27 22:54:26 +00001416
1417 def test_irepeat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001418 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001419 b1 = b
1420 b *= 3
Guido van Rossum9c627722007-08-27 18:31:48 +00001421 self.assertEqual(b, b"abcabcabc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001422 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001423 self.assertIs(b, b1)
Guido van Rossum13e57212006-04-27 22:54:26 +00001424
1425 def test_irepeat_1char(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001426 b = bytearray(b"x")
Guido van Rossum13e57212006-04-27 22:54:26 +00001427 b1 = b
1428 b *= 100
Guido van Rossum98297ee2007-11-06 21:34:58 +00001429 self.assertEqual(b, b"x"*100)
Guido van Rossum13e57212006-04-27 22:54:26 +00001430 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001431 self.assertIs(b, b1)
Guido van Rossum13e57212006-04-27 22:54:26 +00001432
Guido van Rossum20188312006-05-05 15:15:40 +00001433 def test_alloc(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001434 b = bytearray()
Guido van Rossum20188312006-05-05 15:15:40 +00001435 alloc = b.__alloc__()
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001436 self.assertGreaterEqual(alloc, 0)
Guido van Rossum20188312006-05-05 15:15:40 +00001437 seq = [alloc]
1438 for i in range(100):
Guido van Rossum9c627722007-08-27 18:31:48 +00001439 b += b"x"
Guido van Rossum20188312006-05-05 15:15:40 +00001440 alloc = b.__alloc__()
Serhiy Storchaka7b6e3b92015-06-29 21:14:06 +03001441 self.assertGreater(alloc, len(b)) # including trailing null byte
Guido van Rossum20188312006-05-05 15:15:40 +00001442 if alloc not in seq:
1443 seq.append(alloc)
Guido van Rossum20188312006-05-05 15:15:40 +00001444
Serhiy Storchaka7b6e3b92015-06-29 21:14:06 +03001445 def test_init_alloc(self):
1446 b = bytearray()
1447 def g():
1448 for i in range(1, 100):
1449 yield i
1450 a = list(b)
1451 self.assertEqual(a, list(range(1, len(a)+1)))
1452 self.assertEqual(len(b), len(a))
1453 self.assertLessEqual(len(b), i)
1454 alloc = b.__alloc__()
1455 self.assertGreater(alloc, len(b)) # including trailing null byte
1456 b.__init__(g())
1457 self.assertEqual(list(b), list(range(1, 100)))
1458 self.assertEqual(len(b), 99)
1459 alloc = b.__alloc__()
1460 self.assertGreater(alloc, len(b))
1461
Neal Norwitz6968b052007-02-27 19:02:19 +00001462 def test_extend(self):
1463 orig = b'hello'
Guido van Rossum254348e2007-11-21 19:29:53 +00001464 a = bytearray(orig)
Neal Norwitz6968b052007-02-27 19:02:19 +00001465 a.extend(a)
1466 self.assertEqual(a, orig + orig)
1467 self.assertEqual(a[5:], orig)
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001468 a = bytearray(b'')
1469 # Test iterators that don't have a __length_hint__
1470 a.extend(map(int, orig * 25))
1471 a.extend(int(x) for x in orig * 25)
1472 self.assertEqual(a, orig * 50)
1473 self.assertEqual(a[-5:], orig)
1474 a = bytearray(b'')
1475 a.extend(iter(map(int, orig * 50)))
1476 self.assertEqual(a, orig * 50)
1477 self.assertEqual(a[-5:], orig)
1478 a = bytearray(b'')
1479 a.extend(list(map(int, orig * 50)))
1480 self.assertEqual(a, orig * 50)
1481 self.assertEqual(a[-5:], orig)
1482 a = bytearray(b'')
1483 self.assertRaises(ValueError, a.extend, [0, 1, 2, 256])
1484 self.assertRaises(ValueError, a.extend, [0, 1, 2, -1])
1485 self.assertEqual(len(a), 0)
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001486 a = bytearray(b'')
1487 a.extend([Indexable(ord('a'))])
1488 self.assertEqual(a, b'a')
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001489
Neal Norwitz6968b052007-02-27 19:02:19 +00001490 def test_remove(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001491 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +00001492 b.remove(ord('l'))
1493 self.assertEqual(b, b'helo')
1494 b.remove(ord('l'))
1495 self.assertEqual(b, b'heo')
1496 self.assertRaises(ValueError, lambda: b.remove(ord('l')))
1497 self.assertRaises(ValueError, lambda: b.remove(400))
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001498 self.assertRaises(TypeError, lambda: b.remove('e'))
Neal Norwitz6968b052007-02-27 19:02:19 +00001499 # remove first and last
1500 b.remove(ord('o'))
1501 b.remove(ord('h'))
1502 self.assertEqual(b, b'e')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001503 self.assertRaises(TypeError, lambda: b.remove(b'e'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001504 b.remove(Indexable(ord('e')))
1505 self.assertEqual(b, b'')
Neal Norwitz6968b052007-02-27 19:02:19 +00001506
Serhiy Storchaka7bf36da2016-05-16 22:15:38 +03001507 # test values outside of the ascii range: (0, 127)
1508 c = bytearray([126, 127, 128, 129])
1509 c.remove(127)
1510 self.assertEqual(c, bytes([126, 128, 129]))
1511 c.remove(129)
1512 self.assertEqual(c, bytes([126, 128]))
1513
Neal Norwitz6968b052007-02-27 19:02:19 +00001514 def test_pop(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001515 b = bytearray(b'world')
Neal Norwitz6968b052007-02-27 19:02:19 +00001516 self.assertEqual(b.pop(), ord('d'))
1517 self.assertEqual(b.pop(0), ord('w'))
1518 self.assertEqual(b.pop(-2), ord('r'))
1519 self.assertRaises(IndexError, lambda: b.pop(10))
Eli Bendersky1bc4f192011-03-04 04:55:25 +00001520 self.assertRaises(IndexError, lambda: bytearray().pop())
Mark Dickinson54a3db92009-09-06 10:19:23 +00001521 # test for issue #6846
1522 self.assertEqual(bytearray(b'\xff').pop(), 0xff)
Neal Norwitz6968b052007-02-27 19:02:19 +00001523
1524 def test_nosort(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001525 self.assertRaises(AttributeError, lambda: bytearray().sort())
Neal Norwitz6968b052007-02-27 19:02:19 +00001526
Neal Norwitz6968b052007-02-27 19:02:19 +00001527 def test_append(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001528 b = bytearray(b'hell')
Neal Norwitz6968b052007-02-27 19:02:19 +00001529 b.append(ord('o'))
1530 self.assertEqual(b, b'hello')
1531 self.assertEqual(b.append(100), None)
Guido van Rossum254348e2007-11-21 19:29:53 +00001532 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +00001533 b.append(ord('A'))
1534 self.assertEqual(len(b), 1)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001535 self.assertRaises(TypeError, lambda: b.append(b'o'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001536 b = bytearray()
1537 b.append(Indexable(ord('A')))
1538 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001539
1540 def test_insert(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001541 b = bytearray(b'msssspp')
Neal Norwitz6968b052007-02-27 19:02:19 +00001542 b.insert(1, ord('i'))
1543 b.insert(4, ord('i'))
1544 b.insert(-2, ord('i'))
1545 b.insert(1000, ord('i'))
1546 self.assertEqual(b, b'mississippi')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001547 self.assertRaises(TypeError, lambda: b.insert(0, b'1'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001548 b = bytearray()
1549 b.insert(0, Indexable(ord('A')))
1550 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001551
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001552 def test_copied(self):
1553 # Issue 4348. Make sure that operations that don't mutate the array
1554 # copy the bytes.
1555 b = bytearray(b'abc')
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001556 self.assertIsNot(b, b.replace(b'abc', b'cde', 0))
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001557
1558 t = bytearray([i for i in range(256)])
1559 x = bytearray(b'')
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001560 self.assertIsNot(x, x.translate(t))
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001561
Guido van Rossum254348e2007-11-21 19:29:53 +00001562 def test_partition_bytearray_doesnt_share_nullstring(self):
1563 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001564 self.assertEqual(b, b"")
1565 self.assertEqual(c, b"")
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001566 self.assertIsNot(b, c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001567 b += b"!"
1568 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001569 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001570 self.assertEqual(b, b"")
1571 self.assertEqual(c, b"")
1572 # Same for rpartition
Guido van Rossum254348e2007-11-21 19:29:53 +00001573 b, c, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001574 self.assertEqual(b, b"")
1575 self.assertEqual(c, b"")
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001576 self.assertIsNot(b, c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001577 b += b"!"
1578 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001579 c, b, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001580 self.assertEqual(b, b"")
1581 self.assertEqual(c, b"")
1582
Antoine Pitrou5504e892008-12-06 21:27:53 +00001583 def test_resize_forbidden(self):
1584 # #4509: can't resize a bytearray when there are buffer exports, even
1585 # if it wouldn't reallocate the underlying buffer.
1586 # Furthermore, no destructive changes to the buffer may be applied
1587 # before raising the error.
1588 b = bytearray(range(10))
1589 v = memoryview(b)
1590 def resize(n):
1591 b[1:-1] = range(n + 1, 2*n - 1)
1592 resize(10)
1593 orig = b[:]
1594 self.assertRaises(BufferError, resize, 11)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001595 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001596 self.assertRaises(BufferError, resize, 9)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001597 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001598 self.assertRaises(BufferError, resize, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001599 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001600 # Other operations implying resize
1601 self.assertRaises(BufferError, b.pop, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001602 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001603 self.assertRaises(BufferError, b.remove, b[1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001604 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001605 def delitem():
1606 del b[1]
1607 self.assertRaises(BufferError, delitem)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001608 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001609 # deleting a non-contiguous slice
1610 def delslice():
1611 b[1:-1:2] = b""
1612 self.assertRaises(BufferError, delslice)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001613 self.assertEqual(b, orig)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001614
Stefan Krah650c1e82015-02-03 21:43:23 +01001615 @test.support.cpython_only
1616 def test_obsolete_write_lock(self):
1617 from _testcapi import getbuffer_with_null_view
1618 self.assertRaises(BufferError, getbuffer_with_null_view, bytearray())
Antoine Pitroufc8d6f42010-01-17 12:38:54 +00001619
Serhiy Storchakaaabafe72016-03-06 14:10:24 +02001620 def test_iterator_pickling2(self):
1621 orig = bytearray(b'abc')
1622 data = list(b'qwerty')
1623 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
1624 # initial iterator
1625 itorig = iter(orig)
1626 d = pickle.dumps((itorig, orig), proto)
1627 it, b = pickle.loads(d)
1628 b[:] = data
1629 self.assertEqual(type(it), type(itorig))
1630 self.assertEqual(list(it), data)
1631
1632 # running iterator
1633 next(itorig)
1634 d = pickle.dumps((itorig, orig), proto)
1635 it, b = pickle.loads(d)
1636 b[:] = data
1637 self.assertEqual(type(it), type(itorig))
1638 self.assertEqual(list(it), data[1:])
1639
1640 # empty iterator
1641 for i in range(1, len(orig)):
1642 next(itorig)
1643 d = pickle.dumps((itorig, orig), proto)
1644 it, b = pickle.loads(d)
1645 b[:] = data
1646 self.assertEqual(type(it), type(itorig))
1647 self.assertEqual(list(it), data[len(orig):])
1648
1649 # exhausted iterator
1650 self.assertRaises(StopIteration, next, itorig)
1651 d = pickle.dumps((itorig, orig), proto)
1652 it, b = pickle.loads(d)
1653 b[:] = data
1654 self.assertEqual(list(it), [])
1655
Serhiy Storchaka8dc2ec12016-03-30 21:01:26 +03001656 test_exhausted_iterator = test.list_tests.CommonTest.test_exhausted_iterator
Serhiy Storchakaaabafe72016-03-06 14:10:24 +02001657
Serhiy Storchakaaf658722016-07-03 14:41:36 +03001658 def test_iterator_length_hint(self):
1659 # Issue 27443: __length_hint__ can return negative integer
1660 ba = bytearray(b'ab')
1661 it = iter(ba)
1662 next(it)
1663 ba.clear()
1664 # Shouldn't raise an error
1665 self.assertEqual(list(it), [])
1666
1667
Christian Heimes510711d2008-01-30 11:57:58 +00001668class AssortedBytesTest(unittest.TestCase):
1669 #
1670 # Test various combinations of bytes and bytearray
1671 #
1672
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001673 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001674 def test_repr_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001675 for f in str, repr:
1676 self.assertEqual(f(bytearray()), "bytearray(b'')")
1677 self.assertEqual(f(bytearray([0])), "bytearray(b'\\x00')")
1678 self.assertEqual(f(bytearray([0, 1, 254, 255])),
1679 "bytearray(b'\\x00\\x01\\xfe\\xff')")
1680 self.assertEqual(f(b"abc"), "b'abc'")
1681 self.assertEqual(f(b"'"), '''b"'"''') # '''
1682 self.assertEqual(f(b"'\""), r"""b'\'"'""") # '
Christian Heimes510711d2008-01-30 11:57:58 +00001683
Serhiy Storchakad1af5ef2016-10-30 19:33:54 +02001684 @check_bytes_warnings
1685 def test_format(self):
1686 for b in b'abc', bytearray(b'abc'):
1687 self.assertEqual(format(b), str(b))
1688 self.assertEqual(format(b, ''), str(b))
1689 with self.assertRaisesRegex(TypeError,
1690 r'\b%s\b' % re.escape(type(b).__name__)):
1691 format(b, 's')
1692
Christian Heimes510711d2008-01-30 11:57:58 +00001693 def test_compare_bytes_to_bytearray(self):
1694 self.assertEqual(b"abc" == bytes(b"abc"), True)
1695 self.assertEqual(b"ab" != bytes(b"abc"), True)
1696 self.assertEqual(b"ab" <= bytes(b"abc"), True)
1697 self.assertEqual(b"ab" < bytes(b"abc"), True)
1698 self.assertEqual(b"abc" >= bytes(b"ab"), True)
1699 self.assertEqual(b"abc" > bytes(b"ab"), True)
1700
1701 self.assertEqual(b"abc" != bytes(b"abc"), False)
1702 self.assertEqual(b"ab" == bytes(b"abc"), False)
1703 self.assertEqual(b"ab" > bytes(b"abc"), False)
1704 self.assertEqual(b"ab" >= bytes(b"abc"), False)
1705 self.assertEqual(b"abc" < bytes(b"ab"), False)
1706 self.assertEqual(b"abc" <= bytes(b"ab"), False)
1707
1708 self.assertEqual(bytes(b"abc") == b"abc", True)
1709 self.assertEqual(bytes(b"ab") != b"abc", True)
1710 self.assertEqual(bytes(b"ab") <= b"abc", True)
1711 self.assertEqual(bytes(b"ab") < b"abc", True)
1712 self.assertEqual(bytes(b"abc") >= b"ab", True)
1713 self.assertEqual(bytes(b"abc") > b"ab", True)
1714
1715 self.assertEqual(bytes(b"abc") != b"abc", False)
1716 self.assertEqual(bytes(b"ab") == b"abc", False)
1717 self.assertEqual(bytes(b"ab") > b"abc", False)
1718 self.assertEqual(bytes(b"ab") >= b"abc", False)
1719 self.assertEqual(bytes(b"abc") < b"ab", False)
1720 self.assertEqual(bytes(b"abc") <= b"ab", False)
1721
Stefan Krah6e572b82013-01-26 13:06:36 +01001722 @test.support.requires_docstrings
Christian Heimes510711d2008-01-30 11:57:58 +00001723 def test_doc(self):
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001724 self.assertIsNotNone(bytearray.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001725 self.assertTrue(bytearray.__doc__.startswith("bytearray("), bytearray.__doc__)
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001726 self.assertIsNotNone(bytes.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001727 self.assertTrue(bytes.__doc__.startswith("bytes("), bytes.__doc__)
Christian Heimes510711d2008-01-30 11:57:58 +00001728
1729 def test_from_bytearray(self):
1730 sample = bytes(b"Hello world\n\x80\x81\xfe\xff")
1731 buf = memoryview(sample)
1732 b = bytearray(buf)
1733 self.assertEqual(b, bytearray(sample))
1734
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001735 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001736 def test_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001737 self.assertEqual(str(b''), "b''")
1738 self.assertEqual(str(b'x'), "b'x'")
1739 self.assertEqual(str(b'\x80'), "b'\\x80'")
1740 self.assertEqual(str(bytearray(b'')), "bytearray(b'')")
1741 self.assertEqual(str(bytearray(b'x')), "bytearray(b'x')")
1742 self.assertEqual(str(bytearray(b'\x80')), "bytearray(b'\\x80')")
Christian Heimes510711d2008-01-30 11:57:58 +00001743
1744 def test_literal(self):
1745 tests = [
1746 (b"Wonderful spam", "Wonderful spam"),
1747 (br"Wonderful spam too", "Wonderful spam too"),
1748 (b"\xaa\x00\000\200", "\xaa\x00\000\200"),
1749 (br"\xaa\x00\000\200", r"\xaa\x00\000\200"),
1750 ]
1751 for b, s in tests:
1752 self.assertEqual(b, bytearray(s, 'latin-1'))
1753 for c in range(128, 256):
1754 self.assertRaises(SyntaxError, eval,
1755 'b"%s"' % chr(c))
1756
Christian Heimes510711d2008-01-30 11:57:58 +00001757 def test_split_bytearray(self):
1758 self.assertEqual(b'a b'.split(memoryview(b' ')), [b'a', b'b'])
1759
1760 def test_rsplit_bytearray(self):
1761 self.assertEqual(b'a b'.rsplit(memoryview(b' ')), [b'a', b'b'])
1762
Georg Brandleb2c9642008-05-30 12:05:02 +00001763 def test_return_self(self):
1764 # bytearray.replace must always return a new bytearray
1765 b = bytearray()
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001766 self.assertIsNot(b.replace(b'', b''), b)
Georg Brandleb2c9642008-05-30 12:05:02 +00001767
Serhiy Storchaka1dd49822015-03-20 16:54:57 +02001768 @unittest.skipUnless(sys.flags.bytes_warning,
1769 "BytesWarning is needed for this test: use -bb option")
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001770 def test_compare(self):
Serhiy Storchaka1dd49822015-03-20 16:54:57 +02001771 def bytes_warning():
1772 return test.support.check_warnings(('', BytesWarning))
1773 with bytes_warning():
1774 b'' == ''
1775 with bytes_warning():
1776 '' == b''
1777 with bytes_warning():
1778 b'' != ''
1779 with bytes_warning():
1780 '' != b''
1781 with bytes_warning():
1782 bytearray(b'') == ''
1783 with bytes_warning():
1784 '' == bytearray(b'')
1785 with bytes_warning():
1786 bytearray(b'') != ''
1787 with bytes_warning():
1788 '' != bytearray(b'')
1789 with bytes_warning():
1790 b'\0' == 0
1791 with bytes_warning():
1792 0 == b'\0'
1793 with bytes_warning():
1794 b'\0' != 0
1795 with bytes_warning():
1796 0 != b'\0'
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001797
Guido van Rossumd624f182006-04-24 13:47:05 +00001798 # Optimizations:
Guido van Rossume06b6b82006-04-23 07:43:54 +00001799 # __iter__? (optimization)
Guido van Rossumd624f182006-04-24 13:47:05 +00001800 # __reversed__? (optimization)
1801
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001802 # XXX More string methods? (Those that don't use character properties)
1803
Neal Norwitz6968b052007-02-27 19:02:19 +00001804 # There are tests in string_tests.py that are more
Martin Panter0d0db6c2016-04-10 08:45:26 +00001805 # comprehensive for things like partition, etc.
Neal Norwitz6968b052007-02-27 19:02:19 +00001806 # Unfortunately they are all bundled with tests that
1807 # are not appropriate for bytes
Guido van Rossume06b6b82006-04-23 07:43:54 +00001808
Guido van Rossum254348e2007-11-21 19:29:53 +00001809 # I've started porting some of those into bytearray_tests.py, we should port
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001810 # the rest that make sense (the code can be cleaned up to use modern
1811 # unittest methods at the same time).
1812
Martin Panter152a19c2016-04-06 06:37:17 +00001813class BytearrayPEP3137Test(unittest.TestCase):
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001814 def marshal(self, x):
Guido van Rossum254348e2007-11-21 19:29:53 +00001815 return bytearray(x)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001816
1817 def test_returns_new_copy(self):
1818 val = self.marshal(b'1234')
1819 # On immutable types these MAY return a reference to themselves
Guido van Rossum254348e2007-11-21 19:29:53 +00001820 # but on mutable types like bytearray they MUST return a new copy.
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001821 for methname in ('zfill', 'rjust', 'ljust', 'center'):
1822 method = getattr(val, methname)
1823 newval = method(3)
1824 self.assertEqual(val, newval)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001825 self.assertIsNot(val, newval,
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001826 methname+' returned self on a mutable object')
Antoine Pitroucbaa62502010-01-13 15:03:39 +00001827 for expr in ('val.split()[0]', 'val.rsplit()[0]',
1828 'val.partition(b".")[0]', 'val.rpartition(b".")[2]',
1829 'val.splitlines()[0]', 'val.replace(b"", b"")'):
1830 newval = eval(expr)
1831 self.assertEqual(val, newval)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001832 self.assertIsNot(val, newval,
Antoine Pitroucbaa62502010-01-13 15:03:39 +00001833 expr+' returned val on a mutable object')
Antoine Pitroucfc22b42012-10-16 21:07:23 +02001834 sep = self.marshal(b'')
1835 newval = sep.join([val])
1836 self.assertEqual(val, newval)
1837 self.assertIsNot(val, newval)
1838
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001839
Christian Heimes510711d2008-01-30 11:57:58 +00001840class FixedStringTest(test.string_tests.BaseTest):
Guido van Rossum9c627722007-08-27 18:31:48 +00001841 def fixtype(self, obj):
1842 if isinstance(obj, str):
Martin Panter152a19c2016-04-06 06:37:17 +00001843 return self.type2test(obj.encode("utf-8"))
Guido van Rossum9c627722007-08-27 18:31:48 +00001844 return super().fixtype(obj)
1845
Martin Panter152a19c2016-04-06 06:37:17 +00001846 contains_bytes = True
1847
Ezio Melotti0dceb562013-01-10 07:43:26 +02001848class ByteArrayAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001849 type2test = bytearray
1850
Ezio Melotti0dceb562013-01-10 07:43:26 +02001851class BytesAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001852 type2test = bytes
1853
Georg Brandlc7885542007-03-06 19:16:20 +00001854
Ezio Melotti0dceb562013-01-10 07:43:26 +02001855class SubclassTest:
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001856
1857 def test_basic(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001858 self.assertTrue(issubclass(self.type2test, self.basetype))
1859 self.assertIsInstance(self.type2test(), self.basetype)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001860
1861 a, b = b"abcd", b"efgh"
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001862 _a, _b = self.type2test(a), self.type2test(b)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001863
1864 # test comparison operators with subclass instances
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001865 self.assertTrue(_a == _a)
1866 self.assertTrue(_a != _b)
1867 self.assertTrue(_a < _b)
1868 self.assertTrue(_a <= _b)
1869 self.assertTrue(_b >= _a)
1870 self.assertTrue(_b > _a)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001871 self.assertIsNot(_a, a)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001872
1873 # test concat of subclass instances
1874 self.assertEqual(a + b, _a + _b)
1875 self.assertEqual(a + b, a + _b)
1876 self.assertEqual(a + b, _a + b)
1877
1878 # test repeat
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001879 self.assertTrue(a*5 == _a*5)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001880
1881 def test_join(self):
1882 # Make sure join returns a NEW object for single item sequences
1883 # involving a subclass.
1884 # Make sure that it is of the appropriate type.
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001885 s1 = self.type2test(b"abcd")
1886 s2 = self.basetype().join([s1])
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001887 self.assertIsNot(s1, s2)
1888 self.assertIs(type(s2), self.basetype, type(s2))
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001889
1890 # Test reverse, calling join on subclass
1891 s3 = s1.join([b"abcd"])
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001892 self.assertIs(type(s3), self.basetype)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001893
1894 def test_pickle(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001895 a = self.type2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001896 a.x = 10
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001897 a.y = self.type2test(b"efgh")
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +00001898 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001899 b = pickle.loads(pickle.dumps(a, proto))
1900 self.assertNotEqual(id(a), id(b))
1901 self.assertEqual(a, b)
1902 self.assertEqual(a.x, b.x)
1903 self.assertEqual(a.y, b.y)
1904 self.assertEqual(type(a), type(b))
1905 self.assertEqual(type(a.y), type(b.y))
1906
1907 def test_copy(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001908 a = self.type2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001909 a.x = 10
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001910 a.y = self.type2test(b"efgh")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001911 for copy_method in (copy.copy, copy.deepcopy):
1912 b = copy_method(a)
1913 self.assertNotEqual(id(a), id(b))
1914 self.assertEqual(a, b)
1915 self.assertEqual(a.x, b.x)
1916 self.assertEqual(a.y, b.y)
1917 self.assertEqual(type(a), type(b))
1918 self.assertEqual(type(a.y), type(b.y))
1919
Serhiy Storchaka0855e702016-07-01 17:22:31 +03001920 def test_fromhex(self):
1921 b = self.type2test.fromhex('1a2B30')
1922 self.assertEqual(b, b'\x1a\x2b\x30')
1923 self.assertIs(type(b), self.type2test)
1924
1925 class B1(self.basetype):
1926 def __new__(cls, value):
1927 me = self.basetype.__new__(cls, value)
1928 me.foo = 'bar'
1929 return me
1930
1931 b = B1.fromhex('1a2B30')
1932 self.assertEqual(b, b'\x1a\x2b\x30')
1933 self.assertIs(type(b), B1)
1934 self.assertEqual(b.foo, 'bar')
1935
1936 class B2(self.basetype):
1937 def __init__(me, *args, **kwargs):
1938 if self.basetype is not bytes:
1939 self.basetype.__init__(me, *args, **kwargs)
1940 me.foo = 'bar'
1941
1942 b = B2.fromhex('1a2B30')
1943 self.assertEqual(b, b'\x1a\x2b\x30')
1944 self.assertIs(type(b), B2)
1945 self.assertEqual(b.foo, 'bar')
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001946
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001947
1948class ByteArraySubclass(bytearray):
1949 pass
1950
1951class BytesSubclass(bytes):
1952 pass
1953
Serhiy Storchaka15095802015-11-25 15:47:01 +02001954class OtherBytesSubclass(bytes):
1955 pass
1956
Ezio Melotti0dceb562013-01-10 07:43:26 +02001957class ByteArraySubclassTest(SubclassTest, unittest.TestCase):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001958 basetype = bytearray
1959 type2test = ByteArraySubclass
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001960
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001961 def test_init_override(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001962 class subclass(bytearray):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001963 def __init__(me, newarg=1, *args, **kwargs):
1964 bytearray.__init__(me, *args, **kwargs)
1965 x = subclass(4, b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001966 x = subclass(4, source=b"abcd")
1967 self.assertEqual(x, b"abcd")
1968 x = subclass(newarg=4, source=b"abcd")
1969 self.assertEqual(x, b"abcd")
1970
1971
Ezio Melotti0dceb562013-01-10 07:43:26 +02001972class BytesSubclassTest(SubclassTest, unittest.TestCase):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001973 basetype = bytes
1974 type2test = BytesSubclass
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001975
1976
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001977if __name__ == "__main__":
Ezio Melotti0dceb562013-01-10 07:43:26 +02001978 unittest.main()