blob: b7246ebf39c694ba174bb7ffc29ef986ecee2615 [file] [log] [blame]
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001"""
2Common tests shared by test_str, test_unicode, test_userstring and test_string.
3"""
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00004
Guido van Rossum360e4b82007-05-14 22:51:27 +00005import unittest, string, sys, struct
Benjamin Petersonee8712c2008-05-20 21:35:26 +00006from test import support
Raymond Hettinger53dbe392008-02-12 20:03:09 +00007from collections import UserList
Jeremy Hylton20f41b62000-07-11 03:31:55 +00008
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00009class Sequence:
Walter Dörwald0fd583c2003-02-21 12:53:50 +000010 def __init__(self, seq='wxyz'): self.seq = seq
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000011 def __len__(self): return len(self.seq)
12 def __getitem__(self, i): return self.seq[i]
13
14class BadSeq1(Sequence):
Guido van Rossume2a383d2007-01-15 16:59:06 +000015 def __init__(self): self.seq = [7, 'hello', 123]
Guido van Rossumf1044292007-09-27 18:01:22 +000016 def __str__(self): return '{0} {1} {2}'.format(*self.seq)
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000017
18class BadSeq2(Sequence):
19 def __init__(self): self.seq = ['a', 'b', 'c']
20 def __len__(self): return 8
21
Georg Brandlc7885542007-03-06 19:16:20 +000022class BaseTest(unittest.TestCase):
23 # These tests are for buffers of values (bytes) and not
24 # specific to character interpretation, used for bytes objects
25 # and various string implementations
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000026
Walter Dörwald0fd583c2003-02-21 12:53:50 +000027 # The type to be tested
28 # Change in subclasses to change the behaviour of fixtesttype()
29 type2test = None
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000030
Antoine Pitrouac65d962011-10-20 23:54:17 +020031 # Whether the "contained items" of the container are integers in
32 # range(0, 256) (i.e. bytes, bytearray) or strings of length 1
33 # (str)
34 contains_bytes = False
35
Walter Dörwald0fd583c2003-02-21 12:53:50 +000036 # All tests pass their arguments to the testing methods
37 # as str objects. fixtesttype() can be used to propagate
38 # these arguments to the appropriate type
39 def fixtype(self, obj):
40 if isinstance(obj, str):
41 return self.__class__.type2test(obj)
42 elif isinstance(obj, list):
43 return [self.fixtype(x) for x in obj]
44 elif isinstance(obj, tuple):
45 return tuple([self.fixtype(x) for x in obj])
46 elif isinstance(obj, dict):
47 return dict([
48 (self.fixtype(key), self.fixtype(value))
Guido van Rossumcc2b0162007-02-11 06:12:03 +000049 for (key, value) in obj.items()
Walter Dörwald0fd583c2003-02-21 12:53:50 +000050 ])
51 else:
52 return obj
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000053
Guido van Rossum09549f42007-08-27 20:40:10 +000054 # check that obj.method(*args) returns result
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010055 def checkequal(self, result, obj, methodname, *args, **kwargs):
Walter Dörwald0fd583c2003-02-21 12:53:50 +000056 result = self.fixtype(result)
Guido van Rossum09549f42007-08-27 20:40:10 +000057 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000058 args = self.fixtype(args)
Ezio Melotticda6b6d2012-02-26 09:39:55 +020059 kwargs = {k: self.fixtype(v) for k,v in kwargs.items()}
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010060 realresult = getattr(obj, methodname)(*args, **kwargs)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000061 self.assertEqual(
62 result,
63 realresult
64 )
65 # if the original is returned make sure that
66 # this doesn't happen with subclasses
Guido van Rossum09549f42007-08-27 20:40:10 +000067 if obj is realresult:
68 try:
69 class subtype(self.__class__.type2test):
70 pass
71 except TypeError:
72 pass # Skip this if we can't subclass
73 else:
74 obj = subtype(obj)
75 realresult = getattr(obj, methodname)(*args)
Ezio Melottib3aedd42010-11-20 19:04:17 +000076 self.assertIsNot(obj, realresult)
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000077
Guido van Rossum09549f42007-08-27 20:40:10 +000078 # check that obj.method(*args) raises exc
79 def checkraises(self, exc, obj, methodname, *args):
80 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000081 args = self.fixtype(args)
82 self.assertRaises(
83 exc,
Guido van Rossum09549f42007-08-27 20:40:10 +000084 getattr(obj, methodname),
Walter Dörwald0fd583c2003-02-21 12:53:50 +000085 *args
86 )
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000087
Guido van Rossum09549f42007-08-27 20:40:10 +000088 # call obj.method(*args) without any checks
89 def checkcall(self, obj, methodname, *args):
90 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000091 args = self.fixtype(args)
Guido van Rossum09549f42007-08-27 20:40:10 +000092 getattr(obj, methodname)(*args)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000093
Walter Dörwald0fd583c2003-02-21 12:53:50 +000094 def test_count(self):
95 self.checkequal(3, 'aaa', 'count', 'a')
96 self.checkequal(0, 'aaa', 'count', 'b')
97 self.checkequal(3, 'aaa', 'count', 'a')
98 self.checkequal(0, 'aaa', 'count', 'b')
99 self.checkequal(3, 'aaa', 'count', 'a')
100 self.checkequal(0, 'aaa', 'count', 'b')
101 self.checkequal(0, 'aaa', 'count', 'b')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000102 self.checkequal(2, 'aaa', 'count', 'a', 1)
103 self.checkequal(0, 'aaa', 'count', 'a', 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000104 self.checkequal(1, 'aaa', 'count', 'a', -1)
105 self.checkequal(3, 'aaa', 'count', 'a', -10)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000106 self.checkequal(1, 'aaa', 'count', 'a', 0, 1)
107 self.checkequal(3, 'aaa', 'count', 'a', 0, 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000108 self.checkequal(2, 'aaa', 'count', 'a', 0, -1)
109 self.checkequal(0, 'aaa', 'count', 'a', 0, -10)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000110 self.checkequal(3, 'aaa', 'count', '', 1)
111 self.checkequal(1, 'aaa', 'count', '', 3)
112 self.checkequal(0, 'aaa', 'count', '', 10)
113 self.checkequal(2, 'aaa', 'count', '', -1)
114 self.checkequal(4, 'aaa', 'count', '', -10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000115
Amaury Forgeot d'Arcf2e93682008-09-26 22:48:41 +0000116 self.checkequal(1, '', 'count', '')
117 self.checkequal(0, '', 'count', '', 1, 1)
118 self.checkequal(0, '', 'count', '', sys.maxsize, 0)
119
120 self.checkequal(0, '', 'count', 'xx')
121 self.checkequal(0, '', 'count', 'xx', 1, 1)
122 self.checkequal(0, '', 'count', 'xx', sys.maxsize, 0)
123
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000124 self.checkraises(TypeError, 'hello', 'count')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200125
126 if self.contains_bytes:
127 self.checkequal(0, 'hello', 'count', 42)
128 else:
129 self.checkraises(TypeError, 'hello', 'count', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000130
Raymond Hettinger57e74472005-02-20 09:54:53 +0000131 # For a variety of combinations,
132 # verify that str.count() matches an equivalent function
133 # replacing all occurrences and then differencing the string lengths
134 charset = ['', 'a', 'b']
135 digits = 7
136 base = len(charset)
137 teststrings = set()
Guido van Rossum805365e2007-05-07 22:24:25 +0000138 for i in range(base ** digits):
Raymond Hettinger57e74472005-02-20 09:54:53 +0000139 entry = []
Guido van Rossum805365e2007-05-07 22:24:25 +0000140 for j in range(digits):
Raymond Hettinger57e74472005-02-20 09:54:53 +0000141 i, m = divmod(i, base)
142 entry.append(charset[m])
143 teststrings.add(''.join(entry))
Guido van Rossum09549f42007-08-27 20:40:10 +0000144 teststrings = [self.fixtype(ts) for ts in teststrings]
Raymond Hettinger57e74472005-02-20 09:54:53 +0000145 for i in teststrings:
Raymond Hettinger57e74472005-02-20 09:54:53 +0000146 n = len(i)
147 for j in teststrings:
148 r1 = i.count(j)
149 if j:
Guido van Rossum09549f42007-08-27 20:40:10 +0000150 r2, rem = divmod(n - len(i.replace(j, self.fixtype(''))),
151 len(j))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000152 else:
153 r2, rem = len(i)+1, 0
154 if rem or r1 != r2:
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000155 self.assertEqual(rem, 0, '%s != 0 for %s' % (rem, i))
156 self.assertEqual(r1, r2, '%s != %s for %s' % (r1, r2, i))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000157
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000158 def test_find(self):
159 self.checkequal(0, 'abcdefghiabc', 'find', 'abc')
160 self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
161 self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
162
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000163 self.checkequal(0, 'abc', 'find', '', 0)
164 self.checkequal(3, 'abc', 'find', '', 3)
165 self.checkequal(-1, 'abc', 'find', '', 4)
166
Christian Heimes9cd17752007-11-18 19:35:23 +0000167 # to check the ability to pass None as defaults
168 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a')
169 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4)
170 self.checkequal(-1, 'rrarrrrrrrrra', 'find', 'a', 4, 6)
171 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4, None)
172 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a', None, 6)
173
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000174 self.checkraises(TypeError, 'hello', 'find')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200175
176 if self.contains_bytes:
177 self.checkequal(-1, 'hello', 'find', 42)
178 else:
179 self.checkraises(TypeError, 'hello', 'find', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000180
Amaury Forgeot d'Arcf2e93682008-09-26 22:48:41 +0000181 self.checkequal(0, '', 'find', '')
182 self.checkequal(-1, '', 'find', '', 1, 1)
183 self.checkequal(-1, '', 'find', '', sys.maxsize, 0)
184
185 self.checkequal(-1, '', 'find', 'xx')
186 self.checkequal(-1, '', 'find', 'xx', 1, 1)
187 self.checkequal(-1, '', 'find', 'xx', sys.maxsize, 0)
188
Antoine Pitrou74edda02010-01-02 21:51:33 +0000189 # issue 7458
190 self.checkequal(-1, 'ab', 'find', 'xxx', sys.maxsize + 1, 0)
191
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000192 # For a variety of combinations,
193 # verify that str.find() matches __contains__
194 # and that the found substring is really at that location
195 charset = ['', 'a', 'b', 'c']
196 digits = 5
197 base = len(charset)
198 teststrings = set()
Guido van Rossum805365e2007-05-07 22:24:25 +0000199 for i in range(base ** digits):
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000200 entry = []
Guido van Rossum805365e2007-05-07 22:24:25 +0000201 for j in range(digits):
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000202 i, m = divmod(i, base)
203 entry.append(charset[m])
204 teststrings.add(''.join(entry))
Guido van Rossum09549f42007-08-27 20:40:10 +0000205 teststrings = [self.fixtype(ts) for ts in teststrings]
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000206 for i in teststrings:
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000207 for j in teststrings:
208 loc = i.find(j)
209 r1 = (loc != -1)
210 r2 = j in i
Antoine Pitrou2e544fb2010-01-02 21:55:17 +0000211 self.assertEqual(r1, r2)
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000212 if loc != -1:
213 self.assertEqual(i[loc:loc+len(j)], j)
214
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000215 def test_rfind(self):
216 self.checkequal(9, 'abcdefghiabc', 'rfind', 'abc')
217 self.checkequal(12, 'abcdefghiabc', 'rfind', '')
218 self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
219 self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
220
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000221 self.checkequal(3, 'abc', 'rfind', '', 0)
222 self.checkequal(3, 'abc', 'rfind', '', 3)
223 self.checkequal(-1, 'abc', 'rfind', '', 4)
224
Christian Heimes9cd17752007-11-18 19:35:23 +0000225 # to check the ability to pass None as defaults
226 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a')
227 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4)
228 self.checkequal(-1, 'rrarrrrrrrrra', 'rfind', 'a', 4, 6)
229 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4, None)
230 self.checkequal( 2, 'rrarrrrrrrrra', 'rfind', 'a', None, 6)
231
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000232 self.checkraises(TypeError, 'hello', 'rfind')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200233
234 if self.contains_bytes:
235 self.checkequal(-1, 'hello', 'rfind', 42)
236 else:
237 self.checkraises(TypeError, 'hello', 'rfind', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000238
Antoine Pitrouda2ecaf2010-01-02 21:40:36 +0000239 # For a variety of combinations,
240 # verify that str.rfind() matches __contains__
241 # and that the found substring is really at that location
242 charset = ['', 'a', 'b', 'c']
243 digits = 5
244 base = len(charset)
245 teststrings = set()
246 for i in range(base ** digits):
247 entry = []
248 for j in range(digits):
249 i, m = divmod(i, base)
250 entry.append(charset[m])
251 teststrings.add(''.join(entry))
252 teststrings = [self.fixtype(ts) for ts in teststrings]
253 for i in teststrings:
254 for j in teststrings:
255 loc = i.rfind(j)
256 r1 = (loc != -1)
257 r2 = j in i
Antoine Pitrou2e544fb2010-01-02 21:55:17 +0000258 self.assertEqual(r1, r2)
Antoine Pitrouda2ecaf2010-01-02 21:40:36 +0000259 if loc != -1:
260 self.assertEqual(i[loc:loc+len(j)], j)
261
Antoine Pitrou74edda02010-01-02 21:51:33 +0000262 # issue 7458
263 self.checkequal(-1, 'ab', 'rfind', 'xxx', sys.maxsize + 1, 0)
264
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000265 def test_index(self):
266 self.checkequal(0, 'abcdefghiabc', 'index', '')
267 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
268 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
269 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
270
271 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
272 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
273 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
274 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
275
Christian Heimes9cd17752007-11-18 19:35:23 +0000276 # to check the ability to pass None as defaults
277 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a')
278 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4)
279 self.checkraises(ValueError, 'rrarrrrrrrrra', 'index', 'a', 4, 6)
280 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4, None)
281 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a', None, 6)
282
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000283 self.checkraises(TypeError, 'hello', 'index')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200284
285 if self.contains_bytes:
286 self.checkraises(ValueError, 'hello', 'index', 42)
287 else:
288 self.checkraises(TypeError, 'hello', 'index', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000289
290 def test_rindex(self):
291 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
292 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
293 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
294 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
295
296 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
297 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
298 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
299 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
300 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
301
Christian Heimes9cd17752007-11-18 19:35:23 +0000302 # to check the ability to pass None as defaults
303 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a')
304 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4)
305 self.checkraises(ValueError, 'rrarrrrrrrrra', 'rindex', 'a', 4, 6)
306 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4, None)
307 self.checkequal( 2, 'rrarrrrrrrrra', 'rindex', 'a', None, 6)
308
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000309 self.checkraises(TypeError, 'hello', 'rindex')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200310
311 if self.contains_bytes:
312 self.checkraises(ValueError, 'hello', 'rindex', 42)
313 else:
314 self.checkraises(TypeError, 'hello', 'rindex', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000315
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000316 def test_lower(self):
317 self.checkequal('hello', 'HeLLo', 'lower')
318 self.checkequal('hello', 'hello', 'lower')
319 self.checkraises(TypeError, 'hello', 'lower', 42)
320
321 def test_upper(self):
322 self.checkequal('HELLO', 'HeLLo', 'upper')
323 self.checkequal('HELLO', 'HELLO', 'upper')
324 self.checkraises(TypeError, 'hello', 'upper', 42)
325
326 def test_expandtabs(self):
327 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
328 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
329 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 4)
330 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi', 'expandtabs', 4)
331 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
332 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
333 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi', 'expandtabs', 4)
334 self.checkequal(' a\n b', ' \ta\n\tb', 'expandtabs', 1)
335
336 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
337 # This test is only valid when sizeof(int) == sizeof(void*) == 4.
Christian Heimesa37d4c62007-12-04 23:02:19 +0000338 if sys.maxsize < (1 << 32) and struct.calcsize('P') == 4:
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000339 self.checkraises(OverflowError,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000340 '\ta\n\tb', 'expandtabs', sys.maxsize)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000341
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000342 def test_split(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000343 # by a char
344 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000345 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000346 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
347 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
348 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
349 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000350 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000351 sys.maxsize-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000352 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
353 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
354 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000355 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
356 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000357 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
358
Thomas Wouters477c8d52006-05-27 19:21:47 +0000359 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
360 self.checkequal(['a']*15 +['a|a|a|a|a'],
361 ('a|'*20)[:-1], 'split', '|', 15)
362
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000363 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000364 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000365 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
366 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
367 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
368 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000369 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000370 sys.maxsize-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000371 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
372 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000373 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000374 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
375 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
376 'split', 'test')
377 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
378 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
379 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
380 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
381 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
382 self.checkequal([''], '', 'split', 'aaa')
383 self.checkequal(['aa'], 'aa', 'split', 'aaa')
384 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
385 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
386
387 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
388 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
389 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
390 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000391
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200392 # with keyword args
393 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', sep='|')
394 self.checkequal(['a', 'b|c|d'],
395 'a|b|c|d', 'split', '|', maxsplit=1)
396 self.checkequal(['a', 'b|c|d'],
397 'a|b|c|d', 'split', sep='|', maxsplit=1)
398 self.checkequal(['a', 'b|c|d'],
399 'a|b|c|d', 'split', maxsplit=1, sep='|')
400 self.checkequal(['a', 'b c d'],
401 'a b c d', 'split', maxsplit=1)
402
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000403 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000404 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
405
Thomas Wouters477c8d52006-05-27 19:21:47 +0000406 # null case
407 self.checkraises(ValueError, 'hello', 'split', '')
408 self.checkraises(ValueError, 'hello', 'split', '', 0)
409
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000410 def test_rsplit(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000411 # by a char
412 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
413 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
414 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
415 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
416 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000417 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000418 sys.maxsize-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000419 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
420 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
421 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000422 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
423 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
424
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000425 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
426
Thomas Wouters477c8d52006-05-27 19:21:47 +0000427 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
428 self.checkequal(['a|a|a|a|a']+['a']*15,
429 ('a|'*20)[:-1], 'rsplit', '|', 15)
430
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000431 # by string
432 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
433 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
434 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
435 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
436 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000437 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000438 sys.maxsize-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000439 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
440 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
441 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000442 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
443 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
444 'rsplit', 'test')
445 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
446 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
447 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
448 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
449 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
450 self.checkequal([''], '', 'rsplit', 'aaa')
451 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
452 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
453 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
454
455 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
456 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
457 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
458 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000459
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200460 # with keyword args
461 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', sep='|')
462 self.checkequal(['a|b|c', 'd'],
463 'a|b|c|d', 'rsplit', '|', maxsplit=1)
464 self.checkequal(['a|b|c', 'd'],
465 'a|b|c|d', 'rsplit', sep='|', maxsplit=1)
466 self.checkequal(['a|b|c', 'd'],
467 'a|b|c|d', 'rsplit', maxsplit=1, sep='|')
468 self.checkequal(['a b c', 'd'],
469 'a b c d', 'rsplit', maxsplit=1)
470
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000471 # argument type
472 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000473
Thomas Wouters477c8d52006-05-27 19:21:47 +0000474 # null case
475 self.checkraises(ValueError, 'hello', 'rsplit', '')
476 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
477
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000478 def test_replace(self):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000479 EQ = self.checkequal
480
481 # Operations on the empty string
482 EQ("", "", "replace", "", "")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000483 EQ("A", "", "replace", "", "A")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000484 EQ("", "", "replace", "A", "")
485 EQ("", "", "replace", "A", "A")
486 EQ("", "", "replace", "", "", 100)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000487 EQ("", "", "replace", "", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000488
489 # interleave (from=="", 'to' gets inserted everywhere)
490 EQ("A", "A", "replace", "", "")
491 EQ("*A*", "A", "replace", "", "*")
492 EQ("*1A*1", "A", "replace", "", "*1")
493 EQ("*-#A*-#", "A", "replace", "", "*-#")
494 EQ("*-A*-A*-", "AA", "replace", "", "*-")
495 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000496 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000497 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
498 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
499 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
500 EQ("*-AA", "AA", "replace", "", "*-", 1)
501 EQ("AA", "AA", "replace", "", "*-", 0)
502
503 # single character deletion (from=="A", to=="")
504 EQ("", "A", "replace", "A", "")
505 EQ("", "AAA", "replace", "A", "")
506 EQ("", "AAA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000507 EQ("", "AAA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000508 EQ("", "AAA", "replace", "A", "", 4)
509 EQ("", "AAA", "replace", "A", "", 3)
510 EQ("A", "AAA", "replace", "A", "", 2)
511 EQ("AA", "AAA", "replace", "A", "", 1)
512 EQ("AAA", "AAA", "replace", "A", "", 0)
513 EQ("", "AAAAAAAAAA", "replace", "A", "")
514 EQ("BCD", "ABACADA", "replace", "A", "")
515 EQ("BCD", "ABACADA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000516 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000517 EQ("BCD", "ABACADA", "replace", "A", "", 5)
518 EQ("BCD", "ABACADA", "replace", "A", "", 4)
519 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
520 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
521 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
522 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
523 EQ("BCD", "ABCAD", "replace", "A", "")
524 EQ("BCD", "ABCADAA", "replace", "A", "")
525 EQ("BCD", "BCD", "replace", "A", "")
526 EQ("*************", "*************", "replace", "A", "")
527 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
528
529 # substring deletion (from=="the", to=="")
530 EQ("", "the", "replace", "the", "")
531 EQ("ater", "theater", "replace", "the", "")
532 EQ("", "thethe", "replace", "the", "")
533 EQ("", "thethethethe", "replace", "the", "")
534 EQ("aaaa", "theatheatheathea", "replace", "the", "")
535 EQ("that", "that", "replace", "the", "")
536 EQ("thaet", "thaet", "replace", "the", "")
537 EQ("here and re", "here and there", "replace", "the", "")
538 EQ("here and re and re", "here and there and there",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000539 "replace", "the", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000540 EQ("here and re and re", "here and there and there",
541 "replace", "the", "", -1)
542 EQ("here and re and re", "here and there and there",
543 "replace", "the", "", 3)
544 EQ("here and re and re", "here and there and there",
545 "replace", "the", "", 2)
546 EQ("here and re and there", "here and there and there",
547 "replace", "the", "", 1)
548 EQ("here and there and there", "here and there and there",
549 "replace", "the", "", 0)
550 EQ("here and re and re", "here and there and there", "replace", "the", "")
551
552 EQ("abc", "abc", "replace", "the", "")
553 EQ("abcdefg", "abcdefg", "replace", "the", "")
554
555 # substring deletion (from=="bob", to=="")
556 EQ("bob", "bbobob", "replace", "bob", "")
557 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
558 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
559 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
560
561 # single character replace in place (len(from)==len(to)==1)
562 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
563 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000564 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000565 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
566 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
567 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
568 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
569 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
570
571 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
572 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
573 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
574 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
575 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
576
577 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
578
579 # substring replace in place (len(from)==len(to) > 1)
580 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000581 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000582 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
583 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
584 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
585 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
586 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
587 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
588 EQ("cobob", "bobob", "replace", "bob", "cob")
589 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
590 EQ("bobob", "bobob", "replace", "bot", "bot")
591
592 # replace single character (len(from)==1, len(to)>1)
593 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
594 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000595 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000596 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
597 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
598 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
599 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
600
601 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
602
603 # replace substring (len(from)>1, len(to)!=len(from))
604 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
605 "replace", "spam", "ham")
606 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000607 "replace", "spam", "ham", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000608 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
609 "replace", "spam", "ham", -1)
610 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
611 "replace", "spam", "ham", 4)
612 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
613 "replace", "spam", "ham", 3)
614 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
615 "replace", "spam", "ham", 2)
616 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
617 "replace", "spam", "ham", 1)
618 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
619 "replace", "spam", "ham", 0)
620
621 EQ("bobob", "bobobob", "replace", "bobob", "bob")
622 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
623 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
624
Guido van Rossum39478e82007-08-27 17:23:59 +0000625 # XXX Commented out. Is there any reason to support buffer objects
626 # as arguments for str.replace()? GvR
Guido van Rossum254348e2007-11-21 19:29:53 +0000627## ba = bytearray('a')
628## bb = bytearray('b')
Guido van Rossum39478e82007-08-27 17:23:59 +0000629## EQ("bbc", "abc", "replace", ba, bb)
630## EQ("aac", "abc", "replace", bb, ba)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000631
Thomas Wouters477c8d52006-05-27 19:21:47 +0000632 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000633 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
634 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
635 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
636 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
637 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
638 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
639 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
640 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
641 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
642 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
643 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
644 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
645 self.checkequal('', '', 'replace', '', '')
646 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
647 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
648 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
649 # MemoryError due to empty result (platform malloc issue when requesting
650 # 0 bytes).
651 self.checkequal('', '123', 'replace', '123', '')
652 self.checkequal('', '123123', 'replace', '123', '')
653 self.checkequal('x', '123x123', 'replace', '123', '')
654
655 self.checkraises(TypeError, 'hello', 'replace')
656 self.checkraises(TypeError, 'hello', 'replace', 42)
657 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
658 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
659
Thomas Wouters477c8d52006-05-27 19:21:47 +0000660 def test_replace_overflow(self):
661 # Check for overflow checking on 32 bit machines
Christian Heimesa37d4c62007-12-04 23:02:19 +0000662 if sys.maxsize != 2147483647 or struct.calcsize("P") > 4:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000663 return
664 A2_16 = "A" * (2**16)
665 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
666 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
667 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
668
Georg Brandlc7885542007-03-06 19:16:20 +0000669
670
671class CommonTest(BaseTest):
672 # This testcase contains test that can be used in all
673 # stringlike classes. Currently this is str, unicode
674 # UserString and the string module.
675
676 def test_hash(self):
677 # SF bug 1054139: += optimization was not invalidating cached hash value
678 a = self.type2test('DNSSEC')
679 b = self.type2test('')
680 for c in a:
681 b += c
682 hash(b)
683 self.assertEqual(hash(a), hash(b))
684
685 def test_capitalize(self):
686 self.checkequal(' hello ', ' hello ', 'capitalize')
687 self.checkequal('Hello ', 'Hello ','capitalize')
688 self.checkequal('Hello ', 'hello ','capitalize')
689 self.checkequal('Aaaa', 'aaaa', 'capitalize')
690 self.checkequal('Aaaa', 'AaAa', 'capitalize')
691
Ezio Melottiee8d9982011-08-15 09:09:57 +0300692 # check that titlecased chars are lowered correctly
693 # \u1ffc is the titlecased char
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -0500694 self.checkequal('\u03a9\u0399\u1ff3\u1ff3\u1ff3',
Ezio Melottiee8d9982011-08-15 09:09:57 +0300695 '\u1ff3\u1ff3\u1ffc\u1ffc', 'capitalize')
696 # check with cased non-letter chars
697 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
698 '\u24c5\u24ce\u24c9\u24bd\u24c4\u24c3', 'capitalize')
699 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
700 '\u24df\u24e8\u24e3\u24d7\u24de\u24dd', 'capitalize')
701 self.checkequal('\u2160\u2171\u2172',
702 '\u2160\u2161\u2162', 'capitalize')
703 self.checkequal('\u2160\u2171\u2172',
704 '\u2170\u2171\u2172', 'capitalize')
705 # check with Ll chars with no upper - nothing changes here
706 self.checkequal('\u019b\u1d00\u1d86\u0221\u1fb7',
707 '\u019b\u1d00\u1d86\u0221\u1fb7', 'capitalize')
708
Georg Brandlc7885542007-03-06 19:16:20 +0000709 self.checkraises(TypeError, 'hello', 'capitalize', 42)
710
711 def test_lower(self):
712 self.checkequal('hello', 'HeLLo', 'lower')
713 self.checkequal('hello', 'hello', 'lower')
714 self.checkraises(TypeError, 'hello', 'lower', 42)
715
716 def test_upper(self):
717 self.checkequal('HELLO', 'HeLLo', 'upper')
718 self.checkequal('HELLO', 'HELLO', 'upper')
719 self.checkraises(TypeError, 'hello', 'upper', 42)
720
721 def test_expandtabs(self):
722 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
723 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
724 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 4)
725 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi', 'expandtabs', 4)
726 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
727 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
728 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi', 'expandtabs', 4)
729
730 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
731
732 def test_additional_split(self):
733 self.checkequal(['this', 'is', 'the', 'split', 'function'],
734 'this is the split function', 'split')
735
736 # by whitespace
737 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
738 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
739 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
740 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
741 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
742 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000743 sys.maxsize-1)
Georg Brandlc7885542007-03-06 19:16:20 +0000744 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
745 self.checkequal(['a b c d'], ' a b c d', 'split', None, 0)
746 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
747
748 self.checkequal([], ' ', 'split')
749 self.checkequal(['a'], ' a ', 'split')
750 self.checkequal(['a', 'b'], ' a b ', 'split')
751 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
752 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
753 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
754 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
755 aaa = ' a '*20
756 self.checkequal(['a']*20, aaa, 'split')
757 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
758 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
759
760 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000761 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000762
763 def test_additional_rsplit(self):
764 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
765 'this is the rsplit function', 'rsplit')
766
767 # by whitespace
768 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
769 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
770 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
771 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
772 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
773 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000774 sys.maxsize-20)
Georg Brandlc7885542007-03-06 19:16:20 +0000775 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
776 self.checkequal(['a b c d'], 'a b c d ', 'rsplit', None, 0)
777 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
778
779 self.checkequal([], ' ', 'rsplit')
780 self.checkequal(['a'], ' a ', 'rsplit')
781 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
782 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
783 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
784 None, 1)
785 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
786 None, 2)
787 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
788 aaa = ' a '*20
789 self.checkequal(['a']*20, aaa, 'rsplit')
790 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
791 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
792
793 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000794 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000795
796 def test_strip(self):
797 self.checkequal('hello', ' hello ', 'strip')
798 self.checkequal('hello ', ' hello ', 'lstrip')
799 self.checkequal(' hello', ' hello ', 'rstrip')
800 self.checkequal('hello', 'hello', 'strip')
801
802 # strip/lstrip/rstrip with None arg
803 self.checkequal('hello', ' hello ', 'strip', None)
804 self.checkequal('hello ', ' hello ', 'lstrip', None)
805 self.checkequal(' hello', ' hello ', 'rstrip', None)
806 self.checkequal('hello', 'hello', 'strip', None)
807
808 # strip/lstrip/rstrip with str arg
809 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
810 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
811 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
812 self.checkequal('hello', 'hello', 'strip', 'xyz')
813
Georg Brandlc7885542007-03-06 19:16:20 +0000814 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
815 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
816 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
817
818 def test_ljust(self):
819 self.checkequal('abc ', 'abc', 'ljust', 10)
820 self.checkequal('abc ', 'abc', 'ljust', 6)
821 self.checkequal('abc', 'abc', 'ljust', 3)
822 self.checkequal('abc', 'abc', 'ljust', 2)
823 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
824 self.checkraises(TypeError, 'abc', 'ljust')
825
826 def test_rjust(self):
827 self.checkequal(' abc', 'abc', 'rjust', 10)
828 self.checkequal(' abc', 'abc', 'rjust', 6)
829 self.checkequal('abc', 'abc', 'rjust', 3)
830 self.checkequal('abc', 'abc', 'rjust', 2)
831 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
832 self.checkraises(TypeError, 'abc', 'rjust')
833
834 def test_center(self):
835 self.checkequal(' abc ', 'abc', 'center', 10)
836 self.checkequal(' abc ', 'abc', 'center', 6)
837 self.checkequal('abc', 'abc', 'center', 3)
838 self.checkequal('abc', 'abc', 'center', 2)
839 self.checkequal('***abc****', 'abc', 'center', 10, '*')
840 self.checkraises(TypeError, 'abc', 'center')
841
842 def test_swapcase(self):
843 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
844
845 self.checkraises(TypeError, 'hello', 'swapcase', 42)
846
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000847 def test_zfill(self):
848 self.checkequal('123', '123', 'zfill', 2)
849 self.checkequal('123', '123', 'zfill', 3)
850 self.checkequal('0123', '123', 'zfill', 4)
851 self.checkequal('+123', '+123', 'zfill', 3)
852 self.checkequal('+123', '+123', 'zfill', 4)
853 self.checkequal('+0123', '+123', 'zfill', 5)
854 self.checkequal('-123', '-123', 'zfill', 3)
855 self.checkequal('-123', '-123', 'zfill', 4)
856 self.checkequal('-0123', '-123', 'zfill', 5)
857 self.checkequal('000', '', 'zfill', 3)
858 self.checkequal('34', '34', 'zfill', 1)
859 self.checkequal('0034', '34', 'zfill', 4)
860
861 self.checkraises(TypeError, '123', 'zfill')
862
863class MixinStrUnicodeUserStringTest:
864 # additional tests that only work for
865 # stringlike objects, i.e. str, unicode, UserString
866 # (but not the string module)
867
868 def test_islower(self):
869 self.checkequal(False, '', 'islower')
870 self.checkequal(True, 'a', 'islower')
871 self.checkequal(False, 'A', 'islower')
872 self.checkequal(False, '\n', 'islower')
873 self.checkequal(True, 'abc', 'islower')
874 self.checkequal(False, 'aBc', 'islower')
875 self.checkequal(True, 'abc\n', 'islower')
876 self.checkraises(TypeError, 'abc', 'islower', 42)
877
878 def test_isupper(self):
879 self.checkequal(False, '', 'isupper')
880 self.checkequal(False, 'a', 'isupper')
881 self.checkequal(True, 'A', 'isupper')
882 self.checkequal(False, '\n', 'isupper')
883 self.checkequal(True, 'ABC', 'isupper')
884 self.checkequal(False, 'AbC', 'isupper')
885 self.checkequal(True, 'ABC\n', 'isupper')
886 self.checkraises(TypeError, 'abc', 'isupper', 42)
887
888 def test_istitle(self):
889 self.checkequal(False, '', 'istitle')
890 self.checkequal(False, 'a', 'istitle')
891 self.checkequal(True, 'A', 'istitle')
892 self.checkequal(False, '\n', 'istitle')
893 self.checkequal(True, 'A Titlecased Line', 'istitle')
894 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
895 self.checkequal(True, 'A Titlecased, Line', 'istitle')
896 self.checkequal(False, 'Not a capitalized String', 'istitle')
897 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
898 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
899 self.checkequal(False, 'NOT', 'istitle')
900 self.checkraises(TypeError, 'abc', 'istitle', 42)
901
902 def test_isspace(self):
903 self.checkequal(False, '', 'isspace')
904 self.checkequal(False, 'a', 'isspace')
905 self.checkequal(True, ' ', 'isspace')
906 self.checkequal(True, '\t', 'isspace')
907 self.checkequal(True, '\r', 'isspace')
908 self.checkequal(True, '\n', 'isspace')
909 self.checkequal(True, ' \t\r\n', 'isspace')
910 self.checkequal(False, ' \t\r\na', 'isspace')
911 self.checkraises(TypeError, 'abc', 'isspace', 42)
912
913 def test_isalpha(self):
914 self.checkequal(False, '', 'isalpha')
915 self.checkequal(True, 'a', 'isalpha')
916 self.checkequal(True, 'A', 'isalpha')
917 self.checkequal(False, '\n', 'isalpha')
918 self.checkequal(True, 'abc', 'isalpha')
919 self.checkequal(False, 'aBc123', 'isalpha')
920 self.checkequal(False, 'abc\n', 'isalpha')
921 self.checkraises(TypeError, 'abc', 'isalpha', 42)
922
923 def test_isalnum(self):
924 self.checkequal(False, '', 'isalnum')
925 self.checkequal(True, 'a', 'isalnum')
926 self.checkequal(True, 'A', 'isalnum')
927 self.checkequal(False, '\n', 'isalnum')
928 self.checkequal(True, '123abc456', 'isalnum')
929 self.checkequal(True, 'a1b3c', 'isalnum')
930 self.checkequal(False, 'aBc000 ', 'isalnum')
931 self.checkequal(False, 'abc\n', 'isalnum')
932 self.checkraises(TypeError, 'abc', 'isalnum', 42)
933
934 def test_isdigit(self):
935 self.checkequal(False, '', 'isdigit')
936 self.checkequal(False, 'a', 'isdigit')
937 self.checkequal(True, '0', 'isdigit')
938 self.checkequal(True, '0123456789', 'isdigit')
939 self.checkequal(False, '0123456789a', 'isdigit')
940
941 self.checkraises(TypeError, 'abc', 'isdigit', 42)
942
943 def test_title(self):
944 self.checkequal(' Hello ', ' hello ', 'title')
945 self.checkequal('Hello ', 'hello ', 'title')
946 self.checkequal('Hello ', 'Hello ', 'title')
947 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
948 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
949 self.checkequal('Getint', "getInt", 'title')
950 self.checkraises(TypeError, 'hello', 'title', 42)
951
952 def test_splitlines(self):
953 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
954 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
955 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
956 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
957 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
958 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +0100959 self.checkequal(['', 'abc', 'def', 'ghi', ''],
960 "\nabc\ndef\r\nghi\n\r", 'splitlines', False)
961 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
962 "\nabc\ndef\r\nghi\n\r", 'splitlines', True)
963 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r",
964 'splitlines', keepends=False)
965 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
966 "\nabc\ndef\r\nghi\n\r", 'splitlines', keepends=True)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000967
968 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
969
970 def test_startswith(self):
971 self.checkequal(True, 'hello', 'startswith', 'he')
972 self.checkequal(True, 'hello', 'startswith', 'hello')
973 self.checkequal(False, 'hello', 'startswith', 'hello world')
974 self.checkequal(True, 'hello', 'startswith', '')
975 self.checkequal(False, 'hello', 'startswith', 'ello')
976 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
977 self.checkequal(True, 'hello', 'startswith', 'o', 4)
978 self.checkequal(False, 'hello', 'startswith', 'o', 5)
979 self.checkequal(True, 'hello', 'startswith', '', 5)
980 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
981 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
982 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
983 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
984
985 # test negative indices
986 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
987 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
988 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
989 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
990 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
991 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
992 self.checkequal(False, 'hello', 'startswith', 'o', -2)
993 self.checkequal(True, 'hello', 'startswith', 'o', -1)
994 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
995 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
996
997 self.checkraises(TypeError, 'hello', 'startswith')
998 self.checkraises(TypeError, 'hello', 'startswith', 42)
999
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001000 # test tuple arguments
1001 self.checkequal(True, 'hello', 'startswith', ('he', 'ha'))
1002 self.checkequal(False, 'hello', 'startswith', ('lo', 'llo'))
1003 self.checkequal(True, 'hello', 'startswith', ('hellox', 'hello'))
1004 self.checkequal(False, 'hello', 'startswith', ())
1005 self.checkequal(True, 'helloworld', 'startswith', ('hellowo',
1006 'rld', 'lowo'), 3)
1007 self.checkequal(False, 'helloworld', 'startswith', ('hellowo', 'ello',
1008 'rld'), 3)
1009 self.checkequal(True, 'hello', 'startswith', ('lo', 'he'), 0, -1)
1010 self.checkequal(False, 'hello', 'startswith', ('he', 'hel'), 0, 1)
1011 self.checkequal(True, 'hello', 'startswith', ('he', 'hel'), 0, 2)
1012
1013 self.checkraises(TypeError, 'hello', 'startswith', (42,))
1014
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001015 def test_endswith(self):
1016 self.checkequal(True, 'hello', 'endswith', 'lo')
1017 self.checkequal(False, 'hello', 'endswith', 'he')
1018 self.checkequal(True, 'hello', 'endswith', '')
1019 self.checkequal(False, 'hello', 'endswith', 'hello world')
1020 self.checkequal(False, 'helloworld', 'endswith', 'worl')
1021 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
1022 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
1023 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
1024 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
1025 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
1026 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
1027 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
1028 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
1029 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
1030
1031 # test negative indices
1032 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
1033 self.checkequal(False, 'hello', 'endswith', 'he', -2)
1034 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
1035 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
1036 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
1037 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
1038 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
1039 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
1040 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
1041 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
1042 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
1043 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
1044 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
1045
1046 self.checkraises(TypeError, 'hello', 'endswith')
1047 self.checkraises(TypeError, 'hello', 'endswith', 42)
1048
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001049 # test tuple arguments
1050 self.checkequal(False, 'hello', 'endswith', ('he', 'ha'))
1051 self.checkequal(True, 'hello', 'endswith', ('lo', 'llo'))
1052 self.checkequal(True, 'hello', 'endswith', ('hellox', 'hello'))
1053 self.checkequal(False, 'hello', 'endswith', ())
1054 self.checkequal(True, 'helloworld', 'endswith', ('hellowo',
1055 'rld', 'lowo'), 3)
1056 self.checkequal(False, 'helloworld', 'endswith', ('hellowo', 'ello',
1057 'rld'), 3, -1)
1058 self.checkequal(True, 'hello', 'endswith', ('hell', 'ell'), 0, -1)
1059 self.checkequal(False, 'hello', 'endswith', ('he', 'hel'), 0, 1)
1060 self.checkequal(True, 'hello', 'endswith', ('he', 'hell'), 0, 4)
1061
1062 self.checkraises(TypeError, 'hello', 'endswith', (42,))
1063
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001064 def test___contains__(self):
Ezio Melottib19f43d2010-01-24 20:59:24 +00001065 self.checkequal(True, '', '__contains__', '')
1066 self.checkequal(True, 'abc', '__contains__', '')
1067 self.checkequal(False, 'abc', '__contains__', '\0')
1068 self.checkequal(True, '\0abc', '__contains__', '\0')
1069 self.checkequal(True, 'abc\0', '__contains__', '\0')
1070 self.checkequal(True, '\0abc', '__contains__', 'a')
1071 self.checkequal(True, 'asdf', '__contains__', 'asdf')
1072 self.checkequal(False, 'asd', '__contains__', 'asdf')
1073 self.checkequal(False, '', '__contains__', 'asdf')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001074
1075 def test_subscript(self):
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001076 self.checkequal('a', 'abc', '__getitem__', 0)
1077 self.checkequal('c', 'abc', '__getitem__', -1)
1078 self.checkequal('a', 'abc', '__getitem__', 0)
1079 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1080 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1081 self.checkequal('a', 'abc', '__getitem__', slice(0, 1))
1082 self.checkequal('', 'abc', '__getitem__', slice(0, 0))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001083
1084 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
1085
1086 def test_slice(self):
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001087 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1088 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1089 self.checkequal('ab', 'abc', '__getitem__', slice(0, 2))
1090 self.checkequal('bc', 'abc', '__getitem__', slice(1, 3))
1091 self.checkequal('b', 'abc', '__getitem__', slice(1, 2))
1092 self.checkequal('', 'abc', '__getitem__', slice(2, 2))
1093 self.checkequal('', 'abc', '__getitem__', slice(1000, 1000))
1094 self.checkequal('', 'abc', '__getitem__', slice(2000, 1000))
1095 self.checkequal('', 'abc', '__getitem__', slice(2, 1))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001096
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001097 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001098
Thomas Woutersed03b412007-08-28 21:37:11 +00001099 def test_extended_getslice(self):
1100 # Test extended slicing by comparing with list slicing.
1101 s = string.ascii_letters + string.digits
1102 indices = (0, None, 1, 3, 41, -1, -2, -37)
1103 for start in indices:
1104 for stop in indices:
1105 # Skip step 0 (invalid)
1106 for step in indices[1:]:
1107 L = list(s)[start:stop:step]
1108 self.checkequal("".join(L), s, '__getitem__',
1109 slice(start, stop, step))
1110
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001111 def test_mul(self):
1112 self.checkequal('', 'abc', '__mul__', -1)
1113 self.checkequal('', 'abc', '__mul__', 0)
1114 self.checkequal('abc', 'abc', '__mul__', 1)
1115 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
1116 self.checkraises(TypeError, 'abc', '__mul__')
1117 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +00001118 # XXX: on a 64-bit system, this doesn't raise an overflow error,
1119 # but either raises a MemoryError, or succeeds (if you have 54TiB)
1120 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001121
1122 def test_join(self):
1123 # join now works with any sequence type
1124 # moved here, because the argument order is
1125 # different in string.join (see the test in
1126 # test.test_string.StringTest.test_join)
1127 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
1128 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001129 self.checkequal('bd', '', 'join', ('', 'b', '', 'd'))
1130 self.checkequal('ac', '', 'join', ('a', '', 'c', ''))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001131 self.checkequal('w x y z', ' ', 'join', Sequence())
1132 self.checkequal('abc', 'a', 'join', ('abc',))
1133 self.checkequal('z', 'a', 'join', UserList(['z']))
Walter Dörwald67e83882007-05-05 12:26:27 +00001134 self.checkequal('a.b.c', '.', 'join', ['a', 'b', 'c'])
Guido van Rossum98297ee2007-11-06 21:34:58 +00001135 self.assertRaises(TypeError, '.'.join, ['a', 'b', 3])
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001136 for i in [5, 25, 125]:
1137 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1138 ['a' * i] * i)
1139 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1140 ('a' * i,) * i)
1141
Guido van Rossum98297ee2007-11-06 21:34:58 +00001142 #self.checkequal(str(BadSeq1()), ' ', 'join', BadSeq1())
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001143 self.checkequal('a b c', ' ', 'join', BadSeq2())
1144
1145 self.checkraises(TypeError, ' ', 'join')
1146 self.checkraises(TypeError, ' ', 'join', 7)
Guido van Rossumf1044292007-09-27 18:01:22 +00001147 self.checkraises(TypeError, ' ', 'join', [1, 2, bytes()])
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001148 try:
1149 def f():
1150 yield 4 + ""
1151 self.fixtype(' ').join(f())
Guido van Rossumb940e112007-01-10 16:19:56 +00001152 except TypeError as e:
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001153 if '+' not in str(e):
1154 self.fail('join() ate exception message')
1155 else:
1156 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001157
1158 def test_formatting(self):
1159 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
1160 self.checkequal('+10+', '+%d+', '__mod__', 10)
1161 self.checkequal('a', "%c", '__mod__', "a")
1162 self.checkequal('a', "%c", '__mod__', "a")
1163 self.checkequal('"', "%c", '__mod__', 34)
1164 self.checkequal('$', "%c", '__mod__', 36)
1165 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +00001166 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001167
1168 for ordinal in (-100, 0x200000):
1169 # unicode raises ValueError, str raises OverflowError
1170 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
1171
Christian Heimesa612dc02008-02-24 13:08:18 +00001172 longvalue = sys.maxsize + 10
1173 slongvalue = str(longvalue)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001174 self.checkequal(' 42', '%3ld', '__mod__', 42)
Christian Heimesa612dc02008-02-24 13:08:18 +00001175 self.checkequal('42', '%d', '__mod__', 42.0)
1176 self.checkequal(slongvalue, '%d', '__mod__', longvalue)
1177 self.checkcall('%d', '__mod__', float(longvalue))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001178 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +00001179 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001180
1181 self.checkraises(TypeError, 'abc', '__mod__')
1182 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
1183 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
1184 self.checkraises(TypeError, '%c', '__mod__', (None,))
1185 self.checkraises(ValueError, '%(foo', '__mod__', {})
1186 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
Christian Heimesa612dc02008-02-24 13:08:18 +00001187 self.checkraises(TypeError, '%d', '__mod__', "42") # not numeric
Mark Dickinson5c2db372009-12-05 20:28:34 +00001188 self.checkraises(TypeError, '%d', '__mod__', (42+0j)) # no int conversion provided
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001189
1190 # argument names with properly nested brackets are supported
1191 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
1192
1193 # 100 is a magic number in PyUnicode_Format, this forces a resize
1194 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
1195
1196 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
1197 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
1198 self.checkraises(ValueError, '%10', '__mod__', (42,))
1199
1200 def test_floatformatting(self):
1201 # float formatting
Guido van Rossum805365e2007-05-07 22:24:25 +00001202 for prec in range(100):
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001203 format = '%%.%if' % prec
1204 value = 0.01
Guido van Rossum805365e2007-05-07 22:24:25 +00001205 for x in range(60):
Florent Xiclunaa87b3832010-09-13 02:28:18 +00001206 value = value * 3.14159265359 / 3.0 * 10.0
Mark Dickinsonf489caf2009-05-01 11:42:00 +00001207 self.checkcall(format, "__mod__", value)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001208
Thomas Wouters477c8d52006-05-27 19:21:47 +00001209 def test_inplace_rewrites(self):
1210 # Check that strings don't copy and modify cached single-character strings
1211 self.checkequal('a', 'A', 'lower')
1212 self.checkequal(True, 'A', 'isupper')
1213 self.checkequal('A', 'a', 'upper')
1214 self.checkequal(True, 'a', 'islower')
1215
1216 self.checkequal('a', 'A', 'replace', 'A', 'a')
1217 self.checkequal(True, 'A', 'isupper')
1218
1219 self.checkequal('A', 'a', 'capitalize')
1220 self.checkequal(True, 'a', 'islower')
1221
1222 self.checkequal('A', 'a', 'swapcase')
1223 self.checkequal(True, 'a', 'islower')
1224
1225 self.checkequal('A', 'a', 'title')
1226 self.checkequal(True, 'a', 'islower')
1227
1228 def test_partition(self):
1229
1230 self.checkequal(('this is the par', 'ti', 'tion method'),
1231 'this is the partition method', 'partition', 'ti')
1232
1233 # from raymond's original specification
1234 S = 'http://www.python.org'
1235 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1236 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1237 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1238 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1239
1240 self.checkraises(ValueError, S, 'partition', '')
1241 self.checkraises(TypeError, S, 'partition', None)
1242
1243 def test_rpartition(self):
1244
1245 self.checkequal(('this is the rparti', 'ti', 'on method'),
1246 'this is the rpartition method', 'rpartition', 'ti')
1247
1248 # from raymond's original specification
1249 S = 'http://www.python.org'
1250 self.checkequal(('http', '://', 'www.python.org'), S, 'rpartition', '://')
Thomas Wouters89f507f2006-12-13 04:49:30 +00001251 self.checkequal(('', '', 'http://www.python.org'), S, 'rpartition', '?')
Thomas Wouters477c8d52006-05-27 19:21:47 +00001252 self.checkequal(('', 'http://', 'www.python.org'), S, 'rpartition', 'http://')
1253 self.checkequal(('http://www.python.', 'org', ''), S, 'rpartition', 'org')
1254
1255 self.checkraises(ValueError, S, 'rpartition', '')
1256 self.checkraises(TypeError, S, 'rpartition', None)
1257
Jesus Ceaac451502011-04-20 17:09:23 +02001258 def test_none_arguments(self):
1259 # issue 11828
1260 s = 'hello'
1261 self.checkequal(2, s, 'find', 'l', None)
1262 self.checkequal(3, s, 'find', 'l', -2, None)
1263 self.checkequal(2, s, 'find', 'l', None, -2)
1264 self.checkequal(0, s, 'find', 'h', None, None)
1265
1266 self.checkequal(3, s, 'rfind', 'l', None)
1267 self.checkequal(3, s, 'rfind', 'l', -2, None)
1268 self.checkequal(2, s, 'rfind', 'l', None, -2)
1269 self.checkequal(0, s, 'rfind', 'h', None, None)
1270
1271 self.checkequal(2, s, 'index', 'l', None)
1272 self.checkequal(3, s, 'index', 'l', -2, None)
1273 self.checkequal(2, s, 'index', 'l', None, -2)
1274 self.checkequal(0, s, 'index', 'h', None, None)
1275
1276 self.checkequal(3, s, 'rindex', 'l', None)
1277 self.checkequal(3, s, 'rindex', 'l', -2, None)
1278 self.checkequal(2, s, 'rindex', 'l', None, -2)
1279 self.checkequal(0, s, 'rindex', 'h', None, None)
1280
1281 self.checkequal(2, s, 'count', 'l', None)
1282 self.checkequal(1, s, 'count', 'l', -2, None)
1283 self.checkequal(1, s, 'count', 'l', None, -2)
1284 self.checkequal(0, s, 'count', 'x', None, None)
1285
1286 self.checkequal(True, s, 'endswith', 'o', None)
1287 self.checkequal(True, s, 'endswith', 'lo', -2, None)
1288 self.checkequal(True, s, 'endswith', 'l', None, -2)
1289 self.checkequal(False, s, 'endswith', 'x', None, None)
1290
1291 self.checkequal(True, s, 'startswith', 'h', None)
1292 self.checkequal(True, s, 'startswith', 'l', -2, None)
1293 self.checkequal(True, s, 'startswith', 'h', None, -2)
1294 self.checkequal(False, s, 'startswith', 'x', None, None)
1295
1296 def test_find_etc_raise_correct_error_messages(self):
1297 # issue 11828
1298 s = 'hello'
1299 x = 'x'
Ezio Melottiaf928422011-04-20 21:56:21 +03001300 self.assertRaisesRegex(TypeError, r'^find\(', s.find,
Jesus Ceaac451502011-04-20 17:09:23 +02001301 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001302 self.assertRaisesRegex(TypeError, r'^rfind\(', s.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +02001303 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001304 self.assertRaisesRegex(TypeError, r'^index\(', s.index,
Jesus Ceaac451502011-04-20 17:09:23 +02001305 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001306 self.assertRaisesRegex(TypeError, r'^rindex\(', s.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +02001307 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001308 self.assertRaisesRegex(TypeError, r'^count\(', s.count,
Jesus Ceaac451502011-04-20 17:09:23 +02001309 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001310 self.assertRaisesRegex(TypeError, r'^startswith\(', s.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001311 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001312 self.assertRaisesRegex(TypeError, r'^endswith\(', s.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001313 x, None, None, None)
1314
Walter Dörwald57d88e52004-08-26 16:53:04 +00001315
Walter Dörwald57d88e52004-08-26 16:53:04 +00001316class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001317 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001318
1319 def test_bug1001011(self):
1320 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001321 # involving a subclass.
1322 # Make sure that it is of the appropriate type.
1323 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001324 t = self.type2test
1325 class subclass(t):
1326 pass
1327 s1 = subclass("abcd")
1328 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001329 self.assertIsNot(s1, s2)
1330 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001331
1332 s1 = t("abcd")
1333 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001334 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001335
1336 # Should also test mixed-type join.
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001337 if t is str:
Tim Peters108f1372004-08-27 05:36:07 +00001338 s1 = subclass("abcd")
1339 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001340 self.assertIsNot(s1, s2)
1341 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001342
1343 s1 = t("abcd")
1344 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001345 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001346
Guido van Rossum98297ee2007-11-06 21:34:58 +00001347## elif t is str8:
1348## s1 = subclass("abcd")
1349## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001350## self.assertIsNot(s1, s2)
1351## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001352
Guido van Rossum98297ee2007-11-06 21:34:58 +00001353## s1 = t("abcd")
1354## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001355## self.assertIsNot(s1, s2)
1356## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001357
1358 else:
1359 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)