blob: 30784d4ec5b3fc3baaa4d8e2c333877d9750908e [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
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02008import _testcapi
Jeremy Hylton20f41b62000-07-11 03:31:55 +00009
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000010class Sequence:
Walter Dörwald0fd583c2003-02-21 12:53:50 +000011 def __init__(self, seq='wxyz'): self.seq = seq
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000012 def __len__(self): return len(self.seq)
13 def __getitem__(self, i): return self.seq[i]
14
15class BadSeq1(Sequence):
Guido van Rossume2a383d2007-01-15 16:59:06 +000016 def __init__(self): self.seq = [7, 'hello', 123]
Guido van Rossumf1044292007-09-27 18:01:22 +000017 def __str__(self): return '{0} {1} {2}'.format(*self.seq)
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000018
19class BadSeq2(Sequence):
20 def __init__(self): self.seq = ['a', 'b', 'c']
21 def __len__(self): return 8
22
Ezio Melotti0dceb562013-01-10 07:43:26 +020023class BaseTest:
Georg Brandlc7885542007-03-06 19:16:20 +000024 # These tests are for buffers of values (bytes) and not
25 # specific to character interpretation, used for bytes objects
26 # and various string implementations
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000027
Walter Dörwald0fd583c2003-02-21 12:53:50 +000028 # The type to be tested
29 # Change in subclasses to change the behaviour of fixtesttype()
30 type2test = None
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000031
Antoine Pitrouac65d962011-10-20 23:54:17 +020032 # Whether the "contained items" of the container are integers in
33 # range(0, 256) (i.e. bytes, bytearray) or strings of length 1
34 # (str)
35 contains_bytes = False
36
Walter Dörwald0fd583c2003-02-21 12:53:50 +000037 # All tests pass their arguments to the testing methods
38 # as str objects. fixtesttype() can be used to propagate
39 # these arguments to the appropriate type
40 def fixtype(self, obj):
41 if isinstance(obj, str):
42 return self.__class__.type2test(obj)
43 elif isinstance(obj, list):
44 return [self.fixtype(x) for x in obj]
45 elif isinstance(obj, tuple):
46 return tuple([self.fixtype(x) for x in obj])
47 elif isinstance(obj, dict):
48 return dict([
49 (self.fixtype(key), self.fixtype(value))
Guido van Rossumcc2b0162007-02-11 06:12:03 +000050 for (key, value) in obj.items()
Walter Dörwald0fd583c2003-02-21 12:53:50 +000051 ])
52 else:
53 return obj
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000054
Guido van Rossum09549f42007-08-27 20:40:10 +000055 # check that obj.method(*args) returns result
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010056 def checkequal(self, result, obj, methodname, *args, **kwargs):
Walter Dörwald0fd583c2003-02-21 12:53:50 +000057 result = self.fixtype(result)
Guido van Rossum09549f42007-08-27 20:40:10 +000058 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000059 args = self.fixtype(args)
Ezio Melotticda6b6d2012-02-26 09:39:55 +020060 kwargs = {k: self.fixtype(v) for k,v in kwargs.items()}
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010061 realresult = getattr(obj, methodname)(*args, **kwargs)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000062 self.assertEqual(
63 result,
64 realresult
65 )
66 # if the original is returned make sure that
67 # this doesn't happen with subclasses
Guido van Rossum09549f42007-08-27 20:40:10 +000068 if obj is realresult:
69 try:
70 class subtype(self.__class__.type2test):
71 pass
72 except TypeError:
73 pass # Skip this if we can't subclass
74 else:
75 obj = subtype(obj)
76 realresult = getattr(obj, methodname)(*args)
Ezio Melottib3aedd42010-11-20 19:04:17 +000077 self.assertIsNot(obj, realresult)
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000078
Guido van Rossum09549f42007-08-27 20:40:10 +000079 # check that obj.method(*args) raises exc
80 def checkraises(self, exc, obj, methodname, *args):
81 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000082 args = self.fixtype(args)
83 self.assertRaises(
84 exc,
Guido van Rossum09549f42007-08-27 20:40:10 +000085 getattr(obj, methodname),
Walter Dörwald0fd583c2003-02-21 12:53:50 +000086 *args
87 )
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000088
Guido van Rossum09549f42007-08-27 20:40:10 +000089 # call obj.method(*args) without any checks
90 def checkcall(self, obj, methodname, *args):
91 obj = self.fixtype(obj)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000092 args = self.fixtype(args)
Guido van Rossum09549f42007-08-27 20:40:10 +000093 getattr(obj, methodname)(*args)
Walter Dörwald0fd583c2003-02-21 12:53:50 +000094
Walter Dörwald0fd583c2003-02-21 12:53:50 +000095 def test_count(self):
96 self.checkequal(3, 'aaa', 'count', 'a')
97 self.checkequal(0, 'aaa', 'count', 'b')
98 self.checkequal(3, 'aaa', 'count', 'a')
99 self.checkequal(0, 'aaa', 'count', 'b')
100 self.checkequal(3, 'aaa', 'count', 'a')
101 self.checkequal(0, 'aaa', 'count', 'b')
102 self.checkequal(0, 'aaa', 'count', 'b')
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000103 self.checkequal(2, 'aaa', 'count', 'a', 1)
104 self.checkequal(0, 'aaa', 'count', 'a', 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000105 self.checkequal(1, 'aaa', 'count', 'a', -1)
106 self.checkequal(3, 'aaa', 'count', 'a', -10)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000107 self.checkequal(1, 'aaa', 'count', 'a', 0, 1)
108 self.checkequal(3, 'aaa', 'count', 'a', 0, 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000109 self.checkequal(2, 'aaa', 'count', 'a', 0, -1)
110 self.checkequal(0, 'aaa', 'count', 'a', 0, -10)
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000111 self.checkequal(3, 'aaa', 'count', '', 1)
112 self.checkequal(1, 'aaa', 'count', '', 3)
113 self.checkequal(0, 'aaa', 'count', '', 10)
114 self.checkequal(2, 'aaa', 'count', '', -1)
115 self.checkequal(4, 'aaa', 'count', '', -10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000116
Amaury Forgeot d'Arcf2e93682008-09-26 22:48:41 +0000117 self.checkequal(1, '', 'count', '')
118 self.checkequal(0, '', 'count', '', 1, 1)
119 self.checkequal(0, '', 'count', '', sys.maxsize, 0)
120
121 self.checkequal(0, '', 'count', 'xx')
122 self.checkequal(0, '', 'count', 'xx', 1, 1)
123 self.checkequal(0, '', 'count', 'xx', sys.maxsize, 0)
124
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000125 self.checkraises(TypeError, 'hello', 'count')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200126
127 if self.contains_bytes:
128 self.checkequal(0, 'hello', 'count', 42)
129 else:
130 self.checkraises(TypeError, 'hello', 'count', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000131
Raymond Hettinger57e74472005-02-20 09:54:53 +0000132 # For a variety of combinations,
133 # verify that str.count() matches an equivalent function
134 # replacing all occurrences and then differencing the string lengths
135 charset = ['', 'a', 'b']
136 digits = 7
137 base = len(charset)
138 teststrings = set()
Guido van Rossum805365e2007-05-07 22:24:25 +0000139 for i in range(base ** digits):
Raymond Hettinger57e74472005-02-20 09:54:53 +0000140 entry = []
Guido van Rossum805365e2007-05-07 22:24:25 +0000141 for j in range(digits):
Raymond Hettinger57e74472005-02-20 09:54:53 +0000142 i, m = divmod(i, base)
143 entry.append(charset[m])
144 teststrings.add(''.join(entry))
Guido van Rossum09549f42007-08-27 20:40:10 +0000145 teststrings = [self.fixtype(ts) for ts in teststrings]
Raymond Hettinger57e74472005-02-20 09:54:53 +0000146 for i in teststrings:
Raymond Hettinger57e74472005-02-20 09:54:53 +0000147 n = len(i)
148 for j in teststrings:
149 r1 = i.count(j)
150 if j:
Guido van Rossum09549f42007-08-27 20:40:10 +0000151 r2, rem = divmod(n - len(i.replace(j, self.fixtype(''))),
152 len(j))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000153 else:
154 r2, rem = len(i)+1, 0
155 if rem or r1 != r2:
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000156 self.assertEqual(rem, 0, '%s != 0 for %s' % (rem, i))
157 self.assertEqual(r1, r2, '%s != %s for %s' % (r1, r2, i))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000158
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000159 def test_find(self):
160 self.checkequal(0, 'abcdefghiabc', 'find', 'abc')
161 self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
162 self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
163
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000164 self.checkequal(0, 'abc', 'find', '', 0)
165 self.checkequal(3, 'abc', 'find', '', 3)
166 self.checkequal(-1, 'abc', 'find', '', 4)
167
Christian Heimes9cd17752007-11-18 19:35:23 +0000168 # to check the ability to pass None as defaults
169 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a')
170 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4)
171 self.checkequal(-1, 'rrarrrrrrrrra', 'find', 'a', 4, 6)
172 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4, None)
173 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a', None, 6)
174
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000175 self.checkraises(TypeError, 'hello', 'find')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200176
177 if self.contains_bytes:
178 self.checkequal(-1, 'hello', 'find', 42)
179 else:
180 self.checkraises(TypeError, 'hello', 'find', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000181
Amaury Forgeot d'Arcf2e93682008-09-26 22:48:41 +0000182 self.checkequal(0, '', 'find', '')
183 self.checkequal(-1, '', 'find', '', 1, 1)
184 self.checkequal(-1, '', 'find', '', sys.maxsize, 0)
185
186 self.checkequal(-1, '', 'find', 'xx')
187 self.checkequal(-1, '', 'find', 'xx', 1, 1)
188 self.checkequal(-1, '', 'find', 'xx', sys.maxsize, 0)
189
Antoine Pitrou74edda02010-01-02 21:51:33 +0000190 # issue 7458
191 self.checkequal(-1, 'ab', 'find', 'xxx', sys.maxsize + 1, 0)
192
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000193 # For a variety of combinations,
194 # verify that str.find() matches __contains__
195 # and that the found substring is really at that location
196 charset = ['', 'a', 'b', 'c']
197 digits = 5
198 base = len(charset)
199 teststrings = set()
Guido van Rossum805365e2007-05-07 22:24:25 +0000200 for i in range(base ** digits):
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000201 entry = []
Guido van Rossum805365e2007-05-07 22:24:25 +0000202 for j in range(digits):
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000203 i, m = divmod(i, base)
204 entry.append(charset[m])
205 teststrings.add(''.join(entry))
Guido van Rossum09549f42007-08-27 20:40:10 +0000206 teststrings = [self.fixtype(ts) for ts in teststrings]
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000207 for i in teststrings:
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000208 for j in teststrings:
209 loc = i.find(j)
210 r1 = (loc != -1)
211 r2 = j in i
Antoine Pitrou2e544fb2010-01-02 21:55:17 +0000212 self.assertEqual(r1, r2)
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000213 if loc != -1:
214 self.assertEqual(i[loc:loc+len(j)], j)
215
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000216 def test_rfind(self):
217 self.checkequal(9, 'abcdefghiabc', 'rfind', 'abc')
218 self.checkequal(12, 'abcdefghiabc', 'rfind', '')
219 self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
220 self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
221
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000222 self.checkequal(3, 'abc', 'rfind', '', 0)
223 self.checkequal(3, 'abc', 'rfind', '', 3)
224 self.checkequal(-1, 'abc', 'rfind', '', 4)
225
Christian Heimes9cd17752007-11-18 19:35:23 +0000226 # to check the ability to pass None as defaults
227 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a')
228 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4)
229 self.checkequal(-1, 'rrarrrrrrrrra', 'rfind', 'a', 4, 6)
230 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4, None)
231 self.checkequal( 2, 'rrarrrrrrrrra', 'rfind', 'a', None, 6)
232
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000233 self.checkraises(TypeError, 'hello', 'rfind')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200234
235 if self.contains_bytes:
236 self.checkequal(-1, 'hello', 'rfind', 42)
237 else:
238 self.checkraises(TypeError, 'hello', 'rfind', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000239
Antoine Pitrouda2ecaf2010-01-02 21:40:36 +0000240 # For a variety of combinations,
241 # verify that str.rfind() matches __contains__
242 # and that the found substring is really at that location
243 charset = ['', 'a', 'b', 'c']
244 digits = 5
245 base = len(charset)
246 teststrings = set()
247 for i in range(base ** digits):
248 entry = []
249 for j in range(digits):
250 i, m = divmod(i, base)
251 entry.append(charset[m])
252 teststrings.add(''.join(entry))
253 teststrings = [self.fixtype(ts) for ts in teststrings]
254 for i in teststrings:
255 for j in teststrings:
256 loc = i.rfind(j)
257 r1 = (loc != -1)
258 r2 = j in i
Antoine Pitrou2e544fb2010-01-02 21:55:17 +0000259 self.assertEqual(r1, r2)
Antoine Pitrouda2ecaf2010-01-02 21:40:36 +0000260 if loc != -1:
261 self.assertEqual(i[loc:loc+len(j)], j)
262
Antoine Pitrou74edda02010-01-02 21:51:33 +0000263 # issue 7458
264 self.checkequal(-1, 'ab', 'rfind', 'xxx', sys.maxsize + 1, 0)
265
Victor Stinnerb3f55012012-08-02 23:05:01 +0200266 # issue #15534
267 self.checkequal(0, '<......\u043c...', "rfind", "<")
268
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000269 def test_index(self):
270 self.checkequal(0, 'abcdefghiabc', 'index', '')
271 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
272 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
273 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
274
275 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
276 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
277 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
278 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
279
Christian Heimes9cd17752007-11-18 19:35:23 +0000280 # to check the ability to pass None as defaults
281 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a')
282 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4)
283 self.checkraises(ValueError, 'rrarrrrrrrrra', 'index', 'a', 4, 6)
284 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4, None)
285 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a', None, 6)
286
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000287 self.checkraises(TypeError, 'hello', 'index')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200288
289 if self.contains_bytes:
290 self.checkraises(ValueError, 'hello', 'index', 42)
291 else:
292 self.checkraises(TypeError, 'hello', 'index', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000293
294 def test_rindex(self):
295 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
296 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
297 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
298 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
299
300 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
301 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
302 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
303 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
304 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
305
Christian Heimes9cd17752007-11-18 19:35:23 +0000306 # to check the ability to pass None as defaults
307 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a')
308 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4)
309 self.checkraises(ValueError, 'rrarrrrrrrrra', 'rindex', 'a', 4, 6)
310 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4, None)
311 self.checkequal( 2, 'rrarrrrrrrrra', 'rindex', 'a', None, 6)
312
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000313 self.checkraises(TypeError, 'hello', 'rindex')
Antoine Pitrouac65d962011-10-20 23:54:17 +0200314
315 if self.contains_bytes:
316 self.checkraises(ValueError, 'hello', 'rindex', 42)
317 else:
318 self.checkraises(TypeError, 'hello', 'rindex', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000319
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000320 def test_lower(self):
321 self.checkequal('hello', 'HeLLo', 'lower')
322 self.checkequal('hello', 'hello', 'lower')
323 self.checkraises(TypeError, 'hello', 'lower', 42)
324
325 def test_upper(self):
326 self.checkequal('HELLO', 'HeLLo', 'upper')
327 self.checkequal('HELLO', 'HELLO', 'upper')
328 self.checkraises(TypeError, 'hello', 'upper', 42)
329
330 def test_expandtabs(self):
Ezio Melotti745d54d2013-11-16 19:10:57 +0200331 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
332 'expandtabs')
333 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
334 'expandtabs', 8)
335 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
336 'expandtabs', 4)
337 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
338 'expandtabs')
339 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
340 'expandtabs', 8)
341 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi',
342 'expandtabs', 4)
343 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi',
344 'expandtabs', 4)
345 # check keyword args
346 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
347 'expandtabs', tabsize=8)
348 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi',
349 'expandtabs', tabsize=4)
350
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000351 self.checkequal(' a\n b', ' \ta\n\tb', 'expandtabs', 1)
352
353 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
354 # This test is only valid when sizeof(int) == sizeof(void*) == 4.
Christian Heimesa37d4c62007-12-04 23:02:19 +0000355 if sys.maxsize < (1 << 32) and struct.calcsize('P') == 4:
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000356 self.checkraises(OverflowError,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000357 '\ta\n\tb', 'expandtabs', sys.maxsize)
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000358
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000359 def test_split(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000360 # by a char
361 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000362 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000363 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
364 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
365 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
366 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000367 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000368 sys.maxsize-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000369 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
370 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
371 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000372 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
373 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000374 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
375
Thomas Wouters477c8d52006-05-27 19:21:47 +0000376 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
377 self.checkequal(['a']*15 +['a|a|a|a|a'],
378 ('a|'*20)[:-1], 'split', '|', 15)
379
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000380 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000381 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000382 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
383 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
384 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
385 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000386 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000387 sys.maxsize-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000388 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
389 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000390 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000391 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
392 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
393 'split', 'test')
394 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
395 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
396 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
397 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
398 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
399 self.checkequal([''], '', 'split', 'aaa')
400 self.checkequal(['aa'], 'aa', 'split', 'aaa')
401 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
402 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
403
404 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
405 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
406 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
407 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000408
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200409 # with keyword args
410 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', sep='|')
411 self.checkequal(['a', 'b|c|d'],
412 'a|b|c|d', 'split', '|', maxsplit=1)
413 self.checkequal(['a', 'b|c|d'],
414 'a|b|c|d', 'split', sep='|', maxsplit=1)
415 self.checkequal(['a', 'b|c|d'],
416 'a|b|c|d', 'split', maxsplit=1, sep='|')
417 self.checkequal(['a', 'b c d'],
418 'a b c d', 'split', maxsplit=1)
419
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000420 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000421 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
422
Thomas Wouters477c8d52006-05-27 19:21:47 +0000423 # null case
424 self.checkraises(ValueError, 'hello', 'split', '')
425 self.checkraises(ValueError, 'hello', 'split', '', 0)
426
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000427 def test_rsplit(self):
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000428 # by a char
429 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
430 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
431 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
432 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
433 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000434 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000435 sys.maxsize-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000436 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
437 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
438 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000439 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
440 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
441
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000442 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
443
Thomas Wouters477c8d52006-05-27 19:21:47 +0000444 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
445 self.checkequal(['a|a|a|a|a']+['a']*15,
446 ('a|'*20)[:-1], 'rsplit', '|', 15)
447
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000448 # by string
449 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
450 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
451 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
452 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
453 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000454 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
Christian Heimesa37d4c62007-12-04 23:02:19 +0000455 sys.maxsize-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000456 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
457 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
458 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Thomas Wouters477c8d52006-05-27 19:21:47 +0000459 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
460 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
461 'rsplit', 'test')
462 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
463 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
464 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
465 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
466 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
467 self.checkequal([''], '', 'rsplit', 'aaa')
468 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
469 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
470 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
471
472 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
473 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
474 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
475 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000476
Ezio Melotticda6b6d2012-02-26 09:39:55 +0200477 # with keyword args
478 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', sep='|')
479 self.checkequal(['a|b|c', 'd'],
480 'a|b|c|d', 'rsplit', '|', maxsplit=1)
481 self.checkequal(['a|b|c', 'd'],
482 'a|b|c|d', 'rsplit', sep='|', maxsplit=1)
483 self.checkequal(['a|b|c', 'd'],
484 'a|b|c|d', 'rsplit', maxsplit=1, sep='|')
485 self.checkequal(['a b c', 'd'],
486 'a b c d', 'rsplit', maxsplit=1)
487
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000488 # argument type
489 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000490
Thomas Wouters477c8d52006-05-27 19:21:47 +0000491 # null case
492 self.checkraises(ValueError, 'hello', 'rsplit', '')
493 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
494
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000495 def test_replace(self):
Thomas Wouters477c8d52006-05-27 19:21:47 +0000496 EQ = self.checkequal
497
498 # Operations on the empty string
499 EQ("", "", "replace", "", "")
Thomas Wouters4d70c3d2006-06-08 14:42:34 +0000500 EQ("A", "", "replace", "", "A")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000501 EQ("", "", "replace", "A", "")
502 EQ("", "", "replace", "A", "A")
503 EQ("", "", "replace", "", "", 100)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000504 EQ("", "", "replace", "", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000505
506 # interleave (from=="", 'to' gets inserted everywhere)
507 EQ("A", "A", "replace", "", "")
508 EQ("*A*", "A", "replace", "", "*")
509 EQ("*1A*1", "A", "replace", "", "*1")
510 EQ("*-#A*-#", "A", "replace", "", "*-#")
511 EQ("*-A*-A*-", "AA", "replace", "", "*-")
512 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000513 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000514 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
515 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
516 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
517 EQ("*-AA", "AA", "replace", "", "*-", 1)
518 EQ("AA", "AA", "replace", "", "*-", 0)
519
520 # single character deletion (from=="A", to=="")
521 EQ("", "A", "replace", "A", "")
522 EQ("", "AAA", "replace", "A", "")
523 EQ("", "AAA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000524 EQ("", "AAA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000525 EQ("", "AAA", "replace", "A", "", 4)
526 EQ("", "AAA", "replace", "A", "", 3)
527 EQ("A", "AAA", "replace", "A", "", 2)
528 EQ("AA", "AAA", "replace", "A", "", 1)
529 EQ("AAA", "AAA", "replace", "A", "", 0)
530 EQ("", "AAAAAAAAAA", "replace", "A", "")
531 EQ("BCD", "ABACADA", "replace", "A", "")
532 EQ("BCD", "ABACADA", "replace", "A", "", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000533 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000534 EQ("BCD", "ABACADA", "replace", "A", "", 5)
535 EQ("BCD", "ABACADA", "replace", "A", "", 4)
536 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
537 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
538 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
539 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
540 EQ("BCD", "ABCAD", "replace", "A", "")
541 EQ("BCD", "ABCADAA", "replace", "A", "")
542 EQ("BCD", "BCD", "replace", "A", "")
543 EQ("*************", "*************", "replace", "A", "")
544 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
545
546 # substring deletion (from=="the", to=="")
547 EQ("", "the", "replace", "the", "")
548 EQ("ater", "theater", "replace", "the", "")
549 EQ("", "thethe", "replace", "the", "")
550 EQ("", "thethethethe", "replace", "the", "")
551 EQ("aaaa", "theatheatheathea", "replace", "the", "")
552 EQ("that", "that", "replace", "the", "")
553 EQ("thaet", "thaet", "replace", "the", "")
554 EQ("here and re", "here and there", "replace", "the", "")
555 EQ("here and re and re", "here and there and there",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000556 "replace", "the", "", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000557 EQ("here and re and re", "here and there and there",
558 "replace", "the", "", -1)
559 EQ("here and re and re", "here and there and there",
560 "replace", "the", "", 3)
561 EQ("here and re and re", "here and there and there",
562 "replace", "the", "", 2)
563 EQ("here and re and there", "here and there and there",
564 "replace", "the", "", 1)
565 EQ("here and there and there", "here and there and there",
566 "replace", "the", "", 0)
567 EQ("here and re and re", "here and there and there", "replace", "the", "")
568
569 EQ("abc", "abc", "replace", "the", "")
570 EQ("abcdefg", "abcdefg", "replace", "the", "")
571
572 # substring deletion (from=="bob", to=="")
573 EQ("bob", "bbobob", "replace", "bob", "")
574 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
575 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
576 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
577
578 # single character replace in place (len(from)==len(to)==1)
579 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
580 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000581 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000582 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
583 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
584 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
585 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
586 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
587
588 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
589 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
590 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
591 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
592 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
593
594 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
595
596 # substring replace in place (len(from)==len(to) > 1)
597 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
Christian Heimesa37d4c62007-12-04 23:02:19 +0000598 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000599 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
600 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
601 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
602 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
603 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
604 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
605 EQ("cobob", "bobob", "replace", "bob", "cob")
606 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
607 EQ("bobob", "bobob", "replace", "bot", "bot")
608
609 # replace single character (len(from)==1, len(to)>1)
610 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
611 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
Christian Heimesa37d4c62007-12-04 23:02:19 +0000612 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000613 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
614 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
615 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
616 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
Victor Stinnerb3f55012012-08-02 23:05:01 +0200617 # issue #15534
618 EQ('...\u043c......&lt;', '...\u043c......<', "replace", "<", "&lt;")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000619
620 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
621
622 # replace substring (len(from)>1, len(to)!=len(from))
623 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
624 "replace", "spam", "ham")
625 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
Christian Heimesa37d4c62007-12-04 23:02:19 +0000626 "replace", "spam", "ham", sys.maxsize)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000627 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
628 "replace", "spam", "ham", -1)
629 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
630 "replace", "spam", "ham", 4)
631 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
632 "replace", "spam", "ham", 3)
633 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
634 "replace", "spam", "ham", 2)
635 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
636 "replace", "spam", "ham", 1)
637 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
638 "replace", "spam", "ham", 0)
639
640 EQ("bobob", "bobobob", "replace", "bobob", "bob")
641 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
642 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
643
Guido van Rossum39478e82007-08-27 17:23:59 +0000644 # XXX Commented out. Is there any reason to support buffer objects
645 # as arguments for str.replace()? GvR
Guido van Rossum254348e2007-11-21 19:29:53 +0000646## ba = bytearray('a')
647## bb = bytearray('b')
Guido van Rossum39478e82007-08-27 17:23:59 +0000648## EQ("bbc", "abc", "replace", ba, bb)
649## EQ("aac", "abc", "replace", bb, ba)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000650
Thomas Wouters477c8d52006-05-27 19:21:47 +0000651 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000652 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
653 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
654 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
655 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
656 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
657 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
658 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
659 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
660 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
661 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
662 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
663 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
664 self.checkequal('', '', 'replace', '', '')
665 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
666 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
667 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
668 # MemoryError due to empty result (platform malloc issue when requesting
669 # 0 bytes).
670 self.checkequal('', '123', 'replace', '123', '')
671 self.checkequal('', '123123', 'replace', '123', '')
672 self.checkequal('x', '123x123', 'replace', '123', '')
673
674 self.checkraises(TypeError, 'hello', 'replace')
675 self.checkraises(TypeError, 'hello', 'replace', 42)
676 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
677 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
678
Thomas Wouters477c8d52006-05-27 19:21:47 +0000679 def test_replace_overflow(self):
680 # Check for overflow checking on 32 bit machines
Christian Heimesa37d4c62007-12-04 23:02:19 +0000681 if sys.maxsize != 2147483647 or struct.calcsize("P") > 4:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000682 return
683 A2_16 = "A" * (2**16)
684 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
685 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
686 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
687
Georg Brandlc7885542007-03-06 19:16:20 +0000688
689
690class CommonTest(BaseTest):
691 # This testcase contains test that can be used in all
692 # stringlike classes. Currently this is str, unicode
693 # UserString and the string module.
694
695 def test_hash(self):
696 # SF bug 1054139: += optimization was not invalidating cached hash value
697 a = self.type2test('DNSSEC')
698 b = self.type2test('')
699 for c in a:
700 b += c
701 hash(b)
702 self.assertEqual(hash(a), hash(b))
703
704 def test_capitalize(self):
705 self.checkequal(' hello ', ' hello ', 'capitalize')
706 self.checkequal('Hello ', 'Hello ','capitalize')
707 self.checkequal('Hello ', 'hello ','capitalize')
708 self.checkequal('Aaaa', 'aaaa', 'capitalize')
709 self.checkequal('Aaaa', 'AaAa', 'capitalize')
710
Ezio Melottiee8d9982011-08-15 09:09:57 +0300711 # check that titlecased chars are lowered correctly
712 # \u1ffc is the titlecased char
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -0500713 self.checkequal('\u03a9\u0399\u1ff3\u1ff3\u1ff3',
Ezio Melottiee8d9982011-08-15 09:09:57 +0300714 '\u1ff3\u1ff3\u1ffc\u1ffc', 'capitalize')
715 # check with cased non-letter chars
716 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
717 '\u24c5\u24ce\u24c9\u24bd\u24c4\u24c3', 'capitalize')
718 self.checkequal('\u24c5\u24e8\u24e3\u24d7\u24de\u24dd',
719 '\u24df\u24e8\u24e3\u24d7\u24de\u24dd', 'capitalize')
720 self.checkequal('\u2160\u2171\u2172',
721 '\u2160\u2161\u2162', 'capitalize')
722 self.checkequal('\u2160\u2171\u2172',
723 '\u2170\u2171\u2172', 'capitalize')
724 # check with Ll chars with no upper - nothing changes here
725 self.checkequal('\u019b\u1d00\u1d86\u0221\u1fb7',
726 '\u019b\u1d00\u1d86\u0221\u1fb7', 'capitalize')
727
Georg Brandlc7885542007-03-06 19:16:20 +0000728 self.checkraises(TypeError, 'hello', 'capitalize', 42)
729
Georg Brandlc7885542007-03-06 19:16:20 +0000730 def test_additional_split(self):
731 self.checkequal(['this', 'is', 'the', 'split', 'function'],
732 'this is the split function', 'split')
733
734 # by whitespace
735 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
736 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
737 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
738 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
739 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
740 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000741 sys.maxsize-1)
Georg Brandlc7885542007-03-06 19:16:20 +0000742 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, 0)
744 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
745
746 self.checkequal([], ' ', 'split')
747 self.checkequal(['a'], ' a ', 'split')
748 self.checkequal(['a', 'b'], ' a b ', 'split')
749 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
750 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
751 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
752 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
753 aaa = ' a '*20
754 self.checkequal(['a']*20, aaa, 'split')
755 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
756 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
757
758 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000759 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000760
761 def test_additional_rsplit(self):
762 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
763 'this is the rsplit function', 'rsplit')
764
765 # by whitespace
766 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
767 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
768 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
769 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
770 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
771 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
Christian Heimesa37d4c62007-12-04 23:02:19 +0000772 sys.maxsize-20)
Georg Brandlc7885542007-03-06 19:16:20 +0000773 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, 0)
775 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
776
777 self.checkequal([], ' ', 'rsplit')
778 self.checkequal(['a'], ' a ', 'rsplit')
779 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
780 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
781 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
782 None, 1)
783 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
784 None, 2)
785 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
786 aaa = ' a '*20
787 self.checkequal(['a']*20, aaa, 'rsplit')
788 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
789 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
790
791 # mixed use of str and unicode
Guido van Rossumef87d6e2007-05-02 19:09:54 +0000792 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', ' ', 2)
Georg Brandlc7885542007-03-06 19:16:20 +0000793
794 def test_strip(self):
795 self.checkequal('hello', ' hello ', 'strip')
796 self.checkequal('hello ', ' hello ', 'lstrip')
797 self.checkequal(' hello', ' hello ', 'rstrip')
798 self.checkequal('hello', 'hello', 'strip')
799
800 # strip/lstrip/rstrip with None arg
801 self.checkequal('hello', ' hello ', 'strip', None)
802 self.checkequal('hello ', ' hello ', 'lstrip', None)
803 self.checkequal(' hello', ' hello ', 'rstrip', None)
804 self.checkequal('hello', 'hello', 'strip', None)
805
806 # strip/lstrip/rstrip with str arg
807 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
808 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
809 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
810 self.checkequal('hello', 'hello', 'strip', 'xyz')
811
Georg Brandlc7885542007-03-06 19:16:20 +0000812 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
813 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
814 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
815
816 def test_ljust(self):
817 self.checkequal('abc ', 'abc', 'ljust', 10)
818 self.checkequal('abc ', 'abc', 'ljust', 6)
819 self.checkequal('abc', 'abc', 'ljust', 3)
820 self.checkequal('abc', 'abc', 'ljust', 2)
821 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
822 self.checkraises(TypeError, 'abc', 'ljust')
823
824 def test_rjust(self):
825 self.checkequal(' abc', 'abc', 'rjust', 10)
826 self.checkequal(' abc', 'abc', 'rjust', 6)
827 self.checkequal('abc', 'abc', 'rjust', 3)
828 self.checkequal('abc', 'abc', 'rjust', 2)
829 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
830 self.checkraises(TypeError, 'abc', 'rjust')
831
832 def test_center(self):
833 self.checkequal(' abc ', 'abc', 'center', 10)
834 self.checkequal(' abc ', 'abc', 'center', 6)
835 self.checkequal('abc', 'abc', 'center', 3)
836 self.checkequal('abc', 'abc', 'center', 2)
837 self.checkequal('***abc****', 'abc', 'center', 10, '*')
838 self.checkraises(TypeError, 'abc', 'center')
839
840 def test_swapcase(self):
841 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
842
843 self.checkraises(TypeError, 'hello', 'swapcase', 42)
844
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000845 def test_zfill(self):
846 self.checkequal('123', '123', 'zfill', 2)
847 self.checkequal('123', '123', 'zfill', 3)
848 self.checkequal('0123', '123', 'zfill', 4)
849 self.checkequal('+123', '+123', 'zfill', 3)
850 self.checkequal('+123', '+123', 'zfill', 4)
851 self.checkequal('+0123', '+123', 'zfill', 5)
852 self.checkequal('-123', '-123', 'zfill', 3)
853 self.checkequal('-123', '-123', 'zfill', 4)
854 self.checkequal('-0123', '-123', 'zfill', 5)
855 self.checkequal('000', '', 'zfill', 3)
856 self.checkequal('34', '34', 'zfill', 1)
857 self.checkequal('0034', '34', 'zfill', 4)
858
859 self.checkraises(TypeError, '123', 'zfill')
860
861class MixinStrUnicodeUserStringTest:
862 # additional tests that only work for
863 # stringlike objects, i.e. str, unicode, UserString
864 # (but not the string module)
865
866 def test_islower(self):
867 self.checkequal(False, '', 'islower')
868 self.checkequal(True, 'a', 'islower')
869 self.checkequal(False, 'A', 'islower')
870 self.checkequal(False, '\n', 'islower')
871 self.checkequal(True, 'abc', 'islower')
872 self.checkequal(False, 'aBc', 'islower')
873 self.checkequal(True, 'abc\n', 'islower')
874 self.checkraises(TypeError, 'abc', 'islower', 42)
875
876 def test_isupper(self):
877 self.checkequal(False, '', 'isupper')
878 self.checkequal(False, 'a', 'isupper')
879 self.checkequal(True, 'A', 'isupper')
880 self.checkequal(False, '\n', 'isupper')
881 self.checkequal(True, 'ABC', 'isupper')
882 self.checkequal(False, 'AbC', 'isupper')
883 self.checkequal(True, 'ABC\n', 'isupper')
884 self.checkraises(TypeError, 'abc', 'isupper', 42)
885
886 def test_istitle(self):
887 self.checkequal(False, '', 'istitle')
888 self.checkequal(False, 'a', 'istitle')
889 self.checkequal(True, 'A', 'istitle')
890 self.checkequal(False, '\n', 'istitle')
891 self.checkequal(True, 'A Titlecased Line', 'istitle')
892 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
893 self.checkequal(True, 'A Titlecased, Line', 'istitle')
894 self.checkequal(False, 'Not a capitalized String', 'istitle')
895 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
896 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
897 self.checkequal(False, 'NOT', 'istitle')
898 self.checkraises(TypeError, 'abc', 'istitle', 42)
899
900 def test_isspace(self):
901 self.checkequal(False, '', 'isspace')
902 self.checkequal(False, 'a', 'isspace')
903 self.checkequal(True, ' ', 'isspace')
904 self.checkequal(True, '\t', 'isspace')
905 self.checkequal(True, '\r', 'isspace')
906 self.checkequal(True, '\n', 'isspace')
907 self.checkequal(True, ' \t\r\n', 'isspace')
908 self.checkequal(False, ' \t\r\na', 'isspace')
909 self.checkraises(TypeError, 'abc', 'isspace', 42)
910
911 def test_isalpha(self):
912 self.checkequal(False, '', 'isalpha')
913 self.checkequal(True, 'a', 'isalpha')
914 self.checkequal(True, 'A', 'isalpha')
915 self.checkequal(False, '\n', 'isalpha')
916 self.checkequal(True, 'abc', 'isalpha')
917 self.checkequal(False, 'aBc123', 'isalpha')
918 self.checkequal(False, 'abc\n', 'isalpha')
919 self.checkraises(TypeError, 'abc', 'isalpha', 42)
920
921 def test_isalnum(self):
922 self.checkequal(False, '', 'isalnum')
923 self.checkequal(True, 'a', 'isalnum')
924 self.checkequal(True, 'A', 'isalnum')
925 self.checkequal(False, '\n', 'isalnum')
926 self.checkequal(True, '123abc456', 'isalnum')
927 self.checkequal(True, 'a1b3c', 'isalnum')
928 self.checkequal(False, 'aBc000 ', 'isalnum')
929 self.checkequal(False, 'abc\n', 'isalnum')
930 self.checkraises(TypeError, 'abc', 'isalnum', 42)
931
932 def test_isdigit(self):
933 self.checkequal(False, '', 'isdigit')
934 self.checkequal(False, 'a', 'isdigit')
935 self.checkequal(True, '0', 'isdigit')
936 self.checkequal(True, '0123456789', 'isdigit')
937 self.checkequal(False, '0123456789a', 'isdigit')
938
939 self.checkraises(TypeError, 'abc', 'isdigit', 42)
940
941 def test_title(self):
942 self.checkequal(' Hello ', ' hello ', 'title')
943 self.checkequal('Hello ', 'hello ', 'title')
944 self.checkequal('Hello ', 'Hello ', 'title')
945 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
946 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
947 self.checkequal('Getint', "getInt", 'title')
948 self.checkraises(TypeError, 'hello', 'title', 42)
949
950 def test_splitlines(self):
951 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
952 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
953 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
954 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
955 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
956 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +0100957 self.checkequal(['', 'abc', 'def', 'ghi', ''],
958 "\nabc\ndef\r\nghi\n\r", 'splitlines', False)
959 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
960 "\nabc\ndef\r\nghi\n\r", 'splitlines', True)
961 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r",
962 'splitlines', keepends=False)
963 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'],
964 "\nabc\ndef\r\nghi\n\r", 'splitlines', keepends=True)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000965
966 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
967
968 def test_startswith(self):
969 self.checkequal(True, 'hello', 'startswith', 'he')
970 self.checkequal(True, 'hello', 'startswith', 'hello')
971 self.checkequal(False, 'hello', 'startswith', 'hello world')
972 self.checkequal(True, 'hello', 'startswith', '')
973 self.checkequal(False, 'hello', 'startswith', 'ello')
974 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
975 self.checkequal(True, 'hello', 'startswith', 'o', 4)
976 self.checkequal(False, 'hello', 'startswith', 'o', 5)
977 self.checkequal(True, 'hello', 'startswith', '', 5)
978 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
979 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
980 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
981 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
982
983 # test negative indices
984 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
985 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
986 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
987 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
988 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
989 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
990 self.checkequal(False, 'hello', 'startswith', 'o', -2)
991 self.checkequal(True, 'hello', 'startswith', 'o', -1)
992 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
993 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
994
995 self.checkraises(TypeError, 'hello', 'startswith')
996 self.checkraises(TypeError, 'hello', 'startswith', 42)
997
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000998 # test tuple arguments
999 self.checkequal(True, 'hello', 'startswith', ('he', 'ha'))
1000 self.checkequal(False, 'hello', 'startswith', ('lo', 'llo'))
1001 self.checkequal(True, 'hello', 'startswith', ('hellox', 'hello'))
1002 self.checkequal(False, 'hello', 'startswith', ())
1003 self.checkequal(True, 'helloworld', 'startswith', ('hellowo',
1004 'rld', 'lowo'), 3)
1005 self.checkequal(False, 'helloworld', 'startswith', ('hellowo', 'ello',
1006 'rld'), 3)
1007 self.checkequal(True, 'hello', 'startswith', ('lo', 'he'), 0, -1)
1008 self.checkequal(False, 'hello', 'startswith', ('he', 'hel'), 0, 1)
1009 self.checkequal(True, 'hello', 'startswith', ('he', 'hel'), 0, 2)
1010
1011 self.checkraises(TypeError, 'hello', 'startswith', (42,))
1012
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001013 def test_endswith(self):
1014 self.checkequal(True, 'hello', 'endswith', 'lo')
1015 self.checkequal(False, 'hello', 'endswith', 'he')
1016 self.checkequal(True, 'hello', 'endswith', '')
1017 self.checkequal(False, 'hello', 'endswith', 'hello world')
1018 self.checkequal(False, 'helloworld', 'endswith', 'worl')
1019 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
1020 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
1021 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
1022 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
1023 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
1024 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
1025 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
1026 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
1027 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
1028
1029 # test negative indices
1030 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
1031 self.checkequal(False, 'hello', 'endswith', 'he', -2)
1032 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
1033 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
1034 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
1035 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
1036 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
1037 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
1038 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
1039 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
1040 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
1041 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
1042 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
1043
1044 self.checkraises(TypeError, 'hello', 'endswith')
1045 self.checkraises(TypeError, 'hello', 'endswith', 42)
1046
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001047 # test tuple arguments
1048 self.checkequal(False, 'hello', 'endswith', ('he', 'ha'))
1049 self.checkequal(True, 'hello', 'endswith', ('lo', 'llo'))
1050 self.checkequal(True, 'hello', 'endswith', ('hellox', 'hello'))
1051 self.checkequal(False, 'hello', 'endswith', ())
1052 self.checkequal(True, 'helloworld', 'endswith', ('hellowo',
1053 'rld', 'lowo'), 3)
1054 self.checkequal(False, 'helloworld', 'endswith', ('hellowo', 'ello',
1055 'rld'), 3, -1)
1056 self.checkequal(True, 'hello', 'endswith', ('hell', 'ell'), 0, -1)
1057 self.checkequal(False, 'hello', 'endswith', ('he', 'hel'), 0, 1)
1058 self.checkequal(True, 'hello', 'endswith', ('he', 'hell'), 0, 4)
1059
1060 self.checkraises(TypeError, 'hello', 'endswith', (42,))
1061
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001062 def test___contains__(self):
Ezio Melottib19f43d2010-01-24 20:59:24 +00001063 self.checkequal(True, '', '__contains__', '')
1064 self.checkequal(True, 'abc', '__contains__', '')
1065 self.checkequal(False, 'abc', '__contains__', '\0')
1066 self.checkequal(True, '\0abc', '__contains__', '\0')
1067 self.checkequal(True, 'abc\0', '__contains__', '\0')
1068 self.checkequal(True, '\0abc', '__contains__', 'a')
1069 self.checkequal(True, 'asdf', '__contains__', 'asdf')
1070 self.checkequal(False, 'asd', '__contains__', 'asdf')
1071 self.checkequal(False, '', '__contains__', 'asdf')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001072
1073 def test_subscript(self):
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001074 self.checkequal('a', 'abc', '__getitem__', 0)
1075 self.checkequal('c', 'abc', '__getitem__', -1)
1076 self.checkequal('a', 'abc', '__getitem__', 0)
1077 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1078 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1079 self.checkequal('a', 'abc', '__getitem__', slice(0, 1))
1080 self.checkequal('', 'abc', '__getitem__', slice(0, 0))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001081
1082 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
1083
1084 def test_slice(self):
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001085 self.checkequal('abc', 'abc', '__getitem__', slice(0, 1000))
1086 self.checkequal('abc', 'abc', '__getitem__', slice(0, 3))
1087 self.checkequal('ab', 'abc', '__getitem__', slice(0, 2))
1088 self.checkequal('bc', 'abc', '__getitem__', slice(1, 3))
1089 self.checkequal('b', 'abc', '__getitem__', slice(1, 2))
1090 self.checkequal('', 'abc', '__getitem__', slice(2, 2))
1091 self.checkequal('', 'abc', '__getitem__', slice(1000, 1000))
1092 self.checkequal('', 'abc', '__getitem__', slice(2000, 1000))
1093 self.checkequal('', 'abc', '__getitem__', slice(2, 1))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001094
Thomas Woutersd2cf20e2007-08-30 22:57:53 +00001095 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001096
Thomas Woutersed03b412007-08-28 21:37:11 +00001097 def test_extended_getslice(self):
1098 # Test extended slicing by comparing with list slicing.
1099 s = string.ascii_letters + string.digits
1100 indices = (0, None, 1, 3, 41, -1, -2, -37)
1101 for start in indices:
1102 for stop in indices:
1103 # Skip step 0 (invalid)
1104 for step in indices[1:]:
1105 L = list(s)[start:stop:step]
1106 self.checkequal("".join(L), s, '__getitem__',
1107 slice(start, stop, step))
1108
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001109 def test_mul(self):
1110 self.checkequal('', 'abc', '__mul__', -1)
1111 self.checkequal('', 'abc', '__mul__', 0)
1112 self.checkequal('abc', 'abc', '__mul__', 1)
1113 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
1114 self.checkraises(TypeError, 'abc', '__mul__')
1115 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +00001116 # XXX: on a 64-bit system, this doesn't raise an overflow error,
1117 # but either raises a MemoryError, or succeeds (if you have 54TiB)
1118 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001119
1120 def test_join(self):
1121 # join now works with any sequence type
1122 # moved here, because the argument order is
1123 # different in string.join (see the test in
1124 # test.test_string.StringTest.test_join)
1125 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
1126 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001127 self.checkequal('bd', '', 'join', ('', 'b', '', 'd'))
1128 self.checkequal('ac', '', 'join', ('a', '', 'c', ''))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001129 self.checkequal('w x y z', ' ', 'join', Sequence())
1130 self.checkequal('abc', 'a', 'join', ('abc',))
1131 self.checkequal('z', 'a', 'join', UserList(['z']))
Walter Dörwald67e83882007-05-05 12:26:27 +00001132 self.checkequal('a.b.c', '.', 'join', ['a', 'b', 'c'])
Guido van Rossum98297ee2007-11-06 21:34:58 +00001133 self.assertRaises(TypeError, '.'.join, ['a', 'b', 3])
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001134 for i in [5, 25, 125]:
1135 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1136 ['a' * i] * i)
1137 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1138 ('a' * i,) * i)
1139
Guido van Rossum98297ee2007-11-06 21:34:58 +00001140 #self.checkequal(str(BadSeq1()), ' ', 'join', BadSeq1())
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001141 self.checkequal('a b c', ' ', 'join', BadSeq2())
1142
1143 self.checkraises(TypeError, ' ', 'join')
1144 self.checkraises(TypeError, ' ', 'join', 7)
Guido van Rossumf1044292007-09-27 18:01:22 +00001145 self.checkraises(TypeError, ' ', 'join', [1, 2, bytes()])
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001146 try:
1147 def f():
1148 yield 4 + ""
1149 self.fixtype(' ').join(f())
Guido van Rossumb940e112007-01-10 16:19:56 +00001150 except TypeError as e:
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001151 if '+' not in str(e):
1152 self.fail('join() ate exception message')
1153 else:
1154 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001155
1156 def test_formatting(self):
1157 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
1158 self.checkequal('+10+', '+%d+', '__mod__', 10)
1159 self.checkequal('a', "%c", '__mod__', "a")
1160 self.checkequal('a', "%c", '__mod__', "a")
1161 self.checkequal('"', "%c", '__mod__', 34)
1162 self.checkequal('$', "%c", '__mod__', 36)
1163 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +00001164 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001165
1166 for ordinal in (-100, 0x200000):
1167 # unicode raises ValueError, str raises OverflowError
1168 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
1169
Christian Heimesa612dc02008-02-24 13:08:18 +00001170 longvalue = sys.maxsize + 10
1171 slongvalue = str(longvalue)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001172 self.checkequal(' 42', '%3ld', '__mod__', 42)
Christian Heimesa612dc02008-02-24 13:08:18 +00001173 self.checkequal('42', '%d', '__mod__', 42.0)
1174 self.checkequal(slongvalue, '%d', '__mod__', longvalue)
1175 self.checkcall('%d', '__mod__', float(longvalue))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001176 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +00001177 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001178
1179 self.checkraises(TypeError, 'abc', '__mod__')
1180 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
1181 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
1182 self.checkraises(TypeError, '%c', '__mod__', (None,))
1183 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))
1201
Serhiy Storchaka441d30f2013-01-19 12:26:26 +02001202 self.checkraises(OverflowError, '%*s', '__mod__',
1203 (_testcapi.PY_SSIZE_T_MAX + 1, ''))
1204 self.checkraises(OverflowError, '%.*f', '__mod__',
1205 (_testcapi.INT_MAX + 1, 1. / 7))
1206 # Issue 15989
1207 self.checkraises(OverflowError, '%*s', '__mod__',
1208 (1 << (_testcapi.PY_SSIZE_T_MAX.bit_length() + 1), ''))
1209 self.checkraises(OverflowError, '%.*f', '__mod__',
1210 (_testcapi.UINT_MAX + 1, 1. / 7))
1211
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -04001212 class X(object): pass
1213 self.checkraises(TypeError, 'abc', '__mod__', X())
1214
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001215 def test_floatformatting(self):
1216 # float formatting
Guido van Rossum805365e2007-05-07 22:24:25 +00001217 for prec in range(100):
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001218 format = '%%.%if' % prec
1219 value = 0.01
Guido van Rossum805365e2007-05-07 22:24:25 +00001220 for x in range(60):
Florent Xiclunaa87b3832010-09-13 02:28:18 +00001221 value = value * 3.14159265359 / 3.0 * 10.0
Mark Dickinsonf489caf2009-05-01 11:42:00 +00001222 self.checkcall(format, "__mod__", value)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001223
Thomas Wouters477c8d52006-05-27 19:21:47 +00001224 def test_inplace_rewrites(self):
1225 # Check that strings don't copy and modify cached single-character strings
1226 self.checkequal('a', 'A', 'lower')
1227 self.checkequal(True, 'A', 'isupper')
1228 self.checkequal('A', 'a', 'upper')
1229 self.checkequal(True, 'a', 'islower')
1230
1231 self.checkequal('a', 'A', 'replace', 'A', 'a')
1232 self.checkequal(True, 'A', 'isupper')
1233
1234 self.checkequal('A', 'a', 'capitalize')
1235 self.checkequal(True, 'a', 'islower')
1236
1237 self.checkequal('A', 'a', 'swapcase')
1238 self.checkequal(True, 'a', 'islower')
1239
1240 self.checkequal('A', 'a', 'title')
1241 self.checkequal(True, 'a', 'islower')
1242
1243 def test_partition(self):
1244
1245 self.checkequal(('this is the par', 'ti', 'tion method'),
1246 'this is the partition method', 'partition', 'ti')
1247
1248 # from raymond's original specification
1249 S = 'http://www.python.org'
1250 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1251 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1252 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1253 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1254
1255 self.checkraises(ValueError, S, 'partition', '')
1256 self.checkraises(TypeError, S, 'partition', None)
1257
1258 def test_rpartition(self):
1259
1260 self.checkequal(('this is the rparti', 'ti', 'on method'),
1261 'this is the rpartition method', 'rpartition', 'ti')
1262
1263 # from raymond's original specification
1264 S = 'http://www.python.org'
1265 self.checkequal(('http', '://', 'www.python.org'), S, 'rpartition', '://')
Thomas Wouters89f507f2006-12-13 04:49:30 +00001266 self.checkequal(('', '', 'http://www.python.org'), S, 'rpartition', '?')
Thomas Wouters477c8d52006-05-27 19:21:47 +00001267 self.checkequal(('', 'http://', 'www.python.org'), S, 'rpartition', 'http://')
1268 self.checkequal(('http://www.python.', 'org', ''), S, 'rpartition', 'org')
1269
1270 self.checkraises(ValueError, S, 'rpartition', '')
1271 self.checkraises(TypeError, S, 'rpartition', None)
1272
Jesus Ceaac451502011-04-20 17:09:23 +02001273 def test_none_arguments(self):
1274 # issue 11828
1275 s = 'hello'
1276 self.checkequal(2, s, 'find', 'l', None)
1277 self.checkequal(3, s, 'find', 'l', -2, None)
1278 self.checkequal(2, s, 'find', 'l', None, -2)
1279 self.checkequal(0, s, 'find', 'h', None, None)
1280
1281 self.checkequal(3, s, 'rfind', 'l', None)
1282 self.checkequal(3, s, 'rfind', 'l', -2, None)
1283 self.checkequal(2, s, 'rfind', 'l', None, -2)
1284 self.checkequal(0, s, 'rfind', 'h', None, None)
1285
1286 self.checkequal(2, s, 'index', 'l', None)
1287 self.checkequal(3, s, 'index', 'l', -2, None)
1288 self.checkequal(2, s, 'index', 'l', None, -2)
1289 self.checkequal(0, s, 'index', 'h', None, None)
1290
1291 self.checkequal(3, s, 'rindex', 'l', None)
1292 self.checkequal(3, s, 'rindex', 'l', -2, None)
1293 self.checkequal(2, s, 'rindex', 'l', None, -2)
1294 self.checkequal(0, s, 'rindex', 'h', None, None)
1295
1296 self.checkequal(2, s, 'count', 'l', None)
1297 self.checkequal(1, s, 'count', 'l', -2, None)
1298 self.checkequal(1, s, 'count', 'l', None, -2)
1299 self.checkequal(0, s, 'count', 'x', None, None)
1300
1301 self.checkequal(True, s, 'endswith', 'o', None)
1302 self.checkequal(True, s, 'endswith', 'lo', -2, None)
1303 self.checkequal(True, s, 'endswith', 'l', None, -2)
1304 self.checkequal(False, s, 'endswith', 'x', None, None)
1305
1306 self.checkequal(True, s, 'startswith', 'h', None)
1307 self.checkequal(True, s, 'startswith', 'l', -2, None)
1308 self.checkequal(True, s, 'startswith', 'h', None, -2)
1309 self.checkequal(False, s, 'startswith', 'x', None, None)
1310
1311 def test_find_etc_raise_correct_error_messages(self):
1312 # issue 11828
1313 s = 'hello'
1314 x = 'x'
Ezio Melottiaf928422011-04-20 21:56:21 +03001315 self.assertRaisesRegex(TypeError, r'^find\(', s.find,
Jesus Ceaac451502011-04-20 17:09:23 +02001316 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001317 self.assertRaisesRegex(TypeError, r'^rfind\(', s.rfind,
Jesus Ceaac451502011-04-20 17:09:23 +02001318 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001319 self.assertRaisesRegex(TypeError, r'^index\(', s.index,
Jesus Ceaac451502011-04-20 17:09:23 +02001320 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001321 self.assertRaisesRegex(TypeError, r'^rindex\(', s.rindex,
Jesus Ceaac451502011-04-20 17:09:23 +02001322 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001323 self.assertRaisesRegex(TypeError, r'^count\(', s.count,
Jesus Ceaac451502011-04-20 17:09:23 +02001324 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001325 self.assertRaisesRegex(TypeError, r'^startswith\(', s.startswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001326 x, None, None, None)
Ezio Melottiaf928422011-04-20 21:56:21 +03001327 self.assertRaisesRegex(TypeError, r'^endswith\(', s.endswith,
Jesus Ceaac451502011-04-20 17:09:23 +02001328 x, None, None, None)
1329
Victor Stinnerb3f55012012-08-02 23:05:01 +02001330 # issue #15534
1331 self.checkequal(10, "...\u043c......<", "find", "<")
1332
Walter Dörwald57d88e52004-08-26 16:53:04 +00001333
Walter Dörwald57d88e52004-08-26 16:53:04 +00001334class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001335 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001336
1337 def test_bug1001011(self):
1338 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001339 # involving a subclass.
1340 # Make sure that it is of the appropriate type.
1341 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001342 t = self.type2test
1343 class subclass(t):
1344 pass
1345 s1 = subclass("abcd")
1346 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001347 self.assertIsNot(s1, s2)
1348 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001349
1350 s1 = t("abcd")
1351 s2 = t().join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001352 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001353
1354 # Should also test mixed-type join.
Guido van Rossumef87d6e2007-05-02 19:09:54 +00001355 if t is str:
Tim Peters108f1372004-08-27 05:36:07 +00001356 s1 = subclass("abcd")
1357 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001358 self.assertIsNot(s1, s2)
1359 self.assertIs(type(s2), t)
Tim Peters108f1372004-08-27 05:36:07 +00001360
1361 s1 = t("abcd")
1362 s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001363 self.assertIs(s1, s2)
Tim Peters108f1372004-08-27 05:36:07 +00001364
Guido van Rossum98297ee2007-11-06 21:34:58 +00001365## elif t is str8:
1366## s1 = subclass("abcd")
1367## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001368## self.assertIsNot(s1, s2)
1369## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001370
Guido van Rossum98297ee2007-11-06 21:34:58 +00001371## s1 = t("abcd")
1372## s2 = "".join([s1])
Ezio Melottib3aedd42010-11-20 19:04:17 +00001373## self.assertIsNot(s1, s2)
1374## self.assertIs(type(s2), str) # promotes!
Tim Peters108f1372004-08-27 05:36:07 +00001375
1376 else:
1377 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)