blob: 61b4b9162ccc54a46ef3cf423067bdc17ee54db5 [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
Hai Shia089d212020-07-06 17:15:08 +080019from test.support import import_helper
Georg Brandlc7885542007-03-06 19:16:20 +000020import test.string_tests
Serhiy Storchaka8dc2ec12016-03-30 21:01:26 +030021import test.list_tests
Antoine Pitroucc231542014-11-02 18:40:09 +010022from test.support import bigaddrspacetest, MAX_Py_ssize_t
Victor Stinner22eb6892019-06-26 00:51:05 +020023from test.support.script_helper import assert_python_failure
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000024
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000025
26if sys.flags.bytes_warning:
27 def check_bytes_warnings(func):
28 @functools.wraps(func)
29 def wrapper(*args, **kw):
30 with test.support.check_warnings(('', BytesWarning)):
31 return func(*args, **kw)
32 return wrapper
33else:
34 # no-op
35 def check_bytes_warnings(func):
36 return func
37
38
Georg Brandl9a54d7c2008-07-16 23:15:30 +000039class Indexable:
40 def __init__(self, value=0):
41 self.value = value
42 def __index__(self):
43 return self.value
44
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000045
Ezio Melotti0dceb562013-01-10 07:43:26 +020046class BaseBytesTest:
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000047
48 def test_basics(self):
Christian Heimes510711d2008-01-30 11:57:58 +000049 b = self.type2test()
50 self.assertEqual(type(b), self.type2test)
51 self.assertEqual(b.__class__, self.type2test)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000052
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +000053 def test_copy(self):
54 a = self.type2test(b"abcd")
55 for copy_method in (copy.copy, copy.deepcopy):
56 b = copy_method(a)
57 self.assertEqual(a, b)
58 self.assertEqual(type(a), type(b))
59
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000060 def test_empty_sequence(self):
Christian Heimes510711d2008-01-30 11:57:58 +000061 b = self.type2test()
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000062 self.assertEqual(len(b), 0)
63 self.assertRaises(IndexError, lambda: b[0])
64 self.assertRaises(IndexError, lambda: b[1])
Christian Heimesa37d4c62007-12-04 23:02:19 +000065 self.assertRaises(IndexError, lambda: b[sys.maxsize])
66 self.assertRaises(IndexError, lambda: b[sys.maxsize+1])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000067 self.assertRaises(IndexError, lambda: b[10**100])
68 self.assertRaises(IndexError, lambda: b[-1])
69 self.assertRaises(IndexError, lambda: b[-2])
Christian Heimesa37d4c62007-12-04 23:02:19 +000070 self.assertRaises(IndexError, lambda: b[-sys.maxsize])
71 self.assertRaises(IndexError, lambda: b[-sys.maxsize-1])
72 self.assertRaises(IndexError, lambda: b[-sys.maxsize-2])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000073 self.assertRaises(IndexError, lambda: b[-10**100])
74
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030075 def test_from_iterable(self):
76 b = self.type2test(range(256))
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000077 self.assertEqual(len(b), 256)
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030078 self.assertEqual(list(b), list(range(256)))
79
80 # Non-sequence iterable.
81 b = self.type2test({42})
82 self.assertEqual(b, b"*")
83 b = self.type2test({43, 45})
84 self.assertIn(tuple(b), {(43, 45), (45, 43)})
85
86 # Iterator that has a __length_hint__.
87 b = self.type2test(iter(range(256)))
88 self.assertEqual(len(b), 256)
89 self.assertEqual(list(b), list(range(256)))
90
91 # Iterator that doesn't have a __length_hint__.
92 b = self.type2test(i for i in range(256) if i % 2)
93 self.assertEqual(len(b), 128)
94 self.assertEqual(list(b), list(range(256))[1::2])
95
96 # Sequence without __iter__.
97 class S:
98 def __getitem__(self, i):
99 return (1, 2, 3)[i]
100 b = self.type2test(S())
101 self.assertEqual(b, b"\x01\x02\x03")
102
103 def test_from_tuple(self):
104 # There is a special case for tuples.
105 b = self.type2test(tuple(range(256)))
106 self.assertEqual(len(b), 256)
107 self.assertEqual(list(b), list(range(256)))
108 b = self.type2test((1, 2, 3))
109 self.assertEqual(b, b"\x01\x02\x03")
110
111 def test_from_list(self):
112 # There is a special case for lists.
113 b = self.type2test(list(range(256)))
114 self.assertEqual(len(b), 256)
115 self.assertEqual(list(b), list(range(256)))
116 b = self.type2test([1, 2, 3])
117 self.assertEqual(b, b"\x01\x02\x03")
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000118
Serhiy Storchaka914f9a02018-10-21 15:25:53 +0300119 def test_from_mutating_list(self):
120 # Issue #34973: Crash in bytes constructor with mutating list.
121 class X:
122 def __index__(self):
123 a.clear()
124 return 42
125 a = [X(), X()]
126 self.assertEqual(bytes(a), b'*')
127
128 class Y:
129 def __index__(self):
130 if len(a) < 1000:
131 a.append(self)
132 return 42
133 a = [Y()]
134 self.assertEqual(bytes(a), b'*' * 1000) # should not crash
135
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000136 def test_from_index(self):
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000137 b = self.type2test([Indexable(), Indexable(1), Indexable(254),
138 Indexable(255)])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000139 self.assertEqual(list(b), [0, 1, 254, 255])
Benjamin Peterson4c045832010-04-16 22:35:32 +0000140 self.assertRaises(ValueError, self.type2test, [Indexable(-1)])
141 self.assertRaises(ValueError, self.type2test, [Indexable(256)])
Guido van Rossum98297ee2007-11-06 21:34:58 +0000142
INADA Naokia634e232017-01-06 17:32:01 +0900143 def test_from_buffer(self):
144 a = self.type2test(array.array('B', [1, 2, 3]))
145 self.assertEqual(a, b"\x01\x02\x03")
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +0300146 a = self.type2test(b"\x01\x02\x03")
147 self.assertEqual(a, b"\x01\x02\x03")
INADA Naokia634e232017-01-06 17:32:01 +0900148
Serhiy Storchakae8904212018-10-15 00:02:57 +0300149 # Issues #29159 and #34974.
150 # Fallback when __index__ raises a TypeError
INADA Naokia634e232017-01-06 17:32:01 +0900151 class B(bytes):
152 def __index__(self):
153 raise TypeError
154
155 self.assertEqual(self.type2test(B(b"foobar")), b"foobar")
156
Guido van Rossum98297ee2007-11-06 21:34:58 +0000157 def test_from_ssize(self):
Benjamin Peterson4c045832010-04-16 22:35:32 +0000158 self.assertEqual(self.type2test(0), b'')
159 self.assertEqual(self.type2test(1), b'\x00')
160 self.assertEqual(self.type2test(5), b'\x00\x00\x00\x00\x00')
161 self.assertRaises(ValueError, self.type2test, -1)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000162
Benjamin Peterson4c045832010-04-16 22:35:32 +0000163 self.assertEqual(self.type2test('0', 'ascii'), b'0')
164 self.assertEqual(self.type2test(b'0'), b'0')
Benjamin Peterson8380dd52010-04-16 22:51:37 +0000165 self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000166
167 def test_constructor_type_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000168 self.assertRaises(TypeError, self.type2test, 0.0)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000169 class C:
170 pass
Christian Heimes510711d2008-01-30 11:57:58 +0000171 self.assertRaises(TypeError, self.type2test, ["0"])
172 self.assertRaises(TypeError, self.type2test, [0.0])
173 self.assertRaises(TypeError, self.type2test, [None])
174 self.assertRaises(TypeError, self.type2test, [C()])
Serhiy Storchaka2c2044e2018-10-21 15:29:12 +0300175 self.assertRaises(TypeError, self.type2test, encoding='ascii')
176 self.assertRaises(TypeError, self.type2test, errors='ignore')
Serhiy Storchaka83cf99d2014-12-02 09:24:06 +0200177 self.assertRaises(TypeError, self.type2test, 0, 'ascii')
178 self.assertRaises(TypeError, self.type2test, b'', 'ascii')
179 self.assertRaises(TypeError, self.type2test, 0, errors='ignore')
180 self.assertRaises(TypeError, self.type2test, b'', errors='ignore')
181 self.assertRaises(TypeError, self.type2test, '')
182 self.assertRaises(TypeError, self.type2test, '', errors='ignore')
183 self.assertRaises(TypeError, self.type2test, '', b'ascii')
184 self.assertRaises(TypeError, self.type2test, '', 'ascii', b'ignore')
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000185
186 def test_constructor_value_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000187 self.assertRaises(ValueError, self.type2test, [-1])
188 self.assertRaises(ValueError, self.type2test, [-sys.maxsize])
189 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-1])
190 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-2])
191 self.assertRaises(ValueError, self.type2test, [-10**100])
192 self.assertRaises(ValueError, self.type2test, [256])
193 self.assertRaises(ValueError, self.type2test, [257])
194 self.assertRaises(ValueError, self.type2test, [sys.maxsize])
195 self.assertRaises(ValueError, self.type2test, [sys.maxsize+1])
196 self.assertRaises(ValueError, self.type2test, [10**100])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000197
Antoine Pitroucc231542014-11-02 18:40:09 +0100198 @bigaddrspacetest
199 def test_constructor_overflow(self):
200 size = MAX_Py_ssize_t
201 self.assertRaises((OverflowError, MemoryError), self.type2test, size)
202 try:
203 # Should either pass or raise an error (e.g. on debug builds with
204 # additional malloc() overhead), but shouldn't crash.
205 bytearray(size - 4)
206 except (OverflowError, MemoryError):
207 pass
208
Serhiy Storchakae8904212018-10-15 00:02:57 +0300209 def test_constructor_exceptions(self):
210 # Issue #34974: bytes and bytearray constructors replace unexpected
211 # exceptions.
212 class BadInt:
213 def __index__(self):
214 1/0
215 self.assertRaises(ZeroDivisionError, self.type2test, BadInt())
216 self.assertRaises(ZeroDivisionError, self.type2test, [BadInt()])
217
218 class BadIterable:
219 def __iter__(self):
220 1/0
221 self.assertRaises(ZeroDivisionError, self.type2test, BadIterable())
222
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000223 def test_compare(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000224 b1 = self.type2test([1, 2, 3])
225 b2 = self.type2test([1, 2, 3])
226 b3 = self.type2test([1, 3])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000227
Guido van Rossume61fd5b2007-07-11 12:20:59 +0000228 self.assertEqual(b1, b2)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000229 self.assertTrue(b2 != b3)
230 self.assertTrue(b1 <= b2)
231 self.assertTrue(b1 <= b3)
232 self.assertTrue(b1 < b3)
233 self.assertTrue(b1 >= b2)
234 self.assertTrue(b3 >= b2)
235 self.assertTrue(b3 > b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000236
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000237 self.assertFalse(b1 != b2)
238 self.assertFalse(b2 == b3)
239 self.assertFalse(b1 > b2)
240 self.assertFalse(b1 > b3)
241 self.assertFalse(b1 >= b3)
242 self.assertFalse(b1 < b2)
243 self.assertFalse(b3 < b2)
244 self.assertFalse(b3 <= b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000245
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000246 @check_bytes_warnings
Guido van Rossum343e97f2007-04-09 00:43:24 +0000247 def test_compare_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000248 # Byte comparisons with unicode should always fail!
249 # Test this for all expected byte orders and Unicode character
250 # sizes.
251 self.assertEqual(self.type2test(b"\0a\0b\0c") == "abc", False)
252 self.assertEqual(self.type2test(b"\0\0\0a\0\0\0b\0\0\0c") == "abc",
253 False)
254 self.assertEqual(self.type2test(b"a\0b\0c\0") == "abc", False)
255 self.assertEqual(self.type2test(b"a\0\0\0b\0\0\0c\0\0\0") == "abc",
256 False)
257 self.assertEqual(self.type2test() == str(), False)
258 self.assertEqual(self.type2test() != str(), True)
Christian Heimes510711d2008-01-30 11:57:58 +0000259
260 def test_reversed(self):
261 input = list(map(ord, "Hello"))
262 b = self.type2test(input)
263 output = list(reversed(b))
264 input.reverse()
265 self.assertEqual(output, input)
266
267 def test_getslice(self):
268 def by(s):
269 return self.type2test(map(ord, s))
270 b = by("Hello, world")
271
272 self.assertEqual(b[:5], by("Hello"))
273 self.assertEqual(b[1:5], by("ello"))
274 self.assertEqual(b[5:7], by(", "))
275 self.assertEqual(b[7:], by("world"))
276 self.assertEqual(b[7:12], by("world"))
277 self.assertEqual(b[7:100], by("world"))
278
279 self.assertEqual(b[:-7], by("Hello"))
280 self.assertEqual(b[-11:-7], by("ello"))
281 self.assertEqual(b[-7:-5], by(", "))
282 self.assertEqual(b[-5:], by("world"))
283 self.assertEqual(b[-5:12], by("world"))
284 self.assertEqual(b[-5:100], by("world"))
285 self.assertEqual(b[-100:5], by("Hello"))
286
287 def test_extended_getslice(self):
288 # Test extended slicing by comparing with list slicing.
289 L = list(range(255))
290 b = self.type2test(L)
Zackery Spytz14514d92019-05-17 01:13:03 -0600291 indices = (0, None, 1, 3, 19, 100, sys.maxsize, -1, -2, -31, -100)
Christian Heimes510711d2008-01-30 11:57:58 +0000292 for start in indices:
293 for stop in indices:
294 # Skip step 0 (invalid)
295 for step in indices[1:]:
296 self.assertEqual(b[start:stop:step], self.type2test(L[start:stop:step]))
297
298 def test_encoding(self):
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000299 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000300 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000301 b = self.type2test(sample, enc)
302 self.assertEqual(b, self.type2test(sample.encode(enc)))
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000303 self.assertRaises(UnicodeEncodeError, self.type2test, sample, "latin-1")
304 b = self.type2test(sample, "latin-1", "ignore")
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000305 self.assertEqual(b, self.type2test(sample[:-3], "utf-8"))
Christian Heimes510711d2008-01-30 11:57:58 +0000306
307 def test_decode(self):
Serhiy Storchakac49a5162016-09-08 15:47:27 +0300308 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000309 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000310 b = self.type2test(sample, enc)
311 self.assertEqual(b.decode(enc), sample)
312 sample = "Hello world\n\x80\x81\xfe\xff"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000313 b = self.type2test(sample, "latin-1")
314 self.assertRaises(UnicodeDecodeError, b.decode, "utf-8")
315 self.assertEqual(b.decode("utf-8", "ignore"), "Hello world\n")
316 self.assertEqual(b.decode(errors="ignore", encoding="utf-8"),
Benjamin Peterson308d6372009-09-18 21:42:35 +0000317 "Hello world\n")
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000318 # Default encoding is utf-8
319 self.assertEqual(self.type2test(b'\xe2\x98\x83').decode(), '\u2603')
Christian Heimes510711d2008-01-30 11:57:58 +0000320
Victor Stinner22eb6892019-06-26 00:51:05 +0200321 def test_check_encoding_errors(self):
322 # bpo-37388: bytes(str) and bytes.encode() must check encoding
323 # and errors arguments in dev mode
324 invalid = 'Boom, Shaka Laka, Boom!'
325 encodings = ('ascii', 'utf8', 'latin1')
326 code = textwrap.dedent(f'''
327 import sys
328 type2test = {self.type2test.__name__}
329 encodings = {encodings!r}
330
331 for data in ('', 'short string'):
332 try:
333 type2test(data, encoding={invalid!r})
334 except LookupError:
335 pass
336 else:
337 sys.exit(21)
338
339 for encoding in encodings:
340 try:
341 type2test(data, encoding=encoding, errors={invalid!r})
342 except LookupError:
343 pass
344 else:
345 sys.exit(22)
346
347 for data in (b'', b'short string'):
348 data = type2test(data)
349 print(repr(data))
350 try:
351 data.decode(encoding={invalid!r})
352 except LookupError:
353 sys.exit(10)
354 else:
355 sys.exit(23)
356
357 try:
358 data.decode(errors={invalid!r})
359 except LookupError:
360 pass
361 else:
362 sys.exit(24)
363
364 for encoding in encodings:
365 try:
366 data.decode(encoding=encoding, errors={invalid!r})
367 except LookupError:
368 pass
369 else:
370 sys.exit(25)
371
372 sys.exit(10)
373 ''')
374 proc = assert_python_failure('-X', 'dev', '-c', code)
375 self.assertEqual(proc.rc, 10, proc)
376
Christian Heimes510711d2008-01-30 11:57:58 +0000377 def test_from_int(self):
378 b = self.type2test(0)
379 self.assertEqual(b, self.type2test())
380 b = self.type2test(10)
381 self.assertEqual(b, self.type2test([0]*10))
382 b = self.type2test(10000)
383 self.assertEqual(b, self.type2test([0]*10000))
384
385 def test_concat(self):
386 b1 = self.type2test(b"abc")
387 b2 = self.type2test(b"def")
388 self.assertEqual(b1 + b2, b"abcdef")
389 self.assertEqual(b1 + bytes(b"def"), b"abcdef")
390 self.assertEqual(bytes(b"def") + b1, b"defabc")
391 self.assertRaises(TypeError, lambda: b1 + "def")
392 self.assertRaises(TypeError, lambda: "abc" + b2)
393
394 def test_repeat(self):
395 for b in b"abc", self.type2test(b"abc"):
396 self.assertEqual(b * 3, b"abcabcabc")
397 self.assertEqual(b * 0, b"")
398 self.assertEqual(b * -1, b"")
399 self.assertRaises(TypeError, lambda: b * 3.14)
400 self.assertRaises(TypeError, lambda: 3.14 * b)
401 # XXX Shouldn't bytes and bytearray agree on what to raise?
Mark Dickinsoncf940c72010-08-10 18:35:01 +0000402 with self.assertRaises((OverflowError, MemoryError)):
403 c = b * sys.maxsize
404 with self.assertRaises((OverflowError, MemoryError)):
405 b *= sys.maxsize
Christian Heimes510711d2008-01-30 11:57:58 +0000406
407 def test_repeat_1char(self):
408 self.assertEqual(self.type2test(b'x')*100, self.type2test([ord('x')]*100))
409
410 def test_contains(self):
411 b = self.type2test(b"abc")
Benjamin Peterson577473f2010-01-19 00:09:57 +0000412 self.assertIn(ord('a'), b)
413 self.assertIn(int(ord('a')), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000414 self.assertNotIn(200, b)
Christian Heimes510711d2008-01-30 11:57:58 +0000415 self.assertRaises(ValueError, lambda: 300 in b)
416 self.assertRaises(ValueError, lambda: -1 in b)
Serhiy Storchakaf9efb8b2016-07-10 12:37:30 +0300417 self.assertRaises(ValueError, lambda: sys.maxsize+1 in b)
Christian Heimes510711d2008-01-30 11:57:58 +0000418 self.assertRaises(TypeError, lambda: None in b)
419 self.assertRaises(TypeError, lambda: float(ord('a')) in b)
420 self.assertRaises(TypeError, lambda: "a" in b)
421 for f in bytes, bytearray:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000422 self.assertIn(f(b""), b)
423 self.assertIn(f(b"a"), b)
424 self.assertIn(f(b"b"), b)
425 self.assertIn(f(b"c"), b)
426 self.assertIn(f(b"ab"), b)
427 self.assertIn(f(b"bc"), b)
428 self.assertIn(f(b"abc"), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000429 self.assertNotIn(f(b"ac"), b)
430 self.assertNotIn(f(b"d"), b)
431 self.assertNotIn(f(b"dab"), b)
432 self.assertNotIn(f(b"abd"), b)
Christian Heimes510711d2008-01-30 11:57:58 +0000433
434 def test_fromhex(self):
435 self.assertRaises(TypeError, self.type2test.fromhex)
436 self.assertRaises(TypeError, self.type2test.fromhex, 1)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000437 self.assertEqual(self.type2test.fromhex(''), self.type2test())
Christian Heimes510711d2008-01-30 11:57:58 +0000438 b = bytearray([0x1a, 0x2b, 0x30])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000439 self.assertEqual(self.type2test.fromhex('1a2B30'), b)
440 self.assertEqual(self.type2test.fromhex(' 1A 2B 30 '), b)
Serhiy Storchakadd1da7f2016-12-19 18:51:37 +0200441
442 # check that ASCII whitespace is ignored
443 self.assertEqual(self.type2test.fromhex(' 1A\n2B\t30\v'), b)
444 for c in "\x09\x0A\x0B\x0C\x0D\x20":
445 self.assertEqual(self.type2test.fromhex(c), self.type2test())
446 for c in "\x1C\x1D\x1E\x1F\x85\xa0\u2000\u2002\u2028":
447 self.assertRaises(ValueError, self.type2test.fromhex, c)
448
Ezio Melottib3aedd42010-11-20 19:04:17 +0000449 self.assertEqual(self.type2test.fromhex('0000'), b'\0\0')
Christian Heimes510711d2008-01-30 11:57:58 +0000450 self.assertRaises(TypeError, self.type2test.fromhex, b'1B')
451 self.assertRaises(ValueError, self.type2test.fromhex, 'a')
452 self.assertRaises(ValueError, self.type2test.fromhex, 'rt')
453 self.assertRaises(ValueError, self.type2test.fromhex, '1a b cd')
454 self.assertRaises(ValueError, self.type2test.fromhex, '\x00')
455 self.assertRaises(ValueError, self.type2test.fromhex, '12 \x00 34')
456
Victor Stinner2bf89932015-10-14 11:25:33 +0200457 for data, pos in (
458 # invalid first hexadecimal character
459 ('12 x4 56', 3),
460 # invalid second hexadecimal character
461 ('12 3x 56', 4),
462 # two invalid hexadecimal characters
463 ('12 xy 56', 3),
464 # test non-ASCII string
465 ('12 3\xff 56', 4),
466 ):
467 with self.assertRaises(ValueError) as cm:
468 self.type2test.fromhex(data)
469 self.assertIn('at position %s' % pos, str(cm.exception))
470
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000471 def test_hex(self):
472 self.assertRaises(TypeError, self.type2test.hex)
473 self.assertRaises(TypeError, self.type2test.hex, 1)
Gregory P. Smith32d34bc2015-04-26 05:05:53 +0000474 self.assertEqual(self.type2test(b"").hex(), "")
475 self.assertEqual(bytearray([0x1a, 0x2b, 0x30]).hex(), '1a2b30')
476 self.assertEqual(self.type2test(b"\x1a\x2b\x30").hex(), '1a2b30')
477 self.assertEqual(memoryview(b"\x1a\x2b\x30").hex(), '1a2b30')
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000478
Gregory P. Smith0c2f9302019-05-29 11:46:58 -0700479 def test_hex_separator_basics(self):
480 three_bytes = self.type2test(b'\xb9\x01\xef')
481 self.assertEqual(three_bytes.hex(), 'b901ef')
482 with self.assertRaises(ValueError):
483 three_bytes.hex('')
484 with self.assertRaises(ValueError):
485 three_bytes.hex('xx')
486 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
487 with self.assertRaises(TypeError):
488 three_bytes.hex(None, 0)
489 with self.assertRaises(ValueError):
490 three_bytes.hex('\xff')
491 with self.assertRaises(ValueError):
492 three_bytes.hex(b'\xff')
493 with self.assertRaises(ValueError):
494 three_bytes.hex(b'\x80')
495 with self.assertRaises(ValueError):
496 three_bytes.hex(chr(0x100))
497 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
498 self.assertEqual(three_bytes.hex(b'\x00'), 'b9\x0001\x00ef')
499 self.assertEqual(three_bytes.hex('\x00'), 'b9\x0001\x00ef')
500 self.assertEqual(three_bytes.hex(b'\x7f'), 'b9\x7f01\x7fef')
501 self.assertEqual(three_bytes.hex('\x7f'), 'b9\x7f01\x7fef')
502 self.assertEqual(three_bytes.hex(':', 3), 'b901ef')
503 self.assertEqual(three_bytes.hex(':', 4), 'b901ef')
504 self.assertEqual(three_bytes.hex(':', -4), 'b901ef')
505 self.assertEqual(three_bytes.hex(':'), 'b9:01:ef')
506 self.assertEqual(three_bytes.hex(b'$'), 'b9$01$ef')
507 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
508 self.assertEqual(three_bytes.hex(':', -1), 'b9:01:ef')
509 self.assertEqual(three_bytes.hex(':', 2), 'b9:01ef')
510 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
511 self.assertEqual(three_bytes.hex('*', -2), 'b901*ef')
512
513 value = b'{s\005\000\000\000worldi\002\000\000\000s\005\000\000\000helloi\001\000\000\0000'
514 self.assertEqual(value.hex('.', 8), '7b7305000000776f.726c646902000000.730500000068656c.6c6f690100000030')
515
516 def test_hex_separator_five_bytes(self):
517 five_bytes = self.type2test(range(90,95))
518 self.assertEqual(five_bytes.hex(), '5a5b5c5d5e')
519
520 def test_hex_separator_six_bytes(self):
521 six_bytes = self.type2test(x*3 for x in range(1, 7))
522 self.assertEqual(six_bytes.hex(), '0306090c0f12')
523 self.assertEqual(six_bytes.hex('.', 1), '03.06.09.0c.0f.12')
524 self.assertEqual(six_bytes.hex(' ', 2), '0306 090c 0f12')
525 self.assertEqual(six_bytes.hex('-', 3), '030609-0c0f12')
526 self.assertEqual(six_bytes.hex(':', 4), '0306:090c0f12')
527 self.assertEqual(six_bytes.hex(':', 5), '03:06090c0f12')
528 self.assertEqual(six_bytes.hex(':', 6), '0306090c0f12')
529 self.assertEqual(six_bytes.hex(':', 95), '0306090c0f12')
530 self.assertEqual(six_bytes.hex('_', -3), '030609_0c0f12')
531 self.assertEqual(six_bytes.hex(':', -4), '0306090c:0f12')
532 self.assertEqual(six_bytes.hex(b'@', -5), '0306090c0f@12')
533 self.assertEqual(six_bytes.hex(':', -6), '0306090c0f12')
534 self.assertEqual(six_bytes.hex(' ', -95), '0306090c0f12')
535
Christian Heimes510711d2008-01-30 11:57:58 +0000536 def test_join(self):
537 self.assertEqual(self.type2test(b"").join([]), b"")
538 self.assertEqual(self.type2test(b"").join([b""]), b"")
539 for lst in [[b"abc"], [b"a", b"bc"], [b"ab", b"c"], [b"a", b"b", b"c"]]:
540 lst = list(map(self.type2test, lst))
541 self.assertEqual(self.type2test(b"").join(lst), b"abc")
542 self.assertEqual(self.type2test(b"").join(tuple(lst)), b"abc")
543 self.assertEqual(self.type2test(b"").join(iter(lst)), b"abc")
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200544 dot_join = self.type2test(b".:").join
545 self.assertEqual(dot_join([b"ab", b"cd"]), b"ab.:cd")
546 self.assertEqual(dot_join([memoryview(b"ab"), b"cd"]), b"ab.:cd")
547 self.assertEqual(dot_join([b"ab", memoryview(b"cd")]), b"ab.:cd")
548 self.assertEqual(dot_join([bytearray(b"ab"), b"cd"]), b"ab.:cd")
549 self.assertEqual(dot_join([b"ab", bytearray(b"cd")]), b"ab.:cd")
550 # Stress it with many items
Bruce Merryd07d9f42020-01-29 09:09:24 +0200551 seq = [b"abc"] * 100000
552 expected = b"abc" + b".:abc" * 99999
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200553 self.assertEqual(dot_join(seq), expected)
Bruce Merryd07d9f42020-01-29 09:09:24 +0200554 # Stress test with empty separator
555 seq = [b"abc"] * 100000
556 expected = b"abc" * 100000
557 self.assertEqual(self.type2test(b"").join(seq), expected)
Benjamin Petersonc31f12d2014-09-28 12:56:42 -0400558 self.assertRaises(TypeError, self.type2test(b" ").join, None)
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200559 # Error handling and cleanup when some item in the middle of the
560 # sequence has the wrong type.
561 with self.assertRaises(TypeError):
562 dot_join([bytearray(b"ab"), "cd", b"ef"])
563 with self.assertRaises(TypeError):
564 dot_join([memoryview(b"ab"), "cd", b"ef"])
Christian Heimes510711d2008-01-30 11:57:58 +0000565
Christian Heimes510711d2008-01-30 11:57:58 +0000566 def test_count(self):
567 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200568 i = 105
569 p = 112
570 w = 119
571
Christian Heimes510711d2008-01-30 11:57:58 +0000572 self.assertEqual(b.count(b'i'), 4)
573 self.assertEqual(b.count(b'ss'), 2)
574 self.assertEqual(b.count(b'w'), 0)
575
Antoine Pitrouac65d962011-10-20 23:54:17 +0200576 self.assertEqual(b.count(i), 4)
577 self.assertEqual(b.count(w), 0)
578
579 self.assertEqual(b.count(b'i', 6), 2)
580 self.assertEqual(b.count(b'p', 6), 2)
581 self.assertEqual(b.count(b'i', 1, 3), 1)
582 self.assertEqual(b.count(b'p', 7, 9), 1)
583
584 self.assertEqual(b.count(i, 6), 2)
585 self.assertEqual(b.count(p, 6), 2)
586 self.assertEqual(b.count(i, 1, 3), 1)
587 self.assertEqual(b.count(p, 7, 9), 1)
588
Christian Heimes510711d2008-01-30 11:57:58 +0000589 def test_startswith(self):
590 b = self.type2test(b'hello')
591 self.assertFalse(self.type2test().startswith(b"anything"))
592 self.assertTrue(b.startswith(b"hello"))
593 self.assertTrue(b.startswith(b"hel"))
594 self.assertTrue(b.startswith(b"h"))
595 self.assertFalse(b.startswith(b"hellow"))
596 self.assertFalse(b.startswith(b"ha"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300597 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300598 b.startswith([b'h'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300599 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300600 self.assertIn('bytes', exc)
601 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000602
603 def test_endswith(self):
604 b = self.type2test(b'hello')
605 self.assertFalse(bytearray().endswith(b"anything"))
606 self.assertTrue(b.endswith(b"hello"))
607 self.assertTrue(b.endswith(b"llo"))
608 self.assertTrue(b.endswith(b"o"))
609 self.assertFalse(b.endswith(b"whello"))
610 self.assertFalse(b.endswith(b"no"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300611 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300612 b.endswith([b'o'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300613 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300614 self.assertIn('bytes', exc)
615 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000616
617 def test_find(self):
618 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200619 i = 105
620 w = 119
621
Christian Heimes510711d2008-01-30 11:57:58 +0000622 self.assertEqual(b.find(b'ss'), 2)
Christian Heimes510711d2008-01-30 11:57:58 +0000623 self.assertEqual(b.find(b'w'), -1)
624 self.assertEqual(b.find(b'mississippian'), -1)
625
Antoine Pitrouac65d962011-10-20 23:54:17 +0200626 self.assertEqual(b.find(i), 1)
627 self.assertEqual(b.find(w), -1)
628
629 self.assertEqual(b.find(b'ss', 3), 5)
630 self.assertEqual(b.find(b'ss', 1, 7), 2)
631 self.assertEqual(b.find(b'ss', 1, 3), -1)
632
633 self.assertEqual(b.find(i, 6), 7)
634 self.assertEqual(b.find(i, 1, 3), 1)
635 self.assertEqual(b.find(w, 1, 3), -1)
636
Victor Stinnerf8eac002011-12-18 01:17:41 +0100637 for index in (-1, 256, sys.maxsize + 1):
638 self.assertRaisesRegex(
639 ValueError, r'byte must be in range\(0, 256\)',
640 b.find, index)
641
Christian Heimes510711d2008-01-30 11:57:58 +0000642 def test_rfind(self):
643 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200644 i = 105
645 w = 119
646
Christian Heimes510711d2008-01-30 11:57:58 +0000647 self.assertEqual(b.rfind(b'ss'), 5)
Christian Heimes510711d2008-01-30 11:57:58 +0000648 self.assertEqual(b.rfind(b'w'), -1)
649 self.assertEqual(b.rfind(b'mississippian'), -1)
650
Antoine Pitrouac65d962011-10-20 23:54:17 +0200651 self.assertEqual(b.rfind(i), 10)
652 self.assertEqual(b.rfind(w), -1)
653
654 self.assertEqual(b.rfind(b'ss', 3), 5)
655 self.assertEqual(b.rfind(b'ss', 0, 6), 2)
656
657 self.assertEqual(b.rfind(i, 1, 3), 1)
658 self.assertEqual(b.rfind(i, 3, 9), 7)
659 self.assertEqual(b.rfind(w, 1, 3), -1)
660
Christian Heimes510711d2008-01-30 11:57:58 +0000661 def test_index(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200662 b = self.type2test(b'mississippi')
663 i = 105
664 w = 119
665
666 self.assertEqual(b.index(b'ss'), 2)
667 self.assertRaises(ValueError, b.index, b'w')
668 self.assertRaises(ValueError, b.index, b'mississippian')
669
670 self.assertEqual(b.index(i), 1)
671 self.assertRaises(ValueError, b.index, w)
672
673 self.assertEqual(b.index(b'ss', 3), 5)
674 self.assertEqual(b.index(b'ss', 1, 7), 2)
675 self.assertRaises(ValueError, b.index, b'ss', 1, 3)
676
677 self.assertEqual(b.index(i, 6), 7)
678 self.assertEqual(b.index(i, 1, 3), 1)
679 self.assertRaises(ValueError, b.index, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000680
681 def test_rindex(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200682 b = self.type2test(b'mississippi')
683 i = 105
684 w = 119
685
686 self.assertEqual(b.rindex(b'ss'), 5)
687 self.assertRaises(ValueError, b.rindex, b'w')
688 self.assertRaises(ValueError, b.rindex, b'mississippian')
689
690 self.assertEqual(b.rindex(i), 10)
691 self.assertRaises(ValueError, b.rindex, w)
692
693 self.assertEqual(b.rindex(b'ss', 3), 5)
694 self.assertEqual(b.rindex(b'ss', 0, 6), 2)
695
696 self.assertEqual(b.rindex(i, 1, 3), 1)
697 self.assertEqual(b.rindex(i, 3, 9), 7)
698 self.assertRaises(ValueError, b.rindex, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000699
Ethan Furmanb95b5612015-01-23 20:05:18 -0800700 def test_mod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300701 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800702 orig = b
703 b = b % b'world'
704 self.assertEqual(b, b'hello, world!')
705 self.assertEqual(orig, b'hello, %b!')
706 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300707 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800708 a = b % (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200709 self.assertEqual(a, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200710 self.assertIs(type(a), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800711 # issue 29714
712 b = self.type2test(b'hello,\x00%b!')
713 b = b % b'world'
714 self.assertEqual(b, b'hello,\x00world!')
715 self.assertIs(type(b), self.type2test)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800716
717 def test_imod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300718 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800719 orig = b
720 b %= b'world'
721 self.assertEqual(b, b'hello, world!')
722 self.assertEqual(orig, b'hello, %b!')
723 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300724 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800725 b %= (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200726 self.assertEqual(b, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200727 self.assertIs(type(b), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800728 # issue 29714
729 b = self.type2test(b'hello,\x00%b!')
730 b %= b'world'
731 self.assertEqual(b, b'hello,\x00world!')
732 self.assertIs(type(b), self.type2test)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300733
734 def test_rmod(self):
735 with self.assertRaises(TypeError):
736 object() % self.type2test(b'abc')
737 self.assertIs(self.type2test(b'abc').__rmod__('%r'), NotImplemented)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800738
Christian Heimes510711d2008-01-30 11:57:58 +0000739 def test_replace(self):
740 b = self.type2test(b'mississippi')
741 self.assertEqual(b.replace(b'i', b'a'), b'massassappa')
742 self.assertEqual(b.replace(b'ss', b'x'), b'mixixippi')
743
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300744 def test_replace_int_error(self):
745 self.assertRaises(TypeError, self.type2test(b'a b').replace, 32, b'')
746
Christian Heimes510711d2008-01-30 11:57:58 +0000747 def test_split_string_error(self):
748 self.assertRaises(TypeError, self.type2test(b'a b').split, ' ')
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300749 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, ' ')
750
751 def test_split_int_error(self):
752 self.assertRaises(TypeError, self.type2test(b'a b').split, 32)
753 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000754
Antoine Pitrouf8453022010-01-12 22:05:42 +0000755 def test_split_unicodewhitespace(self):
Martin Panter0d0db6c2016-04-10 08:45:26 +0000756 for b in (b'a\x1Cb', b'a\x1Db', b'a\x1Eb', b'a\x1Fb'):
757 b = self.type2test(b)
758 self.assertEqual(b.split(), [b])
Antoine Pitrouf8453022010-01-12 22:05:42 +0000759 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
760 self.assertEqual(b.split(), [b'\x1c\x1d\x1e\x1f'])
761
Christian Heimes510711d2008-01-30 11:57:58 +0000762 def test_rsplit_unicodewhitespace(self):
763 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
Christian Heimes510711d2008-01-30 11:57:58 +0000764 self.assertEqual(b.rsplit(), [b'\x1c\x1d\x1e\x1f'])
765
766 def test_partition(self):
767 b = self.type2test(b'mississippi')
768 self.assertEqual(b.partition(b'ss'), (b'mi', b'ss', b'issippi'))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000769 self.assertEqual(b.partition(b'w'), (b'mississippi', b'', b''))
Christian Heimes510711d2008-01-30 11:57:58 +0000770
771 def test_rpartition(self):
772 b = self.type2test(b'mississippi')
773 self.assertEqual(b.rpartition(b'ss'), (b'missi', b'ss', b'ippi'))
774 self.assertEqual(b.rpartition(b'i'), (b'mississipp', b'i', b''))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000775 self.assertEqual(b.rpartition(b'w'), (b'', b'', b'mississippi'))
Christian Heimes510711d2008-01-30 11:57:58 +0000776
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300777 def test_partition_string_error(self):
778 self.assertRaises(TypeError, self.type2test(b'a b').partition, ' ')
779 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, ' ')
780
781 def test_partition_int_error(self):
782 self.assertRaises(TypeError, self.type2test(b'a b').partition, 32)
783 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, 32)
784
Christian Heimes510711d2008-01-30 11:57:58 +0000785 def test_pickling(self):
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +0000786 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Christian Heimes510711d2008-01-30 11:57:58 +0000787 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
788 b = self.type2test(b)
789 ps = pickle.dumps(b, proto)
790 q = pickle.loads(ps)
791 self.assertEqual(b, q)
792
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000793 def test_iterator_pickling(self):
Serhiy Storchakabad12572014-12-15 14:03:42 +0200794 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
795 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
796 it = itorg = iter(self.type2test(b))
797 data = list(self.type2test(b))
798 d = pickle.dumps(it, proto)
799 it = pickle.loads(d)
800 self.assertEqual(type(itorg), type(it))
801 self.assertEqual(list(it), data)
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000802
Serhiy Storchakabad12572014-12-15 14:03:42 +0200803 it = pickle.loads(d)
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200804 if not b:
Serhiy Storchakabad12572014-12-15 14:03:42 +0200805 continue
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200806 next(it)
Serhiy Storchakabad12572014-12-15 14:03:42 +0200807 d = pickle.dumps(it, proto)
808 it = pickle.loads(d)
809 self.assertEqual(list(it), data[1:])
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000810
Christian Heimes510711d2008-01-30 11:57:58 +0000811 def test_strip_bytearray(self):
812 self.assertEqual(self.type2test(b'abc').strip(memoryview(b'ac')), b'b')
813 self.assertEqual(self.type2test(b'abc').lstrip(memoryview(b'ac')), b'bc')
814 self.assertEqual(self.type2test(b'abc').rstrip(memoryview(b'ac')), b'ab')
815
816 def test_strip_string_error(self):
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300817 self.assertRaises(TypeError, self.type2test(b'abc').strip, 'ac')
818 self.assertRaises(TypeError, self.type2test(b'abc').lstrip, 'ac')
819 self.assertRaises(TypeError, self.type2test(b'abc').rstrip, 'ac')
820
821 def test_strip_int_error(self):
822 self.assertRaises(TypeError, self.type2test(b' abc ').strip, 32)
823 self.assertRaises(TypeError, self.type2test(b' abc ').lstrip, 32)
824 self.assertRaises(TypeError, self.type2test(b' abc ').rstrip, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000825
Eli Bendersky906b88f2011-07-29 07:05:08 +0300826 def test_center(self):
827 # Fill character can be either bytes or bytearray (issue 12380)
828 b = self.type2test(b'abc')
829 for fill_type in (bytes, bytearray):
830 self.assertEqual(b.center(7, fill_type(b'-')),
831 self.type2test(b'--abc--'))
832
833 def test_ljust(self):
834 # Fill character can be either bytes or bytearray (issue 12380)
835 b = self.type2test(b'abc')
836 for fill_type in (bytes, bytearray):
837 self.assertEqual(b.ljust(7, fill_type(b'-')),
838 self.type2test(b'abc----'))
839
840 def test_rjust(self):
841 # Fill character can be either bytes or bytearray (issue 12380)
842 b = self.type2test(b'abc')
843 for fill_type in (bytes, bytearray):
844 self.assertEqual(b.rjust(7, fill_type(b'-')),
845 self.type2test(b'----abc'))
846
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300847 def test_xjust_int_error(self):
848 self.assertRaises(TypeError, self.type2test(b'abc').center, 7, 32)
849 self.assertRaises(TypeError, self.type2test(b'abc').ljust, 7, 32)
850 self.assertRaises(TypeError, self.type2test(b'abc').rjust, 7, 32)
851
Christian Heimes510711d2008-01-30 11:57:58 +0000852 def test_ord(self):
853 b = self.type2test(b'\0A\x7f\x80\xff')
854 self.assertEqual([ord(b[i:i+1]) for i in range(len(b))],
855 [0, 65, 127, 128, 255])
856
Georg Brandlabc38772009-04-12 15:51:51 +0000857 def test_maketrans(self):
858 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 +0000859 self.assertEqual(self.type2test.maketrans(b'abc', b'xyz'), transtable)
Antoine Pitrou403ce782009-10-14 17:14:16 +0000860 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'
861 self.assertEqual(self.type2test.maketrans(b'\375\376\377', b'xyz'), transtable)
Georg Brandlabc38772009-04-12 15:51:51 +0000862 self.assertRaises(ValueError, self.type2test.maketrans, b'abc', b'xyzq')
863 self.assertRaises(TypeError, self.type2test.maketrans, 'abc', 'def')
864
Jesus Ceaac451502011-04-20 17:09:23 +0200865 def test_none_arguments(self):
866 # issue 11828
867 b = self.type2test(b'hello')
868 l = self.type2test(b'l')
869 h = self.type2test(b'h')
870 x = self.type2test(b'x')
871 o = self.type2test(b'o')
872
873 self.assertEqual(2, b.find(l, None))
874 self.assertEqual(3, b.find(l, -2, None))
875 self.assertEqual(2, b.find(l, None, -2))
876 self.assertEqual(0, b.find(h, None, None))
877
878 self.assertEqual(3, b.rfind(l, None))
879 self.assertEqual(3, b.rfind(l, -2, None))
880 self.assertEqual(2, b.rfind(l, None, -2))
881 self.assertEqual(0, b.rfind(h, None, None))
882
883 self.assertEqual(2, b.index(l, None))
884 self.assertEqual(3, b.index(l, -2, None))
885 self.assertEqual(2, b.index(l, None, -2))
886 self.assertEqual(0, b.index(h, None, None))
887
888 self.assertEqual(3, b.rindex(l, None))
889 self.assertEqual(3, b.rindex(l, -2, None))
890 self.assertEqual(2, b.rindex(l, None, -2))
891 self.assertEqual(0, b.rindex(h, None, None))
892
893 self.assertEqual(2, b.count(l, None))
894 self.assertEqual(1, b.count(l, -2, None))
895 self.assertEqual(1, b.count(l, None, -2))
896 self.assertEqual(0, b.count(x, None, None))
897
898 self.assertEqual(True, b.endswith(o, None))
899 self.assertEqual(True, b.endswith(o, -2, None))
900 self.assertEqual(True, b.endswith(l, None, -2))
901 self.assertEqual(False, b.endswith(x, None, None))
902
903 self.assertEqual(True, b.startswith(h, None))
904 self.assertEqual(True, b.startswith(l, -2, None))
905 self.assertEqual(True, b.startswith(h, None, -2))
906 self.assertEqual(False, b.startswith(x, None, None))
907
Antoine Pitrouac65d962011-10-20 23:54:17 +0200908 def test_integer_arguments_out_of_byte_range(self):
909 b = self.type2test(b'hello')
910
911 for method in (b.count, b.find, b.index, b.rfind, b.rindex):
912 self.assertRaises(ValueError, method, -1)
913 self.assertRaises(ValueError, method, 256)
914 self.assertRaises(ValueError, method, 9999)
915
Jesus Ceaac451502011-04-20 17:09:23 +0200916 def test_find_etc_raise_correct_error_messages(self):
917 # issue 11828
918 b = self.type2test(b'hello')
919 x = self.type2test(b'x')
Ezio Melottiaf928422011-04-20 21:56:21 +0300920 self.assertRaisesRegex(TypeError, r'\bfind\b', b.find,
Jesus Ceaac451502011-04-20 17:09:23 +0200921 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300922 self.assertRaisesRegex(TypeError, r'\brfind\b', b.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +0200923 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300924 self.assertRaisesRegex(TypeError, r'\bindex\b', b.index,
Jesus Ceaac451502011-04-20 17:09:23 +0200925 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300926 self.assertRaisesRegex(TypeError, r'\brindex\b', b.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +0200927 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300928 self.assertRaisesRegex(TypeError, r'\bcount\b', b.count,
Jesus Ceaac451502011-04-20 17:09:23 +0200929 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300930 self.assertRaisesRegex(TypeError, r'\bstartswith\b', b.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200931 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300932 self.assertRaisesRegex(TypeError, r'\bendswith\b', b.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200933 x, None, None, None)
934
Serhiy Storchakafbb1c5e2016-03-30 20:40:02 +0300935 def test_free_after_iterating(self):
936 test.support.check_free_after_iterating(self, iter, self.type2test)
937 test.support.check_free_after_iterating(self, reversed, self.type2test)
938
Martin Panter1b6c6da2016-08-27 08:35:02 +0000939 def test_translate(self):
940 b = self.type2test(b'hello')
941 rosetta = bytearray(range(256))
942 rosetta[ord('o')] = ord('e')
943
944 self.assertRaises(TypeError, b.translate)
945 self.assertRaises(TypeError, b.translate, None, None)
946 self.assertRaises(ValueError, b.translate, bytes(range(255)))
947
948 c = b.translate(rosetta, b'hello')
949 self.assertEqual(b, b'hello')
950 self.assertIsInstance(c, self.type2test)
951
952 c = b.translate(rosetta)
953 d = b.translate(rosetta, b'')
954 self.assertEqual(c, d)
955 self.assertEqual(c, b'helle')
956
957 c = b.translate(rosetta, b'l')
958 self.assertEqual(c, b'hee')
959 c = b.translate(None, b'e')
960 self.assertEqual(c, b'hllo')
961
962 # test delete as a keyword argument
963 c = b.translate(rosetta, delete=b'')
964 self.assertEqual(c, b'helle')
965 c = b.translate(rosetta, delete=b'l')
966 self.assertEqual(c, b'hee')
967 c = b.translate(None, delete=b'e')
968 self.assertEqual(c, b'hllo')
969
Sergey Fedoseev92709a22019-09-09 21:28:34 +0500970 def test_sq_item(self):
Hai Shia089d212020-07-06 17:15:08 +0800971 _testcapi = import_helper.import_module('_testcapi')
Sergey Fedoseev92709a22019-09-09 21:28:34 +0500972 obj = self.type2test((42,))
973 with self.assertRaises(IndexError):
974 _testcapi.sequence_getitem(obj, -2)
975 with self.assertRaises(IndexError):
976 _testcapi.sequence_getitem(obj, 1)
977 self.assertEqual(_testcapi.sequence_getitem(obj, 0), 42)
978
Christian Heimes510711d2008-01-30 11:57:58 +0000979
Ezio Melotti0dceb562013-01-10 07:43:26 +0200980class BytesTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +0000981 type2test = bytes
982
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400983 def test_getitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200984 b = b'python'
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400985 msg = "byte indices must be integers or slices"
986 with self.assertRaisesRegex(TypeError, msg):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200987 b['a']
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400988
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000989 def test_buffer_is_readonly(self):
Victor Stinner76119642014-07-28 22:07:07 +0200990 fd = os.open(__file__, os.O_RDONLY)
Antoine Pitrou47d305d2008-08-16 23:28:44 +0000991 with open(fd, "rb", buffering=0) as f:
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000992 self.assertRaises(TypeError, f.readinto, b"")
993
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000994 def test_custom(self):
995 class A:
996 def __bytes__(self):
997 return b'abc'
998 self.assertEqual(bytes(A()), b'abc')
999 class A: pass
1000 self.assertRaises(TypeError, bytes, A())
1001 class A:
1002 def __bytes__(self):
1003 return None
1004 self.assertRaises(TypeError, bytes, A())
Benjamin Peterson5ff3f732012-12-19 15:27:41 -06001005 class A:
1006 def __bytes__(self):
1007 return b'a'
1008 def __index__(self):
1009 return 42
1010 self.assertEqual(bytes(A()), b'a')
Serhiy Storchaka5aac3ed2015-12-20 16:36:34 +02001011 # Issue #25766
1012 class A(str):
1013 def __bytes__(self):
1014 return b'abc'
1015 self.assertEqual(bytes(A('\u20ac')), b'abc')
1016 self.assertEqual(bytes(A('\u20ac'), 'iso8859-15'), b'\xa4')
Serhiy Storchaka15095802015-11-25 15:47:01 +02001017 # Issue #24731
1018 class A:
1019 def __bytes__(self):
1020 return OtherBytesSubclass(b'abc')
1021 self.assertEqual(bytes(A()), b'abc')
1022 self.assertIs(type(bytes(A())), OtherBytesSubclass)
1023 self.assertEqual(BytesSubclass(A()), b'abc')
1024 self.assertIs(type(BytesSubclass(A())), BytesSubclass)
Benjamin Petersonc15a0732008-08-26 16:46:47 +00001025
Victor Stinner29e762c2011-01-05 03:33:28 +00001026 # Test PyBytes_FromFormat()
1027 def test_from_format(self):
Hai Shia089d212020-07-06 17:15:08 +08001028 ctypes = import_helper.import_module('ctypes')
1029 _testcapi = import_helper.import_module('_testcapi')
Victor Stinner03dab782015-10-14 00:21:35 +02001030 from ctypes import pythonapi, py_object
1031 from ctypes import (
1032 c_int, c_uint,
1033 c_long, c_ulong,
1034 c_size_t, c_ssize_t,
1035 c_char_p)
1036
Victor Stinner29e762c2011-01-05 03:33:28 +00001037 PyBytes_FromFormat = pythonapi.PyBytes_FromFormat
1038 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()