blob: d4fdd8fc0cc31407424a75265190d4bcc3b0f357 [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
Walter Dörwald0fd583c2003-02-21 12:53:50 +00005import unittest, string, sys
6from test import test_support
Jeremy Hylton20f41b62000-07-11 03:31:55 +00007from UserList import UserList
8
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):
15 def __init__(self): self.seq = [7, 'hello', 123L]
16
17class BadSeq2(Sequence):
18 def __init__(self): self.seq = ['a', 'b', 'c']
19 def __len__(self): return 8
20
Walter Dörwald0fd583c2003-02-21 12:53:50 +000021class CommonTest(unittest.TestCase):
22 # This testcase contains test that can be used in all
23 # stringlike classes. Currently this is str, unicode
24 # UserString and the string module.
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000025
Walter Dörwald0fd583c2003-02-21 12:53:50 +000026 # The type to be tested
27 # Change in subclasses to change the behaviour of fixtesttype()
28 type2test = None
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000029
Walter Dörwald0fd583c2003-02-21 12:53:50 +000030 # All tests pass their arguments to the testing methods
31 # as str objects. fixtesttype() can be used to propagate
32 # these arguments to the appropriate type
33 def fixtype(self, obj):
34 if isinstance(obj, str):
35 return self.__class__.type2test(obj)
36 elif isinstance(obj, list):
37 return [self.fixtype(x) for x in obj]
38 elif isinstance(obj, tuple):
39 return tuple([self.fixtype(x) for x in obj])
40 elif isinstance(obj, dict):
41 return dict([
42 (self.fixtype(key), self.fixtype(value))
43 for (key, value) in obj.iteritems()
44 ])
45 else:
46 return obj
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000047
Walter Dörwald0fd583c2003-02-21 12:53:50 +000048 # check that object.method(*args) returns result
49 def checkequal(self, result, object, methodname, *args):
50 result = self.fixtype(result)
51 object = self.fixtype(object)
52 args = self.fixtype(args)
53 realresult = getattr(object, methodname)(*args)
54 self.assertEqual(
55 result,
56 realresult
57 )
58 # if the original is returned make sure that
59 # this doesn't happen with subclasses
60 if object == realresult:
61 class subtype(self.__class__.type2test):
62 pass
63 object = subtype(object)
64 realresult = getattr(object, methodname)(*args)
65 self.assert_(object is not realresult)
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000066
Walter Dörwald0fd583c2003-02-21 12:53:50 +000067 # check that object.method(*args) raises exc
68 def checkraises(self, exc, object, methodname, *args):
69 object = self.fixtype(object)
70 args = self.fixtype(args)
71 self.assertRaises(
72 exc,
73 getattr(object, methodname),
74 *args
75 )
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +000076
Walter Dörwald0fd583c2003-02-21 12:53:50 +000077 # call object.method(*args) without any checks
78 def checkcall(self, object, methodname, *args):
79 object = self.fixtype(object)
80 args = self.fixtype(args)
81 getattr(object, methodname)(*args)
82
Raymond Hettinger561fbf12004-10-26 01:52:37 +000083 def test_hash(self):
84 # SF bug 1054139: += optimization was not invalidating cached hash value
85 a = self.type2test('DNSSEC')
86 b = self.type2test('')
87 for c in a:
88 b += c
89 hash(b)
90 self.assertEqual(hash(a), hash(b))
91
Walter Dörwald0fd583c2003-02-21 12:53:50 +000092 def test_capitalize(self):
93 self.checkequal(' hello ', ' hello ', 'capitalize')
94 self.checkequal('Hello ', 'Hello ','capitalize')
95 self.checkequal('Hello ', 'hello ','capitalize')
96 self.checkequal('Aaaa', 'aaaa', 'capitalize')
97 self.checkequal('Aaaa', 'AaAa', 'capitalize')
98
99 self.checkraises(TypeError, 'hello', 'capitalize', 42)
100
101 def test_count(self):
102 self.checkequal(3, 'aaa', 'count', 'a')
103 self.checkequal(0, 'aaa', 'count', 'b')
104 self.checkequal(3, 'aaa', 'count', 'a')
105 self.checkequal(0, 'aaa', 'count', 'b')
106 self.checkequal(3, 'aaa', 'count', 'a')
107 self.checkequal(0, 'aaa', 'count', 'b')
108 self.checkequal(0, 'aaa', 'count', 'b')
109 self.checkequal(1, 'aaa', 'count', 'a', -1)
110 self.checkequal(3, 'aaa', 'count', 'a', -10)
111 self.checkequal(2, 'aaa', 'count', 'a', 0, -1)
112 self.checkequal(0, 'aaa', 'count', 'a', 0, -10)
113
114 self.checkraises(TypeError, 'hello', 'count')
115 self.checkraises(TypeError, 'hello', 'count', 42)
116
Raymond Hettinger57e74472005-02-20 09:54:53 +0000117 # For a variety of combinations,
118 # verify that str.count() matches an equivalent function
119 # replacing all occurrences and then differencing the string lengths
120 charset = ['', 'a', 'b']
121 digits = 7
122 base = len(charset)
123 teststrings = set()
124 for i in xrange(base ** digits):
125 entry = []
126 for j in xrange(digits):
127 i, m = divmod(i, base)
128 entry.append(charset[m])
129 teststrings.add(''.join(entry))
130 teststrings = list(teststrings)
131 for i in teststrings:
132 i = self.fixtype(i)
133 n = len(i)
134 for j in teststrings:
135 r1 = i.count(j)
136 if j:
137 r2, rem = divmod(n - len(i.replace(j, '')), len(j))
138 else:
139 r2, rem = len(i)+1, 0
140 if rem or r1 != r2:
141 self.assertEqual(rem, 0)
142 self.assertEqual(r1, r2)
143
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000144 def test_find(self):
145 self.checkequal(0, 'abcdefghiabc', 'find', 'abc')
146 self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
147 self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
148
149 self.checkraises(TypeError, 'hello', 'find')
150 self.checkraises(TypeError, 'hello', 'find', 42)
151
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000152 # For a variety of combinations,
153 # verify that str.find() matches __contains__
154 # and that the found substring is really at that location
155 charset = ['', 'a', 'b', 'c']
156 digits = 5
157 base = len(charset)
158 teststrings = set()
159 for i in xrange(base ** digits):
160 entry = []
161 for j in xrange(digits):
162 i, m = divmod(i, base)
163 entry.append(charset[m])
164 teststrings.add(''.join(entry))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000165 teststrings = list(teststrings)
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000166 for i in teststrings:
167 i = self.fixtype(i)
168 for j in teststrings:
169 loc = i.find(j)
170 r1 = (loc != -1)
171 r2 = j in i
172 if r1 != r2:
173 self.assertEqual(r1, r2)
174 if loc != -1:
175 self.assertEqual(i[loc:loc+len(j)], j)
176
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000177 def test_rfind(self):
178 self.checkequal(9, 'abcdefghiabc', 'rfind', 'abc')
179 self.checkequal(12, 'abcdefghiabc', 'rfind', '')
180 self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
181 self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
182
183 self.checkraises(TypeError, 'hello', 'rfind')
184 self.checkraises(TypeError, 'hello', 'rfind', 42)
185
186 def test_index(self):
187 self.checkequal(0, 'abcdefghiabc', 'index', '')
188 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
189 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
190 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
191
192 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
193 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
194 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
195 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
196
197 self.checkraises(TypeError, 'hello', 'index')
198 self.checkraises(TypeError, 'hello', 'index', 42)
199
200 def test_rindex(self):
201 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
202 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
203 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
204 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
205
206 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
207 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
208 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
209 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
210 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
211
212 self.checkraises(TypeError, 'hello', 'rindex')
213 self.checkraises(TypeError, 'hello', 'rindex', 42)
214
215 def test_lower(self):
216 self.checkequal('hello', 'HeLLo', 'lower')
217 self.checkequal('hello', 'hello', 'lower')
218 self.checkraises(TypeError, 'hello', 'lower', 42)
219
220 def test_upper(self):
221 self.checkequal('HELLO', 'HeLLo', 'upper')
222 self.checkequal('HELLO', 'HELLO', 'upper')
223 self.checkraises(TypeError, 'hello', 'upper', 42)
224
225 def test_expandtabs(self):
226 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
227 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
228 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 4)
229 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi', 'expandtabs', 4)
230 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
231 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
232 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi', 'expandtabs', 4)
233
234 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
235
236 def test_split(self):
237 self.checkequal(['this', 'is', 'the', 'split', 'function'],
238 'this is the split function', 'split')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000239
240 # by whitespace
241 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000242 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
243 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
244 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
245 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
246 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
247 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000248
249 # by a char
250 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
251 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
252 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
253 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
254 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
255 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
256 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
257 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
258 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
259
260 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000261 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000262 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
263 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
264 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
265 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
266 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
267 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000268 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
269
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000270 # mixed use of str and unicode
271 self.checkequal([u'a', u'b', u'c d'], 'a b c d', 'split', u' ', 2)
272
273 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000274 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
275
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000276 def test_rsplit(self):
277 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
278 'this is the rsplit function', 'rsplit')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000279
280 # by whitespace
281 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000282 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
283 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
284 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
285 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
286 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000287 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000288
289 # by a char
290 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
291 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
292 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
293 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
294 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
295 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
296 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
297 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
298 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
299
300 # by string
301 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
302 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
303 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
304 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
305 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
306 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
307 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
308 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
309
310 # mixed use of str and unicode
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000311 self.checkequal([u'a b', u'c', u'd'], 'a b c d', 'rsplit', u' ', 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000312
313 # argument type
314 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000315
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000316 def test_strip(self):
317 self.checkequal('hello', ' hello ', 'strip')
318 self.checkequal('hello ', ' hello ', 'lstrip')
319 self.checkequal(' hello', ' hello ', 'rstrip')
320 self.checkequal('hello', 'hello', 'strip')
321
Neal Norwitzffe33b72003-04-10 22:35:32 +0000322 # strip/lstrip/rstrip with None arg
323 self.checkequal('hello', ' hello ', 'strip', None)
324 self.checkequal('hello ', ' hello ', 'lstrip', None)
325 self.checkequal(' hello', ' hello ', 'rstrip', None)
326 self.checkequal('hello', 'hello', 'strip', None)
327
328 # strip/lstrip/rstrip with str arg
329 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
330 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
331 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
332 self.checkequal('hello', 'hello', 'strip', 'xyz')
333
334 # strip/lstrip/rstrip with unicode arg
335 if test_support.have_unicode:
336 self.checkequal(unicode('hello', 'ascii'), 'xyzzyhelloxyzzy',
337 'strip', unicode('xyz', 'ascii'))
338 self.checkequal(unicode('helloxyzzy', 'ascii'), 'xyzzyhelloxyzzy',
339 'lstrip', unicode('xyz', 'ascii'))
340 self.checkequal(unicode('xyzzyhello', 'ascii'), 'xyzzyhelloxyzzy',
341 'rstrip', unicode('xyz', 'ascii'))
342 self.checkequal(unicode('hello', 'ascii'), 'hello',
343 'strip', unicode('xyz', 'ascii'))
344
345 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
346 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
347 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
348
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000349 def test_ljust(self):
350 self.checkequal('abc ', 'abc', 'ljust', 10)
351 self.checkequal('abc ', 'abc', 'ljust', 6)
352 self.checkequal('abc', 'abc', 'ljust', 3)
353 self.checkequal('abc', 'abc', 'ljust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000354 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000355 self.checkraises(TypeError, 'abc', 'ljust')
356
357 def test_rjust(self):
358 self.checkequal(' abc', 'abc', 'rjust', 10)
359 self.checkequal(' abc', 'abc', 'rjust', 6)
360 self.checkequal('abc', 'abc', 'rjust', 3)
361 self.checkequal('abc', 'abc', 'rjust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000362 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000363 self.checkraises(TypeError, 'abc', 'rjust')
364
365 def test_center(self):
366 self.checkequal(' abc ', 'abc', 'center', 10)
367 self.checkequal(' abc ', 'abc', 'center', 6)
368 self.checkequal('abc', 'abc', 'center', 3)
369 self.checkequal('abc', 'abc', 'center', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000370 self.checkequal('***abc****', 'abc', 'center', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000371 self.checkraises(TypeError, 'abc', 'center')
372
373 def test_swapcase(self):
374 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
375
376 self.checkraises(TypeError, 'hello', 'swapcase', 42)
377
378 def test_replace(self):
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000379 EQ = self.checkequal
380
381 # Operations on the empty string
382 EQ("", "", "replace", "", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000383
384 #EQ("A", "", "replace", "", "A")
385 # That was the correct result; this is the result we actually get
Tim Petersf4049082006-05-24 21:00:45 +0000386 # now (for str, but not for unicode):
387 #EQ("", "", "replace", "", "A")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000388
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000389 EQ("", "", "replace", "A", "")
390 EQ("", "", "replace", "A", "A")
391 EQ("", "", "replace", "", "", 100)
392 EQ("", "", "replace", "", "", sys.maxint)
393
394 # interleave (from=="", 'to' gets inserted everywhere)
395 EQ("A", "A", "replace", "", "")
396 EQ("*A*", "A", "replace", "", "*")
397 EQ("*1A*1", "A", "replace", "", "*1")
398 EQ("*-#A*-#", "A", "replace", "", "*-#")
399 EQ("*-A*-A*-", "AA", "replace", "", "*-")
400 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
401 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxint)
402 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
403 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
404 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
405 EQ("*-AA", "AA", "replace", "", "*-", 1)
406 EQ("AA", "AA", "replace", "", "*-", 0)
407
408 # single character deletion (from=="A", to=="")
409 EQ("", "A", "replace", "A", "")
410 EQ("", "AAA", "replace", "A", "")
411 EQ("", "AAA", "replace", "A", "", -1)
412 EQ("", "AAA", "replace", "A", "", sys.maxint)
413 EQ("", "AAA", "replace", "A", "", 4)
414 EQ("", "AAA", "replace", "A", "", 3)
415 EQ("A", "AAA", "replace", "A", "", 2)
416 EQ("AA", "AAA", "replace", "A", "", 1)
417 EQ("AAA", "AAA", "replace", "A", "", 0)
418 EQ("", "AAAAAAAAAA", "replace", "A", "")
419 EQ("BCD", "ABACADA", "replace", "A", "")
420 EQ("BCD", "ABACADA", "replace", "A", "", -1)
421 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxint)
422 EQ("BCD", "ABACADA", "replace", "A", "", 5)
423 EQ("BCD", "ABACADA", "replace", "A", "", 4)
424 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
425 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
426 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
427 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
428 EQ("BCD", "ABCAD", "replace", "A", "")
429 EQ("BCD", "ABCADAA", "replace", "A", "")
430 EQ("BCD", "BCD", "replace", "A", "")
431 EQ("*************", "*************", "replace", "A", "")
432 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
433
434 # substring deletion (from=="the", to=="")
435 EQ("", "the", "replace", "the", "")
436 EQ("ater", "theater", "replace", "the", "")
437 EQ("", "thethe", "replace", "the", "")
438 EQ("", "thethethethe", "replace", "the", "")
439 EQ("aaaa", "theatheatheathea", "replace", "the", "")
440 EQ("that", "that", "replace", "the", "")
441 EQ("thaet", "thaet", "replace", "the", "")
442 EQ("here and re", "here and there", "replace", "the", "")
443 EQ("here and re and re", "here and there and there",
444 "replace", "the", "", sys.maxint)
445 EQ("here and re and re", "here and there and there",
446 "replace", "the", "", -1)
447 EQ("here and re and re", "here and there and there",
448 "replace", "the", "", 3)
449 EQ("here and re and re", "here and there and there",
450 "replace", "the", "", 2)
451 EQ("here and re and there", "here and there and there",
452 "replace", "the", "", 1)
453 EQ("here and there and there", "here and there and there",
454 "replace", "the", "", 0)
455 EQ("here and re and re", "here and there and there", "replace", "the", "")
456
457 EQ("abc", "abc", "replace", "the", "")
458 EQ("abcdefg", "abcdefg", "replace", "the", "")
459
460 # substring deletion (from=="bob", to=="")
461 EQ("bob", "bbobob", "replace", "bob", "")
462 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
463 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
464 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000465
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000466 # single character replace in place (len(from)==len(to)==1)
467 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
468 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
469 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxint)
470 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
471 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
472 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
473 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
474 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
475
476 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
477 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
478 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
479 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
480 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
481
482 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000483
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000484 # substring replace in place (len(from)==len(to) > 1)
485 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
486 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxint)
487 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
488 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
489 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
490 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
491 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
492 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
493 EQ("cobob", "bobob", "replace", "bob", "cob")
494 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
495 EQ("bobob", "bobob", "replace", "bot", "bot")
496
497 # replace single character (len(from)==1, len(to)>1)
498 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
499 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
500 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxint)
501 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
502 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
503 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
504 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
505
506 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
507
508 # replace substring (len(from)>1, len(to)!=len(from))
509 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
510 "replace", "spam", "ham")
511 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
512 "replace", "spam", "ham", sys.maxint)
513 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
514 "replace", "spam", "ham", -1)
515 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
516 "replace", "spam", "ham", 4)
517 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
518 "replace", "spam", "ham", 3)
519 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
520 "replace", "spam", "ham", 2)
521 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
522 "replace", "spam", "ham", 1)
523 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
524 "replace", "spam", "ham", 0)
525
526 EQ("bobob", "bobobob", "replace", "bobob", "bob")
527 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
528 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000529
530 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000531 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
532 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
533 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
534 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
535 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
536 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
537 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
538 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
539 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
540 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
541 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
542 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
543 self.checkequal('', '', 'replace', '', '')
544 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
545 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
546 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
547 # MemoryError due to empty result (platform malloc issue when requesting
548 # 0 bytes).
549 self.checkequal('', '123', 'replace', '123', '')
550 self.checkequal('', '123123', 'replace', '123', '')
551 self.checkequal('x', '123x123', 'replace', '123', '')
552
553 self.checkraises(TypeError, 'hello', 'replace')
554 self.checkraises(TypeError, 'hello', 'replace', 42)
555 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
556 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
557
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000558 def test_replace_overflow(self):
559 # Check for overflow checking on 32 bit machines
560 if sys.maxint != 2147483647:
561 return
562 A2_16 = "A" * (2**16)
563 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
564 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
565 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000566
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000567 def test_zfill(self):
568 self.checkequal('123', '123', 'zfill', 2)
569 self.checkequal('123', '123', 'zfill', 3)
570 self.checkequal('0123', '123', 'zfill', 4)
571 self.checkequal('+123', '+123', 'zfill', 3)
572 self.checkequal('+123', '+123', 'zfill', 4)
573 self.checkequal('+0123', '+123', 'zfill', 5)
574 self.checkequal('-123', '-123', 'zfill', 3)
575 self.checkequal('-123', '-123', 'zfill', 4)
576 self.checkequal('-0123', '-123', 'zfill', 5)
577 self.checkequal('000', '', 'zfill', 3)
578 self.checkequal('34', '34', 'zfill', 1)
579 self.checkequal('0034', '34', 'zfill', 4)
580
581 self.checkraises(TypeError, '123', 'zfill')
582
583class MixinStrUnicodeUserStringTest:
584 # additional tests that only work for
585 # stringlike objects, i.e. str, unicode, UserString
586 # (but not the string module)
587
588 def test_islower(self):
589 self.checkequal(False, '', 'islower')
590 self.checkequal(True, 'a', 'islower')
591 self.checkequal(False, 'A', 'islower')
592 self.checkequal(False, '\n', 'islower')
593 self.checkequal(True, 'abc', 'islower')
594 self.checkequal(False, 'aBc', 'islower')
595 self.checkequal(True, 'abc\n', 'islower')
596 self.checkraises(TypeError, 'abc', 'islower', 42)
597
598 def test_isupper(self):
599 self.checkequal(False, '', 'isupper')
600 self.checkequal(False, 'a', 'isupper')
601 self.checkequal(True, 'A', 'isupper')
602 self.checkequal(False, '\n', 'isupper')
603 self.checkequal(True, 'ABC', 'isupper')
604 self.checkequal(False, 'AbC', 'isupper')
605 self.checkequal(True, 'ABC\n', 'isupper')
606 self.checkraises(TypeError, 'abc', 'isupper', 42)
607
608 def test_istitle(self):
609 self.checkequal(False, '', 'istitle')
610 self.checkequal(False, 'a', 'istitle')
611 self.checkequal(True, 'A', 'istitle')
612 self.checkequal(False, '\n', 'istitle')
613 self.checkequal(True, 'A Titlecased Line', 'istitle')
614 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
615 self.checkequal(True, 'A Titlecased, Line', 'istitle')
616 self.checkequal(False, 'Not a capitalized String', 'istitle')
617 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
618 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
619 self.checkequal(False, 'NOT', 'istitle')
620 self.checkraises(TypeError, 'abc', 'istitle', 42)
621
622 def test_isspace(self):
623 self.checkequal(False, '', 'isspace')
624 self.checkequal(False, 'a', 'isspace')
625 self.checkequal(True, ' ', 'isspace')
626 self.checkequal(True, '\t', 'isspace')
627 self.checkequal(True, '\r', 'isspace')
628 self.checkequal(True, '\n', 'isspace')
629 self.checkequal(True, ' \t\r\n', 'isspace')
630 self.checkequal(False, ' \t\r\na', 'isspace')
631 self.checkraises(TypeError, 'abc', 'isspace', 42)
632
633 def test_isalpha(self):
634 self.checkequal(False, '', 'isalpha')
635 self.checkequal(True, 'a', 'isalpha')
636 self.checkequal(True, 'A', 'isalpha')
637 self.checkequal(False, '\n', 'isalpha')
638 self.checkequal(True, 'abc', 'isalpha')
639 self.checkequal(False, 'aBc123', 'isalpha')
640 self.checkequal(False, 'abc\n', 'isalpha')
641 self.checkraises(TypeError, 'abc', 'isalpha', 42)
642
643 def test_isalnum(self):
644 self.checkequal(False, '', 'isalnum')
645 self.checkequal(True, 'a', 'isalnum')
646 self.checkequal(True, 'A', 'isalnum')
647 self.checkequal(False, '\n', 'isalnum')
648 self.checkequal(True, '123abc456', 'isalnum')
649 self.checkequal(True, 'a1b3c', 'isalnum')
650 self.checkequal(False, 'aBc000 ', 'isalnum')
651 self.checkequal(False, 'abc\n', 'isalnum')
652 self.checkraises(TypeError, 'abc', 'isalnum', 42)
653
654 def test_isdigit(self):
655 self.checkequal(False, '', 'isdigit')
656 self.checkequal(False, 'a', 'isdigit')
657 self.checkequal(True, '0', 'isdigit')
658 self.checkequal(True, '0123456789', 'isdigit')
659 self.checkequal(False, '0123456789a', 'isdigit')
660
661 self.checkraises(TypeError, 'abc', 'isdigit', 42)
662
663 def test_title(self):
664 self.checkequal(' Hello ', ' hello ', 'title')
665 self.checkequal('Hello ', 'hello ', 'title')
666 self.checkequal('Hello ', 'Hello ', 'title')
667 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
668 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
669 self.checkequal('Getint', "getInt", 'title')
670 self.checkraises(TypeError, 'hello', 'title', 42)
671
672 def test_splitlines(self):
673 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
674 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
675 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
676 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
677 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
678 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
679 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'], "\nabc\ndef\r\nghi\n\r", 'splitlines', 1)
680
681 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
682
683 def test_startswith(self):
684 self.checkequal(True, 'hello', 'startswith', 'he')
685 self.checkequal(True, 'hello', 'startswith', 'hello')
686 self.checkequal(False, 'hello', 'startswith', 'hello world')
687 self.checkequal(True, 'hello', 'startswith', '')
688 self.checkequal(False, 'hello', 'startswith', 'ello')
689 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
690 self.checkequal(True, 'hello', 'startswith', 'o', 4)
691 self.checkequal(False, 'hello', 'startswith', 'o', 5)
692 self.checkequal(True, 'hello', 'startswith', '', 5)
693 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
694 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
695 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
696 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
697
698 # test negative indices
699 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
700 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
701 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
702 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
703 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
704 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
705 self.checkequal(False, 'hello', 'startswith', 'o', -2)
706 self.checkequal(True, 'hello', 'startswith', 'o', -1)
707 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
708 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
709
710 self.checkraises(TypeError, 'hello', 'startswith')
711 self.checkraises(TypeError, 'hello', 'startswith', 42)
712
713 def test_endswith(self):
714 self.checkequal(True, 'hello', 'endswith', 'lo')
715 self.checkequal(False, 'hello', 'endswith', 'he')
716 self.checkequal(True, 'hello', 'endswith', '')
717 self.checkequal(False, 'hello', 'endswith', 'hello world')
718 self.checkequal(False, 'helloworld', 'endswith', 'worl')
719 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
720 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
721 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
722 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
723 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
724 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
725 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
726 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
727 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
728
729 # test negative indices
730 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
731 self.checkequal(False, 'hello', 'endswith', 'he', -2)
732 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
733 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
734 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
735 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
736 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
737 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
738 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
739 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
740 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
741 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
742 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
743
744 self.checkraises(TypeError, 'hello', 'endswith')
745 self.checkraises(TypeError, 'hello', 'endswith', 42)
746
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000747 def test___contains__(self):
748 self.checkequal(True, '', '__contains__', '') # vereq('' in '', True)
749 self.checkequal(True, 'abc', '__contains__', '') # vereq('' in 'abc', True)
750 self.checkequal(False, 'abc', '__contains__', '\0') # vereq('\0' in 'abc', False)
751 self.checkequal(True, '\0abc', '__contains__', '\0') # vereq('\0' in '\0abc', True)
752 self.checkequal(True, 'abc\0', '__contains__', '\0') # vereq('\0' in 'abc\0', True)
753 self.checkequal(True, '\0abc', '__contains__', 'a') # vereq('a' in '\0abc', True)
754 self.checkequal(True, 'asdf', '__contains__', 'asdf') # vereq('asdf' in 'asdf', True)
755 self.checkequal(False, 'asd', '__contains__', 'asdf') # vereq('asdf' in 'asd', False)
756 self.checkequal(False, '', '__contains__', 'asdf') # vereq('asdf' in '', False)
757
758 def test_subscript(self):
759 self.checkequal(u'a', 'abc', '__getitem__', 0)
760 self.checkequal(u'c', 'abc', '__getitem__', -1)
761 self.checkequal(u'a', 'abc', '__getitem__', 0L)
762 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 3))
763 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 1000))
764 self.checkequal(u'a', 'abc', '__getitem__', slice(0, 1))
765 self.checkequal(u'', 'abc', '__getitem__', slice(0, 0))
766 # FIXME What about negative indizes? This is handled differently by [] and __getitem__(slice)
767
768 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
769
770 def test_slice(self):
771 self.checkequal('abc', 'abc', '__getslice__', 0, 1000)
772 self.checkequal('abc', 'abc', '__getslice__', 0, 3)
773 self.checkequal('ab', 'abc', '__getslice__', 0, 2)
774 self.checkequal('bc', 'abc', '__getslice__', 1, 3)
775 self.checkequal('b', 'abc', '__getslice__', 1, 2)
776 self.checkequal('', 'abc', '__getslice__', 2, 2)
777 self.checkequal('', 'abc', '__getslice__', 1000, 1000)
778 self.checkequal('', 'abc', '__getslice__', 2000, 1000)
779 self.checkequal('', 'abc', '__getslice__', 2, 1)
780 # FIXME What about negative indizes? This is handled differently by [] and __getslice__
781
782 self.checkraises(TypeError, 'abc', '__getslice__', 'def')
783
784 def test_mul(self):
785 self.checkequal('', 'abc', '__mul__', -1)
786 self.checkequal('', 'abc', '__mul__', 0)
787 self.checkequal('abc', 'abc', '__mul__', 1)
788 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
789 self.checkraises(TypeError, 'abc', '__mul__')
790 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +0000791 # XXX: on a 64-bit system, this doesn't raise an overflow error,
792 # but either raises a MemoryError, or succeeds (if you have 54TiB)
793 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000794
795 def test_join(self):
796 # join now works with any sequence type
797 # moved here, because the argument order is
798 # different in string.join (see the test in
799 # test.test_string.StringTest.test_join)
800 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
801 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
802 self.checkequal('w x y z', ' ', 'join', Sequence())
803 self.checkequal('abc', 'a', 'join', ('abc',))
804 self.checkequal('z', 'a', 'join', UserList(['z']))
805 if test_support.have_unicode:
806 self.checkequal(unicode('a.b.c'), unicode('.'), 'join', ['a', 'b', 'c'])
807 self.checkequal(unicode('a.b.c'), '.', 'join', [unicode('a'), 'b', 'c'])
808 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', unicode('b'), 'c'])
809 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', 'b', unicode('c')])
810 self.checkraises(TypeError, '.', 'join', ['a', unicode('b'), 3])
811 for i in [5, 25, 125]:
812 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
813 ['a' * i] * i)
814 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
815 ('a' * i,) * i)
816
817 self.checkraises(TypeError, ' ', 'join', BadSeq1())
818 self.checkequal('a b c', ' ', 'join', BadSeq2())
819
820 self.checkraises(TypeError, ' ', 'join')
821 self.checkraises(TypeError, ' ', 'join', 7)
822 self.checkraises(TypeError, ' ', 'join', Sequence([7, 'hello', 123L]))
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +0000823 try:
824 def f():
825 yield 4 + ""
826 self.fixtype(' ').join(f())
827 except TypeError, e:
828 if '+' not in str(e):
829 self.fail('join() ate exception message')
830 else:
831 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000832
833 def test_formatting(self):
834 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
835 self.checkequal('+10+', '+%d+', '__mod__', 10)
836 self.checkequal('a', "%c", '__mod__', "a")
837 self.checkequal('a', "%c", '__mod__', "a")
838 self.checkequal('"', "%c", '__mod__', 34)
839 self.checkequal('$', "%c", '__mod__', 36)
840 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +0000841 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000842
843 for ordinal in (-100, 0x200000):
844 # unicode raises ValueError, str raises OverflowError
845 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
846
847 self.checkequal(' 42', '%3ld', '__mod__', 42)
848 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +0000849 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000850
851 self.checkraises(TypeError, 'abc', '__mod__')
852 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
853 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
854 self.checkraises(TypeError, '%c', '__mod__', (None,))
855 self.checkraises(ValueError, '%(foo', '__mod__', {})
856 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
857
858 # argument names with properly nested brackets are supported
859 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
860
861 # 100 is a magic number in PyUnicode_Format, this forces a resize
862 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
863
864 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
865 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
866 self.checkraises(ValueError, '%10', '__mod__', (42,))
867
868 def test_floatformatting(self):
869 # float formatting
870 for prec in xrange(100):
871 format = '%%.%if' % prec
872 value = 0.01
873 for x in xrange(60):
874 value = value * 3.141592655 / 3.0 * 10.0
875 # The formatfloat() code in stringobject.c and
876 # unicodeobject.c uses a 120 byte buffer and switches from
877 # 'f' formatting to 'g' at precision 50, so we expect
878 # OverflowErrors for the ranges x < 50 and prec >= 67.
879 if x < 50 and prec >= 67:
880 self.checkraises(OverflowError, format, "__mod__", value)
881 else:
882 self.checkcall(format, "__mod__", value)
883
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000884 def test_inplace_rewrites(self):
885 # Check that strings don't copy and modify cached single-character strings
886 self.checkequal('a', 'A', 'lower')
887 self.checkequal(True, 'A', 'isupper')
888 self.checkequal('A', 'a', 'upper')
889 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000890
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000891 self.checkequal('a', 'A', 'replace', 'A', 'a')
892 self.checkequal(True, 'A', 'isupper')
893
894 self.checkequal('A', 'a', 'capitalize')
895 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000896
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000897 self.checkequal('A', 'a', 'swapcase')
898 self.checkequal(True, 'a', 'islower')
899
900 self.checkequal('A', 'a', 'title')
901 self.checkequal(True, 'a', 'islower')
902
Walter Dörwald57d88e52004-08-26 16:53:04 +0000903
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000904class MixinStrStringUserStringTest:
905 # Additional tests for 8bit strings, i.e. str, UserString and
906 # the string module
907
908 def test_maketrans(self):
909 self.assertEqual(
910 ''.join(map(chr, xrange(256))).replace('abc', 'xyz'),
911 string.maketrans('abc', 'xyz')
912 )
913 self.assertRaises(ValueError, string.maketrans, 'abc', 'xyzw')
914
915 def test_translate(self):
916 table = string.maketrans('abc', 'xyz')
917 self.checkequal('xyzxyz', 'xyzabcdef', 'translate', table, 'def')
918
919 table = string.maketrans('a', 'A')
920 self.checkequal('Abc', 'abc', 'translate', table)
921 self.checkequal('xyz', 'xyz', 'translate', table)
922 self.checkequal('yz', 'xyz', 'translate', table, 'x')
923 self.checkraises(ValueError, 'xyz', 'translate', 'too short', 'strip')
924 self.checkraises(ValueError, 'xyz', 'translate', 'too short')
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +0000925
926
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000927class MixinStrUserStringTest:
928 # Additional tests that only work with
929 # 8bit compatible object, i.e. str and UserString
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +0000930
Walter Dörwald6eea7892005-07-28 16:49:15 +0000931 if test_support.have_unicode:
932 def test_encoding_decoding(self):
933 codecs = [('rot13', 'uryyb jbeyq'),
934 ('base64', 'aGVsbG8gd29ybGQ=\n'),
935 ('hex', '68656c6c6f20776f726c64'),
936 ('uu', 'begin 666 <data>\n+:&5L;&\\@=V]R;&0 \n \nend\n')]
937 for encoding, data in codecs:
938 self.checkequal(data, 'hello world', 'encode', encoding)
939 self.checkequal('hello world', data, 'decode', encoding)
940 # zlib is optional, so we make the test optional too...
941 try:
942 import zlib
943 except ImportError:
944 pass
945 else:
946 data = 'x\x9c\xcbH\xcd\xc9\xc9W(\xcf/\xcaI\x01\x00\x1a\x0b\x04]'
947 self.checkequal(data, 'hello world', 'encode', 'zlib')
948 self.checkequal('hello world', data, 'decode', 'zlib')
Walter Dörwald97951de2003-03-26 14:31:25 +0000949
Walter Dörwald6eea7892005-07-28 16:49:15 +0000950 self.checkraises(TypeError, 'xyz', 'decode', 42)
951 self.checkraises(TypeError, 'xyz', 'encode', 42)
Walter Dörwald57d88e52004-08-26 16:53:04 +0000952
953
954class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +0000955 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +0000956
957 def test_bug1001011(self):
958 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +0000959 # involving a subclass.
960 # Make sure that it is of the appropriate type.
961 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +0000962 t = self.type2test
963 class subclass(t):
964 pass
965 s1 = subclass("abcd")
966 s2 = t().join([s1])
967 self.assert_(s1 is not s2)
968 self.assert_(type(s2) is t)
Tim Peters108f1372004-08-27 05:36:07 +0000969
970 s1 = t("abcd")
971 s2 = t().join([s1])
972 self.assert_(s1 is s2)
973
974 # Should also test mixed-type join.
975 if t is unicode:
976 s1 = subclass("abcd")
977 s2 = "".join([s1])
978 self.assert_(s1 is not s2)
979 self.assert_(type(s2) is t)
980
981 s1 = t("abcd")
982 s2 = "".join([s1])
983 self.assert_(s1 is s2)
984
985 elif t is str:
986 s1 = subclass("abcd")
987 s2 = u"".join([s1])
988 self.assert_(s1 is not s2)
989 self.assert_(type(s2) is unicode) # promotes!
990
991 s1 = t("abcd")
992 s2 = u"".join([s1])
993 self.assert_(s1 is not s2)
994 self.assert_(type(s2) is unicode) # promotes!
995
996 else:
997 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)