blob: cefdcacfec6abc06312e30bc24ba1681f29d7ed0 [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
123 self.checkraises(TypeError, 'hello', 'count')
124 self.checkraises(TypeError, 'hello', 'count', 42)
125
Raymond Hettinger57e74472005-02-20 09:54:53 +0000126 # For a variety of combinations,
127 # verify that str.count() matches an equivalent function
128 # replacing all occurrences and then differencing the string lengths
129 charset = ['', 'a', 'b']
130 digits = 7
131 base = len(charset)
132 teststrings = set()
133 for i in xrange(base ** digits):
134 entry = []
135 for j in xrange(digits):
136 i, m = divmod(i, base)
137 entry.append(charset[m])
138 teststrings.add(''.join(entry))
139 teststrings = list(teststrings)
140 for i in teststrings:
141 i = self.fixtype(i)
142 n = len(i)
143 for j in teststrings:
144 r1 = i.count(j)
145 if j:
146 r2, rem = divmod(n - len(i.replace(j, '')), len(j))
147 else:
148 r2, rem = len(i)+1, 0
149 if rem or r1 != r2:
Neal Norwitzf71ec5a2006-07-30 06:57:04 +0000150 self.assertEqual(rem, 0, '%s != 0 for %s' % (rem, i))
151 self.assertEqual(r1, r2, '%s != %s for %s' % (r1, r2, i))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000152
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000153 def test_find(self):
154 self.checkequal(0, 'abcdefghiabc', 'find', 'abc')
155 self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
156 self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
157
Fredrik Lundh93eff6f2006-05-30 17:11:48 +0000158 self.checkequal(0, 'abc', 'find', '', 0)
159 self.checkequal(3, 'abc', 'find', '', 3)
160 self.checkequal(-1, 'abc', 'find', '', 4)
161
Facundo Batista57d56692007-11-16 18:04:14 +0000162 # to check the ability to pass None as defaults
163 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a')
164 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4)
165 self.checkequal(-1, 'rrarrrrrrrrra', 'find', 'a', 4, 6)
166 self.checkequal(12, 'rrarrrrrrrrra', 'find', 'a', 4, None)
167 self.checkequal( 2, 'rrarrrrrrrrra', 'find', 'a', None, 6)
168
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000169 self.checkraises(TypeError, 'hello', 'find')
170 self.checkraises(TypeError, 'hello', 'find', 42)
171
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000172 # For a variety of combinations,
173 # verify that str.find() matches __contains__
174 # and that the found substring is really at that location
175 charset = ['', 'a', 'b', 'c']
176 digits = 5
177 base = len(charset)
178 teststrings = set()
179 for i in xrange(base ** digits):
180 entry = []
181 for j in xrange(digits):
182 i, m = divmod(i, base)
183 entry.append(charset[m])
184 teststrings.add(''.join(entry))
Raymond Hettinger57e74472005-02-20 09:54:53 +0000185 teststrings = list(teststrings)
Raymond Hettinger7cbf1bc2005-02-20 04:07:08 +0000186 for i in teststrings:
187 i = self.fixtype(i)
188 for j in teststrings:
189 loc = i.find(j)
190 r1 = (loc != -1)
191 r2 = j in i
192 if r1 != r2:
193 self.assertEqual(r1, r2)
194 if loc != -1:
195 self.assertEqual(i[loc:loc+len(j)], j)
196
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000197 def test_rfind(self):
198 self.checkequal(9, 'abcdefghiabc', 'rfind', 'abc')
199 self.checkequal(12, 'abcdefghiabc', 'rfind', '')
200 self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
201 self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
202
Fredrik Lundh93eff6f2006-05-30 17:11:48 +0000203 self.checkequal(3, 'abc', 'rfind', '', 0)
204 self.checkequal(3, 'abc', 'rfind', '', 3)
205 self.checkequal(-1, 'abc', 'rfind', '', 4)
206
Facundo Batista57d56692007-11-16 18:04:14 +0000207 # to check the ability to pass None as defaults
208 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a')
209 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4)
210 self.checkequal(-1, 'rrarrrrrrrrra', 'rfind', 'a', 4, 6)
211 self.checkequal(12, 'rrarrrrrrrrra', 'rfind', 'a', 4, None)
212 self.checkequal( 2, 'rrarrrrrrrrra', 'rfind', 'a', None, 6)
213
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000214 self.checkraises(TypeError, 'hello', 'rfind')
215 self.checkraises(TypeError, 'hello', 'rfind', 42)
216
217 def test_index(self):
218 self.checkequal(0, 'abcdefghiabc', 'index', '')
219 self.checkequal(3, 'abcdefghiabc', 'index', 'def')
220 self.checkequal(0, 'abcdefghiabc', 'index', 'abc')
221 self.checkequal(9, 'abcdefghiabc', 'index', 'abc', 1)
222
223 self.checkraises(ValueError, 'abcdefghiabc', 'index', 'hib')
224 self.checkraises(ValueError, 'abcdefghiab', 'index', 'abc', 1)
225 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', 8)
226 self.checkraises(ValueError, 'abcdefghi', 'index', 'ghi', -1)
227
Facundo Batista57d56692007-11-16 18:04:14 +0000228 # to check the ability to pass None as defaults
229 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a')
230 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4)
231 self.checkraises(ValueError, 'rrarrrrrrrrra', 'index', 'a', 4, 6)
232 self.checkequal(12, 'rrarrrrrrrrra', 'index', 'a', 4, None)
233 self.checkequal( 2, 'rrarrrrrrrrra', 'index', 'a', None, 6)
234
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000235 self.checkraises(TypeError, 'hello', 'index')
236 self.checkraises(TypeError, 'hello', 'index', 42)
237
238 def test_rindex(self):
239 self.checkequal(12, 'abcdefghiabc', 'rindex', '')
240 self.checkequal(3, 'abcdefghiabc', 'rindex', 'def')
241 self.checkequal(9, 'abcdefghiabc', 'rindex', 'abc')
242 self.checkequal(0, 'abcdefghiabc', 'rindex', 'abc', 0, -1)
243
244 self.checkraises(ValueError, 'abcdefghiabc', 'rindex', 'hib')
245 self.checkraises(ValueError, 'defghiabc', 'rindex', 'def', 1)
246 self.checkraises(ValueError, 'defghiabc', 'rindex', 'abc', 0, -1)
247 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, 8)
248 self.checkraises(ValueError, 'abcdefghi', 'rindex', 'ghi', 0, -1)
249
Facundo Batista57d56692007-11-16 18:04:14 +0000250 # to check the ability to pass None as defaults
251 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a')
252 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4)
253 self.checkraises(ValueError, 'rrarrrrrrrrra', 'rindex', 'a', 4, 6)
254 self.checkequal(12, 'rrarrrrrrrrra', 'rindex', 'a', 4, None)
255 self.checkequal( 2, 'rrarrrrrrrrra', 'rindex', 'a', None, 6)
256
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000257 self.checkraises(TypeError, 'hello', 'rindex')
258 self.checkraises(TypeError, 'hello', 'rindex', 42)
259
260 def test_lower(self):
261 self.checkequal('hello', 'HeLLo', 'lower')
262 self.checkequal('hello', 'hello', 'lower')
263 self.checkraises(TypeError, 'hello', 'lower', 42)
264
265 def test_upper(self):
266 self.checkequal('HELLO', 'HeLLo', 'upper')
267 self.checkequal('HELLO', 'HELLO', 'upper')
268 self.checkraises(TypeError, 'hello', 'upper', 42)
269
270 def test_expandtabs(self):
271 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
272 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
273 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 4)
274 self.checkequal('abc\r\nab def\ng hi', 'abc\r\nab\tdef\ng\thi', 'expandtabs', 4)
275 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs')
276 self.checkequal('abc\rab def\ng hi', 'abc\rab\tdef\ng\thi', 'expandtabs', 8)
277 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 +0000278 self.checkequal(' a\n b', ' \ta\n\tb', 'expandtabs', 1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000279
280 self.checkraises(TypeError, 'hello', 'expandtabs', 42, 42)
Neal Norwitz5c9a81a2007-06-11 02:16:10 +0000281 # This test is only valid when sizeof(int) == sizeof(void*) == 4.
282 if sys.maxint < (1 << 32) and struct.calcsize('P') == 4:
283 self.checkraises(OverflowError,
284 '\ta\n\tb', 'expandtabs', sys.maxint)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000285
286 def test_split(self):
287 self.checkequal(['this', 'is', 'the', 'split', 'function'],
288 'this is the split function', 'split')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000289
290 # by whitespace
291 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'split')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000292 self.checkequal(['a', 'b c d'], 'a b c d', 'split', None, 1)
293 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
294 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 3)
295 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None, 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000296 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'split', None,
297 sys.maxint-1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000298 self.checkequal(['a b c d'], 'a b c d', 'split', None, 0)
Andrew Dalke725fe402006-05-26 16:22:52 +0000299 self.checkequal(['a b c d'], ' a b c d', 'split', None, 0)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000300 self.checkequal(['a', 'b', 'c d'], 'a b c d', 'split', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000301
Andrew Dalke984b9712006-05-26 11:11:38 +0000302 self.checkequal([], ' ', 'split')
303 self.checkequal(['a'], ' a ', 'split')
304 self.checkequal(['a', 'b'], ' a b ', 'split')
305 self.checkequal(['a', 'b '], ' a b ', 'split', None, 1)
306 self.checkequal(['a', 'b c '], ' a b c ', 'split', None, 1)
307 self.checkequal(['a', 'b', 'c '], ' a b c ', 'split', None, 2)
Andrew Dalke03fb4442006-05-26 11:15:22 +0000308 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'split')
Andrew Dalke005aee22006-05-26 12:28:15 +0000309 aaa = ' a '*20
310 self.checkequal(['a']*20, aaa, 'split')
311 self.checkequal(['a'] + [aaa[4:]], aaa, 'split', None, 1)
Andrew Dalke669fa182006-05-26 13:05:55 +0000312 self.checkequal(['a']*19 + ['a '], aaa, 'split', None, 19)
Andrew Dalke984b9712006-05-26 11:11:38 +0000313
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000314 # by a char
315 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000316 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000317 self.checkequal(['a', 'b|c|d'], 'a|b|c|d', 'split', '|', 1)
318 self.checkequal(['a', 'b', 'c|d'], 'a|b|c|d', 'split', '|', 2)
319 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 3)
320 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000321 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'split', '|',
322 sys.maxint-2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000323 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'split', '|', 0)
324 self.checkequal(['a', '', 'b||c||d'], 'a||b||c||d', 'split', '|', 2)
325 self.checkequal(['endcase ', ''], 'endcase |', 'split', '|')
Andrew Dalke005aee22006-05-26 12:28:15 +0000326 self.checkequal(['', ' startcase'], '| startcase', 'split', '|')
327 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'split', '|')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000328 self.checkequal(['a', '', 'b\x00c\x00d'], 'a\x00\x00b\x00c\x00d', 'split', '\x00', 2)
329
Andrew Dalke005aee22006-05-26 12:28:15 +0000330 self.checkequal(['a']*20, ('a|'*20)[:-1], 'split', '|')
331 self.checkequal(['a']*15 +['a|a|a|a|a'],
332 ('a|'*20)[:-1], 'split', '|', 15)
333
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000334 # by string
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000335 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000336 self.checkequal(['a', 'b//c//d'], 'a//b//c//d', 'split', '//', 1)
337 self.checkequal(['a', 'b', 'c//d'], 'a//b//c//d', 'split', '//', 2)
338 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 3)
339 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//', 4)
Andrew Dalke005aee22006-05-26 12:28:15 +0000340 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'split', '//',
341 sys.maxint-10)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000342 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'split', '//', 0)
343 self.checkequal(['a', '', 'b////c////d'], 'a////b////c////d', 'split', '//', 2)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000344 self.checkequal(['endcase ', ''], 'endcase test', 'split', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000345 self.checkequal(['', ' begincase'], 'test begincase', 'split', 'test')
346 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
347 'split', 'test')
348 self.checkequal(['a', 'bc'], 'abbbc', 'split', 'bb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000349 self.checkequal(['', ''], 'aaa', 'split', 'aaa')
350 self.checkequal(['aaa'], 'aaa', 'split', 'aaa', 0)
351 self.checkequal(['ab', 'ab'], 'abbaab', 'split', 'ba')
352 self.checkequal(['aaaa'], 'aaaa', 'split', 'aab')
353 self.checkequal([''], '', 'split', 'aaa')
354 self.checkequal(['aa'], 'aa', 'split', 'aaa')
Andrew Dalke5cc60092006-05-26 12:31:00 +0000355 self.checkequal(['A', 'bobb'], 'Abbobbbobb', 'split', 'bbobb')
356 self.checkequal(['A', 'B', ''], 'AbbobbBbbobb', 'split', 'bbobb')
Andrew Dalke005aee22006-05-26 12:28:15 +0000357
358 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH')
359 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'split', 'BLAH', 19)
360 self.checkequal(['a']*18 + ['aBLAHa'], ('aBLAH'*20)[:-4],
361 'split', 'BLAH', 18)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000362
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000363 # mixed use of str and unicode
364 self.checkequal([u'a', u'b', u'c d'], 'a b c d', 'split', u' ', 2)
365
366 # argument type
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000367 self.checkraises(TypeError, 'hello', 'split', 42, 42, 42)
368
Andrew Dalke005aee22006-05-26 12:28:15 +0000369 # null case
370 self.checkraises(ValueError, 'hello', 'split', '')
371 self.checkraises(ValueError, 'hello', 'split', '', 0)
372
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000373 def test_rsplit(self):
374 self.checkequal(['this', 'is', 'the', 'rsplit', 'function'],
375 'this is the rsplit function', 'rsplit')
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000376
377 # by whitespace
378 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d ', 'rsplit')
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000379 self.checkequal(['a b c', 'd'], 'a b c d', 'rsplit', None, 1)
380 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
381 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 3)
382 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None, 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000383 self.checkequal(['a', 'b', 'c', 'd'], 'a b c d', 'rsplit', None,
384 sys.maxint-20)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000385 self.checkequal(['a b c d'], 'a b c d', 'rsplit', None, 0)
Andrew Dalke725fe402006-05-26 16:22:52 +0000386 self.checkequal(['a b c d'], 'a b c d ', 'rsplit', None, 0)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000387 self.checkequal(['a b', 'c', 'd'], 'a b c d', 'rsplit', None, 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000388
Andrew Dalke669fa182006-05-26 13:05:55 +0000389 self.checkequal([], ' ', 'rsplit')
390 self.checkequal(['a'], ' a ', 'rsplit')
391 self.checkequal(['a', 'b'], ' a b ', 'rsplit')
392 self.checkequal([' a', 'b'], ' a b ', 'rsplit', None, 1)
393 self.checkequal([' a b','c'], ' a b c ', 'rsplit',
394 None, 1)
395 self.checkequal([' a', 'b', 'c'], ' a b c ', 'rsplit',
396 None, 2)
397 self.checkequal(['a', 'b'], '\n\ta \t\r b \v ', 'rsplit', None, 88)
398 aaa = ' a '*20
399 self.checkequal(['a']*20, aaa, 'rsplit')
400 self.checkequal([aaa[:-4]] + ['a'], aaa, 'rsplit', None, 1)
401 self.checkequal([' a a'] + ['a']*18, aaa, 'rsplit', None, 18)
402
403
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000404 # by a char
405 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|')
406 self.checkequal(['a|b|c', 'd'], 'a|b|c|d', 'rsplit', '|', 1)
407 self.checkequal(['a|b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 2)
408 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 3)
409 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000410 self.checkequal(['a', 'b', 'c', 'd'], 'a|b|c|d', 'rsplit', '|',
411 sys.maxint-100)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000412 self.checkequal(['a|b|c|d'], 'a|b|c|d', 'rsplit', '|', 0)
413 self.checkequal(['a||b||c', '', 'd'], 'a||b||c||d', 'rsplit', '|', 2)
414 self.checkequal(['', ' begincase'], '| begincase', 'rsplit', '|')
Andrew Dalke669fa182006-05-26 13:05:55 +0000415 self.checkequal(['endcase ', ''], 'endcase |', 'rsplit', '|')
416 self.checkequal(['', 'bothcase', ''], '|bothcase|', 'rsplit', '|')
417
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000418 self.checkequal(['a\x00\x00b', 'c', 'd'], 'a\x00\x00b\x00c\x00d', 'rsplit', '\x00', 2)
419
Andrew Dalke669fa182006-05-26 13:05:55 +0000420 self.checkequal(['a']*20, ('a|'*20)[:-1], 'rsplit', '|')
421 self.checkequal(['a|a|a|a|a']+['a']*15,
422 ('a|'*20)[:-1], 'rsplit', '|', 15)
423
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000424 # by string
425 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//')
426 self.checkequal(['a//b//c', 'd'], 'a//b//c//d', 'rsplit', '//', 1)
427 self.checkequal(['a//b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 2)
428 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 3)
429 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//', 4)
Andrew Dalke669fa182006-05-26 13:05:55 +0000430 self.checkequal(['a', 'b', 'c', 'd'], 'a//b//c//d', 'rsplit', '//',
431 sys.maxint-5)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000432 self.checkequal(['a//b//c//d'], 'a//b//c//d', 'rsplit', '//', 0)
433 self.checkequal(['a////b////c', '', 'd'], 'a////b////c////d', 'rsplit', '//', 2)
434 self.checkequal(['', ' begincase'], 'test begincase', 'rsplit', 'test')
Andrew Dalke669fa182006-05-26 13:05:55 +0000435 self.checkequal(['endcase ', ''], 'endcase test', 'rsplit', 'test')
436 self.checkequal(['', ' bothcase ', ''], 'test bothcase test',
437 'rsplit', 'test')
438 self.checkequal(['ab', 'c'], 'abbbc', 'rsplit', 'bb')
439 self.checkequal(['', ''], 'aaa', 'rsplit', 'aaa')
440 self.checkequal(['aaa'], 'aaa', 'rsplit', 'aaa', 0)
441 self.checkequal(['ab', 'ab'], 'abbaab', 'rsplit', 'ba')
442 self.checkequal(['aaaa'], 'aaaa', 'rsplit', 'aab')
443 self.checkequal([''], '', 'rsplit', 'aaa')
444 self.checkequal(['aa'], 'aa', 'rsplit', 'aaa')
445 self.checkequal(['bbob', 'A'], 'bbobbbobbA', 'rsplit', 'bbobb')
446 self.checkequal(['', 'B', 'A'], 'bbobbBbbobbA', 'rsplit', 'bbobb')
447
448 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH')
449 self.checkequal(['a']*20, ('aBLAH'*20)[:-4], 'rsplit', 'BLAH', 19)
450 self.checkequal(['aBLAHa'] + ['a']*18, ('aBLAH'*20)[:-4],
451 'rsplit', 'BLAH', 18)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000452
453 # mixed use of str and unicode
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000454 self.checkequal([u'a b', u'c', u'd'], 'a b c d', 'rsplit', u' ', 2)
Hye-Shik Chang75c00ef2004-01-05 00:29:51 +0000455
456 # argument type
457 self.checkraises(TypeError, 'hello', 'rsplit', 42, 42, 42)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +0000458
Andrew Dalke669fa182006-05-26 13:05:55 +0000459 # null case
460 self.checkraises(ValueError, 'hello', 'rsplit', '')
461 self.checkraises(ValueError, 'hello', 'rsplit', '', 0)
462
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000463 def test_strip(self):
464 self.checkequal('hello', ' hello ', 'strip')
465 self.checkequal('hello ', ' hello ', 'lstrip')
466 self.checkequal(' hello', ' hello ', 'rstrip')
467 self.checkequal('hello', 'hello', 'strip')
468
Neal Norwitzffe33b72003-04-10 22:35:32 +0000469 # strip/lstrip/rstrip with None arg
470 self.checkequal('hello', ' hello ', 'strip', None)
471 self.checkequal('hello ', ' hello ', 'lstrip', None)
472 self.checkequal(' hello', ' hello ', 'rstrip', None)
473 self.checkequal('hello', 'hello', 'strip', None)
474
475 # strip/lstrip/rstrip with str arg
476 self.checkequal('hello', 'xyzzyhelloxyzzy', 'strip', 'xyz')
477 self.checkequal('helloxyzzy', 'xyzzyhelloxyzzy', 'lstrip', 'xyz')
478 self.checkequal('xyzzyhello', 'xyzzyhelloxyzzy', 'rstrip', 'xyz')
479 self.checkequal('hello', 'hello', 'strip', 'xyz')
480
481 # strip/lstrip/rstrip with unicode arg
482 if test_support.have_unicode:
483 self.checkequal(unicode('hello', 'ascii'), 'xyzzyhelloxyzzy',
484 'strip', unicode('xyz', 'ascii'))
485 self.checkequal(unicode('helloxyzzy', 'ascii'), 'xyzzyhelloxyzzy',
486 'lstrip', unicode('xyz', 'ascii'))
487 self.checkequal(unicode('xyzzyhello', 'ascii'), 'xyzzyhelloxyzzy',
488 'rstrip', unicode('xyz', 'ascii'))
489 self.checkequal(unicode('hello', 'ascii'), 'hello',
490 'strip', unicode('xyz', 'ascii'))
491
492 self.checkraises(TypeError, 'hello', 'strip', 42, 42)
493 self.checkraises(TypeError, 'hello', 'lstrip', 42, 42)
494 self.checkraises(TypeError, 'hello', 'rstrip', 42, 42)
495
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000496 def test_ljust(self):
497 self.checkequal('abc ', 'abc', 'ljust', 10)
498 self.checkequal('abc ', 'abc', 'ljust', 6)
499 self.checkequal('abc', 'abc', 'ljust', 3)
500 self.checkequal('abc', 'abc', 'ljust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000501 self.checkequal('abc*******', 'abc', 'ljust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000502 self.checkraises(TypeError, 'abc', 'ljust')
503
504 def test_rjust(self):
505 self.checkequal(' abc', 'abc', 'rjust', 10)
506 self.checkequal(' abc', 'abc', 'rjust', 6)
507 self.checkequal('abc', 'abc', 'rjust', 3)
508 self.checkequal('abc', 'abc', 'rjust', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000509 self.checkequal('*******abc', 'abc', 'rjust', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000510 self.checkraises(TypeError, 'abc', 'rjust')
511
512 def test_center(self):
513 self.checkequal(' abc ', 'abc', 'center', 10)
514 self.checkequal(' abc ', 'abc', 'center', 6)
515 self.checkequal('abc', 'abc', 'center', 3)
516 self.checkequal('abc', 'abc', 'center', 2)
Raymond Hettinger4f8f9762003-11-26 08:21:35 +0000517 self.checkequal('***abc****', 'abc', 'center', 10, '*')
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000518 self.checkraises(TypeError, 'abc', 'center')
519
520 def test_swapcase(self):
521 self.checkequal('hEllO CoMPuTErS', 'HeLLo cOmpUteRs', 'swapcase')
522
523 self.checkraises(TypeError, 'hello', 'swapcase', 42)
524
525 def test_replace(self):
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000526 EQ = self.checkequal
527
528 # Operations on the empty string
529 EQ("", "", "replace", "", "")
Tim Peters80a18f02006-06-01 13:56:26 +0000530 EQ("A", "", "replace", "", "A")
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000531 EQ("", "", "replace", "A", "")
532 EQ("", "", "replace", "A", "A")
533 EQ("", "", "replace", "", "", 100)
534 EQ("", "", "replace", "", "", sys.maxint)
535
536 # interleave (from=="", 'to' gets inserted everywhere)
537 EQ("A", "A", "replace", "", "")
538 EQ("*A*", "A", "replace", "", "*")
539 EQ("*1A*1", "A", "replace", "", "*1")
540 EQ("*-#A*-#", "A", "replace", "", "*-#")
541 EQ("*-A*-A*-", "AA", "replace", "", "*-")
542 EQ("*-A*-A*-", "AA", "replace", "", "*-", -1)
543 EQ("*-A*-A*-", "AA", "replace", "", "*-", sys.maxint)
544 EQ("*-A*-A*-", "AA", "replace", "", "*-", 4)
545 EQ("*-A*-A*-", "AA", "replace", "", "*-", 3)
546 EQ("*-A*-A", "AA", "replace", "", "*-", 2)
547 EQ("*-AA", "AA", "replace", "", "*-", 1)
548 EQ("AA", "AA", "replace", "", "*-", 0)
549
550 # single character deletion (from=="A", to=="")
551 EQ("", "A", "replace", "A", "")
552 EQ("", "AAA", "replace", "A", "")
553 EQ("", "AAA", "replace", "A", "", -1)
554 EQ("", "AAA", "replace", "A", "", sys.maxint)
555 EQ("", "AAA", "replace", "A", "", 4)
556 EQ("", "AAA", "replace", "A", "", 3)
557 EQ("A", "AAA", "replace", "A", "", 2)
558 EQ("AA", "AAA", "replace", "A", "", 1)
559 EQ("AAA", "AAA", "replace", "A", "", 0)
560 EQ("", "AAAAAAAAAA", "replace", "A", "")
561 EQ("BCD", "ABACADA", "replace", "A", "")
562 EQ("BCD", "ABACADA", "replace", "A", "", -1)
563 EQ("BCD", "ABACADA", "replace", "A", "", sys.maxint)
564 EQ("BCD", "ABACADA", "replace", "A", "", 5)
565 EQ("BCD", "ABACADA", "replace", "A", "", 4)
566 EQ("BCDA", "ABACADA", "replace", "A", "", 3)
567 EQ("BCADA", "ABACADA", "replace", "A", "", 2)
568 EQ("BACADA", "ABACADA", "replace", "A", "", 1)
569 EQ("ABACADA", "ABACADA", "replace", "A", "", 0)
570 EQ("BCD", "ABCAD", "replace", "A", "")
571 EQ("BCD", "ABCADAA", "replace", "A", "")
572 EQ("BCD", "BCD", "replace", "A", "")
573 EQ("*************", "*************", "replace", "A", "")
574 EQ("^A^", "^"+"A"*1000+"^", "replace", "A", "", 999)
575
576 # substring deletion (from=="the", to=="")
577 EQ("", "the", "replace", "the", "")
578 EQ("ater", "theater", "replace", "the", "")
579 EQ("", "thethe", "replace", "the", "")
580 EQ("", "thethethethe", "replace", "the", "")
581 EQ("aaaa", "theatheatheathea", "replace", "the", "")
582 EQ("that", "that", "replace", "the", "")
583 EQ("thaet", "thaet", "replace", "the", "")
584 EQ("here and re", "here and there", "replace", "the", "")
585 EQ("here and re and re", "here and there and there",
586 "replace", "the", "", sys.maxint)
587 EQ("here and re and re", "here and there and there",
588 "replace", "the", "", -1)
589 EQ("here and re and re", "here and there and there",
590 "replace", "the", "", 3)
591 EQ("here and re and re", "here and there and there",
592 "replace", "the", "", 2)
593 EQ("here and re and there", "here and there and there",
594 "replace", "the", "", 1)
595 EQ("here and there and there", "here and there and there",
596 "replace", "the", "", 0)
597 EQ("here and re and re", "here and there and there", "replace", "the", "")
598
599 EQ("abc", "abc", "replace", "the", "")
600 EQ("abcdefg", "abcdefg", "replace", "the", "")
601
602 # substring deletion (from=="bob", to=="")
603 EQ("bob", "bbobob", "replace", "bob", "")
604 EQ("bobXbob", "bbobobXbbobob", "replace", "bob", "")
605 EQ("aaaaaaa", "aaaaaaabob", "replace", "bob", "")
606 EQ("aaaaaaa", "aaaaaaa", "replace", "bob", "")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000607
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000608 # single character replace in place (len(from)==len(to)==1)
609 EQ("Who goes there?", "Who goes there?", "replace", "o", "o")
610 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O")
611 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", sys.maxint)
612 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", -1)
613 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 3)
614 EQ("WhO gOes there?", "Who goes there?", "replace", "o", "O", 2)
615 EQ("WhO goes there?", "Who goes there?", "replace", "o", "O", 1)
616 EQ("Who goes there?", "Who goes there?", "replace", "o", "O", 0)
617
618 EQ("Who goes there?", "Who goes there?", "replace", "a", "q")
619 EQ("who goes there?", "Who goes there?", "replace", "W", "w")
620 EQ("wwho goes there?ww", "WWho goes there?WW", "replace", "W", "w")
621 EQ("Who goes there!", "Who goes there?", "replace", "?", "!")
622 EQ("Who goes there!!", "Who goes there??", "replace", "?", "!")
623
624 EQ("Who goes there?", "Who goes there?", "replace", ".", "!")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000625
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000626 # substring replace in place (len(from)==len(to) > 1)
627 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**")
628 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", sys.maxint)
629 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", -1)
630 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 4)
631 EQ("Th** ** a t**sue", "This is a tissue", "replace", "is", "**", 3)
632 EQ("Th** ** a tissue", "This is a tissue", "replace", "is", "**", 2)
633 EQ("Th** is a tissue", "This is a tissue", "replace", "is", "**", 1)
634 EQ("This is a tissue", "This is a tissue", "replace", "is", "**", 0)
635 EQ("cobob", "bobob", "replace", "bob", "cob")
636 EQ("cobobXcobocob", "bobobXbobobob", "replace", "bob", "cob")
637 EQ("bobob", "bobob", "replace", "bot", "bot")
638
639 # replace single character (len(from)==1, len(to)>1)
640 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK")
641 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", -1)
642 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", sys.maxint)
643 EQ("ReyKKjaviKK", "Reykjavik", "replace", "k", "KK", 2)
644 EQ("ReyKKjavik", "Reykjavik", "replace", "k", "KK", 1)
645 EQ("Reykjavik", "Reykjavik", "replace", "k", "KK", 0)
646 EQ("A----B----C----", "A.B.C.", "replace", ".", "----")
647
648 EQ("Reykjavik", "Reykjavik", "replace", "q", "KK")
649
650 # replace substring (len(from)>1, len(to)!=len(from))
651 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
652 "replace", "spam", "ham")
653 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
654 "replace", "spam", "ham", sys.maxint)
655 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
656 "replace", "spam", "ham", -1)
657 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
658 "replace", "spam", "ham", 4)
659 EQ("ham, ham, eggs and ham", "spam, spam, eggs and spam",
660 "replace", "spam", "ham", 3)
661 EQ("ham, ham, eggs and spam", "spam, spam, eggs and spam",
662 "replace", "spam", "ham", 2)
663 EQ("ham, spam, eggs and spam", "spam, spam, eggs and spam",
664 "replace", "spam", "ham", 1)
665 EQ("spam, spam, eggs and spam", "spam, spam, eggs and spam",
666 "replace", "spam", "ham", 0)
667
668 EQ("bobob", "bobobob", "replace", "bobob", "bob")
669 EQ("bobobXbobob", "bobobobXbobobob", "replace", "bobob", "bob")
670 EQ("BOBOBOB", "BOBOBOB", "replace", "bob", "bobby")
Tim Petersbeaec0c2006-05-24 20:27:18 +0000671
Neal Norwitzf71ec5a2006-07-30 06:57:04 +0000672 ba = buffer('a')
673 bb = buffer('b')
674 EQ("bbc", "abc", "replace", ba, bb)
675 EQ("aac", "abc", "replace", bb, ba)
676
Tim Petersbeaec0c2006-05-24 20:27:18 +0000677 #
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000678 self.checkequal('one@two!three!', 'one!two!three!', 'replace', '!', '@', 1)
679 self.checkequal('onetwothree', 'one!two!three!', 'replace', '!', '')
680 self.checkequal('one@two@three!', 'one!two!three!', 'replace', '!', '@', 2)
681 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 3)
682 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@', 4)
683 self.checkequal('one!two!three!', 'one!two!three!', 'replace', '!', '@', 0)
684 self.checkequal('one@two@three@', 'one!two!three!', 'replace', '!', '@')
685 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@')
686 self.checkequal('one!two!three!', 'one!two!three!', 'replace', 'x', '@', 2)
687 self.checkequal('-a-b-c-', 'abc', 'replace', '', '-')
688 self.checkequal('-a-b-c', 'abc', 'replace', '', '-', 3)
689 self.checkequal('abc', 'abc', 'replace', '', '-', 0)
690 self.checkequal('', '', 'replace', '', '')
691 self.checkequal('abc', 'abc', 'replace', 'ab', '--', 0)
692 self.checkequal('abc', 'abc', 'replace', 'xy', '--')
693 # Next three for SF bug 422088: [OSF1 alpha] string.replace(); died with
694 # MemoryError due to empty result (platform malloc issue when requesting
695 # 0 bytes).
696 self.checkequal('', '123', 'replace', '123', '')
697 self.checkequal('', '123123', 'replace', '123', '')
698 self.checkequal('x', '123x123', 'replace', '123', '')
699
700 self.checkraises(TypeError, 'hello', 'replace')
701 self.checkraises(TypeError, 'hello', 'replace', 42)
702 self.checkraises(TypeError, 'hello', 'replace', 42, 'h')
703 self.checkraises(TypeError, 'hello', 'replace', 'h', 42)
704
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000705 def test_replace_overflow(self):
706 # Check for overflow checking on 32 bit machines
Kristján Valur Jónsson170eee92007-05-03 20:09:56 +0000707 if sys.maxint != 2147483647 or struct.calcsize("P") > 4:
Fredrik Lundh0c71f882006-05-25 16:46:54 +0000708 return
709 A2_16 = "A" * (2**16)
710 self.checkraises(OverflowError, A2_16, "replace", "", A2_16)
711 self.checkraises(OverflowError, A2_16, "replace", "A", A2_16)
712 self.checkraises(OverflowError, A2_16, "replace", "AA", A2_16+A2_16)
Andrew Dalkee5488ec2006-05-24 18:55:37 +0000713
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000714 def test_zfill(self):
715 self.checkequal('123', '123', 'zfill', 2)
716 self.checkequal('123', '123', 'zfill', 3)
717 self.checkequal('0123', '123', 'zfill', 4)
718 self.checkequal('+123', '+123', 'zfill', 3)
719 self.checkequal('+123', '+123', 'zfill', 4)
720 self.checkequal('+0123', '+123', 'zfill', 5)
721 self.checkequal('-123', '-123', 'zfill', 3)
722 self.checkequal('-123', '-123', 'zfill', 4)
723 self.checkequal('-0123', '-123', 'zfill', 5)
724 self.checkequal('000', '', 'zfill', 3)
725 self.checkequal('34', '34', 'zfill', 1)
726 self.checkequal('0034', '34', 'zfill', 4)
727
728 self.checkraises(TypeError, '123', 'zfill')
729
730class MixinStrUnicodeUserStringTest:
731 # additional tests that only work for
732 # stringlike objects, i.e. str, unicode, UserString
733 # (but not the string module)
734
735 def test_islower(self):
736 self.checkequal(False, '', 'islower')
737 self.checkequal(True, 'a', 'islower')
738 self.checkequal(False, 'A', 'islower')
739 self.checkequal(False, '\n', 'islower')
740 self.checkequal(True, 'abc', 'islower')
741 self.checkequal(False, 'aBc', 'islower')
742 self.checkequal(True, 'abc\n', 'islower')
743 self.checkraises(TypeError, 'abc', 'islower', 42)
744
745 def test_isupper(self):
746 self.checkequal(False, '', 'isupper')
747 self.checkequal(False, 'a', 'isupper')
748 self.checkequal(True, 'A', 'isupper')
749 self.checkequal(False, '\n', 'isupper')
750 self.checkequal(True, 'ABC', 'isupper')
751 self.checkequal(False, 'AbC', 'isupper')
752 self.checkequal(True, 'ABC\n', 'isupper')
753 self.checkraises(TypeError, 'abc', 'isupper', 42)
754
755 def test_istitle(self):
756 self.checkequal(False, '', 'istitle')
757 self.checkequal(False, 'a', 'istitle')
758 self.checkequal(True, 'A', 'istitle')
759 self.checkequal(False, '\n', 'istitle')
760 self.checkequal(True, 'A Titlecased Line', 'istitle')
761 self.checkequal(True, 'A\nTitlecased Line', 'istitle')
762 self.checkequal(True, 'A Titlecased, Line', 'istitle')
763 self.checkequal(False, 'Not a capitalized String', 'istitle')
764 self.checkequal(False, 'Not\ta Titlecase String', 'istitle')
765 self.checkequal(False, 'Not--a Titlecase String', 'istitle')
766 self.checkequal(False, 'NOT', 'istitle')
767 self.checkraises(TypeError, 'abc', 'istitle', 42)
768
769 def test_isspace(self):
770 self.checkequal(False, '', 'isspace')
771 self.checkequal(False, 'a', 'isspace')
772 self.checkequal(True, ' ', 'isspace')
773 self.checkequal(True, '\t', 'isspace')
774 self.checkequal(True, '\r', 'isspace')
775 self.checkequal(True, '\n', 'isspace')
776 self.checkequal(True, ' \t\r\n', 'isspace')
777 self.checkequal(False, ' \t\r\na', 'isspace')
778 self.checkraises(TypeError, 'abc', 'isspace', 42)
779
780 def test_isalpha(self):
781 self.checkequal(False, '', 'isalpha')
782 self.checkequal(True, 'a', 'isalpha')
783 self.checkequal(True, 'A', 'isalpha')
784 self.checkequal(False, '\n', 'isalpha')
785 self.checkequal(True, 'abc', 'isalpha')
786 self.checkequal(False, 'aBc123', 'isalpha')
787 self.checkequal(False, 'abc\n', 'isalpha')
788 self.checkraises(TypeError, 'abc', 'isalpha', 42)
789
790 def test_isalnum(self):
791 self.checkequal(False, '', 'isalnum')
792 self.checkequal(True, 'a', 'isalnum')
793 self.checkequal(True, 'A', 'isalnum')
794 self.checkequal(False, '\n', 'isalnum')
795 self.checkequal(True, '123abc456', 'isalnum')
796 self.checkequal(True, 'a1b3c', 'isalnum')
797 self.checkequal(False, 'aBc000 ', 'isalnum')
798 self.checkequal(False, 'abc\n', 'isalnum')
799 self.checkraises(TypeError, 'abc', 'isalnum', 42)
800
801 def test_isdigit(self):
802 self.checkequal(False, '', 'isdigit')
803 self.checkequal(False, 'a', 'isdigit')
804 self.checkequal(True, '0', 'isdigit')
805 self.checkequal(True, '0123456789', 'isdigit')
806 self.checkequal(False, '0123456789a', 'isdigit')
807
808 self.checkraises(TypeError, 'abc', 'isdigit', 42)
809
810 def test_title(self):
811 self.checkequal(' Hello ', ' hello ', 'title')
812 self.checkequal('Hello ', 'hello ', 'title')
813 self.checkequal('Hello ', 'Hello ', 'title')
814 self.checkequal('Format This As Title String', "fOrMaT thIs aS titLe String", 'title')
815 self.checkequal('Format,This-As*Title;String', "fOrMaT,thIs-aS*titLe;String", 'title', )
816 self.checkequal('Getint', "getInt", 'title')
817 self.checkraises(TypeError, 'hello', 'title', 42)
818
819 def test_splitlines(self):
820 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\rghi", 'splitlines')
821 self.checkequal(['abc', 'def', '', 'ghi'], "abc\ndef\n\r\nghi", 'splitlines')
822 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi", 'splitlines')
823 self.checkequal(['abc', 'def', 'ghi'], "abc\ndef\r\nghi\n", 'splitlines')
824 self.checkequal(['abc', 'def', 'ghi', ''], "abc\ndef\r\nghi\n\r", 'splitlines')
825 self.checkequal(['', 'abc', 'def', 'ghi', ''], "\nabc\ndef\r\nghi\n\r", 'splitlines')
826 self.checkequal(['\n', 'abc\n', 'def\r\n', 'ghi\n', '\r'], "\nabc\ndef\r\nghi\n\r", 'splitlines', 1)
827
828 self.checkraises(TypeError, 'abc', 'splitlines', 42, 42)
829
830 def test_startswith(self):
831 self.checkequal(True, 'hello', 'startswith', 'he')
832 self.checkequal(True, 'hello', 'startswith', 'hello')
833 self.checkequal(False, 'hello', 'startswith', 'hello world')
834 self.checkequal(True, 'hello', 'startswith', '')
835 self.checkequal(False, 'hello', 'startswith', 'ello')
836 self.checkequal(True, 'hello', 'startswith', 'ello', 1)
837 self.checkequal(True, 'hello', 'startswith', 'o', 4)
838 self.checkequal(False, 'hello', 'startswith', 'o', 5)
839 self.checkequal(True, 'hello', 'startswith', '', 5)
840 self.checkequal(False, 'hello', 'startswith', 'lo', 6)
841 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3)
842 self.checkequal(True, 'helloworld', 'startswith', 'lowo', 3, 7)
843 self.checkequal(False, 'helloworld', 'startswith', 'lowo', 3, 6)
844
845 # test negative indices
846 self.checkequal(True, 'hello', 'startswith', 'he', 0, -1)
847 self.checkequal(True, 'hello', 'startswith', 'he', -53, -1)
848 self.checkequal(False, 'hello', 'startswith', 'hello', 0, -1)
849 self.checkequal(False, 'hello', 'startswith', 'hello world', -1, -10)
850 self.checkequal(False, 'hello', 'startswith', 'ello', -5)
851 self.checkequal(True, 'hello', 'startswith', 'ello', -4)
852 self.checkequal(False, 'hello', 'startswith', 'o', -2)
853 self.checkequal(True, 'hello', 'startswith', 'o', -1)
854 self.checkequal(True, 'hello', 'startswith', '', -3, -3)
855 self.checkequal(False, 'hello', 'startswith', 'lo', -9)
856
857 self.checkraises(TypeError, 'hello', 'startswith')
858 self.checkraises(TypeError, 'hello', 'startswith', 42)
859
Georg Brandl24250812006-06-09 18:45:48 +0000860 # test tuple arguments
861 self.checkequal(True, 'hello', 'startswith', ('he', 'ha'))
862 self.checkequal(False, 'hello', 'startswith', ('lo', 'llo'))
863 self.checkequal(True, 'hello', 'startswith', ('hellox', 'hello'))
864 self.checkequal(False, 'hello', 'startswith', ())
865 self.checkequal(True, 'helloworld', 'startswith', ('hellowo',
866 'rld', 'lowo'), 3)
867 self.checkequal(False, 'helloworld', 'startswith', ('hellowo', 'ello',
868 'rld'), 3)
869 self.checkequal(True, 'hello', 'startswith', ('lo', 'he'), 0, -1)
870 self.checkequal(False, 'hello', 'startswith', ('he', 'hel'), 0, 1)
871 self.checkequal(True, 'hello', 'startswith', ('he', 'hel'), 0, 2)
872
873 self.checkraises(TypeError, 'hello', 'startswith', (42,))
874
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000875 def test_endswith(self):
876 self.checkequal(True, 'hello', 'endswith', 'lo')
877 self.checkequal(False, 'hello', 'endswith', 'he')
878 self.checkequal(True, 'hello', 'endswith', '')
879 self.checkequal(False, 'hello', 'endswith', 'hello world')
880 self.checkequal(False, 'helloworld', 'endswith', 'worl')
881 self.checkequal(True, 'helloworld', 'endswith', 'worl', 3, 9)
882 self.checkequal(True, 'helloworld', 'endswith', 'world', 3, 12)
883 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 1, 7)
884 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 2, 7)
885 self.checkequal(True, 'helloworld', 'endswith', 'lowo', 3, 7)
886 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 4, 7)
887 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, 8)
888 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 1)
889 self.checkequal(False, 'ab', 'endswith', 'ab', 0, 0)
890
891 # test negative indices
892 self.checkequal(True, 'hello', 'endswith', 'lo', -2)
893 self.checkequal(False, 'hello', 'endswith', 'he', -2)
894 self.checkequal(True, 'hello', 'endswith', '', -3, -3)
895 self.checkequal(False, 'hello', 'endswith', 'hello world', -10, -2)
896 self.checkequal(False, 'helloworld', 'endswith', 'worl', -6)
897 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, -1)
898 self.checkequal(True, 'helloworld', 'endswith', 'worl', -5, 9)
899 self.checkequal(True, 'helloworld', 'endswith', 'world', -7, 12)
900 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -99, -3)
901 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -8, -3)
902 self.checkequal(True, 'helloworld', 'endswith', 'lowo', -7, -3)
903 self.checkequal(False, 'helloworld', 'endswith', 'lowo', 3, -4)
904 self.checkequal(False, 'helloworld', 'endswith', 'lowo', -8, -2)
905
906 self.checkraises(TypeError, 'hello', 'endswith')
907 self.checkraises(TypeError, 'hello', 'endswith', 42)
908
Georg Brandl24250812006-06-09 18:45:48 +0000909 # test tuple arguments
910 self.checkequal(False, 'hello', 'endswith', ('he', 'ha'))
911 self.checkequal(True, 'hello', 'endswith', ('lo', 'llo'))
912 self.checkequal(True, 'hello', 'endswith', ('hellox', 'hello'))
913 self.checkequal(False, 'hello', 'endswith', ())
914 self.checkequal(True, 'helloworld', 'endswith', ('hellowo',
915 'rld', 'lowo'), 3)
916 self.checkequal(False, 'helloworld', 'endswith', ('hellowo', 'ello',
917 'rld'), 3, -1)
918 self.checkequal(True, 'hello', 'endswith', ('hell', 'ell'), 0, -1)
919 self.checkequal(False, 'hello', 'endswith', ('he', 'hel'), 0, 1)
920 self.checkequal(True, 'hello', 'endswith', ('he', 'hell'), 0, 4)
921
922 self.checkraises(TypeError, 'hello', 'endswith', (42,))
923
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000924 def test___contains__(self):
925 self.checkequal(True, '', '__contains__', '') # vereq('' in '', True)
926 self.checkequal(True, 'abc', '__contains__', '') # vereq('' in 'abc', True)
927 self.checkequal(False, 'abc', '__contains__', '\0') # vereq('\0' in 'abc', False)
928 self.checkequal(True, '\0abc', '__contains__', '\0') # vereq('\0' in '\0abc', True)
929 self.checkequal(True, 'abc\0', '__contains__', '\0') # vereq('\0' in 'abc\0', True)
930 self.checkequal(True, '\0abc', '__contains__', 'a') # vereq('a' in '\0abc', True)
931 self.checkequal(True, 'asdf', '__contains__', 'asdf') # vereq('asdf' in 'asdf', True)
932 self.checkequal(False, 'asd', '__contains__', 'asdf') # vereq('asdf' in 'asd', False)
933 self.checkequal(False, '', '__contains__', 'asdf') # vereq('asdf' in '', False)
934
935 def test_subscript(self):
936 self.checkequal(u'a', 'abc', '__getitem__', 0)
937 self.checkequal(u'c', 'abc', '__getitem__', -1)
938 self.checkequal(u'a', 'abc', '__getitem__', 0L)
939 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 3))
940 self.checkequal(u'abc', 'abc', '__getitem__', slice(0, 1000))
941 self.checkequal(u'a', 'abc', '__getitem__', slice(0, 1))
942 self.checkequal(u'', 'abc', '__getitem__', slice(0, 0))
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000943
944 self.checkraises(TypeError, 'abc', '__getitem__', 'def')
945
946 def test_slice(self):
947 self.checkequal('abc', 'abc', '__getslice__', 0, 1000)
948 self.checkequal('abc', 'abc', '__getslice__', 0, 3)
949 self.checkequal('ab', 'abc', '__getslice__', 0, 2)
950 self.checkequal('bc', 'abc', '__getslice__', 1, 3)
951 self.checkequal('b', 'abc', '__getslice__', 1, 2)
952 self.checkequal('', 'abc', '__getslice__', 2, 2)
953 self.checkequal('', 'abc', '__getslice__', 1000, 1000)
954 self.checkequal('', 'abc', '__getslice__', 2000, 1000)
955 self.checkequal('', 'abc', '__getslice__', 2, 1)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000956
957 self.checkraises(TypeError, 'abc', '__getslice__', 'def')
958
Thomas Wouters3ccec682007-08-28 15:28:19 +0000959 def test_extended_getslice(self):
960 # Test extended slicing by comparing with list slicing.
961 s = string.ascii_letters + string.digits
962 indices = (0, None, 1, 3, 41, -1, -2, -37)
963 for start in indices:
964 for stop in indices:
965 # Skip step 0 (invalid)
966 for step in indices[1:]:
967 L = list(s)[start:stop:step]
968 self.checkequal(u"".join(L), s, '__getitem__',
969 slice(start, stop, step))
970
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000971 def test_mul(self):
972 self.checkequal('', 'abc', '__mul__', -1)
973 self.checkequal('', 'abc', '__mul__', 0)
974 self.checkequal('abc', 'abc', '__mul__', 1)
975 self.checkequal('abcabcabc', 'abc', '__mul__', 3)
976 self.checkraises(TypeError, 'abc', '__mul__')
977 self.checkraises(TypeError, 'abc', '__mul__', '')
Martin v. Löwis18e16552006-02-15 17:27:45 +0000978 # XXX: on a 64-bit system, this doesn't raise an overflow error,
979 # but either raises a MemoryError, or succeeds (if you have 54TiB)
980 #self.checkraises(OverflowError, 10000*'abc', '__mul__', 2000000000)
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000981
982 def test_join(self):
983 # join now works with any sequence type
984 # moved here, because the argument order is
985 # different in string.join (see the test in
986 # test.test_string.StringTest.test_join)
987 self.checkequal('a b c d', ' ', 'join', ['a', 'b', 'c', 'd'])
988 self.checkequal('abcd', '', 'join', ('a', 'b', 'c', 'd'))
Georg Brandl90e27d32006-06-10 06:40:50 +0000989 self.checkequal('bd', '', 'join', ('', 'b', '', 'd'))
990 self.checkequal('ac', '', 'join', ('a', '', 'c', ''))
Walter Dörwald0fd583c2003-02-21 12:53:50 +0000991 self.checkequal('w x y z', ' ', 'join', Sequence())
992 self.checkequal('abc', 'a', 'join', ('abc',))
993 self.checkequal('z', 'a', 'join', UserList(['z']))
994 if test_support.have_unicode:
995 self.checkequal(unicode('a.b.c'), unicode('.'), 'join', ['a', 'b', 'c'])
996 self.checkequal(unicode('a.b.c'), '.', 'join', [unicode('a'), 'b', 'c'])
997 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', unicode('b'), 'c'])
998 self.checkequal(unicode('a.b.c'), '.', 'join', ['a', 'b', unicode('c')])
999 self.checkraises(TypeError, '.', 'join', ['a', unicode('b'), 3])
1000 for i in [5, 25, 125]:
1001 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1002 ['a' * i] * i)
1003 self.checkequal(((('a' * i) + '-') * i)[:-1], '-', 'join',
1004 ('a' * i,) * i)
1005
1006 self.checkraises(TypeError, ' ', 'join', BadSeq1())
1007 self.checkequal('a b c', ' ', 'join', BadSeq2())
1008
1009 self.checkraises(TypeError, ' ', 'join')
1010 self.checkraises(TypeError, ' ', 'join', 7)
1011 self.checkraises(TypeError, ' ', 'join', Sequence([7, 'hello', 123L]))
Michael W. Hudsonb2308bb2005-10-21 11:45:01 +00001012 try:
1013 def f():
1014 yield 4 + ""
1015 self.fixtype(' ').join(f())
1016 except TypeError, e:
1017 if '+' not in str(e):
1018 self.fail('join() ate exception message')
1019 else:
1020 self.fail('exception not raised')
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001021
1022 def test_formatting(self):
1023 self.checkequal('+hello+', '+%s+', '__mod__', 'hello')
1024 self.checkequal('+10+', '+%d+', '__mod__', 10)
1025 self.checkequal('a', "%c", '__mod__', "a")
1026 self.checkequal('a', "%c", '__mod__', "a")
1027 self.checkequal('"', "%c", '__mod__', 34)
1028 self.checkequal('$', "%c", '__mod__', 36)
1029 self.checkequal('10', "%d", '__mod__', 10)
Walter Dörwald43440a62003-03-31 18:07:50 +00001030 self.checkequal('\x7f', "%c", '__mod__', 0x7f)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001031
1032 for ordinal in (-100, 0x200000):
1033 # unicode raises ValueError, str raises OverflowError
1034 self.checkraises((ValueError, OverflowError), '%c', '__mod__', ordinal)
1035
1036 self.checkequal(' 42', '%3ld', '__mod__', 42)
1037 self.checkequal('0042.00', '%07.2f', '__mod__', 42)
Raymond Hettinger9bfe5332003-08-27 04:55:52 +00001038 self.checkequal('0042.00', '%07.2F', '__mod__', 42)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001039
1040 self.checkraises(TypeError, 'abc', '__mod__')
1041 self.checkraises(TypeError, '%(foo)s', '__mod__', 42)
1042 self.checkraises(TypeError, '%s%s', '__mod__', (42,))
1043 self.checkraises(TypeError, '%c', '__mod__', (None,))
1044 self.checkraises(ValueError, '%(foo', '__mod__', {})
1045 self.checkraises(TypeError, '%(foo)s %(bar)s', '__mod__', ('foo', 42))
1046
1047 # argument names with properly nested brackets are supported
1048 self.checkequal('bar', '%((foo))s', '__mod__', {'(foo)': 'bar'})
1049
1050 # 100 is a magic number in PyUnicode_Format, this forces a resize
1051 self.checkequal(103*'a'+'x', '%sx', '__mod__', 103*'a')
1052
1053 self.checkraises(TypeError, '%*s', '__mod__', ('foo', 'bar'))
1054 self.checkraises(TypeError, '%10.*f', '__mod__', ('foo', 42.))
1055 self.checkraises(ValueError, '%10', '__mod__', (42,))
1056
1057 def test_floatformatting(self):
1058 # float formatting
1059 for prec in xrange(100):
1060 format = '%%.%if' % prec
1061 value = 0.01
1062 for x in xrange(60):
1063 value = value * 3.141592655 / 3.0 * 10.0
1064 # The formatfloat() code in stringobject.c and
1065 # unicodeobject.c uses a 120 byte buffer and switches from
1066 # 'f' formatting to 'g' at precision 50, so we expect
1067 # OverflowErrors for the ranges x < 50 and prec >= 67.
1068 if x < 50 and prec >= 67:
1069 self.checkraises(OverflowError, format, "__mod__", value)
1070 else:
1071 self.checkcall(format, "__mod__", value)
1072
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001073 def test_inplace_rewrites(self):
1074 # Check that strings don't copy and modify cached single-character strings
1075 self.checkequal('a', 'A', 'lower')
1076 self.checkequal(True, 'A', 'isupper')
1077 self.checkequal('A', 'a', 'upper')
1078 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +00001079
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001080 self.checkequal('a', 'A', 'replace', 'A', 'a')
1081 self.checkequal(True, 'A', 'isupper')
1082
1083 self.checkequal('A', 'a', 'capitalize')
1084 self.checkequal(True, 'a', 'islower')
Tim Petersd95d5932006-05-25 21:52:19 +00001085
Andrew Dalke2bddcbf2006-05-25 16:30:52 +00001086 self.checkequal('A', 'a', 'swapcase')
1087 self.checkequal(True, 'a', 'islower')
1088
1089 self.checkequal('A', 'a', 'title')
1090 self.checkequal(True, 'a', 'islower')
1091
Fredrik Lundh06a69dd2006-05-26 08:54:28 +00001092 def test_partition(self):
1093
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001094 self.checkequal(('this is the par', 'ti', 'tion method'),
1095 'this is the partition method', 'partition', 'ti')
Fredrik Lundh06a69dd2006-05-26 08:54:28 +00001096
1097 # from raymond's original specification
1098 S = 'http://www.python.org'
1099 self.checkequal(('http', '://', 'www.python.org'), S, 'partition', '://')
1100 self.checkequal(('http://www.python.org', '', ''), S, 'partition', '?')
1101 self.checkequal(('', 'http://', 'www.python.org'), S, 'partition', 'http://')
1102 self.checkequal(('http://www.python.', 'org', ''), S, 'partition', 'org')
1103
1104 self.checkraises(ValueError, S, 'partition', '')
1105 self.checkraises(TypeError, S, 'partition', None)
1106
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001107 def test_rpartition(self):
1108
1109 self.checkequal(('this is the rparti', 'ti', 'on method'),
1110 'this is the rpartition method', 'rpartition', 'ti')
1111
1112 # from raymond's original specification
1113 S = 'http://www.python.org'
1114 self.checkequal(('http', '://', 'www.python.org'), S, 'rpartition', '://')
Raymond Hettingera0c95fa2006-09-04 15:32:48 +00001115 self.checkequal(('', '', 'http://www.python.org'), S, 'rpartition', '?')
Fredrik Lundh9c0e9c02006-05-26 18:24:15 +00001116 self.checkequal(('', 'http://', 'www.python.org'), S, 'rpartition', 'http://')
1117 self.checkequal(('http://www.python.', 'org', ''), S, 'rpartition', 'org')
1118
1119 self.checkraises(ValueError, S, 'rpartition', '')
1120 self.checkraises(TypeError, S, 'rpartition', None)
1121
Walter Dörwald57d88e52004-08-26 16:53:04 +00001122
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001123class MixinStrStringUserStringTest:
1124 # Additional tests for 8bit strings, i.e. str, UserString and
1125 # the string module
1126
1127 def test_maketrans(self):
1128 self.assertEqual(
1129 ''.join(map(chr, xrange(256))).replace('abc', 'xyz'),
1130 string.maketrans('abc', 'xyz')
1131 )
1132 self.assertRaises(ValueError, string.maketrans, 'abc', 'xyzw')
1133
1134 def test_translate(self):
1135 table = string.maketrans('abc', 'xyz')
1136 self.checkequal('xyzxyz', 'xyzabcdef', 'translate', table, 'def')
1137
1138 table = string.maketrans('a', 'A')
1139 self.checkequal('Abc', 'abc', 'translate', table)
1140 self.checkequal('xyz', 'xyz', 'translate', table)
1141 self.checkequal('yz', 'xyz', 'translate', table, 'x')
Neal Norwitz0d4c06e2007-04-25 06:30:05 +00001142 self.checkequal('yx', 'zyzzx', 'translate', None, 'z')
Raymond Hettinger4db5fe92007-04-12 04:10:00 +00001143 self.checkequal('zyzzx', 'zyzzx', 'translate', None, '')
Neal Norwitz0d4c06e2007-04-25 06:30:05 +00001144 self.checkequal('zyzzx', 'zyzzx', 'translate', None)
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001145 self.checkraises(ValueError, 'xyz', 'translate', 'too short', 'strip')
1146 self.checkraises(ValueError, 'xyz', 'translate', 'too short')
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001147
1148
Walter Dörwald0fd583c2003-02-21 12:53:50 +00001149class MixinStrUserStringTest:
1150 # Additional tests that only work with
1151 # 8bit compatible object, i.e. str and UserString
Jeremy Hyltonf82b04e2000-07-10 17:08:42 +00001152
Walter Dörwald6eea7892005-07-28 16:49:15 +00001153 if test_support.have_unicode:
1154 def test_encoding_decoding(self):
1155 codecs = [('rot13', 'uryyb jbeyq'),
1156 ('base64', 'aGVsbG8gd29ybGQ=\n'),
1157 ('hex', '68656c6c6f20776f726c64'),
1158 ('uu', 'begin 666 <data>\n+:&5L;&\\@=V]R;&0 \n \nend\n')]
1159 for encoding, data in codecs:
1160 self.checkequal(data, 'hello world', 'encode', encoding)
1161 self.checkequal('hello world', data, 'decode', encoding)
1162 # zlib is optional, so we make the test optional too...
1163 try:
1164 import zlib
1165 except ImportError:
1166 pass
1167 else:
1168 data = 'x\x9c\xcbH\xcd\xc9\xc9W(\xcf/\xcaI\x01\x00\x1a\x0b\x04]'
1169 self.checkequal(data, 'hello world', 'encode', 'zlib')
1170 self.checkequal('hello world', data, 'decode', 'zlib')
Walter Dörwald97951de2003-03-26 14:31:25 +00001171
Walter Dörwald6eea7892005-07-28 16:49:15 +00001172 self.checkraises(TypeError, 'xyz', 'decode', 42)
1173 self.checkraises(TypeError, 'xyz', 'encode', 42)
Walter Dörwald57d88e52004-08-26 16:53:04 +00001174
1175
1176class MixinStrUnicodeTest:
Tim Peters108f1372004-08-27 05:36:07 +00001177 # Additional tests that only work with str and unicode.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001178
1179 def test_bug1001011(self):
1180 # Make sure join returns a NEW object for single item sequences
Tim Peters108f1372004-08-27 05:36:07 +00001181 # involving a subclass.
1182 # Make sure that it is of the appropriate type.
1183 # Check the optimisation still occurs for standard objects.
Walter Dörwald57d88e52004-08-26 16:53:04 +00001184 t = self.type2test
1185 class subclass(t):
1186 pass
1187 s1 = subclass("abcd")
1188 s2 = t().join([s1])
1189 self.assert_(s1 is not s2)
1190 self.assert_(type(s2) is t)
Tim Peters108f1372004-08-27 05:36:07 +00001191
1192 s1 = t("abcd")
1193 s2 = t().join([s1])
1194 self.assert_(s1 is s2)
1195
1196 # Should also test mixed-type join.
1197 if t is unicode:
1198 s1 = subclass("abcd")
1199 s2 = "".join([s1])
1200 self.assert_(s1 is not s2)
1201 self.assert_(type(s2) is t)
1202
1203 s1 = t("abcd")
1204 s2 = "".join([s1])
1205 self.assert_(s1 is s2)
1206
1207 elif t is str:
1208 s1 = subclass("abcd")
1209 s2 = u"".join([s1])
1210 self.assert_(s1 is not s2)
1211 self.assert_(type(s2) is unicode) # promotes!
1212
1213 s1 = t("abcd")
1214 s2 = u"".join([s1])
1215 self.assert_(s1 is not s2)
1216 self.assert_(type(s2) is unicode) # promotes!
1217
1218 else:
1219 self.fail("unexpected type for MixinStrUnicodeTest %r" % t)