annotate Lib/test/test_bytes.py @ 70626:aa07c1237f4e

(Merge 3.2) Issue #12016: Reindent decoders of HK and JP codecs
author Victor Stinner <victor.stinner@haypocalc.com>
date Fri, 03 Jun 2011 23:34:32 +0200
parents a1a1296556d7
children 536fccc75f5a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1 """Unit tests for the bytes and bytearray types.
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
2
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
3 XXX This is a mess. Common tests should be moved to buffer_tests.py,
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
4 which itself ought to be unified with string_tests.py (and the latter
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
5 should be modernized).
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
6 """
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
7
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
8 import os
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
9 import re
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
10 import sys
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
11 import copy
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
12 import functools
41425
0486deaaade3 An example of action-at-a-distance: fix the problems I had in test_io.py
Guido van Rossum <guido@python.org>
parents: 41412
diff changeset
13 import pickle
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
14 import tempfile
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
15 import unittest
47201
9c930a32cebb #2621 rename test.test_support to test.support
Benjamin Peterson <benjamin@python.org>
parents: 45058
diff changeset
16 import test.support
41038
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
17 import test.string_tests
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
18 import test.buffer_tests
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
19
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
20
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
21 if sys.flags.bytes_warning:
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
22 def check_bytes_warnings(func):
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
23 @functools.wraps(func)
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
24 def wrapper(*args, **kw):
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
25 with test.support.check_warnings(('', BytesWarning)):
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
26 return func(*args, **kw)
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
27 return wrapper
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
28 else:
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
29 # no-op
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
30 def check_bytes_warnings(func):
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
31 return func
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
32
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
33
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
34 class Indexable:
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
35 def __init__(self, value=0):
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
36 self.value = value
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
37 def __index__(self):
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
38 return self.value
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
39
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
40
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
41 class BaseBytesTest(unittest.TestCase):
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
42
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
43 def test_basics(self):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
44 b = self.type2test()
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
45 self.assertEqual(type(b), self.type2test)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
46 self.assertEqual(b.__class__, self.type2test)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
47
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
48 def test_copy(self):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
49 a = self.type2test(b"abcd")
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
50 for copy_method in (copy.copy, copy.deepcopy):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
51 b = copy_method(a)
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
52 self.assertEqual(a, b)
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
53 self.assertEqual(type(a), type(b))
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
54
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
55 def test_empty_sequence(self):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
56 b = self.type2test()
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
57 self.assertEqual(len(b), 0)
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
58 self.assertRaises(IndexError, lambda: b[0])
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
59 self.assertRaises(IndexError, lambda: b[1])
44160
9d152530910a Removed PyInt_GetMax and sys.maxint
Christian Heimes <christian@cheimes.de>
parents: 44137
diff changeset
60 self.assertRaises(IndexError, lambda: b[sys.maxsize])
9d152530910a Removed PyInt_GetMax and sys.maxint
Christian Heimes <christian@cheimes.de>
parents: 44137
diff changeset
61 self.assertRaises(IndexError, lambda: b[sys.maxsize+1])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
62 self.assertRaises(IndexError, lambda: b[10**100])
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
63 self.assertRaises(IndexError, lambda: b[-1])
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
64 self.assertRaises(IndexError, lambda: b[-2])
44160
9d152530910a Removed PyInt_GetMax and sys.maxint
Christian Heimes <christian@cheimes.de>
parents: 44137
diff changeset
65 self.assertRaises(IndexError, lambda: b[-sys.maxsize])
9d152530910a Removed PyInt_GetMax and sys.maxint
Christian Heimes <christian@cheimes.de>
parents: 44137
diff changeset
66 self.assertRaises(IndexError, lambda: b[-sys.maxsize-1])
9d152530910a Removed PyInt_GetMax and sys.maxint
Christian Heimes <christian@cheimes.de>
parents: 44137
diff changeset
67 self.assertRaises(IndexError, lambda: b[-sys.maxsize-2])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
68 self.assertRaises(IndexError, lambda: b[-10**100])
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
69
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
70 def test_from_list(self):
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
71 ints = list(range(256))
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
72 b = self.type2test(i for i in ints)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
73 self.assertEqual(len(b), 256)
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
74 self.assertEqual(list(b), ints)
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
75
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
76 def test_from_index(self):
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
77 b = self.type2test([Indexable(), Indexable(1), Indexable(254),
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
78 Indexable(255)])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
79 self.assertEqual(list(b), [0, 1, 254, 255])
60644
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
80 self.assertRaises(ValueError, self.type2test, [Indexable(-1)])
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
81 self.assertRaises(ValueError, self.type2test, [Indexable(256)])
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
82
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
83 def test_from_ssize(self):
60644
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
84 self.assertEqual(self.type2test(0), b'')
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
85 self.assertEqual(self.type2test(1), b'\x00')
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
86 self.assertEqual(self.type2test(5), b'\x00\x00\x00\x00\x00')
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
87 self.assertRaises(ValueError, self.type2test, -1)
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
88
60644
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
89 self.assertEqual(self.type2test('0', 'ascii'), b'0')
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
90 self.assertEqual(self.type2test(b'0'), b'0')
5e1dbc3ae767 Merged revisions 80125,80128,80130 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 58564
diff changeset
91 self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
92
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
93 def test_constructor_type_errors(self):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
94 self.assertRaises(TypeError, self.type2test, 0.0)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
95 class C:
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
96 pass
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
97 self.assertRaises(TypeError, self.type2test, ["0"])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
98 self.assertRaises(TypeError, self.type2test, [0.0])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
99 self.assertRaises(TypeError, self.type2test, [None])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
100 self.assertRaises(TypeError, self.type2test, [C()])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
101
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
102 def test_constructor_value_errors(self):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
103 self.assertRaises(ValueError, self.type2test, [-1])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
104 self.assertRaises(ValueError, self.type2test, [-sys.maxsize])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
105 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-1])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
106 self.assertRaises(ValueError, self.type2test, [-sys.maxsize-2])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
107 self.assertRaises(ValueError, self.type2test, [-10**100])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
108 self.assertRaises(ValueError, self.type2test, [256])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
109 self.assertRaises(ValueError, self.type2test, [257])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
110 self.assertRaises(ValueError, self.type2test, [sys.maxsize])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
111 self.assertRaises(ValueError, self.type2test, [sys.maxsize+1])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
112 self.assertRaises(ValueError, self.type2test, [10**100])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
113
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
114 def test_compare(self):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
115 b1 = self.type2test([1, 2, 3])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
116 b2 = self.type2test([1, 2, 3])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
117 b3 = self.type2test([1, 3])
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
118
42188
81930eaae715 Patch by Christian Heimes to change self.assert_(x == y) into
Guido van Rossum <guido@python.org>
parents: 42142
diff changeset
119 self.assertEqual(b1, b2)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
120 self.assertTrue(b2 != b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
121 self.assertTrue(b1 <= b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
122 self.assertTrue(b1 <= b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
123 self.assertTrue(b1 < b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
124 self.assertTrue(b1 >= b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
125 self.assertTrue(b3 >= b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
126 self.assertTrue(b3 > b2)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
127
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
128 self.assertFalse(b1 != b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
129 self.assertFalse(b2 == b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
130 self.assertFalse(b1 > b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
131 self.assertFalse(b1 > b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
132 self.assertFalse(b1 >= b3)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
133 self.assertFalse(b1 < b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
134 self.assertFalse(b3 < b2)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
135 self.assertFalse(b3 <= b2)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
136
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
137 @check_bytes_warnings
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
138 def test_compare_to_str(self):
43508
47c02791ec6b Patch #1049 by Thomas Lee.
Guido van Rossum <guido@python.org>
parents: 43487
diff changeset
139 # Byte comparisons with unicode should always fail!
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
140 # Test this for all expected byte orders and Unicode character
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
141 # sizes.
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
142 self.assertEqual(self.type2test(b"\0a\0b\0c") == "abc", False)
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
143 self.assertEqual(self.type2test(b"\0\0\0a\0\0\0b\0\0\0c") == "abc",
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
144 False)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
145 self.assertEqual(self.type2test(b"a\0b\0c\0") == "abc", False)
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
146 self.assertEqual(self.type2test(b"a\0\0\0b\0\0\0c\0\0\0") == "abc",
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
147 False)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
148 self.assertEqual(self.type2test() == str(), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
149 self.assertEqual(self.type2test() != str(), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
150
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
151 def test_reversed(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
152 input = list(map(ord, "Hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
153 b = self.type2test(input)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
154 output = list(reversed(b))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
155 input.reverse()
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
156 self.assertEqual(output, input)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
157
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
158 def test_getslice(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
159 def by(s):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
160 return self.type2test(map(ord, s))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
161 b = by("Hello, world")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
162
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
163 self.assertEqual(b[:5], by("Hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
164 self.assertEqual(b[1:5], by("ello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
165 self.assertEqual(b[5:7], by(", "))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
166 self.assertEqual(b[7:], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
167 self.assertEqual(b[7:12], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
168 self.assertEqual(b[7:100], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
169
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
170 self.assertEqual(b[:-7], by("Hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
171 self.assertEqual(b[-11:-7], by("ello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
172 self.assertEqual(b[-7:-5], by(", "))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
173 self.assertEqual(b[-5:], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
174 self.assertEqual(b[-5:12], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
175 self.assertEqual(b[-5:100], by("world"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
176 self.assertEqual(b[-100:5], by("Hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
177
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
178 def test_extended_getslice(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
179 # Test extended slicing by comparing with list slicing.
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
180 L = list(range(255))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
181 b = self.type2test(L)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
182 indices = (0, None, 1, 3, 19, 100, -1, -2, -31, -100)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
183 for start in indices:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
184 for stop in indices:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
185 # Skip step 0 (invalid)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
186 for step in indices[1:]:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
187 self.assertEqual(b[start:stop:step], self.type2test(L[start:stop:step]))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
188
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
189 def test_encoding(self):
53681
52a77ef069cd Issue #3672: Reject surrogates in utf-8 codec; add surrogates error
Martin v. Löwis <martin@v.loewis.de>
parents: 53094
diff changeset
190 sample = "Hello world\n\u1234\u5678\u9abc"
68142
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
191 for enc in ("utf-8", "utf-16"):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
192 b = self.type2test(sample, enc)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
193 self.assertEqual(b, self.type2test(sample.encode(enc)))
68142
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
194 self.assertRaises(UnicodeEncodeError, self.type2test, sample, "latin-1")
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
195 b = self.type2test(sample, "latin-1", "ignore")
53681
52a77ef069cd Issue #3672: Reject surrogates in utf-8 codec; add surrogates error
Martin v. Löwis <martin@v.loewis.de>
parents: 53094
diff changeset
196 self.assertEqual(b, self.type2test(sample[:-3], "utf-8"))
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
197
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
198 def test_decode(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
199 sample = "Hello world\n\u1234\u5678\u9abc\def0\def0"
68142
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
200 for enc in ("utf-8", "utf-16"):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
201 b = self.type2test(sample, enc)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
202 self.assertEqual(b.decode(enc), sample)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
203 sample = "Hello world\n\x80\x81\xfe\xff"
68142
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
204 b = self.type2test(sample, "latin-1")
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
205 self.assertRaises(UnicodeDecodeError, b.decode, "utf-8")
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
206 self.assertEqual(b.decode("utf-8", "ignore"), "Hello world\n")
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
207 self.assertEqual(b.decode(errors="ignore", encoding="utf-8"),
56099
1b7bdae02179 Merged revisions 74929 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 55848
diff changeset
208 "Hello world\n")
68142
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
209 # Default encoding is utf-8
be6c38d1817b Normalize the encoding names for Latin-1 and UTF-8 to
Marc-André Lemburg <mal@egenix.com>
parents: 67368
diff changeset
210 self.assertEqual(self.type2test(b'\xe2\x98\x83').decode(), '\u2603')
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
211
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
212 def test_from_int(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
213 b = self.type2test(0)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
214 self.assertEqual(b, self.type2test())
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
215 b = self.type2test(10)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
216 self.assertEqual(b, self.type2test([0]*10))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
217 b = self.type2test(10000)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
218 self.assertEqual(b, self.type2test([0]*10000))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
219
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
220 def test_concat(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
221 b1 = self.type2test(b"abc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
222 b2 = self.type2test(b"def")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
223 self.assertEqual(b1 + b2, b"abcdef")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
224 self.assertEqual(b1 + bytes(b"def"), b"abcdef")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
225 self.assertEqual(bytes(b"def") + b1, b"defabc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
226 self.assertRaises(TypeError, lambda: b1 + "def")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
227 self.assertRaises(TypeError, lambda: "abc" + b2)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
228
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
229 def test_repeat(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
230 for b in b"abc", self.type2test(b"abc"):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
231 self.assertEqual(b * 3, b"abcabcabc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
232 self.assertEqual(b * 0, b"")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
233 self.assertEqual(b * -1, b"")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
234 self.assertRaises(TypeError, lambda: b * 3.14)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
235 self.assertRaises(TypeError, lambda: 3.14 * b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
236 # XXX Shouldn't bytes and bytearray agree on what to raise?
63913
7dd5f5cfa179 Issue #9530: Fix undefined-behaviour-inducing overflow checks in bytes and bytearray implementations.
Mark Dickinson <dickinsm@gmail.com>
parents: 60856
diff changeset
237 with self.assertRaises((OverflowError, MemoryError)):
7dd5f5cfa179 Issue #9530: Fix undefined-behaviour-inducing overflow checks in bytes and bytearray implementations.
Mark Dickinson <dickinsm@gmail.com>
parents: 60856
diff changeset
238 c = b * sys.maxsize
7dd5f5cfa179 Issue #9530: Fix undefined-behaviour-inducing overflow checks in bytes and bytearray implementations.
Mark Dickinson <dickinsm@gmail.com>
parents: 60856
diff changeset
239 with self.assertRaises((OverflowError, MemoryError)):
7dd5f5cfa179 Issue #9530: Fix undefined-behaviour-inducing overflow checks in bytes and bytearray implementations.
Mark Dickinson <dickinsm@gmail.com>
parents: 60856
diff changeset
240 b *= sys.maxsize
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
241
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
242 def test_repeat_1char(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
243 self.assertEqual(self.type2test(b'x')*100, self.type2test([ord('x')]*100))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
244
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
245 def test_contains(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
246 b = self.type2test(b"abc")
58415
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
247 self.assertIn(ord('a'), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
248 self.assertIn(int(ord('a')), b)
58456
f0f221e1a7c0 use assert[Not]In where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58415
diff changeset
249 self.assertNotIn(200, b)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
250 self.assertRaises(ValueError, lambda: 300 in b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
251 self.assertRaises(ValueError, lambda: -1 in b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
252 self.assertRaises(TypeError, lambda: None in b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
253 self.assertRaises(TypeError, lambda: float(ord('a')) in b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
254 self.assertRaises(TypeError, lambda: "a" in b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
255 for f in bytes, bytearray:
58415
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
256 self.assertIn(f(b""), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
257 self.assertIn(f(b"a"), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
258 self.assertIn(f(b"b"), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
259 self.assertIn(f(b"c"), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
260 self.assertIn(f(b"ab"), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
261 self.assertIn(f(b"bc"), b)
f6a626f34840 use assert[Not]In where appropriate
Benjamin Peterson <benjamin@python.org>
parents: 58388
diff changeset
262 self.assertIn(f(b"abc"), b)
58456
f0f221e1a7c0 use assert[Not]In where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58415
diff changeset
263 self.assertNotIn(f(b"ac"), b)
f0f221e1a7c0 use assert[Not]In where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58415
diff changeset
264 self.assertNotIn(f(b"d"), b)
f0f221e1a7c0 use assert[Not]In where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58415
diff changeset
265 self.assertNotIn(f(b"dab"), b)
f0f221e1a7c0 use assert[Not]In where appropriate
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58415
diff changeset
266 self.assertNotIn(f(b"abd"), b)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
267
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
268 def test_fromhex(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
269 self.assertRaises(TypeError, self.type2test.fromhex)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
270 self.assertRaises(TypeError, self.type2test.fromhex, 1)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
271 self.assertEqual(self.type2test.fromhex(''), self.type2test())
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
272 b = bytearray([0x1a, 0x2b, 0x30])
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
273 self.assertEqual(self.type2test.fromhex('1a2B30'), b)
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
274 self.assertEqual(self.type2test.fromhex(' 1A 2B 30 '), b)
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
275 self.assertEqual(self.type2test.fromhex('0000'), b'\0\0')
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
276 self.assertRaises(TypeError, self.type2test.fromhex, b'1B')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
277 self.assertRaises(ValueError, self.type2test.fromhex, 'a')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
278 self.assertRaises(ValueError, self.type2test.fromhex, 'rt')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
279 self.assertRaises(ValueError, self.type2test.fromhex, '1a b cd')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
280 self.assertRaises(ValueError, self.type2test.fromhex, '\x00')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
281 self.assertRaises(ValueError, self.type2test.fromhex, '12 \x00 34')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
282
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
283 def test_join(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
284 self.assertEqual(self.type2test(b"").join([]), b"")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
285 self.assertEqual(self.type2test(b"").join([b""]), b"")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
286 for lst in [[b"abc"], [b"a", b"bc"], [b"ab", b"c"], [b"a", b"b", b"c"]]:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
287 lst = list(map(self.type2test, lst))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
288 self.assertEqual(self.type2test(b"").join(lst), b"abc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
289 self.assertEqual(self.type2test(b"").join(tuple(lst)), b"abc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
290 self.assertEqual(self.type2test(b"").join(iter(lst)), b"abc")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
291 self.assertEqual(self.type2test(b".").join([b"ab", b"cd"]), b"ab.cd")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
292 # XXX more...
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
293
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
294 def test_count(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
295 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
296 self.assertEqual(b.count(b'i'), 4)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
297 self.assertEqual(b.count(b'ss'), 2)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
298 self.assertEqual(b.count(b'w'), 0)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
299
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
300 def test_startswith(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
301 b = self.type2test(b'hello')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
302 self.assertFalse(self.type2test().startswith(b"anything"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
303 self.assertTrue(b.startswith(b"hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
304 self.assertTrue(b.startswith(b"hel"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
305 self.assertTrue(b.startswith(b"h"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
306 self.assertFalse(b.startswith(b"hellow"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
307 self.assertFalse(b.startswith(b"ha"))
69564
f393c507717a #6780: merge with 3.1.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69479 69563
diff changeset
308 with self.assertRaises(TypeError) as cm:
69563
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
309 b.startswith([b'h'])
69564
f393c507717a #6780: merge with 3.1.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69479 69563
diff changeset
310 exc = str(cm.exception)
69563
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
311 self.assertIn('bytes', exc)
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
312 self.assertIn('tuple', exc)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
313
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
314 def test_endswith(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
315 b = self.type2test(b'hello')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
316 self.assertFalse(bytearray().endswith(b"anything"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
317 self.assertTrue(b.endswith(b"hello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
318 self.assertTrue(b.endswith(b"llo"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
319 self.assertTrue(b.endswith(b"o"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
320 self.assertFalse(b.endswith(b"whello"))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
321 self.assertFalse(b.endswith(b"no"))
69564
f393c507717a #6780: merge with 3.1.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69479 69563
diff changeset
322 with self.assertRaises(TypeError) as cm:
69563
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
323 b.endswith([b'o'])
69564
f393c507717a #6780: merge with 3.1.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69479 69563
diff changeset
324 exc = str(cm.exception)
69563
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
325 self.assertIn('bytes', exc)
bcbf8c3c4a88 #6780: fix starts/endswith error message to mention that tuples are accepted too.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69472
diff changeset
326 self.assertIn('tuple', exc)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
327
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
328 def test_find(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
329 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
330 self.assertEqual(b.find(b'ss'), 2)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
331 self.assertEqual(b.find(b'ss', 3), 5)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
332 self.assertEqual(b.find(b'ss', 1, 7), 2)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
333 self.assertEqual(b.find(b'ss', 1, 3), -1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
334 self.assertEqual(b.find(b'w'), -1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
335 self.assertEqual(b.find(b'mississippian'), -1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
336
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
337 def test_rfind(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
338 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
339 self.assertEqual(b.rfind(b'ss'), 5)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
340 self.assertEqual(b.rfind(b'ss', 3), 5)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
341 self.assertEqual(b.rfind(b'ss', 0, 6), 2)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
342 self.assertEqual(b.rfind(b'w'), -1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
343 self.assertEqual(b.rfind(b'mississippian'), -1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
344
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
345 def test_index(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
346 b = self.type2test(b'world')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
347 self.assertEqual(b.index(b'w'), 0)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
348 self.assertEqual(b.index(b'orl'), 1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
349 self.assertRaises(ValueError, b.index, b'worm')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
350 self.assertRaises(ValueError, b.index, b'ldo')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
351
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
352 def test_rindex(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
353 # XXX could be more rigorous
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
354 b = self.type2test(b'world')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
355 self.assertEqual(b.rindex(b'w'), 0)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
356 self.assertEqual(b.rindex(b'orl'), 1)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
357 self.assertRaises(ValueError, b.rindex, b'worm')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
358 self.assertRaises(ValueError, b.rindex, b'ldo')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
359
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
360 def test_replace(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
361 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
362 self.assertEqual(b.replace(b'i', b'a'), b'massassappa')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
363 self.assertEqual(b.replace(b'ss', b'x'), b'mixixippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
364
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
365 def test_split(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
366 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
367 self.assertEqual(b.split(b'i'), [b'm', b'ss', b'ss', b'pp', b''])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
368 self.assertEqual(b.split(b'ss'), [b'mi', b'i', b'ippi'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
369 self.assertEqual(b.split(b'w'), [b])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
370
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
371 def test_split_whitespace(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
372 for b in (b' arf barf ', b'arf\tbarf', b'arf\nbarf', b'arf\rbarf',
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
373 b'arf\fbarf', b'arf\vbarf'):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
374 b = self.type2test(b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
375 self.assertEqual(b.split(), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
376 self.assertEqual(b.split(None), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
377 self.assertEqual(b.split(None, 2), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
378 for b in (b'a\x1Cb', b'a\x1Db', b'a\x1Eb', b'a\x1Fb'):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
379 b = self.type2test(b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
380 self.assertEqual(b.split(), [b])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
381 self.assertEqual(self.type2test(b' a bb c ').split(None, 0), [b'a bb c '])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
382 self.assertEqual(self.type2test(b' a bb c ').split(None, 1), [b'a', b'bb c '])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
383 self.assertEqual(self.type2test(b' a bb c ').split(None, 2), [b'a', b'bb', b'c '])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
384 self.assertEqual(self.type2test(b' a bb c ').split(None, 3), [b'a', b'bb', b'c'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
385
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
386 def test_split_string_error(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
387 self.assertRaises(TypeError, self.type2test(b'a b').split, ' ')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
388
58276
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
389 def test_split_unicodewhitespace(self):
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
390 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
391 self.assertEqual(b.split(), [b'\x1c\x1d\x1e\x1f'])
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
392
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
393 def test_rsplit(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
394 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
395 self.assertEqual(b.rsplit(b'i'), [b'm', b'ss', b'ss', b'pp', b''])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
396 self.assertEqual(b.rsplit(b'ss'), [b'mi', b'i', b'ippi'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
397 self.assertEqual(b.rsplit(b'w'), [b])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
398
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
399 def test_rsplit_whitespace(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
400 for b in (b' arf barf ', b'arf\tbarf', b'arf\nbarf', b'arf\rbarf',
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
401 b'arf\fbarf', b'arf\vbarf'):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
402 b = self.type2test(b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
403 self.assertEqual(b.rsplit(), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
404 self.assertEqual(b.rsplit(None), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
405 self.assertEqual(b.rsplit(None, 2), [b'arf', b'barf'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
406 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 0), [b' a bb c'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
407 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 1), [b' a bb', b'c'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
408 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 2), [b' a', b'bb', b'c'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
409 self.assertEqual(self.type2test(b' a bb c ').rsplit(None, 3), [b'a', b'bb', b'c'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
410
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
411 def test_rsplit_string_error(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
412 self.assertRaises(TypeError, self.type2test(b'a b').rsplit, ' ')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
413
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
414 def test_rsplit_unicodewhitespace(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
415 b = self.type2test(b"\x09\x0A\x0B\x0C\x0D\x1C\x1D\x1E\x1F")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
416 self.assertEqual(b.rsplit(), [b'\x1c\x1d\x1e\x1f'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
417
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
418 def test_partition(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
419 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
420 self.assertEqual(b.partition(b'ss'), (b'mi', b'ss', b'issippi'))
58276
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
421 self.assertEqual(b.partition(b'w'), (b'mississippi', b'', b''))
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
422
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
423 def test_rpartition(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
424 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
425 self.assertEqual(b.rpartition(b'ss'), (b'missi', b'ss', b'ippi'))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
426 self.assertEqual(b.rpartition(b'i'), (b'mississipp', b'i', b''))
58276
b7649733a354 Merged revisions 77448 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58264
diff changeset
427 self.assertEqual(b.rpartition(b'w'), (b'', b'', b'mississippi'))
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
428
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
429 def test_pickling(self):
49574
bbab11cdb175 Issue #4183: Some tests didn't run with pickle.HIGHEST_PROTOCOL.
Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp>
parents: 49538
diff changeset
430 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
431 for b in b"", b"a", b"abc", b"\xffab\x80", b"\0\0\377\0\0":
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
432 b = self.type2test(b)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
433 ps = pickle.dumps(b, proto)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
434 q = pickle.loads(ps)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
435 self.assertEqual(b, q)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
436
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
437 def test_strip(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
438 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
439 self.assertEqual(b.strip(b'i'), b'mississipp')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
440 self.assertEqual(b.strip(b'm'), b'ississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
441 self.assertEqual(b.strip(b'pi'), b'mississ')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
442 self.assertEqual(b.strip(b'im'), b'ssissipp')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
443 self.assertEqual(b.strip(b'pim'), b'ssiss')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
444 self.assertEqual(b.strip(b), b'')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
445
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
446 def test_lstrip(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
447 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
448 self.assertEqual(b.lstrip(b'i'), b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
449 self.assertEqual(b.lstrip(b'm'), b'ississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
450 self.assertEqual(b.lstrip(b'pi'), b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
451 self.assertEqual(b.lstrip(b'im'), b'ssissippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
452 self.assertEqual(b.lstrip(b'pim'), b'ssissippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
453
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
454 def test_rstrip(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
455 b = self.type2test(b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
456 self.assertEqual(b.rstrip(b'i'), b'mississipp')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
457 self.assertEqual(b.rstrip(b'm'), b'mississippi')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
458 self.assertEqual(b.rstrip(b'pi'), b'mississ')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
459 self.assertEqual(b.rstrip(b'im'), b'mississipp')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
460 self.assertEqual(b.rstrip(b'pim'), b'mississ')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
461
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
462 def test_strip_whitespace(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
463 b = self.type2test(b' \t\n\r\f\vabc \t\n\r\f\v')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
464 self.assertEqual(b.strip(), b'abc')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
465 self.assertEqual(b.lstrip(), b'abc \t\n\r\f\v')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
466 self.assertEqual(b.rstrip(), b' \t\n\r\f\vabc')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
467
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
468 def test_strip_bytearray(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
469 self.assertEqual(self.type2test(b'abc').strip(memoryview(b'ac')), b'b')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
470 self.assertEqual(self.type2test(b'abc').lstrip(memoryview(b'ac')), b'bc')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
471 self.assertEqual(self.type2test(b'abc').rstrip(memoryview(b'ac')), b'ab')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
472
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
473 def test_strip_string_error(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
474 self.assertRaises(TypeError, self.type2test(b'abc').strip, 'b')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
475 self.assertRaises(TypeError, self.type2test(b'abc').lstrip, 'b')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
476 self.assertRaises(TypeError, self.type2test(b'abc').rstrip, 'b')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
477
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
478 def test_ord(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
479 b = self.type2test(b'\0A\x7f\x80\xff')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
480 self.assertEqual([ord(b[i:i+1]) for i in range(len(b))],
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
481 [0, 65, 127, 128, 255])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
482
53094
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
483 def test_maketrans(self):
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
484 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'
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
485 self.assertEqual(self.type2test.maketrans(b'abc', b'xyz'), transtable)
56515
67bb17c2e868 Merged revisions 75404 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 55848
diff changeset
486 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'
67bb17c2e868 Merged revisions 75404 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 55848
diff changeset
487 self.assertEqual(self.type2test.maketrans(b'\375\376\377', b'xyz'), transtable)
53094
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
488 self.assertRaises(ValueError, self.type2test.maketrans, b'abc', b'xyzq')
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
489 self.assertRaises(TypeError, self.type2test.maketrans, 'abc', 'def')
4407e4d07084 Add bytes/bytearray.maketrans() to mirror str.maketrans(), and deprecate
Georg Brandl <georg@python.org>
parents: 50328
diff changeset
490
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
491 def test_none_arguments(self):
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
492 # issue 11828
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
493 b = self.type2test(b'hello')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
494 l = self.type2test(b'l')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
495 h = self.type2test(b'h')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
496 x = self.type2test(b'x')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
497 o = self.type2test(b'o')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
498
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
499 self.assertEqual(2, b.find(l, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
500 self.assertEqual(3, b.find(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
501 self.assertEqual(2, b.find(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
502 self.assertEqual(0, b.find(h, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
503
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
504 self.assertEqual(3, b.rfind(l, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
505 self.assertEqual(3, b.rfind(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
506 self.assertEqual(2, b.rfind(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
507 self.assertEqual(0, b.rfind(h, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
508
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
509 self.assertEqual(2, b.index(l, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
510 self.assertEqual(3, b.index(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
511 self.assertEqual(2, b.index(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
512 self.assertEqual(0, b.index(h, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
513
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
514 self.assertEqual(3, b.rindex(l, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
515 self.assertEqual(3, b.rindex(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
516 self.assertEqual(2, b.rindex(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
517 self.assertEqual(0, b.rindex(h, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
518
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
519 self.assertEqual(2, b.count(l, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
520 self.assertEqual(1, b.count(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
521 self.assertEqual(1, b.count(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
522 self.assertEqual(0, b.count(x, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
523
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
524 self.assertEqual(True, b.endswith(o, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
525 self.assertEqual(True, b.endswith(o, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
526 self.assertEqual(True, b.endswith(l, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
527 self.assertEqual(False, b.endswith(x, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
528
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
529 self.assertEqual(True, b.startswith(h, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
530 self.assertEqual(True, b.startswith(l, -2, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
531 self.assertEqual(True, b.startswith(h, None, -2))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
532 self.assertEqual(False, b.startswith(x, None, None))
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
533
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
534 def test_find_etc_raise_correct_error_messages(self):
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
535 # issue 11828
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
536 b = self.type2test(b'hello')
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
537 x = self.type2test(b'x')
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
538 self.assertRaisesRegex(TypeError, r'\bfind\b', b.find,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
539 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
540 self.assertRaisesRegex(TypeError, r'\brfind\b', b.rfind,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
541 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
542 self.assertRaisesRegex(TypeError, r'\bindex\b', b.index,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
543 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
544 self.assertRaisesRegex(TypeError, r'\brindex\b', b.rindex,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
545 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
546 self.assertRaisesRegex(TypeError, r'\bcount\b', b.count,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
547 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
548 self.assertRaisesRegex(TypeError, r'\bstartswith\b', b.startswith,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
549 x, None, None, None)
69479
87f9519203f4 Use non-deprecated method name.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 69473
diff changeset
550 self.assertRaisesRegex(TypeError, r'\bendswith\b', b.endswith,
69472
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
551 x, None, None, None)
77c657e47b5c startswith and endswith don't accept None as slice index. Patch by Torsten Becker. (closes #11828)
Jesus Cea <jcea@jcea.es>
parents: 66340
diff changeset
552
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
553
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
554 class BytesTest(BaseBytesTest):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
555 type2test = bytes
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
556
48623
bf50085fbcd3 #2538: bytes objects can only provide read-only buffers
Antoine Pitrou <solipsis@pitrou.net>
parents: 48508
diff changeset
557 def test_buffer_is_readonly(self):
48710
e0c65922b878 Issue #3571: test_bytes mistakingly closed stdin
Antoine Pitrou <solipsis@pitrou.net>
parents: 48623
diff changeset
558 fd = os.dup(sys.stdin.fileno())
e0c65922b878 Issue #3571: test_bytes mistakingly closed stdin
Antoine Pitrou <solipsis@pitrou.net>
parents: 48623
diff changeset
559 with open(fd, "rb", buffering=0) as f:
48623
bf50085fbcd3 #2538: bytes objects can only provide read-only buffers
Antoine Pitrou <solipsis@pitrou.net>
parents: 48508
diff changeset
560 self.assertRaises(TypeError, f.readinto, b"")
bf50085fbcd3 #2538: bytes objects can only provide read-only buffers
Antoine Pitrou <solipsis@pitrou.net>
parents: 48508
diff changeset
561
48871
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
562 def test_custom(self):
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
563 class A:
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
564 def __bytes__(self):
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
565 return b'abc'
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
566 self.assertEqual(bytes(A()), b'abc')
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
567 class A: pass
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
568 self.assertRaises(TypeError, bytes, A())
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
569 class A:
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
570 def __bytes__(self):
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
571 return None
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
572 self.assertRaises(TypeError, bytes, A())
548dccc4008e make bytes(o) respect __bytes__ #2415
Benjamin Peterson <benjamin@python.org>
parents: 48769
diff changeset
573
67368
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
574 # Test PyBytes_FromFormat()
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
575 def test_from_format(self):
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
576 test.support.import_module('ctypes')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
577 from ctypes import pythonapi, py_object, c_int, c_char_p
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
578 PyBytes_FromFormat = pythonapi.PyBytes_FromFormat
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
579 PyBytes_FromFormat.restype = py_object
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
580
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
581 self.assertEqual(PyBytes_FromFormat(b'format'),
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
582 b'format')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
583
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
584 self.assertEqual(PyBytes_FromFormat(b'%'), b'%')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
585 self.assertEqual(PyBytes_FromFormat(b'%%'), b'%')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
586 self.assertEqual(PyBytes_FromFormat(b'%%s'), b'%s')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
587 self.assertEqual(PyBytes_FromFormat(b'[%%]'), b'[%]')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
588 self.assertEqual(PyBytes_FromFormat(b'%%%c', c_int(ord('_'))), b'%_')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
589
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
590 self.assertEqual(PyBytes_FromFormat(b'c:%c', c_int(255)),
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
591 b'c:\xff')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
592 self.assertEqual(PyBytes_FromFormat(b's:%s', c_char_p(b'cstr')),
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
593 b's:cstr')
4cc47a9083e3 test_bytes: test PyBytes_FromFormat() using ctypes
Victor Stinner <victor.stinner@haypocalc.com>
parents: 66837
diff changeset
594
48623
bf50085fbcd3 #2538: bytes objects can only provide read-only buffers
Antoine Pitrou <solipsis@pitrou.net>
parents: 48508
diff changeset
595
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
596 class ByteArrayTest(BaseBytesTest):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
597 type2test = bytearray
41412
522b3f35d0eb Bytes should never equal unicode.
Guido van Rossum <guido@python.org>
parents: 41411
diff changeset
598
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
599 def test_nohash(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
600 self.assertRaises(TypeError, hash, bytearray())
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
601
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
602 def test_bytearray_api(self):
41855
057a9c3d067d Fix tset_bytes.py.
Guido van Rossum <guido@python.org>
parents: 41664
diff changeset
603 short_sample = b"Hello world\n"
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
604 sample = short_sample + b"\0"*(20 - len(short_sample))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
605 tfn = tempfile.mktemp()
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
606 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
607 # Prepare
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
608 with open(tfn, "wb") as f:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
609 f.write(short_sample)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
610 # Test readinto
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
611 with open(tfn, "rb") as f:
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
612 b = bytearray(20)
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
613 n = f.readinto(b)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
614 self.assertEqual(n, len(short_sample))
42142
a53e48be6df9 Merged revisions 56125-56153 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 42084
diff changeset
615 self.assertEqual(list(b), list(sample))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
616 # Test writing in binary mode
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
617 with open(tfn, "wb") as f:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
618 f.write(b)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
619 with open(tfn, "rb") as f:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
620 self.assertEqual(f.read(), sample)
41455
5dfd2f4bc834 Support name and mode attributes on all file types.
Guido van Rossum <guido@python.org>
parents: 41451
diff changeset
621 # Text mode is ambiguous; don't test
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
622 finally:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
623 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
624 os.remove(tfn)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
625 except os.error:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
626 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
627
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
628 def test_reverse(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
629 b = bytearray(b'hello')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
630 self.assertEqual(b.reverse(), None)
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
631 self.assertEqual(b, b'olleh')
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
632 b = bytearray(b'hello1') # test even number of items
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
633 b.reverse()
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
634 self.assertEqual(b, b'1olleh')
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
635 b = bytearray()
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
636 b.reverse()
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
637 self.assertFalse(b)
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
638
68221
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
639 def test_clear(self):
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
640 b = bytearray(b'python')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
641 b.clear()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
642 self.assertEqual(b, b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
643
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
644 b = bytearray(b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
645 b.clear()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
646 self.assertEqual(b, b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
647
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
648 b = bytearray(b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
649 b.append(ord('r'))
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
650 b.clear()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
651 b.append(ord('p'))
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
652 self.assertEqual(b, b'p')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
653
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
654 def test_copy(self):
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
655 b = bytearray(b'abc')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
656 bb = b.copy()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
657 self.assertEqual(bb, b'abc')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
658
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
659 b = bytearray(b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
660 bb = b.copy()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
661 self.assertEqual(bb, b'')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
662
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
663 # test that it's indeed a copy and not a reference
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
664 b = bytearray(b'abc')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
665 bb = b.copy()
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
666 self.assertEqual(b, bb)
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
667 self.assertIsNot(b, bb)
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
668 bb.append(ord('d'))
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
669 self.assertEqual(bb, b'abcd')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
670 self.assertEqual(b, b'abc')
08804aea08f1 Issue #10516: added copy() and clear() methods to bytearrays as well
Eli Bendersky <eliben@gmail.com>
parents: 68142
diff changeset
671
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
672 def test_regexps(self):
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
673 def by(s):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
674 return bytearray(map(ord, s))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
675 b = by("Hello, world")
48769
83b2d07b475b #2834: Change re module semantics, so that str and bytes mixing is forbidden,
Antoine Pitrou <solipsis@pitrou.net>
parents: 48710
diff changeset
676 self.assertEqual(re.findall(br"\w+", b), [by("Hello"), by("world")])
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
677
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
678 def test_setitem(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
679 b = bytearray([1, 2, 3])
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
680 b[1] = 100
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
681 self.assertEqual(b, bytearray([1, 100, 3]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
682 b[-1] = 200
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
683 self.assertEqual(b, bytearray([1, 100, 200]))
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
684 b[0] = Indexable(10)
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
685 self.assertEqual(b, bytearray([10, 100, 200]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
686 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
687 b[3] = 0
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
688 self.fail("Didn't raise IndexError")
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
689 except IndexError:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
690 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
691 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
692 b[-10] = 0
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
693 self.fail("Didn't raise IndexError")
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
694 except IndexError:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
695 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
696 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
697 b[0] = 256
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
698 self.fail("Didn't raise ValueError")
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
699 except ValueError:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
700 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
701 try:
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
702 b[0] = Indexable(-1)
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
703 self.fail("Didn't raise ValueError")
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
704 except ValueError:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
705 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
706 try:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
707 b[0] = None
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
708 self.fail("Didn't raise TypeError")
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
709 except TypeError:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
710 pass
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
711
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
712 def test_delitem(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
713 b = bytearray(range(10))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
714 del b[0]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
715 self.assertEqual(b, bytearray(range(1, 10)))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
716 del b[-1]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
717 self.assertEqual(b, bytearray(range(1, 9)))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
718 del b[4]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
719 self.assertEqual(b, bytearray([1, 2, 3, 4, 6, 7, 8]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
720
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
721 def test_setslice(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
722 b = bytearray(range(10))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
723 self.assertEqual(list(b), list(range(10)))
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
724
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
725 b[0:5] = bytearray([1, 1, 1, 1, 1])
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
726 self.assertEqual(b, bytearray([1, 1, 1, 1, 1, 5, 6, 7, 8, 9]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
727
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
728 del b[0:-5]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
729 self.assertEqual(b, bytearray([5, 6, 7, 8, 9]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
730
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
731 b[0:0] = bytearray([0, 1, 2, 3, 4])
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
732 self.assertEqual(b, bytearray(range(10)))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
733
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
734 b[-7:-3] = bytearray([100, 101])
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
735 self.assertEqual(b, bytearray([0, 1, 2, 100, 101, 7, 8, 9]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
736
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
737 b[3:5] = [3, 4, 5, 6]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
738 self.assertEqual(b, bytearray(range(10)))
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
739
39643
3dd61ffd049e Fix buglet in slice assignment of bytesobjects: assigning to b[3:0] ('stop'
Thomas Wouters <thomas@python.org>
parents: 39631
diff changeset
740 b[3:0] = [42, 42, 42]
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
741 self.assertEqual(b, bytearray([0, 1, 2, 42, 42, 42, 3, 4, 5, 6, 7, 8, 9]))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
742
40472
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
743 def test_extended_set_del_slice(self):
58564
6de64896c6f8 Merged revisions 77823 via svnmerge from
Mark Dickinson <dickinsm@gmail.com>
parents: 58389
diff changeset
744 indices = (0, None, 1, 3, 19, 300, 1<<333, -1, -2, -31, -300)
40472
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
745 for start in indices:
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
746 for stop in indices:
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
747 # Skip invalid step 0
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
748 for step in indices[1:]:
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
749 L = list(range(255))
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
750 b = bytearray(L)
40472
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
751 # Make sure we have a slice of exactly the right length,
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
752 # but with different data.
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
753 data = L[start:stop:step]
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
754 data.reverse()
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
755 L[start:stop:step] = data
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
756 b[start:stop:step] = data
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
757 self.assertEqual(b, bytearray(L))
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
758
40472
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
759 del L[start:stop:step]
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
760 del b[start:stop:step]
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
761 self.assertEqual(b, bytearray(L))
40472
9548ca0e6560 Implement extended slicing in bytes objects.
Thomas Wouters <thomas@python.org>
parents: 39643
diff changeset
762
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
763 def test_setslice_trap(self):
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
764 # This test verifies that we correctly handle assigning self
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
765 # to a slice of self (the old Lambert Meertens trap).
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
766 b = bytearray(range(256))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
767 b[8:] = b
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
768 self.assertEqual(b, bytearray(list(range(8)) + list(range(256))))
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
769
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
770 def test_iconcat(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
771 b = bytearray(b"abc")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
772 b1 = b
42907
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
773 b += b"def"
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
774 self.assertEqual(b, b"abcdef")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
775 self.assertEqual(b, b1)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
776 self.assertTrue(b is b1)
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
777 b += b"xyz"
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
778 self.assertEqual(b, b"abcdefxyz")
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
779 try:
41575
beacf40b5960 Rip out all the u"..." literals and calls to unicode().
Guido van Rossum <guido@python.org>
parents: 41455
diff changeset
780 b += ""
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
781 except TypeError:
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
782 pass
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
783 else:
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
784 self.fail("bytes += unicode didn't raise TypeError")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
785
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
786 def test_irepeat(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
787 b = bytearray(b"abc")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
788 b1 = b
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
789 b *= 3
42907
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
790 self.assertEqual(b, b"abcabcabc")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
791 self.assertEqual(b, b1)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
792 self.assertTrue(b is b1)
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
793
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
794 def test_irepeat_1char(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
795 b = bytearray(b"x")
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
796 b1 = b
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
797 b *= 100
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
798 self.assertEqual(b, b"x"*100)
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
799 self.assertEqual(b, b1)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
800 self.assertTrue(b is b1)
38017
0b8e782b8484 Implement bytes += bytes, bytes *= int, int in bytes, bytes in bytes.
Guido van Rossum <guido@python.org>
parents: 37960
diff changeset
801
38117
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
802 def test_alloc(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
803 b = bytearray()
38117
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
804 alloc = b.__alloc__()
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
805 self.assertTrue(alloc >= 0)
38117
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
806 seq = [alloc]
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
807 for i in range(100):
42907
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
808 b += b"x"
38117
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
809 alloc = b.__alloc__()
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
810 self.assertTrue(alloc >= len(b))
38117
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
811 if alloc not in seq:
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
812 seq.append(alloc)
57bd7c7d91e3 Adding bytes.join() -- a class methods that concatenates an iterable of bytes.
Guido van Rossum <guido@python.org>
parents: 38017
diff changeset
813
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
814 def test_extend(self):
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
815 orig = b'hello'
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
816 a = bytearray(orig)
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
817 a.extend(a)
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
818 self.assertEqual(a, orig + orig)
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
819 self.assertEqual(a[5:], orig)
44137
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
820 a = bytearray(b'')
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
821 # Test iterators that don't have a __length_hint__
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
822 a.extend(map(int, orig * 25))
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
823 a.extend(int(x) for x in orig * 25)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
824 self.assertEqual(a, orig * 50)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
825 self.assertEqual(a[-5:], orig)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
826 a = bytearray(b'')
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
827 a.extend(iter(map(int, orig * 50)))
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
828 self.assertEqual(a, orig * 50)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
829 self.assertEqual(a[-5:], orig)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
830 a = bytearray(b'')
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
831 a.extend(list(map(int, orig * 50)))
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
832 self.assertEqual(a, orig * 50)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
833 self.assertEqual(a[-5:], orig)
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
834 a = bytearray(b'')
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
835 self.assertRaises(ValueError, a.extend, [0, 1, 2, 256])
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
836 self.assertRaises(ValueError, a.extend, [0, 1, 2, -1])
825cffd6c0cd Issue #1283: Allow any iterable of integers to be passed to
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 43963
diff changeset
837 self.assertEqual(len(a), 0)
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
838 a = bytearray(b'')
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
839 a.extend([Indexable(ord('a'))])
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
840 self.assertEqual(a, b'a')
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
841
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
842 def test_remove(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
843 b = bytearray(b'hello')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
844 b.remove(ord('l'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
845 self.assertEqual(b, b'helo')
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
846 b.remove(ord('l'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
847 self.assertEqual(b, b'heo')
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
848 self.assertRaises(ValueError, lambda: b.remove(ord('l')))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
849 self.assertRaises(ValueError, lambda: b.remove(400))
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
850 self.assertRaises(TypeError, lambda: b.remove('e'))
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
851 # remove first and last
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
852 b.remove(ord('o'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
853 b.remove(ord('h'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
854 self.assertEqual(b, b'e')
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
855 self.assertRaises(TypeError, lambda: b.remove(b'e'))
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
856 b.remove(Indexable(ord('e')))
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
857 self.assertEqual(b, b'')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
858
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
859 def test_pop(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
860 b = bytearray(b'world')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
861 self.assertEqual(b.pop(), ord('d'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
862 self.assertEqual(b.pop(0), ord('w'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
863 self.assertEqual(b.pop(-2), ord('r'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
864 self.assertRaises(IndexError, lambda: b.pop(10))
68067
1a05bc36a334 Merged revisions 88735 via svnmerge from
Eli Bendersky <eliben@gmail.com>
parents: 67368
diff changeset
865 self.assertRaises(IndexError, lambda: bytearray().pop())
55848
1f4f342bfe2e Merged revisions 74675 via svnmerge from
Mark Dickinson <dickinsm@gmail.com>
parents: 55593
diff changeset
866 # test for issue #6846
1f4f342bfe2e Merged revisions 74675 via svnmerge from
Mark Dickinson <dickinsm@gmail.com>
parents: 55593
diff changeset
867 self.assertEqual(bytearray(b'\xff').pop(), 0xff)
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
868
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
869 def test_nosort(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
870 self.assertRaises(AttributeError, lambda: bytearray().sort())
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
871
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
872 def test_append(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
873 b = bytearray(b'hell')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
874 b.append(ord('o'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
875 self.assertEqual(b, b'hello')
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
876 self.assertEqual(b.append(100), None)
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
877 b = bytearray()
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
878 b.append(ord('A'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
879 self.assertEqual(len(b), 1)
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
880 self.assertRaises(TypeError, lambda: b.append(b'o'))
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
881 b = bytearray()
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
882 b.append(Indexable(ord('A')))
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
883 self.assertEqual(b, b'A')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
884
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
885 def test_insert(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
886 b = bytearray(b'msssspp')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
887 b.insert(1, ord('i'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
888 b.insert(4, ord('i'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
889 b.insert(-2, ord('i'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
890 b.insert(1000, ord('i'))
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
891 self.assertEqual(b, b'mississippi')
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
892 self.assertRaises(TypeError, lambda: b.insert(0, b'1'))
48508
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
893 b = bytearray()
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
894 b.insert(0, Indexable(ord('A')))
b9828b2a51c0 Merged revisions 65041 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47391
diff changeset
895 self.assertEqual(b, b'A')
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
896
49770
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
897 def test_copied(self):
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
898 # Issue 4348. Make sure that operations that don't mutate the array
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
899 # copy the bytes.
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
900 b = bytearray(b'abc')
49819
630c174b465e Merged revisions 67295,67301-67302,67318,67330,67342-67343 via svnmerge from
Amaury Forgeot d'Arc <amauryfa@gmail.com>
parents: 49770
diff changeset
901 self.assertFalse(b is b.replace(b'abc', b'cde', 0))
49770
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
902
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
903 t = bytearray([i for i in range(256)])
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
904 x = bytearray(b'')
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
905 self.assertFalse(x is x.translate(t))
fdc933951964 Merged revisions 67291 via svnmerge from
Benjamin Peterson <benjamin@python.org>
parents: 49574
diff changeset
906
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
907 def test_partition_bytearray_doesnt_share_nullstring(self):
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
908 a, b, c = bytearray(b"x").partition(b"y")
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
909 self.assertEqual(b, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
910 self.assertEqual(c, b"")
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
911 self.assertTrue(b is not c)
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
912 b += b"!"
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
913 self.assertEqual(c, b"")
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
914 a, b, c = bytearray(b"x").partition(b"y")
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
915 self.assertEqual(b, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
916 self.assertEqual(c, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
917 # Same for rpartition
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
918 b, c, a = bytearray(b"x").rpartition(b"y")
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
919 self.assertEqual(b, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
920 self.assertEqual(c, b"")
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
921 self.assertTrue(b is not c)
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
922 b += b"!"
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
923 self.assertEqual(c, b"")
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
924 c, b, a = bytearray(b"x").rpartition(b"y")
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
925 self.assertEqual(b, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
926 self.assertEqual(c, b"")
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
927
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
928 def test_resize_forbidden(self):
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
929 # #4509: can't resize a bytearray when there are buffer exports, even
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
930 # if it wouldn't reallocate the underlying buffer.
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
931 # Furthermore, no destructive changes to the buffer may be applied
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
932 # before raising the error.
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
933 b = bytearray(range(10))
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
934 v = memoryview(b)
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
935 def resize(n):
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
936 b[1:-1] = range(n + 1, 2*n - 1)
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
937 resize(10)
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
938 orig = b[:]
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
939 self.assertRaises(BufferError, resize, 11)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
940 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
941 self.assertRaises(BufferError, resize, 9)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
942 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
943 self.assertRaises(BufferError, resize, 0)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
944 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
945 # Other operations implying resize
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
946 self.assertRaises(BufferError, b.pop, 0)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
947 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
948 self.assertRaises(BufferError, b.remove, b[1])
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
949 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
950 def delitem():
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
951 del b[1]
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
952 self.assertRaises(BufferError, delitem)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
953 self.assertEqual(b, orig)
50056
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
954 # deleting a non-contiguous slice
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
955 def delslice():
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
956 b[1:-1:2] = b""
1693fd94d829 Merged revisions 67618 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 49819
diff changeset
957 self.assertRaises(BufferError, delslice)
66340
2beb25369289 Merged revisions 86596 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 60644
diff changeset
958 self.assertEqual(b, orig)
43777
0d462d789b18 Merging the py3k-pep3137 branch back into the py3k branch.
Guido van Rossum <guido@python.org>
parents: 43757
diff changeset
959
58389
fb337cb46b9d Merged revisions 77576 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 57026
diff changeset
960
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
961 class AssortedBytesTest(unittest.TestCase):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
962 #
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
963 # Test various combinations of bytes and bytearray
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
964 #
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
965
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
966 @check_bytes_warnings
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
967 def test_repr_str(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
968 for f in str, repr:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
969 self.assertEqual(f(bytearray()), "bytearray(b'')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
970 self.assertEqual(f(bytearray([0])), "bytearray(b'\\x00')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
971 self.assertEqual(f(bytearray([0, 1, 254, 255])),
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
972 "bytearray(b'\\x00\\x01\\xfe\\xff')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
973 self.assertEqual(f(b"abc"), "b'abc'")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
974 self.assertEqual(f(b"'"), '''b"'"''') # '''
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
975 self.assertEqual(f(b"'\""), r"""b'\'"'""") # '
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
976
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
977 def test_compare_bytes_to_bytearray(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
978 self.assertEqual(b"abc" == bytes(b"abc"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
979 self.assertEqual(b"ab" != bytes(b"abc"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
980 self.assertEqual(b"ab" <= bytes(b"abc"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
981 self.assertEqual(b"ab" < bytes(b"abc"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
982 self.assertEqual(b"abc" >= bytes(b"ab"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
983 self.assertEqual(b"abc" > bytes(b"ab"), True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
984
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
985 self.assertEqual(b"abc" != bytes(b"abc"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
986 self.assertEqual(b"ab" == bytes(b"abc"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
987 self.assertEqual(b"ab" > bytes(b"abc"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
988 self.assertEqual(b"ab" >= bytes(b"abc"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
989 self.assertEqual(b"abc" < bytes(b"ab"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
990 self.assertEqual(b"abc" <= bytes(b"ab"), False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
991
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
992 self.assertEqual(bytes(b"abc") == b"abc", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
993 self.assertEqual(bytes(b"ab") != b"abc", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
994 self.assertEqual(bytes(b"ab") <= b"abc", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
995 self.assertEqual(bytes(b"ab") < b"abc", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
996 self.assertEqual(bytes(b"abc") >= b"ab", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
997 self.assertEqual(bytes(b"abc") > b"ab", True)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
998
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
999 self.assertEqual(bytes(b"abc") != b"abc", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1000 self.assertEqual(bytes(b"ab") == b"abc", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1001 self.assertEqual(bytes(b"ab") > b"abc", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1002 self.assertEqual(bytes(b"ab") >= b"abc", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1003 self.assertEqual(bytes(b"abc") < b"ab", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1004 self.assertEqual(bytes(b"abc") <= b"ab", False)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1005
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1006 def test_doc(self):
59728
4587e1f74ed8 Merged revisions 79100 via svnmerge from
Florent Xicluna <florent.xicluna@gmail.com>
parents: 58563
diff changeset
1007 self.assertIsNotNone(bytearray.__doc__)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1008 self.assertTrue(bytearray.__doc__.startswith("bytearray("), bytearray.__doc__)
59728
4587e1f74ed8 Merged revisions 79100 via svnmerge from
Florent Xicluna <florent.xicluna@gmail.com>
parents: 58563
diff changeset
1009 self.assertIsNotNone(bytes.__doc__)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1010 self.assertTrue(bytes.__doc__.startswith("bytes("), bytes.__doc__)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1011
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1012 def test_from_bytearray(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1013 sample = bytes(b"Hello world\n\x80\x81\xfe\xff")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1014 buf = memoryview(sample)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1015 b = bytearray(buf)
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1016 self.assertEqual(b, bytearray(sample))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1017
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1018 @check_bytes_warnings
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1019 def test_to_str(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1020 self.assertEqual(str(b''), "b''")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1021 self.assertEqual(str(b'x'), "b'x'")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1022 self.assertEqual(str(b'\x80'), "b'\\x80'")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1023 self.assertEqual(str(bytearray(b'')), "bytearray(b'')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1024 self.assertEqual(str(bytearray(b'x')), "bytearray(b'x')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1025 self.assertEqual(str(bytearray(b'\x80')), "bytearray(b'\\x80')")
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1026
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1027 def test_literal(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1028 tests = [
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1029 (b"Wonderful spam", "Wonderful spam"),
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1030 (br"Wonderful spam too", "Wonderful spam too"),
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1031 (b"\xaa\x00\000\200", "\xaa\x00\000\200"),
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1032 (br"\xaa\x00\000\200", r"\xaa\x00\000\200"),
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1033 ]
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1034 for b, s in tests:
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1035 self.assertEqual(b, bytearray(s, 'latin-1'))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1036 for c in range(128, 256):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1037 self.assertRaises(SyntaxError, eval,
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1038 'b"%s"' % chr(c))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1039
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1040 def test_translate(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1041 b = b'hello'
50328
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1042 ba = bytearray(b)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1043 rosetta = bytearray(range(0, 256))
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1044 rosetta[ord('o')] = ord('e')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1045 c = b.translate(rosetta, b'l')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1046 self.assertEqual(b, b'hello')
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1047 self.assertEqual(c, b'hee')
50328
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1048 c = ba.translate(rosetta, b'l')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1049 self.assertEqual(ba, b'hello')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1050 self.assertEqual(c, b'hee')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1051 c = b.translate(None, b'e')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1052 self.assertEqual(c, b'hllo')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1053 c = ba.translate(None, b'e')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1054 self.assertEqual(c, b'hllo')
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1055 self.assertRaises(TypeError, b.translate, None, None)
7a011680e151 #4759: allow None as first argument of bytearray.translate(), for consistency with bytes.translate().
Georg Brandl <georg@python.org>
parents: 50056
diff changeset
1056 self.assertRaises(TypeError, ba.translate, None, None)
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1057
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1058 def test_split_bytearray(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1059 self.assertEqual(b'a b'.split(memoryview(b' ')), [b'a', b'b'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1060
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1061 def test_rsplit_bytearray(self):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1062 self.assertEqual(b'a b'.rsplit(memoryview(b' ')), [b'a', b'b'])
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1063
47391
eca882731194 Add test case for r63810.
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
1064 def test_return_self(self):
eca882731194 Add test case for r63810.
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
1065 # bytearray.replace must always return a new bytearray
eca882731194 Add test case for r63810.
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
1066 b = bytearray()
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1067 self.assertFalse(b.replace(b'', b'') is b)
47391
eca882731194 Add test case for r63810.
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
1068
49538
645fe56c117a STINNER Victor (haypo)'s patch for bug 3988, Byte warning mode and b'' != ''
Barry Warsaw <barry@python.org>
parents: 48871
diff changeset
1069 def test_compare(self):
645fe56c117a STINNER Victor (haypo)'s patch for bug 3988, Byte warning mode and b'' != ''
Barry Warsaw <barry@python.org>
parents: 48871
diff changeset
1070 if sys.flags.bytes_warning:
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1071 def bytes_warning():
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1072 return test.support.check_warnings(('', BytesWarning))
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1073 with bytes_warning():
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1074 b'' == ''
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1075 with bytes_warning():
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1076 b'' != ''
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1077 with bytes_warning():
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1078 bytearray(b'') == ''
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1079 with bytes_warning():
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1080 bytearray(b'') != ''
49538
645fe56c117a STINNER Victor (haypo)'s patch for bug 3988, Byte warning mode and b'' != ''
Barry Warsaw <barry@python.org>
parents: 48871
diff changeset
1081 else:
64408
3f7764c40c0b Strengthen BytesWarning tests.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 63913
diff changeset
1082 self.skipTest("BytesWarning is needed for this test: use -bb option")
49538
645fe56c117a STINNER Victor (haypo)'s patch for bug 3988, Byte warning mode and b'' != ''
Barry Warsaw <barry@python.org>
parents: 48871
diff changeset
1083
37960
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
1084 # Optimizations:
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
1085 # __iter__? (optimization)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
1086 # __reversed__? (optimization)
e0b7e34b5971 Added much functionality to the bytes type.
Guido van Rossum <guido@python.org>
parents: 37930
diff changeset
1087
41450
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
1088 # XXX More string methods? (Those that don't use character properties)
d1c3dd184e38 Rough and dirty job -- allow concatenation of bytes and arbitrary
Guido van Rossum <guido@python.org>
parents: 41425
diff changeset
1089
40978
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
1090 # There are tests in string_tests.py that are more
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
1091 # comprehensive for things like split, partition, etc.
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
1092 # Unfortunately they are all bundled with tests that
61fee1853ca8 SF patch #1669633, add methods for bytes from Pete Shinners.
Neal Norwitz <nnorwitz@gmail.com>
parents: 40967
diff changeset
1093 # are not appropriate for bytes
37930
e1252cea439c Fix a leak and a buglet discovered by Thomas.
Guido van Rossum <guido@python.org>
parents: 37928
diff changeset
1094
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1095 # I've started porting some of those into bytearray_tests.py, we should port
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1096 # the rest that make sense (the code can be cleaned up to use modern
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1097 # unittest methods at the same time).
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1098
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1099 class BytearrayPEP3137Test(unittest.TestCase,
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1100 test.buffer_tests.MixinBytesBufferCommonTests):
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1101 def marshal(self, x):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1102 return bytearray(x)
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1103
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1104 def test_returns_new_copy(self):
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1105 val = self.marshal(b'1234')
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1106 # On immutable types these MAY return a reference to themselves
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1107 # but on mutable types like bytearray they MUST return a new copy.
43576
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1108 for methname in ('zfill', 'rjust', 'ljust', 'center'):
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1109 method = getattr(val, methname)
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1110 newval = method(3)
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1111 self.assertEqual(val, newval)
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1112 self.assertTrue(val is not newval,
697cc31a83f6 For PEP3137: Adds missing methods to the mutable PyBytes object (soon
Gregory P. Smith <greg@mad-scientist.com>
parents: 43508
diff changeset
1113 methname+' returned self on a mutable object')
58303
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1114 for expr in ('val.split()[0]', 'val.rsplit()[0]',
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1115 'val.partition(b".")[0]', 'val.rpartition(b".")[2]',
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1116 'val.splitlines()[0]', 'val.replace(b"", b"")'):
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1117 newval = eval(expr)
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1118 self.assertEqual(val, newval)
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1119 self.assertTrue(val is not newval,
9fc7f297d132 Merged revisions 77475 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 58276
diff changeset
1120 expr+' returned val on a mutable object')
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
1121
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1122 class FixedStringTest(test.string_tests.BaseTest):
41038
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1123
42907
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
1124 def fixtype(self, obj):
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
1125 if isinstance(obj, str):
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
1126 return obj.encode("utf-8")
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
1127 return super().fixtype(obj)
0170418dadea Changes in anticipation of stricter str vs. bytes enforcement.
Guido van Rossum <guido@python.org>
parents: 42499
diff changeset
1128
41038
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1129 # Currently the bytes containment testing uses a single integer
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1130 # value. This may not be the final design, but until then the
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1131 # bytes section with in a bytes containment not valid
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1132 def test_contains(self):
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1133 pass
42084
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1134 def test_expandtabs(self):
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1135 pass
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1136 def test_upper(self):
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1137 pass
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1138 def test_lower(self):
229c28cb9afb Merged revisions 55817-55961 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41855
diff changeset
1139 pass
41038
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1140
45055
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1141 class ByteArrayAsStringTest(FixedStringTest):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1142 type2test = bytearray
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1143
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1144 class BytesAsStringTest(FixedStringTest):
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1145 type2test = bytes
510a48289d1d Patch #1972 by Antoine Pitrou: improve bytes and bytearray tests
Christian Heimes <christian@cheimes.de>
parents: 45050
diff changeset
1146
41038
efa0a761918c Addendum of patch #1669633: additional tests for bytes methods.
Georg Brandl <georg@python.org>
parents: 40978
diff changeset
1147
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1148 class SubclassTest(unittest.TestCase):
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1149
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1150 def test_basic(self):
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1151 self.assertTrue(issubclass(self.subclass2test, self.type2test))
58473
71acd4482e65 Merged revisions 77727 via svnmerge from
Ezio Melotti <ezio.melotti@gmail.com>
parents: 58456
diff changeset
1152 self.assertIsInstance(self.subclass2test(), self.type2test)
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1153
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1154 a, b = b"abcd", b"efgh"
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1155 _a, _b = self.subclass2test(a), self.subclass2test(b)
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1156
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1157 # test comparison operators with subclass instances
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1158 self.assertTrue(_a == _a)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1159 self.assertTrue(_a != _b)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1160 self.assertTrue(_a < _b)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1161 self.assertTrue(_a <= _b)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1162 self.assertTrue(_b >= _a)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1163 self.assertTrue(_b > _a)
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1164 self.assertTrue(_a is not a)
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1165
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1166 # test concat of subclass instances
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1167 self.assertEqual(a + b, _a + _b)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1168 self.assertEqual(a + b, a + _b)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1169 self.assertEqual(a + b, _a + b)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1170
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1171 # test repeat
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1172 self.assertTrue(a*5 == _a*5)
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1173
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1174 def test_join(self):
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1175 # Make sure join returns a NEW object for single item sequences
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1176 # involving a subclass.
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1177 # Make sure that it is of the appropriate type.
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1178 s1 = self.subclass2test(b"abcd")
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1179 s2 = self.type2test().join([s1])
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 53681
diff changeset
1180 self.assertTrue(s1 is not s2)
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1181 self.assertTrue(type(s2) is self.type2test, type(s2))
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1182
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1183 # Test reverse, calling join on subclass
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1184 s3 = s1.join([b"abcd"])
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1185 self.assertTrue(type(s3) is self.type2test)
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1186
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1187 def test_pickle(self):
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1188 a = self.subclass2test(b"abcd")
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1189 a.x = 10
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1190 a.y = self.subclass2test(b"efgh")
49574
bbab11cdb175 Issue #4183: Some tests didn't run with pickle.HIGHEST_PROTOCOL.
Hirokazu Yamamoto <ocean-city@m2.ccsnet.ne.jp>
parents: 49538
diff changeset
1191 for proto in range(pickle.HIGHEST_PROTOCOL + 1):
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1192 b = pickle.loads(pickle.dumps(a, proto))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1193 self.assertNotEqual(id(a), id(b))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1194 self.assertEqual(a, b)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1195 self.assertEqual(a.x, b.x)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1196 self.assertEqual(a.y, b.y)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1197 self.assertEqual(type(a), type(b))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1198 self.assertEqual(type(a.y), type(b.y))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1199
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1200 def test_copy(self):
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1201 a = self.subclass2test(b"abcd")
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1202 a.x = 10
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1203 a.y = self.subclass2test(b"efgh")
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1204 for copy_method in (copy.copy, copy.deepcopy):
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1205 b = copy_method(a)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1206 self.assertNotEqual(id(a), id(b))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1207 self.assertEqual(a, b)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1208 self.assertEqual(a.x, b.x)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1209 self.assertEqual(a.y, b.y)
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1210 self.assertEqual(type(a), type(b))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1211 self.assertEqual(type(a.y), type(b.y))
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1212
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1213
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1214 class ByteArraySubclass(bytearray):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1215 pass
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1216
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1217 class BytesSubclass(bytes):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1218 pass
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1219
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1220 class ByteArraySubclassTest(SubclassTest):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1221 type2test = bytearray
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1222 subclass2test = ByteArraySubclass
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1223
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1224 def test_init_override(self):
43947
cbc8b3bd3157 Rename buffer -> bytearray.
Guido van Rossum <guido@python.org>
parents: 43777
diff changeset
1225 class subclass(bytearray):
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1226 def __init__(me, newarg=1, *args, **kwargs):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1227 bytearray.__init__(me, *args, **kwargs)
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1228 x = subclass(4, b"abcd")
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1229 x = subclass(4, source=b"abcd")
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1230 self.assertEqual(x, b"abcd")
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1231 x = subclass(newarg=4, source=b"abcd")
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1232 self.assertEqual(x, b"abcd")
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1233
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1234
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1235 class BytesSubclassTest(SubclassTest):
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1236 type2test = bytes
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1237 subclass2test = BytesSubclass
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1238
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1239
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
1240 def test_main():
56515
67bb17c2e868 Merged revisions 75404 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 55848
diff changeset
1241 test.support.run_unittest(
67bb17c2e868 Merged revisions 75404 via svnmerge from
Antoine Pitrou <solipsis@pitrou.net>
parents: 55848
diff changeset
1242 BytesTest, AssortedBytesTest, BytesAsStringTest,
58264
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1243 ByteArrayTest, ByteArrayAsStringTest, BytesSubclassTest,
efa5867d8a14 Issue #7382: Fix bytes.__getnewargs__.
Alexandre Vassalotti <alexandre@peadrop.com>
parents: 57025
diff changeset
1244 ByteArraySubclassTest, BytearrayPEP3137Test)
37927
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
1245
a7f3765ea464 Here is a bytes type. It's very minimal but it's a start.
Guido van Rossum <guido@python.org>
parents:
diff changeset
1246 if __name__ == "__main__":
43757
2ff0e81e68fc Patch 1171 by mfenniak -- allow subclassing of bytes.
Guido van Rossum <guido@python.org>
parents: 43693
diff changeset
1247 test_main()