blob: 5ed01f2979952af74bd80c29a810366719c4b21a [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
Ezio Melotti0dceb562013-01-10 07:43:26 +020022class BaseTest:
Georg Brandlc7885542007-03-06 19:16:20 +000023 # 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
Victor Stinnerb3f55012012-08-02 23:05:01 +0200265 # issue #15534
266 self.checkequal(0, '<......\u043c...', "rfind", "<")
267
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000268 def test_index(self):
269 self.checkequal(0, 'abcdefghiabc', 'index', '')
270 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
271 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
272 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
273
274 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
275 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
276 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
277 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
278
Christian Heimes9cd17752007-11-18 19:35:23 +0000279 # to check the ability to pass None as defaults
280 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a')
281 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4)
282 self.checkraises(ValueError, 'rrarrrrrrrrra', 'index', 'a', 4, 6)
283 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4, None)
284 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a', None, 6)
285
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000286 self.checkraises(TypeError, 'hello', 'index')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200287
288 if self.contains_bytes:
289 self.checkraises(ValueError, 'hello', 'index', 42)
290 else:
291 self.checkraises(TypeError, 'hello', 'index', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000292
293 def test_rindex(self):
294 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
295 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
296 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
297 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
298
299 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
300 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
301 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
302 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
303 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
304
Christian Heimes9cd17752007-11-18 19:35:23 +0000305 # to check the ability to pass None as defaults
306 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a')
307 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4)
308 self.checkraises(ValueError, 'rrarrrrrrrrra', 'rindex', 'a', 4, 6)
309 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4, None)
310 self.checkequal( 2, 'rrarrrrrrrrra', 'rindex', 'a', None, 6)
311
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000312 self.checkraises(TypeError, 'hello', 'rindex')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200313
314 if self.contains_bytes:
315 self.checkraises(ValueError, 'hello', 'rindex', 42)
316 else:
317 self.checkraises(TypeError, 'hello', 'rindex', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000318
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000319 def test_lower(self):
320 self.checkequal('hello', 'HeLLo', 'lower')
321 self.checkequal('hello', 'hello', 'lower')
322 self.checkraises(TypeError, 'hello', 'lower', 42)
323
324 def test_upper(self):
325 self.checkequal('HELLO', 'HeLLo', 'upper')
326 self.checkequal('HELLO', 'HELLO', 'upper')
327 self.checkraises(TypeError, 'hello', 'upper', 42)
328
329 def test_expandtabs(self):
Ezio Melotti745d54d2013-11-16 19:10:57 +0200330 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
331 'expandtabs')
332 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
333 'expandtabs', 8)
334 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
335 'expandtabs', 4)
336 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
337 'expandtabs')
338 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
339 'expandtabs', 8)
340 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
341 'expandtabs', 4)
342 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi',
343 'expandtabs', 4)
344 # check keyword args
345 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
346 'expandtabs', tabsize=8)
347 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
348 'expandtabs', tabsize=4)
349
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000350 self.checkequal(' a\n b', ' \ta\n\tb', 'expandtabs', 1)
351
352 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
353 # This test is only valid when sizeof(int) == sizeof(void*) == 4.
Christian Heimesa37d4c62007-12-04 23:02:19 +0000354 if sys.maxsize < (1 << 32) and struct.calcsize('P') == 4:
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000355 self.checkraises(OverflowError,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000356 '\ta\n\tb', 'expandtabs', sys.maxsize)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000357
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000358 def test_split(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000359 # by a char
360 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000361 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000362 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
363 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
364 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
365 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000366 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000367 sys.maxsize-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000368 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
369 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
370 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000371 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
372 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000373 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
374
Thomas Wouters477c8d52006-05-27 19:21:47 +0000375 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
376 self.checkequal(['a']*15 +['a|a|a|a|a'],
377 ('a|'*20)[:-1], 'split', '|', 15)
378
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000379 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000380 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000381 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
382 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
383 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
384 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000385 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000386 sys.maxsize-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000387 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
388 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000389 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000390 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
391 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
392 'split', 'test')
393 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
394 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
395 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
396 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
397 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
398 self.checkequal([''], '', 'split', 'aaa')
399 self.checkequal(['aa'], 'aa', 'split', 'aaa')
400 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
401 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
402
403 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
404 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
405 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
406 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000407
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200408 # with keyword args
409 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', sep='|')
410 self.checkequal(['a', 'b|c|d'],
411 'a|b|c|d', 'split', '|', maxsplit=1)
412 self.checkequal(['a', 'b|c|d'],
413 'a|b|c|d', 'split', sep='|', maxsplit=1)
414 self.checkequal(['a', 'b|c|d'],
415 'a|b|c|d', 'split', maxsplit=1, sep='|')
416 self.checkequal(['a', 'b c d'],
417 'a b c d', 'split', maxsplit=1)
418
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000419 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000420 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
421
Thomas Wouters477c8d52006-05-27 19:21:47 +0000422 # null case
423 self.checkraises(ValueError, 'hello', 'split', '')
424 self.checkraises(ValueError, 'hello', 'split', '', 0)
425
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000426 def test_rsplit(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000427 # by a char
428 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
429 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
430 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
431 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
432 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000433 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000434 sys.maxsize-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000435 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
436 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
437 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000438 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
439 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
440
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000441 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
442
Thomas Wouters477c8d52006-05-27 19:21:47 +0000443 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
444 self.checkequal(['a|a|a|a|a']+['a']*15,
445 ('a|'*20)[:-1], 'rsplit', '|', 15)
446
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000447 # by string
448 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
449 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
450 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
451 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
452 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000453 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000454 sys.maxsize-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000455 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
456 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
457 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000458 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
459 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
460 'rsplit', 'test')
461 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
462 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
463 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
464 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
465 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
466 self.checkequal([''], '', 'rsplit', 'aaa')
467 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
468 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
469 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
470
471 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
472 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
473 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
474 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000475
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200476 # with keyword args
477 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', sep='|')
478 self.checkequal(['a|b|c', 'd'],
479 'a|b|c|d', 'rsplit', '|', maxsplit=1)
480 self.checkequal(['a|b|c', 'd'],
481 'a|b|c|d', 'rsplit', sep='|', maxsplit=1)
482 self.checkequal(['a|b|c', 'd'],
483 'a|b|c|d', 'rsplit', maxsplit=1, sep='|')
484 self.checkequal(['a b c', 'd'],
485 'a b c d', 'rsplit', maxsplit=1)
486
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000487 # argument type
488 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000489
Thomas Wouters477c8d52006-05-27 19:21:47 +0000490 # null case
491 self.checkraises(ValueError, 'hello', 'rsplit', '')
492 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
493
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000494 def test_replace(self):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000495 EQ = self.checkequal
496
497 # Operations on the empty string
498 EQ("", "", "replace", "", "")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000499 EQ("A", "", "replace", "", "A")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000500 EQ("", "", "replace", "A", "")
501 EQ("", "", "replace", "A", "A")
502 EQ("", "", "replace", "", "", 100)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000503 EQ("", "", "replace", "", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000504
505 # interleave (from=="", 'to' gets inserted everywhere)
506 EQ("A", "A", "replace", "", "")
507 EQ("*A*", "A", "replace", "", "*")
508 EQ("*1A*1", "A", "replace", "", "*1")
509 EQ("*-#A*-#", "A", "replace", "", "*-#")
510 EQ("*-A*-A*-", "AA", "replace", "", "*-")
511 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000512 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000513 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
514 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
515 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
516 EQ("*-AA", "AA", "replace", "", "*-", 1)
517 EQ("AA", "AA", "replace", "", "*-", 0)
518
519 # single character deletion (from=="A", to=="")
520 EQ("", "A", "replace", "A", "")
521 EQ("", "AAA", "replace", "A", "")
522 EQ("", "AAA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000523 EQ("", "AAA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000524 EQ("", "AAA", "replace", "A", "", 4)
525 EQ("", "AAA", "replace", "A", "", 3)
526 EQ("A", "AAA", "replace", "A", "", 2)
527 EQ("AA", "AAA", "replace", "A", "", 1)
528 EQ("AAA", "AAA", "replace", "A", "", 0)
529 EQ("", "AAAAAAAAAA", "replace", "A", "")
530 EQ("BCD", "ABACADA", "replace", "A", "")
531 EQ("BCD", "ABACADA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000532 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000533 EQ("BCD", "ABACADA", "replace", "A", "", 5)
534 EQ("BCD", "ABACADA", "replace", "A", "", 4)
535 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
536 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
537 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
538 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
539 EQ("BCD", "ABCAD", "replace", "A", "")
540 EQ("BCD", "ABCADAA", "replace", "A", "")
541 EQ("BCD", "BCD", "replace", "A", "")
542 EQ("*************", "*************", "replace", "A", "")
543 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
544
545 # substring deletion (from=="the", to=="")
546 EQ("", "the", "replace", "the", "")
547 EQ("ater", "theater", "replace", "the", "")
548 EQ("", "thethe", "replace", "the", "")
549 EQ("", "thethethethe", "replace", "the", "")
550 EQ("aaaa", "theatheatheathea", "replace", "the", "")
551 EQ("that", "that", "replace", "the", "")
552 EQ("thaet", "thaet", "replace", "the", "")
553 EQ("here and re", "here and there", "replace", "the", "")
554 EQ("here and re and re", "here and there and there",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000555 "replace", "the", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000556 EQ("here and re and re", "here and there and there",
557 "replace", "the", "", -1)
558 EQ("here and re and re", "here and there and there",
559 "replace", "the", "", 3)
560 EQ("here and re and re", "here and there and there",
561 "replace", "the", "", 2)
562 EQ("here and re and there", "here and there and there",
563 "replace", "the", "", 1)
564 EQ("here and there and there", "here and there and there",
565 "replace", "the", "", 0)
566 EQ("here and re and re", "here and there and there", "replace", "the", "")
567
568 EQ("abc", "abc", "replace", "the", "")
569 EQ("abcdefg", "abcdefg", "replace", "the", "")
570
571 # substring deletion (from=="bob", to=="")
572 EQ("bob", "bbobob", "replace", "bob", "")
573 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
574 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
575 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
576
577 # single character replace in place (len(from)==len(to)==1)
578 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
579 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000580 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000581 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
582 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
583 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
584 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
585 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
586
587 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
588 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
589 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
590 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
591 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
592
593 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
594
595 # substring replace in place (len(from)==len(to) > 1)
596 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000597 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000598 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
599 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
600 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
601 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
602 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
603 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
604 EQ("cobob", "bobob", "replace", "bob", "cob")
605 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
606 EQ("bobob", "bobob", "replace", "bot", "bot")
607
608 # replace single character (len(from)==1, len(to)>1)
609 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
610 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000611 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000612 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
613 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
614 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
615 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
Victor Stinnerb3f55012012-08-02 23:05:01 +0200616 # issue #15534
617 EQ('...\u043c......&lt;', '...\u043c......<', "replace", "<", "&lt;")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000618
619 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
620
621 # replace substring (len(from)>1, len(to)!=len(from))
622 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
623 "replace", "spam", "ham")
624 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000625 "replace", "spam", "ham", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000626 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
627 "replace", "spam", "ham", -1)
628 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
629 "replace", "spam", "ham", 4)
630 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
631 "replace", "spam", "ham", 3)
632 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
633 "replace", "spam", "ham", 2)
634 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
635 "replace", "spam", "ham", 1)
636 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
637 "replace", "spam", "ham", 0)
638
639 EQ("bobob", "bobobob", "replace", "bobob", "bob")
640 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
641 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
642
Guido van Rossum39478e82007-08-27 17:23:59 +0000643 # XXX Commented out. Is there any reason to support buffer objects
644 # as arguments for str.replace()? GvR
Guido van Rossum254348e2007-11-21 19:29:53 +0000645## ba = bytearray('a')
646## bb = bytearray('b')
Guido van Rossum39478e82007-08-27 17:23:59 +0000647## EQ("bbc", "abc", "replace", ba, bb)
648## EQ("aac", "abc", "replace", bb, ba)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000649
Thomas Wouters477c8d52006-05-27 19:21:47 +0000650 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000651 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
652 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
653 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
654 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
655 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
656 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
657 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
658 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
659 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
660 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
661 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
662 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
663 self.checkequal('', '', 'replace', '', '')
664 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
665 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
666 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
667 # MemoryError due to empty result (platform malloc issue when requesting
668 # 0 bytes).
669 self.checkequal('', '123', 'replace', '123', '')
670 self.checkequal('', '123123', 'replace', '123', '')
671 self.checkequal('x', '123x123', 'replace', '123', '')
672
673 self.checkraises(TypeError, 'hello', 'replace')
674 self.checkraises(TypeError, 'hello', 'replace', 42)
675 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
676 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
677
Zachary Ware9fe6d862013-12-08 00:20:35 -0600678 @unittest.skipIf(sys.maxsize > (1 << 32) or struct.calcsize('P') != 4,
679 'only applies to 32-bit platforms')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000680 def test_replace_overflow(self):
681 # Check for overflow checking on 32 bit machines
Thomas Wouters477c8d52006-05-27 19:21:47 +0000682 A2_16 = "A" * (2**16)
683 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
684 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
685 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
686
Georg Brandlc7885542007-03-06 19:16:20 +0000687
688
689class CommonTest(BaseTest):
690 # This testcase contains test that can be used in all
691 # stringlike classes. Currently this is str, unicode
692 # UserString and the string module.
693
694 def test_hash(self):
695 # SF bug 1054139: += optimization was not invalidating cached hash value
696 a = self.type2test('DNSSEC')
697 b = self.type2test('')
698 for c in a:
699 b += c
700 hash(b)
701 self.assertEqual(hash(a), hash(b))
702
703 def test_capitalize(self):
704 self.checkequal(' hello ', ' hello ', 'capitalize')
705 self.checkequal('Hello ', 'Hello ','capitalize')
706 self.checkequal('Hello ', 'hello ','capitalize')
707 self.checkequal('Aaaa', 'aaaa', 'capitalize')
708 self.checkequal('Aaaa', 'AaAa', 'capitalize')
709
Ezio Melottiee8d9982011-08-15 09:09:57 +0300710 # check that titlecased chars are lowered correctly
711 # \u1ffc is the titlecased char
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -0500712 self.checkequal('\u03a9\u0399\u1ff3\u1ff3\u1ff3',
Ezio Melottiee8d9982011-08-15 09:09:57 +0300713 '\u1ff3\u1ff3\u1ffc\u1ffc', 'capitalize')
714 # check with cased non-letter chars
715 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
716 '\u24c5\u24ce\u24c9\u24bd\u24c4\u24c3', 'capitalize')
717 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
718 '\u24df\u24e8\u24e3\u24d7\u24de\u24dd', 'capitalize')
719 self.checkequal('\u2160\u2171\u2172',
720 '\u2160\u2161\u2162', 'capitalize')
721 self.checkequal('\u2160\u2171\u2172',
722 '\u2170\u2171\u2172', 'capitalize')
723 # check with Ll chars with no upper - nothing changes here
724 self.checkequal('\u019b\u1d00\u1d86\u0221\u1fb7',
725 '\u019b\u1d00\u1d86\u0221\u1fb7', 'capitalize')
726
Georg Brandlc7885542007-03-06 19:16:20 +0000727 self.checkraises(TypeError, 'hello', 'capitalize', 42)
728
Georg Brandlc7885542007-03-06 19:16:20 +0000729 def test_additional_split(self):
730 self.checkequal(['this', 'is', 'the', 'split', 'function'],
731 'this is the split function', 'split')
732
733 # by whitespace
734 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
735 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
736 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
737 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
738 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
739 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000740 sys.maxsize-1)
Georg Brandlc7885542007-03-06 19:16:20 +0000741 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
742 self.checkequal(['a b c d'], ' a b c d', 'split', None, 0)
743 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
744
745 self.checkequal([], ' ', 'split')
746 self.checkequal(['a'], ' a ', 'split')
747 self.checkequal(['a', 'b'], ' a b ', 'split')
748 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
749 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
750 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
751 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
752 aaa = ' a '*20
753 self.checkequal(['a']*20, aaa, 'split')
754 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
755 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
756
757 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000758 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000759
760 def test_additional_rsplit(self):
761 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
762 'this is the rsplit function', 'rsplit')
763
764 # by whitespace
765 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
766 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
767 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
768 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
769 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
770 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000771 sys.maxsize-20)
Georg Brandlc7885542007-03-06 19:16:20 +0000772 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
773 self.checkequal(['a b c d'], 'a b c d ', 'rsplit', None, 0)
774 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
775
776 self.checkequal([], ' ', 'rsplit')
777 self.checkequal(['a'], ' a ', 'rsplit')
778 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
779 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
780 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
781 None, 1)
782 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
783 None, 2)
784 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
785 aaa = ' a '*20
786 self.checkequal(['a']*20, aaa, 'rsplit')
787 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
788 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
789
790 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000791 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000792
793 def test_strip(self):
794 self.checkequal('hello', ' hello ', 'strip')
795 self.checkequal('hello ', ' hello ', 'lstrip')
796 self.checkequal(' hello', ' hello ', 'rstrip')
797 self.checkequal('hello', 'hello', 'strip')
798
799 # strip/lstrip/rstrip with None arg
800 self.checkequal('hello', ' hello ', 'strip', None)
801 self.checkequal('hello ', ' hello ', 'lstrip', None)
802 self.checkequal(' hello', ' hello ', 'rstrip', None)
803 self.checkequal('hello', 'hello', 'strip', None)
804
805 # strip/lstrip/rstrip with str arg
806 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
807 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
808 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
809 self.checkequal('hello', 'hello', 'strip', 'xyz')
810
Georg Brandlc7885542007-03-06 19:16:20 +0000811 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
812 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
813 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
814
815 def test_ljust(self):
816 self.checkequal('abc ', 'abc', 'ljust', 10)
817 self.checkequal('abc ', 'abc', 'ljust', 6)
818 self.checkequal('abc', 'abc', 'ljust', 3)
819 self.checkequal('abc', 'abc', 'ljust', 2)
820 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
821 self.checkraises(TypeError, 'abc', 'ljust')
822
823 def test_rjust(self):
824 self.checkequal(' abc', 'abc', 'rjust', 10)
825 self.checkequal(' abc', 'abc', 'rjust', 6)
826 self.checkequal('abc', 'abc', 'rjust', 3)
827 self.checkequal('abc', 'abc', 'rjust', 2)
828 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
829 self.checkraises(TypeError, 'abc', 'rjust')
830
831 def test_center(self):
832 self.checkequal(' abc ', 'abc', 'center', 10)
833 self.checkequal(' abc ', 'abc', 'center', 6)
834 self.checkequal('abc', 'abc', 'center', 3)
835 self.checkequal('abc', 'abc', 'center', 2)
836 self.checkequal('***abc****', 'abc', 'center', 10, '*')
837 self.checkraises(TypeError, 'abc', 'center')
838
839 def test_swapcase(self):
840 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
841
842 self.checkraises(TypeError, 'hello', 'swapcase', 42)
843
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000844 def test_zfill(self):
845 self.checkequal('123', '123', 'zfill', 2)
846 self.checkequal('123', '123', 'zfill', 3)
847 self.checkequal('0123', '123', 'zfill', 4)
848 self.checkequal('+123', '+123', 'zfill', 3)
849 self.checkequal('+123', '+123', 'zfill', 4)
850 self.checkequal('+0123', '+123', 'zfill', 5)
851 self.checkequal('-123', '-123', 'zfill', 3)
852 self.checkequal('-123', '-123', 'zfill', 4)
853 self.checkequal('-0123', '-123', 'zfill', 5)
854 self.checkequal('000', '', 'zfill', 3)
855 self.checkequal('34', '34', 'zfill', 1)
856 self.checkequal('0034', '34', 'zfill', 4)
857
858 self.checkraises(TypeError, '123', 'zfill')
859
860class MixinStrUnicodeUserStringTest:
861 # additional tests that only work for
862 # stringlike objects, i.e. str, unicode, UserString
863 # (but not the string module)
864
865 def test_islower(self):
866 self.checkequal(False, '', 'islower')
867 self.checkequal(True, 'a', 'islower')
868 self.checkequal(False, 'A', 'islower')
869 self.checkequal(False, '\n', 'islower')
870 self.checkequal(True, 'abc', 'islower')
871 self.checkequal(False, 'aBc', 'islower')
872 self.checkequal(True, 'abc\n', 'islower')
873 self.checkraises(TypeError, 'abc', 'islower', 42)
874
875 def test_isupper(self):
876 self.checkequal(False, '', 'isupper')
877 self.checkequal(False, 'a', 'isupper')
878 self.checkequal(True, 'A', 'isupper')
879 self.checkequal(False, '\n', 'isupper')
880 self.checkequal(True, 'ABC', 'isupper')
881 self.checkequal(False, 'AbC', 'isupper')
882 self.checkequal(True, 'ABC\n', 'isupper')
883 self.checkraises(TypeError, 'abc', 'isupper', 42)
884
885 def test_istitle(self):
886 self.checkequal(False, '', 'istitle')
887 self.checkequal(False, 'a', 'istitle')
888 self.checkequal(True, 'A', 'istitle')
889 self.checkequal(False, '\n', 'istitle')
890 self.checkequal(True, 'A Titlecased Line', 'istitle')
891 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
892 self.checkequal(True, 'A Titlecased, Line', 'istitle')
893 self.checkequal(False, 'Not a capitalized String', 'istitle')
894 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
895 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
896 self.checkequal(False, 'NOT', 'istitle')
897 self.checkraises(TypeError, 'abc', 'istitle', 42)
898
899 def test_isspace(self):
900 self.checkequal(False, '', 'isspace')
901 self.checkequal(False, 'a', 'isspace')
902 self.checkequal(True, ' ', 'isspace')
903 self.checkequal(True, '\t', 'isspace')
904 self.checkequal(True, '\r', 'isspace')
905 self.checkequal(True, '\n', 'isspace')
906 self.checkequal(True, ' \t\r\n', 'isspace')
907 self.checkequal(False, ' \t\r\na', 'isspace')
908 self.checkraises(TypeError, 'abc', 'isspace', 42)
909
910 def test_isalpha(self):
911 self.checkequal(False, '', 'isalpha')
912 self.checkequal(True, 'a', 'isalpha')
913 self.checkequal(True, 'A', 'isalpha')
914 self.checkequal(False, '\n', 'isalpha')
915 self.checkequal(True, 'abc', 'isalpha')
916 self.checkequal(False, 'aBc123', 'isalpha')
917 self.checkequal(False, 'abc\n', 'isalpha')
918 self.checkraises(TypeError, 'abc', 'isalpha', 42)
919
920 def test_isalnum(self):
921 self.checkequal(False, '', 'isalnum')
922 self.checkequal(True, 'a', 'isalnum')
923 self.checkequal(True, 'A', 'isalnum')
924 self.checkequal(False, '\n', 'isalnum')
925 self.checkequal(True, '123abc456', 'isalnum')
926 self.checkequal(True, 'a1b3c', 'isalnum')
927 self.checkequal(False, 'aBc000 ', 'isalnum')
928 self.checkequal(False, 'abc\n', 'isalnum')
929 self.checkraises(TypeError, 'abc', 'isalnum', 42)
930
931 def test_isdigit(self):
932 self.checkequal(False, '', 'isdigit')
933 self.checkequal(False, 'a', 'isdigit')
934 self.checkequal(True, '0', 'isdigit')
935 self.checkequal(True, '0123456789', 'isdigit')
936 self.checkequal(False, '0123456789a', 'isdigit')
937
938 self.checkraises(TypeError, 'abc', 'isdigit', 42)
939
940 def test_title(self):
941 self.checkequal(' Hello ', ' hello ', 'title')
942 self.checkequal('Hello ', 'hello ', 'title')
943 self.checkequal('Hello ', 'Hello ', 'title')
944 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
945 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
946 self.checkequal('Getint', "getInt", 'title')
947 self.checkraises(TypeError, 'hello', 'title', 42)
948
949 def test_splitlines(self):
950 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
951 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
952 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
953 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
954 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
955 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +0100956 self.checkequal(['', 'abc', 'def', 'ghi', ''],
957 "\nabc\ndef\r\nghi\n\r", 'splitlines', False)
958 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
959 "\nabc\ndef\r\nghi\n\r", 'splitlines', True)
960 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r",
961 'splitlines', keepends=False)
962 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
963 "\nabc\ndef\r\nghi\n\r", 'splitlines', keepends=True)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000964
965 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
966
967 def test_startswith(self):
968 self.checkequal(True, 'hello', 'startswith', 'he')
969 self.checkequal(True, 'hello', 'startswith', 'hello')
970 self.checkequal(False, 'hello', 'startswith', 'hello world')
971 self.checkequal(True, 'hello', 'startswith', '')
972 self.checkequal(False, 'hello', 'startswith', 'ello')
973 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
974 self.checkequal(True, 'hello', 'startswith', 'o', 4)
975 self.checkequal(False, 'hello', 'startswith', 'o', 5)
976 self.checkequal(True, 'hello', 'startswith', '', 5)
977 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
978 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
979 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
980 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
981
982 # test negative indices
983 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
984 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
985 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
986 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
987 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
988 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
989 self.checkequal(False, 'hello', 'startswith', 'o', -2)
990 self.checkequal(True, 'hello', 'startswith', 'o', -1)
991 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
992 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
993
994 self.checkraises(TypeError, 'hello', 'startswith')
995 self.checkraises(TypeError, 'hello', 'startswith', 42)
996
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000997 # test tuple arguments
998 self.checkequal(True, 'hello', 'startswith', ('he', 'ha'))
999 self.checkequal(False, 'hello', 'startswith', ('lo', 'llo'))
1000 self.checkequal(True, 'hello', 'startswith', ('hellox', 'hello'))
1001 self.checkequal(False, 'hello', 'startswith', ())
1002 self.checkequal(True, 'helloworld', 'startswith', ('hellowo',
1003 'rld', 'lowo'), 3)
1004 self.checkequal(False, 'helloworld', 'startswith', ('hellowo', 'ello',
1005 'rld'), 3)
1006 self.checkequal(True, 'hello', 'startswith', ('lo', 'he'), 0, -1)
1007 self.checkequal(False, 'hello', 'startswith', ('he', 'hel'), 0, 1)
1008 self.checkequal(True, 'hello', 'startswith', ('he', 'hel'), 0, 2)
1009
1010 self.checkraises(TypeError, 'hello', 'startswith', (42,))
1011
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001012 def test_endswith(self):
1013 self.checkequal(True, 'hello', 'endswith', 'lo')
1014 self.checkequal(False, 'hello', 'endswith', 'he')
1015 self.checkequal(True, 'hello', 'endswith', '')
1016 self.checkequal(False, 'hello', 'endswith', 'hello world')
1017 self.checkequal(False, 'helloworld', 'endswith', 'worl')
1018 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
1019 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
1020 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
1021 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
1022 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
1023 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
1024 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
1025 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
1026 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
1027
1028 # test negative indices
1029 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
1030 self.checkequal(False, 'hello', 'endswith', 'he', -2)
1031 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
1032 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
1033 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
1034 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
1035 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
1036 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
1037 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
1038 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
1039 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
1040 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
1041 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
1042
1043 self.checkraises(TypeError, 'hello', 'endswith')
1044 self.checkraises(TypeError, 'hello', 'endswith', 42)
1045
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001046 # test tuple arguments
1047 self.checkequal(False, 'hello', 'endswith', ('he', 'ha'))
1048 self.checkequal(True, 'hello', 'endswith', ('lo', 'llo'))
1049 self.checkequal(True, 'hello', 'endswith', ('hellox', 'hello'))
1050 self.checkequal(False, 'hello', 'endswith', ())
1051 self.checkequal(True, 'helloworld', 'endswith', ('hellowo',
1052 'rld', 'lowo'), 3)
1053 self.checkequal(False, 'helloworld', 'endswith', ('hellowo', 'ello',
1054 'rld'), 3, -1)
1055 self.checkequal(True, 'hello', 'endswith', ('hell', 'ell'), 0, -1)
1056 self.checkequal(False, 'hello', 'endswith', ('he', 'hel'), 0, 1)
1057 self.checkequal(True, 'hello', 'endswith', ('he', 'hell'), 0, 4)
1058
1059 self.checkraises(TypeError, 'hello', 'endswith', (42,))
1060
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001061 def test___contains__(self):
Ezio Melottib19f43d2010-01-24 20:59:24 +00001062 self.checkequal(True, '', '__contains__', '')
1063 self.checkequal(True, 'abc', '__contains__', '')
1064 self.checkequal(False, 'abc', '__contains__', '\0')
1065 self.checkequal(True, '\0abc', '__contains__', '\0')
1066 self.checkequal(True, 'abc\0', '__contains__', '\0')
1067 self.checkequal(True, '\0abc', '__contains__', 'a')
1068 self.checkequal(True, 'asdf', '__contains__', 'asdf')
1069 self.checkequal(False, 'asd', '__contains__', 'asdf')
1070 self.checkequal(False, '', '__contains__', 'asdf')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001071
1072 def test_subscript(self):
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001073 self.checkequal('a', 'abc', '__getitem__', 0)
1074 self.checkequal('c', 'abc', '__getitem__', -1)
1075 self.checkequal('a', 'abc', '__getitem__', 0)
1076 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1077 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1078 self.checkequal('a', 'abc', '__getitem__', slice(0, 1))
1079 self.checkequal('', 'abc', '__getitem__', slice(0, 0))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001080
1081 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
1082
1083 def test_slice(self):
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001084 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1085 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1086 self.checkequal('ab', 'abc', '__getitem__', slice(0, 2))
1087 self.checkequal('bc', 'abc', '__getitem__', slice(1, 3))
1088 self.checkequal('b', 'abc', '__getitem__', slice(1, 2))
1089 self.checkequal('', 'abc', '__getitem__', slice(2, 2))
1090 self.checkequal('', 'abc', '__getitem__', slice(1000, 1000))
1091 self.checkequal('', 'abc', '__getitem__', slice(2000, 1000))
1092 self.checkequal('', 'abc', '__getitem__', slice(2, 1))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001093
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001094 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001095
Thomas Woutersed03b412007-08-28 21:37:11 +00001096 def test_extended_getslice(self):
1097 # Test extended slicing by comparing with list slicing.
1098 s = string.ascii_letters + string.digits
1099 indices = (0, None, 1, 3, 41, -1, -2, -37)
1100 for start in indices:
1101 for stop in indices:
1102 # Skip step 0 (invalid)
1103 for step in indices[1:]:
1104 L = list(s)[start:stop:step]
1105 self.checkequal("".join(L), s, '__getitem__',
1106 slice(start, stop, step))
1107
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001108 def test_mul(self):
1109 self.checkequal('', 'abc', '__mul__', -1)
1110 self.checkequal('', 'abc', '__mul__', 0)
1111 self.checkequal('abc', 'abc', '__mul__', 1)
1112 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
1113 self.checkraises(TypeError, 'abc', '__mul__')
1114 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +00001115 # XXX: on a 64-bit system, this doesn't raise an overflow error,
1116 # but either raises a MemoryError, or succeeds (if you have 54TiB)
1117 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001118
1119 def test_join(self):
1120 # join now works with any sequence type
1121 # moved here, because the argument order is
1122 # different in string.join (see the test in
1123 # test.test_string.StringTest.test_join)
1124 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
1125 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 self.checkequal('bd', '', 'join', ('', 'b', '', 'd'))
1127 self.checkequal('ac', '', 'join', ('a', '', 'c', ''))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001128 self.checkequal('w x y z', ' ', 'join', Sequence())
1129 self.checkequal('abc', 'a', 'join', ('abc',))
1130 self.checkequal('z', 'a', 'join', UserList(['z']))
Walter Dörwald67e83882007-05-05 12:26:27 +00001131 self.checkequal('a.b.c', '.', 'join', ['a', 'b', 'c'])
Guido van Rossum98297ee2007-11-06 21:34:58 +00001132 self.assertRaises(TypeError, '.'.join, ['a', 'b', 3])
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001133 for i in [5, 25, 125]:
1134 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1135 ['a' * i] * i)
1136 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1137 ('a' * i,) * i)
1138
Guido van Rossum98297ee2007-11-06 21:34:58 +00001139 #self.checkequal(str(BadSeq1()), ' ', 'join', BadSeq1())
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001140 self.checkequal('a b c', ' ', 'join', BadSeq2())
1141
1142 self.checkraises(TypeError, ' ', 'join')
1143 self.checkraises(TypeError, ' ', 'join', 7)
Guido van Rossumf1044292007-09-27 18:01:22 +00001144 self.checkraises(TypeError, ' ', 'join', [1, 2, bytes()])
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001145 try:
1146 def f():
1147 yield 4 + ""
1148 self.fixtype(' ').join(f())
Guido van Rossumb940e112007-01-10 16:19:56 +00001149 except TypeError as e:
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001150 if '+' not in str(e):
1151 self.fail('join() ate exception message')
1152 else:
1153 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001154
1155 def test_formatting(self):
1156 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
1157 self.checkequal('+10+', '+%d+', '__mod__', 10)
1158 self.checkequal('a', "%c", '__mod__', "a")
1159 self.checkequal('a', "%c", '__mod__', "a")
1160 self.checkequal('"', "%c", '__mod__', 34)
1161 self.checkequal('$', "%c", '__mod__', 36)
1162 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +00001163 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001164
1165 for ordinal in (-100, 0x200000):
1166 # unicode raises ValueError, str raises OverflowError
1167 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
1168
Christian Heimesa612dc02008-02-24 13:08:18 +00001169 longvalue = sys.maxsize + 10
1170 slongvalue = str(longvalue)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001171 self.checkequal(' 42', '%3ld', '__mod__', 42)
Christian Heimesa612dc02008-02-24 13:08:18 +00001172 self.checkequal('42', '%d', '__mod__', 42.0)
1173 self.checkequal(slongvalue, '%d', '__mod__', longvalue)
1174 self.checkcall('%d', '__mod__', float(longvalue))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001175 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +00001176 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001177
1178 self.checkraises(TypeError, 'abc', '__mod__')
1179 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
1180 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
Serhiy Storchakaa146bef2014-02-06 22:44:27 +02001181 with self.assertWarns(DeprecationWarning):
1182 self.checkraises(TypeError, '%c', '__mod__', (None,))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001183 self.checkraises(ValueError, '%(foo', '__mod__', {})
1184 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
Christian Heimesa612dc02008-02-24 13:08:18 +00001185 self.checkraises(TypeError, '%d', '__mod__', "42") # not numeric
Mark Dickinson5c2db372009-12-05 20:28:34 +00001186 self.checkraises(TypeError, '%d', '__mod__', (42+0j)) # no int conversion provided
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001187
1188 # argument names with properly nested brackets are supported
1189 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
1190
1191 # 100 is a magic number in PyUnicode_Format, this forces a resize
1192 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
1193
1194 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
1195 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
1196 self.checkraises(ValueError, '%10', '__mod__', (42,))
1197
Mark Dickinson99e2e552012-05-07 11:20:50 +01001198 # Outrageously large width or precision should raise ValueError.
1199 self.checkraises(ValueError, '%%%df' % (2**64), '__mod__', (3.2))
1200 self.checkraises(ValueError, '%%.%df' % (2**64), '__mod__', (3.2))
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02001201 self.checkraises(OverflowError, '%*s', '__mod__',
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001202 (sys.maxsize + 1, ''))
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02001203 self.checkraises(OverflowError, '%.*f', '__mod__',
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001204 (sys.maxsize + 1, 1. / 7))
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02001205
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -04001206 class X(object): pass
1207 self.checkraises(TypeError, 'abc', '__mod__', X())
1208
Serhiy Storchaka5cfc79d2014-02-07 10:06:39 +02001209 @support.cpython_only
1210 def test_formatting_c_limits(self):
1211 from _testcapi import PY_SSIZE_T_MAX, INT_MAX, UINT_MAX
1212 SIZE_MAX = (1 << (PY_SSIZE_T_MAX.bit_length() + 1)) - 1
1213 self.checkraises(OverflowError, '%*s', '__mod__',
1214 (PY_SSIZE_T_MAX + 1, ''))
1215 self.checkraises(OverflowError, '%.*f', '__mod__',
1216 (INT_MAX + 1, 1. / 7))
1217 # Issue 15989
1218 self.checkraises(OverflowError, '%*s', '__mod__',
1219 (SIZE_MAX + 1, ''))
1220 self.checkraises(OverflowError, '%.*f', '__mod__',
1221 (UINT_MAX + 1, 1. / 7))
1222
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001223 def test_floatformatting(self):
1224 # float formatting
Guido van Rossum805365e2007-05-07 22:24:25 +00001225 for prec in range(100):
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001226 format = '%%.%if' % prec
1227 value = 0.01
Guido van Rossum805365e2007-05-07 22:24:25 +00001228 for x in range(60):
Florent Xiclunaa87b3832010-09-13 02:28:18 +00001229 value = value * 3.14159265359 / 3.0 * 10.0
Mark Dickinsonf489caf2009-05-01 11:42:00 +00001230 self.checkcall(format, "__mod__", value)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001231
Thomas Wouters477c8d52006-05-27 19:21:47 +00001232 def test_inplace_rewrites(self):
1233 # Check that strings don't copy and modify cached single-character strings
1234 self.checkequal('a', 'A', 'lower')
1235 self.checkequal(True, 'A', 'isupper')
1236 self.checkequal('A', 'a', 'upper')
1237 self.checkequal(True, 'a', 'islower')
1238
1239 self.checkequal('a', 'A', 'replace', 'A', 'a')
1240 self.checkequal(True, 'A', 'isupper')
1241
1242 self.checkequal('A', 'a', 'capitalize')
1243 self.checkequal(True, 'a', 'islower')
1244
1245 self.checkequal('A', 'a', 'swapcase')
1246 self.checkequal(True, 'a', 'islower')
1247
1248 self.checkequal('A', 'a', 'title')
1249 self.checkequal(True, 'a', 'islower')
1250
1251 def test_partition(self):
1252
1253 self.checkequal(('this is the par', 'ti', 'tion method'),
1254 'this is the partition method', 'partition', 'ti')
1255
1256 # from raymond's original specification
1257 S = 'http://www.python.org'
1258 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1259 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1260 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1261 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1262
1263 self.checkraises(ValueError, S, 'partition', '')
1264 self.checkraises(TypeError, S, 'partition', None)
1265
1266 def test_rpartition(self):
1267
1268 self.checkequal(('this is the rparti', 'ti', 'on method'),
1269 'this is the rpartition method', 'rpartition', 'ti')
1270
1271 # from raymond's original specification
1272 S = 'http://www.python.org'
1273 self.checkequal(('http', '://', 'www.python.org'), S, 'rpartition', '://')
Thomas Wouters89f507f2006-12-13 04:49:30 +00001274 self.checkequal(('', '', 'http://www.python.org'), S, 'rpartition', '?')
Thomas Wouters477c8d52006-05-27 19:21:47 +00001275 self.checkequal(('', 'http://', 'www.python.org'), S, 'rpartition', 'http://')
1276 self.checkequal(('http://www.python.', 'org', ''), S, 'rpartition', 'org')
1277
1278 self.checkraises(ValueError, S, 'rpartition', '')
1279 self.checkraises(TypeError, S, 'rpartition', None)
1280
Jesus Ceaac451502011-04-20 17:09:23 +02001281 def test_none_arguments(self):
1282 # issue 11828
1283 s = 'hello'
1284 self.checkequal(2, s, 'find', 'l', None)
1285 self.checkequal(3, s, 'find', 'l', -2, None)
1286 self.checkequal(2, s, 'find', 'l', None, -2)
1287 self.checkequal(0, s, 'find', 'h', None, None)
1288
1289 self.checkequal(3, s, 'rfind', 'l', None)
1290 self.checkequal(3, s, 'rfind', 'l', -2, None)
1291 self.checkequal(2, s, 'rfind', 'l', None, -2)
1292 self.checkequal(0, s, 'rfind', 'h', None, None)
1293
1294 self.checkequal(2, s, 'index', 'l', None)
1295 self.checkequal(3, s, 'index', 'l', -2, None)
1296 self.checkequal(2, s, 'index', 'l', None, -2)
1297 self.checkequal(0, s, 'index', 'h', None, None)
1298
1299 self.checkequal(3, s, 'rindex', 'l', None)
1300 self.checkequal(3, s, 'rindex', 'l', -2, None)
1301 self.checkequal(2, s, 'rindex', 'l', None, -2)
1302 self.checkequal(0, s, 'rindex', 'h', None, None)
1303
1304 self.checkequal(2, s, 'count', 'l', None)
1305 self.checkequal(1, s, 'count', 'l', -2, None)
1306 self.checkequal(1, s, 'count', 'l', None, -2)
1307 self.checkequal(0, s, 'count', 'x', None, None)
1308
1309 self.checkequal(True, s, 'endswith', 'o', None)
1310 self.checkequal(True, s, 'endswith', 'lo', -2, None)
1311 self.checkequal(True, s, 'endswith', 'l', None, -2)
1312 self.checkequal(False, s, 'endswith', 'x', None, None)
1313
1314 self.checkequal(True, s, 'startswith', 'h', None)
1315 self.checkequal(True, s, 'startswith', 'l', -2, None)
1316 self.checkequal(True, s, 'startswith', 'h', None, -2)
1317 self.checkequal(False, s, 'startswith', 'x', None, None)
1318
1319 def test_find_etc_raise_correct_error_messages(self):
1320 # issue 11828
1321 s = 'hello'
1322 x = 'x'
Ezio Melottiaf928422011-04-20 21:56:21 +03001323 self.assertRaisesRegex(TypeError, r'^find\(', s.find,
Jesus Ceaac451502011-04-20 17:09:23 +02001324 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001325 self.assertRaisesRegex(TypeError, r'^rfind\(', s.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +02001326 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001327 self.assertRaisesRegex(TypeError, r'^index\(', s.index,
Jesus Ceaac451502011-04-20 17:09:23 +02001328 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001329 self.assertRaisesRegex(TypeError, r'^rindex\(', s.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +02001330 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001331 self.assertRaisesRegex(TypeError, r'^count\(', s.count,
Jesus Ceaac451502011-04-20 17:09:23 +02001332 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001333 self.assertRaisesRegex(TypeError, r'^startswith\(', s.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001334 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001335 self.assertRaisesRegex(TypeError, r'^endswith\(', s.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001336 x, None, None, None)
1337
Victor Stinnerb3f55012012-08-02 23:05:01 +02001338 # issue #15534
1339 self.checkequal(10, "...\u043c......<", "find", "<")
1340
Walter Dörwald57d88e52004-08-26 16:53:04 +00001341
Walter Dörwald57d88e52004-08-26 16:53:04 +00001342class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001343 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001344
1345 def test_bug1001011(self):
1346 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001347 # involving a subclass.
1348 # Make sure that it is of the appropriate type.
1349 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001350 t = self.type2test
1351 class subclass(t):
1352 pass
1353 s1 = subclass("abcd")
1354 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001355 self.assertIsNot(s1, s2)
1356 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001357
1358 s1 = t("abcd")
1359 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001360 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001361
1362 # Should also test mixed-type join.
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001363 if t is str:
Tim Peters108f1372004-08-27 05:36:07 +00001364 s1 = subclass("abcd")
1365 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001366 self.assertIsNot(s1, s2)
1367 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001368
1369 s1 = t("abcd")
1370 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001371 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001372
Guido van Rossum98297ee2007-11-06 21:34:58 +00001373## elif t is str8:
1374## s1 = subclass("abcd")
1375## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001376## self.assertIsNot(s1, s2)
1377## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001378
Guido van Rossum98297ee2007-11-06 21:34:58 +00001379## s1 = t("abcd")
1380## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001381## self.assertIsNot(s1, s2)
1382## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001383
1384 else:
1385 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)