blob: 770e2c5592cc6167483bad32aeabad68146e0cda [file] [log] [blame]
Guido van Rossum254348e2007-11-21 19:29:53 +00001"""Unit tests for the bytes and bytearray types.
Guido van Rossum98297ee2007-11-06 21:34:58 +00002
Martin Panter152a19c2016-04-06 06:37:17 +00003XXX This is a mess. Common tests should be unified with string_tests.py (and
4the latter should be modernized).
Guido van Rossum98297ee2007-11-06 21:34:58 +00005"""
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00006
INADA Naokia634e232017-01-06 17:32:01 +09007import array
Guido van Rossumd624f182006-04-24 13:47:05 +00008import os
9import re
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000010import sys
Guido van Rossuma6c04be2007-11-03 00:24:24 +000011import copy
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000012import functools
Guido van Rossum0ad08122007-04-11 04:37:43 +000013import pickle
Guido van Rossumd624f182006-04-24 13:47:05 +000014import tempfile
Victor Stinner22eb6892019-06-26 00:51:05 +020015import textwrap
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000016import unittest
Antoine Pitroucc231542014-11-02 18:40:09 +010017
Benjamin Petersonee8712c2008-05-20 21:35:26 +000018import test.support
Georg Brandlc7885542007-03-06 19:16:20 +000019import test.string_tests
Serhiy Storchaka8dc2ec12016-03-30 21:01:26 +030020import test.list_tests
Antoine Pitroucc231542014-11-02 18:40:09 +010021from test.support import bigaddrspacetest, MAX_Py_ssize_t
Victor Stinner22eb6892019-06-26 00:51:05 +020022from test.support.script_helper import assert_python_failure
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000023
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +000024
25if sys.flags.bytes_warning:
26 def check_bytes_warnings(func):
27 @functools.wraps(func)
28 def wrapper(*args, **kw):
29 with test.support.check_warnings(('', BytesWarning)):
30 return func(*args, **kw)
31 return wrapper
32else:
33 # no-op
34 def check_bytes_warnings(func):
35 return func
36
37
Georg Brandl9a54d7c2008-07-16 23:15:30 +000038class Indexable:
39 def __init__(self, value=0):
40 self.value = value
41 def __index__(self):
42 return self.value
43
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000044
Ezio Melotti0dceb562013-01-10 07:43:26 +020045class BaseBytesTest:
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000046
47 def test_basics(self):
Christian Heimes510711d2008-01-30 11:57:58 +000048 b = self.type2test()
49 self.assertEqual(type(b), self.type2test)
50 self.assertEqual(b.__class__, self.type2test)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000051
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +000052 def test_copy(self):
53 a = self.type2test(b"abcd")
54 for copy_method in (copy.copy, copy.deepcopy):
55 b = copy_method(a)
56 self.assertEqual(a, b)
57 self.assertEqual(type(a), type(b))
58
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000059 def test_empty_sequence(self):
Christian Heimes510711d2008-01-30 11:57:58 +000060 b = self.type2test()
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000061 self.assertEqual(len(b), 0)
62 self.assertRaises(IndexError, lambda: b[0])
63 self.assertRaises(IndexError, lambda: b[1])
Christian Heimesa37d4c62007-12-04 23:02:19 +000064 self.assertRaises(IndexError, lambda: b[sys.maxsize])
65 self.assertRaises(IndexError, lambda: b[sys.maxsize+1])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000066 self.assertRaises(IndexError, lambda: b[10**100])
67 self.assertRaises(IndexError, lambda: b[-1])
68 self.assertRaises(IndexError, lambda: b[-2])
Christian Heimesa37d4c62007-12-04 23:02:19 +000069 self.assertRaises(IndexError, lambda: b[-sys.maxsize])
70 self.assertRaises(IndexError, lambda: b[-sys.maxsize-1])
71 self.assertRaises(IndexError, lambda: b[-sys.maxsize-2])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000072 self.assertRaises(IndexError, lambda: b[-10**100])
73
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030074 def test_from_iterable(self):
75 b = self.type2test(range(256))
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000076 self.assertEqual(len(b), 256)
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +030077 self.assertEqual(list(b), list(range(256)))
78
79 # Non-sequence iterable.
80 b = self.type2test({42})
81 self.assertEqual(b, b"*")
82 b = self.type2test({43, 45})
83 self.assertIn(tuple(b), {(43, 45), (45, 43)})
84
85 # Iterator that has a __length_hint__.
86 b = self.type2test(iter(range(256)))
87 self.assertEqual(len(b), 256)
88 self.assertEqual(list(b), list(range(256)))
89
90 # Iterator that doesn't have a __length_hint__.
91 b = self.type2test(i for i in range(256) if i % 2)
92 self.assertEqual(len(b), 128)
93 self.assertEqual(list(b), list(range(256))[1::2])
94
95 # Sequence without __iter__.
96 class S:
97 def __getitem__(self, i):
98 return (1, 2, 3)[i]
99 b = self.type2test(S())
100 self.assertEqual(b, b"\x01\x02\x03")
101
102 def test_from_tuple(self):
103 # There is a special case for tuples.
104 b = self.type2test(tuple(range(256)))
105 self.assertEqual(len(b), 256)
106 self.assertEqual(list(b), list(range(256)))
107 b = self.type2test((1, 2, 3))
108 self.assertEqual(b, b"\x01\x02\x03")
109
110 def test_from_list(self):
111 # There is a special case for lists.
112 b = self.type2test(list(range(256)))
113 self.assertEqual(len(b), 256)
114 self.assertEqual(list(b), list(range(256)))
115 b = self.type2test([1, 2, 3])
116 self.assertEqual(b, b"\x01\x02\x03")
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000117
Serhiy Storchaka914f9a02018-10-21 15:25:53 +0300118 def test_from_mutating_list(self):
119 # Issue #34973: Crash in bytes constructor with mutating list.
120 class X:
121 def __index__(self):
122 a.clear()
123 return 42
124 a = [X(), X()]
125 self.assertEqual(bytes(a), b'*')
126
127 class Y:
128 def __index__(self):
129 if len(a) < 1000:
130 a.append(self)
131 return 42
132 a = [Y()]
133 self.assertEqual(bytes(a), b'*' * 1000) # should not crash
134
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000135 def test_from_index(self):
Georg Brandl9a54d7c2008-07-16 23:15:30 +0000136 b = self.type2test([Indexable(), Indexable(1), Indexable(254),
137 Indexable(255)])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000138 self.assertEqual(list(b), [0, 1, 254, 255])
Benjamin Peterson4c045832010-04-16 22:35:32 +0000139 self.assertRaises(ValueError, self.type2test, [Indexable(-1)])
140 self.assertRaises(ValueError, self.type2test, [Indexable(256)])
Guido van Rossum98297ee2007-11-06 21:34:58 +0000141
INADA Naokia634e232017-01-06 17:32:01 +0900142 def test_from_buffer(self):
143 a = self.type2test(array.array('B', [1, 2, 3]))
144 self.assertEqual(a, b"\x01\x02\x03")
Serhiy Storchaka1a997eb2018-10-13 20:39:45 +0300145 a = self.type2test(b"\x01\x02\x03")
146 self.assertEqual(a, b"\x01\x02\x03")
INADA Naokia634e232017-01-06 17:32:01 +0900147
Serhiy Storchakae8904212018-10-15 00:02:57 +0300148 # Issues #29159 and #34974.
149 # Fallback when __index__ raises a TypeError
INADA Naokia634e232017-01-06 17:32:01 +0900150 class B(bytes):
151 def __index__(self):
152 raise TypeError
153
154 self.assertEqual(self.type2test(B(b"foobar")), b"foobar")
155
Guido van Rossum98297ee2007-11-06 21:34:58 +0000156 def test_from_ssize(self):
Benjamin Peterson4c045832010-04-16 22:35:32 +0000157 self.assertEqual(self.type2test(0), b'')
158 self.assertEqual(self.type2test(1), b'\x00')
159 self.assertEqual(self.type2test(5), b'\x00\x00\x00\x00\x00')
160 self.assertRaises(ValueError, self.type2test, -1)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000161
Benjamin Peterson4c045832010-04-16 22:35:32 +0000162 self.assertEqual(self.type2test('0', 'ascii'), b'0')
163 self.assertEqual(self.type2test(b'0'), b'0')
Benjamin Peterson8380dd52010-04-16 22:51:37 +0000164 self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000165
166 def test_constructor_type_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000167 self.assertRaises(TypeError, self.type2test, 0.0)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000168 class C:
169 pass
Christian Heimes510711d2008-01-30 11:57:58 +0000170 self.assertRaises(TypeError, self.type2test, ["0"])
171 self.assertRaises(TypeError, self.type2test, [0.0])
172 self.assertRaises(TypeError, self.type2test, [None])
173 self.assertRaises(TypeError, self.type2test, [C()])
Serhiy Storchaka2c2044e2018-10-21 15:29:12 +0300174 self.assertRaises(TypeError, self.type2test, encoding='ascii')
175 self.assertRaises(TypeError, self.type2test, errors='ignore')
Serhiy Storchaka83cf99d2014-12-02 09:24:06 +0200176 self.assertRaises(TypeError, self.type2test, 0, 'ascii')
177 self.assertRaises(TypeError, self.type2test, b'', 'ascii')
178 self.assertRaises(TypeError, self.type2test, 0, errors='ignore')
179 self.assertRaises(TypeError, self.type2test, b'', errors='ignore')
180 self.assertRaises(TypeError, self.type2test, '')
181 self.assertRaises(TypeError, self.type2test, '', errors='ignore')
182 self.assertRaises(TypeError, self.type2test, '', b'ascii')
183 self.assertRaises(TypeError, self.type2test, '', 'ascii', b'ignore')
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000184
185 def test_constructor_value_errors(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000186 self.assertRaises(ValueError, self.type2test, [-1])
187 self.assertRaises(ValueError, self.type2test, [-sys.maxsize])
188 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-1])
189 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-2])
190 self.assertRaises(ValueError, self.type2test, [-10**100])
191 self.assertRaises(ValueError, self.type2test, [256])
192 self.assertRaises(ValueError, self.type2test, [257])
193 self.assertRaises(ValueError, self.type2test, [sys.maxsize])
194 self.assertRaises(ValueError, self.type2test, [sys.maxsize+1])
195 self.assertRaises(ValueError, self.type2test, [10**100])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000196
Antoine Pitroucc231542014-11-02 18:40:09 +0100197 @bigaddrspacetest
198 def test_constructor_overflow(self):
199 size = MAX_Py_ssize_t
200 self.assertRaises((OverflowError, MemoryError), self.type2test, size)
201 try:
202 # Should either pass or raise an error (e.g. on debug builds with
203 # additional malloc() overhead), but shouldn't crash.
204 bytearray(size - 4)
205 except (OverflowError, MemoryError):
206 pass
207
Serhiy Storchakae8904212018-10-15 00:02:57 +0300208 def test_constructor_exceptions(self):
209 # Issue #34974: bytes and bytearray constructors replace unexpected
210 # exceptions.
211 class BadInt:
212 def __index__(self):
213 1/0
214 self.assertRaises(ZeroDivisionError, self.type2test, BadInt())
215 self.assertRaises(ZeroDivisionError, self.type2test, [BadInt()])
216
217 class BadIterable:
218 def __iter__(self):
219 1/0
220 self.assertRaises(ZeroDivisionError, self.type2test, BadIterable())
221
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000222 def test_compare(self):
Christian Heimes510711d2008-01-30 11:57:58 +0000223 b1 = self.type2test([1, 2, 3])
224 b2 = self.type2test([1, 2, 3])
225 b3 = self.type2test([1, 3])
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000226
Guido van Rossume61fd5b2007-07-11 12:20:59 +0000227 self.assertEqual(b1, b2)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000228 self.assertTrue(b2 != b3)
229 self.assertTrue(b1 <= b2)
230 self.assertTrue(b1 <= b3)
231 self.assertTrue(b1 < b3)
232 self.assertTrue(b1 >= b2)
233 self.assertTrue(b3 >= b2)
234 self.assertTrue(b3 > b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000235
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000236 self.assertFalse(b1 != b2)
237 self.assertFalse(b2 == b3)
238 self.assertFalse(b1 > b2)
239 self.assertFalse(b1 > b3)
240 self.assertFalse(b1 >= b3)
241 self.assertFalse(b1 < b2)
242 self.assertFalse(b3 < b2)
243 self.assertFalse(b3 <= b2)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000244
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000245 @check_bytes_warnings
Guido van Rossum343e97f2007-04-09 00:43:24 +0000246 def test_compare_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +0000247 # Byte comparisons with unicode should always fail!
248 # Test this for all expected byte orders and Unicode character
249 # sizes.
250 self.assertEqual(self.type2test(b"\0a\0b\0c") == "abc", False)
251 self.assertEqual(self.type2test(b"\0\0\0a\0\0\0b\0\0\0c") == "abc",
252 False)
253 self.assertEqual(self.type2test(b"a\0b\0c\0") == "abc", False)
254 self.assertEqual(self.type2test(b"a\0\0\0b\0\0\0c\0\0\0") == "abc",
255 False)
256 self.assertEqual(self.type2test() == str(), False)
257 self.assertEqual(self.type2test() != str(), True)
Christian Heimes510711d2008-01-30 11:57:58 +0000258
259 def test_reversed(self):
260 input = list(map(ord, "Hello"))
261 b = self.type2test(input)
262 output = list(reversed(b))
263 input.reverse()
264 self.assertEqual(output, input)
265
266 def test_getslice(self):
267 def by(s):
268 return self.type2test(map(ord, s))
269 b = by("Hello, world")
270
271 self.assertEqual(b[:5], by("Hello"))
272 self.assertEqual(b[1:5], by("ello"))
273 self.assertEqual(b[5:7], by(", "))
274 self.assertEqual(b[7:], by("world"))
275 self.assertEqual(b[7:12], by("world"))
276 self.assertEqual(b[7:100], by("world"))
277
278 self.assertEqual(b[:-7], by("Hello"))
279 self.assertEqual(b[-11:-7], by("ello"))
280 self.assertEqual(b[-7:-5], by(", "))
281 self.assertEqual(b[-5:], by("world"))
282 self.assertEqual(b[-5:12], by("world"))
283 self.assertEqual(b[-5:100], by("world"))
284 self.assertEqual(b[-100:5], by("Hello"))
285
286 def test_extended_getslice(self):
287 # Test extended slicing by comparing with list slicing.
288 L = list(range(255))
289 b = self.type2test(L)
Zackery Spytz14514d92019-05-17 01:13:03 -0600290 indices = (0, None, 1, 3, 19, 100, sys.maxsize, -1, -2, -31, -100)
Christian Heimes510711d2008-01-30 11:57:58 +0000291 for start in indices:
292 for stop in indices:
293 # Skip step 0 (invalid)
294 for step in indices[1:]:
295 self.assertEqual(b[start:stop:step], self.type2test(L[start:stop:step]))
296
297 def test_encoding(self):
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000298 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000299 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000300 b = self.type2test(sample, enc)
301 self.assertEqual(b, self.type2test(sample.encode(enc)))
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000302 self.assertRaises(UnicodeEncodeError, self.type2test, sample, "latin-1")
303 b = self.type2test(sample, "latin-1", "ignore")
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000304 self.assertEqual(b, self.type2test(sample[:-3], "utf-8"))
Christian Heimes510711d2008-01-30 11:57:58 +0000305
306 def test_decode(self):
Serhiy Storchakac49a5162016-09-08 15:47:27 +0300307 sample = "Hello world\n\u1234\u5678\u9abc"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000308 for enc in ("utf-8", "utf-16"):
Christian Heimes510711d2008-01-30 11:57:58 +0000309 b = self.type2test(sample, enc)
310 self.assertEqual(b.decode(enc), sample)
311 sample = "Hello world\n\x80\x81\xfe\xff"
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000312 b = self.type2test(sample, "latin-1")
313 self.assertRaises(UnicodeDecodeError, b.decode, "utf-8")
314 self.assertEqual(b.decode("utf-8", "ignore"), "Hello world\n")
315 self.assertEqual(b.decode(errors="ignore", encoding="utf-8"),
Benjamin Peterson308d6372009-09-18 21:42:35 +0000316 "Hello world\n")
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000317 # Default encoding is utf-8
318 self.assertEqual(self.type2test(b'\xe2\x98\x83').decode(), '\u2603')
Christian Heimes510711d2008-01-30 11:57:58 +0000319
Victor Stinner22eb6892019-06-26 00:51:05 +0200320 def test_check_encoding_errors(self):
321 # bpo-37388: bytes(str) and bytes.encode() must check encoding
322 # and errors arguments in dev mode
323 invalid = 'Boom, Shaka Laka, Boom!'
324 encodings = ('ascii', 'utf8', 'latin1')
325 code = textwrap.dedent(f'''
326 import sys
327 type2test = {self.type2test.__name__}
328 encodings = {encodings!r}
329
330 for data in ('', 'short string'):
331 try:
332 type2test(data, encoding={invalid!r})
333 except LookupError:
334 pass
335 else:
336 sys.exit(21)
337
338 for encoding in encodings:
339 try:
340 type2test(data, encoding=encoding, errors={invalid!r})
341 except LookupError:
342 pass
343 else:
344 sys.exit(22)
345
346 for data in (b'', b'short string'):
347 data = type2test(data)
348 print(repr(data))
349 try:
350 data.decode(encoding={invalid!r})
351 except LookupError:
352 sys.exit(10)
353 else:
354 sys.exit(23)
355
356 try:
357 data.decode(errors={invalid!r})
358 except LookupError:
359 pass
360 else:
361 sys.exit(24)
362
363 for encoding in encodings:
364 try:
365 data.decode(encoding=encoding, errors={invalid!r})
366 except LookupError:
367 pass
368 else:
369 sys.exit(25)
370
371 sys.exit(10)
372 ''')
373 proc = assert_python_failure('-X', 'dev', '-c', code)
374 self.assertEqual(proc.rc, 10, proc)
375
Christian Heimes510711d2008-01-30 11:57:58 +0000376 def test_from_int(self):
377 b = self.type2test(0)
378 self.assertEqual(b, self.type2test())
379 b = self.type2test(10)
380 self.assertEqual(b, self.type2test([0]*10))
381 b = self.type2test(10000)
382 self.assertEqual(b, self.type2test([0]*10000))
383
384 def test_concat(self):
385 b1 = self.type2test(b"abc")
386 b2 = self.type2test(b"def")
387 self.assertEqual(b1 + b2, b"abcdef")
388 self.assertEqual(b1 + bytes(b"def"), b"abcdef")
389 self.assertEqual(bytes(b"def") + b1, b"defabc")
390 self.assertRaises(TypeError, lambda: b1 + "def")
391 self.assertRaises(TypeError, lambda: "abc" + b2)
392
393 def test_repeat(self):
394 for b in b"abc", self.type2test(b"abc"):
395 self.assertEqual(b * 3, b"abcabcabc")
396 self.assertEqual(b * 0, b"")
397 self.assertEqual(b * -1, b"")
398 self.assertRaises(TypeError, lambda: b * 3.14)
399 self.assertRaises(TypeError, lambda: 3.14 * b)
400 # XXX Shouldn't bytes and bytearray agree on what to raise?
Mark Dickinsoncf940c72010-08-10 18:35:01 +0000401 with self.assertRaises((OverflowError, MemoryError)):
402 c = b * sys.maxsize
403 with self.assertRaises((OverflowError, MemoryError)):
404 b *= sys.maxsize
Christian Heimes510711d2008-01-30 11:57:58 +0000405
406 def test_repeat_1char(self):
407 self.assertEqual(self.type2test(b'x')*100, self.type2test([ord('x')]*100))
408
409 def test_contains(self):
410 b = self.type2test(b"abc")
Benjamin Peterson577473f2010-01-19 00:09:57 +0000411 self.assertIn(ord('a'), b)
412 self.assertIn(int(ord('a')), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000413 self.assertNotIn(200, b)
Christian Heimes510711d2008-01-30 11:57:58 +0000414 self.assertRaises(ValueError, lambda: 300 in b)
415 self.assertRaises(ValueError, lambda: -1 in b)
Serhiy Storchakaf9efb8b2016-07-10 12:37:30 +0300416 self.assertRaises(ValueError, lambda: sys.maxsize+1 in b)
Christian Heimes510711d2008-01-30 11:57:58 +0000417 self.assertRaises(TypeError, lambda: None in b)
418 self.assertRaises(TypeError, lambda: float(ord('a')) in b)
419 self.assertRaises(TypeError, lambda: "a" in b)
420 for f in bytes, bytearray:
Benjamin Peterson577473f2010-01-19 00:09:57 +0000421 self.assertIn(f(b""), b)
422 self.assertIn(f(b"a"), b)
423 self.assertIn(f(b"b"), b)
424 self.assertIn(f(b"c"), b)
425 self.assertIn(f(b"ab"), b)
426 self.assertIn(f(b"bc"), b)
427 self.assertIn(f(b"abc"), b)
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000428 self.assertNotIn(f(b"ac"), b)
429 self.assertNotIn(f(b"d"), b)
430 self.assertNotIn(f(b"dab"), b)
431 self.assertNotIn(f(b"abd"), b)
Christian Heimes510711d2008-01-30 11:57:58 +0000432
433 def test_fromhex(self):
434 self.assertRaises(TypeError, self.type2test.fromhex)
435 self.assertRaises(TypeError, self.type2test.fromhex, 1)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000436 self.assertEqual(self.type2test.fromhex(''), self.type2test())
Christian Heimes510711d2008-01-30 11:57:58 +0000437 b = bytearray([0x1a, 0x2b, 0x30])
Ezio Melottib3aedd42010-11-20 19:04:17 +0000438 self.assertEqual(self.type2test.fromhex('1a2B30'), b)
439 self.assertEqual(self.type2test.fromhex(' 1A 2B 30 '), b)
Serhiy Storchakadd1da7f2016-12-19 18:51:37 +0200440
441 # check that ASCII whitespace is ignored
442 self.assertEqual(self.type2test.fromhex(' 1A\n2B\t30\v'), b)
443 for c in "\x09\x0A\x0B\x0C\x0D\x20":
444 self.assertEqual(self.type2test.fromhex(c), self.type2test())
445 for c in "\x1C\x1D\x1E\x1F\x85\xa0\u2000\u2002\u2028":
446 self.assertRaises(ValueError, self.type2test.fromhex, c)
447
Ezio Melottib3aedd42010-11-20 19:04:17 +0000448 self.assertEqual(self.type2test.fromhex('0000'), b'\0\0')
Christian Heimes510711d2008-01-30 11:57:58 +0000449 self.assertRaises(TypeError, self.type2test.fromhex, b'1B')
450 self.assertRaises(ValueError, self.type2test.fromhex, 'a')
451 self.assertRaises(ValueError, self.type2test.fromhex, 'rt')
452 self.assertRaises(ValueError, self.type2test.fromhex, '1a b cd')
453 self.assertRaises(ValueError, self.type2test.fromhex, '\x00')
454 self.assertRaises(ValueError, self.type2test.fromhex, '12 \x00 34')
455
Victor Stinner2bf89932015-10-14 11:25:33 +0200456 for data, pos in (
457 # invalid first hexadecimal character
458 ('12 x4 56', 3),
459 # invalid second hexadecimal character
460 ('12 3x 56', 4),
461 # two invalid hexadecimal characters
462 ('12 xy 56', 3),
463 # test non-ASCII string
464 ('12 3\xff 56', 4),
465 ):
466 with self.assertRaises(ValueError) as cm:
467 self.type2test.fromhex(data)
468 self.assertIn('at position %s' % pos, str(cm.exception))
469
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000470 def test_hex(self):
471 self.assertRaises(TypeError, self.type2test.hex)
472 self.assertRaises(TypeError, self.type2test.hex, 1)
Gregory P. Smith32d34bc2015-04-26 05:05:53 +0000473 self.assertEqual(self.type2test(b"").hex(), "")
474 self.assertEqual(bytearray([0x1a, 0x2b, 0x30]).hex(), '1a2b30')
475 self.assertEqual(self.type2test(b"\x1a\x2b\x30").hex(), '1a2b30')
476 self.assertEqual(memoryview(b"\x1a\x2b\x30").hex(), '1a2b30')
Gregory P. Smith8cb65692015-04-25 23:22:26 +0000477
Gregory P. Smith0c2f9302019-05-29 11:46:58 -0700478 def test_hex_separator_basics(self):
479 three_bytes = self.type2test(b'\xb9\x01\xef')
480 self.assertEqual(three_bytes.hex(), 'b901ef')
481 with self.assertRaises(ValueError):
482 three_bytes.hex('')
483 with self.assertRaises(ValueError):
484 three_bytes.hex('xx')
485 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
486 with self.assertRaises(TypeError):
487 three_bytes.hex(None, 0)
488 with self.assertRaises(ValueError):
489 three_bytes.hex('\xff')
490 with self.assertRaises(ValueError):
491 three_bytes.hex(b'\xff')
492 with self.assertRaises(ValueError):
493 three_bytes.hex(b'\x80')
494 with self.assertRaises(ValueError):
495 three_bytes.hex(chr(0x100))
496 self.assertEqual(three_bytes.hex(':', 0), 'b901ef')
497 self.assertEqual(three_bytes.hex(b'\x00'), 'b9\x0001\x00ef')
498 self.assertEqual(three_bytes.hex('\x00'), 'b9\x0001\x00ef')
499 self.assertEqual(three_bytes.hex(b'\x7f'), 'b9\x7f01\x7fef')
500 self.assertEqual(three_bytes.hex('\x7f'), 'b9\x7f01\x7fef')
501 self.assertEqual(three_bytes.hex(':', 3), 'b901ef')
502 self.assertEqual(three_bytes.hex(':', 4), 'b901ef')
503 self.assertEqual(three_bytes.hex(':', -4), 'b901ef')
504 self.assertEqual(three_bytes.hex(':'), 'b9:01:ef')
505 self.assertEqual(three_bytes.hex(b'$'), 'b9$01$ef')
506 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
507 self.assertEqual(three_bytes.hex(':', -1), 'b9:01:ef')
508 self.assertEqual(three_bytes.hex(':', 2), 'b9:01ef')
509 self.assertEqual(three_bytes.hex(':', 1), 'b9:01:ef')
510 self.assertEqual(three_bytes.hex('*', -2), 'b901*ef')
511
512 value = b'{s\005\000\000\000worldi\002\000\000\000s\005\000\000\000helloi\001\000\000\0000'
513 self.assertEqual(value.hex('.', 8), '7b7305000000776f.726c646902000000.730500000068656c.6c6f690100000030')
514
515 def test_hex_separator_five_bytes(self):
516 five_bytes = self.type2test(range(90,95))
517 self.assertEqual(five_bytes.hex(), '5a5b5c5d5e')
518
519 def test_hex_separator_six_bytes(self):
520 six_bytes = self.type2test(x*3 for x in range(1, 7))
521 self.assertEqual(six_bytes.hex(), '0306090c0f12')
522 self.assertEqual(six_bytes.hex('.', 1), '03.06.09.0c.0f.12')
523 self.assertEqual(six_bytes.hex(' ', 2), '0306 090c 0f12')
524 self.assertEqual(six_bytes.hex('-', 3), '030609-0c0f12')
525 self.assertEqual(six_bytes.hex(':', 4), '0306:090c0f12')
526 self.assertEqual(six_bytes.hex(':', 5), '03:06090c0f12')
527 self.assertEqual(six_bytes.hex(':', 6), '0306090c0f12')
528 self.assertEqual(six_bytes.hex(':', 95), '0306090c0f12')
529 self.assertEqual(six_bytes.hex('_', -3), '030609_0c0f12')
530 self.assertEqual(six_bytes.hex(':', -4), '0306090c:0f12')
531 self.assertEqual(six_bytes.hex(b'@', -5), '0306090c0f@12')
532 self.assertEqual(six_bytes.hex(':', -6), '0306090c0f12')
533 self.assertEqual(six_bytes.hex(' ', -95), '0306090c0f12')
534
Christian Heimes510711d2008-01-30 11:57:58 +0000535 def test_join(self):
536 self.assertEqual(self.type2test(b"").join([]), b"")
537 self.assertEqual(self.type2test(b"").join([b""]), b"")
538 for lst in [[b"abc"], [b"a", b"bc"], [b"ab", b"c"], [b"a", b"b", b"c"]]:
539 lst = list(map(self.type2test, lst))
540 self.assertEqual(self.type2test(b"").join(lst), b"abc")
541 self.assertEqual(self.type2test(b"").join(tuple(lst)), b"abc")
542 self.assertEqual(self.type2test(b"").join(iter(lst)), b"abc")
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200543 dot_join = self.type2test(b".:").join
544 self.assertEqual(dot_join([b"ab", b"cd"]), b"ab.:cd")
545 self.assertEqual(dot_join([memoryview(b"ab"), b"cd"]), b"ab.:cd")
546 self.assertEqual(dot_join([b"ab", memoryview(b"cd")]), b"ab.:cd")
547 self.assertEqual(dot_join([bytearray(b"ab"), b"cd"]), b"ab.:cd")
548 self.assertEqual(dot_join([b"ab", bytearray(b"cd")]), b"ab.:cd")
549 # Stress it with many items
Bruce Merryd07d9f42020-01-29 09:09:24 +0200550 seq = [b"abc"] * 100000
551 expected = b"abc" + b".:abc" * 99999
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200552 self.assertEqual(dot_join(seq), expected)
Bruce Merryd07d9f42020-01-29 09:09:24 +0200553 # Stress test with empty separator
554 seq = [b"abc"] * 100000
555 expected = b"abc" * 100000
556 self.assertEqual(self.type2test(b"").join(seq), expected)
Benjamin Petersonc31f12d2014-09-28 12:56:42 -0400557 self.assertRaises(TypeError, self.type2test(b" ").join, None)
Antoine Pitroucfc22b42012-10-16 21:07:23 +0200558 # Error handling and cleanup when some item in the middle of the
559 # sequence has the wrong type.
560 with self.assertRaises(TypeError):
561 dot_join([bytearray(b"ab"), "cd", b"ef"])
562 with self.assertRaises(TypeError):
563 dot_join([memoryview(b"ab"), "cd", b"ef"])
Christian Heimes510711d2008-01-30 11:57:58 +0000564
Christian Heimes510711d2008-01-30 11:57:58 +0000565 def test_count(self):
566 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200567 i = 105
568 p = 112
569 w = 119
570
Christian Heimes510711d2008-01-30 11:57:58 +0000571 self.assertEqual(b.count(b'i'), 4)
572 self.assertEqual(b.count(b'ss'), 2)
573 self.assertEqual(b.count(b'w'), 0)
574
Antoine Pitrouac65d962011-10-20 23:54:17 +0200575 self.assertEqual(b.count(i), 4)
576 self.assertEqual(b.count(w), 0)
577
578 self.assertEqual(b.count(b'i', 6), 2)
579 self.assertEqual(b.count(b'p', 6), 2)
580 self.assertEqual(b.count(b'i', 1, 3), 1)
581 self.assertEqual(b.count(b'p', 7, 9), 1)
582
583 self.assertEqual(b.count(i, 6), 2)
584 self.assertEqual(b.count(p, 6), 2)
585 self.assertEqual(b.count(i, 1, 3), 1)
586 self.assertEqual(b.count(p, 7, 9), 1)
587
Christian Heimes510711d2008-01-30 11:57:58 +0000588 def test_startswith(self):
589 b = self.type2test(b'hello')
590 self.assertFalse(self.type2test().startswith(b"anything"))
591 self.assertTrue(b.startswith(b"hello"))
592 self.assertTrue(b.startswith(b"hel"))
593 self.assertTrue(b.startswith(b"h"))
594 self.assertFalse(b.startswith(b"hellow"))
595 self.assertFalse(b.startswith(b"ha"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300596 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300597 b.startswith([b'h'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300598 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300599 self.assertIn('bytes', exc)
600 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000601
602 def test_endswith(self):
603 b = self.type2test(b'hello')
604 self.assertFalse(bytearray().endswith(b"anything"))
605 self.assertTrue(b.endswith(b"hello"))
606 self.assertTrue(b.endswith(b"llo"))
607 self.assertTrue(b.endswith(b"o"))
608 self.assertFalse(b.endswith(b"whello"))
609 self.assertFalse(b.endswith(b"no"))
Ezio Melottif2b3f782011-04-26 06:40:59 +0300610 with self.assertRaises(TypeError) as cm:
Ezio Melottiba42fd52011-04-26 06:09:45 +0300611 b.endswith([b'o'])
Ezio Melottif2b3f782011-04-26 06:40:59 +0300612 exc = str(cm.exception)
Ezio Melottiba42fd52011-04-26 06:09:45 +0300613 self.assertIn('bytes', exc)
614 self.assertIn('tuple', exc)
Christian Heimes510711d2008-01-30 11:57:58 +0000615
616 def test_find(self):
617 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200618 i = 105
619 w = 119
620
Christian Heimes510711d2008-01-30 11:57:58 +0000621 self.assertEqual(b.find(b'ss'), 2)
Christian Heimes510711d2008-01-30 11:57:58 +0000622 self.assertEqual(b.find(b'w'), -1)
623 self.assertEqual(b.find(b'mississippian'), -1)
624
Antoine Pitrouac65d962011-10-20 23:54:17 +0200625 self.assertEqual(b.find(i), 1)
626 self.assertEqual(b.find(w), -1)
627
628 self.assertEqual(b.find(b'ss', 3), 5)
629 self.assertEqual(b.find(b'ss', 1, 7), 2)
630 self.assertEqual(b.find(b'ss', 1, 3), -1)
631
632 self.assertEqual(b.find(i, 6), 7)
633 self.assertEqual(b.find(i, 1, 3), 1)
634 self.assertEqual(b.find(w, 1, 3), -1)
635
Victor Stinnerf8eac002011-12-18 01:17:41 +0100636 for index in (-1, 256, sys.maxsize + 1):
637 self.assertRaisesRegex(
638 ValueError, r'byte must be in range\(0, 256\)',
639 b.find, index)
640
Christian Heimes510711d2008-01-30 11:57:58 +0000641 def test_rfind(self):
642 b = self.type2test(b'mississippi')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200643 i = 105
644 w = 119
645
Christian Heimes510711d2008-01-30 11:57:58 +0000646 self.assertEqual(b.rfind(b'ss'), 5)
Christian Heimes510711d2008-01-30 11:57:58 +0000647 self.assertEqual(b.rfind(b'w'), -1)
648 self.assertEqual(b.rfind(b'mississippian'), -1)
649
Antoine Pitrouac65d962011-10-20 23:54:17 +0200650 self.assertEqual(b.rfind(i), 10)
651 self.assertEqual(b.rfind(w), -1)
652
653 self.assertEqual(b.rfind(b'ss', 3), 5)
654 self.assertEqual(b.rfind(b'ss', 0, 6), 2)
655
656 self.assertEqual(b.rfind(i, 1, 3), 1)
657 self.assertEqual(b.rfind(i, 3, 9), 7)
658 self.assertEqual(b.rfind(w, 1, 3), -1)
659
Christian Heimes510711d2008-01-30 11:57:58 +0000660 def test_index(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200661 b = self.type2test(b'mississippi')
662 i = 105
663 w = 119
664
665 self.assertEqual(b.index(b'ss'), 2)
666 self.assertRaises(ValueError, b.index, b'w')
667 self.assertRaises(ValueError, b.index, b'mississippian')
668
669 self.assertEqual(b.index(i), 1)
670 self.assertRaises(ValueError, b.index, w)
671
672 self.assertEqual(b.index(b'ss', 3), 5)
673 self.assertEqual(b.index(b'ss', 1, 7), 2)
674 self.assertRaises(ValueError, b.index, b'ss', 1, 3)
675
676 self.assertEqual(b.index(i, 6), 7)
677 self.assertEqual(b.index(i, 1, 3), 1)
678 self.assertRaises(ValueError, b.index, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000679
680 def test_rindex(self):
Antoine Pitrouac65d962011-10-20 23:54:17 +0200681 b = self.type2test(b'mississippi')
682 i = 105
683 w = 119
684
685 self.assertEqual(b.rindex(b'ss'), 5)
686 self.assertRaises(ValueError, b.rindex, b'w')
687 self.assertRaises(ValueError, b.rindex, b'mississippian')
688
689 self.assertEqual(b.rindex(i), 10)
690 self.assertRaises(ValueError, b.rindex, w)
691
692 self.assertEqual(b.rindex(b'ss', 3), 5)
693 self.assertEqual(b.rindex(b'ss', 0, 6), 2)
694
695 self.assertEqual(b.rindex(i, 1, 3), 1)
696 self.assertEqual(b.rindex(i, 3, 9), 7)
697 self.assertRaises(ValueError, b.rindex, w, 1, 3)
Christian Heimes510711d2008-01-30 11:57:58 +0000698
Ethan Furmanb95b5612015-01-23 20:05:18 -0800699 def test_mod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300700 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800701 orig = b
702 b = b % b'world'
703 self.assertEqual(b, b'hello, world!')
704 self.assertEqual(orig, b'hello, %b!')
705 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300706 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800707 a = b % (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200708 self.assertEqual(a, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200709 self.assertIs(type(a), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800710 # issue 29714
711 b = self.type2test(b'hello,\x00%b!')
712 b = b % b'world'
713 self.assertEqual(b, b'hello,\x00world!')
714 self.assertIs(type(b), self.type2test)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800715
716 def test_imod(self):
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300717 b = self.type2test(b'hello, %b!')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800718 orig = b
719 b %= b'world'
720 self.assertEqual(b, b'hello, world!')
721 self.assertEqual(orig, b'hello, %b!')
722 self.assertFalse(b is orig)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300723 b = self.type2test(b'%s / 100 = %d%%')
Ethan Furmanb95b5612015-01-23 20:05:18 -0800724 b %= (b'seventy-nine', 79)
Serhiy Storchaka4e1942b2015-01-26 09:28:11 +0200725 self.assertEqual(b, b'seventy-nine / 100 = 79%')
Victor Stinnere914d412016-04-15 17:52:27 +0200726 self.assertIs(type(b), self.type2test)
Xiang Zhangb76ad512017-03-06 17:17:05 +0800727 # issue 29714
728 b = self.type2test(b'hello,\x00%b!')
729 b %= b'world'
730 self.assertEqual(b, b'hello,\x00world!')
731 self.assertIs(type(b), self.type2test)
Serhiy Storchakac9a59e62016-04-15 14:11:10 +0300732
733 def test_rmod(self):
734 with self.assertRaises(TypeError):
735 object() % self.type2test(b'abc')
736 self.assertIs(self.type2test(b'abc').__rmod__('%r'), NotImplemented)
Ethan Furmanb95b5612015-01-23 20:05:18 -0800737
Christian Heimes510711d2008-01-30 11:57:58 +0000738 def test_replace(self):
739 b = self.type2test(b'mississippi')
740 self.assertEqual(b.replace(b'i', b'a'), b'massassappa')
741 self.assertEqual(b.replace(b'ss', b'x'), b'mixixippi')
742
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300743 def test_replace_int_error(self):
744 self.assertRaises(TypeError, self.type2test(b'a b').replace, 32, b'')
745
Christian Heimes510711d2008-01-30 11:57:58 +0000746 def test_split_string_error(self):
747 self.assertRaises(TypeError, self.type2test(b'a b').split, ' ')
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300748 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, ' ')
749
750 def test_split_int_error(self):
751 self.assertRaises(TypeError, self.type2test(b'a b').split, 32)
752 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000753
Antoine Pitrouf8453022010-01-12 22:05:42 +0000754 def test_split_unicodewhitespace(self):
Martin Panter0d0db6c2016-04-10 08:45:26 +0000755 for b in (b'a\x1Cb', b'a\x1Db', b'a\x1Eb', b'a\x1Fb'):
756 b = self.type2test(b)
757 self.assertEqual(b.split(), [b])
Antoine Pitrouf8453022010-01-12 22:05:42 +0000758 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
759 self.assertEqual(b.split(), [b'\x1c\x1d\x1e\x1f'])
760
Christian Heimes510711d2008-01-30 11:57:58 +0000761 def test_rsplit_unicodewhitespace(self):
762 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
Christian Heimes510711d2008-01-30 11:57:58 +0000763 self.assertEqual(b.rsplit(), [b'\x1c\x1d\x1e\x1f'])
764
765 def test_partition(self):
766 b = self.type2test(b'mississippi')
767 self.assertEqual(b.partition(b'ss'), (b'mi', b'ss', b'issippi'))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000768 self.assertEqual(b.partition(b'w'), (b'mississippi', b'', b''))
Christian Heimes510711d2008-01-30 11:57:58 +0000769
770 def test_rpartition(self):
771 b = self.type2test(b'mississippi')
772 self.assertEqual(b.rpartition(b'ss'), (b'missi', b'ss', b'ippi'))
773 self.assertEqual(b.rpartition(b'i'), (b'mississipp', b'i', b''))
Antoine Pitrouf8453022010-01-12 22:05:42 +0000774 self.assertEqual(b.rpartition(b'w'), (b'', b'', b'mississippi'))
Christian Heimes510711d2008-01-30 11:57:58 +0000775
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300776 def test_partition_string_error(self):
777 self.assertRaises(TypeError, self.type2test(b'a b').partition, ' ')
778 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, ' ')
779
780 def test_partition_int_error(self):
781 self.assertRaises(TypeError, self.type2test(b'a b').partition, 32)
782 self.assertRaises(TypeError, self.type2test(b'a b').rpartition, 32)
783
Christian Heimes510711d2008-01-30 11:57:58 +0000784 def test_pickling(self):
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +0000785 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Christian Heimes510711d2008-01-30 11:57:58 +0000786 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
787 b = self.type2test(b)
788 ps = pickle.dumps(b, proto)
789 q = pickle.loads(ps)
790 self.assertEqual(b, q)
791
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000792 def test_iterator_pickling(self):
Serhiy Storchakabad12572014-12-15 14:03:42 +0200793 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
794 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
795 it = itorg = iter(self.type2test(b))
796 data = list(self.type2test(b))
797 d = pickle.dumps(it, proto)
798 it = pickle.loads(d)
799 self.assertEqual(type(itorg), type(it))
800 self.assertEqual(list(it), data)
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000801
Serhiy Storchakabad12572014-12-15 14:03:42 +0200802 it = pickle.loads(d)
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200803 if not b:
Serhiy Storchakabad12572014-12-15 14:03:42 +0200804 continue
Serhiy Storchakaaabafe72016-03-06 14:10:24 +0200805 next(it)
Serhiy Storchakabad12572014-12-15 14:03:42 +0200806 d = pickle.dumps(it, proto)
807 it = pickle.loads(d)
808 self.assertEqual(list(it), data[1:])
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +0000809
Christian Heimes510711d2008-01-30 11:57:58 +0000810 def test_strip_bytearray(self):
811 self.assertEqual(self.type2test(b'abc').strip(memoryview(b'ac')), b'b')
812 self.assertEqual(self.type2test(b'abc').lstrip(memoryview(b'ac')), b'bc')
813 self.assertEqual(self.type2test(b'abc').rstrip(memoryview(b'ac')), b'ab')
814
815 def test_strip_string_error(self):
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300816 self.assertRaises(TypeError, self.type2test(b'abc').strip, 'ac')
817 self.assertRaises(TypeError, self.type2test(b'abc').lstrip, 'ac')
818 self.assertRaises(TypeError, self.type2test(b'abc').rstrip, 'ac')
819
820 def test_strip_int_error(self):
821 self.assertRaises(TypeError, self.type2test(b' abc ').strip, 32)
822 self.assertRaises(TypeError, self.type2test(b' abc ').lstrip, 32)
823 self.assertRaises(TypeError, self.type2test(b' abc ').rstrip, 32)
Christian Heimes510711d2008-01-30 11:57:58 +0000824
Eli Bendersky906b88f2011-07-29 07:05:08 +0300825 def test_center(self):
826 # Fill character can be either bytes or bytearray (issue 12380)
827 b = self.type2test(b'abc')
828 for fill_type in (bytes, bytearray):
829 self.assertEqual(b.center(7, fill_type(b'-')),
830 self.type2test(b'--abc--'))
831
832 def test_ljust(self):
833 # Fill character can be either bytes or bytearray (issue 12380)
834 b = self.type2test(b'abc')
835 for fill_type in (bytes, bytearray):
836 self.assertEqual(b.ljust(7, fill_type(b'-')),
837 self.type2test(b'abc----'))
838
839 def test_rjust(self):
840 # Fill character can be either bytes or bytearray (issue 12380)
841 b = self.type2test(b'abc')
842 for fill_type in (bytes, bytearray):
843 self.assertEqual(b.rjust(7, fill_type(b'-')),
844 self.type2test(b'----abc'))
845
Serhiy Storchakaa2314282017-10-29 02:11:54 +0300846 def test_xjust_int_error(self):
847 self.assertRaises(TypeError, self.type2test(b'abc').center, 7, 32)
848 self.assertRaises(TypeError, self.type2test(b'abc').ljust, 7, 32)
849 self.assertRaises(TypeError, self.type2test(b'abc').rjust, 7, 32)
850
Christian Heimes510711d2008-01-30 11:57:58 +0000851 def test_ord(self):
852 b = self.type2test(b'\0A\x7f\x80\xff')
853 self.assertEqual([ord(b[i:i+1]) for i in range(len(b))],
854 [0, 65, 127, 128, 255])
855
Georg Brandlabc38772009-04-12 15:51:51 +0000856 def test_maketrans(self):
857 transtable = b'\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037 !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`xyzdefghijklmnopqrstuvwxyz{|}~\177\200\201\202\203\204\205\206\207\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374\375\376\377'
Georg Brandlabc38772009-04-12 15:51:51 +0000858 self.assertEqual(self.type2test.maketrans(b'abc', b'xyz'), transtable)
Antoine Pitrou403ce782009-10-14 17:14:16 +0000859 transtable = b'\000\001\002\003\004\005\006\007\010\011\012\013\014\015\016\017\020\021\022\023\024\025\026\027\030\031\032\033\034\035\036\037 !"#$%&\'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~\177\200\201\202\203\204\205\206\207\210\211\212\213\214\215\216\217\220\221\222\223\224\225\226\227\230\231\232\233\234\235\236\237\240\241\242\243\244\245\246\247\250\251\252\253\254\255\256\257\260\261\262\263\264\265\266\267\270\271\272\273\274\275\276\277\300\301\302\303\304\305\306\307\310\311\312\313\314\315\316\317\320\321\322\323\324\325\326\327\330\331\332\333\334\335\336\337\340\341\342\343\344\345\346\347\350\351\352\353\354\355\356\357\360\361\362\363\364\365\366\367\370\371\372\373\374xyz'
860 self.assertEqual(self.type2test.maketrans(b'\375\376\377', b'xyz'), transtable)
Georg Brandlabc38772009-04-12 15:51:51 +0000861 self.assertRaises(ValueError, self.type2test.maketrans, b'abc', b'xyzq')
862 self.assertRaises(TypeError, self.type2test.maketrans, 'abc', 'def')
863
Jesus Ceaac451502011-04-20 17:09:23 +0200864 def test_none_arguments(self):
865 # issue 11828
866 b = self.type2test(b'hello')
867 l = self.type2test(b'l')
868 h = self.type2test(b'h')
869 x = self.type2test(b'x')
870 o = self.type2test(b'o')
871
872 self.assertEqual(2, b.find(l, None))
873 self.assertEqual(3, b.find(l, -2, None))
874 self.assertEqual(2, b.find(l, None, -2))
875 self.assertEqual(0, b.find(h, None, None))
876
877 self.assertEqual(3, b.rfind(l, None))
878 self.assertEqual(3, b.rfind(l, -2, None))
879 self.assertEqual(2, b.rfind(l, None, -2))
880 self.assertEqual(0, b.rfind(h, None, None))
881
882 self.assertEqual(2, b.index(l, None))
883 self.assertEqual(3, b.index(l, -2, None))
884 self.assertEqual(2, b.index(l, None, -2))
885 self.assertEqual(0, b.index(h, None, None))
886
887 self.assertEqual(3, b.rindex(l, None))
888 self.assertEqual(3, b.rindex(l, -2, None))
889 self.assertEqual(2, b.rindex(l, None, -2))
890 self.assertEqual(0, b.rindex(h, None, None))
891
892 self.assertEqual(2, b.count(l, None))
893 self.assertEqual(1, b.count(l, -2, None))
894 self.assertEqual(1, b.count(l, None, -2))
895 self.assertEqual(0, b.count(x, None, None))
896
897 self.assertEqual(True, b.endswith(o, None))
898 self.assertEqual(True, b.endswith(o, -2, None))
899 self.assertEqual(True, b.endswith(l, None, -2))
900 self.assertEqual(False, b.endswith(x, None, None))
901
902 self.assertEqual(True, b.startswith(h, None))
903 self.assertEqual(True, b.startswith(l, -2, None))
904 self.assertEqual(True, b.startswith(h, None, -2))
905 self.assertEqual(False, b.startswith(x, None, None))
906
Antoine Pitrouac65d962011-10-20 23:54:17 +0200907 def test_integer_arguments_out_of_byte_range(self):
908 b = self.type2test(b'hello')
909
910 for method in (b.count, b.find, b.index, b.rfind, b.rindex):
911 self.assertRaises(ValueError, method, -1)
912 self.assertRaises(ValueError, method, 256)
913 self.assertRaises(ValueError, method, 9999)
914
Jesus Ceaac451502011-04-20 17:09:23 +0200915 def test_find_etc_raise_correct_error_messages(self):
916 # issue 11828
917 b = self.type2test(b'hello')
918 x = self.type2test(b'x')
Ezio Melottiaf928422011-04-20 21:56:21 +0300919 self.assertRaisesRegex(TypeError, r'\bfind\b', b.find,
Jesus Ceaac451502011-04-20 17:09:23 +0200920 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300921 self.assertRaisesRegex(TypeError, r'\brfind\b', b.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +0200922 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300923 self.assertRaisesRegex(TypeError, r'\bindex\b', b.index,
Jesus Ceaac451502011-04-20 17:09:23 +0200924 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300925 self.assertRaisesRegex(TypeError, r'\brindex\b', b.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +0200926 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300927 self.assertRaisesRegex(TypeError, r'\bcount\b', b.count,
Jesus Ceaac451502011-04-20 17:09:23 +0200928 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300929 self.assertRaisesRegex(TypeError, r'\bstartswith\b', b.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200930 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +0300931 self.assertRaisesRegex(TypeError, r'\bendswith\b', b.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +0200932 x, None, None, None)
933
Serhiy Storchakafbb1c5e2016-03-30 20:40:02 +0300934 def test_free_after_iterating(self):
935 test.support.check_free_after_iterating(self, iter, self.type2test)
936 test.support.check_free_after_iterating(self, reversed, self.type2test)
937
Martin Panter1b6c6da2016-08-27 08:35:02 +0000938 def test_translate(self):
939 b = self.type2test(b'hello')
940 rosetta = bytearray(range(256))
941 rosetta[ord('o')] = ord('e')
942
943 self.assertRaises(TypeError, b.translate)
944 self.assertRaises(TypeError, b.translate, None, None)
945 self.assertRaises(ValueError, b.translate, bytes(range(255)))
946
947 c = b.translate(rosetta, b'hello')
948 self.assertEqual(b, b'hello')
949 self.assertIsInstance(c, self.type2test)
950
951 c = b.translate(rosetta)
952 d = b.translate(rosetta, b'')
953 self.assertEqual(c, d)
954 self.assertEqual(c, b'helle')
955
956 c = b.translate(rosetta, b'l')
957 self.assertEqual(c, b'hee')
958 c = b.translate(None, b'e')
959 self.assertEqual(c, b'hllo')
960
961 # test delete as a keyword argument
962 c = b.translate(rosetta, delete=b'')
963 self.assertEqual(c, b'helle')
964 c = b.translate(rosetta, delete=b'l')
965 self.assertEqual(c, b'hee')
966 c = b.translate(None, delete=b'e')
967 self.assertEqual(c, b'hllo')
968
Sergey Fedoseev92709a22019-09-09 21:28:34 +0500969 def test_sq_item(self):
970 _testcapi = test.support.import_module('_testcapi')
971 obj = self.type2test((42,))
972 with self.assertRaises(IndexError):
973 _testcapi.sequence_getitem(obj, -2)
974 with self.assertRaises(IndexError):
975 _testcapi.sequence_getitem(obj, 1)
976 self.assertEqual(_testcapi.sequence_getitem(obj, 0), 42)
977
Christian Heimes510711d2008-01-30 11:57:58 +0000978
Ezio Melotti0dceb562013-01-10 07:43:26 +0200979class BytesTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +0000980 type2test = bytes
981
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400982 def test_getitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200983 b = b'python'
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400984 msg = "byte indices must be integers or slices"
985 with self.assertRaisesRegex(TypeError, msg):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +0200986 b['a']
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400987
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000988 def test_buffer_is_readonly(self):
Victor Stinner76119642014-07-28 22:07:07 +0200989 fd = os.open(__file__, os.O_RDONLY)
Antoine Pitrou47d305d2008-08-16 23:28:44 +0000990 with open(fd, "rb", buffering=0) as f:
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000991 self.assertRaises(TypeError, f.readinto, b"")
992
Benjamin Petersonc15a0732008-08-26 16:46:47 +0000993 def test_custom(self):
994 class A:
995 def __bytes__(self):
996 return b'abc'
997 self.assertEqual(bytes(A()), b'abc')
998 class A: pass
999 self.assertRaises(TypeError, bytes, A())
1000 class A:
1001 def __bytes__(self):
1002 return None
1003 self.assertRaises(TypeError, bytes, A())
Benjamin Peterson5ff3f732012-12-19 15:27:41 -06001004 class A:
1005 def __bytes__(self):
1006 return b'a'
1007 def __index__(self):
1008 return 42
1009 self.assertEqual(bytes(A()), b'a')
Serhiy Storchaka5aac3ed2015-12-20 16:36:34 +02001010 # Issue #25766
1011 class A(str):
1012 def __bytes__(self):
1013 return b'abc'
1014 self.assertEqual(bytes(A('\u20ac')), b'abc')
1015 self.assertEqual(bytes(A('\u20ac'), 'iso8859-15'), b'\xa4')
Serhiy Storchaka15095802015-11-25 15:47:01 +02001016 # Issue #24731
1017 class A:
1018 def __bytes__(self):
1019 return OtherBytesSubclass(b'abc')
1020 self.assertEqual(bytes(A()), b'abc')
1021 self.assertIs(type(bytes(A())), OtherBytesSubclass)
1022 self.assertEqual(BytesSubclass(A()), b'abc')
1023 self.assertIs(type(BytesSubclass(A())), BytesSubclass)
Benjamin Petersonc15a0732008-08-26 16:46:47 +00001024
Victor Stinner29e762c2011-01-05 03:33:28 +00001025 # Test PyBytes_FromFormat()
1026 def test_from_format(self):
Victor Stinner7ab986d2015-10-14 02:55:12 +02001027 ctypes = test.support.import_module('ctypes')
Victor Stinner03dab782015-10-14 00:21:35 +02001028 _testcapi = test.support.import_module('_testcapi')
1029 from ctypes import pythonapi, py_object
1030 from ctypes import (
1031 c_int, c_uint,
1032 c_long, c_ulong,
1033 c_size_t, c_ssize_t,
1034 c_char_p)
1035
Victor Stinner29e762c2011-01-05 03:33:28 +00001036 PyBytes_FromFormat = pythonapi.PyBytes_FromFormat
1037 PyBytes_FromFormat.restype = py_object
1038
Victor Stinner03dab782015-10-14 00:21:35 +02001039 # basic tests
Victor Stinner29e762c2011-01-05 03:33:28 +00001040 self.assertEqual(PyBytes_FromFormat(b'format'),
1041 b'format')
Victor Stinner03dab782015-10-14 00:21:35 +02001042 self.assertEqual(PyBytes_FromFormat(b'Hello %s !', b'world'),
1043 b'Hello world !')
Victor Stinner29e762c2011-01-05 03:33:28 +00001044
Victor Stinner03dab782015-10-14 00:21:35 +02001045 # test formatters
1046 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(0)),
1047 b'c=\0')
1048 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(ord('@'))),
1049 b'c=@')
1050 self.assertEqual(PyBytes_FromFormat(b'c=%c', c_int(255)),
1051 b'c=\xff')
1052 self.assertEqual(PyBytes_FromFormat(b'd=%d ld=%ld zd=%zd',
1053 c_int(1), c_long(2),
1054 c_size_t(3)),
1055 b'd=1 ld=2 zd=3')
1056 self.assertEqual(PyBytes_FromFormat(b'd=%d ld=%ld zd=%zd',
1057 c_int(-1), c_long(-2),
1058 c_size_t(-3)),
1059 b'd=-1 ld=-2 zd=-3')
1060 self.assertEqual(PyBytes_FromFormat(b'u=%u lu=%lu zu=%zu',
1061 c_uint(123), c_ulong(456),
1062 c_size_t(789)),
1063 b'u=123 lu=456 zu=789')
1064 self.assertEqual(PyBytes_FromFormat(b'i=%i', c_int(123)),
1065 b'i=123')
1066 self.assertEqual(PyBytes_FromFormat(b'i=%i', c_int(-123)),
1067 b'i=-123')
1068 self.assertEqual(PyBytes_FromFormat(b'x=%x', c_int(0xabc)),
1069 b'x=abc')
Victor Stinner83ff8a62015-10-14 15:28:59 +02001070
1071 sizeof_ptr = ctypes.sizeof(c_char_p)
1072
1073 if os.name == 'nt':
1074 # Windows (MSCRT)
1075 ptr_format = '0x%0{}X'.format(2 * sizeof_ptr)
1076 def ptr_formatter(ptr):
1077 return (ptr_format % ptr)
1078 else:
1079 # UNIX (glibc)
1080 def ptr_formatter(ptr):
1081 return '%#x' % ptr
1082
Victor Stinner7ab986d2015-10-14 02:55:12 +02001083 ptr = 0xabcdef
Victor Stinner83ff8a62015-10-14 15:28:59 +02001084 self.assertEqual(PyBytes_FromFormat(b'ptr=%p', c_char_p(ptr)),
1085 ('ptr=' + ptr_formatter(ptr)).encode('ascii'))
Victor Stinner03dab782015-10-14 00:21:35 +02001086 self.assertEqual(PyBytes_FromFormat(b's=%s', c_char_p(b'cstr')),
1087 b's=cstr')
1088
1089 # test minimum and maximum integer values
1090 size_max = c_size_t(-1).value
1091 for formatstr, ctypes_type, value, py_formatter in (
1092 (b'%d', c_int, _testcapi.INT_MIN, str),
1093 (b'%d', c_int, _testcapi.INT_MAX, str),
1094 (b'%ld', c_long, _testcapi.LONG_MIN, str),
1095 (b'%ld', c_long, _testcapi.LONG_MAX, str),
1096 (b'%lu', c_ulong, _testcapi.ULONG_MAX, str),
1097 (b'%zd', c_ssize_t, _testcapi.PY_SSIZE_T_MIN, str),
1098 (b'%zd', c_ssize_t, _testcapi.PY_SSIZE_T_MAX, str),
1099 (b'%zu', c_size_t, size_max, str),
Victor Stinner83ff8a62015-10-14 15:28:59 +02001100 (b'%p', c_char_p, size_max, ptr_formatter),
Victor Stinner03dab782015-10-14 00:21:35 +02001101 ):
1102 self.assertEqual(PyBytes_FromFormat(formatstr, ctypes_type(value)),
1103 py_formatter(value).encode('ascii')),
1104
1105 # width and precision (width is currently ignored)
1106 self.assertEqual(PyBytes_FromFormat(b'%5s', b'a'),
1107 b'a')
1108 self.assertEqual(PyBytes_FromFormat(b'%.3s', b'abcdef'),
1109 b'abc')
1110
1111 # '%%' formatter
1112 self.assertEqual(PyBytes_FromFormat(b'%%'),
1113 b'%')
1114 self.assertEqual(PyBytes_FromFormat(b'[%%]'),
1115 b'[%]')
1116 self.assertEqual(PyBytes_FromFormat(b'%%%c', c_int(ord('_'))),
1117 b'%_')
1118 self.assertEqual(PyBytes_FromFormat(b'%%s'),
1119 b'%s')
1120
1121 # Invalid formats and partial formatting
Victor Stinner29e762c2011-01-05 03:33:28 +00001122 self.assertEqual(PyBytes_FromFormat(b'%'), b'%')
Victor Stinner03dab782015-10-14 00:21:35 +02001123 self.assertEqual(PyBytes_FromFormat(b'x=%i y=%', c_int(2), c_int(3)),
1124 b'x=2 y=%')
Victor Stinner29e762c2011-01-05 03:33:28 +00001125
Victor Stinner03dab782015-10-14 00:21:35 +02001126 # Issue #19969: %c must raise OverflowError for values
1127 # not in the range [0; 255]
Victor Stinnerc9362cf2013-12-13 12:14:44 +01001128 self.assertRaises(OverflowError,
1129 PyBytes_FromFormat, b'%c', c_int(-1))
1130 self.assertRaises(OverflowError,
1131 PyBytes_FromFormat, b'%c', c_int(256))
1132
Serhiy Storchaka44cc4822019-01-12 09:22:29 +02001133 # Issue #33817: empty strings
1134 self.assertEqual(PyBytes_FromFormat(b''),
1135 b'')
1136 self.assertEqual(PyBytes_FromFormat(b'%s', b''),
1137 b'')
1138
Guido van Rossum97c1adf2016-08-18 09:22:23 -07001139 def test_bytes_blocking(self):
1140 class IterationBlocked(list):
1141 __bytes__ = None
1142 i = [0, 1, 2, 3]
1143 self.assertEqual(bytes(i), b'\x00\x01\x02\x03')
1144 self.assertRaises(TypeError, bytes, IterationBlocked(i))
1145
1146 # At least in CPython, because bytes.__new__ and the C API
1147 # PyBytes_FromObject have different fallback rules, integer
1148 # fallback is handled specially, so test separately.
1149 class IntBlocked(int):
1150 __bytes__ = None
1151 self.assertEqual(bytes(3), b'\0\0\0')
1152 self.assertRaises(TypeError, bytes, IntBlocked(3))
1153
1154 # While there is no separately-defined rule for handling bytes
1155 # subclasses differently from other buffer-interface classes,
1156 # an implementation may well special-case them (as CPython 2.x
1157 # str did), so test them separately.
1158 class BytesSubclassBlocked(bytes):
1159 __bytes__ = None
1160 self.assertEqual(bytes(b'ab'), b'ab')
1161 self.assertRaises(TypeError, bytes, BytesSubclassBlocked(b'ab'))
1162
1163 class BufferBlocked(bytearray):
1164 __bytes__ = None
1165 ba, bb = bytearray(b'ab'), BufferBlocked(b'ab')
1166 self.assertEqual(bytes(ba), b'ab')
1167 self.assertRaises(TypeError, bytes, bb)
1168
Antoine Pitrou2f89aa62008-08-02 21:02:48 +00001169
Ezio Melotti0dceb562013-01-10 07:43:26 +02001170class ByteArrayTest(BaseBytesTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001171 type2test = bytearray
Guido van Rossumebea9be2007-04-09 00:49:13 +00001172
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001173 def test_getitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001174 b = bytearray(b'python')
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001175 msg = "bytearray indices must be integers or slices"
1176 with self.assertRaisesRegex(TypeError, msg):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001177 b['a']
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001178
1179 def test_setitem_error(self):
Serhiy Storchaka8e79e6e2019-02-19 13:49:09 +02001180 b = bytearray(b'python')
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001181 msg = "bytearray indices must be integers or slices"
1182 with self.assertRaisesRegex(TypeError, msg):
Terry Jan Reedyffff1442014-08-02 01:30:37 -04001183 b['a'] = "python"
1184
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001185 def test_nohash(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001186 self.assertRaises(TypeError, hash, bytearray())
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001187
Guido van Rossum254348e2007-11-21 19:29:53 +00001188 def test_bytearray_api(self):
Guido van Rossum75d6f1a2007-05-24 17:52:00 +00001189 short_sample = b"Hello world\n"
Guido van Rossum98297ee2007-11-06 21:34:58 +00001190 sample = short_sample + b"\0"*(20 - len(short_sample))
Guido van Rossumd624f182006-04-24 13:47:05 +00001191 tfn = tempfile.mktemp()
1192 try:
1193 # Prepare
1194 with open(tfn, "wb") as f:
1195 f.write(short_sample)
1196 # Test readinto
1197 with open(tfn, "rb") as f:
Guido van Rossum254348e2007-11-21 19:29:53 +00001198 b = bytearray(20)
Guido van Rossumd624f182006-04-24 13:47:05 +00001199 n = f.readinto(b)
1200 self.assertEqual(n, len(short_sample))
Guido van Rossumc1f779c2007-07-03 08:25:58 +00001201 self.assertEqual(list(b), list(sample))
Guido van Rossumd624f182006-04-24 13:47:05 +00001202 # Test writing in binary mode
1203 with open(tfn, "wb") as f:
1204 f.write(b)
1205 with open(tfn, "rb") as f:
1206 self.assertEqual(f.read(), sample)
Guido van Rossum13633bb2007-04-13 18:42:35 +00001207 # Text mode is ambiguous; don't test
Guido van Rossumd624f182006-04-24 13:47:05 +00001208 finally:
1209 try:
1210 os.remove(tfn)
Andrew Svetlov8b33dd82012-12-24 19:58:48 +02001211 except OSError:
Guido van Rossumd624f182006-04-24 13:47:05 +00001212 pass
1213
Neal Norwitz6968b052007-02-27 19:02:19 +00001214 def test_reverse(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001215 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +00001216 self.assertEqual(b.reverse(), None)
1217 self.assertEqual(b, b'olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +00001218 b = bytearray(b'hello1') # test even number of items
Neal Norwitz6968b052007-02-27 19:02:19 +00001219 b.reverse()
1220 self.assertEqual(b, b'1olleh')
Guido van Rossum254348e2007-11-21 19:29:53 +00001221 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +00001222 b.reverse()
1223 self.assertFalse(b)
1224
Eli Bendersky4db28d32011-03-03 18:21:02 +00001225 def test_clear(self):
1226 b = bytearray(b'python')
1227 b.clear()
1228 self.assertEqual(b, b'')
1229
1230 b = bytearray(b'')
1231 b.clear()
1232 self.assertEqual(b, b'')
1233
1234 b = bytearray(b'')
1235 b.append(ord('r'))
1236 b.clear()
1237 b.append(ord('p'))
1238 self.assertEqual(b, b'p')
1239
1240 def test_copy(self):
1241 b = bytearray(b'abc')
1242 bb = b.copy()
1243 self.assertEqual(bb, b'abc')
1244
1245 b = bytearray(b'')
1246 bb = b.copy()
1247 self.assertEqual(bb, b'')
1248
1249 # test that it's indeed a copy and not a reference
1250 b = bytearray(b'abc')
1251 bb = b.copy()
1252 self.assertEqual(b, bb)
1253 self.assertIsNot(b, bb)
1254 bb.append(ord('d'))
1255 self.assertEqual(bb, b'abcd')
1256 self.assertEqual(b, b'abc')
1257
Guido van Rossumd624f182006-04-24 13:47:05 +00001258 def test_regexps(self):
1259 def by(s):
Guido van Rossum254348e2007-11-21 19:29:53 +00001260 return bytearray(map(ord, s))
Guido van Rossumd624f182006-04-24 13:47:05 +00001261 b = by("Hello, world")
Antoine Pitroufd036452008-08-19 17:56:33 +00001262 self.assertEqual(re.findall(br"\w+", b), [by("Hello"), by("world")])
Guido van Rossumd624f182006-04-24 13:47:05 +00001263
1264 def test_setitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001265 b = bytearray([1, 2, 3])
Guido van Rossumd624f182006-04-24 13:47:05 +00001266 b[1] = 100
Guido van Rossum254348e2007-11-21 19:29:53 +00001267 self.assertEqual(b, bytearray([1, 100, 3]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001268 b[-1] = 200
Guido van Rossum254348e2007-11-21 19:29:53 +00001269 self.assertEqual(b, bytearray([1, 100, 200]))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001270 b[0] = Indexable(10)
Guido van Rossum254348e2007-11-21 19:29:53 +00001271 self.assertEqual(b, bytearray([10, 100, 200]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001272 try:
1273 b[3] = 0
1274 self.fail("Didn't raise IndexError")
1275 except IndexError:
1276 pass
1277 try:
1278 b[-10] = 0
1279 self.fail("Didn't raise IndexError")
1280 except IndexError:
1281 pass
1282 try:
1283 b[0] = 256
1284 self.fail("Didn't raise ValueError")
1285 except ValueError:
1286 pass
1287 try:
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001288 b[0] = Indexable(-1)
Guido van Rossumd624f182006-04-24 13:47:05 +00001289 self.fail("Didn't raise ValueError")
1290 except ValueError:
1291 pass
1292 try:
1293 b[0] = None
1294 self.fail("Didn't raise TypeError")
1295 except TypeError:
1296 pass
1297
1298 def test_delitem(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001299 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +00001300 del b[0]
Guido van Rossum254348e2007-11-21 19:29:53 +00001301 self.assertEqual(b, bytearray(range(1, 10)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001302 del b[-1]
Guido van Rossum254348e2007-11-21 19:29:53 +00001303 self.assertEqual(b, bytearray(range(1, 9)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001304 del b[4]
Guido van Rossum254348e2007-11-21 19:29:53 +00001305 self.assertEqual(b, bytearray([1, 2, 3, 4, 6, 7, 8]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001306
1307 def test_setslice(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001308 b = bytearray(range(10))
Guido van Rossumd624f182006-04-24 13:47:05 +00001309 self.assertEqual(list(b), list(range(10)))
1310
Guido van Rossum254348e2007-11-21 19:29:53 +00001311 b[0:5] = bytearray([1, 1, 1, 1, 1])
1312 self.assertEqual(b, bytearray([1, 1, 1, 1, 1, 5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001313
1314 del b[0:-5]
Guido van Rossum254348e2007-11-21 19:29:53 +00001315 self.assertEqual(b, bytearray([5, 6, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001316
Guido van Rossum254348e2007-11-21 19:29:53 +00001317 b[0:0] = bytearray([0, 1, 2, 3, 4])
1318 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumd624f182006-04-24 13:47:05 +00001319
Guido van Rossum254348e2007-11-21 19:29:53 +00001320 b[-7:-3] = bytearray([100, 101])
1321 self.assertEqual(b, bytearray([0, 1, 2, 100, 101, 7, 8, 9]))
Guido van Rossumd624f182006-04-24 13:47:05 +00001322
1323 b[3:5] = [3, 4, 5, 6]
Guido van Rossum254348e2007-11-21 19:29:53 +00001324 self.assertEqual(b, bytearray(range(10)))
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001325
Thomas Wouters9a6e62b2006-08-23 23:20:29 +00001326 b[3:0] = [42, 42, 42]
Guido van Rossum254348e2007-11-21 19:29:53 +00001327 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 +00001328
Ezio Melottic64bcbe2012-11-03 21:19:06 +02001329 b[3:] = b'foo'
1330 self.assertEqual(b, bytearray([0, 1, 2, 102, 111, 111]))
1331
1332 b[:3] = memoryview(b'foo')
1333 self.assertEqual(b, bytearray([102, 111, 111, 102, 111, 111]))
1334
1335 b[3:4] = []
1336 self.assertEqual(b, bytearray([102, 111, 111, 111, 111]))
1337
1338 for elem in [5, -5, 0, int(10e20), 'str', 2.3,
1339 ['a', 'b'], [b'a', b'b'], [[]]]:
1340 with self.assertRaises(TypeError):
1341 b[3:4] = elem
1342
1343 for elem in [[254, 255, 256], [-256, 9000]]:
1344 with self.assertRaises(ValueError):
1345 b[3:4] = elem
1346
Antoine Pitrou5df8a8a2013-10-05 21:12:18 +02001347 def test_setslice_extend(self):
1348 # Exercise the resizing logic (see issue #19087)
1349 b = bytearray(range(100))
1350 self.assertEqual(list(b), list(range(100)))
1351 del b[:10]
1352 self.assertEqual(list(b), list(range(10, 100)))
1353 b.extend(range(100, 110))
1354 self.assertEqual(list(b), list(range(10, 110)))
1355
Antoine Pitrou25454112015-05-19 20:52:27 +02001356 def test_fifo_overrun(self):
1357 # Test for issue #23985, a buffer overrun when implementing a FIFO
1358 # Build Python in pydebug mode for best results.
1359 b = bytearray(10)
1360 b.pop() # Defeat expanding buffer off-by-one quirk
1361 del b[:1] # Advance start pointer without reallocating
1362 b += bytes(2) # Append exactly the number of deleted bytes
1363 del b # Free memory buffer, allowing pydebug verification
1364
1365 def test_del_expand(self):
1366 # Reducing the size should not expand the buffer (issue #23985)
1367 b = bytearray(10)
1368 size = sys.getsizeof(b)
1369 del b[:1]
1370 self.assertLessEqual(sys.getsizeof(b), size)
1371
Thomas Wouters376446d2006-12-19 08:30:14 +00001372 def test_extended_set_del_slice(self):
Zackery Spytz14514d92019-05-17 01:13:03 -06001373 indices = (0, None, 1, 3, 19, 300, 1<<333, sys.maxsize,
1374 -1, -2, -31, -300)
Thomas Wouters376446d2006-12-19 08:30:14 +00001375 for start in indices:
1376 for stop in indices:
1377 # Skip invalid step 0
1378 for step in indices[1:]:
1379 L = list(range(255))
Guido van Rossum254348e2007-11-21 19:29:53 +00001380 b = bytearray(L)
Thomas Wouters376446d2006-12-19 08:30:14 +00001381 # Make sure we have a slice of exactly the right length,
1382 # but with different data.
1383 data = L[start:stop:step]
1384 data.reverse()
1385 L[start:stop:step] = data
1386 b[start:stop:step] = data
Ezio Melottib3aedd42010-11-20 19:04:17 +00001387 self.assertEqual(b, bytearray(L))
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001388
Thomas Wouters376446d2006-12-19 08:30:14 +00001389 del L[start:stop:step]
1390 del b[start:stop:step]
Ezio Melottib3aedd42010-11-20 19:04:17 +00001391 self.assertEqual(b, bytearray(L))
Thomas Wouters376446d2006-12-19 08:30:14 +00001392
Guido van Rossumd624f182006-04-24 13:47:05 +00001393 def test_setslice_trap(self):
1394 # This test verifies that we correctly handle assigning self
1395 # to a slice of self (the old Lambert Meertens trap).
Guido van Rossum254348e2007-11-21 19:29:53 +00001396 b = bytearray(range(256))
Guido van Rossumd624f182006-04-24 13:47:05 +00001397 b[8:] = b
Guido van Rossum254348e2007-11-21 19:29:53 +00001398 self.assertEqual(b, bytearray(list(range(8)) + list(range(256))))
Guido van Rossumd624f182006-04-24 13:47:05 +00001399
Guido van Rossum13e57212006-04-27 22:54:26 +00001400 def test_iconcat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001401 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001402 b1 = b
Guido van Rossum9c627722007-08-27 18:31:48 +00001403 b += b"def"
1404 self.assertEqual(b, b"abcdef")
Guido van Rossum13e57212006-04-27 22:54:26 +00001405 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001406 self.assertIs(b, b1)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001407 b += b"xyz"
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001408 self.assertEqual(b, b"abcdefxyz")
1409 try:
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001410 b += ""
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001411 except TypeError:
1412 pass
1413 else:
1414 self.fail("bytes += unicode didn't raise TypeError")
Guido van Rossum13e57212006-04-27 22:54:26 +00001415
1416 def test_irepeat(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001417 b = bytearray(b"abc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001418 b1 = b
1419 b *= 3
Guido van Rossum9c627722007-08-27 18:31:48 +00001420 self.assertEqual(b, b"abcabcabc")
Guido van Rossum13e57212006-04-27 22:54:26 +00001421 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001422 self.assertIs(b, b1)
Guido van Rossum13e57212006-04-27 22:54:26 +00001423
1424 def test_irepeat_1char(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001425 b = bytearray(b"x")
Guido van Rossum13e57212006-04-27 22:54:26 +00001426 b1 = b
1427 b *= 100
Guido van Rossum98297ee2007-11-06 21:34:58 +00001428 self.assertEqual(b, b"x"*100)
Guido van Rossum13e57212006-04-27 22:54:26 +00001429 self.assertEqual(b, b1)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001430 self.assertIs(b, b1)
Guido van Rossum13e57212006-04-27 22:54:26 +00001431
Guido van Rossum20188312006-05-05 15:15:40 +00001432 def test_alloc(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001433 b = bytearray()
Guido van Rossum20188312006-05-05 15:15:40 +00001434 alloc = b.__alloc__()
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001435 self.assertGreaterEqual(alloc, 0)
Guido van Rossum20188312006-05-05 15:15:40 +00001436 seq = [alloc]
1437 for i in range(100):
Guido van Rossum9c627722007-08-27 18:31:48 +00001438 b += b"x"
Guido van Rossum20188312006-05-05 15:15:40 +00001439 alloc = b.__alloc__()
Serhiy Storchaka7b6e3b92015-06-29 21:14:06 +03001440 self.assertGreater(alloc, len(b)) # including trailing null byte
Guido van Rossum20188312006-05-05 15:15:40 +00001441 if alloc not in seq:
1442 seq.append(alloc)
Guido van Rossum20188312006-05-05 15:15:40 +00001443
Serhiy Storchaka7b6e3b92015-06-29 21:14:06 +03001444 def test_init_alloc(self):
1445 b = bytearray()
1446 def g():
1447 for i in range(1, 100):
1448 yield i
1449 a = list(b)
1450 self.assertEqual(a, list(range(1, len(a)+1)))
1451 self.assertEqual(len(b), len(a))
1452 self.assertLessEqual(len(b), i)
1453 alloc = b.__alloc__()
1454 self.assertGreater(alloc, len(b)) # including trailing null byte
1455 b.__init__(g())
1456 self.assertEqual(list(b), list(range(1, 100)))
1457 self.assertEqual(len(b), 99)
1458 alloc = b.__alloc__()
1459 self.assertGreater(alloc, len(b))
1460
Neal Norwitz6968b052007-02-27 19:02:19 +00001461 def test_extend(self):
1462 orig = b'hello'
Guido van Rossum254348e2007-11-21 19:29:53 +00001463 a = bytearray(orig)
Neal Norwitz6968b052007-02-27 19:02:19 +00001464 a.extend(a)
1465 self.assertEqual(a, orig + orig)
1466 self.assertEqual(a[5:], orig)
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001467 a = bytearray(b'')
1468 # Test iterators that don't have a __length_hint__
1469 a.extend(map(int, orig * 25))
1470 a.extend(int(x) for x in orig * 25)
1471 self.assertEqual(a, orig * 50)
1472 self.assertEqual(a[-5:], orig)
1473 a = bytearray(b'')
1474 a.extend(iter(map(int, orig * 50)))
1475 self.assertEqual(a, orig * 50)
1476 self.assertEqual(a[-5:], orig)
1477 a = bytearray(b'')
1478 a.extend(list(map(int, orig * 50)))
1479 self.assertEqual(a, orig * 50)
1480 self.assertEqual(a[-5:], orig)
1481 a = bytearray(b'')
1482 self.assertRaises(ValueError, a.extend, [0, 1, 2, 256])
1483 self.assertRaises(ValueError, a.extend, [0, 1, 2, -1])
1484 self.assertEqual(len(a), 0)
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001485 a = bytearray(b'')
1486 a.extend([Indexable(ord('a'))])
1487 self.assertEqual(a, b'a')
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001488
Neal Norwitz6968b052007-02-27 19:02:19 +00001489 def test_remove(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001490 b = bytearray(b'hello')
Neal Norwitz6968b052007-02-27 19:02:19 +00001491 b.remove(ord('l'))
1492 self.assertEqual(b, b'helo')
1493 b.remove(ord('l'))
1494 self.assertEqual(b, b'heo')
1495 self.assertRaises(ValueError, lambda: b.remove(ord('l')))
1496 self.assertRaises(ValueError, lambda: b.remove(400))
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001497 self.assertRaises(TypeError, lambda: b.remove('e'))
Neal Norwitz6968b052007-02-27 19:02:19 +00001498 # remove first and last
1499 b.remove(ord('o'))
1500 b.remove(ord('h'))
1501 self.assertEqual(b, b'e')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001502 self.assertRaises(TypeError, lambda: b.remove(b'e'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001503 b.remove(Indexable(ord('e')))
1504 self.assertEqual(b, b'')
Neal Norwitz6968b052007-02-27 19:02:19 +00001505
Serhiy Storchaka7bf36da2016-05-16 22:15:38 +03001506 # test values outside of the ascii range: (0, 127)
1507 c = bytearray([126, 127, 128, 129])
1508 c.remove(127)
1509 self.assertEqual(c, bytes([126, 128, 129]))
1510 c.remove(129)
1511 self.assertEqual(c, bytes([126, 128]))
1512
Neal Norwitz6968b052007-02-27 19:02:19 +00001513 def test_pop(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001514 b = bytearray(b'world')
Neal Norwitz6968b052007-02-27 19:02:19 +00001515 self.assertEqual(b.pop(), ord('d'))
1516 self.assertEqual(b.pop(0), ord('w'))
1517 self.assertEqual(b.pop(-2), ord('r'))
1518 self.assertRaises(IndexError, lambda: b.pop(10))
Eli Bendersky1bc4f192011-03-04 04:55:25 +00001519 self.assertRaises(IndexError, lambda: bytearray().pop())
Mark Dickinson54a3db92009-09-06 10:19:23 +00001520 # test for issue #6846
1521 self.assertEqual(bytearray(b'\xff').pop(), 0xff)
Neal Norwitz6968b052007-02-27 19:02:19 +00001522
1523 def test_nosort(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001524 self.assertRaises(AttributeError, lambda: bytearray().sort())
Neal Norwitz6968b052007-02-27 19:02:19 +00001525
Neal Norwitz6968b052007-02-27 19:02:19 +00001526 def test_append(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001527 b = bytearray(b'hell')
Neal Norwitz6968b052007-02-27 19:02:19 +00001528 b.append(ord('o'))
1529 self.assertEqual(b, b'hello')
1530 self.assertEqual(b.append(100), None)
Guido van Rossum254348e2007-11-21 19:29:53 +00001531 b = bytearray()
Neal Norwitz6968b052007-02-27 19:02:19 +00001532 b.append(ord('A'))
1533 self.assertEqual(len(b), 1)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001534 self.assertRaises(TypeError, lambda: b.append(b'o'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001535 b = bytearray()
1536 b.append(Indexable(ord('A')))
1537 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001538
1539 def test_insert(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001540 b = bytearray(b'msssspp')
Neal Norwitz6968b052007-02-27 19:02:19 +00001541 b.insert(1, ord('i'))
1542 b.insert(4, ord('i'))
1543 b.insert(-2, ord('i'))
1544 b.insert(1000, ord('i'))
1545 self.assertEqual(b, b'mississippi')
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001546 self.assertRaises(TypeError, lambda: b.insert(0, b'1'))
Georg Brandl9a54d7c2008-07-16 23:15:30 +00001547 b = bytearray()
1548 b.insert(0, Indexable(ord('A')))
1549 self.assertEqual(b, b'A')
Neal Norwitz6968b052007-02-27 19:02:19 +00001550
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001551 def test_copied(self):
1552 # Issue 4348. Make sure that operations that don't mutate the array
1553 # copy the bytes.
1554 b = bytearray(b'abc')
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001555 self.assertIsNot(b, b.replace(b'abc', b'cde', 0))
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001556
1557 t = bytearray([i for i in range(256)])
1558 x = bytearray(b'')
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001559 self.assertIsNot(x, x.translate(t))
Benjamin Peterson0f3641c2008-11-19 22:05:52 +00001560
Guido van Rossum254348e2007-11-21 19:29:53 +00001561 def test_partition_bytearray_doesnt_share_nullstring(self):
1562 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001563 self.assertEqual(b, b"")
1564 self.assertEqual(c, b"")
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001565 self.assertIsNot(b, c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001566 b += b"!"
1567 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001568 a, b, c = bytearray(b"x").partition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001569 self.assertEqual(b, b"")
1570 self.assertEqual(c, b"")
1571 # Same for rpartition
Guido van Rossum254348e2007-11-21 19:29:53 +00001572 b, c, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001573 self.assertEqual(b, b"")
1574 self.assertEqual(c, b"")
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001575 self.assertIsNot(b, c)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001576 b += b"!"
1577 self.assertEqual(c, b"")
Guido van Rossum254348e2007-11-21 19:29:53 +00001578 c, b, a = bytearray(b"x").rpartition(b"y")
Guido van Rossum98297ee2007-11-06 21:34:58 +00001579 self.assertEqual(b, b"")
1580 self.assertEqual(c, b"")
1581
Antoine Pitrou5504e892008-12-06 21:27:53 +00001582 def test_resize_forbidden(self):
1583 # #4509: can't resize a bytearray when there are buffer exports, even
1584 # if it wouldn't reallocate the underlying buffer.
1585 # Furthermore, no destructive changes to the buffer may be applied
1586 # before raising the error.
1587 b = bytearray(range(10))
1588 v = memoryview(b)
1589 def resize(n):
1590 b[1:-1] = range(n + 1, 2*n - 1)
1591 resize(10)
1592 orig = b[:]
1593 self.assertRaises(BufferError, resize, 11)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001594 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001595 self.assertRaises(BufferError, resize, 9)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001596 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001597 self.assertRaises(BufferError, resize, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001598 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001599 # Other operations implying resize
1600 self.assertRaises(BufferError, b.pop, 0)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001601 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001602 self.assertRaises(BufferError, b.remove, b[1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001603 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001604 def delitem():
1605 del b[1]
1606 self.assertRaises(BufferError, delitem)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001607 self.assertEqual(b, orig)
Antoine Pitrou5504e892008-12-06 21:27:53 +00001608 # deleting a non-contiguous slice
1609 def delslice():
1610 b[1:-1:2] = b""
1611 self.assertRaises(BufferError, delslice)
Ezio Melottib3aedd42010-11-20 19:04:17 +00001612 self.assertEqual(b, orig)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001613
Stefan Krah650c1e82015-02-03 21:43:23 +01001614 @test.support.cpython_only
1615 def test_obsolete_write_lock(self):
1616 from _testcapi import getbuffer_with_null_view
1617 self.assertRaises(BufferError, getbuffer_with_null_view, bytearray())
Antoine Pitroufc8d6f42010-01-17 12:38:54 +00001618
Serhiy Storchakaaabafe72016-03-06 14:10:24 +02001619 def test_iterator_pickling2(self):
1620 orig = bytearray(b'abc')
1621 data = list(b'qwerty')
1622 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
1623 # initial iterator
1624 itorig = iter(orig)
1625 d = pickle.dumps((itorig, orig), proto)
1626 it, b = pickle.loads(d)
1627 b[:] = data
1628 self.assertEqual(type(it), type(itorig))
1629 self.assertEqual(list(it), data)
1630
1631 # running iterator
1632 next(itorig)
1633 d = pickle.dumps((itorig, orig), proto)
1634 it, b = pickle.loads(d)
1635 b[:] = data
1636 self.assertEqual(type(it), type(itorig))
1637 self.assertEqual(list(it), data[1:])
1638
1639 # empty iterator
1640 for i in range(1, len(orig)):
1641 next(itorig)
1642 d = pickle.dumps((itorig, orig), proto)
1643 it, b = pickle.loads(d)
1644 b[:] = data
1645 self.assertEqual(type(it), type(itorig))
1646 self.assertEqual(list(it), data[len(orig):])
1647
1648 # exhausted iterator
1649 self.assertRaises(StopIteration, next, itorig)
1650 d = pickle.dumps((itorig, orig), proto)
1651 it, b = pickle.loads(d)
1652 b[:] = data
1653 self.assertEqual(list(it), [])
1654
Serhiy Storchaka8dc2ec12016-03-30 21:01:26 +03001655 test_exhausted_iterator = test.list_tests.CommonTest.test_exhausted_iterator
Serhiy Storchakaaabafe72016-03-06 14:10:24 +02001656
Serhiy Storchakaaf658722016-07-03 14:41:36 +03001657 def test_iterator_length_hint(self):
1658 # Issue 27443: __length_hint__ can return negative integer
1659 ba = bytearray(b'ab')
1660 it = iter(ba)
1661 next(it)
1662 ba.clear()
1663 # Shouldn't raise an error
1664 self.assertEqual(list(it), [])
1665
1666
Christian Heimes510711d2008-01-30 11:57:58 +00001667class AssortedBytesTest(unittest.TestCase):
1668 #
1669 # Test various combinations of bytes and bytearray
1670 #
1671
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001672 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001673 def test_repr_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001674 for f in str, repr:
1675 self.assertEqual(f(bytearray()), "bytearray(b'')")
1676 self.assertEqual(f(bytearray([0])), "bytearray(b'\\x00')")
1677 self.assertEqual(f(bytearray([0, 1, 254, 255])),
1678 "bytearray(b'\\x00\\x01\\xfe\\xff')")
1679 self.assertEqual(f(b"abc"), "b'abc'")
1680 self.assertEqual(f(b"'"), '''b"'"''') # '''
1681 self.assertEqual(f(b"'\""), r"""b'\'"'""") # '
Christian Heimes510711d2008-01-30 11:57:58 +00001682
Serhiy Storchakad1af5ef2016-10-30 19:33:54 +02001683 @check_bytes_warnings
1684 def test_format(self):
1685 for b in b'abc', bytearray(b'abc'):
1686 self.assertEqual(format(b), str(b))
1687 self.assertEqual(format(b, ''), str(b))
1688 with self.assertRaisesRegex(TypeError,
1689 r'\b%s\b' % re.escape(type(b).__name__)):
1690 format(b, 's')
1691
Christian Heimes510711d2008-01-30 11:57:58 +00001692 def test_compare_bytes_to_bytearray(self):
1693 self.assertEqual(b"abc" == bytes(b"abc"), True)
1694 self.assertEqual(b"ab" != 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"abc" >= bytes(b"ab"), True)
1698 self.assertEqual(b"abc" > bytes(b"ab"), True)
1699
1700 self.assertEqual(b"abc" != bytes(b"abc"), False)
1701 self.assertEqual(b"ab" == 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"abc" < bytes(b"ab"), False)
1705 self.assertEqual(b"abc" <= bytes(b"ab"), False)
1706
1707 self.assertEqual(bytes(b"abc") == b"abc", True)
1708 self.assertEqual(bytes(b"ab") != 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"abc") >= b"ab", True)
1712 self.assertEqual(bytes(b"abc") > b"ab", True)
1713
1714 self.assertEqual(bytes(b"abc") != b"abc", False)
1715 self.assertEqual(bytes(b"ab") == 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"abc") < b"ab", False)
1719 self.assertEqual(bytes(b"abc") <= b"ab", False)
1720
Stefan Krah6e572b82013-01-26 13:06:36 +01001721 @test.support.requires_docstrings
Christian Heimes510711d2008-01-30 11:57:58 +00001722 def test_doc(self):
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001723 self.assertIsNotNone(bytearray.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001724 self.assertTrue(bytearray.__doc__.startswith("bytearray("), bytearray.__doc__)
Florent Xicluna9b86b9a2010-03-19 19:00:44 +00001725 self.assertIsNotNone(bytes.__doc__)
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001726 self.assertTrue(bytes.__doc__.startswith("bytes("), bytes.__doc__)
Christian Heimes510711d2008-01-30 11:57:58 +00001727
1728 def test_from_bytearray(self):
1729 sample = bytes(b"Hello world\n\x80\x81\xfe\xff")
1730 buf = memoryview(sample)
1731 b = bytearray(buf)
1732 self.assertEqual(b, bytearray(sample))
1733
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001734 @check_bytes_warnings
Christian Heimes510711d2008-01-30 11:57:58 +00001735 def test_to_str(self):
Florent Xiclunaedf5f0d2010-09-03 20:00:37 +00001736 self.assertEqual(str(b''), "b''")
1737 self.assertEqual(str(b'x'), "b'x'")
1738 self.assertEqual(str(b'\x80'), "b'\\x80'")
1739 self.assertEqual(str(bytearray(b'')), "bytearray(b'')")
1740 self.assertEqual(str(bytearray(b'x')), "bytearray(b'x')")
1741 self.assertEqual(str(bytearray(b'\x80')), "bytearray(b'\\x80')")
Christian Heimes510711d2008-01-30 11:57:58 +00001742
1743 def test_literal(self):
1744 tests = [
1745 (b"Wonderful spam", "Wonderful spam"),
1746 (br"Wonderful spam too", "Wonderful spam too"),
1747 (b"\xaa\x00\000\200", "\xaa\x00\000\200"),
1748 (br"\xaa\x00\000\200", r"\xaa\x00\000\200"),
1749 ]
1750 for b, s in tests:
1751 self.assertEqual(b, bytearray(s, 'latin-1'))
1752 for c in range(128, 256):
1753 self.assertRaises(SyntaxError, eval,
1754 'b"%s"' % chr(c))
1755
Christian Heimes510711d2008-01-30 11:57:58 +00001756 def test_split_bytearray(self):
1757 self.assertEqual(b'a b'.split(memoryview(b' ')), [b'a', b'b'])
1758
1759 def test_rsplit_bytearray(self):
1760 self.assertEqual(b'a b'.rsplit(memoryview(b' ')), [b'a', b'b'])
1761
Georg Brandleb2c9642008-05-30 12:05:02 +00001762 def test_return_self(self):
1763 # bytearray.replace must always return a new bytearray
1764 b = bytearray()
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001765 self.assertIsNot(b.replace(b'', b''), b)
Georg Brandleb2c9642008-05-30 12:05:02 +00001766
Serhiy Storchaka1dd49822015-03-20 16:54:57 +02001767 @unittest.skipUnless(sys.flags.bytes_warning,
1768 "BytesWarning is needed for this test: use -bb option")
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001769 def test_compare(self):
Serhiy Storchaka1dd49822015-03-20 16:54:57 +02001770 def bytes_warning():
1771 return test.support.check_warnings(('', BytesWarning))
1772 with bytes_warning():
1773 b'' == ''
1774 with bytes_warning():
1775 '' == b''
1776 with bytes_warning():
1777 b'' != ''
1778 with bytes_warning():
1779 '' != b''
1780 with bytes_warning():
1781 bytearray(b'') == ''
1782 with bytes_warning():
1783 '' == bytearray(b'')
1784 with bytes_warning():
1785 bytearray(b'') != ''
1786 with bytes_warning():
1787 '' != bytearray(b'')
1788 with bytes_warning():
1789 b'\0' == 0
1790 with bytes_warning():
1791 0 == b'\0'
1792 with bytes_warning():
1793 b'\0' != 0
1794 with bytes_warning():
1795 0 != b'\0'
Barry Warsaw9e9dcd62008-10-17 01:50:37 +00001796
Guido van Rossumd624f182006-04-24 13:47:05 +00001797 # Optimizations:
Guido van Rossume06b6b82006-04-23 07:43:54 +00001798 # __iter__? (optimization)
Guido van Rossumd624f182006-04-24 13:47:05 +00001799 # __reversed__? (optimization)
1800
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001801 # XXX More string methods? (Those that don't use character properties)
1802
Neal Norwitz6968b052007-02-27 19:02:19 +00001803 # There are tests in string_tests.py that are more
Martin Panter0d0db6c2016-04-10 08:45:26 +00001804 # comprehensive for things like partition, etc.
Neal Norwitz6968b052007-02-27 19:02:19 +00001805 # Unfortunately they are all bundled with tests that
1806 # are not appropriate for bytes
Guido van Rossume06b6b82006-04-23 07:43:54 +00001807
Guido van Rossum254348e2007-11-21 19:29:53 +00001808 # I've started porting some of those into bytearray_tests.py, we should port
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001809 # the rest that make sense (the code can be cleaned up to use modern
1810 # unittest methods at the same time).
1811
Martin Panter152a19c2016-04-06 06:37:17 +00001812class BytearrayPEP3137Test(unittest.TestCase):
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001813 def marshal(self, x):
Guido van Rossum254348e2007-11-21 19:29:53 +00001814 return bytearray(x)
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001815
1816 def test_returns_new_copy(self):
1817 val = self.marshal(b'1234')
1818 # On immutable types these MAY return a reference to themselves
Guido van Rossum254348e2007-11-21 19:29:53 +00001819 # but on mutable types like bytearray they MUST return a new copy.
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001820 for methname in ('zfill', 'rjust', 'ljust', 'center'):
1821 method = getattr(val, methname)
1822 newval = method(3)
1823 self.assertEqual(val, newval)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001824 self.assertIsNot(val, newval,
Gregory P. Smith60d241f2007-10-16 06:31:30 +00001825 methname+' returned self on a mutable object')
Antoine Pitroucbaa62502010-01-13 15:03:39 +00001826 for expr in ('val.split()[0]', 'val.rsplit()[0]',
1827 'val.partition(b".")[0]', 'val.rpartition(b".")[2]',
1828 'val.splitlines()[0]', 'val.replace(b"", b"")'):
1829 newval = eval(expr)
1830 self.assertEqual(val, newval)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001831 self.assertIsNot(val, newval,
Antoine Pitroucbaa62502010-01-13 15:03:39 +00001832 expr+' returned val on a mutable object')
Antoine Pitroucfc22b42012-10-16 21:07:23 +02001833 sep = self.marshal(b'')
1834 newval = sep.join([val])
1835 self.assertEqual(val, newval)
1836 self.assertIsNot(val, newval)
1837
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001838
Christian Heimes510711d2008-01-30 11:57:58 +00001839class FixedStringTest(test.string_tests.BaseTest):
Guido van Rossum9c627722007-08-27 18:31:48 +00001840 def fixtype(self, obj):
1841 if isinstance(obj, str):
Martin Panter152a19c2016-04-06 06:37:17 +00001842 return self.type2test(obj.encode("utf-8"))
Guido van Rossum9c627722007-08-27 18:31:48 +00001843 return super().fixtype(obj)
1844
Martin Panter152a19c2016-04-06 06:37:17 +00001845 contains_bytes = True
1846
Ezio Melotti0dceb562013-01-10 07:43:26 +02001847class ByteArrayAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001848 type2test = bytearray
1849
Ezio Melotti0dceb562013-01-10 07:43:26 +02001850class BytesAsStringTest(FixedStringTest, unittest.TestCase):
Christian Heimes510711d2008-01-30 11:57:58 +00001851 type2test = bytes
1852
Georg Brandlc7885542007-03-06 19:16:20 +00001853
Ezio Melotti0dceb562013-01-10 07:43:26 +02001854class SubclassTest:
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001855
1856 def test_basic(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001857 self.assertTrue(issubclass(self.type2test, self.basetype))
1858 self.assertIsInstance(self.type2test(), self.basetype)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001859
1860 a, b = b"abcd", b"efgh"
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001861 _a, _b = self.type2test(a), self.type2test(b)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001862
1863 # test comparison operators with subclass instances
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001864 self.assertTrue(_a == _a)
1865 self.assertTrue(_a != _b)
1866 self.assertTrue(_a < _b)
1867 self.assertTrue(_a <= _b)
1868 self.assertTrue(_b >= _a)
1869 self.assertTrue(_b > _a)
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001870 self.assertIsNot(_a, a)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001871
1872 # test concat of subclass instances
1873 self.assertEqual(a + b, _a + _b)
1874 self.assertEqual(a + b, a + _b)
1875 self.assertEqual(a + b, _a + b)
1876
1877 # test repeat
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001878 self.assertTrue(a*5 == _a*5)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001879
1880 def test_join(self):
1881 # Make sure join returns a NEW object for single item sequences
1882 # involving a subclass.
1883 # Make sure that it is of the appropriate type.
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001884 s1 = self.type2test(b"abcd")
1885 s2 = self.basetype().join([s1])
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001886 self.assertIsNot(s1, s2)
1887 self.assertIs(type(s2), self.basetype, type(s2))
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001888
1889 # Test reverse, calling join on subclass
1890 s3 = s1.join([b"abcd"])
Serhiy Storchaka604e74c2017-03-27 13:59:07 +03001891 self.assertIs(type(s3), self.basetype)
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001892
1893 def test_pickle(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001894 a = self.type2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001895 a.x = 10
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001896 a.y = self.type2test(b"efgh")
Hirokazu Yamamotod0d0b652008-10-23 00:38:15 +00001897 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001898 b = pickle.loads(pickle.dumps(a, proto))
1899 self.assertNotEqual(id(a), id(b))
1900 self.assertEqual(a, b)
1901 self.assertEqual(a.x, b.x)
1902 self.assertEqual(a.y, b.y)
1903 self.assertEqual(type(a), type(b))
1904 self.assertEqual(type(a.y), type(b.y))
1905
1906 def test_copy(self):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001907 a = self.type2test(b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001908 a.x = 10
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001909 a.y = self.type2test(b"efgh")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001910 for copy_method in (copy.copy, copy.deepcopy):
1911 b = copy_method(a)
1912 self.assertNotEqual(id(a), id(b))
1913 self.assertEqual(a, b)
1914 self.assertEqual(a.x, b.x)
1915 self.assertEqual(a.y, b.y)
1916 self.assertEqual(type(a), type(b))
1917 self.assertEqual(type(a.y), type(b.y))
1918
Serhiy Storchaka0855e702016-07-01 17:22:31 +03001919 def test_fromhex(self):
1920 b = self.type2test.fromhex('1a2B30')
1921 self.assertEqual(b, b'\x1a\x2b\x30')
1922 self.assertIs(type(b), self.type2test)
1923
1924 class B1(self.basetype):
1925 def __new__(cls, value):
1926 me = self.basetype.__new__(cls, value)
1927 me.foo = 'bar'
1928 return me
1929
1930 b = B1.fromhex('1a2B30')
1931 self.assertEqual(b, b'\x1a\x2b\x30')
1932 self.assertIs(type(b), B1)
1933 self.assertEqual(b.foo, 'bar')
1934
1935 class B2(self.basetype):
1936 def __init__(me, *args, **kwargs):
1937 if self.basetype is not bytes:
1938 self.basetype.__init__(me, *args, **kwargs)
1939 me.foo = 'bar'
1940
1941 b = B2.fromhex('1a2B30')
1942 self.assertEqual(b, b'\x1a\x2b\x30')
1943 self.assertIs(type(b), B2)
1944 self.assertEqual(b.foo, 'bar')
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001945
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001946
1947class ByteArraySubclass(bytearray):
1948 pass
1949
1950class BytesSubclass(bytes):
1951 pass
1952
Serhiy Storchaka15095802015-11-25 15:47:01 +02001953class OtherBytesSubclass(bytes):
1954 pass
1955
Ezio Melotti0dceb562013-01-10 07:43:26 +02001956class ByteArraySubclassTest(SubclassTest, unittest.TestCase):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001957 basetype = bytearray
1958 type2test = ByteArraySubclass
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001959
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001960 def test_init_override(self):
Guido van Rossum254348e2007-11-21 19:29:53 +00001961 class subclass(bytearray):
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001962 def __init__(me, newarg=1, *args, **kwargs):
1963 bytearray.__init__(me, *args, **kwargs)
1964 x = subclass(4, b"abcd")
Guido van Rossuma6c04be2007-11-03 00:24:24 +00001965 x = subclass(4, source=b"abcd")
1966 self.assertEqual(x, b"abcd")
1967 x = subclass(newarg=4, source=b"abcd")
1968 self.assertEqual(x, b"abcd")
1969
1970
Ezio Melotti0dceb562013-01-10 07:43:26 +02001971class BytesSubclassTest(SubclassTest, unittest.TestCase):
Serhiy Storchakae78592d2016-05-12 12:33:21 +03001972 basetype = bytes
1973 type2test = BytesSubclass
Alexandre Vassalotti41f58a72010-01-12 01:23:09 +00001974
1975
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00001976if __name__ == "__main__":
Ezio Melotti0dceb562013-01-10 07:43:26 +02001977 unittest.main()