blob: 87a70fb18dc9d1f012ff5c1edae7325d4dae7d07 [file] [log] [blame]
Guido van Rossum85f18201992-11-27 22:53:50 +00001# Python test set -- part 6, built-in types
2
3from test_support import *
4
5print '6. Built-in types'
6
7print '6.1 Truth value testing'
8if None: raise TestFailed, 'None is true instead of false'
9if 0: raise TestFailed, '0 is true instead of false'
10if 0L: raise TestFailed, '0L is true instead of false'
11if 0.0: raise TestFailed, '0.0 is true instead of false'
12if '': raise TestFailed, '\'\' is true instead of false'
13if (): raise TestFailed, '() is true instead of false'
14if []: raise TestFailed, '[] is true instead of false'
15if {}: raise TestFailed, '{} is true instead of false'
16if not 1: raise TestFailed, '1 is false instead of true'
17if not 1L: raise TestFailed, '1L is false instead of true'
18if not 1.0: raise TestFailed, '1.0 is false instead of true'
19if not 'x': raise TestFailed, '\'x\' is false instead of true'
20if not (1, 1): raise TestFailed, '(1, 1) is false instead of true'
21if not [1]: raise TestFailed, '[1] is false instead of true'
22if not {'x': 1}: raise TestFailed, '{\'x\': 1} is false instead of true'
23def f(): pass
Guido van Rossumd3166071993-05-24 14:16:22 +000024class C: pass
Guido van Rossum85f18201992-11-27 22:53:50 +000025import sys
26x = C()
27if not f: raise TestFailed, 'f is false instead of true'
28if not C: raise TestFailed, 'C is false instead of true'
29if not sys: raise TestFailed, 'sys is false instead of true'
30if not x: raise TestFailed, 'x is false instead of true'
31
32print '6.2 Boolean operations'
33if 0 or 0: raise TestFailed, '0 or 0 is true instead of false'
34if 1 and 1: pass
35else: raise TestFailed, '1 and 1 is false instead of false'
36if not 1: raise TestFailed, 'not 1 is true instead of false'
37
38print '6.3 Comparisons'
Fred Drake132dce22000-12-12 23:11:42 +000039if 0 < 1 <= 1 == 1 >= 1 > 0 != 1: pass
Guido van Rossum85f18201992-11-27 22:53:50 +000040else: raise TestFailed, 'int comparisons failed'
Fred Drake132dce22000-12-12 23:11:42 +000041if 0L < 1L <= 1L == 1L >= 1L > 0L != 1L: pass
Guido van Rossum85f18201992-11-27 22:53:50 +000042else: raise TestFailed, 'long int comparisons failed'
Fred Drake132dce22000-12-12 23:11:42 +000043if 0.0 < 1.0 <= 1.0 == 1.0 >= 1.0 > 0.0 != 1.0: pass
Guido van Rossum85f18201992-11-27 22:53:50 +000044else: raise TestFailed, 'float comparisons failed'
45if '' < 'a' <= 'a' == 'a' < 'abc' < 'abd' < 'b': pass
46else: raise TestFailed, 'string comparisons failed'
47if 0 in [0] and 0 not in [1]: pass
48else: raise TestFailed, 'membership test failed'
49if None is None and [] is not []: pass
50else: raise TestFailed, 'identity test failed'
51
Neil Schemenauereff72442002-03-24 01:24:54 +000052try: float('')
53except ValueError: pass
54else: raise TestFailed, "float('') didn't raise ValueError"
55
56try: float('5\0')
57except ValueError: pass
58else: raise TestFailed, "float('5\0') didn't raise ValueError"
59
60try: 5.0 / 0.0
61except ZeroDivisionError: pass
62else: raise TestFailed, "5.0 / 0.0 didn't raise ZeroDivisionError"
63
64try: 5.0 // 0.0
65except ZeroDivisionError: pass
66else: raise TestFailed, "5.0 // 0.0 didn't raise ZeroDivisionError"
67
68try: 5.0 % 0.0
69except ZeroDivisionError: pass
70else: raise TestFailed, "5.0 % 0.0 didn't raise ZeroDivisionError"
71
72try: 5 / 0L
73except ZeroDivisionError: pass
74else: raise TestFailed, "5 / 0L didn't raise ZeroDivisionError"
75
76try: 5 // 0L
77except ZeroDivisionError: pass
78else: raise TestFailed, "5 // 0L didn't raise ZeroDivisionError"
79
80try: 5 % 0L
81except ZeroDivisionError: pass
82else: raise TestFailed, "5 % 0L didn't raise ZeroDivisionError"
83
Guido van Rossum85f18201992-11-27 22:53:50 +000084print '6.4 Numeric types (mostly conversions)'
Fred Drake132dce22000-12-12 23:11:42 +000085if 0 != 0L or 0 != 0.0 or 0L != 0.0: raise TestFailed, 'mixed comparisons'
86if 1 != 1L or 1 != 1.0 or 1L != 1.0: raise TestFailed, 'mixed comparisons'
87if -1 != -1L or -1 != -1.0 or -1L != -1.0:
Fred Drake004d5e62000-10-23 17:22:08 +000088 raise TestFailed, 'int/long/float value not equal'
Guido van Rossum85f18201992-11-27 22:53:50 +000089if int(1.9) == 1 == int(1.1) and int(-1.1) == -1 == int(-1.9): pass
90else: raise TestFailed, 'int() does not round properly'
91if long(1.9) == 1L == long(1.1) and long(-1.1) == -1L == long(-1.9): pass
92else: raise TestFailed, 'long() does not round properly'
93if float(1) == 1.0 and float(-1) == -1.0 and float(0) == 0.0: pass
94else: raise TestFailed, 'float() does not work properly'
Guido van Rossum80530ce1993-01-21 15:36:40 +000095print '6.4.1 32-bit integers'
Fred Drake132dce22000-12-12 23:11:42 +000096if 12 + 24 != 36: raise TestFailed, 'int op'
97if 12 + (-24) != -12: raise TestFailed, 'int op'
98if (-12) + 24 != 12: raise TestFailed, 'int op'
99if (-12) + (-24) != -36: raise TestFailed, 'int op'
Guido van Rossum80530ce1993-01-21 15:36:40 +0000100if not 12 < 24: raise TestFailed, 'int op'
101if not -24 < -12: raise TestFailed, 'int op'
Guido van Rossumb6775db1994-08-01 11:34:53 +0000102# Test for a particular bug in integer multiply
103xsize, ysize, zsize = 238, 356, 4
104if not (xsize*ysize*zsize == zsize*xsize*ysize == 338912):
Fred Drake004d5e62000-10-23 17:22:08 +0000105 raise TestFailed, 'int mul commutativity'
Tim Petersa3c01ce2001-12-04 23:05:10 +0000106# And another.
107m = -sys.maxint - 1
108for divisor in 1, 2, 4, 8, 16, 32:
Tim Peters6d30c3e2001-12-05 00:30:09 +0000109 j = m // divisor
Tim Petersa3c01ce2001-12-04 23:05:10 +0000110 prod = divisor * j
111 if prod != m:
112 raise TestFailed, "%r * %r == %r != %r" % (divisor, j, prod, m)
113 if type(prod) is not int:
114 raise TestFailed, ("expected type(prod) to be int, not %r" %
115 type(prod))
116# Check for expected * overflow to long.
117for divisor in 1, 2, 4, 8, 16, 32:
Tim Peters6d30c3e2001-12-05 00:30:09 +0000118 j = m // divisor - 1
Tim Petersa3c01ce2001-12-04 23:05:10 +0000119 prod = divisor * j
120 if type(prod) is not long:
121 raise TestFailed, ("expected type(%r) to be long, not %r" %
122 (prod, type(prod)))
123# Check for expected * overflow to long.
124m = sys.maxint
125for divisor in 1, 2, 4, 8, 16, 32:
Tim Peters6d30c3e2001-12-05 00:30:09 +0000126 j = m // divisor + 1
Tim Petersa3c01ce2001-12-04 23:05:10 +0000127 prod = divisor * j
128 if type(prod) is not long:
129 raise TestFailed, ("expected type(%r) to be long, not %r" %
130 (prod, type(prod)))
131
Guido van Rossum80530ce1993-01-21 15:36:40 +0000132print '6.4.2 Long integers'
Fred Drake132dce22000-12-12 23:11:42 +0000133if 12L + 24L != 36L: raise TestFailed, 'long op'
134if 12L + (-24L) != -12L: raise TestFailed, 'long op'
135if (-12L) + 24L != 12L: raise TestFailed, 'long op'
136if (-12L) + (-24L) != -36L: raise TestFailed, 'long op'
Guido van Rossum80530ce1993-01-21 15:36:40 +0000137if not 12L < 24L: raise TestFailed, 'long op'
138if not -24L < -12L: raise TestFailed, 'long op'
Guido van Rossum74629421998-05-26 14:51:55 +0000139x = sys.maxint
140if int(long(x)) != x: raise TestFailed, 'long op'
141try: int(long(x)+1L)
142except OverflowError: pass
143else:raise TestFailed, 'long op'
144x = -x
145if int(long(x)) != x: raise TestFailed, 'long op'
146x = x-1
147if int(long(x)) != x: raise TestFailed, 'long op'
148try: int(long(x)-1L)
149except OverflowError: pass
150else:raise TestFailed, 'long op'
Neil Schemenauereff72442002-03-24 01:24:54 +0000151
152try: 5 << -5
153except ValueError: pass
154else: raise TestFailed, 'int negative shift <<'
155
156try: 5L << -5L
157except ValueError: pass
158else: raise TestFailed, 'long negative shift <<'
159
160try: 5 >> -5
161except ValueError: pass
162else: raise TestFailed, 'int negative shift >>'
163
164try: 5L >> -5L
165except ValueError: pass
166else: raise TestFailed, 'long negative shift >>'
167
Guido van Rossum80530ce1993-01-21 15:36:40 +0000168print '6.4.3 Floating point numbers'
Fred Drake132dce22000-12-12 23:11:42 +0000169if 12.0 + 24.0 != 36.0: raise TestFailed, 'float op'
170if 12.0 + (-24.0) != -12.0: raise TestFailed, 'float op'
171if (-12.0) + 24.0 != 12.0: raise TestFailed, 'float op'
172if (-12.0) + (-24.0) != -36.0: raise TestFailed, 'float op'
Guido van Rossum80530ce1993-01-21 15:36:40 +0000173if not 12.0 < 24.0: raise TestFailed, 'float op'
174if not -24.0 < -12.0: raise TestFailed, 'float op'
Guido van Rossum85f18201992-11-27 22:53:50 +0000175
176print '6.5 Sequence types'
177
178print '6.5.1 Strings'
Fred Drake132dce22000-12-12 23:11:42 +0000179if len('') != 0: raise TestFailed, 'len(\'\')'
180if len('a') != 1: raise TestFailed, 'len(\'a\')'
181if len('abcdef') != 6: raise TestFailed, 'len(\'abcdef\')'
182if 'xyz' + 'abcde' != 'xyzabcde': raise TestFailed, 'string concatenation'
183if 'xyz'*3 != 'xyzxyzxyz': raise TestFailed, 'string repetition *3'
184if 0*'abcde' != '': raise TestFailed, 'string repetition 0*'
185if min('abc') != 'a' or max('abc') != 'c': raise TestFailed, 'min/max string'
Guido van Rossum85f18201992-11-27 22:53:50 +0000186if 'a' in 'abc' and 'b' in 'abc' and 'c' in 'abc' and 'd' not in 'abc': pass
187else: raise TestFailed, 'in/not in string'
Guido van Rossumb6775db1994-08-01 11:34:53 +0000188x = 'x'*103
189if '%s!'%x != x+'!': raise TestFailed, 'nasty string formatting bug'
Guido van Rossum85f18201992-11-27 22:53:50 +0000190
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000191#extended slices for strings
192a = '0123456789'
193vereq(a[::], a)
194vereq(a[::2], '02468')
195vereq(a[1::2], '13579')
196vereq(a[::-1],'9876543210')
197vereq(a[::-2], '97531')
198vereq(a[3::-2], '31')
199vereq(a[-100:100:], a)
200vereq(a[100:-100:-1], a[::-1])
201vereq(a[-100L:100L:2L], '02468')
202
203if have_unicode:
204 a = unicode('0123456789', 'ascii')
205 vereq(a[::], a)
206 vereq(a[::2], unicode('02468', 'ascii'))
207 vereq(a[1::2], unicode('13579', 'ascii'))
208 vereq(a[::-1], unicode('9876543210', 'ascii'))
209 vereq(a[::-2], unicode('97531', 'ascii'))
210 vereq(a[3::-2], unicode('31', 'ascii'))
211 vereq(a[-100:100:], a)
212 vereq(a[100:-100:-1], a[::-1])
213 vereq(a[-100L:100L:2L], unicode('02468', 'ascii'))
Tim Petersc411dba2002-07-16 21:35:23 +0000214
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000215
Guido van Rossum85f18201992-11-27 22:53:50 +0000216print '6.5.2 Tuples'
Fred Drake132dce22000-12-12 23:11:42 +0000217if len(()) != 0: raise TestFailed, 'len(())'
218if len((1,)) != 1: raise TestFailed, 'len((1,))'
219if len((1,2,3,4,5,6)) != 6: raise TestFailed, 'len((1,2,3,4,5,6))'
220if (1,2)+(3,4) != (1,2,3,4): raise TestFailed, 'tuple concatenation'
221if (1,2)*3 != (1,2,1,2,1,2): raise TestFailed, 'tuple repetition *3'
222if 0*(1,2,3) != (): raise TestFailed, 'tuple repetition 0*'
223if min((1,2)) != 1 or max((1,2)) != 2: raise TestFailed, 'min/max tuple'
Guido van Rossum85f18201992-11-27 22:53:50 +0000224if 0 in (0,1,2) and 1 in (0,1,2) and 2 in (0,1,2) and 3 not in (0,1,2): pass
225else: raise TestFailed, 'in/not in tuple'
Neil Schemenauereff72442002-03-24 01:24:54 +0000226try: ()[0]
227except IndexError: pass
228else: raise TestFailed, "tuple index error didn't raise IndexError"
229x = ()
230x += ()
231if x != (): raise TestFailed, 'tuple inplace add from () to () failed'
232x += (1,)
233if x != (1,): raise TestFailed, 'tuple resize from () failed'
Guido van Rossum85f18201992-11-27 22:53:50 +0000234
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000235# extended slicing - subscript only for tuples
236a = (0,1,2,3,4)
237vereq(a[::], a)
238vereq(a[::2], (0,2,4))
239vereq(a[1::2], (1,3))
240vereq(a[::-1], (4,3,2,1,0))
241vereq(a[::-2], (4,2,0))
242vereq(a[3::-2], (3,1))
243vereq(a[-100:100:], a)
244vereq(a[100:-100:-1], a[::-1])
245vereq(a[-100L:100L:2L], (0,2,4))
246
Guido van Rossum10f36d92002-06-21 02:14:10 +0000247# Check that a specific bug in _PyTuple_Resize() is squashed.
248def f():
249 for i in range(1000):
250 yield i
251vereq(list(tuple(f())), range(1000))
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000252
Guido van Rossum85f18201992-11-27 22:53:50 +0000253print '6.5.3 Lists'
Fred Drake132dce22000-12-12 23:11:42 +0000254if len([]) != 0: raise TestFailed, 'len([])'
255if len([1,]) != 1: raise TestFailed, 'len([1,])'
256if len([1,2,3,4,5,6]) != 6: raise TestFailed, 'len([1,2,3,4,5,6])'
257if [1,2]+[3,4] != [1,2,3,4]: raise TestFailed, 'list concatenation'
258if [1,2]*3 != [1,2,1,2,1,2]: raise TestFailed, 'list repetition *3'
259if [1,2]*3L != [1,2,1,2,1,2]: raise TestFailed, 'list repetition *3L'
260if 0*[1,2,3] != []: raise TestFailed, 'list repetition 0*'
261if 0L*[1,2,3] != []: raise TestFailed, 'list repetition 0L*'
262if min([1,2]) != 1 or max([1,2]) != 2: raise TestFailed, 'min/max list'
Guido van Rossum85f18201992-11-27 22:53:50 +0000263if 0 in [0,1,2] and 1 in [0,1,2] and 2 in [0,1,2] and 3 not in [0,1,2]: pass
264else: raise TestFailed, 'in/not in list'
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000265a = [1, 2, 3, 4, 5]
266a[:-1] = a
267if a != [1, 2, 3, 4, 5, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000268 raise TestFailed, "list self-slice-assign (head)"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000269a = [1, 2, 3, 4, 5]
270a[1:] = a
271if a != [1, 1, 2, 3, 4, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000272 raise TestFailed, "list self-slice-assign (tail)"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000273a = [1, 2, 3, 4, 5]
274a[1:-1] = a
275if a != [1, 1, 2, 3, 4, 5, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000276 raise TestFailed, "list self-slice-assign (center)"
Neil Schemenauereff72442002-03-24 01:24:54 +0000277try: [][0]
278except IndexError: pass
279else: raise TestFailed, "list index error didn't raise IndexError"
280try: [][0] = 5
281except IndexError: pass
282else: raise TestFailed, "list assignment index error didn't raise IndexError"
283try: [].pop()
284except IndexError: pass
285else: raise TestFailed, "empty list.pop() didn't raise IndexError"
286try: [1].pop(5)
287except IndexError: pass
288else: raise TestFailed, "[1].pop(5) didn't raise IndexError"
289try: [][0:1] = 5
290except TypeError: pass
291else: raise TestFailed, "bad list slice assignment didn't raise TypeError"
292try: [].extend(None)
293except TypeError: pass
294else: raise TestFailed, "list.extend(None) didn't raise TypeError"
295a = [1, 2, 3, 4]
296a *= 0
297if a != []:
298 raise TestFailed, "list inplace repeat"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000299
Guido van Rossum85f18201992-11-27 22:53:50 +0000300
301print '6.5.3a Additional list operations'
302a = [0,1,2,3,4]
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000303a[0L] = 1
304a[1L] = 2
305a[2L] = 3
Fred Drake132dce22000-12-12 23:11:42 +0000306if a != [1,2,3,3,4]: raise TestFailed, 'list item assignment [0L], [1L], [2L]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000307a[0] = 5
308a[1] = 6
309a[2] = 7
Fred Drake132dce22000-12-12 23:11:42 +0000310if a != [5,6,7,3,4]: raise TestFailed, 'list item assignment [0], [1], [2]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000311a[-2L] = 88
312a[-1L] = 99
Fred Drake132dce22000-12-12 23:11:42 +0000313if a != [5,6,7,88,99]: raise TestFailed, 'list item assignment [-2L], [-1L]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000314a[-2] = 8
315a[-1] = 9
Fred Drake132dce22000-12-12 23:11:42 +0000316if a != [5,6,7,8,9]: raise TestFailed, 'list item assignment [-2], [-1]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000317a[:2] = [0,4]
318a[-3:] = []
319a[1:1] = [1,2,3]
Fred Drake132dce22000-12-12 23:11:42 +0000320if a != [0,1,2,3,4]: raise TestFailed, 'list slice assignment'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000321a[ 1L : 4L] = [7,8,9]
Fred Drake132dce22000-12-12 23:11:42 +0000322if a != [0,7,8,9,4]: raise TestFailed, 'list slice assignment using long ints'
Guido van Rossum85f18201992-11-27 22:53:50 +0000323del a[1:4]
Fred Drake132dce22000-12-12 23:11:42 +0000324if a != [0,4]: raise TestFailed, 'list slice deletion'
Guido van Rossum85f18201992-11-27 22:53:50 +0000325del a[0]
Fred Drake132dce22000-12-12 23:11:42 +0000326if a != [4]: raise TestFailed, 'list item deletion [0]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000327del a[-1]
Fred Drake132dce22000-12-12 23:11:42 +0000328if a != []: raise TestFailed, 'list item deletion [-1]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000329a=range(0,5)
330del a[1L:4L]
Fred Drake132dce22000-12-12 23:11:42 +0000331if a != [0,4]: raise TestFailed, 'list slice deletion'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000332del a[0L]
Fred Drake132dce22000-12-12 23:11:42 +0000333if a != [4]: raise TestFailed, 'list item deletion [0]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000334del a[-1L]
Fred Drake132dce22000-12-12 23:11:42 +0000335if a != []: raise TestFailed, 'list item deletion [-1]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000336a.append(0)
337a.append(1)
338a.append(2)
Fred Drake132dce22000-12-12 23:11:42 +0000339if a != [0,1,2]: raise TestFailed, 'list append'
Guido van Rossum85f18201992-11-27 22:53:50 +0000340a.insert(0, -2)
341a.insert(1, -1)
342a.insert(2,0)
Fred Drake132dce22000-12-12 23:11:42 +0000343if a != [-2,-1,0,0,1,2]: raise TestFailed, 'list insert'
344if a.count(0) != 2: raise TestFailed, ' list count'
345if a.index(0) != 2: raise TestFailed, 'list index'
Guido van Rossum85f18201992-11-27 22:53:50 +0000346a.remove(0)
Fred Drake132dce22000-12-12 23:11:42 +0000347if a != [-2,-1,0,1,2]: raise TestFailed, 'list remove'
Guido van Rossum85f18201992-11-27 22:53:50 +0000348a.reverse()
Fred Drake132dce22000-12-12 23:11:42 +0000349if a != [2,1,0,-1,-2]: raise TestFailed, 'list reverse'
Guido van Rossum85f18201992-11-27 22:53:50 +0000350a.sort()
Fred Drake132dce22000-12-12 23:11:42 +0000351if a != [-2,-1,0,1,2]: raise TestFailed, 'list sort'
Guido van Rossume61fa0a1993-10-22 13:56:35 +0000352def revcmp(a, b): return cmp(b, a)
353a.sort(revcmp)
Fred Drake132dce22000-12-12 23:11:42 +0000354if a != [2,1,0,-1,-2]: raise TestFailed, 'list sort with cmp func'
Guido van Rossumd151d341998-02-25 17:51:50 +0000355# The following dumps core in unpatched Python 1.5:
356def myComparison(x,y):
357 return cmp(x%3, y%7)
358z = range(12)
359z.sort(myComparison)
Guido van Rossum85f18201992-11-27 22:53:50 +0000360
Neal Norwitz6fc36c52002-06-13 22:23:06 +0000361try: z.sort(2)
362except TypeError: pass
363else: raise TestFailed, 'list sort compare function is not callable'
364
365def selfmodifyingComparison(x,y):
366 z[0] = 1
367 return cmp(x, y)
368try: z.sort(selfmodifyingComparison)
369except TypeError: pass
370else: raise TestFailed, 'modifying list during sort'
371
372try: z.sort(lambda x, y: 's')
373except TypeError: pass
374else: raise TestFailed, 'list sort compare function does not return int'
375
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000376# Test extreme cases with long ints
377a = [0,1,2,3,4]
Fred Drake004d5e62000-10-23 17:22:08 +0000378if a[ -pow(2,128L): 3 ] != [0,1,2]:
379 raise TestFailed, "list slicing with too-small long integer"
380if a[ 3: pow(2,145L) ] != [3,4]:
381 raise TestFailed, "list slicing with too-large long integer"
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000382
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000383
384# extended slicing
385
386# subscript
387a = [0,1,2,3,4]
388vereq(a[::], a)
389vereq(a[::2], [0,2,4])
390vereq(a[1::2], [1,3])
391vereq(a[::-1], [4,3,2,1,0])
392vereq(a[::-2], [4,2,0])
393vereq(a[3::-2], [3,1])
394vereq(a[-100:100:], a)
395vereq(a[100:-100:-1], a[::-1])
396vereq(a[-100L:100L:2L], [0,2,4])
Michael W. Hudson589dc932002-06-11 13:38:42 +0000397vereq(a[1000:2000:2], [])
398vereq(a[-1000:-2000:-2], [])
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000399# deletion
400del a[::2]
401vereq(a, [1,3])
402a = range(5)
403del a[1::2]
404vereq(a, [0,2,4])
405a = range(5)
406del a[1::-2]
407vereq(a, [0,2,3,4])
408# assignment
409a = range(10)
410a[::2] = [-1]*5
411vereq(a, [-1, 1, -1, 3, -1, 5, -1, 7, -1, 9])
412a = range(10)
413a[::-4] = [10]*3
414vereq(a, [0, 10, 2, 3, 4, 10, 6, 7, 8 ,10])
415a = range(4)
416a[::-1] = a
417vereq(a, [3, 2, 1, 0])
Michael W. Hudson9c14bad2002-06-19 15:44:15 +0000418a = range(10)
419b = a[:]
420c = a[:]
421a[2:3] = ["two", "elements"]
422b[slice(2,3)] = ["two", "elements"]
423c[2:3:] = ["two", "elements"]
424vereq(a, b)
425vereq(a, c)
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000426
Guido van Rossum85f18201992-11-27 22:53:50 +0000427print '6.6 Mappings == Dictionaries'
428d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000429if d.keys() != []: raise TestFailed, '{}.keys()'
Neil Schemenauereff72442002-03-24 01:24:54 +0000430if d.values() != []: raise TestFailed, '{}.values()'
431if d.items() != []: raise TestFailed, '{}.items()'
Fred Drake132dce22000-12-12 23:11:42 +0000432if d.has_key('a') != 0: raise TestFailed, '{}.has_key(\'a\')'
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +0000433if ('a' in d) != 0: raise TestFailed, "'a' in {}"
434if ('a' not in d) != 1: raise TestFailed, "'a' not in {}"
Fred Drake132dce22000-12-12 23:11:42 +0000435if len(d) != 0: raise TestFailed, 'len({})'
Guido van Rossum85f18201992-11-27 22:53:50 +0000436d = {'a': 1, 'b': 2}
Fred Drake132dce22000-12-12 23:11:42 +0000437if len(d) != 2: raise TestFailed, 'len(dict)'
Guido van Rossum85f18201992-11-27 22:53:50 +0000438k = d.keys()
439k.sort()
Fred Drake132dce22000-12-12 23:11:42 +0000440if k != ['a', 'b']: raise TestFailed, 'dict keys()'
Guido van Rossum85f18201992-11-27 22:53:50 +0000441if d.has_key('a') and d.has_key('b') and not d.has_key('c'): pass
442else: raise TestFailed, 'dict keys()'
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +0000443if 'a' in d and 'b' in d and 'c' not in d: pass
444else: raise TestFailed, 'dict keys() # in/not in version'
Fred Drake132dce22000-12-12 23:11:42 +0000445if d['a'] != 1 or d['b'] != 2: raise TestFailed, 'dict item'
Guido van Rossum85f18201992-11-27 22:53:50 +0000446d['c'] = 3
447d['a'] = 4
Fred Drake132dce22000-12-12 23:11:42 +0000448if d['c'] != 3 or d['a'] != 4: raise TestFailed, 'dict item assignment'
Guido van Rossum85f18201992-11-27 22:53:50 +0000449del d['b']
Fred Drake132dce22000-12-12 23:11:42 +0000450if d != {'a': 4, 'c': 3}: raise TestFailed, 'dict item deletion'
Barry Warsaw41775382001-06-26 20:09:28 +0000451# dict.clear()
Guido van Rossumce1fa261997-06-02 23:14:00 +0000452d = {1:1, 2:2, 3:3}
453d.clear()
454if d != {}: raise TestFailed, 'dict clear'
Barry Warsaw41775382001-06-26 20:09:28 +0000455# dict.update()
Guido van Rossumce1fa261997-06-02 23:14:00 +0000456d.update({1:100})
457d.update({2:20})
458d.update({1:1, 2:2, 3:3})
459if d != {1:1, 2:2, 3:3}: raise TestFailed, 'dict update'
Barry Warsaw41775382001-06-26 20:09:28 +0000460d.clear()
461try: d.update(None)
462except AttributeError: pass
463else: raise TestFailed, 'dict.update(None), AttributeError expected'
464class SimpleUserDict:
465 def __init__(self):
466 self.d = {1:1, 2:2, 3:3}
467 def keys(self):
468 return self.d.keys()
469 def __getitem__(self, i):
470 return self.d[i]
471d.update(SimpleUserDict())
472if d != {1:1, 2:2, 3:3}: raise TestFailed, 'dict.update(instance)'
473d.clear()
474class FailingUserDict:
475 def keys(self):
476 raise ValueError
477try: d.update(FailingUserDict())
478except ValueError: pass
479else: raise TestFailed, 'dict.keys() expected ValueError'
480class FailingUserDict:
481 def keys(self):
482 class BogonIter:
483 def __iter__(self):
484 raise ValueError
485 return BogonIter()
486try: d.update(FailingUserDict())
487except ValueError: pass
488else: raise TestFailed, 'iter(dict.keys()) expected ValueError'
489class FailingUserDict:
490 def keys(self):
491 class BogonIter:
492 def __init__(self):
493 self.i = 1
494 def __iter__(self):
495 return self
496 def next(self):
497 if self.i:
498 self.i = 0
499 return 'a'
500 raise ValueError
501 return BogonIter()
502 def __getitem__(self, key):
503 return key
504try: d.update(FailingUserDict())
505except ValueError: pass
506else: raise TestFailed, 'iter(dict.keys()).next() expected ValueError'
507class FailingUserDict:
508 def keys(self):
509 class BogonIter:
510 def __init__(self):
511 self.i = ord('a')
512 def __iter__(self):
513 return self
514 def next(self):
515 if self.i <= ord('z'):
516 rtn = chr(self.i)
517 self.i += 1
518 return rtn
519 raise StopIteration
520 return BogonIter()
521 def __getitem__(self, key):
522 raise ValueError
523try: d.update(FailingUserDict())
524except ValueError: pass
525else: raise TestFailed, 'dict.update(), __getitem__ expected ValueError'
526# dict.copy()
527d = {1:1, 2:2, 3:3}
Guido van Rossumce1fa261997-06-02 23:14:00 +0000528if d.copy() != {1:1, 2:2, 3:3}: raise TestFailed, 'dict copy'
529if {}.copy() != {}: raise TestFailed, 'empty dict copy'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000530# dict.get()
Guido van Rossumfb5cef11997-10-20 20:10:43 +0000531d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000532if d.get('c') is not None: raise TestFailed, 'missing {} get, no 2nd arg'
Guido van Rossumfb5cef11997-10-20 20:10:43 +0000533if d.get('c', 3) != 3: raise TestFailed, 'missing {} get, w/ 2nd arg'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000534d = {'a' : 1, 'b' : 2}
Fred Drake132dce22000-12-12 23:11:42 +0000535if d.get('c') is not None: raise TestFailed, 'missing dict get, no 2nd arg'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000536if d.get('c', 3) != 3: raise TestFailed, 'missing dict get, w/ 2nd arg'
537if d.get('a') != 1: raise TestFailed, 'present dict get, no 2nd arg'
538if d.get('a', 3) != 1: raise TestFailed, 'present dict get, w/ 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000539# dict.setdefault()
540d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000541if d.setdefault('key0') is not None:
Fred Drake004d5e62000-10-23 17:22:08 +0000542 raise TestFailed, 'missing {} setdefault, no 2nd arg'
Fred Drake132dce22000-12-12 23:11:42 +0000543if d.setdefault('key0') is not None:
Fred Drake004d5e62000-10-23 17:22:08 +0000544 raise TestFailed, 'present {} setdefault, no 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000545d.setdefault('key', []).append(3)
Fred Drake132dce22000-12-12 23:11:42 +0000546if d['key'][0] != 3:
Fred Drake004d5e62000-10-23 17:22:08 +0000547 raise TestFailed, 'missing {} setdefault, w/ 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000548d.setdefault('key', []).append(4)
Fred Drake132dce22000-12-12 23:11:42 +0000549if len(d['key']) != 2:
Fred Drake004d5e62000-10-23 17:22:08 +0000550 raise TestFailed, 'present {} setdefault, w/ 2nd arg'
Guido van Rossumb822c612000-12-12 22:02:59 +0000551# dict.popitem()
552for copymode in -1, +1:
553 # -1: b has same structure as a
554 # +1: b is a.copy()
555 for log2size in range(12):
556 size = 2**log2size
557 a = {}
558 b = {}
559 for i in range(size):
560 a[`i`] = i
561 if copymode < 0:
562 b[`i`] = i
563 if copymode > 0:
564 b = a.copy()
565 for i in range(size):
566 ka, va = ta = a.popitem()
567 if va != int(ka): raise TestFailed, "a.popitem: %s" % str(ta)
568 kb, vb = tb = b.popitem()
569 if vb != int(kb): raise TestFailed, "b.popitem: %s" % str(tb)
570 if copymode < 0 and ta != tb:
571 raise TestFailed, "a.popitem != b.popitem: %s, %s" % (
572 str(ta), str(tb))
573 if a: raise TestFailed, 'a not empty after popitems: %s' % str(a)
574 if b: raise TestFailed, 'b not empty after popitems: %s' % str(b)
Guido van Rossum29d26062001-12-11 04:37:34 +0000575
Neil Schemenauereff72442002-03-24 01:24:54 +0000576d.clear()
577try: d.popitem()
578except KeyError: pass
579else: raise TestFailed, "{}.popitem doesn't raise KeyError"
580
Guido van Rossume027d982002-04-12 15:11:59 +0000581# Tests for pop with specified key
582d.clear()
583k, v = 'abc', 'def'
584d[k] = v
585try: d.pop('ghi')
586except KeyError: pass
587else: raise TestFailed, "{}.pop(k) doesn't raise KeyError when k not in dictionary"
588
589if d.pop(k) != v: raise TestFailed, "{}.pop(k) doesn't find known key/value pair"
590if len(d) > 0: raise TestFailed, "{}.pop(k) failed to remove the specified pair"
591
592try: d.pop(k)
593except KeyError: pass
594else: raise TestFailed, "{}.pop(k) doesn't raise KeyError when dictionary is empty"
595
Neil Schemenauereff72442002-03-24 01:24:54 +0000596d[1] = 1
597try:
598 for i in d:
Tim Peters863ac442002-04-16 01:38:40 +0000599 d[i+1] = 1
Neil Schemenauereff72442002-03-24 01:24:54 +0000600except RuntimeError:
601 pass
602else:
603 raise TestFailed, "changing dict size during iteration doesn't raise Error"
604
Guido van Rossum29d26062001-12-11 04:37:34 +0000605try: type(1, 2)
606except TypeError: pass
607else: raise TestFailed, 'type(), w/2 args expected TypeError'
608
609try: type(1, 2, 3, 4)
610except TypeError: pass
611else: raise TestFailed, 'type(), w/4 args expected TypeError'
Neil Schemenauereff72442002-03-24 01:24:54 +0000612
613print 'Buffers'
614try: buffer('asdf', -1)
615except ValueError: pass
616else: raise TestFailed, "buffer('asdf', -1) should raise ValueError"
617
618try: buffer(None)
619except TypeError: pass
620else: raise TestFailed, "buffer(None) should raise TypeError"
621
622a = buffer('asdf')
623hash(a)
624b = a * 5
625if a == b:
626 raise TestFailed, 'buffers should not be equal'
Fred Drake485f3402002-05-02 04:27:20 +0000627if str(b) != ('asdf' * 5):
628 raise TestFailed, 'repeated buffer has wrong content'
629if str(a * 0) != '':
630 raise TestFailed, 'repeated buffer zero times has wrong content'
631if str(a + buffer('def')) != 'asdfdef':
632 raise TestFailed, 'concatenation of buffers yields wrong content'
Neil Schemenauereff72442002-03-24 01:24:54 +0000633
634try: a[1] = 'g'
635except TypeError: pass
636else: raise TestFailed, "buffer assignment should raise TypeError"
637
638try: a[0:1] = 'g'
639except TypeError: pass
640else: raise TestFailed, "buffer slice assignment should raise TypeError"