blob: 8452cecdc4855cc195ef6630c80e5d1f1e906352 [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'))
214
215
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
247
Guido van Rossum85f18201992-11-27 22:53:50 +0000248print '6.5.3 Lists'
Fred Drake132dce22000-12-12 23:11:42 +0000249if len([]) != 0: raise TestFailed, 'len([])'
250if len([1,]) != 1: raise TestFailed, 'len([1,])'
251if len([1,2,3,4,5,6]) != 6: raise TestFailed, 'len([1,2,3,4,5,6])'
252if [1,2]+[3,4] != [1,2,3,4]: raise TestFailed, 'list concatenation'
253if [1,2]*3 != [1,2,1,2,1,2]: raise TestFailed, 'list repetition *3'
254if [1,2]*3L != [1,2,1,2,1,2]: raise TestFailed, 'list repetition *3L'
255if 0*[1,2,3] != []: raise TestFailed, 'list repetition 0*'
256if 0L*[1,2,3] != []: raise TestFailed, 'list repetition 0L*'
257if min([1,2]) != 1 or max([1,2]) != 2: raise TestFailed, 'min/max list'
Guido van Rossum85f18201992-11-27 22:53:50 +0000258if 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
259else: raise TestFailed, 'in/not in list'
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000260a = [1, 2, 3, 4, 5]
261a[:-1] = a
262if a != [1, 2, 3, 4, 5, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000263 raise TestFailed, "list self-slice-assign (head)"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000264a = [1, 2, 3, 4, 5]
265a[1:] = a
266if a != [1, 1, 2, 3, 4, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000267 raise TestFailed, "list self-slice-assign (tail)"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000268a = [1, 2, 3, 4, 5]
269a[1:-1] = a
270if a != [1, 1, 2, 3, 4, 5, 5]:
Fred Drake004d5e62000-10-23 17:22:08 +0000271 raise TestFailed, "list self-slice-assign (center)"
Neil Schemenauereff72442002-03-24 01:24:54 +0000272try: [][0]
273except IndexError: pass
274else: raise TestFailed, "list index error didn't raise IndexError"
275try: [][0] = 5
276except IndexError: pass
277else: raise TestFailed, "list assignment index error didn't raise IndexError"
278try: [].pop()
279except IndexError: pass
280else: raise TestFailed, "empty list.pop() didn't raise IndexError"
281try: [1].pop(5)
282except IndexError: pass
283else: raise TestFailed, "[1].pop(5) didn't raise IndexError"
284try: [][0:1] = 5
285except TypeError: pass
286else: raise TestFailed, "bad list slice assignment didn't raise TypeError"
287try: [].extend(None)
288except TypeError: pass
289else: raise TestFailed, "list.extend(None) didn't raise TypeError"
290a = [1, 2, 3, 4]
291a *= 0
292if a != []:
293 raise TestFailed, "list inplace repeat"
Guido van Rossumaffd77f1998-07-16 15:29:06 +0000294
Guido van Rossum85f18201992-11-27 22:53:50 +0000295
296print '6.5.3a Additional list operations'
297a = [0,1,2,3,4]
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000298a[0L] = 1
299a[1L] = 2
300a[2L] = 3
Fred Drake132dce22000-12-12 23:11:42 +0000301if a != [1,2,3,3,4]: raise TestFailed, 'list item assignment [0L], [1L], [2L]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000302a[0] = 5
303a[1] = 6
304a[2] = 7
Fred Drake132dce22000-12-12 23:11:42 +0000305if a != [5,6,7,3,4]: raise TestFailed, 'list item assignment [0], [1], [2]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000306a[-2L] = 88
307a[-1L] = 99
Fred Drake132dce22000-12-12 23:11:42 +0000308if a != [5,6,7,88,99]: raise TestFailed, 'list item assignment [-2L], [-1L]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000309a[-2] = 8
310a[-1] = 9
Fred Drake132dce22000-12-12 23:11:42 +0000311if a != [5,6,7,8,9]: raise TestFailed, 'list item assignment [-2], [-1]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000312a[:2] = [0,4]
313a[-3:] = []
314a[1:1] = [1,2,3]
Fred Drake132dce22000-12-12 23:11:42 +0000315if a != [0,1,2,3,4]: raise TestFailed, 'list slice assignment'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000316a[ 1L : 4L] = [7,8,9]
Fred Drake132dce22000-12-12 23:11:42 +0000317if a != [0,7,8,9,4]: raise TestFailed, 'list slice assignment using long ints'
Guido van Rossum85f18201992-11-27 22:53:50 +0000318del a[1:4]
Fred Drake132dce22000-12-12 23:11:42 +0000319if a != [0,4]: raise TestFailed, 'list slice deletion'
Guido van Rossum85f18201992-11-27 22:53:50 +0000320del a[0]
Fred Drake132dce22000-12-12 23:11:42 +0000321if a != [4]: raise TestFailed, 'list item deletion [0]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000322del a[-1]
Fred Drake132dce22000-12-12 23:11:42 +0000323if a != []: raise TestFailed, 'list item deletion [-1]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000324a=range(0,5)
325del a[1L:4L]
Fred Drake132dce22000-12-12 23:11:42 +0000326if a != [0,4]: raise TestFailed, 'list slice deletion'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000327del a[0L]
Fred Drake132dce22000-12-12 23:11:42 +0000328if a != [4]: raise TestFailed, 'list item deletion [0]'
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000329del a[-1L]
Fred Drake132dce22000-12-12 23:11:42 +0000330if a != []: raise TestFailed, 'list item deletion [-1]'
Guido van Rossum85f18201992-11-27 22:53:50 +0000331a.append(0)
332a.append(1)
333a.append(2)
Fred Drake132dce22000-12-12 23:11:42 +0000334if a != [0,1,2]: raise TestFailed, 'list append'
Guido van Rossum85f18201992-11-27 22:53:50 +0000335a.insert(0, -2)
336a.insert(1, -1)
337a.insert(2,0)
Fred Drake132dce22000-12-12 23:11:42 +0000338if a != [-2,-1,0,0,1,2]: raise TestFailed, 'list insert'
339if a.count(0) != 2: raise TestFailed, ' list count'
340if a.index(0) != 2: raise TestFailed, 'list index'
Guido van Rossum85f18201992-11-27 22:53:50 +0000341a.remove(0)
Fred Drake132dce22000-12-12 23:11:42 +0000342if a != [-2,-1,0,1,2]: raise TestFailed, 'list remove'
Guido van Rossum85f18201992-11-27 22:53:50 +0000343a.reverse()
Fred Drake132dce22000-12-12 23:11:42 +0000344if a != [2,1,0,-1,-2]: raise TestFailed, 'list reverse'
Guido van Rossum85f18201992-11-27 22:53:50 +0000345a.sort()
Fred Drake132dce22000-12-12 23:11:42 +0000346if a != [-2,-1,0,1,2]: raise TestFailed, 'list sort'
Guido van Rossume61fa0a1993-10-22 13:56:35 +0000347def revcmp(a, b): return cmp(b, a)
348a.sort(revcmp)
Fred Drake132dce22000-12-12 23:11:42 +0000349if a != [2,1,0,-1,-2]: raise TestFailed, 'list sort with cmp func'
Guido van Rossumd151d341998-02-25 17:51:50 +0000350# The following dumps core in unpatched Python 1.5:
351def myComparison(x,y):
352 return cmp(x%3, y%7)
353z = range(12)
354z.sort(myComparison)
Guido van Rossum85f18201992-11-27 22:53:50 +0000355
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000356# Test extreme cases with long ints
357a = [0,1,2,3,4]
Fred Drake004d5e62000-10-23 17:22:08 +0000358if a[ -pow(2,128L): 3 ] != [0,1,2]:
359 raise TestFailed, "list slicing with too-small long integer"
360if a[ 3: pow(2,145L) ] != [3,4]:
361 raise TestFailed, "list slicing with too-large long integer"
Andrew M. Kuchling5ebfa2a2000-02-23 22:23:17 +0000362
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000363
364# extended slicing
365
366# subscript
367a = [0,1,2,3,4]
368vereq(a[::], a)
369vereq(a[::2], [0,2,4])
370vereq(a[1::2], [1,3])
371vereq(a[::-1], [4,3,2,1,0])
372vereq(a[::-2], [4,2,0])
373vereq(a[3::-2], [3,1])
374vereq(a[-100:100:], a)
375vereq(a[100:-100:-1], a[::-1])
376vereq(a[-100L:100L:2L], [0,2,4])
377# deletion
378del a[::2]
379vereq(a, [1,3])
380a = range(5)
381del a[1::2]
382vereq(a, [0,2,4])
383a = range(5)
384del a[1::-2]
385vereq(a, [0,2,3,4])
386# assignment
387a = range(10)
388a[::2] = [-1]*5
389vereq(a, [-1, 1, -1, 3, -1, 5, -1, 7, -1, 9])
390a = range(10)
391a[::-4] = [10]*3
392vereq(a, [0, 10, 2, 3, 4, 10, 6, 7, 8 ,10])
393a = range(4)
394a[::-1] = a
395vereq(a, [3, 2, 1, 0])
396
Guido van Rossum85f18201992-11-27 22:53:50 +0000397print '6.6 Mappings == Dictionaries'
398d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000399if d.keys() != []: raise TestFailed, '{}.keys()'
Neil Schemenauereff72442002-03-24 01:24:54 +0000400if d.values() != []: raise TestFailed, '{}.values()'
401if d.items() != []: raise TestFailed, '{}.items()'
Fred Drake132dce22000-12-12 23:11:42 +0000402if d.has_key('a') != 0: raise TestFailed, '{}.has_key(\'a\')'
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +0000403if ('a' in d) != 0: raise TestFailed, "'a' in {}"
404if ('a' not in d) != 1: raise TestFailed, "'a' not in {}"
Fred Drake132dce22000-12-12 23:11:42 +0000405if len(d) != 0: raise TestFailed, 'len({})'
Guido van Rossum85f18201992-11-27 22:53:50 +0000406d = {'a': 1, 'b': 2}
Fred Drake132dce22000-12-12 23:11:42 +0000407if len(d) != 2: raise TestFailed, 'len(dict)'
Guido van Rossum85f18201992-11-27 22:53:50 +0000408k = d.keys()
409k.sort()
Fred Drake132dce22000-12-12 23:11:42 +0000410if k != ['a', 'b']: raise TestFailed, 'dict keys()'
Guido van Rossum85f18201992-11-27 22:53:50 +0000411if d.has_key('a') and d.has_key('b') and not d.has_key('c'): pass
412else: raise TestFailed, 'dict keys()'
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +0000413if 'a' in d and 'b' in d and 'c' not in d: pass
414else: raise TestFailed, 'dict keys() # in/not in version'
Fred Drake132dce22000-12-12 23:11:42 +0000415if d['a'] != 1 or d['b'] != 2: raise TestFailed, 'dict item'
Guido van Rossum85f18201992-11-27 22:53:50 +0000416d['c'] = 3
417d['a'] = 4
Fred Drake132dce22000-12-12 23:11:42 +0000418if d['c'] != 3 or d['a'] != 4: raise TestFailed, 'dict item assignment'
Guido van Rossum85f18201992-11-27 22:53:50 +0000419del d['b']
Fred Drake132dce22000-12-12 23:11:42 +0000420if d != {'a': 4, 'c': 3}: raise TestFailed, 'dict item deletion'
Barry Warsaw41775382001-06-26 20:09:28 +0000421# dict.clear()
Guido van Rossumce1fa261997-06-02 23:14:00 +0000422d = {1:1, 2:2, 3:3}
423d.clear()
424if d != {}: raise TestFailed, 'dict clear'
Barry Warsaw41775382001-06-26 20:09:28 +0000425# dict.update()
Guido van Rossumce1fa261997-06-02 23:14:00 +0000426d.update({1:100})
427d.update({2:20})
428d.update({1:1, 2:2, 3:3})
429if d != {1:1, 2:2, 3:3}: raise TestFailed, 'dict update'
Barry Warsaw41775382001-06-26 20:09:28 +0000430d.clear()
431try: d.update(None)
432except AttributeError: pass
433else: raise TestFailed, 'dict.update(None), AttributeError expected'
434class SimpleUserDict:
435 def __init__(self):
436 self.d = {1:1, 2:2, 3:3}
437 def keys(self):
438 return self.d.keys()
439 def __getitem__(self, i):
440 return self.d[i]
441d.update(SimpleUserDict())
442if d != {1:1, 2:2, 3:3}: raise TestFailed, 'dict.update(instance)'
443d.clear()
444class FailingUserDict:
445 def keys(self):
446 raise ValueError
447try: d.update(FailingUserDict())
448except ValueError: pass
449else: raise TestFailed, 'dict.keys() expected ValueError'
450class FailingUserDict:
451 def keys(self):
452 class BogonIter:
453 def __iter__(self):
454 raise ValueError
455 return BogonIter()
456try: d.update(FailingUserDict())
457except ValueError: pass
458else: raise TestFailed, 'iter(dict.keys()) expected ValueError'
459class FailingUserDict:
460 def keys(self):
461 class BogonIter:
462 def __init__(self):
463 self.i = 1
464 def __iter__(self):
465 return self
466 def next(self):
467 if self.i:
468 self.i = 0
469 return 'a'
470 raise ValueError
471 return BogonIter()
472 def __getitem__(self, key):
473 return key
474try: d.update(FailingUserDict())
475except ValueError: pass
476else: raise TestFailed, 'iter(dict.keys()).next() expected ValueError'
477class FailingUserDict:
478 def keys(self):
479 class BogonIter:
480 def __init__(self):
481 self.i = ord('a')
482 def __iter__(self):
483 return self
484 def next(self):
485 if self.i <= ord('z'):
486 rtn = chr(self.i)
487 self.i += 1
488 return rtn
489 raise StopIteration
490 return BogonIter()
491 def __getitem__(self, key):
492 raise ValueError
493try: d.update(FailingUserDict())
494except ValueError: pass
495else: raise TestFailed, 'dict.update(), __getitem__ expected ValueError'
496# dict.copy()
497d = {1:1, 2:2, 3:3}
Guido van Rossumce1fa261997-06-02 23:14:00 +0000498if d.copy() != {1:1, 2:2, 3:3}: raise TestFailed, 'dict copy'
499if {}.copy() != {}: raise TestFailed, 'empty dict copy'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000500# dict.get()
Guido van Rossumfb5cef11997-10-20 20:10:43 +0000501d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000502if d.get('c') is not None: raise TestFailed, 'missing {} get, no 2nd arg'
Guido van Rossumfb5cef11997-10-20 20:10:43 +0000503if d.get('c', 3) != 3: raise TestFailed, 'missing {} get, w/ 2nd arg'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000504d = {'a' : 1, 'b' : 2}
Fred Drake132dce22000-12-12 23:11:42 +0000505if d.get('c') is not None: raise TestFailed, 'missing dict get, no 2nd arg'
Barry Warsaw9b887c71997-10-20 17:34:43 +0000506if d.get('c', 3) != 3: raise TestFailed, 'missing dict get, w/ 2nd arg'
507if d.get('a') != 1: raise TestFailed, 'present dict get, no 2nd arg'
508if d.get('a', 3) != 1: raise TestFailed, 'present dict get, w/ 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000509# dict.setdefault()
510d = {}
Fred Drake132dce22000-12-12 23:11:42 +0000511if d.setdefault('key0') is not None:
Fred Drake004d5e62000-10-23 17:22:08 +0000512 raise TestFailed, 'missing {} setdefault, no 2nd arg'
Fred Drake132dce22000-12-12 23:11:42 +0000513if d.setdefault('key0') is not None:
Fred Drake004d5e62000-10-23 17:22:08 +0000514 raise TestFailed, 'present {} setdefault, no 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000515d.setdefault('key', []).append(3)
Fred Drake132dce22000-12-12 23:11:42 +0000516if d['key'][0] != 3:
Fred Drake004d5e62000-10-23 17:22:08 +0000517 raise TestFailed, 'missing {} setdefault, w/ 2nd arg'
Guido van Rossum79c9b172000-08-08 16:13:23 +0000518d.setdefault('key', []).append(4)
Fred Drake132dce22000-12-12 23:11:42 +0000519if len(d['key']) != 2:
Fred Drake004d5e62000-10-23 17:22:08 +0000520 raise TestFailed, 'present {} setdefault, w/ 2nd arg'
Guido van Rossumb822c612000-12-12 22:02:59 +0000521# dict.popitem()
522for copymode in -1, +1:
523 # -1: b has same structure as a
524 # +1: b is a.copy()
525 for log2size in range(12):
526 size = 2**log2size
527 a = {}
528 b = {}
529 for i in range(size):
530 a[`i`] = i
531 if copymode < 0:
532 b[`i`] = i
533 if copymode > 0:
534 b = a.copy()
535 for i in range(size):
536 ka, va = ta = a.popitem()
537 if va != int(ka): raise TestFailed, "a.popitem: %s" % str(ta)
538 kb, vb = tb = b.popitem()
539 if vb != int(kb): raise TestFailed, "b.popitem: %s" % str(tb)
540 if copymode < 0 and ta != tb:
541 raise TestFailed, "a.popitem != b.popitem: %s, %s" % (
542 str(ta), str(tb))
543 if a: raise TestFailed, 'a not empty after popitems: %s' % str(a)
544 if b: raise TestFailed, 'b not empty after popitems: %s' % str(b)
Guido van Rossum29d26062001-12-11 04:37:34 +0000545
Neil Schemenauereff72442002-03-24 01:24:54 +0000546d.clear()
547try: d.popitem()
548except KeyError: pass
549else: raise TestFailed, "{}.popitem doesn't raise KeyError"
550
Guido van Rossume027d982002-04-12 15:11:59 +0000551# Tests for pop with specified key
552d.clear()
553k, v = 'abc', 'def'
554d[k] = v
555try: d.pop('ghi')
556except KeyError: pass
557else: raise TestFailed, "{}.pop(k) doesn't raise KeyError when k not in dictionary"
558
559if d.pop(k) != v: raise TestFailed, "{}.pop(k) doesn't find known key/value pair"
560if len(d) > 0: raise TestFailed, "{}.pop(k) failed to remove the specified pair"
561
562try: d.pop(k)
563except KeyError: pass
564else: raise TestFailed, "{}.pop(k) doesn't raise KeyError when dictionary is empty"
565
Neil Schemenauereff72442002-03-24 01:24:54 +0000566d[1] = 1
567try:
568 for i in d:
Tim Peters863ac442002-04-16 01:38:40 +0000569 d[i+1] = 1
Neil Schemenauereff72442002-03-24 01:24:54 +0000570except RuntimeError:
571 pass
572else:
573 raise TestFailed, "changing dict size during iteration doesn't raise Error"
574
Guido van Rossum29d26062001-12-11 04:37:34 +0000575try: type(1, 2)
576except TypeError: pass
577else: raise TestFailed, 'type(), w/2 args expected TypeError'
578
579try: type(1, 2, 3, 4)
580except TypeError: pass
581else: raise TestFailed, 'type(), w/4 args expected TypeError'
Neil Schemenauereff72442002-03-24 01:24:54 +0000582
583print 'Buffers'
584try: buffer('asdf', -1)
585except ValueError: pass
586else: raise TestFailed, "buffer('asdf', -1) should raise ValueError"
587
588try: buffer(None)
589except TypeError: pass
590else: raise TestFailed, "buffer(None) should raise TypeError"
591
592a = buffer('asdf')
593hash(a)
594b = a * 5
595if a == b:
596 raise TestFailed, 'buffers should not be equal'
Fred Drake485f3402002-05-02 04:27:20 +0000597if str(b) != ('asdf' * 5):
598 raise TestFailed, 'repeated buffer has wrong content'
599if str(a * 0) != '':
600 raise TestFailed, 'repeated buffer zero times has wrong content'
601if str(a + buffer('def')) != 'asdfdef':
602 raise TestFailed, 'concatenation of buffers yields wrong content'
Neil Schemenauereff72442002-03-24 01:24:54 +0000603
604try: a[1] = 'g'
605except TypeError: pass
606else: raise TestFailed, "buffer assignment should raise TypeError"
607
608try: a[0:1] = 'g'
609except TypeError: pass
610else: raise TestFailed, "buffer slice assignment should raise TypeError"