blob: 3a97462a6a7871c42e3765c2e8f1006a1c8eaba4 [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)
261 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 Dalke005aee22006-05-26 12:28:15 +0000294 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
295 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
296 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
297 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
298 self.checkequal([''], '', 'split', 'aaa')
299 self.checkequal(['aa'], 'aa', 'split', 'aaa')
Andrew Dalke5cc60092006-05-26 12:31:00 +0000300 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
301 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000302
303 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
304 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
305 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
306 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000307
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000308 # mixed use of str and unicode
309 self.checkequal([u'a', u'b', u'c d'], 'a b c d', 'split', u' ', 2)
310
311 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000312 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
313
Andrew Dalke005aee22006-05-26 12:28:15 +0000314 # null case
315 self.checkraises(ValueError, 'hello', 'split', '')
316 self.checkraises(ValueError, 'hello', 'split', '', 0)
317
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000318 def test_rsplit(self):
319 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
320 'this is the rsplit function', 'rsplit')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000321
322 # by whitespace
323 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000324 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
325 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
326 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
327 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
328 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000329 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000330
331 # by a char
332 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
333 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
334 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
335 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
336 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
337 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
338 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
339 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
340 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
341
342 # by string
343 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
344 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
345 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
346 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
347 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
348 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
349 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
350 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
351
352 # mixed use of str and unicode
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000353 self.checkequal([u'a b', u'c', u'd'], 'a b c d', 'rsplit', u' ', 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000354
355 # argument type
356 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000357
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000358 def test_strip(self):
359 self.checkequal('hello', ' hello ', 'strip')
360 self.checkequal('hello ', ' hello ', 'lstrip')
361 self.checkequal(' hello', ' hello ', 'rstrip')
362 self.checkequal('hello', 'hello', 'strip')
363
Neal Norwitzffe33b72003-04-10 22:35:32 +0000364 # strip/lstrip/rstrip with None arg
365 self.checkequal('hello', ' hello ', 'strip', None)
366 self.checkequal('hello ', ' hello ', 'lstrip', None)
367 self.checkequal(' hello', ' hello ', 'rstrip', None)
368 self.checkequal('hello', 'hello', 'strip', None)
369
370 # strip/lstrip/rstrip with str arg
371 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
372 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
373 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
374 self.checkequal('hello', 'hello', 'strip', 'xyz')
375
376 # strip/lstrip/rstrip with unicode arg
377 if test_support.have_unicode:
378 self.checkequal(unicode('hello', 'ascii'), 'xyzzyhelloxyzzy',
379 'strip', unicode('xyz', 'ascii'))
380 self.checkequal(unicode('helloxyzzy', 'ascii'), 'xyzzyhelloxyzzy',
381 'lstrip', unicode('xyz', 'ascii'))
382 self.checkequal(unicode('xyzzyhello', 'ascii'), 'xyzzyhelloxyzzy',
383 'rstrip', unicode('xyz', 'ascii'))
384 self.checkequal(unicode('hello', 'ascii'), 'hello',
385 'strip', unicode('xyz', 'ascii'))
386
387 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
388 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
389 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
390
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000391 def test_ljust(self):
392 self.checkequal('abc ', 'abc', 'ljust', 10)
393 self.checkequal('abc ', 'abc', 'ljust', 6)
394 self.checkequal('abc', 'abc', 'ljust', 3)
395 self.checkequal('abc', 'abc', 'ljust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000396 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000397 self.checkraises(TypeError, 'abc', 'ljust')
398
399 def test_rjust(self):
400 self.checkequal(' abc', 'abc', 'rjust', 10)
401 self.checkequal(' abc', 'abc', 'rjust', 6)
402 self.checkequal('abc', 'abc', 'rjust', 3)
403 self.checkequal('abc', 'abc', 'rjust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000404 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000405 self.checkraises(TypeError, 'abc', 'rjust')
406
407 def test_center(self):
408 self.checkequal(' abc ', 'abc', 'center', 10)
409 self.checkequal(' abc ', 'abc', 'center', 6)
410 self.checkequal('abc', 'abc', 'center', 3)
411 self.checkequal('abc', 'abc', 'center', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000412 self.checkequal('***abc****', 'abc', 'center', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000413 self.checkraises(TypeError, 'abc', 'center')
414
415 def test_swapcase(self):
416 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
417
418 self.checkraises(TypeError, 'hello', 'swapcase', 42)
419
420 def test_replace(self):
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000421 EQ = self.checkequal
422
423 # Operations on the empty string
424 EQ("", "", "replace", "", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000425
426 #EQ("A", "", "replace", "", "A")
427 # That was the correct result; this is the result we actually get
Tim Petersf4049082006-05-24 21:00:45 +0000428 # now (for str, but not for unicode):
429 #EQ("", "", "replace", "", "A")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000430
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000431 EQ("", "", "replace", "A", "")
432 EQ("", "", "replace", "A", "A")
433 EQ("", "", "replace", "", "", 100)
434 EQ("", "", "replace", "", "", sys.maxint)
435
436 # interleave (from=="", 'to' gets inserted everywhere)
437 EQ("A", "A", "replace", "", "")
438 EQ("*A*", "A", "replace", "", "*")
439 EQ("*1A*1", "A", "replace", "", "*1")
440 EQ("*-#A*-#", "A", "replace", "", "*-#")
441 EQ("*-A*-A*-", "AA", "replace", "", "*-")
442 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
443 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxint)
444 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
445 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
446 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
447 EQ("*-AA", "AA", "replace", "", "*-", 1)
448 EQ("AA", "AA", "replace", "", "*-", 0)
449
450 # single character deletion (from=="A", to=="")
451 EQ("", "A", "replace", "A", "")
452 EQ("", "AAA", "replace", "A", "")
453 EQ("", "AAA", "replace", "A", "", -1)
454 EQ("", "AAA", "replace", "A", "", sys.maxint)
455 EQ("", "AAA", "replace", "A", "", 4)
456 EQ("", "AAA", "replace", "A", "", 3)
457 EQ("A", "AAA", "replace", "A", "", 2)
458 EQ("AA", "AAA", "replace", "A", "", 1)
459 EQ("AAA", "AAA", "replace", "A", "", 0)
460 EQ("", "AAAAAAAAAA", "replace", "A", "")
461 EQ("BCD", "ABACADA", "replace", "A", "")
462 EQ("BCD", "ABACADA", "replace", "A", "", -1)
463 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxint)
464 EQ("BCD", "ABACADA", "replace", "A", "", 5)
465 EQ("BCD", "ABACADA", "replace", "A", "", 4)
466 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
467 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
468 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
469 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
470 EQ("BCD", "ABCAD", "replace", "A", "")
471 EQ("BCD", "ABCADAA", "replace", "A", "")
472 EQ("BCD", "BCD", "replace", "A", "")
473 EQ("*************", "*************", "replace", "A", "")
474 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
475
476 # substring deletion (from=="the", to=="")
477 EQ("", "the", "replace", "the", "")
478 EQ("ater", "theater", "replace", "the", "")
479 EQ("", "thethe", "replace", "the", "")
480 EQ("", "thethethethe", "replace", "the", "")
481 EQ("aaaa", "theatheatheathea", "replace", "the", "")
482 EQ("that", "that", "replace", "the", "")
483 EQ("thaet", "thaet", "replace", "the", "")
484 EQ("here and re", "here and there", "replace", "the", "")
485 EQ("here and re and re", "here and there and there",
486 "replace", "the", "", sys.maxint)
487 EQ("here and re and re", "here and there and there",
488 "replace", "the", "", -1)
489 EQ("here and re and re", "here and there and there",
490 "replace", "the", "", 3)
491 EQ("here and re and re", "here and there and there",
492 "replace", "the", "", 2)
493 EQ("here and re and there", "here and there and there",
494 "replace", "the", "", 1)
495 EQ("here and there and there", "here and there and there",
496 "replace", "the", "", 0)
497 EQ("here and re and re", "here and there and there", "replace", "the", "")
498
499 EQ("abc", "abc", "replace", "the", "")
500 EQ("abcdefg", "abcdefg", "replace", "the", "")
501
502 # substring deletion (from=="bob", to=="")
503 EQ("bob", "bbobob", "replace", "bob", "")
504 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
505 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
506 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000507
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000508 # single character replace in place (len(from)==len(to)==1)
509 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
510 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
511 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxint)
512 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
513 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
514 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
515 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
516 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
517
518 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
519 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
520 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
521 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
522 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
523
524 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000525
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000526 # substring replace in place (len(from)==len(to) > 1)
527 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
528 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxint)
529 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
530 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
531 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
532 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
533 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
534 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
535 EQ("cobob", "bobob", "replace", "bob", "cob")
536 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
537 EQ("bobob", "bobob", "replace", "bot", "bot")
538
539 # replace single character (len(from)==1, len(to)>1)
540 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
541 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
542 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxint)
543 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
544 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
545 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
546 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
547
548 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
549
550 # replace substring (len(from)>1, len(to)!=len(from))
551 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
552 "replace", "spam", "ham")
553 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
554 "replace", "spam", "ham", sys.maxint)
555 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
556 "replace", "spam", "ham", -1)
557 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
558 "replace", "spam", "ham", 4)
559 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
560 "replace", "spam", "ham", 3)
561 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
562 "replace", "spam", "ham", 2)
563 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
564 "replace", "spam", "ham", 1)
565 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
566 "replace", "spam", "ham", 0)
567
568 EQ("bobob", "bobobob", "replace", "bobob", "bob")
569 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
570 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000571
572 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000573 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
574 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
575 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
576 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
577 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
578 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
579 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
580 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
581 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
582 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
583 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
584 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
585 self.checkequal('', '', 'replace', '', '')
586 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
587 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
588 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
589 # MemoryError due to empty result (platform malloc issue when requesting
590 # 0 bytes).
591 self.checkequal('', '123', 'replace', '123', '')
592 self.checkequal('', '123123', 'replace', '123', '')
593 self.checkequal('x', '123x123', 'replace', '123', '')
594
595 self.checkraises(TypeError, 'hello', 'replace')
596 self.checkraises(TypeError, 'hello', 'replace', 42)
597 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
598 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
599
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000600 def test_replace_overflow(self):
601 # Check for overflow checking on 32 bit machines
602 if sys.maxint != 2147483647:
603 return
604 A2_16 = "A" * (2**16)
605 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
606 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
607 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000608
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000609 def test_zfill(self):
610 self.checkequal('123', '123', 'zfill', 2)
611 self.checkequal('123', '123', 'zfill', 3)
612 self.checkequal('0123', '123', 'zfill', 4)
613 self.checkequal('+123', '+123', 'zfill', 3)
614 self.checkequal('+123', '+123', 'zfill', 4)
615 self.checkequal('+0123', '+123', 'zfill', 5)
616 self.checkequal('-123', '-123', 'zfill', 3)
617 self.checkequal('-123', '-123', 'zfill', 4)
618 self.checkequal('-0123', '-123', 'zfill', 5)
619 self.checkequal('000', '', 'zfill', 3)
620 self.checkequal('34', '34', 'zfill', 1)
621 self.checkequal('0034', '34', 'zfill', 4)
622
623 self.checkraises(TypeError, '123', 'zfill')
624
625class MixinStrUnicodeUserStringTest:
626 # additional tests that only work for
627 # stringlike objects, i.e. str, unicode, UserString
628 # (but not the string module)
629
630 def test_islower(self):
631 self.checkequal(False, '', 'islower')
632 self.checkequal(True, 'a', 'islower')
633 self.checkequal(False, 'A', 'islower')
634 self.checkequal(False, '\n', 'islower')
635 self.checkequal(True, 'abc', 'islower')
636 self.checkequal(False, 'aBc', 'islower')
637 self.checkequal(True, 'abc\n', 'islower')
638 self.checkraises(TypeError, 'abc', 'islower', 42)
639
640 def test_isupper(self):
641 self.checkequal(False, '', 'isupper')
642 self.checkequal(False, 'a', 'isupper')
643 self.checkequal(True, 'A', 'isupper')
644 self.checkequal(False, '\n', 'isupper')
645 self.checkequal(True, 'ABC', 'isupper')
646 self.checkequal(False, 'AbC', 'isupper')
647 self.checkequal(True, 'ABC\n', 'isupper')
648 self.checkraises(TypeError, 'abc', 'isupper', 42)
649
650 def test_istitle(self):
651 self.checkequal(False, '', 'istitle')
652 self.checkequal(False, 'a', 'istitle')
653 self.checkequal(True, 'A', 'istitle')
654 self.checkequal(False, '\n', 'istitle')
655 self.checkequal(True, 'A Titlecased Line', 'istitle')
656 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
657 self.checkequal(True, 'A Titlecased, Line', 'istitle')
658 self.checkequal(False, 'Not a capitalized String', 'istitle')
659 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
660 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
661 self.checkequal(False, 'NOT', 'istitle')
662 self.checkraises(TypeError, 'abc', 'istitle', 42)
663
664 def test_isspace(self):
665 self.checkequal(False, '', 'isspace')
666 self.checkequal(False, 'a', 'isspace')
667 self.checkequal(True, ' ', 'isspace')
668 self.checkequal(True, '\t', 'isspace')
669 self.checkequal(True, '\r', 'isspace')
670 self.checkequal(True, '\n', 'isspace')
671 self.checkequal(True, ' \t\r\n', 'isspace')
672 self.checkequal(False, ' \t\r\na', 'isspace')
673 self.checkraises(TypeError, 'abc', 'isspace', 42)
674
675 def test_isalpha(self):
676 self.checkequal(False, '', 'isalpha')
677 self.checkequal(True, 'a', 'isalpha')
678 self.checkequal(True, 'A', 'isalpha')
679 self.checkequal(False, '\n', 'isalpha')
680 self.checkequal(True, 'abc', 'isalpha')
681 self.checkequal(False, 'aBc123', 'isalpha')
682 self.checkequal(False, 'abc\n', 'isalpha')
683 self.checkraises(TypeError, 'abc', 'isalpha', 42)
684
685 def test_isalnum(self):
686 self.checkequal(False, '', 'isalnum')
687 self.checkequal(True, 'a', 'isalnum')
688 self.checkequal(True, 'A', 'isalnum')
689 self.checkequal(False, '\n', 'isalnum')
690 self.checkequal(True, '123abc456', 'isalnum')
691 self.checkequal(True, 'a1b3c', 'isalnum')
692 self.checkequal(False, 'aBc000 ', 'isalnum')
693 self.checkequal(False, 'abc\n', 'isalnum')
694 self.checkraises(TypeError, 'abc', 'isalnum', 42)
695
696 def test_isdigit(self):
697 self.checkequal(False, '', 'isdigit')
698 self.checkequal(False, 'a', 'isdigit')
699 self.checkequal(True, '0', 'isdigit')
700 self.checkequal(True, '0123456789', 'isdigit')
701 self.checkequal(False, '0123456789a', 'isdigit')
702
703 self.checkraises(TypeError, 'abc', 'isdigit', 42)
704
705 def test_title(self):
706 self.checkequal(' Hello ', ' hello ', 'title')
707 self.checkequal('Hello ', 'hello ', 'title')
708 self.checkequal('Hello ', 'Hello ', 'title')
709 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
710 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
711 self.checkequal('Getint', "getInt", 'title')
712 self.checkraises(TypeError, 'hello', 'title', 42)
713
714 def test_splitlines(self):
715 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
716 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
717 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
718 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
719 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
720 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
721 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'], "\nabc\ndef\r\nghi\n\r", 'splitlines', 1)
722
723 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
724
725 def test_startswith(self):
726 self.checkequal(True, 'hello', 'startswith', 'he')
727 self.checkequal(True, 'hello', 'startswith', 'hello')
728 self.checkequal(False, 'hello', 'startswith', 'hello world')
729 self.checkequal(True, 'hello', 'startswith', '')
730 self.checkequal(False, 'hello', 'startswith', 'ello')
731 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
732 self.checkequal(True, 'hello', 'startswith', 'o', 4)
733 self.checkequal(False, 'hello', 'startswith', 'o', 5)
734 self.checkequal(True, 'hello', 'startswith', '', 5)
735 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
736 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
737 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
738 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
739
740 # test negative indices
741 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
742 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
743 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
744 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
745 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
746 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
747 self.checkequal(False, 'hello', 'startswith', 'o', -2)
748 self.checkequal(True, 'hello', 'startswith', 'o', -1)
749 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
750 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
751
752 self.checkraises(TypeError, 'hello', 'startswith')
753 self.checkraises(TypeError, 'hello', 'startswith', 42)
754
755 def test_endswith(self):
756 self.checkequal(True, 'hello', 'endswith', 'lo')
757 self.checkequal(False, 'hello', 'endswith', 'he')
758 self.checkequal(True, 'hello', 'endswith', '')
759 self.checkequal(False, 'hello', 'endswith', 'hello world')
760 self.checkequal(False, 'helloworld', 'endswith', 'worl')
761 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
762 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
763 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
764 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
765 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
766 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
767 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
768 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
769 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
770
771 # test negative indices
772 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
773 self.checkequal(False, 'hello', 'endswith', 'he', -2)
774 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
775 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
776 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
777 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
778 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
779 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
780 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
781 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
782 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
783 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
784 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
785
786 self.checkraises(TypeError, 'hello', 'endswith')
787 self.checkraises(TypeError, 'hello', 'endswith', 42)
788
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000789 def test___contains__(self):
790 self.checkequal(True, '', '__contains__', '') # vereq('' in '', True)
791 self.checkequal(True, 'abc', '__contains__', '') # vereq('' in 'abc', True)
792 self.checkequal(False, 'abc', '__contains__', '\0') # vereq('\0' in 'abc', False)
793 self.checkequal(True, '\0abc', '__contains__', '\0') # vereq('\0' in '\0abc', True)
794 self.checkequal(True, 'abc\0', '__contains__', '\0') # vereq('\0' in 'abc\0', True)
795 self.checkequal(True, '\0abc', '__contains__', 'a') # vereq('a' in '\0abc', True)
796 self.checkequal(True, 'asdf', '__contains__', 'asdf') # vereq('asdf' in 'asdf', True)
797 self.checkequal(False, 'asd', '__contains__', 'asdf') # vereq('asdf' in 'asd', False)
798 self.checkequal(False, '', '__contains__', 'asdf') # vereq('asdf' in '', False)
799
800 def test_subscript(self):
801 self.checkequal(u'a', 'abc', '__getitem__', 0)
802 self.checkequal(u'c', 'abc', '__getitem__', -1)
803 self.checkequal(u'a', 'abc', '__getitem__', 0L)
804 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 3))
805 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 1000))
806 self.checkequal(u'a', 'abc', '__getitem__', slice(0, 1))
807 self.checkequal(u'', 'abc', '__getitem__', slice(0, 0))
808 # FIXME What about negative indizes? This is handled differently by [] and __getitem__(slice)
809
810 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
811
812 def test_slice(self):
813 self.checkequal('abc', 'abc', '__getslice__', 0, 1000)
814 self.checkequal('abc', 'abc', '__getslice__', 0, 3)
815 self.checkequal('ab', 'abc', '__getslice__', 0, 2)
816 self.checkequal('bc', 'abc', '__getslice__', 1, 3)
817 self.checkequal('b', 'abc', '__getslice__', 1, 2)
818 self.checkequal('', 'abc', '__getslice__', 2, 2)
819 self.checkequal('', 'abc', '__getslice__', 1000, 1000)
820 self.checkequal('', 'abc', '__getslice__', 2000, 1000)
821 self.checkequal('', 'abc', '__getslice__', 2, 1)
822 # FIXME What about negative indizes? This is handled differently by [] and __getslice__
823
824 self.checkraises(TypeError, 'abc', '__getslice__', 'def')
825
826 def test_mul(self):
827 self.checkequal('', 'abc', '__mul__', -1)
828 self.checkequal('', 'abc', '__mul__', 0)
829 self.checkequal('abc', 'abc', '__mul__', 1)
830 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
831 self.checkraises(TypeError, 'abc', '__mul__')
832 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +0000833 # XXX: on a 64-bit system, this doesn't raise an overflow error,
834 # but either raises a MemoryError, or succeeds (if you have 54TiB)
835 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000836
837 def test_join(self):
838 # join now works with any sequence type
839 # moved here, because the argument order is
840 # different in string.join (see the test in
841 # test.test_string.StringTest.test_join)
842 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
843 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
844 self.checkequal('w x y z', ' ', 'join', Sequence())
845 self.checkequal('abc', 'a', 'join', ('abc',))
846 self.checkequal('z', 'a', 'join', UserList(['z']))
847 if test_support.have_unicode:
848 self.checkequal(unicode('a.b.c'), unicode('.'), 'join', ['a', 'b', 'c'])
849 self.checkequal(unicode('a.b.c'), '.', 'join', [unicode('a'), 'b', 'c'])
850 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', unicode('b'), 'c'])
851 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', 'b', unicode('c')])
852 self.checkraises(TypeError, '.', 'join', ['a', unicode('b'), 3])
853 for i in [5, 25, 125]:
854 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
855 ['a' * i] * i)
856 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
857 ('a' * i,) * i)
858
859 self.checkraises(TypeError, ' ', 'join', BadSeq1())
860 self.checkequal('a b c', ' ', 'join', BadSeq2())
861
862 self.checkraises(TypeError, ' ', 'join')
863 self.checkraises(TypeError, ' ', 'join', 7)
864 self.checkraises(TypeError, ' ', 'join', Sequence([7, 'hello', 123L]))
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +0000865 try:
866 def f():
867 yield 4 + ""
868 self.fixtype(' ').join(f())
869 except TypeError, e:
870 if '+' not in str(e):
871 self.fail('join() ate exception message')
872 else:
873 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000874
875 def test_formatting(self):
876 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
877 self.checkequal('+10+', '+%d+', '__mod__', 10)
878 self.checkequal('a', "%c", '__mod__', "a")
879 self.checkequal('a', "%c", '__mod__', "a")
880 self.checkequal('"', "%c", '__mod__', 34)
881 self.checkequal('$', "%c", '__mod__', 36)
882 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +0000883 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000884
885 for ordinal in (-100, 0x200000):
886 # unicode raises ValueError, str raises OverflowError
887 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
888
889 self.checkequal(' 42', '%3ld', '__mod__', 42)
890 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +0000891 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000892
893 self.checkraises(TypeError, 'abc', '__mod__')
894 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
895 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
896 self.checkraises(TypeError, '%c', '__mod__', (None,))
897 self.checkraises(ValueError, '%(foo', '__mod__', {})
898 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
899
900 # argument names with properly nested brackets are supported
901 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
902
903 # 100 is a magic number in PyUnicode_Format, this forces a resize
904 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
905
906 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
907 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
908 self.checkraises(ValueError, '%10', '__mod__', (42,))
909
910 def test_floatformatting(self):
911 # float formatting
912 for prec in xrange(100):
913 format = '%%.%if' % prec
914 value = 0.01
915 for x in xrange(60):
916 value = value * 3.141592655 / 3.0 * 10.0
917 # The formatfloat() code in stringobject.c and
918 # unicodeobject.c uses a 120 byte buffer and switches from
919 # 'f' formatting to 'g' at precision 50, so we expect
920 # OverflowErrors for the ranges x < 50 and prec >= 67.
921 if x < 50 and prec >= 67:
922 self.checkraises(OverflowError, format, "__mod__", value)
923 else:
924 self.checkcall(format, "__mod__", value)
925
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000926 def test_inplace_rewrites(self):
927 # Check that strings don't copy and modify cached single-character strings
928 self.checkequal('a', 'A', 'lower')
929 self.checkequal(True, 'A', 'isupper')
930 self.checkequal('A', 'a', 'upper')
931 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000932
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000933 self.checkequal('a', 'A', 'replace', 'A', 'a')
934 self.checkequal(True, 'A', 'isupper')
935
936 self.checkequal('A', 'a', 'capitalize')
937 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +0000938
Andrew Dalke2bddcbf2006-05-25 16:30:52 +0000939 self.checkequal('A', 'a', 'swapcase')
940 self.checkequal(True, 'a', 'islower')
941
942 self.checkequal('A', 'a', 'title')
943 self.checkequal(True, 'a', 'islower')
944
Fredrik Lundh06a69dd2006-05-26 08:54:28 +0000945 def test_partition(self):
946
947 self.checkequal(('this', ' is ', 'the partition method'),
948 'this is the partition method', 'partition', ' is ')
949
950 # from raymond's original specification
951 S = 'http://www.python.org'
952 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
953 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
954 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
955 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
956
957 self.checkraises(ValueError, S, 'partition', '')
958 self.checkraises(TypeError, S, 'partition', None)
959
Walter Dörwald57d88e52004-08-26 16:53:04 +0000960
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000961class MixinStrStringUserStringTest:
962 # Additional tests for 8bit strings, i.e. str, UserString and
963 # the string module
964
965 def test_maketrans(self):
966 self.assertEqual(
967 ''.join(map(chr, xrange(256))).replace('abc', 'xyz'),
968 string.maketrans('abc', 'xyz')
969 )
970 self.assertRaises(ValueError, string.maketrans, 'abc', 'xyzw')
971
972 def test_translate(self):
973 table = string.maketrans('abc', 'xyz')
974 self.checkequal('xyzxyz', 'xyzabcdef', 'translate', table, 'def')
975
976 table = string.maketrans('a', 'A')
977 self.checkequal('Abc', 'abc', 'translate', table)
978 self.checkequal('xyz', 'xyz', 'translate', table)
979 self.checkequal('yz', 'xyz', 'translate', table, 'x')
980 self.checkraises(ValueError, 'xyz', 'translate', 'too short', 'strip')
981 self.checkraises(ValueError, 'xyz', 'translate', 'too short')
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +0000982
983
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000984class MixinStrUserStringTest:
985 # Additional tests that only work with
986 # 8bit compatible object, i.e. str and UserString
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +0000987
Walter Dörwald6eea7892005-07-28 16:49:15 +0000988 if test_support.have_unicode:
989 def test_encoding_decoding(self):
990 codecs = [('rot13', 'uryyb jbeyq'),
991 ('base64', 'aGVsbG8gd29ybGQ=\n'),
992 ('hex', '68656c6c6f20776f726c64'),
993 ('uu', 'begin 666 <data>\n+:&5L;&\\@=V]R;&0 \n \nend\n')]
994 for encoding, data in codecs:
995 self.checkequal(data, 'hello world', 'encode', encoding)
996 self.checkequal('hello world', data, 'decode', encoding)
997 # zlib is optional, so we make the test optional too...
998 try:
999 import zlib
1000 except ImportError:
1001 pass
1002 else:
1003 data = 'x\x9c\xcbH\xcd\xc9\xc9W(\xcf/\xcaI\x01\x00\x1a\x0b\x04]'
1004 self.checkequal(data, 'hello world', 'encode', 'zlib')
1005 self.checkequal('hello world', data, 'decode', 'zlib')
Walter Dörwald97951de2003-03-26 14:31:25 +00001006
Walter Dörwald6eea7892005-07-28 16:49:15 +00001007 self.checkraises(TypeError, 'xyz', 'decode', 42)
1008 self.checkraises(TypeError, 'xyz', 'encode', 42)
Walter Dörwald57d88e52004-08-26 16:53:04 +00001009
1010
1011class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001012 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001013
1014 def test_bug1001011(self):
1015 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001016 # involving a subclass.
1017 # Make sure that it is of the appropriate type.
1018 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001019 t = self.type2test
1020 class subclass(t):
1021 pass
1022 s1 = subclass("abcd")
1023 s2 = t().join([s1])
1024 self.assert_(s1 is not s2)
1025 self.assert_(type(s2) is t)
Tim Peters108f1372004-08-27 05:36:07 +00001026
1027 s1 = t("abcd")
1028 s2 = t().join([s1])
1029 self.assert_(s1 is s2)
1030
1031 # Should also test mixed-type join.
1032 if t is unicode:
1033 s1 = subclass("abcd")
1034 s2 = "".join([s1])
1035 self.assert_(s1 is not s2)
1036 self.assert_(type(s2) is t)
1037
1038 s1 = t("abcd")
1039 s2 = "".join([s1])
1040 self.assert_(s1 is s2)
1041
1042 elif t is str:
1043 s1 = subclass("abcd")
1044 s2 = u"".join([s1])
1045 self.assert_(s1 is not s2)
1046 self.assert_(type(s2) is unicode) # promotes!
1047
1048 s1 = t("abcd")
1049 s2 = u"".join([s1])
1050 self.assert_(s1 is not s2)
1051 self.assert_(type(s2) is unicode) # promotes!
1052
1053 else:
1054 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)