blob: 2469830d01eeef0acb292dc41188d1f860296ee0 [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)
Andrew Dalke005aee22006-05-26 12:28:15 +0000246 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
247 sys.maxint-1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000248 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
249 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000250
Andrew Dalke984b9712006-05-26 11:11:38 +0000251 self.checkequal([], ' ', 'split')
252 self.checkequal(['a'], ' a ', 'split')
253 self.checkequal(['a', 'b'], ' a b ', 'split')
254 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
255 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
256 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
Andrew Dalke03fb4442006-05-26 11:15:22 +0000257 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
Andrew Dalke005aee22006-05-26 12:28:15 +0000258 aaa = ' a '*20
259 self.checkequal(['a']*20, aaa, 'split')
260 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
Andrew Dalke669fa182006-05-26 13:05:55 +0000261 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
Andrew Dalke984b9712006-05-26 11:11:38 +0000262
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000263 # by a char
264 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000265 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000266 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
267 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
268 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
269 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000270 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
271 sys.maxint-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000272 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
273 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
274 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000275 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
276 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000277 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
278
Andrew Dalke005aee22006-05-26 12:28:15 +0000279 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
280 self.checkequal(['a']*15 +['a|a|a|a|a'],
281 ('a|'*20)[:-1], 'split', '|', 15)
282
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000283 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000284 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000285 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
286 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
287 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
288 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000289 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
290 sys.maxint-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000291 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
292 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000293 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000294 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
295 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
296 'split', 'test')
297 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000298 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
299 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
300 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
301 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
302 self.checkequal([''], '', 'split', 'aaa')
303 self.checkequal(['aa'], 'aa', 'split', 'aaa')
Andrew Dalke5cc60092006-05-26 12:31:00 +0000304 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
305 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000306
307 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
308 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
309 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
310 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000311
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000312 # mixed use of str and unicode
313 self.checkequal([u'a', u'b', u'c d'], 'a b c d', 'split', u' ', 2)
314
315 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000316 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
317
Andrew Dalke005aee22006-05-26 12:28:15 +0000318 # null case
319 self.checkraises(ValueError, 'hello', 'split', '')
320 self.checkraises(ValueError, 'hello', 'split', '', 0)
321
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000322 def test_rsplit(self):
323 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
324 'this is the rsplit function', 'rsplit')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000325
326 # by whitespace
327 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000328 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
329 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
330 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
331 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000332 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
333 sys.maxint-20)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000334 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000335 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000336
Andrew Dalke669fa182006-05-26 13:05:55 +0000337 self.checkequal([], ' ', 'rsplit')
338 self.checkequal(['a'], ' a ', 'rsplit')
339 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
340 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
341 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
342 None, 1)
343 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
344 None, 2)
345 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
346 aaa = ' a '*20
347 self.checkequal(['a']*20, aaa, 'rsplit')
348 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
349 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
350
351
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000352 # by a char
353 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
354 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
355 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
356 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
357 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000358 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
359 sys.maxint-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000360 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
361 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
362 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Andrew Dalke669fa182006-05-26 13:05:55 +0000363 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
364 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
365
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000366 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
367
Andrew Dalke669fa182006-05-26 13:05:55 +0000368 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
369 self.checkequal(['a|a|a|a|a']+['a']*15,
370 ('a|'*20)[:-1], 'rsplit', '|', 15)
371
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000372 # by string
373 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
374 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
375 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
376 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
377 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000378 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
379 sys.maxint-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000380 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
381 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
382 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000383 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
384 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
385 'rsplit', 'test')
386 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
387 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
388 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
389 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
390 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
391 self.checkequal([''], '', 'rsplit', 'aaa')
392 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
393 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
394 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
395
396 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
397 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
398 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
399 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000400
401 # mixed use of str and unicode
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000402 self.checkequal([u'a b', u'c', u'd'], 'a b c d', 'rsplit', u' ', 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000403
404 # argument type
405 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000406
Andrew Dalke669fa182006-05-26 13:05:55 +0000407 # null case
408 self.checkraises(ValueError, 'hello', 'rsplit', '')
409 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
410
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000411 def test_strip(self):
412 self.checkequal('hello', ' hello ', 'strip')
413 self.checkequal('hello ', ' hello ', 'lstrip')
414 self.checkequal(' hello', ' hello ', 'rstrip')
415 self.checkequal('hello', 'hello', 'strip')
416
Neal Norwitzffe33b72003-04-10 22:35:32 +0000417 # strip/lstrip/rstrip with None arg
418 self.checkequal('hello', ' hello ', 'strip', None)
419 self.checkequal('hello ', ' hello ', 'lstrip', None)
420 self.checkequal(' hello', ' hello ', 'rstrip', None)
421 self.checkequal('hello', 'hello', 'strip', None)
422
423 # strip/lstrip/rstrip with str arg
424 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
425 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
426 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
427 self.checkequal('hello', 'hello', 'strip', 'xyz')
428
429 # strip/lstrip/rstrip with unicode arg
430 if test_support.have_unicode:
431 self.checkequal(unicode('hello', 'ascii'), 'xyzzyhelloxyzzy',
432 'strip', unicode('xyz', 'ascii'))
433 self.checkequal(unicode('helloxyzzy', 'ascii'), 'xyzzyhelloxyzzy',
434 'lstrip', unicode('xyz', 'ascii'))
435 self.checkequal(unicode('xyzzyhello', 'ascii'), 'xyzzyhelloxyzzy',
436 'rstrip', unicode('xyz', 'ascii'))
437 self.checkequal(unicode('hello', 'ascii'), 'hello',
438 'strip', unicode('xyz', 'ascii'))
439
440 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
441 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
442 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
443
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000444 def test_ljust(self):
445 self.checkequal('abc ', 'abc', 'ljust', 10)
446 self.checkequal('abc ', 'abc', 'ljust', 6)
447 self.checkequal('abc', 'abc', 'ljust', 3)
448 self.checkequal('abc', 'abc', 'ljust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000449 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000450 self.checkraises(TypeError, 'abc', 'ljust')
451
452 def test_rjust(self):
453 self.checkequal(' abc', 'abc', 'rjust', 10)
454 self.checkequal(' abc', 'abc', 'rjust', 6)
455 self.checkequal('abc', 'abc', 'rjust', 3)
456 self.checkequal('abc', 'abc', 'rjust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000457 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000458 self.checkraises(TypeError, 'abc', 'rjust')
459
460 def test_center(self):
461 self.checkequal(' abc ', 'abc', 'center', 10)
462 self.checkequal(' abc ', 'abc', 'center', 6)
463 self.checkequal('abc', 'abc', 'center', 3)
464 self.checkequal('abc', 'abc', 'center', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000465 self.checkequal('***abc****', 'abc', 'center', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000466 self.checkraises(TypeError, 'abc', 'center')
467
468 def test_swapcase(self):
469 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
470
471 self.checkraises(TypeError, 'hello', 'swapcase', 42)
472
473 def test_replace(self):
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000474 EQ = self.checkequal
475
476 # Operations on the empty string
477 EQ("", "", "replace", "", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000478
479 #EQ("A", "", "replace", "", "A")
480 # That was the correct result; this is the result we actually get
Tim Petersf4049082006-05-24 21:00:45 +0000481 # now (for str, but not for unicode):
482 #EQ("", "", "replace", "", "A")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000483
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000484 EQ("", "", "replace", "A", "")
485 EQ("", "", "replace", "A", "A")
486 EQ("", "", "replace", "", "", 100)
487 EQ("", "", "replace", "", "", sys.maxint)
488
489 # interleave (from=="", 'to' gets inserted everywhere)
490 EQ("A", "A", "replace", "", "")
491 EQ("*A*", "A", "replace", "", "*")
492 EQ("*1A*1", "A", "replace", "", "*1")
493 EQ("*-#A*-#", "A", "replace", "", "*-#")
494 EQ("*-A*-A*-", "AA", "replace", "", "*-")
495 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
496 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxint)
497 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
498 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
499 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
500 EQ("*-AA", "AA", "replace", "", "*-", 1)
501 EQ("AA", "AA", "replace", "", "*-", 0)
502
503 # single character deletion (from=="A", to=="")
504 EQ("", "A", "replace", "A", "")
505 EQ("", "AAA", "replace", "A", "")
506 EQ("", "AAA", "replace", "A", "", -1)
507 EQ("", "AAA", "replace", "A", "", sys.maxint)
508 EQ("", "AAA", "replace", "A", "", 4)
509 EQ("", "AAA", "replace", "A", "", 3)
510 EQ("A", "AAA", "replace", "A", "", 2)
511 EQ("AA", "AAA", "replace", "A", "", 1)
512 EQ("AAA", "AAA", "replace", "A", "", 0)
513 EQ("", "AAAAAAAAAA", "replace", "A", "")
514 EQ("BCD", "ABACADA", "replace", "A", "")
515 EQ("BCD", "ABACADA", "replace", "A", "", -1)
516 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxint)
517 EQ("BCD", "ABACADA", "replace", "A", "", 5)
518 EQ("BCD", "ABACADA", "replace", "A", "", 4)
519 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
520 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
521 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
522 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
523 EQ("BCD", "ABCAD", "replace", "A", "")
524 EQ("BCD", "ABCADAA", "replace", "A", "")
525 EQ("BCD", "BCD", "replace", "A", "")
526 EQ("*************", "*************", "replace", "A", "")
527 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
528
529 # substring deletion (from=="the", to=="")
530 EQ("", "the", "replace", "the", "")
531 EQ("ater", "theater", "replace", "the", "")
532 EQ("", "thethe", "replace", "the", "")
533 EQ("", "thethethethe", "replace", "the", "")
534 EQ("aaaa", "theatheatheathea", "replace", "the", "")
535 EQ("that", "that", "replace", "the", "")
536 EQ("thaet", "thaet", "replace", "the", "")
537 EQ("here and re", "here and there", "replace", "the", "")
538 EQ("here and re and re", "here and there and there",
539 "replace", "the", "", sys.maxint)
540 EQ("here and re and re", "here and there and there",
541 "replace", "the", "", -1)
542 EQ("here and re and re", "here and there and there",
543 "replace", "the", "", 3)
544 EQ("here and re and re", "here and there and there",
545 "replace", "the", "", 2)
546 EQ("here and re and there", "here and there and there",
547 "replace", "the", "", 1)
548 EQ("here and there and there", "here and there and there",
549 "replace", "the", "", 0)
550 EQ("here and re and re", "here and there and there", "replace", "the", "")
551
552 EQ("abc", "abc", "replace", "the", "")
553 EQ("abcdefg", "abcdefg", "replace", "the", "")
554
555 # substring deletion (from=="bob", to=="")
556 EQ("bob", "bbobob", "replace", "bob", "")
557 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
558 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
559 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000560
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000561 # single character replace in place (len(from)==len(to)==1)
562 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
563 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
564 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxint)
565 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
566 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
567 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
568 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
569 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
570
571 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
572 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
573 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
574 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
575 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
576
577 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000578
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000579 # substring replace in place (len(from)==len(to) > 1)
580 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
581 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxint)
582 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
583 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
584 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
585 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
586 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
587 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
588 EQ("cobob", "bobob", "replace", "bob", "cob")
589 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
590 EQ("bobob", "bobob", "replace", "bot", "bot")
591
592 # replace single character (len(from)==1, len(to)>1)
593 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
594 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
595 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxint)
596 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
597 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
598 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
599 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
600
601 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
602
603 # replace substring (len(from)>1, len(to)!=len(from))
604 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
605 "replace", "spam", "ham")
606 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
607 "replace", "spam", "ham", sys.maxint)
608 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
609 "replace", "spam", "ham", -1)
610 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
611 "replace", "spam", "ham", 4)
612 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
613 "replace", "spam", "ham", 3)
614 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
615 "replace", "spam", "ham", 2)
616 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
617 "replace", "spam", "ham", 1)
618 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
619 "replace", "spam", "ham", 0)
620
621 EQ("bobob", "bobobob", "replace", "bobob", "bob")
622 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
623 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000624
625 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000626 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
627 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
628 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
629 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
630 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
631 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
632 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
633 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
634 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
635 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
636 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
637 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
638 self.checkequal('', '', 'replace', '', '')
639 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
640 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
641 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
642 # MemoryError due to empty result (platform malloc issue when requesting
643 # 0 bytes).
644 self.checkequal('', '123', 'replace', '123', '')
645 self.checkequal('', '123123', 'replace', '123', '')
646 self.checkequal('x', '123x123', 'replace', '123', '')
647
648 self.checkraises(TypeError, 'hello', 'replace')
649 self.checkraises(TypeError, 'hello', 'replace', 42)
650 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
651 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
652
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000653 def test_replace_overflow(self):
654 # Check for overflow checking on 32 bit machines
655 if sys.maxint != 2147483647:
656 return
657 A2_16 = "A" * (2**16)
658 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
659 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
660 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000661
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000662 def test_zfill(self):
663 self.checkequal('123', '123', 'zfill', 2)
664 self.checkequal('123', '123', 'zfill', 3)
665 self.checkequal('0123', '123', 'zfill', 4)
666 self.checkequal('+123', '+123', 'zfill', 3)
667 self.checkequal('+123', '+123', 'zfill', 4)
668 self.checkequal('+0123', '+123', 'zfill', 5)
669 self.checkequal('-123', '-123', 'zfill', 3)
670 self.checkequal('-123', '-123', 'zfill', 4)
671 self.checkequal('-0123', '-123', 'zfill', 5)
672 self.checkequal('000', '', 'zfill', 3)
673 self.checkequal('34', '34', 'zfill', 1)
674 self.checkequal('0034', '34', 'zfill', 4)
675
676 self.checkraises(TypeError, '123', 'zfill')
677
678class MixinStrUnicodeUserStringTest:
679 # additional tests that only work for
680 # stringlike objects, i.e. str, unicode, UserString
681 # (but not the string module)
682
683 def test_islower(self):
684 self.checkequal(False, '', 'islower')
685 self.checkequal(True, 'a', 'islower')
686 self.checkequal(False, 'A', 'islower')
687 self.checkequal(False, '\n', 'islower')
688 self.checkequal(True, 'abc', 'islower')
689 self.checkequal(False, 'aBc', 'islower')
690 self.checkequal(True, 'abc\n', 'islower')
691 self.checkraises(TypeError, 'abc', 'islower', 42)
692
693 def test_isupper(self):
694 self.checkequal(False, '', 'isupper')
695 self.checkequal(False, 'a', 'isupper')
696 self.checkequal(True, 'A', 'isupper')
697 self.checkequal(False, '\n', 'isupper')
698 self.checkequal(True, 'ABC', 'isupper')
699 self.checkequal(False, 'AbC', 'isupper')
700 self.checkequal(True, 'ABC\n', 'isupper')
701 self.checkraises(TypeError, 'abc', 'isupper', 42)
702
703 def test_istitle(self):
704 self.checkequal(False, '', 'istitle')
705 self.checkequal(False, 'a', 'istitle')
706 self.checkequal(True, 'A', 'istitle')
707 self.checkequal(False, '\n', 'istitle')
708 self.checkequal(True, 'A Titlecased Line', 'istitle')
709 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
710 self.checkequal(True, 'A Titlecased, Line', 'istitle')
711 self.checkequal(False, 'Not a capitalized String', 'istitle')
712 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
713 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
714 self.checkequal(False, 'NOT', 'istitle')
715 self.checkraises(TypeError, 'abc', 'istitle', 42)
716
717 def test_isspace(self):
718 self.checkequal(False, '', 'isspace')
719 self.checkequal(False, 'a', 'isspace')
720 self.checkequal(True, ' ', 'isspace')
721 self.checkequal(True, '\t', 'isspace')
722 self.checkequal(True, '\r', 'isspace')
723 self.checkequal(True, '\n', 'isspace')
724 self.checkequal(True, ' \t\r\n', 'isspace')
725 self.checkequal(False, ' \t\r\na', 'isspace')
726 self.checkraises(TypeError, 'abc', 'isspace', 42)
727
728 def test_isalpha(self):
729 self.checkequal(False, '', 'isalpha')
730 self.checkequal(True, 'a', 'isalpha')
731 self.checkequal(True, 'A', 'isalpha')
732 self.checkequal(False, '\n', 'isalpha')
733 self.checkequal(True, 'abc', 'isalpha')
734 self.checkequal(False, 'aBc123', 'isalpha')
735 self.checkequal(False, 'abc\n', 'isalpha')
736 self.checkraises(TypeError, 'abc', 'isalpha', 42)
737
738 def test_isalnum(self):
739 self.checkequal(False, '', 'isalnum')
740 self.checkequal(True, 'a', 'isalnum')
741 self.checkequal(True, 'A', 'isalnum')
742 self.checkequal(False, '\n', 'isalnum')
743 self.checkequal(True, '123abc456', 'isalnum')
744 self.checkequal(True, 'a1b3c', 'isalnum')
745 self.checkequal(False, 'aBc000 ', 'isalnum')
746 self.checkequal(False, 'abc\n', 'isalnum')
747 self.checkraises(TypeError, 'abc', 'isalnum', 42)
748
749 def test_isdigit(self):
750 self.checkequal(False, '', 'isdigit')
751 self.checkequal(False, 'a', 'isdigit')
752 self.checkequal(True, '0', 'isdigit')
753 self.checkequal(True, '0123456789', 'isdigit')
754 self.checkequal(False, '0123456789a', 'isdigit')
755
756 self.checkraises(TypeError, 'abc', 'isdigit', 42)
757
758 def test_title(self):
759 self.checkequal(' Hello ', ' hello ', 'title')
760 self.checkequal('Hello ', 'hello ', 'title')
761 self.checkequal('Hello ', 'Hello ', 'title')
762 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
763 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
764 self.checkequal('Getint', "getInt", 'title')
765 self.checkraises(TypeError, 'hello', 'title', 42)
766
767 def test_splitlines(self):
768 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
769 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
770 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
771 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
772 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
773 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
774 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'], "\nabc\ndef\r\nghi\n\r", 'splitlines', 1)
775
776 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
777
778 def test_startswith(self):
779 self.checkequal(True, 'hello', 'startswith', 'he')
780 self.checkequal(True, 'hello', 'startswith', 'hello')
781 self.checkequal(False, 'hello', 'startswith', 'hello world')
782 self.checkequal(True, 'hello', 'startswith', '')
783 self.checkequal(False, 'hello', 'startswith', 'ello')
784 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
785 self.checkequal(True, 'hello', 'startswith', 'o', 4)
786 self.checkequal(False, 'hello', 'startswith', 'o', 5)
787 self.checkequal(True, 'hello', 'startswith', '', 5)
788 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
789 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
790 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
791 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
792
793 # test negative indices
794 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
795 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
796 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
797 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
798 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
799 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
800 self.checkequal(False, 'hello', 'startswith', 'o', -2)
801 self.checkequal(True, 'hello', 'startswith', 'o', -1)
802 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
803 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
804
805 self.checkraises(TypeError, 'hello', 'startswith')
806 self.checkraises(TypeError, 'hello', 'startswith', 42)
807
808 def test_endswith(self):
809 self.checkequal(True, 'hello', 'endswith', 'lo')
810 self.checkequal(False, 'hello', 'endswith', 'he')
811 self.checkequal(True, 'hello', 'endswith', '')
812 self.checkequal(False, 'hello', 'endswith', 'hello world')
813 self.checkequal(False, 'helloworld', 'endswith', 'worl')
814 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
815 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
816 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
817 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
818 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
819 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
820 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
821 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
822 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
823
824 # test negative indices
825 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
826 self.checkequal(False, 'hello', 'endswith', 'he', -2)
827 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
828 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
829 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
830 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
831 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
832 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
833 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
834 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
835 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
836 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
837 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
838
839 self.checkraises(TypeError, 'hello', 'endswith')
840 self.checkraises(TypeError, 'hello', 'endswith', 42)
841
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000842 def test___contains__(self):
843 self.checkequal(True, '', '__contains__', '') # vereq('' in '', True)
844 self.checkequal(True, 'abc', '__contains__', '') # vereq('' in 'abc', True)
845 self.checkequal(False, 'abc', '__contains__', '\0') # vereq('\0' in 'abc', False)
846 self.checkequal(True, '\0abc', '__contains__', '\0') # vereq('\0' in '\0abc', True)
847 self.checkequal(True, 'abc\0', '__contains__', '\0') # vereq('\0' in 'abc\0', True)
848 self.checkequal(True, '\0abc', '__contains__', 'a') # vereq('a' in '\0abc', True)
849 self.checkequal(True, 'asdf', '__contains__', 'asdf') # vereq('asdf' in 'asdf', True)
850 self.checkequal(False, 'asd', '__contains__', 'asdf') # vereq('asdf' in 'asd', False)
851 self.checkequal(False, '', '__contains__', 'asdf') # vereq('asdf' in '', False)
852
853 def test_subscript(self):
854 self.checkequal(u'a', 'abc', '__getitem__', 0)
855 self.checkequal(u'c', 'abc', '__getitem__', -1)
856 self.checkequal(u'a', 'abc', '__getitem__', 0L)
857 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 3))
858 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 1000))
859 self.checkequal(u'a', 'abc', '__getitem__', slice(0, 1))
860 self.checkequal(u'', 'abc', '__getitem__', slice(0, 0))
861 # FIXME What about negative indizes? This is handled differently by [] and __getitem__(slice)
862
863 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
864
865 def test_slice(self):
866 self.checkequal('abc', 'abc', '__getslice__', 0, 1000)
867 self.checkequal('abc', 'abc', '__getslice__', 0, 3)
868 self.checkequal('ab', 'abc', '__getslice__', 0, 2)
869 self.checkequal('bc', 'abc', '__getslice__', 1, 3)
870 self.checkequal('b', 'abc', '__getslice__', 1, 2)
871 self.checkequal('', 'abc', '__getslice__', 2, 2)
872 self.checkequal('', 'abc', '__getslice__', 1000, 1000)
873 self.checkequal('', 'abc', '__getslice__', 2000, 1000)
874 self.checkequal('', 'abc', '__getslice__', 2, 1)
875 # FIXME What about negative indizes? This is handled differently by [] and __getslice__
876
877 self.checkraises(TypeError, 'abc', '__getslice__', 'def')
878
879 def test_mul(self):
880 self.checkequal('', 'abc', '__mul__', -1)
881 self.checkequal('', 'abc', '__mul__', 0)
882 self.checkequal('abc', 'abc', '__mul__', 1)
883 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
884 self.checkraises(TypeError, 'abc', '__mul__')
885 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +0000886 # XXX: on a 64-bit system, this doesn't raise an overflow error,
887 # but either raises a MemoryError, or succeeds (if you have 54TiB)
888 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000889
890 def test_join(self):
891 # join now works with any sequence type
892 # moved here, because the argument order is
893 # different in string.join (see the test in
894 # test.test_string.StringTest.test_join)
895 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
896 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
897 self.checkequal('w x y z', ' ', 'join', Sequence())
898 self.checkequal('abc', 'a', 'join', ('abc',))
899 self.checkequal('z', 'a', 'join', UserList(['z']))
900 if test_support.have_unicode:
901 self.checkequal(unicode('a.b.c'), unicode('.'), 'join', ['a', 'b', 'c'])
902 self.checkequal(unicode('a.b.c'), '.', 'join', [unicode('a'), 'b', 'c'])
903 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', unicode('b'), 'c'])
904 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', 'b', unicode('c')])
905 self.checkraises(TypeError, '.', 'join', ['a', unicode('b'), 3])
906 for i in [5, 25, 125]:
907 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
908 ['a' * i] * i)
909 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
910 ('a' * i,) * i)
911
912 self.checkraises(TypeError, ' ', 'join', BadSeq1())
913 self.checkequal('a b c', ' ', 'join', BadSeq2())
914
915 self.checkraises(TypeError, ' ', 'join')
916 self.checkraises(TypeError, ' ', 'join', 7)
917 self.checkraises(TypeError, ' ', 'join', Sequence([7, 'hello', 123L]))
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +0000918 try:
919 def f():
920 yield 4 + ""
921 self.fixtype(' ').join(f())
922 except TypeError, e:
923 if '+' not in str(e):
924 self.fail('join() ate exception message')
925 else:
926 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000927
928 def test_formatting(self):
929 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
930 self.checkequal('+10+', '+%d+', '__mod__', 10)
931 self.checkequal('a', "%c", '__mod__', "a")
932 self.checkequal('a', "%c", '__mod__', "a")
933 self.checkequal('"', "%c", '__mod__', 34)
934 self.checkequal('$', "%c", '__mod__', 36)
935 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +0000936 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000937
938 for ordinal in (-100, 0x200000):
939 # unicode raises ValueError, str raises OverflowError
940 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
941
942 self.checkequal(' 42', '%3ld', '__mod__', 42)
943 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +0000944 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000945
946 self.checkraises(TypeError, 'abc', '__mod__')
947 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
948 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
949 self.checkraises(TypeError, '%c', '__mod__', (None,))
950 self.checkraises(ValueError, '%(foo', '__mod__', {})
951 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
952
953 # argument names with properly nested brackets are supported
954 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
955
956 # 100 is a magic number in PyUnicode_Format, this forces a resize
957 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
958
959 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
960 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
961 self.checkraises(ValueError, '%10', '__mod__', (42,))
962
963 def test_floatformatting(self):
964 # float formatting
965 for prec in xrange(100):
966 format = '%%.%if' % prec
967 value = 0.01
968 for x in xrange(60):
969 value = value * 3.141592655 / 3.0 * 10.0
970 # The formatfloat() code in stringobject.c and
971 # unicodeobject.c uses a 120 byte buffer and switches from
972 # 'f' formatting to 'g' at precision 50, so we expect
973 # OverflowErrors for the ranges x < 50 and prec >= 67.
974 if x < 50 and prec >= 67:
975 self.checkraises(OverflowError, format, "__mod__", value)
976 else:
977 self.checkcall(format, "__mod__", value)
978
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000979 def test_inplace_rewrites(self):
980 # Check that strings don't copy and modify cached single-character strings
981 self.checkequal('a', 'A', 'lower')
982 self.checkequal(True, 'A', 'isupper')
983 self.checkequal('A', 'a', 'upper')
984 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000985
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000986 self.checkequal('a', 'A', 'replace', 'A', 'a')
987 self.checkequal(True, 'A', 'isupper')
988
989 self.checkequal('A', 'a', 'capitalize')
990 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000991
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000992 self.checkequal('A', 'a', 'swapcase')
993 self.checkequal(True, 'a', 'islower')
994
995 self.checkequal('A', 'a', 'title')
996 self.checkequal(True, 'a', 'islower')
997
Fredrik Lundh06a69dd2006-05-26 08:54:28 +0000998 def test_partition(self):
999
1000 self.checkequal(('this', ' is ', 'the partition method'),
1001 'this is the partition method', 'partition', ' is ')
1002
1003 # from raymond's original specification
1004 S = 'http://www.python.org'
1005 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1006 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1007 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1008 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1009
1010 self.checkraises(ValueError, S, 'partition', '')
1011 self.checkraises(TypeError, S, 'partition', None)
1012
Walter Dörwald57d88e52004-08-26 16:53:04 +00001013
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001014class MixinStrStringUserStringTest:
1015 # Additional tests for 8bit strings, i.e. str, UserString and
1016 # the string module
1017
1018 def test_maketrans(self):
1019 self.assertEqual(
1020 ''.join(map(chr, xrange(256))).replace('abc', 'xyz'),
1021 string.maketrans('abc', 'xyz')
1022 )
1023 self.assertRaises(ValueError, string.maketrans, 'abc', 'xyzw')
1024
1025 def test_translate(self):
1026 table = string.maketrans('abc', 'xyz')
1027 self.checkequal('xyzxyz', 'xyzabcdef', 'translate', table, 'def')
1028
1029 table = string.maketrans('a', 'A')
1030 self.checkequal('Abc', 'abc', 'translate', table)
1031 self.checkequal('xyz', 'xyz', 'translate', table)
1032 self.checkequal('yz', 'xyz', 'translate', table, 'x')
1033 self.checkraises(ValueError, 'xyz', 'translate', 'too short', 'strip')
1034 self.checkraises(ValueError, 'xyz', 'translate', 'too short')
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001035
1036
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001037class MixinStrUserStringTest:
1038 # Additional tests that only work with
1039 # 8bit compatible object, i.e. str and UserString
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001040
Walter Dörwald6eea7892005-07-28 16:49:15 +00001041 if test_support.have_unicode:
1042 def test_encoding_decoding(self):
1043 codecs = [('rot13', 'uryyb jbeyq'),
1044 ('base64', 'aGVsbG8gd29ybGQ=\n'),
1045 ('hex', '68656c6c6f20776f726c64'),
1046 ('uu', 'begin 666 <data>\n+:&5L;&\\@=V]R;&0 \n \nend\n')]
1047 for encoding, data in codecs:
1048 self.checkequal(data, 'hello world', 'encode', encoding)
1049 self.checkequal('hello world', data, 'decode', encoding)
1050 # zlib is optional, so we make the test optional too...
1051 try:
1052 import zlib
1053 except ImportError:
1054 pass
1055 else:
1056 data = 'x\x9c\xcbH\xcd\xc9\xc9W(\xcf/\xcaI\x01\x00\x1a\x0b\x04]'
1057 self.checkequal(data, 'hello world', 'encode', 'zlib')
1058 self.checkequal('hello world', data, 'decode', 'zlib')
Walter Dörwald97951de2003-03-26 14:31:25 +00001059
Walter Dörwald6eea7892005-07-28 16:49:15 +00001060 self.checkraises(TypeError, 'xyz', 'decode', 42)
1061 self.checkraises(TypeError, 'xyz', 'encode', 42)
Walter Dörwald57d88e52004-08-26 16:53:04 +00001062
1063
1064class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001065 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001066
1067 def test_bug1001011(self):
1068 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001069 # involving a subclass.
1070 # Make sure that it is of the appropriate type.
1071 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001072 t = self.type2test
1073 class subclass(t):
1074 pass
1075 s1 = subclass("abcd")
1076 s2 = t().join([s1])
1077 self.assert_(s1 is not s2)
1078 self.assert_(type(s2) is t)
Tim Peters108f1372004-08-27 05:36:07 +00001079
1080 s1 = t("abcd")
1081 s2 = t().join([s1])
1082 self.assert_(s1 is s2)
1083
1084 # Should also test mixed-type join.
1085 if t is unicode:
1086 s1 = subclass("abcd")
1087 s2 = "".join([s1])
1088 self.assert_(s1 is not s2)
1089 self.assert_(type(s2) is t)
1090
1091 s1 = t("abcd")
1092 s2 = "".join([s1])
1093 self.assert_(s1 is s2)
1094
1095 elif t is str:
1096 s1 = subclass("abcd")
1097 s2 = u"".join([s1])
1098 self.assert_(s1 is not s2)
1099 self.assert_(type(s2) is unicode) # promotes!
1100
1101 s1 = t("abcd")
1102 s2 = u"".join([s1])
1103 self.assert_(s1 is not s2)
1104 self.assert_(type(s2) is unicode) # promotes!
1105
1106 else:
1107 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)