blob: caff3d4c344a413b262f748a01bb255528715a84 [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
Kristján Valur Jónsson170eee92007-05-03 20:09:56 +00005import unittest, string, sys, struct
Walter Dörwald0fd583c2003-02-21 12:53:50 +00006from 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')
Fredrik Lundhb51b4702006-05-29 22:42:07 +0000109 self.checkequal(2, 'aaa', 'count', 'a', 1)
110 self.checkequal(0, 'aaa', 'count', 'a', 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000111 self.checkequal(1, 'aaa', 'count', 'a', -1)
112 self.checkequal(3, 'aaa', 'count', 'a', -10)
Fredrik Lundhb51b4702006-05-29 22:42:07 +0000113 self.checkequal(1, 'aaa', 'count', 'a', 0, 1)
114 self.checkequal(3, 'aaa', 'count', 'a', 0, 10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000115 self.checkequal(2, 'aaa', 'count', 'a', 0, -1)
116 self.checkequal(0, 'aaa', 'count', 'a', 0, -10)
Fredrik Lundhb51b4702006-05-29 22:42:07 +0000117 self.checkequal(3, 'aaa', 'count', '', 1)
Fredrik Lundh9e9ef9f2006-05-30 17:39:58 +0000118 self.checkequal(1, 'aaa', 'count', '', 3)
119 self.checkequal(0, 'aaa', 'count', '', 10)
Fredrik Lundhb51b4702006-05-29 22:42:07 +0000120 self.checkequal(2, 'aaa', 'count', '', -1)
121 self.checkequal(4, 'aaa', 'count', '', -10)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000122
Amaury Forgeot d'Arcfc5ea392008-09-26 22:34:08 +0000123 self.checkequal(1, '', 'count', '')
124 self.checkequal(0, '', 'count', '', 1, 1)
125 self.checkequal(0, '', 'count', '', sys.maxint, 0)
126
127 self.checkequal(0, '', 'count', 'xx')
128 self.checkequal(0, '', 'count', 'xx', 1, 1)
129 self.checkequal(0, '', 'count', 'xx', sys.maxint, 0)
130
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000131 self.checkraises(TypeError, 'hello', 'count')
132 self.checkraises(TypeError, 'hello', 'count', 42)
133
Raymond Hettinger57e74472005-02-20 09:54:53 +0000134 # For a variety of combinations,
135 # verify that str.count() matches an equivalent function
136 # replacing all occurrences and then differencing the string lengths
137 charset = ['', 'a', 'b']
138 digits = 7
139 base = len(charset)
140 teststrings = set()
141 for i in xrange(base ** digits):
142 entry = []
143 for j in xrange(digits):
144 i, m = divmod(i, base)
145 entry.append(charset[m])
146 teststrings.add(''.join(entry))
147 teststrings = list(teststrings)
148 for i in teststrings:
149 i = self.fixtype(i)
150 n = len(i)
151 for j in teststrings:
152 r1 = i.count(j)
153 if j:
154 r2, rem = divmod(n - len(i.replace(j, '')), len(j))
155 else:
156 r2, rem = len(i)+1, 0
157 if rem or r1 != r2:
Neal Norwitzf71ec5a2006-07-30 06:57:04 +0000158 self.assertEqual(rem, 0, '%s != 0 for %s' % (rem, i))
159 self.assertEqual(r1, r2, '%s != %s for %s' % (r1, r2, i))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000160
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000161 def test_find(self):
162 self.checkequal(0, 'abcdefghiabc', 'find', 'abc')
163 self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
164 self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
165
Fredrik Lundh93eff6f2006-05-30 17:11:48 +0000166 self.checkequal(0, 'abc', 'find', '', 0)
167 self.checkequal(3, 'abc', 'find', '', 3)
168 self.checkequal(-1, 'abc', 'find', '', 4)
169
Facundo Batista57d56692007-11-16 18:04:14 +0000170 # to check the ability to pass None as defaults
171 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a')
172 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4)
173 self.checkequal(-1, 'rrarrrrrrrrra', 'find', 'a', 4, 6)
174 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4, None)
175 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a', None, 6)
176
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000177 self.checkraises(TypeError, 'hello', 'find')
178 self.checkraises(TypeError, 'hello', 'find', 42)
179
Amaury Forgeot d'Arcfc5ea392008-09-26 22:34:08 +0000180 self.checkequal(0, '', 'find', '')
181 self.checkequal(-1, '', 'find', '', 1, 1)
182 self.checkequal(-1, '', 'find', '', sys.maxint, 0)
183
184 self.checkequal(-1, '', 'find', 'xx')
185 self.checkequal(-1, '', 'find', 'xx', 1, 1)
186 self.checkequal(-1, '', 'find', 'xx', sys.maxint, 0)
187
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000188 # For a variety of combinations,
189 # verify that str.find() matches __contains__
190 # and that the found substring is really at that location
191 charset = ['', 'a', 'b', 'c']
192 digits = 5
193 base = len(charset)
194 teststrings = set()
195 for i in xrange(base ** digits):
196 entry = []
197 for j in xrange(digits):
198 i, m = divmod(i, base)
199 entry.append(charset[m])
200 teststrings.add(''.join(entry))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000201 teststrings = list(teststrings)
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000202 for i in teststrings:
203 i = self.fixtype(i)
204 for j in teststrings:
205 loc = i.find(j)
206 r1 = (loc != -1)
207 r2 = j in i
208 if r1 != r2:
209 self.assertEqual(r1, r2)
210 if loc != -1:
211 self.assertEqual(i[loc:loc+len(j)], j)
212
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000213 def test_rfind(self):
214 self.checkequal(9, 'abcdefghiabc', 'rfind', 'abc')
215 self.checkequal(12, 'abcdefghiabc', 'rfind', '')
216 self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
217 self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
218
Fredrik Lundh93eff6f2006-05-30 17:11:48 +0000219 self.checkequal(3, 'abc', 'rfind', '', 0)
220 self.checkequal(3, 'abc', 'rfind', '', 3)
221 self.checkequal(-1, 'abc', 'rfind', '', 4)
222
Facundo Batista57d56692007-11-16 18:04:14 +0000223 # to check the ability to pass None as defaults
224 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a')
225 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4)
226 self.checkequal(-1, 'rrarrrrrrrrra', 'rfind', 'a', 4, 6)
227 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4, None)
228 self.checkequal( 2, 'rrarrrrrrrrra', 'rfind', 'a', None, 6)
229
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000230 self.checkraises(TypeError, 'hello', 'rfind')
231 self.checkraises(TypeError, 'hello', 'rfind', 42)
232
233 def test_index(self):
234 self.checkequal(0, 'abcdefghiabc', 'index', '')
235 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
236 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
237 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
238
239 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
240 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
241 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
242 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
243
Facundo Batista57d56692007-11-16 18:04:14 +0000244 # to check the ability to pass None as defaults
245 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a')
246 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4)
247 self.checkraises(ValueError, 'rrarrrrrrrrra', 'index', 'a', 4, 6)
248 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4, None)
249 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a', None, 6)
250
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000251 self.checkraises(TypeError, 'hello', 'index')
252 self.checkraises(TypeError, 'hello', 'index', 42)
253
254 def test_rindex(self):
255 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
256 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
257 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
258 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
259
260 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
261 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
262 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
263 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
264 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
265
Facundo Batista57d56692007-11-16 18:04:14 +0000266 # to check the ability to pass None as defaults
267 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a')
268 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4)
269 self.checkraises(ValueError, 'rrarrrrrrrrra', 'rindex', 'a', 4, 6)
270 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4, None)
271 self.checkequal( 2, 'rrarrrrrrrrra', 'rindex', 'a', None, 6)
272
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000273 self.checkraises(TypeError, 'hello', 'rindex')
274 self.checkraises(TypeError, 'hello', 'rindex', 42)
275
276 def test_lower(self):
277 self.checkequal('hello', 'HeLLo', 'lower')
278 self.checkequal('hello', 'hello', 'lower')
279 self.checkraises(TypeError, 'hello', 'lower', 42)
280
281 def test_upper(self):
282 self.checkequal('HELLO', 'HeLLo', 'upper')
283 self.checkequal('HELLO', 'HELLO', 'upper')
284 self.checkraises(TypeError, 'hello', 'upper', 42)
285
286 def test_expandtabs(self):
287 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
288 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
289 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 4)
290 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi', 'expandtabs', 4)
291 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
292 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
293 self.checkequal('abc\r\nab\r\ndef\ng\r\nhi', 'abc\r\nab\r\ndef\ng\r\nhi', 'expandtabs', 4)
Neal Norwitz5c9a81a2007-06-11 02:16:10 +0000294 self.checkequal(' a\n b', ' \ta\n\tb', 'expandtabs', 1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000295
296 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
Neal Norwitz5c9a81a2007-06-11 02:16:10 +0000297 # This test is only valid when sizeof(int) == sizeof(void*) == 4.
298 if sys.maxint < (1 << 32) and struct.calcsize('P') == 4:
299 self.checkraises(OverflowError,
300 '\ta\n\tb', 'expandtabs', sys.maxint)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000301
302 def test_split(self):
303 self.checkequal(['this', 'is', 'the', 'split', 'function'],
304 'this is the split function', 'split')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000305
306 # by whitespace
307 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000308 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
309 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
310 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
311 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000312 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
313 sys.maxint-1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000314 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
Andrew Dalke725fe402006-05-26 16:22:52 +0000315 self.checkequal(['a b c d'], ' a b c d', 'split', None, 0)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000316 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000317
Andrew Dalke984b9712006-05-26 11:11:38 +0000318 self.checkequal([], ' ', 'split')
319 self.checkequal(['a'], ' a ', 'split')
320 self.checkequal(['a', 'b'], ' a b ', 'split')
321 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
322 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
323 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
Andrew Dalke03fb4442006-05-26 11:15:22 +0000324 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
Andrew Dalke005aee22006-05-26 12:28:15 +0000325 aaa = ' a '*20
326 self.checkequal(['a']*20, aaa, 'split')
327 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
Andrew Dalke669fa182006-05-26 13:05:55 +0000328 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
Andrew Dalke984b9712006-05-26 11:11:38 +0000329
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000330 # by a char
331 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000332 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000333 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
334 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
335 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
336 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000337 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
338 sys.maxint-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000339 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
340 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
341 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000342 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
343 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000344 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
345
Andrew Dalke005aee22006-05-26 12:28:15 +0000346 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
347 self.checkequal(['a']*15 +['a|a|a|a|a'],
348 ('a|'*20)[:-1], 'split', '|', 15)
349
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000350 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000351 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000352 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
353 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
354 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
355 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000356 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
357 sys.maxint-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000358 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
359 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000360 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000361 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
362 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
363 'split', 'test')
364 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000365 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
366 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
367 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
368 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
369 self.checkequal([''], '', 'split', 'aaa')
370 self.checkequal(['aa'], 'aa', 'split', 'aaa')
Andrew Dalke5cc60092006-05-26 12:31:00 +0000371 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
372 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000373
374 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
375 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
376 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
377 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000378
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000379 # mixed use of str and unicode
380 self.checkequal([u'a', u'b', u'c d'], 'a b c d', 'split', u' ', 2)
381
382 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000383 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
384
Andrew Dalke005aee22006-05-26 12:28:15 +0000385 # null case
386 self.checkraises(ValueError, 'hello', 'split', '')
387 self.checkraises(ValueError, 'hello', 'split', '', 0)
388
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000389 def test_rsplit(self):
390 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
391 'this is the rsplit function', 'rsplit')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000392
393 # by whitespace
394 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000395 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
396 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
397 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
398 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000399 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
400 sys.maxint-20)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000401 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
Andrew Dalke725fe402006-05-26 16:22:52 +0000402 self.checkequal(['a b c d'], 'a b c d ', 'rsplit', None, 0)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000403 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000404
Andrew Dalke669fa182006-05-26 13:05:55 +0000405 self.checkequal([], ' ', 'rsplit')
406 self.checkequal(['a'], ' a ', 'rsplit')
407 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
408 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
409 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
410 None, 1)
411 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
412 None, 2)
413 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
414 aaa = ' a '*20
415 self.checkequal(['a']*20, aaa, 'rsplit')
416 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
417 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
418
419
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000420 # by a char
421 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
422 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
423 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
424 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
425 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000426 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
427 sys.maxint-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000428 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
429 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
430 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Andrew Dalke669fa182006-05-26 13:05:55 +0000431 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
432 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
433
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000434 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
435
Andrew Dalke669fa182006-05-26 13:05:55 +0000436 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
437 self.checkequal(['a|a|a|a|a']+['a']*15,
438 ('a|'*20)[:-1], 'rsplit', '|', 15)
439
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000440 # by string
441 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
442 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
443 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
444 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
445 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000446 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
447 sys.maxint-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000448 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
449 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
450 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000451 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
452 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
453 'rsplit', 'test')
454 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
455 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
456 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
457 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
458 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
459 self.checkequal([''], '', 'rsplit', 'aaa')
460 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
461 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
462 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
463
464 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
465 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
466 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
467 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000468
469 # mixed use of str and unicode
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000470 self.checkequal([u'a b', u'c', u'd'], 'a b c d', 'rsplit', u' ', 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000471
472 # argument type
473 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000474
Andrew Dalke669fa182006-05-26 13:05:55 +0000475 # null case
476 self.checkraises(ValueError, 'hello', 'rsplit', '')
477 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
478
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000479 def test_strip(self):
480 self.checkequal('hello', ' hello ', 'strip')
481 self.checkequal('hello ', ' hello ', 'lstrip')
482 self.checkequal(' hello', ' hello ', 'rstrip')
483 self.checkequal('hello', 'hello', 'strip')
484
Neal Norwitzffe33b72003-04-10 22:35:32 +0000485 # strip/lstrip/rstrip with None arg
486 self.checkequal('hello', ' hello ', 'strip', None)
487 self.checkequal('hello ', ' hello ', 'lstrip', None)
488 self.checkequal(' hello', ' hello ', 'rstrip', None)
489 self.checkequal('hello', 'hello', 'strip', None)
490
491 # strip/lstrip/rstrip with str arg
492 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
493 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
494 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
495 self.checkequal('hello', 'hello', 'strip', 'xyz')
496
497 # strip/lstrip/rstrip with unicode arg
498 if test_support.have_unicode:
499 self.checkequal(unicode('hello', 'ascii'), 'xyzzyhelloxyzzy',
500 'strip', unicode('xyz', 'ascii'))
501 self.checkequal(unicode('helloxyzzy', 'ascii'), 'xyzzyhelloxyzzy',
502 'lstrip', unicode('xyz', 'ascii'))
503 self.checkequal(unicode('xyzzyhello', 'ascii'), 'xyzzyhelloxyzzy',
504 'rstrip', unicode('xyz', 'ascii'))
Christian Heimes1a6387e2008-03-26 12:49:49 +0000505 # XXX
506 #self.checkequal(unicode('hello', 'ascii'), 'hello',
507 # 'strip', unicode('xyz', 'ascii'))
Neal Norwitzffe33b72003-04-10 22:35:32 +0000508
509 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
510 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
511 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
512
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000513 def test_ljust(self):
514 self.checkequal('abc ', 'abc', 'ljust', 10)
515 self.checkequal('abc ', 'abc', 'ljust', 6)
516 self.checkequal('abc', 'abc', 'ljust', 3)
517 self.checkequal('abc', 'abc', 'ljust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000518 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000519 self.checkraises(TypeError, 'abc', 'ljust')
520
521 def test_rjust(self):
522 self.checkequal(' abc', 'abc', 'rjust', 10)
523 self.checkequal(' abc', 'abc', 'rjust', 6)
524 self.checkequal('abc', 'abc', 'rjust', 3)
525 self.checkequal('abc', 'abc', 'rjust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000526 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000527 self.checkraises(TypeError, 'abc', 'rjust')
528
529 def test_center(self):
530 self.checkequal(' abc ', 'abc', 'center', 10)
531 self.checkequal(' abc ', 'abc', 'center', 6)
532 self.checkequal('abc', 'abc', 'center', 3)
533 self.checkequal('abc', 'abc', 'center', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000534 self.checkequal('***abc****', 'abc', 'center', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000535 self.checkraises(TypeError, 'abc', 'center')
536
537 def test_swapcase(self):
538 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
539
540 self.checkraises(TypeError, 'hello', 'swapcase', 42)
541
542 def test_replace(self):
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000543 EQ = self.checkequal
544
545 # Operations on the empty string
546 EQ("", "", "replace", "", "")
Tim Peters80a18f02006-06-01 13:56:26 +0000547 EQ("A", "", "replace", "", "A")
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000548 EQ("", "", "replace", "A", "")
549 EQ("", "", "replace", "A", "A")
550 EQ("", "", "replace", "", "", 100)
551 EQ("", "", "replace", "", "", sys.maxint)
552
553 # interleave (from=="", 'to' gets inserted everywhere)
554 EQ("A", "A", "replace", "", "")
555 EQ("*A*", "A", "replace", "", "*")
556 EQ("*1A*1", "A", "replace", "", "*1")
557 EQ("*-#A*-#", "A", "replace", "", "*-#")
558 EQ("*-A*-A*-", "AA", "replace", "", "*-")
559 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
560 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxint)
561 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
562 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
563 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
564 EQ("*-AA", "AA", "replace", "", "*-", 1)
565 EQ("AA", "AA", "replace", "", "*-", 0)
566
567 # single character deletion (from=="A", to=="")
568 EQ("", "A", "replace", "A", "")
569 EQ("", "AAA", "replace", "A", "")
570 EQ("", "AAA", "replace", "A", "", -1)
571 EQ("", "AAA", "replace", "A", "", sys.maxint)
572 EQ("", "AAA", "replace", "A", "", 4)
573 EQ("", "AAA", "replace", "A", "", 3)
574 EQ("A", "AAA", "replace", "A", "", 2)
575 EQ("AA", "AAA", "replace", "A", "", 1)
576 EQ("AAA", "AAA", "replace", "A", "", 0)
577 EQ("", "AAAAAAAAAA", "replace", "A", "")
578 EQ("BCD", "ABACADA", "replace", "A", "")
579 EQ("BCD", "ABACADA", "replace", "A", "", -1)
580 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxint)
581 EQ("BCD", "ABACADA", "replace", "A", "", 5)
582 EQ("BCD", "ABACADA", "replace", "A", "", 4)
583 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
584 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
585 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
586 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
587 EQ("BCD", "ABCAD", "replace", "A", "")
588 EQ("BCD", "ABCADAA", "replace", "A", "")
589 EQ("BCD", "BCD", "replace", "A", "")
590 EQ("*************", "*************", "replace", "A", "")
591 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
592
593 # substring deletion (from=="the", to=="")
594 EQ("", "the", "replace", "the", "")
595 EQ("ater", "theater", "replace", "the", "")
596 EQ("", "thethe", "replace", "the", "")
597 EQ("", "thethethethe", "replace", "the", "")
598 EQ("aaaa", "theatheatheathea", "replace", "the", "")
599 EQ("that", "that", "replace", "the", "")
600 EQ("thaet", "thaet", "replace", "the", "")
601 EQ("here and re", "here and there", "replace", "the", "")
602 EQ("here and re and re", "here and there and there",
603 "replace", "the", "", sys.maxint)
604 EQ("here and re and re", "here and there and there",
605 "replace", "the", "", -1)
606 EQ("here and re and re", "here and there and there",
607 "replace", "the", "", 3)
608 EQ("here and re and re", "here and there and there",
609 "replace", "the", "", 2)
610 EQ("here and re and there", "here and there and there",
611 "replace", "the", "", 1)
612 EQ("here and there and there", "here and there and there",
613 "replace", "the", "", 0)
614 EQ("here and re and re", "here and there and there", "replace", "the", "")
615
616 EQ("abc", "abc", "replace", "the", "")
617 EQ("abcdefg", "abcdefg", "replace", "the", "")
618
619 # substring deletion (from=="bob", to=="")
620 EQ("bob", "bbobob", "replace", "bob", "")
621 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
622 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
623 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000624
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000625 # single character replace in place (len(from)==len(to)==1)
626 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
627 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
628 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxint)
629 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
630 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
631 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
632 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
633 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
634
635 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
636 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
637 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
638 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
639 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
640
641 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000642
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000643 # substring replace in place (len(from)==len(to) > 1)
644 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
645 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxint)
646 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
647 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
648 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
649 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
650 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
651 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
652 EQ("cobob", "bobob", "replace", "bob", "cob")
653 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
654 EQ("bobob", "bobob", "replace", "bot", "bot")
655
656 # replace single character (len(from)==1, len(to)>1)
657 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
658 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
659 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxint)
660 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
661 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
662 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
663 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
664
665 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
666
667 # replace substring (len(from)>1, len(to)!=len(from))
668 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
669 "replace", "spam", "ham")
670 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
671 "replace", "spam", "ham", sys.maxint)
672 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
673 "replace", "spam", "ham", -1)
674 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
675 "replace", "spam", "ham", 4)
676 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
677 "replace", "spam", "ham", 3)
678 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
679 "replace", "spam", "ham", 2)
680 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
681 "replace", "spam", "ham", 1)
682 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
683 "replace", "spam", "ham", 0)
684
685 EQ("bobob", "bobobob", "replace", "bobob", "bob")
686 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
687 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000688
Neal Norwitzf71ec5a2006-07-30 06:57:04 +0000689 ba = buffer('a')
690 bb = buffer('b')
691 EQ("bbc", "abc", "replace", ba, bb)
692 EQ("aac", "abc", "replace", bb, ba)
693
Tim Petersbeaec0c2006-05-24 20:27:18 +0000694 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000695 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
696 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
697 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
698 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
699 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
700 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
701 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
702 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
703 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
704 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
705 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
706 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
707 self.checkequal('', '', 'replace', '', '')
708 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
709 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
710 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
711 # MemoryError due to empty result (platform malloc issue when requesting
712 # 0 bytes).
713 self.checkequal('', '123', 'replace', '123', '')
714 self.checkequal('', '123123', 'replace', '123', '')
715 self.checkequal('x', '123x123', 'replace', '123', '')
716
717 self.checkraises(TypeError, 'hello', 'replace')
718 self.checkraises(TypeError, 'hello', 'replace', 42)
719 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
720 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
721
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000722 def test_replace_overflow(self):
723 # Check for overflow checking on 32 bit machines
Kristján Valur Jónsson170eee92007-05-03 20:09:56 +0000724 if sys.maxint != 2147483647 or struct.calcsize("P") > 4:
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000725 return
726 A2_16 = "A" * (2**16)
727 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
728 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
729 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000730
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000731 def test_zfill(self):
732 self.checkequal('123', '123', 'zfill', 2)
733 self.checkequal('123', '123', 'zfill', 3)
734 self.checkequal('0123', '123', 'zfill', 4)
735 self.checkequal('+123', '+123', 'zfill', 3)
736 self.checkequal('+123', '+123', 'zfill', 4)
737 self.checkequal('+0123', '+123', 'zfill', 5)
738 self.checkequal('-123', '-123', 'zfill', 3)
739 self.checkequal('-123', '-123', 'zfill', 4)
740 self.checkequal('-0123', '-123', 'zfill', 5)
741 self.checkequal('000', '', 'zfill', 3)
742 self.checkequal('34', '34', 'zfill', 1)
743 self.checkequal('0034', '34', 'zfill', 4)
744
745 self.checkraises(TypeError, '123', 'zfill')
746
Christian Heimes1a6387e2008-03-26 12:49:49 +0000747# XXX alias for py3k forward compatibility
748BaseTest = CommonTest
749
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000750class MixinStrUnicodeUserStringTest:
751 # additional tests that only work for
752 # stringlike objects, i.e. str, unicode, UserString
753 # (but not the string module)
754
755 def test_islower(self):
756 self.checkequal(False, '', 'islower')
757 self.checkequal(True, 'a', 'islower')
758 self.checkequal(False, 'A', 'islower')
759 self.checkequal(False, '\n', 'islower')
760 self.checkequal(True, 'abc', 'islower')
761 self.checkequal(False, 'aBc', 'islower')
762 self.checkequal(True, 'abc\n', 'islower')
763 self.checkraises(TypeError, 'abc', 'islower', 42)
764
765 def test_isupper(self):
766 self.checkequal(False, '', 'isupper')
767 self.checkequal(False, 'a', 'isupper')
768 self.checkequal(True, 'A', 'isupper')
769 self.checkequal(False, '\n', 'isupper')
770 self.checkequal(True, 'ABC', 'isupper')
771 self.checkequal(False, 'AbC', 'isupper')
772 self.checkequal(True, 'ABC\n', 'isupper')
773 self.checkraises(TypeError, 'abc', 'isupper', 42)
774
775 def test_istitle(self):
776 self.checkequal(False, '', 'istitle')
777 self.checkequal(False, 'a', 'istitle')
778 self.checkequal(True, 'A', 'istitle')
779 self.checkequal(False, '\n', 'istitle')
780 self.checkequal(True, 'A Titlecased Line', 'istitle')
781 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
782 self.checkequal(True, 'A Titlecased, Line', 'istitle')
783 self.checkequal(False, 'Not a capitalized String', 'istitle')
784 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
785 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
786 self.checkequal(False, 'NOT', 'istitle')
787 self.checkraises(TypeError, 'abc', 'istitle', 42)
788
789 def test_isspace(self):
790 self.checkequal(False, '', 'isspace')
791 self.checkequal(False, 'a', 'isspace')
792 self.checkequal(True, ' ', 'isspace')
793 self.checkequal(True, '\t', 'isspace')
794 self.checkequal(True, '\r', 'isspace')
795 self.checkequal(True, '\n', 'isspace')
796 self.checkequal(True, ' \t\r\n', 'isspace')
797 self.checkequal(False, ' \t\r\na', 'isspace')
798 self.checkraises(TypeError, 'abc', 'isspace', 42)
799
800 def test_isalpha(self):
801 self.checkequal(False, '', 'isalpha')
802 self.checkequal(True, 'a', 'isalpha')
803 self.checkequal(True, 'A', 'isalpha')
804 self.checkequal(False, '\n', 'isalpha')
805 self.checkequal(True, 'abc', 'isalpha')
806 self.checkequal(False, 'aBc123', 'isalpha')
807 self.checkequal(False, 'abc\n', 'isalpha')
808 self.checkraises(TypeError, 'abc', 'isalpha', 42)
809
810 def test_isalnum(self):
811 self.checkequal(False, '', 'isalnum')
812 self.checkequal(True, 'a', 'isalnum')
813 self.checkequal(True, 'A', 'isalnum')
814 self.checkequal(False, '\n', 'isalnum')
815 self.checkequal(True, '123abc456', 'isalnum')
816 self.checkequal(True, 'a1b3c', 'isalnum')
817 self.checkequal(False, 'aBc000 ', 'isalnum')
818 self.checkequal(False, 'abc\n', 'isalnum')
819 self.checkraises(TypeError, 'abc', 'isalnum', 42)
820
821 def test_isdigit(self):
822 self.checkequal(False, '', 'isdigit')
823 self.checkequal(False, 'a', 'isdigit')
824 self.checkequal(True, '0', 'isdigit')
825 self.checkequal(True, '0123456789', 'isdigit')
826 self.checkequal(False, '0123456789a', 'isdigit')
827
828 self.checkraises(TypeError, 'abc', 'isdigit', 42)
829
830 def test_title(self):
831 self.checkequal(' Hello ', ' hello ', 'title')
832 self.checkequal('Hello ', 'hello ', 'title')
833 self.checkequal('Hello ', 'Hello ', 'title')
834 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
835 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
836 self.checkequal('Getint', "getInt", 'title')
837 self.checkraises(TypeError, 'hello', 'title', 42)
838
839 def test_splitlines(self):
840 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
841 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
842 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
843 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
844 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
845 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
846 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'], "\nabc\ndef\r\nghi\n\r", 'splitlines', 1)
847
848 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
849
850 def test_startswith(self):
851 self.checkequal(True, 'hello', 'startswith', 'he')
852 self.checkequal(True, 'hello', 'startswith', 'hello')
853 self.checkequal(False, 'hello', 'startswith', 'hello world')
854 self.checkequal(True, 'hello', 'startswith', '')
855 self.checkequal(False, 'hello', 'startswith', 'ello')
856 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
857 self.checkequal(True, 'hello', 'startswith', 'o', 4)
858 self.checkequal(False, 'hello', 'startswith', 'o', 5)
859 self.checkequal(True, 'hello', 'startswith', '', 5)
860 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
861 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
862 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
863 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
864
865 # test negative indices
866 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
867 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
868 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
869 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
870 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
871 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
872 self.checkequal(False, 'hello', 'startswith', 'o', -2)
873 self.checkequal(True, 'hello', 'startswith', 'o', -1)
874 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
875 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
876
877 self.checkraises(TypeError, 'hello', 'startswith')
878 self.checkraises(TypeError, 'hello', 'startswith', 42)
879
Georg Brandl24250812006-06-09 18:45:48 +0000880 # test tuple arguments
881 self.checkequal(True, 'hello', 'startswith', ('he', 'ha'))
882 self.checkequal(False, 'hello', 'startswith', ('lo', 'llo'))
883 self.checkequal(True, 'hello', 'startswith', ('hellox', 'hello'))
884 self.checkequal(False, 'hello', 'startswith', ())
885 self.checkequal(True, 'helloworld', 'startswith', ('hellowo',
886 'rld', 'lowo'), 3)
887 self.checkequal(False, 'helloworld', 'startswith', ('hellowo', 'ello',
888 'rld'), 3)
889 self.checkequal(True, 'hello', 'startswith', ('lo', 'he'), 0, -1)
890 self.checkequal(False, 'hello', 'startswith', ('he', 'hel'), 0, 1)
891 self.checkequal(True, 'hello', 'startswith', ('he', 'hel'), 0, 2)
892
893 self.checkraises(TypeError, 'hello', 'startswith', (42,))
894
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000895 def test_endswith(self):
896 self.checkequal(True, 'hello', 'endswith', 'lo')
897 self.checkequal(False, 'hello', 'endswith', 'he')
898 self.checkequal(True, 'hello', 'endswith', '')
899 self.checkequal(False, 'hello', 'endswith', 'hello world')
900 self.checkequal(False, 'helloworld', 'endswith', 'worl')
901 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
902 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
903 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
904 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
905 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
906 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
907 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
908 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
909 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
910
911 # test negative indices
912 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
913 self.checkequal(False, 'hello', 'endswith', 'he', -2)
914 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
915 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
916 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
917 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
918 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
919 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
920 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
921 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
922 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
923 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
924 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
925
926 self.checkraises(TypeError, 'hello', 'endswith')
927 self.checkraises(TypeError, 'hello', 'endswith', 42)
928
Georg Brandl24250812006-06-09 18:45:48 +0000929 # test tuple arguments
930 self.checkequal(False, 'hello', 'endswith', ('he', 'ha'))
931 self.checkequal(True, 'hello', 'endswith', ('lo', 'llo'))
932 self.checkequal(True, 'hello', 'endswith', ('hellox', 'hello'))
933 self.checkequal(False, 'hello', 'endswith', ())
934 self.checkequal(True, 'helloworld', 'endswith', ('hellowo',
935 'rld', 'lowo'), 3)
936 self.checkequal(False, 'helloworld', 'endswith', ('hellowo', 'ello',
937 'rld'), 3, -1)
938 self.checkequal(True, 'hello', 'endswith', ('hell', 'ell'), 0, -1)
939 self.checkequal(False, 'hello', 'endswith', ('he', 'hel'), 0, 1)
940 self.checkequal(True, 'hello', 'endswith', ('he', 'hell'), 0, 4)
941
942 self.checkraises(TypeError, 'hello', 'endswith', (42,))
943
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000944 def test___contains__(self):
945 self.checkequal(True, '', '__contains__', '') # vereq('' in '', True)
946 self.checkequal(True, 'abc', '__contains__', '') # vereq('' in 'abc', True)
947 self.checkequal(False, 'abc', '__contains__', '\0') # vereq('\0' in 'abc', False)
948 self.checkequal(True, '\0abc', '__contains__', '\0') # vereq('\0' in '\0abc', True)
949 self.checkequal(True, 'abc\0', '__contains__', '\0') # vereq('\0' in 'abc\0', True)
950 self.checkequal(True, '\0abc', '__contains__', 'a') # vereq('a' in '\0abc', True)
951 self.checkequal(True, 'asdf', '__contains__', 'asdf') # vereq('asdf' in 'asdf', True)
952 self.checkequal(False, 'asd', '__contains__', 'asdf') # vereq('asdf' in 'asd', False)
953 self.checkequal(False, '', '__contains__', 'asdf') # vereq('asdf' in '', False)
954
955 def test_subscript(self):
956 self.checkequal(u'a', 'abc', '__getitem__', 0)
957 self.checkequal(u'c', 'abc', '__getitem__', -1)
958 self.checkequal(u'a', 'abc', '__getitem__', 0L)
959 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 3))
960 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 1000))
961 self.checkequal(u'a', 'abc', '__getitem__', slice(0, 1))
962 self.checkequal(u'', 'abc', '__getitem__', slice(0, 0))
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000963
964 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
965
966 def test_slice(self):
967 self.checkequal('abc', 'abc', '__getslice__', 0, 1000)
968 self.checkequal('abc', 'abc', '__getslice__', 0, 3)
969 self.checkequal('ab', 'abc', '__getslice__', 0, 2)
970 self.checkequal('bc', 'abc', '__getslice__', 1, 3)
971 self.checkequal('b', 'abc', '__getslice__', 1, 2)
972 self.checkequal('', 'abc', '__getslice__', 2, 2)
973 self.checkequal('', 'abc', '__getslice__', 1000, 1000)
974 self.checkequal('', 'abc', '__getslice__', 2000, 1000)
975 self.checkequal('', 'abc', '__getslice__', 2, 1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000976
977 self.checkraises(TypeError, 'abc', '__getslice__', 'def')
978
Thomas Wouters3ccec682007-08-28 15:28:19 +0000979 def test_extended_getslice(self):
980 # Test extended slicing by comparing with list slicing.
981 s = string.ascii_letters + string.digits
982 indices = (0, None, 1, 3, 41, -1, -2, -37)
983 for start in indices:
984 for stop in indices:
985 # Skip step 0 (invalid)
986 for step in indices[1:]:
987 L = list(s)[start:stop:step]
988 self.checkequal(u"".join(L), s, '__getitem__',
989 slice(start, stop, step))
990
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000991 def test_mul(self):
992 self.checkequal('', 'abc', '__mul__', -1)
993 self.checkequal('', 'abc', '__mul__', 0)
994 self.checkequal('abc', 'abc', '__mul__', 1)
995 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
996 self.checkraises(TypeError, 'abc', '__mul__')
997 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +0000998 # XXX: on a 64-bit system, this doesn't raise an overflow error,
999 # but either raises a MemoryError, or succeeds (if you have 54TiB)
1000 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001001
1002 def test_join(self):
1003 # join now works with any sequence type
1004 # moved here, because the argument order is
1005 # different in string.join (see the test in
1006 # test.test_string.StringTest.test_join)
1007 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
1008 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
Georg Brandl90e27d32006-06-10 06:40:50 +00001009 self.checkequal('bd', '', 'join', ('', 'b', '', 'd'))
1010 self.checkequal('ac', '', 'join', ('a', '', 'c', ''))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001011 self.checkequal('w x y z', ' ', 'join', Sequence())
1012 self.checkequal('abc', 'a', 'join', ('abc',))
1013 self.checkequal('z', 'a', 'join', UserList(['z']))
1014 if test_support.have_unicode:
1015 self.checkequal(unicode('a.b.c'), unicode('.'), 'join', ['a', 'b', 'c'])
1016 self.checkequal(unicode('a.b.c'), '.', 'join', [unicode('a'), 'b', 'c'])
1017 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', unicode('b'), 'c'])
1018 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', 'b', unicode('c')])
1019 self.checkraises(TypeError, '.', 'join', ['a', unicode('b'), 3])
1020 for i in [5, 25, 125]:
1021 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1022 ['a' * i] * i)
1023 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1024 ('a' * i,) * i)
1025
1026 self.checkraises(TypeError, ' ', 'join', BadSeq1())
1027 self.checkequal('a b c', ' ', 'join', BadSeq2())
1028
1029 self.checkraises(TypeError, ' ', 'join')
1030 self.checkraises(TypeError, ' ', 'join', 7)
1031 self.checkraises(TypeError, ' ', 'join', Sequence([7, 'hello', 123L]))
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001032 try:
1033 def f():
1034 yield 4 + ""
1035 self.fixtype(' ').join(f())
1036 except TypeError, e:
1037 if '+' not in str(e):
1038 self.fail('join() ate exception message')
1039 else:
1040 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001041
1042 def test_formatting(self):
1043 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
1044 self.checkequal('+10+', '+%d+', '__mod__', 10)
1045 self.checkequal('a', "%c", '__mod__', "a")
1046 self.checkequal('a', "%c", '__mod__', "a")
1047 self.checkequal('"', "%c", '__mod__', 34)
1048 self.checkequal('$', "%c", '__mod__', 36)
1049 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +00001050 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001051
1052 for ordinal in (-100, 0x200000):
1053 # unicode raises ValueError, str raises OverflowError
1054 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
1055
Facundo Batistac11cecf2008-02-24 03:17:21 +00001056 longvalue = sys.maxint + 10L
1057 slongvalue = str(longvalue)
1058 if slongvalue[-1] in ("L","l"): slongvalue = slongvalue[:-1]
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001059 self.checkequal(' 42', '%3ld', '__mod__', 42)
Facundo Batistac11cecf2008-02-24 03:17:21 +00001060 self.checkequal('42', '%d', '__mod__', 42L)
1061 self.checkequal('42', '%d', '__mod__', 42.0)
1062 self.checkequal(slongvalue, '%d', '__mod__', longvalue)
1063 self.checkcall('%d', '__mod__', float(longvalue))
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001064 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +00001065 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001066
1067 self.checkraises(TypeError, 'abc', '__mod__')
1068 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
1069 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
1070 self.checkraises(TypeError, '%c', '__mod__', (None,))
1071 self.checkraises(ValueError, '%(foo', '__mod__', {})
1072 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
Facundo Batistac11cecf2008-02-24 03:17:21 +00001073 self.checkraises(TypeError, '%d', '__mod__', "42") # not numeric
1074 self.checkraises(TypeError, '%d', '__mod__', (42+0j)) # no int/long conversion provided
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001075
1076 # argument names with properly nested brackets are supported
1077 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
1078
1079 # 100 is a magic number in PyUnicode_Format, this forces a resize
1080 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
1081
1082 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
1083 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
1084 self.checkraises(ValueError, '%10', '__mod__', (42,))
1085
1086 def test_floatformatting(self):
1087 # float formatting
1088 for prec in xrange(100):
1089 format = '%%.%if' % prec
1090 value = 0.01
1091 for x in xrange(60):
1092 value = value * 3.141592655 / 3.0 * 10.0
Mark Dickinson18cfada2009-11-23 18:46:41 +00001093 self.checkcall(format, "__mod__", value)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001094
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001095 def test_inplace_rewrites(self):
1096 # Check that strings don't copy and modify cached single-character strings
1097 self.checkequal('a', 'A', 'lower')
1098 self.checkequal(True, 'A', 'isupper')
1099 self.checkequal('A', 'a', 'upper')
1100 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +00001101
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001102 self.checkequal('a', 'A', 'replace', 'A', 'a')
1103 self.checkequal(True, 'A', 'isupper')
1104
1105 self.checkequal('A', 'a', 'capitalize')
1106 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +00001107
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001108 self.checkequal('A', 'a', 'swapcase')
1109 self.checkequal(True, 'a', 'islower')
1110
1111 self.checkequal('A', 'a', 'title')
1112 self.checkequal(True, 'a', 'islower')
1113
Fredrik Lundh06a69dd2006-05-26 08:54:28 +00001114 def test_partition(self):
1115
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001116 self.checkequal(('this is the par', 'ti', 'tion method'),
1117 'this is the partition method', 'partition', 'ti')
Fredrik Lundh06a69dd2006-05-26 08:54:28 +00001118
1119 # from raymond's original specification
1120 S = 'http://www.python.org'
1121 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1122 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1123 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1124 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1125
1126 self.checkraises(ValueError, S, 'partition', '')
1127 self.checkraises(TypeError, S, 'partition', None)
1128
Amaury Forgeot d'Arc3571fbf2008-09-01 19:52:00 +00001129 # mixed use of str and unicode
1130 self.assertEqual('a/b/c'.partition(u'/'), ('a', '/', 'b/c'))
1131
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001132 def test_rpartition(self):
1133
1134 self.checkequal(('this is the rparti', 'ti', 'on method'),
1135 'this is the rpartition method', 'rpartition', 'ti')
1136
1137 # from raymond's original specification
1138 S = 'http://www.python.org'
1139 self.checkequal(('http', '://', 'www.python.org'), S, 'rpartition', '://')
Raymond Hettingera0c95fa2006-09-04 15:32:48 +00001140 self.checkequal(('', '', 'http://www.python.org'), S, 'rpartition', '?')
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001141 self.checkequal(('', 'http://', 'www.python.org'), S, 'rpartition', 'http://')
1142 self.checkequal(('http://www.python.', 'org', ''), S, 'rpartition', 'org')
1143
1144 self.checkraises(ValueError, S, 'rpartition', '')
1145 self.checkraises(TypeError, S, 'rpartition', None)
1146
Amaury Forgeot d'Arc3571fbf2008-09-01 19:52:00 +00001147 # mixed use of str and unicode
1148 self.assertEqual('a/b/c'.rpartition(u'/'), ('a/b', '/', 'c'))
Walter Dörwald57d88e52004-08-26 16:53:04 +00001149
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001150class MixinStrStringUserStringTest:
1151 # Additional tests for 8bit strings, i.e. str, UserString and
1152 # the string module
1153
1154 def test_maketrans(self):
1155 self.assertEqual(
1156 ''.join(map(chr, xrange(256))).replace('abc', 'xyz'),
1157 string.maketrans('abc', 'xyz')
1158 )
1159 self.assertRaises(ValueError, string.maketrans, 'abc', 'xyzw')
1160
1161 def test_translate(self):
1162 table = string.maketrans('abc', 'xyz')
1163 self.checkequal('xyzxyz', 'xyzabcdef', 'translate', table, 'def')
1164
1165 table = string.maketrans('a', 'A')
1166 self.checkequal('Abc', 'abc', 'translate', table)
1167 self.checkequal('xyz', 'xyz', 'translate', table)
1168 self.checkequal('yz', 'xyz', 'translate', table, 'x')
Neal Norwitz0d4c06e2007-04-25 06:30:05 +00001169 self.checkequal('yx', 'zyzzx', 'translate', None, 'z')
Raymond Hettinger4db5fe92007-04-12 04:10:00 +00001170 self.checkequal('zyzzx', 'zyzzx', 'translate', None, '')
Neal Norwitz0d4c06e2007-04-25 06:30:05 +00001171 self.checkequal('zyzzx', 'zyzzx', 'translate', None)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001172 self.checkraises(ValueError, 'xyz', 'translate', 'too short', 'strip')
1173 self.checkraises(ValueError, 'xyz', 'translate', 'too short')
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001174
1175
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001176class MixinStrUserStringTest:
1177 # Additional tests that only work with
1178 # 8bit compatible object, i.e. str and UserString
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001179
Walter Dörwald6eea7892005-07-28 16:49:15 +00001180 if test_support.have_unicode:
1181 def test_encoding_decoding(self):
1182 codecs = [('rot13', 'uryyb jbeyq'),
1183 ('base64', 'aGVsbG8gd29ybGQ=\n'),
1184 ('hex', '68656c6c6f20776f726c64'),
1185 ('uu', 'begin 666 <data>\n+:&5L;&\\@=V]R;&0 \n \nend\n')]
1186 for encoding, data in codecs:
1187 self.checkequal(data, 'hello world', 'encode', encoding)
1188 self.checkequal('hello world', data, 'decode', encoding)
1189 # zlib is optional, so we make the test optional too...
1190 try:
1191 import zlib
1192 except ImportError:
1193 pass
1194 else:
1195 data = 'x\x9c\xcbH\xcd\xc9\xc9W(\xcf/\xcaI\x01\x00\x1a\x0b\x04]'
1196 self.checkequal(data, 'hello world', 'encode', 'zlib')
1197 self.checkequal('hello world', data, 'decode', 'zlib')
Walter Dörwald97951de2003-03-26 14:31:25 +00001198
Walter Dörwald6eea7892005-07-28 16:49:15 +00001199 self.checkraises(TypeError, 'xyz', 'decode', 42)
1200 self.checkraises(TypeError, 'xyz', 'encode', 42)
Walter Dörwald57d88e52004-08-26 16:53:04 +00001201
1202
1203class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001204 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001205
1206 def test_bug1001011(self):
1207 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001208 # involving a subclass.
1209 # Make sure that it is of the appropriate type.
1210 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001211 t = self.type2test
1212 class subclass(t):
1213 pass
1214 s1 = subclass("abcd")
1215 s2 = t().join([s1])
1216 self.assert_(s1 is not s2)
1217 self.assert_(type(s2) is t)
Tim Peters108f1372004-08-27 05:36:07 +00001218
1219 s1 = t("abcd")
1220 s2 = t().join([s1])
1221 self.assert_(s1 is s2)
1222
1223 # Should also test mixed-type join.
1224 if t is unicode:
1225 s1 = subclass("abcd")
1226 s2 = "".join([s1])
1227 self.assert_(s1 is not s2)
1228 self.assert_(type(s2) is t)
1229
1230 s1 = t("abcd")
1231 s2 = "".join([s1])
1232 self.assert_(s1 is s2)
1233
1234 elif t is str:
1235 s1 = subclass("abcd")
1236 s2 = u"".join([s1])
1237 self.assert_(s1 is not s2)
1238 self.assert_(type(s2) is unicode) # promotes!
1239
1240 s1 = t("abcd")
1241 s2 = u"".join([s1])
1242 self.assert_(s1 is not s2)
1243 self.assert_(type(s2) is unicode) # promotes!
1244
1245 else:
1246 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)