blob: b2e7e665cc3399423fe3bcd3c0ea5b1ae2123c41 [file] [log] [blame]
Guido van Rossum4bb1e362001-09-28 23:49:48 +00001# Test enhancements related to descriptors and new-style classes
Tim Peters6d6c1a32001-08-02 04:15:00 +00002
Neal Norwitz1a997502003-01-13 20:13:12 +00003from test.test_support import verify, vereq, verbose, TestFailed, TESTFN, get_original_stdout
Tim Peters6d6c1a32001-08-02 04:15:00 +00004from copy import deepcopy
Tim Peters4d9b4662002-04-16 01:59:17 +00005import warnings
6
7warnings.filterwarnings("ignore",
8 r'complex divmod\(\), // and % are deprecated$',
Guido van Rossum155a34d2002-06-03 19:45:32 +00009 DeprecationWarning, r'(<string>|%s)$' % __name__)
Tim Peters6d6c1a32001-08-02 04:15:00 +000010
Guido van Rossum875eeaa2001-10-11 18:33:53 +000011def veris(a, b):
12 if a is not b:
13 raise TestFailed, "%r is %r" % (a, b)
14
Tim Peters6d6c1a32001-08-02 04:15:00 +000015def testunop(a, res, expr="len(a)", meth="__len__"):
16 if verbose: print "checking", expr
17 dict = {'a': a}
Guido van Rossum45704552001-10-08 16:35:45 +000018 vereq(eval(expr, dict), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000019 t = type(a)
20 m = getattr(t, meth)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000021 while meth not in t.__dict__:
22 t = t.__bases__[0]
Guido van Rossum45704552001-10-08 16:35:45 +000023 vereq(m, t.__dict__[meth])
24 vereq(m(a), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000025 bm = getattr(a, meth)
Guido van Rossum45704552001-10-08 16:35:45 +000026 vereq(bm(), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000027
28def testbinop(a, b, res, expr="a+b", meth="__add__"):
29 if verbose: print "checking", expr
30 dict = {'a': a, 'b': b}
Tim Peters3caca232001-12-06 06:23:26 +000031
Guido van Rossum45704552001-10-08 16:35:45 +000032 vereq(eval(expr, dict), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000033 t = type(a)
34 m = getattr(t, meth)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000035 while meth not in t.__dict__:
36 t = t.__bases__[0]
Guido van Rossum45704552001-10-08 16:35:45 +000037 vereq(m, t.__dict__[meth])
38 vereq(m(a, b), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000039 bm = getattr(a, meth)
Guido van Rossum45704552001-10-08 16:35:45 +000040 vereq(bm(b), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000041
42def testternop(a, b, c, res, expr="a[b:c]", meth="__getslice__"):
43 if verbose: print "checking", expr
44 dict = {'a': a, 'b': b, 'c': c}
Guido van Rossum45704552001-10-08 16:35:45 +000045 vereq(eval(expr, dict), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000046 t = type(a)
47 m = getattr(t, meth)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000048 while meth not in t.__dict__:
49 t = t.__bases__[0]
Guido van Rossum45704552001-10-08 16:35:45 +000050 vereq(m, t.__dict__[meth])
51 vereq(m(a, b, c), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000052 bm = getattr(a, meth)
Guido van Rossum45704552001-10-08 16:35:45 +000053 vereq(bm(b, c), res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000054
55def testsetop(a, b, res, stmt="a+=b", meth="__iadd__"):
56 if verbose: print "checking", stmt
57 dict = {'a': deepcopy(a), 'b': b}
Georg Brandl7cae87c2006-09-06 06:51:57 +000058 exec(stmt, dict)
Guido van Rossum45704552001-10-08 16:35:45 +000059 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000060 t = type(a)
61 m = getattr(t, meth)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000062 while meth not in t.__dict__:
63 t = t.__bases__[0]
Guido van Rossum45704552001-10-08 16:35:45 +000064 vereq(m, t.__dict__[meth])
Tim Peters6d6c1a32001-08-02 04:15:00 +000065 dict['a'] = deepcopy(a)
66 m(dict['a'], b)
Guido van Rossum45704552001-10-08 16:35:45 +000067 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000068 dict['a'] = deepcopy(a)
69 bm = getattr(dict['a'], meth)
70 bm(b)
Guido van Rossum45704552001-10-08 16:35:45 +000071 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000072
73def testset2op(a, b, c, res, stmt="a[b]=c", meth="__setitem__"):
74 if verbose: print "checking", stmt
75 dict = {'a': deepcopy(a), 'b': b, 'c': c}
Georg Brandl7cae87c2006-09-06 06:51:57 +000076 exec(stmt, dict)
Guido van Rossum45704552001-10-08 16:35:45 +000077 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000078 t = type(a)
79 m = getattr(t, meth)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000080 while meth not in t.__dict__:
81 t = t.__bases__[0]
Guido van Rossum45704552001-10-08 16:35:45 +000082 vereq(m, t.__dict__[meth])
Tim Peters6d6c1a32001-08-02 04:15:00 +000083 dict['a'] = deepcopy(a)
84 m(dict['a'], b, c)
Guido van Rossum45704552001-10-08 16:35:45 +000085 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000086 dict['a'] = deepcopy(a)
87 bm = getattr(dict['a'], meth)
88 bm(b, c)
Guido van Rossum45704552001-10-08 16:35:45 +000089 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000090
91def testset3op(a, b, c, d, res, stmt="a[b:c]=d", meth="__setslice__"):
92 if verbose: print "checking", stmt
93 dict = {'a': deepcopy(a), 'b': b, 'c': c, 'd': d}
Georg Brandl7cae87c2006-09-06 06:51:57 +000094 exec(stmt, dict)
Guido van Rossum45704552001-10-08 16:35:45 +000095 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +000096 t = type(a)
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +000097 while meth not in t.__dict__:
98 t = t.__bases__[0]
Tim Peters6d6c1a32001-08-02 04:15:00 +000099 m = getattr(t, meth)
Guido van Rossum45704552001-10-08 16:35:45 +0000100 vereq(m, t.__dict__[meth])
Tim Peters6d6c1a32001-08-02 04:15:00 +0000101 dict['a'] = deepcopy(a)
102 m(dict['a'], b, c, d)
Guido van Rossum45704552001-10-08 16:35:45 +0000103 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000104 dict['a'] = deepcopy(a)
105 bm = getattr(dict['a'], meth)
106 bm(b, c, d)
Guido van Rossum45704552001-10-08 16:35:45 +0000107 vereq(dict['a'], res)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000108
Tim Peters2f93e282001-10-04 05:27:00 +0000109def class_docstrings():
110 class Classic:
111 "A classic docstring."
Guido van Rossum45704552001-10-08 16:35:45 +0000112 vereq(Classic.__doc__, "A classic docstring.")
113 vereq(Classic.__dict__['__doc__'], "A classic docstring.")
Tim Peters2f93e282001-10-04 05:27:00 +0000114
115 class Classic2:
116 pass
117 verify(Classic2.__doc__ is None)
118
Tim Peters4fb1fe82001-10-04 05:48:13 +0000119 class NewStatic(object):
Tim Peters2f93e282001-10-04 05:27:00 +0000120 "Another docstring."
Guido van Rossum45704552001-10-08 16:35:45 +0000121 vereq(NewStatic.__doc__, "Another docstring.")
122 vereq(NewStatic.__dict__['__doc__'], "Another docstring.")
Tim Peters2f93e282001-10-04 05:27:00 +0000123
Tim Peters4fb1fe82001-10-04 05:48:13 +0000124 class NewStatic2(object):
Tim Peters2f93e282001-10-04 05:27:00 +0000125 pass
126 verify(NewStatic2.__doc__ is None)
127
Tim Peters4fb1fe82001-10-04 05:48:13 +0000128 class NewDynamic(object):
Tim Peters2f93e282001-10-04 05:27:00 +0000129 "Another docstring."
Guido van Rossum45704552001-10-08 16:35:45 +0000130 vereq(NewDynamic.__doc__, "Another docstring.")
131 vereq(NewDynamic.__dict__['__doc__'], "Another docstring.")
Tim Peters2f93e282001-10-04 05:27:00 +0000132
Tim Peters4fb1fe82001-10-04 05:48:13 +0000133 class NewDynamic2(object):
Tim Peters2f93e282001-10-04 05:27:00 +0000134 pass
135 verify(NewDynamic2.__doc__ is None)
136
Tim Peters6d6c1a32001-08-02 04:15:00 +0000137def lists():
138 if verbose: print "Testing list operations..."
139 testbinop([1], [2], [1,2], "a+b", "__add__")
140 testbinop([1,2,3], 2, 1, "b in a", "__contains__")
141 testbinop([1,2,3], 4, 0, "b in a", "__contains__")
142 testbinop([1,2,3], 1, 2, "a[b]", "__getitem__")
143 testternop([1,2,3], 0, 2, [1,2], "a[b:c]", "__getslice__")
144 testsetop([1], [2], [1,2], "a+=b", "__iadd__")
145 testsetop([1,2], 3, [1,2,1,2,1,2], "a*=b", "__imul__")
146 testunop([1,2,3], 3, "len(a)", "__len__")
147 testbinop([1,2], 3, [1,2,1,2,1,2], "a*b", "__mul__")
148 testbinop([1,2], 3, [1,2,1,2,1,2], "b*a", "__rmul__")
149 testset2op([1,2], 1, 3, [1,3], "a[b]=c", "__setitem__")
150 testset3op([1,2,3,4], 1, 3, [5,6], [1,5,6,4], "a[b:c]=d", "__setslice__")
151
152def dicts():
153 if verbose: print "Testing dict operations..."
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000154 ##testbinop({1:2}, {2:1}, -1, "cmp(a,b)", "__cmp__")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000155 testbinop({1:2,3:4}, 1, 1, "b in a", "__contains__")
156 testbinop({1:2,3:4}, 2, 0, "b in a", "__contains__")
157 testbinop({1:2,3:4}, 1, 2, "a[b]", "__getitem__")
158 d = {1:2,3:4}
159 l1 = []
160 for i in d.keys(): l1.append(i)
161 l = []
162 for i in iter(d): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000163 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000164 l = []
165 for i in d.__iter__(): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000166 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000167 l = []
Tim Petersa427a2b2001-10-29 22:25:45 +0000168 for i in dict.__iter__(d): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000169 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000170 d = {1:2, 3:4}
171 testunop(d, 2, "len(a)", "__len__")
Guido van Rossum45704552001-10-08 16:35:45 +0000172 vereq(eval(repr(d), {}), d)
173 vereq(eval(d.__repr__(), {}), d)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000174 testset2op({1:2,3:4}, 2, 3, {1:2,2:3,3:4}, "a[b]=c", "__setitem__")
175
Tim Peters25786c02001-09-02 08:22:48 +0000176def dict_constructor():
177 if verbose:
Tim Petersa427a2b2001-10-29 22:25:45 +0000178 print "Testing dict constructor ..."
179 d = dict()
Guido van Rossum45704552001-10-08 16:35:45 +0000180 vereq(d, {})
Tim Petersa427a2b2001-10-29 22:25:45 +0000181 d = dict({})
Guido van Rossum45704552001-10-08 16:35:45 +0000182 vereq(d, {})
Tim Petersa427a2b2001-10-29 22:25:45 +0000183 d = dict({1: 2, 'a': 'b'})
Guido van Rossum45704552001-10-08 16:35:45 +0000184 vereq(d, {1: 2, 'a': 'b'})
Tim Petersa427a2b2001-10-29 22:25:45 +0000185 vereq(d, dict(d.items()))
Just van Rossuma797d812002-11-23 09:45:04 +0000186 vereq(d, dict(d.iteritems()))
187 d = dict({'one':1, 'two':2})
188 vereq(d, dict(one=1, two=2))
189 vereq(d, dict(**d))
190 vereq(d, dict({"one": 1}, two=2))
191 vereq(d, dict([("two", 2)], one=1))
192 vereq(d, dict([("one", 100), ("two", 200)], **d))
193 verify(d is not dict(**d))
Tim Peters25786c02001-09-02 08:22:48 +0000194 for badarg in 0, 0L, 0j, "0", [0], (0,):
195 try:
Tim Petersa427a2b2001-10-29 22:25:45 +0000196 dict(badarg)
Tim Peters25786c02001-09-02 08:22:48 +0000197 except TypeError:
198 pass
Tim Peters1fc240e2001-10-26 05:06:50 +0000199 except ValueError:
200 if badarg == "0":
201 # It's a sequence, and its elements are also sequences (gotta
202 # love strings <wink>), but they aren't of length 2, so this
203 # one seemed better as a ValueError than a TypeError.
204 pass
205 else:
Tim Petersa427a2b2001-10-29 22:25:45 +0000206 raise TestFailed("no TypeError from dict(%r)" % badarg)
Tim Peters25786c02001-09-02 08:22:48 +0000207 else:
Tim Petersa427a2b2001-10-29 22:25:45 +0000208 raise TestFailed("no TypeError from dict(%r)" % badarg)
Tim Peters25786c02001-09-02 08:22:48 +0000209
210 try:
Tim Petersa427a2b2001-10-29 22:25:45 +0000211 dict({}, {})
Tim Peters25786c02001-09-02 08:22:48 +0000212 except TypeError:
213 pass
214 else:
Tim Petersa427a2b2001-10-29 22:25:45 +0000215 raise TestFailed("no TypeError from dict({}, {})")
Tim Peters25786c02001-09-02 08:22:48 +0000216
217 class Mapping:
Tim Peters1fc240e2001-10-26 05:06:50 +0000218 # Lacks a .keys() method; will be added later.
Tim Peters25786c02001-09-02 08:22:48 +0000219 dict = {1:2, 3:4, 'a':1j}
220
Tim Peters25786c02001-09-02 08:22:48 +0000221 try:
Tim Petersa427a2b2001-10-29 22:25:45 +0000222 dict(Mapping())
Tim Peters25786c02001-09-02 08:22:48 +0000223 except TypeError:
224 pass
225 else:
Tim Petersa427a2b2001-10-29 22:25:45 +0000226 raise TestFailed("no TypeError from dict(incomplete mapping)")
Tim Peters25786c02001-09-02 08:22:48 +0000227
228 Mapping.keys = lambda self: self.dict.keys()
Tim Peters1fc240e2001-10-26 05:06:50 +0000229 Mapping.__getitem__ = lambda self, i: self.dict[i]
Just van Rossuma797d812002-11-23 09:45:04 +0000230 d = dict(Mapping())
Guido van Rossum45704552001-10-08 16:35:45 +0000231 vereq(d, Mapping.dict)
Tim Peters25786c02001-09-02 08:22:48 +0000232
Tim Peters1fc240e2001-10-26 05:06:50 +0000233 # Init from sequence of iterable objects, each producing a 2-sequence.
234 class AddressBookEntry:
235 def __init__(self, first, last):
236 self.first = first
237 self.last = last
238 def __iter__(self):
239 return iter([self.first, self.last])
240
Tim Petersa427a2b2001-10-29 22:25:45 +0000241 d = dict([AddressBookEntry('Tim', 'Warsaw'),
Tim Petersfe677e22001-10-30 05:41:07 +0000242 AddressBookEntry('Barry', 'Peters'),
243 AddressBookEntry('Tim', 'Peters'),
244 AddressBookEntry('Barry', 'Warsaw')])
Tim Peters1fc240e2001-10-26 05:06:50 +0000245 vereq(d, {'Barry': 'Warsaw', 'Tim': 'Peters'})
246
Tim Petersa427a2b2001-10-29 22:25:45 +0000247 d = dict(zip(range(4), range(1, 5)))
248 vereq(d, dict([(i, i+1) for i in range(4)]))
Tim Peters1fc240e2001-10-26 05:06:50 +0000249
250 # Bad sequence lengths.
Tim Peters9fda73c2001-10-26 20:57:38 +0000251 for bad in [('tooshort',)], [('too', 'long', 'by 1')]:
Tim Peters1fc240e2001-10-26 05:06:50 +0000252 try:
Tim Petersa427a2b2001-10-29 22:25:45 +0000253 dict(bad)
Tim Peters1fc240e2001-10-26 05:06:50 +0000254 except ValueError:
255 pass
256 else:
Tim Petersa427a2b2001-10-29 22:25:45 +0000257 raise TestFailed("no ValueError from dict(%r)" % bad)
Tim Peters1fc240e2001-10-26 05:06:50 +0000258
Tim Peters5d2b77c2001-09-03 05:47:38 +0000259def test_dir():
260 if verbose:
261 print "Testing dir() ..."
262 junk = 12
Guido van Rossum45704552001-10-08 16:35:45 +0000263 vereq(dir(), ['junk'])
Tim Peters5d2b77c2001-09-03 05:47:38 +0000264 del junk
265
266 # Just make sure these don't blow up!
267 for arg in 2, 2L, 2j, 2e0, [2], "2", u"2", (2,), {2:2}, type, test_dir:
268 dir(arg)
269
Thomas Wouters0725cf22006-04-15 09:04:57 +0000270 # Test dir on custom classes. Since these have object as a
271 # base class, a lot of stuff gets sucked in.
Tim Peters37a309d2001-09-04 01:20:04 +0000272 def interesting(strings):
273 return [s for s in strings if not s.startswith('_')]
274
Tim Peters5d2b77c2001-09-03 05:47:38 +0000275 class C(object):
276 Cdata = 1
277 def Cmethod(self): pass
Tim Peters37a309d2001-09-04 01:20:04 +0000278
279 cstuff = ['Cdata', 'Cmethod']
Guido van Rossum45704552001-10-08 16:35:45 +0000280 vereq(interesting(dir(C)), cstuff)
Tim Peters37a309d2001-09-04 01:20:04 +0000281
282 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000283 vereq(interesting(dir(c)), cstuff)
Tim Peters305b5852001-09-17 02:38:46 +0000284 verify('im_self' in dir(C.Cmethod))
Tim Peters37a309d2001-09-04 01:20:04 +0000285
286 c.cdata = 2
287 c.cmethod = lambda self: 0
Guido van Rossum45704552001-10-08 16:35:45 +0000288 vereq(interesting(dir(c)), cstuff + ['cdata', 'cmethod'])
Tim Peters305b5852001-09-17 02:38:46 +0000289 verify('im_self' in dir(c.Cmethod))
Tim Peters37a309d2001-09-04 01:20:04 +0000290
Tim Peters5d2b77c2001-09-03 05:47:38 +0000291 class A(C):
292 Adata = 1
293 def Amethod(self): pass
Tim Peters37a309d2001-09-04 01:20:04 +0000294
295 astuff = ['Adata', 'Amethod'] + cstuff
Guido van Rossum45704552001-10-08 16:35:45 +0000296 vereq(interesting(dir(A)), astuff)
Tim Peters305b5852001-09-17 02:38:46 +0000297 verify('im_self' in dir(A.Amethod))
Tim Peters37a309d2001-09-04 01:20:04 +0000298 a = A()
Guido van Rossum45704552001-10-08 16:35:45 +0000299 vereq(interesting(dir(a)), astuff)
Tim Peters37a309d2001-09-04 01:20:04 +0000300 a.adata = 42
301 a.amethod = lambda self: 3
Guido van Rossum45704552001-10-08 16:35:45 +0000302 vereq(interesting(dir(a)), astuff + ['adata', 'amethod'])
Tim Peters305b5852001-09-17 02:38:46 +0000303 verify('im_self' in dir(a.Amethod))
Tim Peters5d2b77c2001-09-03 05:47:38 +0000304
Tim Peterscaaff8d2001-09-10 23:12:14 +0000305 # Try a module subclass.
306 import sys
307 class M(type(sys)):
308 pass
Guido van Rossum1bdd9b02002-06-04 06:10:37 +0000309 minstance = M("m")
Tim Peterscaaff8d2001-09-10 23:12:14 +0000310 minstance.b = 2
311 minstance.a = 1
Guido van Rossum1bdd9b02002-06-04 06:10:37 +0000312 names = [x for x in dir(minstance) if x not in ["__name__", "__doc__"]]
313 vereq(names, ['a', 'b'])
Tim Peterscaaff8d2001-09-10 23:12:14 +0000314
315 class M2(M):
316 def getdict(self):
317 return "Not a dict!"
318 __dict__ = property(getdict)
319
Guido van Rossum1bdd9b02002-06-04 06:10:37 +0000320 m2instance = M2("m2")
Tim Peterscaaff8d2001-09-10 23:12:14 +0000321 m2instance.b = 2
322 m2instance.a = 1
Guido van Rossum45704552001-10-08 16:35:45 +0000323 vereq(m2instance.__dict__, "Not a dict!")
Tim Peterscaaff8d2001-09-10 23:12:14 +0000324 try:
325 dir(m2instance)
326 except TypeError:
327 pass
328
Tim Peters9e6a3992001-10-30 05:45:26 +0000329 # Two essentially featureless objects, just inheriting stuff from
330 # object.
331 vereq(dir(None), dir(Ellipsis))
332
Guido van Rossum44022412002-05-13 18:29:46 +0000333 # Nasty test case for proxied objects
334 class Wrapper(object):
335 def __init__(self, obj):
336 self.__obj = obj
337 def __repr__(self):
338 return "Wrapper(%s)" % repr(self.__obj)
339 def __getitem__(self, key):
340 return Wrapper(self.__obj[key])
341 def __len__(self):
342 return len(self.__obj)
343 def __getattr__(self, name):
344 return Wrapper(getattr(self.__obj, name))
345
346 class C(object):
347 def __getclass(self):
348 return Wrapper(type(self))
349 __class__ = property(__getclass)
350
351 dir(C()) # This used to segfault
352
Tim Peters6d6c1a32001-08-02 04:15:00 +0000353binops = {
354 'add': '+',
355 'sub': '-',
356 'mul': '*',
357 'div': '/',
358 'mod': '%',
359 'divmod': 'divmod',
360 'pow': '**',
361 'lshift': '<<',
362 'rshift': '>>',
363 'and': '&',
364 'xor': '^',
365 'or': '|',
366 'cmp': 'cmp',
367 'lt': '<',
368 'le': '<=',
369 'eq': '==',
370 'ne': '!=',
371 'gt': '>',
372 'ge': '>=',
373 }
374
375for name, expr in binops.items():
376 if expr.islower():
377 expr = expr + "(a, b)"
378 else:
379 expr = 'a %s b' % expr
380 binops[name] = expr
381
382unops = {
383 'pos': '+',
384 'neg': '-',
385 'abs': 'abs',
386 'invert': '~',
387 'int': 'int',
388 'long': 'long',
389 'float': 'float',
390 'oct': 'oct',
391 'hex': 'hex',
392 }
393
394for name, expr in unops.items():
395 if expr.islower():
396 expr = expr + "(a)"
397 else:
398 expr = '%s a' % expr
399 unops[name] = expr
400
401def numops(a, b, skip=[]):
402 dict = {'a': a, 'b': b}
403 for name, expr in binops.items():
404 if name not in skip:
405 name = "__%s__" % name
406 if hasattr(a, name):
407 res = eval(expr, dict)
408 testbinop(a, b, res, expr, name)
409 for name, expr in unops.items():
Guido van Rossum0eb2a6e2001-10-09 11:07:24 +0000410 if name not in skip:
411 name = "__%s__" % name
412 if hasattr(a, name):
413 res = eval(expr, dict)
414 testunop(a, res, expr, name)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000415
416def ints():
417 if verbose: print "Testing int operations..."
418 numops(100, 3)
Guido van Rossum15d529a2002-03-11 00:07:13 +0000419 # The following crashes in Python 2.2
420 vereq((1).__nonzero__(), 1)
421 vereq((0).__nonzero__(), 0)
Guido van Rossumc9e9e402002-03-11 13:21:25 +0000422 # This returns 'NotImplemented' in Python 2.2
423 class C(int):
424 def __add__(self, other):
425 return NotImplemented
Neal Norwitzde8b94c2003-02-10 02:12:43 +0000426 vereq(C(5L), 5)
Guido van Rossumc9e9e402002-03-11 13:21:25 +0000427 try:
428 C() + ""
429 except TypeError:
430 pass
431 else:
Neal Norwitz1af5e352002-03-11 14:44:12 +0000432 raise TestFailed, "NotImplemented should have caused TypeError"
Neal Norwitzde8b94c2003-02-10 02:12:43 +0000433 import sys
434 try:
435 C(sys.maxint+1)
436 except OverflowError:
437 pass
438 else:
439 raise TestFailed, "should have raised OverflowError"
Tim Peters6d6c1a32001-08-02 04:15:00 +0000440
441def longs():
442 if verbose: print "Testing long operations..."
443 numops(100L, 3L)
444
445def floats():
446 if verbose: print "Testing float operations..."
447 numops(100.0, 3.0)
448
449def complexes():
450 if verbose: print "Testing complex operations..."
Guido van Rossum0eb2a6e2001-10-09 11:07:24 +0000451 numops(100.0j, 3.0j, skip=['lt', 'le', 'gt', 'ge', 'int', 'long', 'float'])
Tim Peters6d6c1a32001-08-02 04:15:00 +0000452 class Number(complex):
453 __slots__ = ['prec']
Tim Peters3f996e72001-09-13 19:18:27 +0000454 def __new__(cls, *args, **kwds):
455 result = complex.__new__(cls, *args)
456 result.prec = kwds.get('prec', 12)
457 return result
Tim Peters6d6c1a32001-08-02 04:15:00 +0000458 def __repr__(self):
459 prec = self.prec
460 if self.imag == 0.0:
461 return "%.*g" % (prec, self.real)
462 if self.real == 0.0:
463 return "%.*gj" % (prec, self.imag)
464 return "(%.*g+%.*gj)" % (prec, self.real, prec, self.imag)
465 __str__ = __repr__
Tim Peters3f996e72001-09-13 19:18:27 +0000466
Tim Peters6d6c1a32001-08-02 04:15:00 +0000467 a = Number(3.14, prec=6)
Walter Dörwald70a6b492004-02-12 17:35:32 +0000468 vereq(repr(a), "3.14")
Guido van Rossum45704552001-10-08 16:35:45 +0000469 vereq(a.prec, 6)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000470
Tim Peters3f996e72001-09-13 19:18:27 +0000471 a = Number(a, prec=2)
Walter Dörwald70a6b492004-02-12 17:35:32 +0000472 vereq(repr(a), "3.1")
Guido van Rossum45704552001-10-08 16:35:45 +0000473 vereq(a.prec, 2)
Tim Peters3f996e72001-09-13 19:18:27 +0000474
475 a = Number(234.5)
Walter Dörwald70a6b492004-02-12 17:35:32 +0000476 vereq(repr(a), "234.5")
Guido van Rossum45704552001-10-08 16:35:45 +0000477 vereq(a.prec, 12)
Tim Peters3f996e72001-09-13 19:18:27 +0000478
Tim Peters6d6c1a32001-08-02 04:15:00 +0000479def spamlists():
480 if verbose: print "Testing spamlist operations..."
481 import copy, xxsubtype as spam
482 def spamlist(l, memo=None):
483 import xxsubtype as spam
484 return spam.spamlist(l)
485 # This is an ugly hack:
486 copy._deepcopy_dispatch[spam.spamlist] = spamlist
487
488 testbinop(spamlist([1]), spamlist([2]), spamlist([1,2]), "a+b", "__add__")
489 testbinop(spamlist([1,2,3]), 2, 1, "b in a", "__contains__")
490 testbinop(spamlist([1,2,3]), 4, 0, "b in a", "__contains__")
491 testbinop(spamlist([1,2,3]), 1, 2, "a[b]", "__getitem__")
492 testternop(spamlist([1,2,3]), 0, 2, spamlist([1,2]),
493 "a[b:c]", "__getslice__")
494 testsetop(spamlist([1]), spamlist([2]), spamlist([1,2]),
495 "a+=b", "__iadd__")
496 testsetop(spamlist([1,2]), 3, spamlist([1,2,1,2,1,2]), "a*=b", "__imul__")
497 testunop(spamlist([1,2,3]), 3, "len(a)", "__len__")
498 testbinop(spamlist([1,2]), 3, spamlist([1,2,1,2,1,2]), "a*b", "__mul__")
499 testbinop(spamlist([1,2]), 3, spamlist([1,2,1,2,1,2]), "b*a", "__rmul__")
500 testset2op(spamlist([1,2]), 1, 3, spamlist([1,3]), "a[b]=c", "__setitem__")
501 testset3op(spamlist([1,2,3,4]), 1, 3, spamlist([5,6]),
502 spamlist([1,5,6,4]), "a[b:c]=d", "__setslice__")
503 # Test subclassing
504 class C(spam.spamlist):
505 def foo(self): return 1
506 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000507 vereq(a, [])
508 vereq(a.foo(), 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000509 a.append(100)
Guido van Rossum45704552001-10-08 16:35:45 +0000510 vereq(a, [100])
511 vereq(a.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000512 a.setstate(42)
Guido van Rossum45704552001-10-08 16:35:45 +0000513 vereq(a.getstate(), 42)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000514
515def spamdicts():
516 if verbose: print "Testing spamdict operations..."
517 import copy, xxsubtype as spam
518 def spamdict(d, memo=None):
519 import xxsubtype as spam
520 sd = spam.spamdict()
521 for k, v in d.items(): sd[k] = v
522 return sd
523 # This is an ugly hack:
524 copy._deepcopy_dispatch[spam.spamdict] = spamdict
525
Guido van Rossum47b9ff62006-08-24 00:41:19 +0000526 ##testbinop(spamdict({1:2}), spamdict({2:1}), -1, "cmp(a,b)", "__cmp__")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000527 testbinop(spamdict({1:2,3:4}), 1, 1, "b in a", "__contains__")
528 testbinop(spamdict({1:2,3:4}), 2, 0, "b in a", "__contains__")
529 testbinop(spamdict({1:2,3:4}), 1, 2, "a[b]", "__getitem__")
530 d = spamdict({1:2,3:4})
531 l1 = []
532 for i in d.keys(): l1.append(i)
533 l = []
534 for i in iter(d): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000535 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000536 l = []
537 for i in d.__iter__(): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000538 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000539 l = []
540 for i in type(spamdict({})).__iter__(d): l.append(i)
Guido van Rossum45704552001-10-08 16:35:45 +0000541 vereq(l, l1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000542 straightd = {1:2, 3:4}
543 spamd = spamdict(straightd)
544 testunop(spamd, 2, "len(a)", "__len__")
545 testunop(spamd, repr(straightd), "repr(a)", "__repr__")
546 testset2op(spamdict({1:2,3:4}), 2, 3, spamdict({1:2,2:3,3:4}),
547 "a[b]=c", "__setitem__")
548 # Test subclassing
549 class C(spam.spamdict):
550 def foo(self): return 1
551 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000552 vereq(a.items(), [])
553 vereq(a.foo(), 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000554 a['foo'] = 'bar'
Guido van Rossum45704552001-10-08 16:35:45 +0000555 vereq(a.items(), [('foo', 'bar')])
556 vereq(a.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000557 a.setstate(100)
Guido van Rossum45704552001-10-08 16:35:45 +0000558 vereq(a.getstate(), 100)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000559
560def pydicts():
561 if verbose: print "Testing Python subclass of dict..."
Tim Petersa427a2b2001-10-29 22:25:45 +0000562 verify(issubclass(dict, dict))
563 verify(isinstance({}, dict))
564 d = dict()
Guido van Rossum45704552001-10-08 16:35:45 +0000565 vereq(d, {})
Tim Petersa427a2b2001-10-29 22:25:45 +0000566 verify(d.__class__ is dict)
567 verify(isinstance(d, dict))
568 class C(dict):
Tim Peters6d6c1a32001-08-02 04:15:00 +0000569 state = -1
570 def __init__(self, *a, **kw):
571 if a:
Guido van Rossum90c45142001-11-24 21:07:01 +0000572 vereq(len(a), 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000573 self.state = a[0]
574 if kw:
575 for k, v in kw.items(): self[v] = k
576 def __getitem__(self, key):
577 return self.get(key, 0)
578 def __setitem__(self, key, value):
Guido van Rossum90c45142001-11-24 21:07:01 +0000579 verify(isinstance(key, type(0)))
Tim Petersa427a2b2001-10-29 22:25:45 +0000580 dict.__setitem__(self, key, value)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000581 def setstate(self, state):
582 self.state = state
583 def getstate(self):
584 return self.state
Tim Petersa427a2b2001-10-29 22:25:45 +0000585 verify(issubclass(C, dict))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000586 a1 = C(12)
Guido van Rossum45704552001-10-08 16:35:45 +0000587 vereq(a1.state, 12)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000588 a2 = C(foo=1, bar=2)
Guido van Rossum45704552001-10-08 16:35:45 +0000589 vereq(a2[1] == 'foo' and a2[2], 'bar')
Tim Peters6d6c1a32001-08-02 04:15:00 +0000590 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000591 vereq(a.state, -1)
592 vereq(a.getstate(), -1)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000593 a.setstate(0)
Guido van Rossum45704552001-10-08 16:35:45 +0000594 vereq(a.state, 0)
595 vereq(a.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000596 a.setstate(10)
Guido van Rossum45704552001-10-08 16:35:45 +0000597 vereq(a.state, 10)
598 vereq(a.getstate(), 10)
599 vereq(a[42], 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000600 a[42] = 24
Guido van Rossum45704552001-10-08 16:35:45 +0000601 vereq(a[42], 24)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000602 if verbose: print "pydict stress test ..."
603 N = 50
604 for i in range(N):
605 a[i] = C()
606 for j in range(N):
607 a[i][j] = i*j
608 for i in range(N):
609 for j in range(N):
Guido van Rossum45704552001-10-08 16:35:45 +0000610 vereq(a[i][j], i*j)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000611
612def pylists():
613 if verbose: print "Testing Python subclass of list..."
614 class C(list):
615 def __getitem__(self, i):
616 return list.__getitem__(self, i) + 100
617 def __getslice__(self, i, j):
618 return (i, j)
619 a = C()
620 a.extend([0,1,2])
Guido van Rossum45704552001-10-08 16:35:45 +0000621 vereq(a[0], 100)
622 vereq(a[1], 101)
623 vereq(a[2], 102)
624 vereq(a[100:200], (100,200))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000625
626def metaclass():
627 if verbose: print "Testing __metaclass__..."
Tim Peters6d6c1a32001-08-02 04:15:00 +0000628 class C:
629 __metaclass__ = type
630 def __init__(self):
631 self.__state = 0
632 def getstate(self):
633 return self.__state
634 def setstate(self, state):
635 self.__state = state
636 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000637 vereq(a.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000638 a.setstate(10)
Guido van Rossum45704552001-10-08 16:35:45 +0000639 vereq(a.getstate(), 10)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000640 class D:
641 class __metaclass__(type):
642 def myself(cls): return cls
Guido van Rossum45704552001-10-08 16:35:45 +0000643 vereq(D.myself(), D)
Guido van Rossum309b5662001-08-17 11:43:17 +0000644 d = D()
645 verify(d.__class__ is D)
646 class M1(type):
647 def __new__(cls, name, bases, dict):
648 dict['__spam__'] = 1
649 return type.__new__(cls, name, bases, dict)
650 class C:
651 __metaclass__ = M1
Guido van Rossum45704552001-10-08 16:35:45 +0000652 vereq(C.__spam__, 1)
Guido van Rossum309b5662001-08-17 11:43:17 +0000653 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000654 vereq(c.__spam__, 1)
Guido van Rossum91ee7982001-08-30 20:52:40 +0000655
Guido van Rossum309b5662001-08-17 11:43:17 +0000656 class _instance(object):
657 pass
658 class M2(object):
Guido van Rossum5a8a0372005-01-16 00:25:31 +0000659 @staticmethod
Guido van Rossum309b5662001-08-17 11:43:17 +0000660 def __new__(cls, name, bases, dict):
661 self = object.__new__(cls)
662 self.name = name
663 self.bases = bases
664 self.dict = dict
665 return self
Guido van Rossum309b5662001-08-17 11:43:17 +0000666 def __call__(self):
667 it = _instance()
Guido van Rossum7e1ff692001-08-17 11:55:58 +0000668 # Early binding of methods
669 for key in self.dict:
670 if key.startswith("__"):
671 continue
672 setattr(it, key, self.dict[key].__get__(it, self))
Guido van Rossum309b5662001-08-17 11:43:17 +0000673 return it
674 class C:
675 __metaclass__ = M2
676 def spam(self):
677 return 42
Guido van Rossum45704552001-10-08 16:35:45 +0000678 vereq(C.name, 'C')
679 vereq(C.bases, ())
Guido van Rossum309b5662001-08-17 11:43:17 +0000680 verify('spam' in C.dict)
681 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000682 vereq(c.spam(), 42)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000683
Guido van Rossum91ee7982001-08-30 20:52:40 +0000684 # More metaclass examples
685
686 class autosuper(type):
687 # Automatically add __super to the class
688 # This trick only works for dynamic classes
Guido van Rossum91ee7982001-08-30 20:52:40 +0000689 def __new__(metaclass, name, bases, dict):
Guido van Rossum91ee7982001-08-30 20:52:40 +0000690 cls = super(autosuper, metaclass).__new__(metaclass,
691 name, bases, dict)
Guido van Rossumbfa47b02001-08-31 04:35:14 +0000692 # Name mangling for __super removes leading underscores
Guido van Rossum91ee7982001-08-30 20:52:40 +0000693 while name[:1] == "_":
694 name = name[1:]
Guido van Rossum91ee7982001-08-30 20:52:40 +0000695 if name:
696 name = "_%s__super" % name
697 else:
698 name = "__super"
699 setattr(cls, name, super(cls))
700 return cls
701 class A:
702 __metaclass__ = autosuper
703 def meth(self):
704 return "A"
705 class B(A):
706 def meth(self):
707 return "B" + self.__super.meth()
708 class C(A):
709 def meth(self):
710 return "C" + self.__super.meth()
711 class D(C, B):
712 def meth(self):
713 return "D" + self.__super.meth()
Guido van Rossum45704552001-10-08 16:35:45 +0000714 vereq(D().meth(), "DCBA")
Guido van Rossum91ee7982001-08-30 20:52:40 +0000715 class E(B, C):
716 def meth(self):
717 return "E" + self.__super.meth()
Guido van Rossum45704552001-10-08 16:35:45 +0000718 vereq(E().meth(), "EBCA")
Guido van Rossum91ee7982001-08-30 20:52:40 +0000719
Guido van Rossum8bce4ac2001-09-06 21:56:42 +0000720 class autoproperty(type):
721 # Automatically create property attributes when methods
Guido van Rossum91ee7982001-08-30 20:52:40 +0000722 # named _get_x and/or _set_x are found
723 def __new__(metaclass, name, bases, dict):
724 hits = {}
725 for key, val in dict.iteritems():
726 if key.startswith("_get_"):
727 key = key[5:]
728 get, set = hits.get(key, (None, None))
729 get = val
730 hits[key] = get, set
731 elif key.startswith("_set_"):
732 key = key[5:]
733 get, set = hits.get(key, (None, None))
734 set = val
735 hits[key] = get, set
736 for key, (get, set) in hits.iteritems():
Guido van Rossum8bce4ac2001-09-06 21:56:42 +0000737 dict[key] = property(get, set)
738 return super(autoproperty, metaclass).__new__(metaclass,
Guido van Rossum91ee7982001-08-30 20:52:40 +0000739 name, bases, dict)
740 class A:
Guido van Rossum8bce4ac2001-09-06 21:56:42 +0000741 __metaclass__ = autoproperty
Guido van Rossum91ee7982001-08-30 20:52:40 +0000742 def _get_x(self):
743 return -self.__x
744 def _set_x(self, x):
745 self.__x = -x
746 a = A()
747 verify(not hasattr(a, "x"))
748 a.x = 12
Guido van Rossum45704552001-10-08 16:35:45 +0000749 vereq(a.x, 12)
750 vereq(a._A__x, -12)
Guido van Rossum91ee7982001-08-30 20:52:40 +0000751
Guido van Rossum8bce4ac2001-09-06 21:56:42 +0000752 class multimetaclass(autoproperty, autosuper):
Guido van Rossum91ee7982001-08-30 20:52:40 +0000753 # Merge of multiple cooperating metaclasses
754 pass
755 class A:
756 __metaclass__ = multimetaclass
757 def _get_x(self):
758 return "A"
759 class B(A):
760 def _get_x(self):
761 return "B" + self.__super._get_x()
762 class C(A):
763 def _get_x(self):
764 return "C" + self.__super._get_x()
765 class D(C, B):
766 def _get_x(self):
767 return "D" + self.__super._get_x()
Guido van Rossum45704552001-10-08 16:35:45 +0000768 vereq(D().x, "DCBA")
Guido van Rossum91ee7982001-08-30 20:52:40 +0000769
Guido van Rossumf76de622001-10-18 15:49:21 +0000770 # Make sure type(x) doesn't call x.__class__.__init__
771 class T(type):
772 counter = 0
773 def __init__(self, *args):
774 T.counter += 1
775 class C:
776 __metaclass__ = T
777 vereq(T.counter, 1)
778 a = C()
779 vereq(type(a), C)
780 vereq(T.counter, 1)
781
Guido van Rossum29d26062001-12-11 04:37:34 +0000782 class C(object): pass
783 c = C()
784 try: c()
785 except TypeError: pass
Neal Norwitzb1295da2002-04-01 18:59:20 +0000786 else: raise TestFailed, "calling object w/o call method should raise TypeError"
Guido van Rossum29d26062001-12-11 04:37:34 +0000787
Tim Peters6d6c1a32001-08-02 04:15:00 +0000788def pymods():
789 if verbose: print "Testing Python subclass of module..."
Tim Peters6d6c1a32001-08-02 04:15:00 +0000790 log = []
Guido van Rossumd3077402001-08-12 05:24:18 +0000791 import sys
792 MT = type(sys)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000793 class MM(MT):
Guido van Rossum1bdd9b02002-06-04 06:10:37 +0000794 def __init__(self, name):
795 MT.__init__(self, name)
Guido van Rossum867a8d22001-09-21 19:29:08 +0000796 def __getattribute__(self, name):
Tim Peters6d6c1a32001-08-02 04:15:00 +0000797 log.append(("getattr", name))
Guido van Rossum867a8d22001-09-21 19:29:08 +0000798 return MT.__getattribute__(self, name)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000799 def __setattr__(self, name, value):
800 log.append(("setattr", name, value))
801 MT.__setattr__(self, name, value)
802 def __delattr__(self, name):
803 log.append(("delattr", name))
804 MT.__delattr__(self, name)
Guido van Rossum1bdd9b02002-06-04 06:10:37 +0000805 a = MM("a")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000806 a.foo = 12
807 x = a.foo
808 del a.foo
Guido van Rossum45704552001-10-08 16:35:45 +0000809 vereq(log, [("setattr", "foo", 12),
810 ("getattr", "foo"),
811 ("delattr", "foo")])
Tim Peters6d6c1a32001-08-02 04:15:00 +0000812
813def multi():
814 if verbose: print "Testing multiple inheritance..."
Tim Peters6d6c1a32001-08-02 04:15:00 +0000815 class C(object):
816 def __init__(self):
817 self.__state = 0
818 def getstate(self):
819 return self.__state
820 def setstate(self, state):
821 self.__state = state
822 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +0000823 vereq(a.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000824 a.setstate(10)
Guido van Rossum45704552001-10-08 16:35:45 +0000825 vereq(a.getstate(), 10)
Tim Petersa427a2b2001-10-29 22:25:45 +0000826 class D(dict, C):
Tim Peters6d6c1a32001-08-02 04:15:00 +0000827 def __init__(self):
828 type({}).__init__(self)
829 C.__init__(self)
830 d = D()
Guido van Rossum45704552001-10-08 16:35:45 +0000831 vereq(d.keys(), [])
Tim Peters6d6c1a32001-08-02 04:15:00 +0000832 d["hello"] = "world"
Guido van Rossum45704552001-10-08 16:35:45 +0000833 vereq(d.items(), [("hello", "world")])
834 vereq(d["hello"], "world")
835 vereq(d.getstate(), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +0000836 d.setstate(10)
Guido van Rossum45704552001-10-08 16:35:45 +0000837 vereq(d.getstate(), 10)
Tim Petersa427a2b2001-10-29 22:25:45 +0000838 vereq(D.__mro__, (D, dict, C, object))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000839
Guido van Rossume45763a2001-08-10 21:28:46 +0000840 # SF bug #442833
841 class Node(object):
842 def __int__(self):
843 return int(self.foo())
844 def foo(self):
845 return "23"
846 class Frag(Node, list):
847 def foo(self):
848 return "42"
Guido van Rossum45704552001-10-08 16:35:45 +0000849 vereq(Node().__int__(), 23)
850 vereq(int(Node()), 23)
851 vereq(Frag().__int__(), 42)
852 vereq(int(Frag()), 42)
Guido van Rossume45763a2001-08-10 21:28:46 +0000853
Tim Peters6d6c1a32001-08-02 04:15:00 +0000854def diamond():
855 if verbose: print "Testing multiple inheritance special cases..."
856 class A(object):
857 def spam(self): return "A"
Guido van Rossum45704552001-10-08 16:35:45 +0000858 vereq(A().spam(), "A")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000859 class B(A):
860 def boo(self): return "B"
861 def spam(self): return "B"
Guido van Rossum45704552001-10-08 16:35:45 +0000862 vereq(B().spam(), "B")
863 vereq(B().boo(), "B")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000864 class C(A):
865 def boo(self): return "C"
Guido van Rossum45704552001-10-08 16:35:45 +0000866 vereq(C().spam(), "A")
867 vereq(C().boo(), "C")
Tim Peters6d6c1a32001-08-02 04:15:00 +0000868 class D(B, C): pass
Guido van Rossum45704552001-10-08 16:35:45 +0000869 vereq(D().spam(), "B")
870 vereq(D().boo(), "B")
871 vereq(D.__mro__, (D, B, C, A, object))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000872 class E(C, B): pass
Guido van Rossum45704552001-10-08 16:35:45 +0000873 vereq(E().spam(), "B")
874 vereq(E().boo(), "C")
875 vereq(E.__mro__, (E, C, B, A, object))
Guido van Rossum9a818922002-11-14 19:50:14 +0000876 # MRO order disagreement
877 try:
878 class F(D, E): pass
879 except TypeError:
880 pass
881 else:
882 raise TestFailed, "expected MRO order disagreement (F)"
883 try:
884 class G(E, D): pass
885 except TypeError:
886 pass
887 else:
888 raise TestFailed, "expected MRO order disagreement (G)"
889
890
891# see thread python-dev/2002-October/029035.html
892def ex5():
893 if verbose: print "Testing ex5 from C3 switch discussion..."
894 class A(object): pass
895 class B(object): pass
896 class C(object): pass
897 class X(A): pass
898 class Y(A): pass
899 class Z(X,B,Y,C): pass
900 vereq(Z.__mro__, (Z, X, B, Y, A, C, object))
901
902# see "A Monotonic Superclass Linearization for Dylan",
903# by Kim Barrett et al. (OOPSLA 1996)
904def monotonicity():
905 if verbose: print "Testing MRO monotonicity..."
906 class Boat(object): pass
907 class DayBoat(Boat): pass
908 class WheelBoat(Boat): pass
909 class EngineLess(DayBoat): pass
910 class SmallMultihull(DayBoat): pass
911 class PedalWheelBoat(EngineLess,WheelBoat): pass
912 class SmallCatamaran(SmallMultihull): pass
913 class Pedalo(PedalWheelBoat,SmallCatamaran): pass
914
915 vereq(PedalWheelBoat.__mro__,
916 (PedalWheelBoat, EngineLess, DayBoat, WheelBoat, Boat,
917 object))
918 vereq(SmallCatamaran.__mro__,
919 (SmallCatamaran, SmallMultihull, DayBoat, Boat, object))
920
921 vereq(Pedalo.__mro__,
922 (Pedalo, PedalWheelBoat, EngineLess, SmallCatamaran,
923 SmallMultihull, DayBoat, WheelBoat, Boat, object))
924
925# see "A Monotonic Superclass Linearization for Dylan",
926# by Kim Barrett et al. (OOPSLA 1996)
927def consistency_with_epg():
928 if verbose: print "Testing consistentcy with EPG..."
929 class Pane(object): pass
930 class ScrollingMixin(object): pass
931 class EditingMixin(object): pass
932 class ScrollablePane(Pane,ScrollingMixin): pass
933 class EditablePane(Pane,EditingMixin): pass
934 class EditableScrollablePane(ScrollablePane,EditablePane): pass
935
936 vereq(EditableScrollablePane.__mro__,
937 (EditableScrollablePane, ScrollablePane, EditablePane,
938 Pane, ScrollingMixin, EditingMixin, object))
Tim Peters6d6c1a32001-08-02 04:15:00 +0000939
Raymond Hettingerf394df42003-04-06 19:13:41 +0000940mro_err_msg = """Cannot create a consistent method resolution
941order (MRO) for bases """
Raymond Hettinger83245b52003-03-12 04:25:42 +0000942
Guido van Rossumd32047f2002-11-25 21:38:52 +0000943def mro_disagreement():
944 if verbose: print "Testing error messages for MRO disagreement..."
945 def raises(exc, expected, callable, *args):
946 try:
947 callable(*args)
948 except exc, msg:
Guido van Rossuma01fa262002-11-27 04:00:59 +0000949 if not str(msg).startswith(expected):
Guido van Rossumd32047f2002-11-25 21:38:52 +0000950 raise TestFailed, "Message %r, expected %r" % (str(msg),
951 expected)
952 else:
953 raise TestFailed, "Expected %s" % exc
954 class A(object): pass
955 class B(A): pass
956 class C(object): pass
957 # Test some very simple errors
958 raises(TypeError, "duplicate base class A",
959 type, "X", (A, A), {})
Raymond Hettinger83245b52003-03-12 04:25:42 +0000960 raises(TypeError, mro_err_msg,
Guido van Rossumd32047f2002-11-25 21:38:52 +0000961 type, "X", (A, B), {})
Raymond Hettinger83245b52003-03-12 04:25:42 +0000962 raises(TypeError, mro_err_msg,
Guido van Rossumd32047f2002-11-25 21:38:52 +0000963 type, "X", (A, C, B), {})
964 # Test a slightly more complex error
965 class GridLayout(object): pass
966 class HorizontalGrid(GridLayout): pass
967 class VerticalGrid(GridLayout): pass
968 class HVGrid(HorizontalGrid, VerticalGrid): pass
969 class VHGrid(VerticalGrid, HorizontalGrid): pass
Raymond Hettinger83245b52003-03-12 04:25:42 +0000970 raises(TypeError, mro_err_msg,
Guido van Rossumd32047f2002-11-25 21:38:52 +0000971 type, "ConfusedGrid", (HVGrid, VHGrid), {})
972
Guido van Rossum37202612001-08-09 19:45:21 +0000973def objects():
974 if verbose: print "Testing object class..."
975 a = object()
Guido van Rossum45704552001-10-08 16:35:45 +0000976 vereq(a.__class__, object)
977 vereq(type(a), object)
Guido van Rossum37202612001-08-09 19:45:21 +0000978 b = object()
979 verify(a is not b)
980 verify(not hasattr(a, "foo"))
981 try:
982 a.foo = 12
Guido van Rossum6d946272001-08-10 19:42:38 +0000983 except (AttributeError, TypeError):
Guido van Rossum37202612001-08-09 19:45:21 +0000984 pass
985 else:
986 verify(0, "object() should not allow setting a foo attribute")
987 verify(not hasattr(object(), "__dict__"))
988
989 class Cdict(object):
990 pass
991 x = Cdict()
Guido van Rossum45704552001-10-08 16:35:45 +0000992 vereq(x.__dict__, {})
Guido van Rossum37202612001-08-09 19:45:21 +0000993 x.foo = 1
Guido van Rossum45704552001-10-08 16:35:45 +0000994 vereq(x.foo, 1)
995 vereq(x.__dict__, {'foo': 1})
Guido van Rossum37202612001-08-09 19:45:21 +0000996
Tim Peters6d6c1a32001-08-02 04:15:00 +0000997def slots():
998 if verbose: print "Testing __slots__..."
999 class C0(object):
1000 __slots__ = []
1001 x = C0()
1002 verify(not hasattr(x, "__dict__"))
1003 verify(not hasattr(x, "foo"))
1004
1005 class C1(object):
1006 __slots__ = ['a']
1007 x = C1()
1008 verify(not hasattr(x, "__dict__"))
Guido van Rossum6b705992001-12-04 16:23:42 +00001009 verify(not hasattr(x, "a"))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001010 x.a = 1
Guido van Rossum45704552001-10-08 16:35:45 +00001011 vereq(x.a, 1)
Guido van Rossum6b705992001-12-04 16:23:42 +00001012 x.a = None
1013 veris(x.a, None)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001014 del x.a
Guido van Rossum6b705992001-12-04 16:23:42 +00001015 verify(not hasattr(x, "a"))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001016
1017 class C3(object):
1018 __slots__ = ['a', 'b', 'c']
1019 x = C3()
1020 verify(not hasattr(x, "__dict__"))
Guido van Rossum6b705992001-12-04 16:23:42 +00001021 verify(not hasattr(x, 'a'))
1022 verify(not hasattr(x, 'b'))
1023 verify(not hasattr(x, 'c'))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001024 x.a = 1
1025 x.b = 2
1026 x.c = 3
Guido van Rossum45704552001-10-08 16:35:45 +00001027 vereq(x.a, 1)
1028 vereq(x.b, 2)
1029 vereq(x.c, 3)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001030
Raymond Hettinger0ae0c072002-06-20 22:23:15 +00001031 class C4(object):
1032 """Validate name mangling"""
1033 __slots__ = ['__a']
1034 def __init__(self, value):
1035 self.__a = value
1036 def get(self):
1037 return self.__a
1038 x = C4(5)
1039 verify(not hasattr(x, '__dict__'))
1040 verify(not hasattr(x, '__a'))
1041 vereq(x.get(), 5)
1042 try:
1043 x.__a = 6
1044 except AttributeError:
1045 pass
1046 else:
1047 raise TestFailed, "Double underscored names not mangled"
1048
Guido van Rossum9923ffe2002-06-04 19:52:53 +00001049 # Make sure slot names are proper identifiers
1050 try:
1051 class C(object):
1052 __slots__ = [None]
1053 except TypeError:
1054 pass
1055 else:
1056 raise TestFailed, "[None] slots not caught"
1057 try:
1058 class C(object):
1059 __slots__ = ["foo bar"]
1060 except TypeError:
1061 pass
1062 else:
1063 raise TestFailed, "['foo bar'] slots not caught"
1064 try:
1065 class C(object):
1066 __slots__ = ["foo\0bar"]
1067 except TypeError:
1068 pass
1069 else:
1070 raise TestFailed, "['foo\\0bar'] slots not caught"
1071 try:
1072 class C(object):
1073 __slots__ = ["1"]
1074 except TypeError:
1075 pass
1076 else:
1077 raise TestFailed, "['1'] slots not caught"
1078 try:
1079 class C(object):
1080 __slots__ = [""]
1081 except TypeError:
1082 pass
1083 else:
1084 raise TestFailed, "[''] slots not caught"
1085 class C(object):
1086 __slots__ = ["a", "a_b", "_a", "A0123456789Z"]
1087
Guido van Rossum33bab012001-12-05 22:45:48 +00001088 # Test leaks
1089 class Counted(object):
1090 counter = 0 # counts the number of instances alive
1091 def __init__(self):
1092 Counted.counter += 1
1093 def __del__(self):
1094 Counted.counter -= 1
1095 class C(object):
1096 __slots__ = ['a', 'b', 'c']
1097 x = C()
1098 x.a = Counted()
1099 x.b = Counted()
1100 x.c = Counted()
1101 vereq(Counted.counter, 3)
1102 del x
1103 vereq(Counted.counter, 0)
1104 class D(C):
1105 pass
1106 x = D()
1107 x.a = Counted()
1108 x.z = Counted()
1109 vereq(Counted.counter, 2)
1110 del x
1111 vereq(Counted.counter, 0)
1112 class E(D):
1113 __slots__ = ['e']
1114 x = E()
1115 x.a = Counted()
1116 x.z = Counted()
1117 x.e = Counted()
1118 vereq(Counted.counter, 3)
1119 del x
1120 vereq(Counted.counter, 0)
1121
Guido van Rossum9923ffe2002-06-04 19:52:53 +00001122 # Test cyclical leaks [SF bug 519621]
1123 class F(object):
1124 __slots__ = ['a', 'b']
1125 log = []
1126 s = F()
1127 s.a = [Counted(), s]
1128 vereq(Counted.counter, 1)
1129 s = None
1130 import gc
1131 gc.collect()
1132 vereq(Counted.counter, 0)
1133
Raymond Hettingerab5dae32002-06-24 13:08:16 +00001134 # Test lookup leaks [SF bug 572567]
1135 import sys,gc
1136 class G(object):
1137 def __cmp__(self, other):
1138 return 0
1139 g = G()
1140 orig_objects = len(gc.get_objects())
1141 for i in xrange(10):
1142 g==g
1143 new_objects = len(gc.get_objects())
1144 vereq(orig_objects, new_objects)
Neal Norwitz98a379e2003-06-16 22:51:22 +00001145 class H(object):
1146 __slots__ = ['a', 'b']
1147 def __init__(self):
1148 self.a = 1
1149 self.b = 2
1150 def __del__(self):
1151 assert self.a == 1
1152 assert self.b == 2
1153
1154 save_stderr = sys.stderr
1155 sys.stderr = sys.stdout
1156 h = H()
1157 try:
1158 del h
1159 finally:
1160 sys.stderr = save_stderr
Raymond Hettingerab5dae32002-06-24 13:08:16 +00001161
Guido van Rossum8b056da2002-08-13 18:26:26 +00001162def slotspecials():
1163 if verbose: print "Testing __dict__ and __weakref__ in __slots__..."
1164
1165 class D(object):
1166 __slots__ = ["__dict__"]
1167 a = D()
1168 verify(hasattr(a, "__dict__"))
1169 verify(not hasattr(a, "__weakref__"))
1170 a.foo = 42
1171 vereq(a.__dict__, {"foo": 42})
1172
1173 class W(object):
1174 __slots__ = ["__weakref__"]
1175 a = W()
1176 verify(hasattr(a, "__weakref__"))
1177 verify(not hasattr(a, "__dict__"))
1178 try:
1179 a.foo = 42
1180 except AttributeError:
1181 pass
1182 else:
1183 raise TestFailed, "shouldn't be allowed to set a.foo"
1184
1185 class C1(W, D):
1186 __slots__ = []
1187 a = C1()
1188 verify(hasattr(a, "__dict__"))
1189 verify(hasattr(a, "__weakref__"))
1190 a.foo = 42
1191 vereq(a.__dict__, {"foo": 42})
1192
1193 class C2(D, W):
1194 __slots__ = []
1195 a = C2()
1196 verify(hasattr(a, "__dict__"))
1197 verify(hasattr(a, "__weakref__"))
1198 a.foo = 42
1199 vereq(a.__dict__, {"foo": 42})
1200
Guido van Rossum9a818922002-11-14 19:50:14 +00001201# MRO order disagreement
1202#
1203# class C3(C1, C2):
1204# __slots__ = []
1205#
1206# class C4(C2, C1):
1207# __slots__ = []
Guido van Rossum8b056da2002-08-13 18:26:26 +00001208
Tim Peters6d6c1a32001-08-02 04:15:00 +00001209def dynamics():
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001210 if verbose: print "Testing class attribute propagation..."
Tim Peters6d6c1a32001-08-02 04:15:00 +00001211 class D(object):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001212 pass
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001213 class E(D):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001214 pass
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001215 class F(D):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001216 pass
Tim Peters6d6c1a32001-08-02 04:15:00 +00001217 D.foo = 1
Guido van Rossum45704552001-10-08 16:35:45 +00001218 vereq(D.foo, 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001219 # Test that dynamic attributes are inherited
Guido van Rossum45704552001-10-08 16:35:45 +00001220 vereq(E.foo, 1)
1221 vereq(F.foo, 1)
Guido van Rossum9d4fe422001-08-12 03:38:18 +00001222 # Test dynamic instances
1223 class C(object):
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001224 pass
Guido van Rossum9d4fe422001-08-12 03:38:18 +00001225 a = C()
Guido van Rossumd3077402001-08-12 05:24:18 +00001226 verify(not hasattr(a, "foobar"))
Guido van Rossum9d4fe422001-08-12 03:38:18 +00001227 C.foobar = 2
Guido van Rossum45704552001-10-08 16:35:45 +00001228 vereq(a.foobar, 2)
Guido van Rossum9d4fe422001-08-12 03:38:18 +00001229 C.method = lambda self: 42
Guido van Rossum45704552001-10-08 16:35:45 +00001230 vereq(a.method(), 42)
Guido van Rossum9d4fe422001-08-12 03:38:18 +00001231 C.__repr__ = lambda self: "C()"
Guido van Rossum45704552001-10-08 16:35:45 +00001232 vereq(repr(a), "C()")
Guido van Rossumd3077402001-08-12 05:24:18 +00001233 C.__int__ = lambda self: 100
Guido van Rossum45704552001-10-08 16:35:45 +00001234 vereq(int(a), 100)
1235 vereq(a.foobar, 2)
Guido van Rossumd3077402001-08-12 05:24:18 +00001236 verify(not hasattr(a, "spam"))
1237 def mygetattr(self, name):
1238 if name == "spam":
1239 return "spam"
Guido van Rossum19c1cd52001-09-21 21:24:49 +00001240 raise AttributeError
1241 C.__getattr__ = mygetattr
Guido van Rossum45704552001-10-08 16:35:45 +00001242 vereq(a.spam, "spam")
Guido van Rossumd3077402001-08-12 05:24:18 +00001243 a.new = 12
Guido van Rossum45704552001-10-08 16:35:45 +00001244 vereq(a.new, 12)
Guido van Rossumd3077402001-08-12 05:24:18 +00001245 def mysetattr(self, name, value):
1246 if name == "spam":
1247 raise AttributeError
1248 return object.__setattr__(self, name, value)
1249 C.__setattr__ = mysetattr
1250 try:
1251 a.spam = "not spam"
1252 except AttributeError:
1253 pass
1254 else:
1255 verify(0, "expected AttributeError")
Guido van Rossum45704552001-10-08 16:35:45 +00001256 vereq(a.spam, "spam")
Guido van Rossum80e36752001-08-14 20:00:33 +00001257 class D(C):
1258 pass
1259 d = D()
1260 d.foo = 1
Guido van Rossum45704552001-10-08 16:35:45 +00001261 vereq(d.foo, 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001262
Guido van Rossum7e35d572001-09-15 03:14:32 +00001263 # Test handling of int*seq and seq*int
1264 class I(int):
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001265 pass
Guido van Rossum45704552001-10-08 16:35:45 +00001266 vereq("a"*I(2), "aa")
1267 vereq(I(2)*"a", "aa")
1268 vereq(2*I(3), 6)
1269 vereq(I(3)*2, 6)
1270 vereq(I(3)*I(2), 6)
Guido van Rossum7e35d572001-09-15 03:14:32 +00001271
1272 # Test handling of long*seq and seq*long
1273 class L(long):
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001274 pass
Guido van Rossum45704552001-10-08 16:35:45 +00001275 vereq("a"*L(2L), "aa")
1276 vereq(L(2L)*"a", "aa")
1277 vereq(2*L(3), 6)
1278 vereq(L(3)*2, 6)
1279 vereq(L(3)*L(2), 6)
Guido van Rossum7e35d572001-09-15 03:14:32 +00001280
Guido van Rossum3d45d8f2001-09-24 18:47:40 +00001281 # Test comparison of classes with dynamic metaclasses
1282 class dynamicmetaclass(type):
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00001283 pass
Guido van Rossum3d45d8f2001-09-24 18:47:40 +00001284 class someclass:
1285 __metaclass__ = dynamicmetaclass
1286 verify(someclass != object)
1287
Tim Peters6d6c1a32001-08-02 04:15:00 +00001288def errors():
1289 if verbose: print "Testing errors..."
1290
1291 try:
Tim Petersa427a2b2001-10-29 22:25:45 +00001292 class C(list, dict):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001293 pass
1294 except TypeError:
1295 pass
1296 else:
1297 verify(0, "inheritance from both list and dict should be illegal")
1298
1299 try:
1300 class C(object, None):
1301 pass
1302 except TypeError:
1303 pass
1304 else:
1305 verify(0, "inheritance from non-type should be illegal")
1306 class Classic:
1307 pass
1308
1309 try:
Guido van Rossum8aea0cc2001-08-29 15:48:43 +00001310 class C(type(len)):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001311 pass
1312 except TypeError:
1313 pass
1314 else:
Guido van Rossum8aea0cc2001-08-29 15:48:43 +00001315 verify(0, "inheritance from CFunction should be illegal")
Tim Peters6d6c1a32001-08-02 04:15:00 +00001316
1317 try:
1318 class C(object):
1319 __slots__ = 1
1320 except TypeError:
1321 pass
1322 else:
1323 verify(0, "__slots__ = 1 should be illegal")
1324
1325 try:
1326 class C(object):
1327 __slots__ = [1]
1328 except TypeError:
1329 pass
1330 else:
1331 verify(0, "__slots__ = [1] should be illegal")
1332
1333def classmethods():
1334 if verbose: print "Testing class methods..."
1335 class C(object):
1336 def foo(*a): return a
1337 goo = classmethod(foo)
1338 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001339 vereq(C.goo(1), (C, 1))
1340 vereq(c.goo(1), (C, 1))
1341 vereq(c.foo(1), (c, 1))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001342 class D(C):
1343 pass
1344 d = D()
Guido van Rossum45704552001-10-08 16:35:45 +00001345 vereq(D.goo(1), (D, 1))
1346 vereq(d.goo(1), (D, 1))
1347 vereq(d.foo(1), (d, 1))
1348 vereq(D.foo(d, 1), (d, 1))
Guido van Rossum7e305482002-03-18 03:09:06 +00001349 # Test for a specific crash (SF bug 528132)
1350 def f(cls, arg): return (cls, arg)
1351 ff = classmethod(f)
1352 vereq(ff.__get__(0, int)(42), (int, 42))
1353 vereq(ff.__get__(0)(42), (int, 42))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001354
Guido van Rossum155db9a2002-04-02 17:53:47 +00001355 # Test super() with classmethods (SF bug 535444)
1356 veris(C.goo.im_self, C)
1357 veris(D.goo.im_self, D)
1358 veris(super(D,D).goo.im_self, D)
1359 veris(super(D,d).goo.im_self, D)
1360 vereq(super(D,D).goo(), (D,))
1361 vereq(super(D,d).goo(), (D,))
1362
Raymond Hettingerbe971532003-06-18 01:13:41 +00001363 # Verify that argument is checked for callability (SF bug 753451)
1364 try:
1365 classmethod(1).__get__(1)
1366 except TypeError:
1367 pass
1368 else:
1369 raise TestFailed, "classmethod should check for callability"
1370
Georg Brandl6a29c322006-02-21 22:17:46 +00001371 # Verify that classmethod() doesn't allow keyword args
1372 try:
1373 classmethod(f, kw=1)
1374 except TypeError:
1375 pass
1376 else:
1377 raise TestFailed, "classmethod shouldn't accept keyword args"
1378
Fred Drakef841aa62002-03-28 15:49:54 +00001379def classmethods_in_c():
1380 if verbose: print "Testing C-based class methods..."
1381 import xxsubtype as spam
1382 a = (1, 2, 3)
1383 d = {'abc': 123}
1384 x, a1, d1 = spam.spamlist.classmeth(*a, **d)
Tim Petersbca1cbc2002-12-09 22:56:13 +00001385 veris(x, spam.spamlist)
1386 vereq(a, a1)
Fred Drakef841aa62002-03-28 15:49:54 +00001387 vereq(d, d1)
1388 x, a1, d1 = spam.spamlist().classmeth(*a, **d)
Tim Petersbca1cbc2002-12-09 22:56:13 +00001389 veris(x, spam.spamlist)
1390 vereq(a, a1)
Fred Drakef841aa62002-03-28 15:49:54 +00001391 vereq(d, d1)
1392
Tim Peters6d6c1a32001-08-02 04:15:00 +00001393def staticmethods():
1394 if verbose: print "Testing static methods..."
1395 class C(object):
1396 def foo(*a): return a
1397 goo = staticmethod(foo)
1398 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001399 vereq(C.goo(1), (1,))
1400 vereq(c.goo(1), (1,))
1401 vereq(c.foo(1), (c, 1,))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001402 class D(C):
1403 pass
1404 d = D()
Guido van Rossum45704552001-10-08 16:35:45 +00001405 vereq(D.goo(1), (1,))
1406 vereq(d.goo(1), (1,))
1407 vereq(d.foo(1), (d, 1))
1408 vereq(D.foo(d, 1), (d, 1))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001409
Fred Drakef841aa62002-03-28 15:49:54 +00001410def staticmethods_in_c():
1411 if verbose: print "Testing C-based static methods..."
1412 import xxsubtype as spam
1413 a = (1, 2, 3)
1414 d = {"abc": 123}
1415 x, a1, d1 = spam.spamlist.staticmeth(*a, **d)
1416 veris(x, None)
1417 vereq(a, a1)
1418 vereq(d, d1)
1419 x, a1, d2 = spam.spamlist().staticmeth(*a, **d)
1420 veris(x, None)
1421 vereq(a, a1)
1422 vereq(d, d1)
1423
Tim Peters6d6c1a32001-08-02 04:15:00 +00001424def classic():
1425 if verbose: print "Testing classic classes..."
1426 class C:
1427 def foo(*a): return a
1428 goo = classmethod(foo)
1429 c = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001430 vereq(C.goo(1), (C, 1))
1431 vereq(c.goo(1), (C, 1))
1432 vereq(c.foo(1), (c, 1))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001433 class D(C):
1434 pass
1435 d = D()
Guido van Rossum45704552001-10-08 16:35:45 +00001436 vereq(D.goo(1), (D, 1))
1437 vereq(d.goo(1), (D, 1))
1438 vereq(d.foo(1), (d, 1))
1439 vereq(D.foo(d, 1), (d, 1))
Guido van Rossum93018762001-08-17 13:40:47 +00001440 class E: # *not* subclassing from C
1441 foo = C.foo
Guido van Rossum45704552001-10-08 16:35:45 +00001442 vereq(E().foo, C.foo) # i.e., unbound
Guido van Rossum84a79a82001-08-17 13:58:31 +00001443 verify(repr(C.foo.__get__(C())).startswith("<bound method "))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001444
1445def compattr():
1446 if verbose: print "Testing computed attributes..."
1447 class C(object):
1448 class computed_attribute(object):
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001449 def __init__(self, get, set=None, delete=None):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001450 self.__get = get
1451 self.__set = set
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001452 self.__delete = delete
Tim Peters6d6c1a32001-08-02 04:15:00 +00001453 def __get__(self, obj, type=None):
1454 return self.__get(obj)
1455 def __set__(self, obj, value):
1456 return self.__set(obj, value)
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001457 def __delete__(self, obj):
1458 return self.__delete(obj)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001459 def __init__(self):
1460 self.__x = 0
1461 def __get_x(self):
1462 x = self.__x
1463 self.__x = x+1
1464 return x
1465 def __set_x(self, x):
1466 self.__x = x
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001467 def __delete_x(self):
1468 del self.__x
1469 x = computed_attribute(__get_x, __set_x, __delete_x)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001470 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001471 vereq(a.x, 0)
1472 vereq(a.x, 1)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001473 a.x = 10
Guido van Rossum45704552001-10-08 16:35:45 +00001474 vereq(a.x, 10)
1475 vereq(a.x, 11)
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001476 del a.x
1477 vereq(hasattr(a, 'x'), 0)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001478
1479def newslot():
1480 if verbose: print "Testing __new__ slot override..."
1481 class C(list):
1482 def __new__(cls):
1483 self = list.__new__(cls)
1484 self.foo = 1
1485 return self
1486 def __init__(self):
1487 self.foo = self.foo + 2
1488 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001489 vereq(a.foo, 3)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001490 verify(a.__class__ is C)
1491 class D(C):
1492 pass
1493 b = D()
Guido van Rossum45704552001-10-08 16:35:45 +00001494 vereq(b.foo, 3)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001495 verify(b.__class__ is D)
1496
Tim Peters6d6c1a32001-08-02 04:15:00 +00001497def altmro():
1498 if verbose: print "Testing mro() and overriding it..."
1499 class A(object):
1500 def f(self): return "A"
1501 class B(A):
1502 pass
1503 class C(A):
1504 def f(self): return "C"
1505 class D(B, C):
1506 pass
Guido van Rossum45704552001-10-08 16:35:45 +00001507 vereq(D.mro(), [D, B, C, A, object])
1508 vereq(D.__mro__, (D, B, C, A, object))
1509 vereq(D().f(), "C")
Guido van Rossum9a818922002-11-14 19:50:14 +00001510
Guido van Rossumd3077402001-08-12 05:24:18 +00001511 class PerverseMetaType(type):
1512 def mro(cls):
1513 L = type.mro(cls)
1514 L.reverse()
1515 return L
Guido van Rossum9a818922002-11-14 19:50:14 +00001516 class X(D,B,C,A):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001517 __metaclass__ = PerverseMetaType
Guido van Rossum45704552001-10-08 16:35:45 +00001518 vereq(X.__mro__, (object, A, C, B, D, X))
1519 vereq(X().f(), "A")
Tim Peters6d6c1a32001-08-02 04:15:00 +00001520
Armin Rigo037d1e02005-12-29 17:07:39 +00001521 try:
1522 class X(object):
1523 class __metaclass__(type):
1524 def mro(self):
1525 return [self, dict, object]
1526 except TypeError:
1527 pass
1528 else:
1529 raise TestFailed, "devious mro() return not caught"
1530
1531 try:
1532 class X(object):
1533 class __metaclass__(type):
1534 def mro(self):
1535 return [1]
1536 except TypeError:
1537 pass
1538 else:
1539 raise TestFailed, "non-class mro() return not caught"
1540
1541 try:
1542 class X(object):
1543 class __metaclass__(type):
1544 def mro(self):
1545 return 1
1546 except TypeError:
1547 pass
1548 else:
1549 raise TestFailed, "non-sequence mro() return not caught"
Tim Peters1b27f862005-12-30 18:42:42 +00001550
Armin Rigo037d1e02005-12-29 17:07:39 +00001551
Tim Peters6d6c1a32001-08-02 04:15:00 +00001552def overloading():
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00001553 if verbose: print "Testing operator overloading..."
Tim Peters6d6c1a32001-08-02 04:15:00 +00001554
1555 class B(object):
1556 "Intermediate class because object doesn't have a __setattr__"
1557
1558 class C(B):
1559
Guido van Rossum19c1cd52001-09-21 21:24:49 +00001560 def __getattr__(self, name):
Tim Peters6d6c1a32001-08-02 04:15:00 +00001561 if name == "foo":
1562 return ("getattr", name)
1563 else:
Guido van Rossum19c1cd52001-09-21 21:24:49 +00001564 raise AttributeError
Tim Peters6d6c1a32001-08-02 04:15:00 +00001565 def __setattr__(self, name, value):
1566 if name == "foo":
1567 self.setattr = (name, value)
1568 else:
1569 return B.__setattr__(self, name, value)
1570 def __delattr__(self, name):
1571 if name == "foo":
1572 self.delattr = name
1573 else:
1574 return B.__delattr__(self, name)
1575
1576 def __getitem__(self, key):
1577 return ("getitem", key)
1578 def __setitem__(self, key, value):
1579 self.setitem = (key, value)
1580 def __delitem__(self, key):
1581 self.delitem = key
1582
1583 def __getslice__(self, i, j):
1584 return ("getslice", i, j)
1585 def __setslice__(self, i, j, value):
1586 self.setslice = (i, j, value)
1587 def __delslice__(self, i, j):
1588 self.delslice = (i, j)
1589
1590 a = C()
Guido van Rossum45704552001-10-08 16:35:45 +00001591 vereq(a.foo, ("getattr", "foo"))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001592 a.foo = 12
Guido van Rossum45704552001-10-08 16:35:45 +00001593 vereq(a.setattr, ("foo", 12))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001594 del a.foo
Guido van Rossum45704552001-10-08 16:35:45 +00001595 vereq(a.delattr, "foo")
Tim Peters6d6c1a32001-08-02 04:15:00 +00001596
Guido van Rossum45704552001-10-08 16:35:45 +00001597 vereq(a[12], ("getitem", 12))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001598 a[12] = 21
Guido van Rossum45704552001-10-08 16:35:45 +00001599 vereq(a.setitem, (12, 21))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001600 del a[12]
Guido van Rossum45704552001-10-08 16:35:45 +00001601 vereq(a.delitem, 12)
Tim Peters6d6c1a32001-08-02 04:15:00 +00001602
Guido van Rossum45704552001-10-08 16:35:45 +00001603 vereq(a[0:10], ("getslice", 0, 10))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001604 a[0:10] = "foo"
Guido van Rossum45704552001-10-08 16:35:45 +00001605 vereq(a.setslice, (0, 10, "foo"))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001606 del a[0:10]
Guido van Rossum45704552001-10-08 16:35:45 +00001607 vereq(a.delslice, (0, 10))
Tim Peters6d6c1a32001-08-02 04:15:00 +00001608
Guido van Rossumb5a136b2001-08-15 17:51:17 +00001609def methods():
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00001610 if verbose: print "Testing methods..."
Guido van Rossumb5a136b2001-08-15 17:51:17 +00001611 class C(object):
1612 def __init__(self, x):
1613 self.x = x
1614 def foo(self):
1615 return self.x
1616 c1 = C(1)
Guido van Rossum45704552001-10-08 16:35:45 +00001617 vereq(c1.foo(), 1)
Guido van Rossumb5a136b2001-08-15 17:51:17 +00001618 class D(C):
1619 boo = C.foo
1620 goo = c1.foo
1621 d2 = D(2)
Guido van Rossum45704552001-10-08 16:35:45 +00001622 vereq(d2.foo(), 2)
1623 vereq(d2.boo(), 2)
1624 vereq(d2.goo(), 1)
Guido van Rossum93018762001-08-17 13:40:47 +00001625 class E(object):
1626 foo = C.foo
Guido van Rossum45704552001-10-08 16:35:45 +00001627 vereq(E().foo, C.foo) # i.e., unbound
Guido van Rossum84a79a82001-08-17 13:58:31 +00001628 verify(repr(C.foo.__get__(C(1))).startswith("<bound method "))
Guido van Rossumb5a136b2001-08-15 17:51:17 +00001629
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001630def specials():
1631 # Test operators like __hash__ for which a built-in default exists
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00001632 if verbose: print "Testing special operators..."
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001633 # Test the default behavior for static classes
1634 class C(object):
1635 def __getitem__(self, i):
1636 if 0 <= i < 10: return i
1637 raise IndexError
1638 c1 = C()
1639 c2 = C()
1640 verify(not not c1)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001641 verify(id(c1) != id(c2))
1642 hash(c1)
1643 hash(c2)
Guido van Rossum47b9ff62006-08-24 00:41:19 +00001644 ##vereq(cmp(c1, c2), cmp(id(c1), id(c2)))
Guido van Rossum45704552001-10-08 16:35:45 +00001645 vereq(c1, c1)
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001646 verify(c1 != c2)
1647 verify(not c1 != c1)
1648 verify(not c1 == c2)
Tim Peters4d2dded2001-08-16 19:50:51 +00001649 # Note that the module name appears in str/repr, and that varies
1650 # depending on whether this test is run standalone or from a framework.
Guido van Rossumff0e6d62001-09-24 16:03:59 +00001651 verify(str(c1).find('C object at ') >= 0)
Guido van Rossum45704552001-10-08 16:35:45 +00001652 vereq(str(c1), repr(c1))
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001653 verify(-1 not in c1)
1654 for i in range(10):
1655 verify(i in c1)
1656 verify(10 not in c1)
1657 # Test the default behavior for dynamic classes
1658 class D(object):
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001659 def __getitem__(self, i):
1660 if 0 <= i < 10: return i
1661 raise IndexError
1662 d1 = D()
1663 d2 = D()
1664 verify(not not d1)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001665 verify(id(d1) != id(d2))
1666 hash(d1)
1667 hash(d2)
Guido van Rossum47b9ff62006-08-24 00:41:19 +00001668 ##vereq(cmp(d1, d2), cmp(id(d1), id(d2)))
Guido van Rossum45704552001-10-08 16:35:45 +00001669 vereq(d1, d1)
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001670 verify(d1 != d2)
1671 verify(not d1 != d1)
1672 verify(not d1 == d2)
Tim Peters4d2dded2001-08-16 19:50:51 +00001673 # Note that the module name appears in str/repr, and that varies
1674 # depending on whether this test is run standalone or from a framework.
Guido van Rossumff0e6d62001-09-24 16:03:59 +00001675 verify(str(d1).find('D object at ') >= 0)
Guido van Rossum45704552001-10-08 16:35:45 +00001676 vereq(str(d1), repr(d1))
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001677 verify(-1 not in d1)
1678 for i in range(10):
1679 verify(i in d1)
1680 verify(10 not in d1)
1681 # Test overridden behavior for static classes
1682 class Proxy(object):
1683 def __init__(self, x):
1684 self.x = x
1685 def __nonzero__(self):
1686 return not not self.x
1687 def __hash__(self):
1688 return hash(self.x)
1689 def __eq__(self, other):
1690 return self.x == other
1691 def __ne__(self, other):
1692 return self.x != other
1693 def __cmp__(self, other):
1694 return cmp(self.x, other.x)
1695 def __str__(self):
1696 return "Proxy:%s" % self.x
1697 def __repr__(self):
1698 return "Proxy(%r)" % self.x
1699 def __contains__(self, value):
1700 return value in self.x
1701 p0 = Proxy(0)
1702 p1 = Proxy(1)
1703 p_1 = Proxy(-1)
1704 verify(not p0)
1705 verify(not not p1)
Guido van Rossum45704552001-10-08 16:35:45 +00001706 vereq(hash(p0), hash(0))
1707 vereq(p0, p0)
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001708 verify(p0 != p1)
1709 verify(not p0 != p0)
Guido van Rossum45704552001-10-08 16:35:45 +00001710 vereq(not p0, p1)
1711 vereq(cmp(p0, p1), -1)
1712 vereq(cmp(p0, p0), 0)
1713 vereq(cmp(p0, p_1), 1)
1714 vereq(str(p0), "Proxy:0")
1715 vereq(repr(p0), "Proxy(0)")
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001716 p10 = Proxy(range(10))
1717 verify(-1 not in p10)
1718 for i in range(10):
1719 verify(i in p10)
1720 verify(10 not in p10)
1721 # Test overridden behavior for dynamic classes
1722 class DProxy(object):
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001723 def __init__(self, x):
1724 self.x = x
1725 def __nonzero__(self):
1726 return not not self.x
1727 def __hash__(self):
1728 return hash(self.x)
1729 def __eq__(self, other):
1730 return self.x == other
1731 def __ne__(self, other):
1732 return self.x != other
1733 def __cmp__(self, other):
1734 return cmp(self.x, other.x)
1735 def __str__(self):
1736 return "DProxy:%s" % self.x
1737 def __repr__(self):
1738 return "DProxy(%r)" % self.x
1739 def __contains__(self, value):
1740 return value in self.x
1741 p0 = DProxy(0)
1742 p1 = DProxy(1)
1743 p_1 = DProxy(-1)
1744 verify(not p0)
1745 verify(not not p1)
Guido van Rossum45704552001-10-08 16:35:45 +00001746 vereq(hash(p0), hash(0))
1747 vereq(p0, p0)
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001748 verify(p0 != p1)
1749 verify(not p0 != p0)
Guido van Rossum45704552001-10-08 16:35:45 +00001750 vereq(not p0, p1)
1751 vereq(cmp(p0, p1), -1)
1752 vereq(cmp(p0, p0), 0)
1753 vereq(cmp(p0, p_1), 1)
1754 vereq(str(p0), "DProxy:0")
1755 vereq(repr(p0), "DProxy(0)")
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001756 p10 = DProxy(range(10))
1757 verify(-1 not in p10)
1758 for i in range(10):
1759 verify(i in p10)
1760 verify(10 not in p10)
Guido van Rossum47b9ff62006-08-24 00:41:19 +00001761## # Safety test for __cmp__
1762## def unsafecmp(a, b):
1763## try:
1764## a.__class__.__cmp__(a, b)
1765## except TypeError:
1766## pass
1767## else:
1768## raise TestFailed, "shouldn't allow %s.__cmp__(%r, %r)" % (
1769## a.__class__, a, b)
1770## unsafecmp(u"123", "123")
1771## unsafecmp("123", u"123")
1772## unsafecmp(1, 1.0)
1773## unsafecmp(1.0, 1)
1774## unsafecmp(1, 1L)
1775## unsafecmp(1L, 1)
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00001776
Neal Norwitz1a997502003-01-13 20:13:12 +00001777 class Letter(str):
1778 def __new__(cls, letter):
1779 if letter == 'EPS':
1780 return str.__new__(cls)
1781 return str.__new__(cls, letter)
1782 def __str__(self):
1783 if not self:
1784 return 'EPS'
Tim Petersf2715e02003-02-19 02:35:07 +00001785 return self
Neal Norwitz1a997502003-01-13 20:13:12 +00001786
1787 # sys.stdout needs to be the original to trigger the recursion bug
1788 import sys
1789 test_stdout = sys.stdout
1790 sys.stdout = get_original_stdout()
1791 try:
1792 # nothing should actually be printed, this should raise an exception
1793 print Letter('w')
1794 except RuntimeError:
1795 pass
1796 else:
1797 raise TestFailed, "expected a RuntimeError for print recursion"
1798 sys.stdout = test_stdout
1799
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00001800def weakrefs():
1801 if verbose: print "Testing weak references..."
1802 import weakref
1803 class C(object):
1804 pass
1805 c = C()
1806 r = weakref.ref(c)
1807 verify(r() is c)
1808 del c
1809 verify(r() is None)
1810 del r
1811 class NoWeak(object):
1812 __slots__ = ['foo']
1813 no = NoWeak()
1814 try:
1815 weakref.ref(no)
1816 except TypeError, msg:
Fred Drake4bf018b2001-10-22 21:45:25 +00001817 verify(str(msg).find("weak reference") >= 0)
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00001818 else:
1819 verify(0, "weakref.ref(no) should be illegal")
1820 class Weak(object):
1821 __slots__ = ['foo', '__weakref__']
1822 yes = Weak()
1823 r = weakref.ref(yes)
1824 verify(r() is yes)
1825 del yes
1826 verify(r() is None)
1827 del r
1828
Guido van Rossum8bce4ac2001-09-06 21:56:42 +00001829def properties():
1830 if verbose: print "Testing property..."
Guido van Rossum76f0cb82001-08-24 15:24:24 +00001831 class C(object):
1832 def getx(self):
1833 return self.__x
1834 def setx(self, value):
1835 self.__x = value
1836 def delx(self):
1837 del self.__x
Tim Peters66c1a522001-09-24 21:17:50 +00001838 x = property(getx, setx, delx, doc="I'm the x property.")
Guido van Rossum76f0cb82001-08-24 15:24:24 +00001839 a = C()
1840 verify(not hasattr(a, "x"))
1841 a.x = 42
Guido van Rossum45704552001-10-08 16:35:45 +00001842 vereq(a._C__x, 42)
1843 vereq(a.x, 42)
Guido van Rossum76f0cb82001-08-24 15:24:24 +00001844 del a.x
1845 verify(not hasattr(a, "x"))
1846 verify(not hasattr(a, "_C__x"))
1847 C.x.__set__(a, 100)
Guido van Rossum45704552001-10-08 16:35:45 +00001848 vereq(C.x.__get__(a), 100)
Guido van Rossum0dbab4c52002-08-01 14:39:25 +00001849 C.x.__delete__(a)
1850 verify(not hasattr(a, "x"))
Guido van Rossum76f0cb82001-08-24 15:24:24 +00001851
Tim Peters66c1a522001-09-24 21:17:50 +00001852 raw = C.__dict__['x']
1853 verify(isinstance(raw, property))
1854
1855 attrs = dir(raw)
1856 verify("__doc__" in attrs)
1857 verify("fget" in attrs)
1858 verify("fset" in attrs)
1859 verify("fdel" in attrs)
1860
Guido van Rossum45704552001-10-08 16:35:45 +00001861 vereq(raw.__doc__, "I'm the x property.")
Tim Peters66c1a522001-09-24 21:17:50 +00001862 verify(raw.fget is C.__dict__['getx'])
1863 verify(raw.fset is C.__dict__['setx'])
1864 verify(raw.fdel is C.__dict__['delx'])
1865
1866 for attr in "__doc__", "fget", "fset", "fdel":
1867 try:
1868 setattr(raw, attr, 42)
1869 except TypeError, msg:
1870 if str(msg).find('readonly') < 0:
1871 raise TestFailed("when setting readonly attr %r on a "
1872 "property, got unexpected TypeError "
1873 "msg %r" % (attr, str(msg)))
1874 else:
1875 raise TestFailed("expected TypeError from trying to set "
1876 "readonly %r attr on a property" % attr)
1877
Neal Norwitz673cd822002-10-18 16:33:13 +00001878 class D(object):
1879 __getitem__ = property(lambda s: 1/0)
1880
1881 d = D()
1882 try:
1883 for i in d:
1884 str(i)
1885 except ZeroDivisionError:
1886 pass
1887 else:
1888 raise TestFailed, "expected ZeroDivisionError from bad property"
1889
Georg Brandl533ff6f2006-03-08 18:09:27 +00001890 class E(object):
1891 def getter(self):
1892 "getter method"
1893 return 0
1894 def setter(self, value):
1895 "setter method"
1896 pass
1897 prop = property(getter)
1898 vereq(prop.__doc__, "getter method")
1899 prop2 = property(fset=setter)
1900 vereq(prop2.__doc__, None)
1901
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001902 # this segfaulted in 2.5b2
1903 try:
1904 import _testcapi
1905 except ImportError:
1906 pass
1907 else:
1908 class X(object):
1909 p = property(_testcapi.test_with_docstring)
1910
1911
Guido van Rossumc4a18802001-08-24 16:55:27 +00001912def supers():
Guido van Rossum9881fc12001-08-24 17:07:20 +00001913 if verbose: print "Testing super..."
Guido van Rossumc4a18802001-08-24 16:55:27 +00001914
1915 class A(object):
1916 def meth(self, a):
1917 return "A(%r)" % a
1918
Guido van Rossum45704552001-10-08 16:35:45 +00001919 vereq(A().meth(1), "A(1)")
Guido van Rossumc4a18802001-08-24 16:55:27 +00001920
1921 class B(A):
1922 def __init__(self):
1923 self.__super = super(B, self)
1924 def meth(self, a):
1925 return "B(%r)" % a + self.__super.meth(a)
1926
Guido van Rossum45704552001-10-08 16:35:45 +00001927 vereq(B().meth(2), "B(2)A(2)")
Guido van Rossumc4a18802001-08-24 16:55:27 +00001928
1929 class C(A):
Guido van Rossumc4a18802001-08-24 16:55:27 +00001930 def meth(self, a):
1931 return "C(%r)" % a + self.__super.meth(a)
1932 C._C__super = super(C)
1933
Guido van Rossum45704552001-10-08 16:35:45 +00001934 vereq(C().meth(3), "C(3)A(3)")
Guido van Rossumc4a18802001-08-24 16:55:27 +00001935
1936 class D(C, B):
1937 def meth(self, a):
1938 return "D(%r)" % a + super(D, self).meth(a)
1939
Guido van Rossum5b443c62001-12-03 15:38:28 +00001940 vereq(D().meth(4), "D(4)C(4)B(4)A(4)")
1941
1942 # Test for subclassing super
1943
1944 class mysuper(super):
1945 def __init__(self, *args):
1946 return super(mysuper, self).__init__(*args)
1947
1948 class E(D):
1949 def meth(self, a):
1950 return "E(%r)" % a + mysuper(E, self).meth(a)
1951
1952 vereq(E().meth(5), "E(5)D(5)C(5)B(5)A(5)")
1953
1954 class F(E):
1955 def meth(self, a):
Guido van Rossuma4541a32003-04-16 20:02:22 +00001956 s = self.__super # == mysuper(F, self)
Guido van Rossum5b443c62001-12-03 15:38:28 +00001957 return "F(%r)[%s]" % (a, s.__class__.__name__) + s.meth(a)
1958 F._F__super = mysuper(F)
1959
1960 vereq(F().meth(6), "F(6)[mysuper]E(6)D(6)C(6)B(6)A(6)")
1961
1962 # Make sure certain errors are raised
1963
1964 try:
1965 super(D, 42)
1966 except TypeError:
1967 pass
1968 else:
1969 raise TestFailed, "shouldn't allow super(D, 42)"
1970
1971 try:
1972 super(D, C())
1973 except TypeError:
1974 pass
1975 else:
1976 raise TestFailed, "shouldn't allow super(D, C())"
1977
1978 try:
1979 super(D).__get__(12)
1980 except TypeError:
1981 pass
1982 else:
1983 raise TestFailed, "shouldn't allow super(D).__get__(12)"
1984
1985 try:
1986 super(D).__get__(C())
1987 except TypeError:
1988 pass
1989 else:
1990 raise TestFailed, "shouldn't allow super(D).__get__(C())"
Guido van Rossumc4a18802001-08-24 16:55:27 +00001991
Guido van Rossuma4541a32003-04-16 20:02:22 +00001992 # Make sure data descriptors can be overridden and accessed via super
1993 # (new feature in Python 2.3)
1994
1995 class DDbase(object):
1996 def getx(self): return 42
1997 x = property(getx)
1998
1999 class DDsub(DDbase):
2000 def getx(self): return "hello"
2001 x = property(getx)
2002
2003 dd = DDsub()
2004 vereq(dd.x, "hello")
2005 vereq(super(DDsub, dd).x, 42)
2006
Phillip J. Eby91a968a2004-03-25 02:19:34 +00002007 # Ensure that super() lookup of descriptor from classmethod
2008 # works (SF ID# 743627)
2009
2010 class Base(object):
2011 aProp = property(lambda self: "foo")
2012
2013 class Sub(Base):
Guido van Rossum5a8a0372005-01-16 00:25:31 +00002014 @classmethod
Phillip J. Eby91a968a2004-03-25 02:19:34 +00002015 def test(klass):
2016 return super(Sub,klass).aProp
Phillip J. Eby91a968a2004-03-25 02:19:34 +00002017
2018 veris(Sub.test(), Base.aProp)
2019
2020
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002021def inherits():
2022 if verbose: print "Testing inheritance from basic types..."
2023
2024 class hexint(int):
2025 def __repr__(self):
2026 return hex(self)
2027 def __add__(self, other):
2028 return hexint(int.__add__(self, other))
2029 # (Note that overriding __radd__ doesn't work,
2030 # because the int type gets first dibs.)
Guido van Rossum45704552001-10-08 16:35:45 +00002031 vereq(repr(hexint(7) + 9), "0x10")
2032 vereq(repr(hexint(1000) + 7), "0x3ef")
Tim Peters64b5ce32001-09-10 20:52:51 +00002033 a = hexint(12345)
Guido van Rossum45704552001-10-08 16:35:45 +00002034 vereq(a, 12345)
2035 vereq(int(a), 12345)
Tim Peters64b5ce32001-09-10 20:52:51 +00002036 verify(int(a).__class__ is int)
Guido van Rossum45704552001-10-08 16:35:45 +00002037 vereq(hash(a), hash(12345))
Tim Peters73a1dfe2001-09-11 21:44:14 +00002038 verify((+a).__class__ is int)
2039 verify((a >> 0).__class__ is int)
2040 verify((a << 0).__class__ is int)
2041 verify((hexint(0) << 12).__class__ is int)
2042 verify((hexint(0) >> 12).__class__ is int)
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002043
2044 class octlong(long):
2045 __slots__ = []
2046 def __str__(self):
2047 s = oct(self)
2048 if s[-1] == 'L':
2049 s = s[:-1]
2050 return s
2051 def __add__(self, other):
2052 return self.__class__(super(octlong, self).__add__(other))
2053 __radd__ = __add__
Guido van Rossum45704552001-10-08 16:35:45 +00002054 vereq(str(octlong(3) + 5), "010")
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002055 # (Note that overriding __radd__ here only seems to work
2056 # because the example uses a short int left argument.)
Guido van Rossum45704552001-10-08 16:35:45 +00002057 vereq(str(5 + octlong(3000)), "05675")
Tim Peters64b5ce32001-09-10 20:52:51 +00002058 a = octlong(12345)
Guido van Rossum45704552001-10-08 16:35:45 +00002059 vereq(a, 12345L)
2060 vereq(long(a), 12345L)
2061 vereq(hash(a), hash(12345L))
Tim Peters64b5ce32001-09-10 20:52:51 +00002062 verify(long(a).__class__ is long)
Tim Peters69c2de32001-09-11 22:31:33 +00002063 verify((+a).__class__ is long)
2064 verify((-a).__class__ is long)
2065 verify((-octlong(0)).__class__ is long)
2066 verify((a >> 0).__class__ is long)
2067 verify((a << 0).__class__ is long)
2068 verify((a - 0).__class__ is long)
2069 verify((a * 1).__class__ is long)
2070 verify((a ** 1).__class__ is long)
2071 verify((a // 1).__class__ is long)
2072 verify((1 * a).__class__ is long)
2073 verify((a | 0).__class__ is long)
2074 verify((a ^ 0).__class__ is long)
2075 verify((a & -1L).__class__ is long)
2076 verify((octlong(0) << 12).__class__ is long)
2077 verify((octlong(0) >> 12).__class__ is long)
2078 verify(abs(octlong(0)).__class__ is long)
2079
2080 # Because octlong overrides __add__, we can't check the absence of +0
2081 # optimizations using octlong.
2082 class longclone(long):
2083 pass
2084 a = longclone(1)
2085 verify((a + 0).__class__ is long)
2086 verify((0 + a).__class__ is long)
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002087
Guido van Rossum2eb0b872002-03-01 22:24:49 +00002088 # Check that negative clones don't segfault
2089 a = longclone(-1)
2090 vereq(a.__dict__, {})
Tim Peters5329cdb2002-03-02 04:18:04 +00002091 vereq(long(a), -1) # verify PyNumber_Long() copies the sign bit
Guido van Rossum2eb0b872002-03-01 22:24:49 +00002092
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002093 class precfloat(float):
2094 __slots__ = ['prec']
2095 def __init__(self, value=0.0, prec=12):
2096 self.prec = int(prec)
2097 float.__init__(value)
2098 def __repr__(self):
2099 return "%.*g" % (self.prec, self)
Guido van Rossum45704552001-10-08 16:35:45 +00002100 vereq(repr(precfloat(1.1)), "1.1")
Tim Peters64b5ce32001-09-10 20:52:51 +00002101 a = precfloat(12345)
Guido van Rossum45704552001-10-08 16:35:45 +00002102 vereq(a, 12345.0)
2103 vereq(float(a), 12345.0)
Tim Peters7a50f252001-09-10 21:28:20 +00002104 verify(float(a).__class__ is float)
Guido van Rossum45704552001-10-08 16:35:45 +00002105 vereq(hash(a), hash(12345.0))
Tim Peters0280cf72001-09-11 21:53:35 +00002106 verify((+a).__class__ is float)
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002107
Tim Peters2400fa42001-09-12 19:12:49 +00002108 class madcomplex(complex):
2109 def __repr__(self):
2110 return "%.17gj%+.17g" % (self.imag, self.real)
2111 a = madcomplex(-3, 4)
Guido van Rossum45704552001-10-08 16:35:45 +00002112 vereq(repr(a), "4j-3")
Tim Peters2400fa42001-09-12 19:12:49 +00002113 base = complex(-3, 4)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002114 veris(base.__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002115 vereq(a, base)
2116 vereq(complex(a), base)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002117 veris(complex(a).__class__, complex)
Tim Peters2400fa42001-09-12 19:12:49 +00002118 a = madcomplex(a) # just trying another form of the constructor
Guido van Rossum45704552001-10-08 16:35:45 +00002119 vereq(repr(a), "4j-3")
2120 vereq(a, base)
2121 vereq(complex(a), base)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002122 veris(complex(a).__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002123 vereq(hash(a), hash(base))
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002124 veris((+a).__class__, complex)
2125 veris((a + 0).__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002126 vereq(a + 0, base)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002127 veris((a - 0).__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002128 vereq(a - 0, base)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002129 veris((a * 1).__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002130 vereq(a * 1, base)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002131 veris((a / 1).__class__, complex)
Guido van Rossum45704552001-10-08 16:35:45 +00002132 vereq(a / 1, base)
Tim Peters2400fa42001-09-12 19:12:49 +00002133
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002134 class madtuple(tuple):
2135 _rev = None
2136 def rev(self):
2137 if self._rev is not None:
2138 return self._rev
2139 L = list(self)
2140 L.reverse()
2141 self._rev = self.__class__(L)
2142 return self._rev
2143 a = madtuple((1,2,3,4,5,6,7,8,9,0))
Guido van Rossum45704552001-10-08 16:35:45 +00002144 vereq(a, (1,2,3,4,5,6,7,8,9,0))
2145 vereq(a.rev(), madtuple((0,9,8,7,6,5,4,3,2,1)))
2146 vereq(a.rev().rev(), madtuple((1,2,3,4,5,6,7,8,9,0)))
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002147 for i in range(512):
2148 t = madtuple(range(i))
2149 u = t.rev()
2150 v = u.rev()
Guido van Rossum45704552001-10-08 16:35:45 +00002151 vereq(v, t)
Tim Peters64b5ce32001-09-10 20:52:51 +00002152 a = madtuple((1,2,3,4,5))
Guido van Rossum45704552001-10-08 16:35:45 +00002153 vereq(tuple(a), (1,2,3,4,5))
Tim Peters4c3a0a32001-09-10 23:37:46 +00002154 verify(tuple(a).__class__ is tuple)
Guido van Rossum45704552001-10-08 16:35:45 +00002155 vereq(hash(a), hash((1,2,3,4,5)))
Tim Peters7b07a412001-09-11 19:48:03 +00002156 verify(a[:].__class__ is tuple)
2157 verify((a * 1).__class__ is tuple)
2158 verify((a * 0).__class__ is tuple)
2159 verify((a + ()).__class__ is tuple)
Tim Peters64b5ce32001-09-10 20:52:51 +00002160 a = madtuple(())
Guido van Rossum45704552001-10-08 16:35:45 +00002161 vereq(tuple(a), ())
Guido van Rossum779ce4a2001-09-11 14:02:22 +00002162 verify(tuple(a).__class__ is tuple)
Tim Peters7b07a412001-09-11 19:48:03 +00002163 verify((a + a).__class__ is tuple)
2164 verify((a * 0).__class__ is tuple)
2165 verify((a * 1).__class__ is tuple)
2166 verify((a * 2).__class__ is tuple)
2167 verify(a[:].__class__ is tuple)
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002168
2169 class madstring(str):
2170 _rev = None
2171 def rev(self):
2172 if self._rev is not None:
2173 return self._rev
2174 L = list(self)
2175 L.reverse()
2176 self._rev = self.__class__("".join(L))
2177 return self._rev
2178 s = madstring("abcdefghijklmnopqrstuvwxyz")
Guido van Rossum45704552001-10-08 16:35:45 +00002179 vereq(s, "abcdefghijklmnopqrstuvwxyz")
2180 vereq(s.rev(), madstring("zyxwvutsrqponmlkjihgfedcba"))
2181 vereq(s.rev().rev(), madstring("abcdefghijklmnopqrstuvwxyz"))
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002182 for i in range(256):
2183 s = madstring("".join(map(chr, range(i))))
2184 t = s.rev()
2185 u = t.rev()
Guido van Rossum45704552001-10-08 16:35:45 +00002186 vereq(u, s)
Tim Peters64b5ce32001-09-10 20:52:51 +00002187 s = madstring("12345")
Guido van Rossum45704552001-10-08 16:35:45 +00002188 vereq(str(s), "12345")
Tim Peters5a49ade2001-09-11 01:41:59 +00002189 verify(str(s).__class__ is str)
Guido van Rossumcaa9f432001-08-30 20:06:08 +00002190
Tim Peters8fa5dd02001-09-12 02:18:30 +00002191 base = "\x00" * 5
2192 s = madstring(base)
Guido van Rossum45704552001-10-08 16:35:45 +00002193 vereq(s, base)
2194 vereq(str(s), base)
Tim Petersc636f562001-09-11 01:52:02 +00002195 verify(str(s).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002196 vereq(hash(s), hash(base))
2197 vereq({s: 1}[base], 1)
2198 vereq({base: 1}[s], 1)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002199 verify((s + "").__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002200 vereq(s + "", base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002201 verify(("" + s).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002202 vereq("" + s, base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002203 verify((s * 0).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002204 vereq(s * 0, "")
Tim Peters8fa5dd02001-09-12 02:18:30 +00002205 verify((s * 1).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002206 vereq(s * 1, base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002207 verify((s * 2).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002208 vereq(s * 2, base + base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002209 verify(s[:].__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002210 vereq(s[:], base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002211 verify(s[0:0].__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002212 vereq(s[0:0], "")
Tim Peters8fa5dd02001-09-12 02:18:30 +00002213 verify(s.strip().__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002214 vereq(s.strip(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002215 verify(s.lstrip().__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002216 vereq(s.lstrip(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002217 verify(s.rstrip().__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002218 vereq(s.rstrip(), base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002219 identitytab = ''.join([chr(i) for i in range(256)])
2220 verify(s.translate(identitytab).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002221 vereq(s.translate(identitytab), base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002222 verify(s.translate(identitytab, "x").__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002223 vereq(s.translate(identitytab, "x"), base)
2224 vereq(s.translate(identitytab, "\x00"), "")
Tim Peters8fa5dd02001-09-12 02:18:30 +00002225 verify(s.replace("x", "x").__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002226 vereq(s.replace("x", "x"), base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002227 verify(s.ljust(len(s)).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002228 vereq(s.ljust(len(s)), base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002229 verify(s.rjust(len(s)).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002230 vereq(s.rjust(len(s)), base)
Tim Peters8fa5dd02001-09-12 02:18:30 +00002231 verify(s.center(len(s)).__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002232 vereq(s.center(len(s)), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002233 verify(s.lower().__class__ is str)
Guido van Rossum45704552001-10-08 16:35:45 +00002234 vereq(s.lower(), base)
Tim Petersc636f562001-09-11 01:52:02 +00002235
Guido van Rossum91ee7982001-08-30 20:52:40 +00002236 class madunicode(unicode):
2237 _rev = None
2238 def rev(self):
2239 if self._rev is not None:
2240 return self._rev
2241 L = list(self)
2242 L.reverse()
2243 self._rev = self.__class__(u"".join(L))
2244 return self._rev
2245 u = madunicode("ABCDEF")
Guido van Rossum45704552001-10-08 16:35:45 +00002246 vereq(u, u"ABCDEF")
2247 vereq(u.rev(), madunicode(u"FEDCBA"))
2248 vereq(u.rev().rev(), madunicode(u"ABCDEF"))
Tim Peters7a29bd52001-09-12 03:03:31 +00002249 base = u"12345"
2250 u = madunicode(base)
Guido van Rossum45704552001-10-08 16:35:45 +00002251 vereq(unicode(u), base)
Tim Peters78e0fc72001-09-11 03:07:38 +00002252 verify(unicode(u).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002253 vereq(hash(u), hash(base))
2254 vereq({u: 1}[base], 1)
2255 vereq({base: 1}[u], 1)
Tim Peters7a29bd52001-09-12 03:03:31 +00002256 verify(u.strip().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002257 vereq(u.strip(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002258 verify(u.lstrip().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002259 vereq(u.lstrip(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002260 verify(u.rstrip().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002261 vereq(u.rstrip(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002262 verify(u.replace(u"x", u"x").__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002263 vereq(u.replace(u"x", u"x"), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002264 verify(u.replace(u"xy", u"xy").__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002265 vereq(u.replace(u"xy", u"xy"), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002266 verify(u.center(len(u)).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002267 vereq(u.center(len(u)), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002268 verify(u.ljust(len(u)).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002269 vereq(u.ljust(len(u)), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002270 verify(u.rjust(len(u)).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002271 vereq(u.rjust(len(u)), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002272 verify(u.lower().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002273 vereq(u.lower(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002274 verify(u.upper().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002275 vereq(u.upper(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002276 verify(u.capitalize().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002277 vereq(u.capitalize(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002278 verify(u.title().__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002279 vereq(u.title(), base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002280 verify((u + u"").__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002281 vereq(u + u"", base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002282 verify((u"" + u).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002283 vereq(u"" + u, base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002284 verify((u * 0).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002285 vereq(u * 0, u"")
Tim Peters7a29bd52001-09-12 03:03:31 +00002286 verify((u * 1).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002287 vereq(u * 1, base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002288 verify((u * 2).__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002289 vereq(u * 2, base + base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002290 verify(u[:].__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002291 vereq(u[:], base)
Tim Peters7a29bd52001-09-12 03:03:31 +00002292 verify(u[0:0].__class__ is unicode)
Guido van Rossum45704552001-10-08 16:35:45 +00002293 vereq(u[0:0], u"")
Guido van Rossum91ee7982001-08-30 20:52:40 +00002294
Guido van Rossum12b22ff2001-10-09 20:36:44 +00002295 class sublist(list):
2296 pass
2297 a = sublist(range(5))
2298 vereq(a, range(5))
2299 a.append("hello")
2300 vereq(a, range(5) + ["hello"])
2301 a[5] = 5
2302 vereq(a, range(6))
2303 a.extend(range(6, 20))
2304 vereq(a, range(20))
2305 a[-5:] = []
2306 vereq(a, range(15))
2307 del a[10:15]
2308 vereq(len(a), 10)
2309 vereq(a, range(10))
2310 vereq(list(a), range(10))
2311 vereq(a[0], 0)
2312 vereq(a[9], 9)
2313 vereq(a[-10], 0)
2314 vereq(a[-1], 9)
2315 vereq(a[:5], range(5))
2316
Tim Peters59c9a642001-09-13 05:38:56 +00002317 class CountedInput(file):
2318 """Counts lines read by self.readline().
2319
2320 self.lineno is the 0-based ordinal of the last line read, up to
2321 a maximum of one greater than the number of lines in the file.
2322
2323 self.ateof is true if and only if the final "" line has been read,
2324 at which point self.lineno stops incrementing, and further calls
2325 to readline() continue to return "".
2326 """
2327
2328 lineno = 0
2329 ateof = 0
2330 def readline(self):
2331 if self.ateof:
2332 return ""
2333 s = file.readline(self)
2334 # Next line works too.
2335 # s = super(CountedInput, self).readline()
2336 self.lineno += 1
2337 if s == "":
2338 self.ateof = 1
2339 return s
2340
Alex Martelli01c77c62006-08-24 02:58:11 +00002341 f = open(name=TESTFN, mode='w')
Tim Peters59c9a642001-09-13 05:38:56 +00002342 lines = ['a\n', 'b\n', 'c\n']
2343 try:
2344 f.writelines(lines)
2345 f.close()
2346 f = CountedInput(TESTFN)
2347 for (i, expected) in zip(range(1, 5) + [4], lines + 2 * [""]):
2348 got = f.readline()
Guido van Rossum45704552001-10-08 16:35:45 +00002349 vereq(expected, got)
2350 vereq(f.lineno, i)
2351 vereq(f.ateof, (i > len(lines)))
Tim Peters59c9a642001-09-13 05:38:56 +00002352 f.close()
2353 finally:
2354 try:
2355 f.close()
2356 except:
2357 pass
2358 try:
2359 import os
2360 os.unlink(TESTFN)
2361 except:
2362 pass
2363
Tim Peters808b94e2001-09-13 19:33:07 +00002364def keywords():
2365 if verbose:
2366 print "Testing keyword args to basic type constructors ..."
Guido van Rossum45704552001-10-08 16:35:45 +00002367 vereq(int(x=1), 1)
2368 vereq(float(x=2), 2.0)
2369 vereq(long(x=3), 3L)
2370 vereq(complex(imag=42, real=666), complex(666, 42))
2371 vereq(str(object=500), '500')
2372 vereq(unicode(string='abc', errors='strict'), u'abc')
2373 vereq(tuple(sequence=range(3)), (0, 1, 2))
2374 vereq(list(sequence=(0, 1, 2)), range(3))
Just van Rossuma797d812002-11-23 09:45:04 +00002375 # note: as of Python 2.3, dict() no longer has an "items" keyword arg
Tim Peters808b94e2001-09-13 19:33:07 +00002376
2377 for constructor in (int, float, long, complex, str, unicode,
Just van Rossuma797d812002-11-23 09:45:04 +00002378 tuple, list, file):
Tim Peters808b94e2001-09-13 19:33:07 +00002379 try:
2380 constructor(bogus_keyword_arg=1)
2381 except TypeError:
2382 pass
2383 else:
2384 raise TestFailed("expected TypeError from bogus keyword "
2385 "argument to %r" % constructor)
Tim Peters561f8992001-09-13 19:36:36 +00002386
Tim Peters8fa45672001-09-13 21:01:29 +00002387def restricted():
Guido van Rossum4f374182003-01-06 16:03:43 +00002388 # XXX This test is disabled because rexec is not deemed safe
2389 return
Tim Peters8fa45672001-09-13 21:01:29 +00002390 import rexec
2391 if verbose:
2392 print "Testing interaction with restricted execution ..."
2393
2394 sandbox = rexec.RExec()
2395
2396 code1 = """f = open(%r, 'w')""" % TESTFN
Alex Martelli01c77c62006-08-24 02:58:11 +00002397 code2 = """f = open(%r, 'w')""" % TESTFN
Tim Peters8fa45672001-09-13 21:01:29 +00002398 code3 = """\
2399f = open(%r)
2400t = type(f) # a sneaky way to get the file() constructor
2401f.close()
2402f = t(%r, 'w') # rexec can't catch this by itself
2403""" % (TESTFN, TESTFN)
2404
2405 f = open(TESTFN, 'w') # Create the file so code3 can find it.
2406 f.close()
2407
2408 try:
2409 for code in code1, code2, code3:
2410 try:
2411 sandbox.r_exec(code)
2412 except IOError, msg:
2413 if str(msg).find("restricted") >= 0:
2414 outcome = "OK"
2415 else:
2416 outcome = "got an exception, but not an expected one"
2417 else:
2418 outcome = "expected a restricted-execution exception"
2419
2420 if outcome != "OK":
2421 raise TestFailed("%s, in %r" % (outcome, code))
2422
2423 finally:
2424 try:
2425 import os
2426 os.unlink(TESTFN)
2427 except:
2428 pass
2429
Tim Peters0ab085c2001-09-14 00:25:33 +00002430def str_subclass_as_dict_key():
2431 if verbose:
2432 print "Testing a str subclass used as dict key .."
2433
2434 class cistr(str):
2435 """Sublcass of str that computes __eq__ case-insensitively.
2436
2437 Also computes a hash code of the string in canonical form.
2438 """
2439
2440 def __init__(self, value):
2441 self.canonical = value.lower()
2442 self.hashcode = hash(self.canonical)
2443
2444 def __eq__(self, other):
2445 if not isinstance(other, cistr):
2446 other = cistr(other)
2447 return self.canonical == other.canonical
2448
2449 def __hash__(self):
2450 return self.hashcode
2451
Guido van Rossum45704552001-10-08 16:35:45 +00002452 vereq(cistr('ABC'), 'abc')
2453 vereq('aBc', cistr('ABC'))
2454 vereq(str(cistr('ABC')), 'ABC')
Tim Peters0ab085c2001-09-14 00:25:33 +00002455
2456 d = {cistr('one'): 1, cistr('two'): 2, cistr('tHree'): 3}
Guido van Rossum45704552001-10-08 16:35:45 +00002457 vereq(d[cistr('one')], 1)
2458 vereq(d[cistr('tWo')], 2)
2459 vereq(d[cistr('THrEE')], 3)
Tim Peters0ab085c2001-09-14 00:25:33 +00002460 verify(cistr('ONe') in d)
Guido van Rossum45704552001-10-08 16:35:45 +00002461 vereq(d.get(cistr('thrEE')), 3)
Tim Peters0ab085c2001-09-14 00:25:33 +00002462
Guido van Rossumab3b0342001-09-18 20:38:53 +00002463def classic_comparisons():
2464 if verbose: print "Testing classic comparisons..."
Guido van Rossum0639f592001-09-18 21:06:04 +00002465 class classic:
2466 pass
2467 for base in (classic, int, object):
Guido van Rossumab3b0342001-09-18 20:38:53 +00002468 if verbose: print " (base = %s)" % base
2469 class C(base):
2470 def __init__(self, value):
2471 self.value = int(value)
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002472 def __eq__(self, other):
Guido van Rossumab3b0342001-09-18 20:38:53 +00002473 if isinstance(other, C):
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002474 return self.value == other.value
Guido van Rossumab3b0342001-09-18 20:38:53 +00002475 if isinstance(other, int) or isinstance(other, long):
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002476 return self.value == other
Guido van Rossumab3b0342001-09-18 20:38:53 +00002477 return NotImplemented
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002478 def __ne__(self, other):
2479 if isinstance(other, C):
2480 return self.value != other.value
2481 if isinstance(other, int) or isinstance(other, long):
2482 return self.value != other
2483 return NotImplemented
2484 def __lt__(self, other):
2485 if isinstance(other, C):
2486 return self.value < other.value
2487 if isinstance(other, int) or isinstance(other, long):
2488 return self.value < other
2489 return NotImplemented
2490 def __le__(self, other):
2491 if isinstance(other, C):
2492 return self.value <= other.value
2493 if isinstance(other, int) or isinstance(other, long):
2494 return self.value <= other
2495 return NotImplemented
2496 def __gt__(self, other):
2497 if isinstance(other, C):
2498 return self.value > other.value
2499 if isinstance(other, int) or isinstance(other, long):
2500 return self.value > other
2501 return NotImplemented
2502 def __ge__(self, other):
2503 if isinstance(other, C):
2504 return self.value >= other.value
2505 if isinstance(other, int) or isinstance(other, long):
2506 return self.value >= other
2507 return NotImplemented
2508
Guido van Rossumab3b0342001-09-18 20:38:53 +00002509 c1 = C(1)
2510 c2 = C(2)
2511 c3 = C(3)
Guido van Rossum45704552001-10-08 16:35:45 +00002512 vereq(c1, 1)
Guido van Rossumab3b0342001-09-18 20:38:53 +00002513 c = {1: c1, 2: c2, 3: c3}
2514 for x in 1, 2, 3:
2515 for y in 1, 2, 3:
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002516 ##verify(cmp(c[x], c[y]) == cmp(x, y), "x=%d, y=%d" % (x, y))
Guido van Rossumab3b0342001-09-18 20:38:53 +00002517 for op in "<", "<=", "==", "!=", ">", ">=":
2518 verify(eval("c[x] %s c[y]" % op) == eval("x %s y" % op),
2519 "x=%d, y=%d" % (x, y))
Guido van Rossum47b9ff62006-08-24 00:41:19 +00002520 ##verify(cmp(c[x], y) == cmp(x, y), "x=%d, y=%d" % (x, y))
2521 ##verify(cmp(x, c[y]) == cmp(x, y), "x=%d, y=%d" % (x, y))
Guido van Rossumab3b0342001-09-18 20:38:53 +00002522
Guido van Rossum0639f592001-09-18 21:06:04 +00002523def rich_comparisons():
2524 if verbose:
2525 print "Testing rich comparisons..."
Guido van Rossum22056422001-09-24 17:52:04 +00002526 class Z(complex):
Guido van Rossum2f3ca6e2001-10-15 21:05:10 +00002527 pass
Guido van Rossum22056422001-09-24 17:52:04 +00002528 z = Z(1)
Guido van Rossum45704552001-10-08 16:35:45 +00002529 vereq(z, 1+0j)
2530 vereq(1+0j, z)
Guido van Rossum22056422001-09-24 17:52:04 +00002531 class ZZ(complex):
2532 def __eq__(self, other):
2533 try:
2534 return abs(self - other) <= 1e-6
2535 except:
2536 return NotImplemented
2537 zz = ZZ(1.0000003)
Guido van Rossum45704552001-10-08 16:35:45 +00002538 vereq(zz, 1+0j)
2539 vereq(1+0j, zz)
Tim Peters66c1a522001-09-24 21:17:50 +00002540
Guido van Rossum0639f592001-09-18 21:06:04 +00002541 class classic:
2542 pass
2543 for base in (classic, int, object, list):
2544 if verbose: print " (base = %s)" % base
2545 class C(base):
2546 def __init__(self, value):
2547 self.value = int(value)
2548 def __cmp__(self, other):
2549 raise TestFailed, "shouldn't call __cmp__"
2550 def __eq__(self, other):
2551 if isinstance(other, C):
2552 return self.value == other.value
2553 if isinstance(other, int) or isinstance(other, long):
2554 return self.value == other
2555 return NotImplemented
2556 def __ne__(self, other):
2557 if isinstance(other, C):
2558 return self.value != other.value
2559 if isinstance(other, int) or isinstance(other, long):
2560 return self.value != other
2561 return NotImplemented
2562 def __lt__(self, other):
2563 if isinstance(other, C):
2564 return self.value < other.value
2565 if isinstance(other, int) or isinstance(other, long):
2566 return self.value < other
2567 return NotImplemented
2568 def __le__(self, other):
2569 if isinstance(other, C):
2570 return self.value <= other.value
2571 if isinstance(other, int) or isinstance(other, long):
2572 return self.value <= other
2573 return NotImplemented
2574 def __gt__(self, other):
2575 if isinstance(other, C):
2576 return self.value > other.value
2577 if isinstance(other, int) or isinstance(other, long):
2578 return self.value > other
2579 return NotImplemented
2580 def __ge__(self, other):
2581 if isinstance(other, C):
2582 return self.value >= other.value
2583 if isinstance(other, int) or isinstance(other, long):
2584 return self.value >= other
2585 return NotImplemented
2586 c1 = C(1)
2587 c2 = C(2)
2588 c3 = C(3)
Guido van Rossum45704552001-10-08 16:35:45 +00002589 vereq(c1, 1)
Guido van Rossum0639f592001-09-18 21:06:04 +00002590 c = {1: c1, 2: c2, 3: c3}
2591 for x in 1, 2, 3:
2592 for y in 1, 2, 3:
2593 for op in "<", "<=", "==", "!=", ">", ">=":
2594 verify(eval("c[x] %s c[y]" % op) == eval("x %s y" % op),
2595 "x=%d, y=%d" % (x, y))
2596 verify(eval("c[x] %s y" % op) == eval("x %s y" % op),
2597 "x=%d, y=%d" % (x, y))
2598 verify(eval("x %s c[y]" % op) == eval("x %s y" % op),
2599 "x=%d, y=%d" % (x, y))
2600
Guido van Rossum8b9cc7e2001-09-20 21:49:53 +00002601def descrdoc():
2602 if verbose: print "Testing descriptor doc strings..."
2603 def check(descr, what):
Guido van Rossum45704552001-10-08 16:35:45 +00002604 vereq(descr.__doc__, what)
Guido van Rossum77f6a652002-04-03 22:41:51 +00002605 check(file.closed, "True if the file is closed") # getset descriptor
Guido van Rossum8b9cc7e2001-09-20 21:49:53 +00002606 check(file.name, "file name") # member descriptor
2607
Guido van Rossum5c294fb2001-09-25 03:43:42 +00002608def setclass():
2609 if verbose: print "Testing __class__ assignment..."
2610 class C(object): pass
2611 class D(object): pass
2612 class E(object): pass
2613 class F(D, E): pass
2614 for cls in C, D, E, F:
2615 for cls2 in C, D, E, F:
2616 x = cls()
2617 x.__class__ = cls2
2618 verify(x.__class__ is cls2)
2619 x.__class__ = cls
2620 verify(x.__class__ is cls)
2621 def cant(x, C):
2622 try:
2623 x.__class__ = C
2624 except TypeError:
2625 pass
2626 else:
2627 raise TestFailed, "shouldn't allow %r.__class__ = %r" % (x, C)
Guido van Rossumb6b89422002-04-15 01:03:30 +00002628 try:
2629 delattr(x, "__class__")
2630 except TypeError:
2631 pass
2632 else:
2633 raise TestFailed, "shouldn't allow del %r.__class__" % x
Guido van Rossum5c294fb2001-09-25 03:43:42 +00002634 cant(C(), list)
2635 cant(list(), C)
2636 cant(C(), 1)
2637 cant(C(), object)
2638 cant(object(), list)
2639 cant(list(), object)
Guido van Rossum40af8892002-08-10 05:42:07 +00002640 class Int(int): __slots__ = []
2641 cant(2, Int)
2642 cant(Int(), int)
2643 cant(True, int)
2644 cant(2, bool)
Neal Norwitz78ce6b12002-12-24 15:26:42 +00002645 o = object()
2646 cant(o, type(1))
2647 cant(o, type(None))
2648 del o
Guido van Rossum5c294fb2001-09-25 03:43:42 +00002649
Guido van Rossum6661be32001-10-26 04:26:12 +00002650def setdict():
2651 if verbose: print "Testing __dict__ assignment..."
2652 class C(object): pass
2653 a = C()
2654 a.__dict__ = {'b': 1}
2655 vereq(a.b, 1)
2656 def cant(x, dict):
2657 try:
2658 x.__dict__ = dict
Barry Warsawb180c062005-04-20 19:41:36 +00002659 except (AttributeError, TypeError):
Guido van Rossum6661be32001-10-26 04:26:12 +00002660 pass
2661 else:
2662 raise TestFailed, "shouldn't allow %r.__dict__ = %r" % (x, dict)
2663 cant(a, None)
2664 cant(a, [])
2665 cant(a, 1)
Guido van Rossumd331cb52001-12-05 19:46:42 +00002666 del a.__dict__ # Deleting __dict__ is allowed
Guido van Rossum6661be32001-10-26 04:26:12 +00002667 # Classes don't allow __dict__ assignment
2668 cant(C, {})
2669
Guido van Rossum3926a632001-09-25 16:25:58 +00002670def pickles():
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002671 if verbose:
2672 print "Testing pickling and copying new-style classes and objects..."
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00002673 import pickle
2674 try:
2675 import cPickle
2676 except ImportError:
2677 cPickle = None
Guido van Rossum3926a632001-09-25 16:25:58 +00002678
2679 def sorteditems(d):
2680 L = d.items()
2681 L.sort()
2682 return L
2683
2684 global C
2685 class C(object):
2686 def __init__(self, a, b):
2687 super(C, self).__init__()
2688 self.a = a
2689 self.b = b
2690 def __repr__(self):
2691 return "C(%r, %r)" % (self.a, self.b)
2692
2693 global C1
2694 class C1(list):
2695 def __new__(cls, a, b):
2696 return super(C1, cls).__new__(cls)
Guido van Rossumf6318592003-02-07 14:59:13 +00002697 def __getnewargs__(self):
2698 return (self.a, self.b)
Guido van Rossum3926a632001-09-25 16:25:58 +00002699 def __init__(self, a, b):
2700 self.a = a
2701 self.b = b
2702 def __repr__(self):
2703 return "C1(%r, %r)<%r>" % (self.a, self.b, list(self))
2704
2705 global C2
2706 class C2(int):
2707 def __new__(cls, a, b, val=0):
2708 return super(C2, cls).__new__(cls, val)
Guido van Rossumf6318592003-02-07 14:59:13 +00002709 def __getnewargs__(self):
2710 return (self.a, self.b, int(self))
Guido van Rossum3926a632001-09-25 16:25:58 +00002711 def __init__(self, a, b, val=0):
2712 self.a = a
2713 self.b = b
2714 def __repr__(self):
2715 return "C2(%r, %r)<%r>" % (self.a, self.b, int(self))
2716
Guido van Rossum90c45142001-11-24 21:07:01 +00002717 global C3
2718 class C3(object):
2719 def __init__(self, foo):
2720 self.foo = foo
2721 def __getstate__(self):
2722 return self.foo
2723 def __setstate__(self, foo):
2724 self.foo = foo
2725
2726 global C4classic, C4
2727 class C4classic: # classic
2728 pass
2729 class C4(C4classic, object): # mixed inheritance
2730 pass
2731
Guido van Rossum3926a632001-09-25 16:25:58 +00002732 for p in pickle, cPickle:
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00002733 if p is None:
2734 continue # cPickle not found -- skip it
Guido van Rossum3926a632001-09-25 16:25:58 +00002735 for bin in 0, 1:
Guido van Rossumfe1fd0e2001-10-02 19:58:32 +00002736 if verbose:
2737 print p.__name__, ["text", "binary"][bin]
Guido van Rossum3926a632001-09-25 16:25:58 +00002738
2739 for cls in C, C1, C2:
2740 s = p.dumps(cls, bin)
2741 cls2 = p.loads(s)
2742 verify(cls2 is cls)
2743
2744 a = C1(1, 2); a.append(42); a.append(24)
2745 b = C2("hello", "world", 42)
2746 s = p.dumps((a, b), bin)
2747 x, y = p.loads(s)
Guido van Rossum90c45142001-11-24 21:07:01 +00002748 vereq(x.__class__, a.__class__)
2749 vereq(sorteditems(x.__dict__), sorteditems(a.__dict__))
2750 vereq(y.__class__, b.__class__)
2751 vereq(sorteditems(y.__dict__), sorteditems(b.__dict__))
Walter Dörwald70a6b492004-02-12 17:35:32 +00002752 vereq(repr(x), repr(a))
2753 vereq(repr(y), repr(b))
Guido van Rossum3926a632001-09-25 16:25:58 +00002754 if verbose:
2755 print "a = x =", a
2756 print "b = y =", b
Guido van Rossum90c45142001-11-24 21:07:01 +00002757 # Test for __getstate__ and __setstate__ on new style class
2758 u = C3(42)
2759 s = p.dumps(u, bin)
2760 v = p.loads(s)
2761 veris(u.__class__, v.__class__)
2762 vereq(u.foo, v.foo)
2763 # Test for picklability of hybrid class
2764 u = C4()
2765 u.foo = 42
2766 s = p.dumps(u, bin)
2767 v = p.loads(s)
2768 veris(u.__class__, v.__class__)
2769 vereq(u.foo, v.foo)
Guido van Rossum3926a632001-09-25 16:25:58 +00002770
Guido van Rossumfe1fd0e2001-10-02 19:58:32 +00002771 # Testing copy.deepcopy()
2772 if verbose:
2773 print "deepcopy"
2774 import copy
2775 for cls in C, C1, C2:
2776 cls2 = copy.deepcopy(cls)
2777 verify(cls2 is cls)
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002778
Guido van Rossumfe1fd0e2001-10-02 19:58:32 +00002779 a = C1(1, 2); a.append(42); a.append(24)
2780 b = C2("hello", "world", 42)
2781 x, y = copy.deepcopy((a, b))
Guido van Rossum90c45142001-11-24 21:07:01 +00002782 vereq(x.__class__, a.__class__)
2783 vereq(sorteditems(x.__dict__), sorteditems(a.__dict__))
2784 vereq(y.__class__, b.__class__)
2785 vereq(sorteditems(y.__dict__), sorteditems(b.__dict__))
Walter Dörwald70a6b492004-02-12 17:35:32 +00002786 vereq(repr(x), repr(a))
2787 vereq(repr(y), repr(b))
Guido van Rossumfe1fd0e2001-10-02 19:58:32 +00002788 if verbose:
2789 print "a = x =", a
2790 print "b = y =", b
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002791
Guido van Rossum8c842552002-03-14 23:05:54 +00002792def pickleslots():
2793 if verbose: print "Testing pickling of classes with __slots__ ..."
Guido van Rossumbf12cdb2006-08-17 20:24:18 +00002794 import pickle, pickle as cPickle
Guido van Rossum8c842552002-03-14 23:05:54 +00002795 # Pickling of classes with __slots__ but without __getstate__ should fail
2796 global B, C, D, E
2797 class B(object):
2798 pass
2799 for base in [object, B]:
2800 class C(base):
2801 __slots__ = ['a']
2802 class D(C):
2803 pass
2804 try:
2805 pickle.dumps(C())
2806 except TypeError:
2807 pass
2808 else:
2809 raise TestFailed, "should fail: pickle C instance - %s" % base
2810 try:
2811 cPickle.dumps(C())
2812 except TypeError:
2813 pass
2814 else:
2815 raise TestFailed, "should fail: cPickle C instance - %s" % base
2816 try:
2817 pickle.dumps(C())
2818 except TypeError:
2819 pass
2820 else:
2821 raise TestFailed, "should fail: pickle D instance - %s" % base
2822 try:
2823 cPickle.dumps(D())
2824 except TypeError:
2825 pass
2826 else:
2827 raise TestFailed, "should fail: cPickle D instance - %s" % base
Guido van Rossum3f50cdc2003-02-10 21:31:27 +00002828 # Give C a nice generic __getstate__ and __setstate__
Guido van Rossum8c842552002-03-14 23:05:54 +00002829 class C(base):
2830 __slots__ = ['a']
2831 def __getstate__(self):
2832 try:
2833 d = self.__dict__.copy()
2834 except AttributeError:
2835 d = {}
Guido van Rossum3f50cdc2003-02-10 21:31:27 +00002836 for cls in self.__class__.__mro__:
2837 for sn in cls.__dict__.get('__slots__', ()):
2838 try:
2839 d[sn] = getattr(self, sn)
2840 except AttributeError:
2841 pass
Guido van Rossum8c842552002-03-14 23:05:54 +00002842 return d
2843 def __setstate__(self, d):
2844 for k, v in d.items():
2845 setattr(self, k, v)
2846 class D(C):
2847 pass
2848 # Now it should work
2849 x = C()
2850 y = pickle.loads(pickle.dumps(x))
2851 vereq(hasattr(y, 'a'), 0)
2852 y = cPickle.loads(cPickle.dumps(x))
2853 vereq(hasattr(y, 'a'), 0)
2854 x.a = 42
2855 y = pickle.loads(pickle.dumps(x))
2856 vereq(y.a, 42)
2857 y = cPickle.loads(cPickle.dumps(x))
2858 vereq(y.a, 42)
2859 x = D()
2860 x.a = 42
2861 x.b = 100
2862 y = pickle.loads(pickle.dumps(x))
2863 vereq(y.a + y.b, 142)
2864 y = cPickle.loads(cPickle.dumps(x))
2865 vereq(y.a + y.b, 142)
Guido van Rossum3f50cdc2003-02-10 21:31:27 +00002866 # A subclass that adds a slot should also work
Guido van Rossum8c842552002-03-14 23:05:54 +00002867 class E(C):
2868 __slots__ = ['b']
Guido van Rossum3f50cdc2003-02-10 21:31:27 +00002869 x = E()
2870 x.a = 42
2871 x.b = "foo"
2872 y = pickle.loads(pickle.dumps(x))
2873 vereq(y.a, x.a)
2874 vereq(y.b, x.b)
2875 y = cPickle.loads(cPickle.dumps(x))
2876 vereq(y.a, x.a)
2877 vereq(y.b, x.b)
Guido van Rossum8c842552002-03-14 23:05:54 +00002878
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002879def copies():
2880 if verbose: print "Testing copy.copy() and copy.deepcopy()..."
2881 import copy
2882 class C(object):
2883 pass
2884
2885 a = C()
2886 a.foo = 12
2887 b = copy.copy(a)
Guido van Rossum45704552001-10-08 16:35:45 +00002888 vereq(b.__dict__, a.__dict__)
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002889
2890 a.bar = [1,2,3]
2891 c = copy.copy(a)
Guido van Rossum45704552001-10-08 16:35:45 +00002892 vereq(c.bar, a.bar)
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002893 verify(c.bar is a.bar)
2894
2895 d = copy.deepcopy(a)
Guido van Rossum45704552001-10-08 16:35:45 +00002896 vereq(d.__dict__, a.__dict__)
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002897 a.bar.append(4)
Guido van Rossum45704552001-10-08 16:35:45 +00002898 vereq(d.bar, [1,2,3])
Guido van Rossum6cef6d52001-09-28 18:13:29 +00002899
Guido van Rossum4bb1e362001-09-28 23:49:48 +00002900def binopoverride():
2901 if verbose: print "Testing overrides of binary operations..."
2902 class I(int):
2903 def __repr__(self):
2904 return "I(%r)" % int(self)
2905 def __add__(self, other):
2906 return I(int(self) + int(other))
2907 __radd__ = __add__
2908 def __pow__(self, other, mod=None):
2909 if mod is None:
2910 return I(pow(int(self), int(other)))
2911 else:
2912 return I(pow(int(self), int(other), int(mod)))
2913 def __rpow__(self, other, mod=None):
2914 if mod is None:
2915 return I(pow(int(other), int(self), mod))
2916 else:
2917 return I(pow(int(other), int(self), int(mod)))
Tim Peters2f93e282001-10-04 05:27:00 +00002918
Walter Dörwald70a6b492004-02-12 17:35:32 +00002919 vereq(repr(I(1) + I(2)), "I(3)")
2920 vereq(repr(I(1) + 2), "I(3)")
2921 vereq(repr(1 + I(2)), "I(3)")
2922 vereq(repr(I(2) ** I(3)), "I(8)")
2923 vereq(repr(2 ** I(3)), "I(8)")
2924 vereq(repr(I(2) ** 3), "I(8)")
2925 vereq(repr(pow(I(2), I(3), I(5))), "I(3)")
Guido van Rossum4bb1e362001-09-28 23:49:48 +00002926 class S(str):
2927 def __eq__(self, other):
2928 return self.lower() == other.lower()
2929
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002930def subclasspropagation():
2931 if verbose: print "Testing propagation of slot functions to subclasses..."
2932 class A(object):
2933 pass
2934 class B(A):
2935 pass
2936 class C(A):
2937 pass
2938 class D(B, C):
2939 pass
2940 d = D()
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002941 orig_hash = hash(d) # related to id(d) in platform-dependent ways
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002942 A.__hash__ = lambda self: 42
2943 vereq(hash(d), 42)
2944 C.__hash__ = lambda self: 314
2945 vereq(hash(d), 314)
2946 B.__hash__ = lambda self: 144
2947 vereq(hash(d), 144)
2948 D.__hash__ = lambda self: 100
2949 vereq(hash(d), 100)
2950 del D.__hash__
2951 vereq(hash(d), 144)
2952 del B.__hash__
2953 vereq(hash(d), 314)
2954 del C.__hash__
2955 vereq(hash(d), 42)
2956 del A.__hash__
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002957 vereq(hash(d), orig_hash)
Guido van Rossum875eeaa2001-10-11 18:33:53 +00002958 d.foo = 42
2959 d.bar = 42
2960 vereq(d.foo, 42)
2961 vereq(d.bar, 42)
2962 def __getattribute__(self, name):
2963 if name == "foo":
2964 return 24
2965 return object.__getattribute__(self, name)
2966 A.__getattribute__ = __getattribute__
2967 vereq(d.foo, 24)
2968 vereq(d.bar, 42)
2969 def __getattr__(self, name):
2970 if name in ("spam", "foo", "bar"):
2971 return "hello"
2972 raise AttributeError, name
2973 B.__getattr__ = __getattr__
2974 vereq(d.spam, "hello")
2975 vereq(d.foo, 24)
2976 vereq(d.bar, 42)
2977 del A.__getattribute__
2978 vereq(d.foo, 42)
2979 del d.foo
2980 vereq(d.foo, "hello")
2981 vereq(d.bar, 42)
2982 del B.__getattr__
2983 try:
2984 d.foo
2985 except AttributeError:
2986 pass
2987 else:
2988 raise TestFailed, "d.foo should be undefined now"
Tim Petersfc57ccb2001-10-12 02:38:24 +00002989
Guido van Rossume7f3e242002-06-14 02:35:45 +00002990 # Test a nasty bug in recurse_down_subclasses()
2991 import gc
2992 class A(object):
2993 pass
2994 class B(A):
2995 pass
2996 del B
2997 gc.collect()
2998 A.__setitem__ = lambda *a: None # crash
2999
Tim Petersfc57ccb2001-10-12 02:38:24 +00003000def buffer_inherit():
3001 import binascii
3002 # SF bug [#470040] ParseTuple t# vs subclasses.
3003 if verbose:
3004 print "Testing that buffer interface is inherited ..."
3005
3006 class MyStr(str):
3007 pass
3008 base = 'abc'
3009 m = MyStr(base)
3010 # b2a_hex uses the buffer interface to get its argument's value, via
3011 # PyArg_ParseTuple 't#' code.
3012 vereq(binascii.b2a_hex(m), binascii.b2a_hex(base))
3013
3014 # It's not clear that unicode will continue to support the character
3015 # buffer interface, and this test will fail if that's taken away.
3016 class MyUni(unicode):
3017 pass
3018 base = u'abc'
3019 m = MyUni(base)
3020 vereq(binascii.b2a_hex(m), binascii.b2a_hex(base))
3021
3022 class MyInt(int):
3023 pass
3024 m = MyInt(42)
3025 try:
3026 binascii.b2a_hex(m)
3027 raise TestFailed('subclass of int should not have a buffer interface')
3028 except TypeError:
3029 pass
Tim Peters0ab085c2001-09-14 00:25:33 +00003030
Tim Petersc9933152001-10-16 20:18:24 +00003031def str_of_str_subclass():
3032 import binascii
3033 import cStringIO
3034
3035 if verbose:
3036 print "Testing __str__ defined in subclass of str ..."
3037
3038 class octetstring(str):
3039 def __str__(self):
3040 return binascii.b2a_hex(self)
3041 def __repr__(self):
3042 return self + " repr"
3043
3044 o = octetstring('A')
3045 vereq(type(o), octetstring)
3046 vereq(type(str(o)), str)
3047 vereq(type(repr(o)), str)
3048 vereq(ord(o), 0x41)
3049 vereq(str(o), '41')
3050 vereq(repr(o), 'A repr')
3051 vereq(o.__str__(), '41')
3052 vereq(o.__repr__(), 'A repr')
3053
3054 capture = cStringIO.StringIO()
3055 # Calling str() or not exercises different internal paths.
3056 print >> capture, o
3057 print >> capture, str(o)
3058 vereq(capture.getvalue(), '41\n41\n')
3059 capture.close()
3060
Guido van Rossumc8e56452001-10-22 00:43:43 +00003061def kwdargs():
3062 if verbose: print "Testing keyword arguments to __init__, __call__..."
3063 def f(a): return a
3064 vereq(f.__call__(a=42), 42)
3065 a = []
3066 list.__init__(a, sequence=[0, 1, 2])
Tim Peters1fc240e2001-10-26 05:06:50 +00003067 vereq(a, [0, 1, 2])
Guido van Rossumc8e56452001-10-22 00:43:43 +00003068
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003069def recursive__call__():
3070 if verbose: print ("Testing recursive __call__() by setting to instance of "
3071 "class ...")
3072 class A(object):
3073 pass
3074
3075 A.__call__ = A()
3076 try:
3077 A()()
3078 except RuntimeError:
3079 pass
3080 else:
3081 raise TestFailed("Recursion limit should have been reached for "
3082 "__call__()")
3083
Guido van Rossumed87ad82001-10-30 02:33:02 +00003084def delhook():
3085 if verbose: print "Testing __del__ hook..."
3086 log = []
3087 class C(object):
3088 def __del__(self):
3089 log.append(1)
3090 c = C()
3091 vereq(log, [])
3092 del c
3093 vereq(log, [1])
3094
Guido van Rossum29d26062001-12-11 04:37:34 +00003095 class D(object): pass
3096 d = D()
3097 try: del d[0]
3098 except TypeError: pass
3099 else: raise TestFailed, "invalid del() didn't raise TypeError"
3100
Guido van Rossumdbb53d92001-12-03 16:32:18 +00003101def hashinherit():
3102 if verbose: print "Testing hash of mutable subclasses..."
3103
3104 class mydict(dict):
3105 pass
3106 d = mydict()
3107 try:
3108 hash(d)
3109 except TypeError:
3110 pass
3111 else:
3112 raise TestFailed, "hash() of dict subclass should fail"
3113
3114 class mylist(list):
3115 pass
3116 d = mylist()
3117 try:
3118 hash(d)
3119 except TypeError:
3120 pass
3121 else:
3122 raise TestFailed, "hash() of list subclass should fail"
3123
Guido van Rossum29d26062001-12-11 04:37:34 +00003124def strops():
3125 try: 'a' + 5
3126 except TypeError: pass
3127 else: raise TestFailed, "'' + 5 doesn't raise TypeError"
3128
3129 try: ''.split('')
3130 except ValueError: pass
3131 else: raise TestFailed, "''.split('') doesn't raise ValueError"
3132
3133 try: ''.join([0])
3134 except TypeError: pass
3135 else: raise TestFailed, "''.join([0]) doesn't raise TypeError"
3136
3137 try: ''.rindex('5')
3138 except ValueError: pass
3139 else: raise TestFailed, "''.rindex('5') doesn't raise ValueError"
3140
Guido van Rossum29d26062001-12-11 04:37:34 +00003141 try: '%(n)s' % None
3142 except TypeError: pass
3143 else: raise TestFailed, "'%(n)s' % None doesn't raise TypeError"
3144
3145 try: '%(n' % {}
3146 except ValueError: pass
3147 else: raise TestFailed, "'%(n' % {} '' doesn't raise ValueError"
3148
3149 try: '%*s' % ('abc')
3150 except TypeError: pass
3151 else: raise TestFailed, "'%*s' % ('abc') doesn't raise TypeError"
3152
3153 try: '%*.*s' % ('abc', 5)
3154 except TypeError: pass
3155 else: raise TestFailed, "'%*.*s' % ('abc', 5) doesn't raise TypeError"
3156
3157 try: '%s' % (1, 2)
3158 except TypeError: pass
3159 else: raise TestFailed, "'%s' % (1, 2) doesn't raise TypeError"
3160
3161 try: '%' % None
3162 except ValueError: pass
3163 else: raise TestFailed, "'%' % None doesn't raise ValueError"
3164
3165 vereq('534253'.isdigit(), 1)
3166 vereq('534253x'.isdigit(), 0)
3167 vereq('%c' % 5, '\x05')
3168 vereq('%c' % '5', '5')
3169
Guido van Rossum2764a3a2001-12-28 21:39:03 +00003170def deepcopyrecursive():
3171 if verbose: print "Testing deepcopy of recursive objects..."
3172 class Node:
3173 pass
3174 a = Node()
3175 b = Node()
3176 a.b = b
3177 b.a = a
3178 z = deepcopy(a) # This blew up before
Guido van Rossum29d26062001-12-11 04:37:34 +00003179
Guido van Rossumd7035672002-03-12 20:43:31 +00003180def modules():
3181 if verbose: print "Testing uninitialized module objects..."
3182 from types import ModuleType as M
3183 m = M.__new__(M)
3184 str(m)
3185 vereq(hasattr(m, "__name__"), 0)
3186 vereq(hasattr(m, "__file__"), 0)
3187 vereq(hasattr(m, "foo"), 0)
3188 vereq(m.__dict__, None)
3189 m.foo = 1
3190 vereq(m.__dict__, {"foo": 1})
Guido van Rossum29d26062001-12-11 04:37:34 +00003191
Walter Dörwalddbd2d252002-03-25 18:36:32 +00003192def dictproxyiterkeys():
3193 class C(object):
3194 def meth(self):
3195 pass
3196 if verbose: print "Testing dict-proxy iterkeys..."
3197 keys = [ key for key in C.__dict__.iterkeys() ]
3198 keys.sort()
3199 vereq(keys, ['__dict__', '__doc__', '__module__', '__weakref__', 'meth'])
3200
3201def dictproxyitervalues():
3202 class C(object):
3203 def meth(self):
3204 pass
3205 if verbose: print "Testing dict-proxy itervalues..."
3206 values = [ values for values in C.__dict__.itervalues() ]
3207 vereq(len(values), 5)
3208
3209def dictproxyiteritems():
3210 class C(object):
3211 def meth(self):
3212 pass
3213 if verbose: print "Testing dict-proxy iteritems..."
3214 keys = [ key for (key, value) in C.__dict__.iteritems() ]
3215 keys.sort()
3216 vereq(keys, ['__dict__', '__doc__', '__module__', '__weakref__', 'meth'])
3217
Guido van Rossum8ace1ab2002-04-06 01:05:01 +00003218def funnynew():
3219 if verbose: print "Testing __new__ returning something unexpected..."
3220 class C(object):
3221 def __new__(cls, arg):
3222 if isinstance(arg, str): return [1, 2, 3]
3223 elif isinstance(arg, int): return object.__new__(D)
3224 else: return object.__new__(cls)
3225 class D(C):
3226 def __init__(self, arg):
3227 self.foo = arg
3228 vereq(C("1"), [1, 2, 3])
3229 vereq(D("1"), [1, 2, 3])
3230 d = D(None)
3231 veris(d.foo, None)
3232 d = C(1)
3233 vereq(isinstance(d, D), True)
3234 vereq(d.foo, 1)
3235 d = D(1)
3236 vereq(isinstance(d, D), True)
3237 vereq(d.foo, 1)
3238
Guido van Rossume8fc6402002-04-16 16:44:51 +00003239def imulbug():
3240 # SF bug 544647
3241 if verbose: print "Testing for __imul__ problems..."
3242 class C(object):
3243 def __imul__(self, other):
3244 return (self, other)
3245 x = C()
3246 y = x
3247 y *= 1.0
3248 vereq(y, (x, 1.0))
3249 y = x
3250 y *= 2
3251 vereq(y, (x, 2))
3252 y = x
3253 y *= 3L
3254 vereq(y, (x, 3L))
3255 y = x
3256 y *= 1L<<100
3257 vereq(y, (x, 1L<<100))
3258 y = x
3259 y *= None
3260 vereq(y, (x, None))
3261 y = x
3262 y *= "foo"
3263 vereq(y, (x, "foo"))
3264
Guido van Rossumd99b3e72002-04-18 00:27:33 +00003265def docdescriptor():
3266 # SF bug 542984
3267 if verbose: print "Testing __doc__ descriptor..."
3268 class DocDescr(object):
3269 def __get__(self, object, otype):
3270 if object:
3271 object = object.__class__.__name__ + ' instance'
3272 if otype:
3273 otype = otype.__name__
3274 return 'object=%s; type=%s' % (object, otype)
3275 class OldClass:
3276 __doc__ = DocDescr()
3277 class NewClass(object):
3278 __doc__ = DocDescr()
3279 vereq(OldClass.__doc__, 'object=None; type=OldClass')
3280 vereq(OldClass().__doc__, 'object=OldClass instance; type=OldClass')
3281 vereq(NewClass.__doc__, 'object=None; type=NewClass')
3282 vereq(NewClass().__doc__, 'object=NewClass instance; type=NewClass')
3283
Guido van Rossuma48cb8f2002-06-06 17:53:03 +00003284def copy_setstate():
3285 if verbose:
3286 print "Testing that copy.*copy() correctly uses __setstate__..."
3287 import copy
3288 class C(object):
3289 def __init__(self, foo=None):
3290 self.foo = foo
3291 self.__foo = foo
3292 def setfoo(self, foo=None):
3293 self.foo = foo
3294 def getfoo(self):
3295 return self.__foo
3296 def __getstate__(self):
3297 return [self.foo]
3298 def __setstate__(self, lst):
3299 assert len(lst) == 1
3300 self.__foo = self.foo = lst[0]
3301 a = C(42)
3302 a.setfoo(24)
3303 vereq(a.foo, 24)
3304 vereq(a.getfoo(), 42)
3305 b = copy.copy(a)
3306 vereq(b.foo, 24)
3307 vereq(b.getfoo(), 24)
3308 b = copy.deepcopy(a)
3309 vereq(b.foo, 24)
3310 vereq(b.getfoo(), 24)
3311
Guido van Rossum09638c12002-06-13 19:17:46 +00003312def slices():
3313 if verbose:
3314 print "Testing cases with slices and overridden __getitem__ ..."
3315 # Strings
3316 vereq("hello"[:4], "hell")
3317 vereq("hello"[slice(4)], "hell")
3318 vereq(str.__getitem__("hello", slice(4)), "hell")
3319 class S(str):
3320 def __getitem__(self, x):
3321 return str.__getitem__(self, x)
3322 vereq(S("hello")[:4], "hell")
3323 vereq(S("hello")[slice(4)], "hell")
3324 vereq(S("hello").__getitem__(slice(4)), "hell")
3325 # Tuples
3326 vereq((1,2,3)[:2], (1,2))
3327 vereq((1,2,3)[slice(2)], (1,2))
3328 vereq(tuple.__getitem__((1,2,3), slice(2)), (1,2))
3329 class T(tuple):
3330 def __getitem__(self, x):
3331 return tuple.__getitem__(self, x)
3332 vereq(T((1,2,3))[:2], (1,2))
3333 vereq(T((1,2,3))[slice(2)], (1,2))
3334 vereq(T((1,2,3)).__getitem__(slice(2)), (1,2))
3335 # Lists
3336 vereq([1,2,3][:2], [1,2])
3337 vereq([1,2,3][slice(2)], [1,2])
3338 vereq(list.__getitem__([1,2,3], slice(2)), [1,2])
3339 class L(list):
3340 def __getitem__(self, x):
3341 return list.__getitem__(self, x)
3342 vereq(L([1,2,3])[:2], [1,2])
3343 vereq(L([1,2,3])[slice(2)], [1,2])
3344 vereq(L([1,2,3]).__getitem__(slice(2)), [1,2])
3345 # Now do lists and __setitem__
3346 a = L([1,2,3])
3347 a[slice(1, 3)] = [3,2]
3348 vereq(a, [1,3,2])
3349 a[slice(0, 2, 1)] = [3,1]
3350 vereq(a, [3,1,2])
3351 a.__setitem__(slice(1, 3), [2,1])
3352 vereq(a, [3,2,1])
3353 a.__setitem__(slice(0, 2, 1), [2,3])
3354 vereq(a, [2,3,1])
3355
Tim Peters2484aae2002-07-11 06:56:07 +00003356def subtype_resurrection():
3357 if verbose:
Tim Peters45228ca2002-07-11 07:09:42 +00003358 print "Testing resurrection of new-style instance..."
Tim Peters2484aae2002-07-11 06:56:07 +00003359
3360 class C(object):
3361 container = []
3362
3363 def __del__(self):
3364 # resurrect the instance
3365 C.container.append(self)
3366
3367 c = C()
3368 c.attr = 42
Tim Peters14cb1e12002-07-11 18:26:21 +00003369 # The most interesting thing here is whether this blows up, due to flawed
Tim Peters45228ca2002-07-11 07:09:42 +00003370 # GC tracking logic in typeobject.c's call_finalizer() (a 2.2.1 bug).
Tim Peters2484aae2002-07-11 06:56:07 +00003371 del c
Tim Peters14cb1e12002-07-11 18:26:21 +00003372
3373 # If that didn't blow up, it's also interesting to see whether clearing
3374 # the last container slot works: that will attempt to delete c again,
3375 # which will cause c to get appended back to the container again "during"
3376 # the del.
3377 del C.container[-1]
3378 vereq(len(C.container), 1)
Tim Peters2484aae2002-07-11 06:56:07 +00003379 vereq(C.container[-1].attr, 42)
Guido van Rossum09638c12002-06-13 19:17:46 +00003380
Tim Peters14cb1e12002-07-11 18:26:21 +00003381 # Make c mortal again, so that the test framework with -l doesn't report
3382 # it as a leak.
3383 del C.__del__
3384
Guido van Rossum2d702462002-08-06 21:28:28 +00003385def slottrash():
3386 # Deallocating deeply nested slotted trash caused stack overflows
3387 if verbose:
3388 print "Testing slot trash..."
3389 class trash(object):
3390 __slots__ = ['x']
3391 def __init__(self, x):
3392 self.x = x
3393 o = None
3394 for i in xrange(50000):
3395 o = trash(o)
3396 del o
3397
Neal Norwitzf9dd0f12002-08-13 17:16:49 +00003398def slotmultipleinheritance():
3399 # SF bug 575229, multiple inheritance w/ slots dumps core
3400 class A(object):
3401 __slots__=()
3402 class B(object):
3403 pass
3404 class C(A,B) :
3405 __slots__=()
Guido van Rossum8b056da2002-08-13 18:26:26 +00003406 vereq(C.__basicsize__, B.__basicsize__)
3407 verify(hasattr(C, '__dict__'))
3408 verify(hasattr(C, '__weakref__'))
3409 C().x = 2
Neal Norwitzf9dd0f12002-08-13 17:16:49 +00003410
Guido van Rossum0f5f0b82002-08-09 16:11:37 +00003411def testrmul():
3412 # SF patch 592646
3413 if verbose:
3414 print "Testing correct invocation of __rmul__..."
3415 class C(object):
3416 def __mul__(self, other):
3417 return "mul"
3418 def __rmul__(self, other):
3419 return "rmul"
3420 a = C()
3421 vereq(a*2, "mul")
3422 vereq(a*2.2, "mul")
3423 vereq(2*a, "rmul")
3424 vereq(2.2*a, "rmul")
3425
Guido van Rossum6e5680f2002-10-15 01:01:53 +00003426def testipow():
3427 # [SF bug 620179]
3428 if verbose:
3429 print "Testing correct invocation of __ipow__..."
3430 class C(object):
3431 def __ipow__(self, other):
3432 pass
3433 a = C()
3434 a **= 2
3435
Guido van Rossum9fc8a292002-05-24 21:40:08 +00003436def do_this_first():
3437 if verbose:
3438 print "Testing SF bug 551412 ..."
3439 # This dumps core when SF bug 551412 isn't fixed --
3440 # but only when test_descr.py is run separately.
3441 # (That can't be helped -- as soon as PyType_Ready()
3442 # is called for PyLong_Type, the bug is gone.)
3443 class UserLong(object):
3444 def __pow__(self, *args):
3445 pass
3446 try:
3447 pow(0L, UserLong(), 0L)
3448 except:
3449 pass
3450
Guido van Rossuma96b0df2002-06-18 16:49:45 +00003451 if verbose:
3452 print "Testing SF bug 570483..."
3453 # Another segfault only when run early
3454 # (before PyType_Ready(tuple) is called)
3455 type.mro(tuple)
3456
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003457def test_mutable_bases():
Guido van Rossum2720b0d2003-01-06 21:26:44 +00003458 if verbose:
3459 print "Testing mutable bases..."
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003460 # stuff that should work:
3461 class C(object):
3462 pass
3463 class C2(object):
3464 def __getattribute__(self, attr):
3465 if attr == 'a':
3466 return 2
3467 else:
Tim Peters6578dc92002-12-24 18:31:27 +00003468 return super(C2, self).__getattribute__(attr)
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003469 def meth(self):
3470 return 1
3471 class D(C):
3472 pass
3473 class E(D):
3474 pass
3475 d = D()
3476 e = E()
Michael W. Hudsoncaf17be2002-11-27 10:24:44 +00003477 D.__bases__ = (C,)
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003478 D.__bases__ = (C2,)
3479 vereq(d.meth(), 1)
3480 vereq(e.meth(), 1)
3481 vereq(d.a, 2)
3482 vereq(e.a, 2)
3483 vereq(C2.__subclasses__(), [D])
3484
3485 # stuff that shouldn't:
3486 class L(list):
3487 pass
3488
3489 try:
3490 L.__bases__ = (dict,)
3491 except TypeError:
3492 pass
3493 else:
3494 raise TestFailed, "shouldn't turn list subclass into dict subclass"
3495
3496 try:
3497 list.__bases__ = (dict,)
3498 except TypeError:
3499 pass
3500 else:
3501 raise TestFailed, "shouldn't be able to assign to list.__bases__"
3502
3503 try:
3504 del D.__bases__
3505 except TypeError:
3506 pass
3507 else:
3508 raise TestFailed, "shouldn't be able to delete .__bases__"
3509
3510 try:
Guido van Rossum3bbc0ee2002-12-13 17:49:38 +00003511 D.__bases__ = ()
3512 except TypeError, msg:
3513 if str(msg) == "a new-style class can't have only classic bases":
3514 raise TestFailed, "wrong error message for .__bases__ = ()"
3515 else:
3516 raise TestFailed, "shouldn't be able to set .__bases__ to ()"
3517
3518 try:
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003519 D.__bases__ = (D,)
3520 except TypeError:
3521 pass
3522 else:
3523 # actually, we'll have crashed by here...
3524 raise TestFailed, "shouldn't be able to create inheritance cycles"
3525
Michael W. Hudsoncaf17be2002-11-27 10:24:44 +00003526 try:
Michael W. Hudsone723e452003-08-07 14:58:10 +00003527 D.__bases__ = (C, C)
3528 except TypeError:
3529 pass
3530 else:
3531 raise TestFailed, "didn't detect repeated base classes"
3532
3533 try:
Michael W. Hudsoncaf17be2002-11-27 10:24:44 +00003534 D.__bases__ = (E,)
3535 except TypeError:
3536 pass
3537 else:
3538 raise TestFailed, "shouldn't be able to create inheritance cycles"
3539
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003540def test_mutable_bases_with_failing_mro():
Guido van Rossum2720b0d2003-01-06 21:26:44 +00003541 if verbose:
3542 print "Testing mutable bases with failing mro..."
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003543 class WorkOnce(type):
3544 def __new__(self, name, bases, ns):
3545 self.flag = 0
3546 return super(WorkOnce, self).__new__(WorkOnce, name, bases, ns)
3547 def mro(self):
3548 if self.flag > 0:
3549 raise RuntimeError, "bozo"
3550 else:
3551 self.flag += 1
3552 return type.mro(self)
3553
3554 class WorkAlways(type):
3555 def mro(self):
3556 # this is here to make sure that .mro()s aren't called
3557 # with an exception set (which was possible at one point).
3558 # An error message will be printed in a debug build.
3559 # What's a good way to test for this?
3560 return type.mro(self)
3561
3562 class C(object):
3563 pass
3564
3565 class C2(object):
3566 pass
3567
3568 class D(C):
3569 pass
3570
3571 class E(D):
3572 pass
3573
3574 class F(D):
3575 __metaclass__ = WorkOnce
3576
3577 class G(D):
3578 __metaclass__ = WorkAlways
3579
3580 # Immediate subclasses have their mro's adjusted in alphabetical
3581 # order, so E's will get adjusted before adjusting F's fails. We
3582 # check here that E's gets restored.
Tim Peters6578dc92002-12-24 18:31:27 +00003583
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003584 E_mro_before = E.__mro__
Michael W. Hudson7e7c00d2002-11-27 15:40:09 +00003585 D_mro_before = D.__mro__
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003586
3587 try:
3588 D.__bases__ = (C2,)
3589 except RuntimeError:
3590 vereq(E.__mro__, E_mro_before)
Michael W. Hudson7e7c00d2002-11-27 15:40:09 +00003591 vereq(D.__mro__, D_mro_before)
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003592 else:
3593 raise TestFailed, "exception not propagated"
3594
3595def test_mutable_bases_catch_mro_conflict():
Guido van Rossum2720b0d2003-01-06 21:26:44 +00003596 if verbose:
3597 print "Testing mutable bases catch mro conflict..."
Michael W. Hudson586da8f2002-11-27 15:20:19 +00003598 class A(object):
3599 pass
3600
3601 class B(object):
3602 pass
3603
3604 class C(A, B):
3605 pass
3606
3607 class D(A, B):
3608 pass
3609
3610 class E(C, D):
3611 pass
3612
3613 try:
3614 C.__bases__ = (B, A)
3615 except TypeError:
3616 pass
3617 else:
3618 raise TestFailed, "didn't catch MRO conflict"
Tim Peters6578dc92002-12-24 18:31:27 +00003619
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003620def mutable_names():
Guido van Rossum2720b0d2003-01-06 21:26:44 +00003621 if verbose:
3622 print "Testing mutable names..."
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003623 class C(object):
3624 pass
3625
Michael W. Hudsonade8c8b22002-11-27 16:29:26 +00003626 # C.__module__ could be 'test_descr' or '__main__'
3627 mod = C.__module__
Tim Peters6578dc92002-12-24 18:31:27 +00003628
Michael W. Hudsonade8c8b22002-11-27 16:29:26 +00003629 C.__name__ = 'D'
3630 vereq((C.__module__, C.__name__), (mod, 'D'))
3631
3632 C.__name__ = 'D.E'
3633 vereq((C.__module__, C.__name__), (mod, 'D.E'))
Tim Peters6578dc92002-12-24 18:31:27 +00003634
Guido van Rossum613f24f2003-01-06 23:00:59 +00003635def subclass_right_op():
3636 if verbose:
3637 print "Testing correct dispatch of subclass overloading __r<op>__..."
3638
3639 # This code tests various cases where right-dispatch of a subclass
3640 # should be preferred over left-dispatch of a base class.
3641
3642 # Case 1: subclass of int; this tests code in abstract.c::binary_op1()
3643
3644 class B(int):
Guido van Rossumf389c772003-02-27 20:04:19 +00003645 def __floordiv__(self, other):
3646 return "B.__floordiv__"
3647 def __rfloordiv__(self, other):
3648 return "B.__rfloordiv__"
Guido van Rossum613f24f2003-01-06 23:00:59 +00003649
Guido van Rossumf389c772003-02-27 20:04:19 +00003650 vereq(B(1) // 1, "B.__floordiv__")
3651 vereq(1 // B(1), "B.__rfloordiv__")
Guido van Rossum613f24f2003-01-06 23:00:59 +00003652
3653 # Case 2: subclass of object; this is just the baseline for case 3
3654
3655 class C(object):
Guido van Rossumf389c772003-02-27 20:04:19 +00003656 def __floordiv__(self, other):
3657 return "C.__floordiv__"
3658 def __rfloordiv__(self, other):
3659 return "C.__rfloordiv__"
Guido van Rossum613f24f2003-01-06 23:00:59 +00003660
Guido van Rossumf389c772003-02-27 20:04:19 +00003661 vereq(C() // 1, "C.__floordiv__")
3662 vereq(1 // C(), "C.__rfloordiv__")
Guido van Rossum613f24f2003-01-06 23:00:59 +00003663
3664 # Case 3: subclass of new-style class; here it gets interesting
3665
3666 class D(C):
Guido van Rossumf389c772003-02-27 20:04:19 +00003667 def __floordiv__(self, other):
3668 return "D.__floordiv__"
3669 def __rfloordiv__(self, other):
3670 return "D.__rfloordiv__"
Guido van Rossum613f24f2003-01-06 23:00:59 +00003671
Guido van Rossumf389c772003-02-27 20:04:19 +00003672 vereq(D() // C(), "D.__floordiv__")
3673 vereq(C() // D(), "D.__rfloordiv__")
Guido van Rossum613f24f2003-01-06 23:00:59 +00003674
3675 # Case 4: this didn't work right in 2.2.2 and 2.3a1
3676
3677 class E(C):
3678 pass
3679
Guido van Rossumf389c772003-02-27 20:04:19 +00003680 vereq(E.__rfloordiv__, C.__rfloordiv__)
Guido van Rossum613f24f2003-01-06 23:00:59 +00003681
Guido van Rossumf389c772003-02-27 20:04:19 +00003682 vereq(E() // 1, "C.__floordiv__")
3683 vereq(1 // E(), "C.__rfloordiv__")
3684 vereq(E() // C(), "C.__floordiv__")
3685 vereq(C() // E(), "C.__floordiv__") # This one would fail
Guido van Rossum613f24f2003-01-06 23:00:59 +00003686
Guido van Rossum373c7412003-01-07 13:41:37 +00003687def dict_type_with_metaclass():
3688 if verbose:
3689 print "Testing type of __dict__ when __metaclass__ set..."
3690
3691 class B(object):
3692 pass
3693 class M(type):
3694 pass
3695 class C:
3696 # In 2.3a1, C.__dict__ was a real dict rather than a dict proxy
3697 __metaclass__ = M
3698 veris(type(C.__dict__), type(B.__dict__))
3699
Guido van Rossumb6e5a0c2003-02-11 18:44:42 +00003700def meth_class_get():
3701 # Full coverage of descrobject.c::classmethod_get()
Guido van Rossum03bc7d32003-02-12 03:32:58 +00003702 if verbose:
3703 print "Testing __get__ method of METH_CLASS C methods..."
Guido van Rossumb6e5a0c2003-02-11 18:44:42 +00003704 # Baseline
3705 arg = [1, 2, 3]
3706 res = {1: None, 2: None, 3: None}
3707 vereq(dict.fromkeys(arg), res)
3708 vereq({}.fromkeys(arg), res)
3709 # Now get the descriptor
3710 descr = dict.__dict__["fromkeys"]
3711 # More baseline using the descriptor directly
3712 vereq(descr.__get__(None, dict)(arg), res)
3713 vereq(descr.__get__({})(arg), res)
3714 # Now check various error cases
3715 try:
3716 descr.__get__(None, None)
3717 except TypeError:
3718 pass
3719 else:
3720 raise TestFailed, "shouldn't have allowed descr.__get__(None, None)"
3721 try:
3722 descr.__get__(42)
3723 except TypeError:
3724 pass
3725 else:
3726 raise TestFailed, "shouldn't have allowed descr.__get__(42)"
3727 try:
3728 descr.__get__(None, 42)
3729 except TypeError:
3730 pass
3731 else:
3732 raise TestFailed, "shouldn't have allowed descr.__get__(None, 42)"
3733 try:
3734 descr.__get__(None, int)
3735 except TypeError:
3736 pass
3737 else:
3738 raise TestFailed, "shouldn't have allowed descr.__get__(None, int)"
3739
Guido van Rossum03bc7d32003-02-12 03:32:58 +00003740def isinst_isclass():
3741 if verbose:
3742 print "Testing proxy isinstance() and isclass()..."
3743 class Proxy(object):
3744 def __init__(self, obj):
3745 self.__obj = obj
3746 def __getattribute__(self, name):
3747 if name.startswith("_Proxy__"):
3748 return object.__getattribute__(self, name)
3749 else:
3750 return getattr(self.__obj, name)
3751 # Test with a classic class
3752 class C:
3753 pass
3754 a = C()
3755 pa = Proxy(a)
3756 verify(isinstance(a, C)) # Baseline
3757 verify(isinstance(pa, C)) # Test
Guido van Rossuma89d10e2003-02-12 03:58:38 +00003758 # Test with a classic subclass
3759 class D(C):
3760 pass
3761 a = D()
3762 pa = Proxy(a)
3763 verify(isinstance(a, C)) # Baseline
3764 verify(isinstance(pa, C)) # Test
Guido van Rossum03bc7d32003-02-12 03:32:58 +00003765 # Test with a new-style class
3766 class C(object):
3767 pass
3768 a = C()
3769 pa = Proxy(a)
3770 verify(isinstance(a, C)) # Baseline
3771 verify(isinstance(pa, C)) # Test
Guido van Rossuma89d10e2003-02-12 03:58:38 +00003772 # Test with a new-style subclass
3773 class D(C):
3774 pass
3775 a = D()
3776 pa = Proxy(a)
3777 verify(isinstance(a, C)) # Baseline
3778 verify(isinstance(pa, C)) # Test
3779
3780def proxysuper():
3781 if verbose:
3782 print "Testing super() for a proxy object..."
3783 class Proxy(object):
3784 def __init__(self, obj):
3785 self.__obj = obj
3786 def __getattribute__(self, name):
3787 if name.startswith("_Proxy__"):
3788 return object.__getattribute__(self, name)
3789 else:
3790 return getattr(self.__obj, name)
3791
3792 class B(object):
3793 def f(self):
3794 return "B.f"
3795
3796 class C(B):
3797 def f(self):
3798 return super(C, self).f() + "->C.f"
3799
3800 obj = C()
3801 p = Proxy(obj)
3802 vereq(C.__dict__["f"](p), "B.f->C.f")
Guido van Rossum03bc7d32003-02-12 03:32:58 +00003803
Guido van Rossum52b27052003-04-15 20:05:10 +00003804def carloverre():
Guido van Rossum4dcdb782003-04-14 21:46:03 +00003805 if verbose:
Guido van Rossum52b27052003-04-15 20:05:10 +00003806 print "Testing prohibition of Carlo Verre's hack..."
Guido van Rossum4dcdb782003-04-14 21:46:03 +00003807 try:
3808 object.__setattr__(str, "foo", 42)
3809 except TypeError:
3810 pass
3811 else:
Guido van Rossum52b27052003-04-15 20:05:10 +00003812 raise TestFailed, "Carlo Verre __setattr__ suceeded!"
Guido van Rossum4dcdb782003-04-14 21:46:03 +00003813 try:
3814 object.__delattr__(str, "lower")
3815 except TypeError:
3816 pass
3817 else:
Guido van Rossum52b27052003-04-15 20:05:10 +00003818 raise TestFailed, "Carlo Verre __delattr__ succeeded!"
Guido van Rossum4dcdb782003-04-14 21:46:03 +00003819
Guido van Rossumaabe0b32003-05-29 14:26:57 +00003820def weakref_segfault():
3821 # SF 742911
3822 if verbose:
3823 print "Testing weakref segfault..."
3824
3825 import weakref
3826
3827 class Provoker:
3828 def __init__(self, referrent):
3829 self.ref = weakref.ref(referrent)
3830
3831 def __del__(self):
3832 x = self.ref()
Guido van Rossumaabe0b32003-05-29 14:26:57 +00003833
3834 class Oops(object):
3835 pass
3836
3837 o = Oops()
3838 o.whatever = Provoker(o)
3839 del o
3840
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003841def wrapper_segfault():
3842 # SF 927248: deeply nested wrappers could cause stack overflow
3843 f = lambda:None
3844 for i in xrange(1000000):
3845 f = f.__call__
3846 f = None
3847
Raymond Hettinger2b6220d2003-06-29 15:44:07 +00003848# Fix SF #762455, segfault when sys.stdout is changed in getattr
3849def filefault():
3850 if verbose:
3851 print "Testing sys.stdout is changed in getattr..."
3852 import sys
3853 class StdoutGuard:
3854 def __getattr__(self, attr):
3855 sys.stdout = sys.__stdout__
3856 raise RuntimeError("Premature access to sys.stdout.%s" % attr)
3857 sys.stdout = StdoutGuard()
3858 try:
3859 print "Oops!"
3860 except RuntimeError:
3861 pass
Michael W. Hudson98bbc492002-11-26 14:47:27 +00003862
Michael W. Hudsonb2c7de42003-08-15 13:07:47 +00003863def vicious_descriptor_nonsense():
3864 # A potential segfault spotted by Thomas Wouters in mail to
3865 # python-dev 2003-04-17, turned into an example & fixed by Michael
3866 # Hudson just less than four months later...
3867 if verbose:
3868 print "Testing vicious_descriptor_nonsense..."
3869
3870 class Evil(object):
3871 def __hash__(self):
3872 return hash('attr')
3873 def __eq__(self, other):
3874 del C.attr
3875 return 0
3876
3877 class Descr(object):
3878 def __get__(self, ob, type=None):
3879 return 1
3880
3881 class C(object):
3882 attr = Descr()
3883
3884 c = C()
3885 c.__dict__[Evil()] = 0
3886
3887 vereq(c.attr, 1)
3888 # this makes a crash more likely:
3889 import gc; gc.collect()
3890 vereq(hasattr(c, 'attr'), False)
Tim Peters58eb11c2004-01-18 20:29:55 +00003891
Raymond Hettingerb67cc802005-03-03 16:45:19 +00003892def test_init():
3893 # SF 1155938
3894 class Foo(object):
3895 def __init__(self):
3896 return 10
3897 try:
3898 Foo()
3899 except TypeError:
3900 pass
3901 else:
3902 raise TestFailed, "did not test __init__() for None return"
3903
Armin Rigoc6686b72005-11-07 08:38:00 +00003904def methodwrapper():
3905 # <type 'method-wrapper'> did not support any reflection before 2.5
3906 if verbose:
3907 print "Testing method-wrapper objects..."
3908
Guido van Rossum47b9ff62006-08-24 00:41:19 +00003909 return # XXX should methods really support __eq__?
3910
Armin Rigoc6686b72005-11-07 08:38:00 +00003911 l = []
3912 vereq(l.__add__, l.__add__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00003913 vereq(l.__add__, [].__add__)
3914 verify(l.__add__ != [5].__add__)
3915 verify(l.__add__ != l.__mul__)
Armin Rigoc6686b72005-11-07 08:38:00 +00003916 verify(l.__add__.__name__ == '__add__')
3917 verify(l.__add__.__self__ is l)
3918 verify(l.__add__.__objclass__ is list)
3919 vereq(l.__add__.__doc__, list.__add__.__doc__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00003920 try:
3921 hash(l.__add__)
3922 except TypeError:
3923 pass
3924 else:
3925 raise TestFailed("no TypeError from hash([].__add__)")
3926
3927 t = ()
3928 t += (7,)
3929 vereq(t.__add__, (7,).__add__)
3930 vereq(hash(t.__add__), hash((7,).__add__))
Michael W. Hudsonb2c7de42003-08-15 13:07:47 +00003931
Armin Rigofd163f92005-12-29 15:59:19 +00003932def notimplemented():
3933 # all binary methods should be able to return a NotImplemented
3934 if verbose:
3935 print "Testing NotImplemented..."
3936
3937 import sys
3938 import types
3939 import operator
3940
3941 def specialmethod(self, other):
3942 return NotImplemented
3943
3944 def check(expr, x, y):
3945 try:
Georg Brandl7cae87c2006-09-06 06:51:57 +00003946 exec(expr, {'x': x, 'y': y, 'operator': operator})
Armin Rigofd163f92005-12-29 15:59:19 +00003947 except TypeError:
3948 pass
3949 else:
3950 raise TestFailed("no TypeError from %r" % (expr,))
3951
3952 N1 = sys.maxint + 1L # might trigger OverflowErrors instead of TypeErrors
3953 N2 = sys.maxint # if sizeof(int) < sizeof(long), might trigger
3954 # ValueErrors instead of TypeErrors
3955 for metaclass in [type, types.ClassType]:
3956 for name, expr, iexpr in [
3957 ('__add__', 'x + y', 'x += y'),
3958 ('__sub__', 'x - y', 'x -= y'),
3959 ('__mul__', 'x * y', 'x *= y'),
Neal Norwitzbcc0db82006-03-24 08:14:36 +00003960 ('__truediv__', 'x / y', None),
3961 ('__floordiv__', 'x // y', None),
Armin Rigofd163f92005-12-29 15:59:19 +00003962 ('__mod__', 'x % y', 'x %= y'),
3963 ('__divmod__', 'divmod(x, y)', None),
3964 ('__pow__', 'x ** y', 'x **= y'),
3965 ('__lshift__', 'x << y', 'x <<= y'),
3966 ('__rshift__', 'x >> y', 'x >>= y'),
3967 ('__and__', 'x & y', 'x &= y'),
3968 ('__or__', 'x | y', 'x |= y'),
3969 ('__xor__', 'x ^ y', 'x ^= y'),
Neal Norwitz4886cc32006-08-21 17:06:07 +00003970 ]:
3971 rname = '__r' + name[2:]
Armin Rigofd163f92005-12-29 15:59:19 +00003972 A = metaclass('A', (), {name: specialmethod})
3973 B = metaclass('B', (), {rname: specialmethod})
3974 a = A()
3975 b = B()
3976 check(expr, a, a)
3977 check(expr, a, b)
3978 check(expr, b, a)
3979 check(expr, b, b)
3980 check(expr, a, N1)
3981 check(expr, a, N2)
3982 check(expr, N1, b)
3983 check(expr, N2, b)
3984 if iexpr:
3985 check(iexpr, a, a)
3986 check(iexpr, a, b)
3987 check(iexpr, b, a)
3988 check(iexpr, b, b)
3989 check(iexpr, a, N1)
3990 check(iexpr, a, N2)
3991 iname = '__i' + name[2:]
3992 C = metaclass('C', (), {iname: specialmethod})
3993 c = C()
3994 check(iexpr, c, a)
3995 check(iexpr, c, b)
3996 check(iexpr, c, N1)
3997 check(iexpr, c, N2)
3998
Guido van Rossuma56b42b2001-09-20 21:39:07 +00003999def test_main():
Guido van Rossumaabe0b32003-05-29 14:26:57 +00004000 weakref_segfault() # Must be first, somehow
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004001 wrapper_segfault()
Guido van Rossum9fc8a292002-05-24 21:40:08 +00004002 do_this_first()
Tim Peters2f93e282001-10-04 05:27:00 +00004003 class_docstrings()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004004 lists()
4005 dicts()
Tim Peters25786c02001-09-02 08:22:48 +00004006 dict_constructor()
Tim Peters5d2b77c2001-09-03 05:47:38 +00004007 test_dir()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004008 ints()
4009 longs()
4010 floats()
4011 complexes()
4012 spamlists()
4013 spamdicts()
4014 pydicts()
4015 pylists()
4016 metaclass()
4017 pymods()
4018 multi()
Guido van Rossumd32047f2002-11-25 21:38:52 +00004019 mro_disagreement()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004020 diamond()
Guido van Rossum9a818922002-11-14 19:50:14 +00004021 ex5()
4022 monotonicity()
4023 consistency_with_epg()
Guido van Rossum37202612001-08-09 19:45:21 +00004024 objects()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004025 slots()
Guido van Rossum8b056da2002-08-13 18:26:26 +00004026 slotspecials()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004027 dynamics()
4028 errors()
4029 classmethods()
Fred Drakef841aa62002-03-28 15:49:54 +00004030 classmethods_in_c()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004031 staticmethods()
Fred Drakef841aa62002-03-28 15:49:54 +00004032 staticmethods_in_c()
Tim Peters6d6c1a32001-08-02 04:15:00 +00004033 classic()
4034 compattr()
4035 newslot()
4036 altmro()
4037 overloading()
Guido van Rossumb5a136b2001-08-15 17:51:17 +00004038 methods()
Guido van Rossuma4ff6ab2001-08-15 23:57:59 +00004039 specials()
Guido van Rossum65d5d7f2001-08-17 21:27:53 +00004040 weakrefs()
Guido van Rossum8bce4ac2001-09-06 21:56:42 +00004041 properties()
Guido van Rossumc4a18802001-08-24 16:55:27 +00004042 supers()
Guido van Rossumcaa9f432001-08-30 20:06:08 +00004043 inherits()
Tim Peters808b94e2001-09-13 19:33:07 +00004044 keywords()
Tim Peters8fa45672001-09-13 21:01:29 +00004045 restricted()
Tim Peters0ab085c2001-09-14 00:25:33 +00004046 str_subclass_as_dict_key()
Guido van Rossumab3b0342001-09-18 20:38:53 +00004047 classic_comparisons()
Guido van Rossum0639f592001-09-18 21:06:04 +00004048 rich_comparisons()
Guido van Rossum8b9cc7e2001-09-20 21:49:53 +00004049 descrdoc()
Guido van Rossum5c294fb2001-09-25 03:43:42 +00004050 setclass()
Guido van Rossum6661be32001-10-26 04:26:12 +00004051 setdict()
Guido van Rossum3926a632001-09-25 16:25:58 +00004052 pickles()
Guido van Rossum6cef6d52001-09-28 18:13:29 +00004053 copies()
Guido van Rossum4bb1e362001-09-28 23:49:48 +00004054 binopoverride()
Guido van Rossum875eeaa2001-10-11 18:33:53 +00004055 subclasspropagation()
Tim Petersfc57ccb2001-10-12 02:38:24 +00004056 buffer_inherit()
Tim Petersc9933152001-10-16 20:18:24 +00004057 str_of_str_subclass()
Guido van Rossumc8e56452001-10-22 00:43:43 +00004058 kwdargs()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004059 recursive__call__()
Guido van Rossumed87ad82001-10-30 02:33:02 +00004060 delhook()
Guido van Rossumdbb53d92001-12-03 16:32:18 +00004061 hashinherit()
Guido van Rossum29d26062001-12-11 04:37:34 +00004062 strops()
Guido van Rossum2764a3a2001-12-28 21:39:03 +00004063 deepcopyrecursive()
Guido van Rossumd7035672002-03-12 20:43:31 +00004064 modules()
Walter Dörwalddbd2d252002-03-25 18:36:32 +00004065 dictproxyiterkeys()
4066 dictproxyitervalues()
4067 dictproxyiteritems()
Guido van Rossum8c842552002-03-14 23:05:54 +00004068 pickleslots()
Guido van Rossum8ace1ab2002-04-06 01:05:01 +00004069 funnynew()
Guido van Rossume8fc6402002-04-16 16:44:51 +00004070 imulbug()
Guido van Rossumd99b3e72002-04-18 00:27:33 +00004071 docdescriptor()
Guido van Rossuma48cb8f2002-06-06 17:53:03 +00004072 copy_setstate()
Guido van Rossum09638c12002-06-13 19:17:46 +00004073 slices()
Tim Peters2484aae2002-07-11 06:56:07 +00004074 subtype_resurrection()
Guido van Rossum2d702462002-08-06 21:28:28 +00004075 slottrash()
Neal Norwitzf9dd0f12002-08-13 17:16:49 +00004076 slotmultipleinheritance()
Guido van Rossum0f5f0b82002-08-09 16:11:37 +00004077 testrmul()
Guido van Rossum6e5680f2002-10-15 01:01:53 +00004078 testipow()
Michael W. Hudson586da8f2002-11-27 15:20:19 +00004079 test_mutable_bases()
4080 test_mutable_bases_with_failing_mro()
4081 test_mutable_bases_catch_mro_conflict()
Michael W. Hudson98bbc492002-11-26 14:47:27 +00004082 mutable_names()
Guido van Rossum613f24f2003-01-06 23:00:59 +00004083 subclass_right_op()
Guido van Rossum373c7412003-01-07 13:41:37 +00004084 dict_type_with_metaclass()
Guido van Rossumb6e5a0c2003-02-11 18:44:42 +00004085 meth_class_get()
Guido van Rossum03bc7d32003-02-12 03:32:58 +00004086 isinst_isclass()
Guido van Rossuma89d10e2003-02-12 03:58:38 +00004087 proxysuper()
Guido van Rossum52b27052003-04-15 20:05:10 +00004088 carloverre()
Raymond Hettinger2b6220d2003-06-29 15:44:07 +00004089 filefault()
Michael W. Hudsonb2c7de42003-08-15 13:07:47 +00004090 vicious_descriptor_nonsense()
Raymond Hettingerb67cc802005-03-03 16:45:19 +00004091 test_init()
Armin Rigoc6686b72005-11-07 08:38:00 +00004092 methodwrapper()
Armin Rigofd163f92005-12-29 15:59:19 +00004093 notimplemented()
Michael W. Hudson586da8f2002-11-27 15:20:19 +00004094
Guido van Rossuma56b42b2001-09-20 21:39:07 +00004095 if verbose: print "All OK"
Tim Peters6d6c1a32001-08-02 04:15:00 +00004096
Guido van Rossuma56b42b2001-09-20 21:39:07 +00004097if __name__ == "__main__":
4098 test_main()