Port test_class to unittest. Patch #1671298.

This commit is contained in:
Georg Brandl 2007-08-24 19:33:53 +00:00
parent ff9b963387
commit c325556505
2 changed files with 498 additions and 388 deletions

View file

@ -1,101 +0,0 @@
test_class
__init__: ()
__coerce__: (1,)
__add__: (1,)
__coerce__: (1,)
__radd__: (1,)
__coerce__: (1,)
__sub__: (1,)
__coerce__: (1,)
__rsub__: (1,)
__coerce__: (1,)
__mul__: (1,)
__coerce__: (1,)
__rmul__: (1,)
__coerce__: (1,)
__div__: (1,)
__coerce__: (1,)
__rdiv__: (1,)
__coerce__: (1,)
__mod__: (1,)
__coerce__: (1,)
__rmod__: (1,)
__coerce__: (1,)
__divmod__: (1,)
__coerce__: (1,)
__rdivmod__: (1,)
__coerce__: (1,)
__pow__: (1,)
__coerce__: (1,)
__rpow__: (1,)
__coerce__: (1,)
__rshift__: (1,)
__coerce__: (1,)
__rrshift__: (1,)
__coerce__: (1,)
__lshift__: (1,)
__coerce__: (1,)
__rlshift__: (1,)
__coerce__: (1,)
__and__: (1,)
__coerce__: (1,)
__rand__: (1,)
__coerce__: (1,)
__or__: (1,)
__coerce__: (1,)
__ror__: (1,)
__coerce__: (1,)
__xor__: (1,)
__coerce__: (1,)
__rxor__: (1,)
__contains__: (1,)
__getitem__: (1,)
__setitem__: (1, 1)
__delitem__: (1,)
__getslice__: (0, 42)
__setslice__: (0, 42, 'The Answer')
__delslice__: (0, 42)
__getitem__: (slice(2, 1024, 10),)
__setitem__: (slice(2, 1024, 10), 'A lot')
__delitem__: (slice(2, 1024, 10),)
__getitem__: ((slice(None, 42, None), Ellipsis, slice(None, 24, None), 24, 100),)
__setitem__: ((slice(None, 42, None), Ellipsis, slice(None, 24, None), 24, 100), 'Strange')
__delitem__: ((slice(None, 42, None), Ellipsis, slice(None, 24, None), 24, 100),)
__getitem__: (slice(0, 42, None),)
__setitem__: (slice(0, 42, None), 'The Answer')
__delitem__: (slice(0, 42, None),)
__neg__: ()
__pos__: ()
__abs__: ()
__int__: ()
__long__: ()
__float__: ()
__oct__: ()
__hex__: ()
__hash__: ()
__repr__: ()
__str__: ()
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__coerce__: (1,)
__cmp__: (1,)
__del__: ()
__getattr__: ('spam',)
__setattr__: ('eggs', 'spam, spam, spam and ham')
__delattr__: ('cardinal',)

View file

@ -1,6 +1,9 @@
"Test the functionality of Python classes implementing operators." "Test the functionality of Python classes implementing operators."
from test.test_support import TestFailed import unittest
import sys
from test import test_support
testmeths = [ testmeths = [
@ -64,55 +67,62 @@ testmeths = [
# "setattr", # "setattr",
# "delattr", # "delattr",
callLst = []
def trackCall(f):
def track(*args, **kwargs):
callLst.append((f.__name__, args))
return f(*args, **kwargs)
return track
class AllTests: class AllTests:
trackCall = trackCall
@trackCall
def __coerce__(self, *args): def __coerce__(self, *args):
print "__coerce__:", args
return (self,) + args return (self,) + args
@trackCall
def __hash__(self, *args): def __hash__(self, *args):
print "__hash__:", args
return hash(id(self)) return hash(id(self))
@trackCall
def __str__(self, *args): def __str__(self, *args):
print "__str__:", args
return "AllTests" return "AllTests"
@trackCall
def __repr__(self, *args): def __repr__(self, *args):
print "__repr__:", args
return "AllTests" return "AllTests"
@trackCall
def __int__(self, *args): def __int__(self, *args):
print "__int__:", args
return 1 return 1
@trackCall
def __float__(self, *args): def __float__(self, *args):
print "__float__:", args
return 1.0 return 1.0
@trackCall
def __long__(self, *args): def __long__(self, *args):
print "__long__:", args
return 1L return 1L
@trackCall
def __oct__(self, *args): def __oct__(self, *args):
print "__oct__:", args
return '01' return '01'
@trackCall
def __hex__(self, *args): def __hex__(self, *args):
print "__hex__:", args
return '0x1' return '0x1'
@trackCall
def __cmp__(self, *args): def __cmp__(self, *args):
print "__cmp__:", args
return 0 return 0
def __del__(self, *args): # Synthesize all the other AllTests methods from the names in testmeths.
print "__del__:", args
# Synthesize AllTests methods from the names in testmeths.
method_template = """\ method_template = """\
@trackCall
def __%(method)s__(self, *args): def __%(method)s__(self, *args):
print "__%(method)s__:", args pass
""" """
for method in testmeths: for method in testmeths:
@ -120,293 +130,494 @@ for method in testmeths:
del method, method_template del method, method_template
# this also tests __init__ of course. class ClassTests(unittest.TestCase):
testme = AllTests() def setUp(self):
callLst[:] = []
# Binary operations def assertCallStack(self, expected_calls):
actualCallList = callLst[:] # need to copy because the comparison below will add
# additional calls to callLst
if expected_calls != actualCallList:
self.fail("Expected call list:\n %s\ndoes not match actual call list\n %s" %
(expected_calls, actualCallList))
testme + 1 def testInit(self):
1 + testme foo = AllTests()
self.assertCallStack([("__init__", (foo,))])
testme - 1 def testBinaryOps(self):
1 - testme testme = AllTests()
# Binary operations
testme * 1 callLst[:] = []
1 * testme testme + 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__add__", (testme, 1))])
if 1/2 == 0: callLst[:] = []
testme / 1 1 + testme
1 / testme self.assertCallStack([("__coerce__", (testme, 1)), ("__radd__", (testme, 1))])
else:
# True division is in effect, so "/" doesn't map to __div__ etc; but
# the canned expected-output file requires that __div__ etc get called.
testme.__coerce__(1)
testme.__div__(1)
testme.__coerce__(1)
testme.__rdiv__(1)
testme % 1 callLst[:] = []
1 % testme testme - 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__sub__", (testme, 1))])
divmod(testme,1) callLst[:] = []
divmod(1, testme) 1 - testme
self.assertCallStack([("__coerce__", (testme, 1)), ("__rsub__", (testme, 1))])
testme ** 1 callLst[:] = []
1 ** testme testme * 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__mul__", (testme, 1))])
testme >> 1 callLst[:] = []
1 >> testme 1 * testme
self.assertCallStack([("__coerce__", (testme, 1)), ("__rmul__", (testme, 1))])
testme << 1 if 1/2 == 0:
1 << testme callLst[:] = []
testme / 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__div__", (testme, 1))])
testme & 1
1 & testme
testme | 1 callLst[:] = []
1 | testme 1 / testme
self.assertCallStack([("__coerce__", (testme, 1)), ("__rdiv__", (testme, 1))])
testme ^ 1 callLst[:] = []
1 ^ testme testme % 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__mod__", (testme, 1))])
callLst[:] = []
1 % testme
self.assertCallStack([("__coerce__", (testme, 1)), ("__rmod__", (testme, 1))])
# List/dict operations
class Empty: pass callLst[:] = []
divmod(testme,1)
self.assertCallStack([("__coerce__", (testme, 1)), ("__divmod__", (testme, 1))])
try: callLst[:] = []
1 in Empty() divmod(1, testme)
print 'failed, should have raised TypeError' self.assertCallStack([("__coerce__", (testme, 1)), ("__rdivmod__", (testme, 1))])
except TypeError:
pass
1 in testme callLst[:] = []
testme ** 1
self.assertCallStack([("__coerce__", (testme, 1)), ("__pow__", (testme, 1))])
testme[1] callLst[:] = []
testme[1] = 1 1 ** testme
del testme[1] self.assertCallStack([("__coerce__", (testme, 1)), ("__rpow__", (testme, 1))])
testme[:42] callLst[:] = []
testme[:42] = "The Answer" testme >> 1
del testme[:42] self.assertCallStack([("__coerce__", (testme, 1)), ("__rshift__", (testme, 1))])
testme[2:1024:10] callLst[:] = []
testme[2:1024:10] = "A lot" 1 >> testme
del testme[2:1024:10] self.assertCallStack([("__coerce__", (testme, 1)), ("__rrshift__", (testme, 1))])
testme[:42, ..., :24:, 24, 100] callLst[:] = []
testme[:42, ..., :24:, 24, 100] = "Strange" testme << 1
del testme[:42, ..., :24:, 24, 100] self.assertCallStack([("__coerce__", (testme, 1)), ("__lshift__", (testme, 1))])
callLst[:] = []
# Now remove the slice hooks to see if converting normal slices to slice 1 << testme
# object works. self.assertCallStack([("__coerce__", (testme, 1)), ("__rlshift__", (testme, 1))])
del AllTests.__getslice__ callLst[:] = []
del AllTests.__setslice__ testme & 1
del AllTests.__delslice__ self.assertCallStack([("__coerce__", (testme, 1)), ("__and__", (testme, 1))])
import sys callLst[:] = []
if sys.platform[:4] != 'java': 1 & testme
testme[:42] self.assertCallStack([("__coerce__", (testme, 1)), ("__rand__", (testme, 1))])
testme[:42] = "The Answer"
del testme[:42] callLst[:] = []
else: testme | 1
# This works under Jython, but the actual slice values are self.assertCallStack([("__coerce__", (testme, 1)), ("__or__", (testme, 1))])
# different.
print "__getitem__: (slice(0, 42, None),)" callLst[:] = []
print "__setitem__: (slice(0, 42, None), 'The Answer')" 1 | testme
print "__delitem__: (slice(0, 42, None),)" self.assertCallStack([("__coerce__", (testme, 1)), ("__ror__", (testme, 1))])
# Unary operations callLst[:] = []
testme ^ 1
-testme self.assertCallStack([("__coerce__", (testme, 1)), ("__xor__", (testme, 1))])
+testme
abs(testme) callLst[:] = []
int(testme) 1 ^ testme
long(testme) self.assertCallStack([("__coerce__", (testme, 1)), ("__rxor__", (testme, 1))])
float(testme)
oct(testme) def testListAndDictOps(self):
hex(testme) testme = AllTests()
# And the rest... # List/dict operations
hash(testme) class Empty: pass
repr(testme)
str(testme) try:
1 in Empty()
testme == 1 self.fail('failed, should have raised TypeError')
testme < 1 except TypeError:
testme > 1 pass
testme <> 1
testme != 1 callLst[:] = []
1 == testme 1 in testme
1 < testme self.assertCallStack([('__contains__', (testme, 1))])
1 > testme
1 <> testme callLst[:] = []
1 != testme testme[1]
self.assertCallStack([('__getitem__', (testme, 1))])
# This test has to be last (duh.)
callLst[:] = []
del testme testme[1] = 1
if sys.platform[:4] == 'java': self.assertCallStack([('__setitem__', (testme, 1, 1))])
import java
java.lang.System.gc() callLst[:] = []
del testme[1]
# Interfering tests self.assertCallStack([('__delitem__', (testme, 1))])
class ExtraTests: callLst[:] = []
def __getattr__(self, *args): testme[:42]
print "__getattr__:", args self.assertCallStack([('__getslice__', (testme, 0, 42))])
return "SomeVal"
callLst[:] = []
def __setattr__(self, *args): testme[:42] = "The Answer"
print "__setattr__:", args self.assertCallStack([('__setslice__', (testme, 0, 42, "The Answer"))])
def __delattr__(self, *args): callLst[:] = []
print "__delattr__:", args del testme[:42]
self.assertCallStack([('__delslice__', (testme, 0, 42))])
testme = ExtraTests()
testme.spam callLst[:] = []
testme.eggs = "spam, spam, spam and ham" testme[2:1024:10]
del testme.cardinal self.assertCallStack([('__getitem__', (testme, slice(2, 1024, 10)))])
callLst[:] = []
# return values of some method are type-checked testme[2:1024:10] = "A lot"
class BadTypeClass: self.assertCallStack([('__setitem__', (testme, slice(2, 1024, 10),
def __int__(self): "A lot"))])
return None callLst[:] = []
__float__ = __int__ del testme[2:1024:10]
__long__ = __int__ self.assertCallStack([('__delitem__', (testme, slice(2, 1024, 10)))])
__str__ = __int__
__repr__ = __int__ callLst[:] = []
__oct__ = __int__ testme[:42, ..., :24:, 24, 100]
__hex__ = __int__ self.assertCallStack([('__getitem__', (testme, (slice(None, 42, None),
Ellipsis,
def check_exc(stmt, exception): slice(None, 24, None),
"""Raise TestFailed if executing 'stmt' does not raise 'exception' 24, 100)))])
""" callLst[:] = []
try: testme[:42, ..., :24:, 24, 100] = "Strange"
exec stmt self.assertCallStack([('__setitem__', (testme, (slice(None, 42, None),
except exception: Ellipsis,
pass slice(None, 24, None),
else: 24, 100), "Strange"))])
raise TestFailed, "%s should raise %s" % (stmt, exception) callLst[:] = []
del testme[:42, ..., :24:, 24, 100]
check_exc("int(BadTypeClass())", TypeError) self.assertCallStack([('__delitem__', (testme, (slice(None, 42, None),
check_exc("float(BadTypeClass())", TypeError) Ellipsis,
check_exc("long(BadTypeClass())", TypeError) slice(None, 24, None),
check_exc("str(BadTypeClass())", TypeError) 24, 100)))])
check_exc("repr(BadTypeClass())", TypeError)
check_exc("oct(BadTypeClass())", TypeError) # Now remove the slice hooks to see if converting normal slices to
check_exc("hex(BadTypeClass())", TypeError) # slice object works.
# mixing up ints and longs is okay getslice = AllTests.__getslice__
class IntLongMixClass: del AllTests.__getslice__
def __int__(self): setslice = AllTests.__setslice__
return 0L del AllTests.__setslice__
delslice = AllTests.__delslice__
def __long__(self): del AllTests.__delslice__
return 0
# XXX when using new-style classes the slice testme[:42] produces
try: # slice(None, 42, None) instead of slice(0, 42, None). py3k will have
int(IntLongMixClass()) # to change this test.
except TypeError: callLst[:] = []
raise TestFailed, "TypeError should not be raised" testme[:42]
self.assertCallStack([('__getitem__', (testme, slice(0, 42, None)))])
try:
long(IntLongMixClass()) callLst[:] = []
except TypeError: testme[:42] = "The Answer"
raise TestFailed, "TypeError should not be raised" self.assertCallStack([('__setitem__', (testme, slice(0, 42, None),
"The Answer"))])
callLst[:] = []
# Test correct errors from hash() on objects with comparisons but no __hash__ del testme[:42]
self.assertCallStack([('__delitem__', (testme, slice(0, 42, None)))])
class C0:
pass # Restore the slice methods, or the tests will fail with regrtest -R.
AllTests.__getslice__ = getslice
hash(C0()) # This should work; the next two should raise TypeError AllTests.__setslice__ = setslice
AllTests.__delslice__ = delslice
class C1:
def __cmp__(self, other): return 0
def testUnaryOps(self):
check_exc("hash(C1())", TypeError) testme = AllTests()
class C2: callLst[:] = []
def __eq__(self, other): return 1 -testme
self.assertCallStack([('__neg__', (testme,))])
check_exc("hash(C2())", TypeError) callLst[:] = []
+testme
# Test for SF bug 532646 self.assertCallStack([('__pos__', (testme,))])
callLst[:] = []
class A: abs(testme)
pass self.assertCallStack([('__abs__', (testme,))])
A.__call__ = A() callLst[:] = []
a = A() int(testme)
try: self.assertCallStack([('__int__', (testme,))])
a() # This should not segfault callLst[:] = []
except RuntimeError: long(testme)
pass self.assertCallStack([('__long__', (testme,))])
else: callLst[:] = []
raise TestFailed, "how could this not have overflowed the stack?" float(testme)
self.assertCallStack([('__float__', (testme,))])
callLst[:] = []
# Tests for exceptions raised in instance_getattr2(). oct(testme)
self.assertCallStack([('__oct__', (testme,))])
def booh(self): callLst[:] = []
raise AttributeError, "booh" hex(testme)
self.assertCallStack([('__hex__', (testme,))])
class A:
a = property(booh)
try: def testMisc(self):
A().a # Raised AttributeError: A instance has no attribute 'a' testme = AllTests()
except AttributeError, x:
if str(x) != "booh": callLst[:] = []
print "attribute error for A().a got masked:", str(x) hash(testme)
self.assertCallStack([('__hash__', (testme,))])
class E:
__eq__ = property(booh) callLst[:] = []
E() == E() # In debug mode, caused a C-level assert() to fail repr(testme)
self.assertCallStack([('__repr__', (testme,))])
class I:
__init__ = property(booh) callLst[:] = []
try: str(testme)
I() # In debug mode, printed XXX undetected error and raises AttributeError self.assertCallStack([('__str__', (testme,))])
except AttributeError, x:
pass callLst[:] = []
else: testme == 1
print "attribute error for I.__init__ got masked" self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (testme, 1))])
callLst[:] = []
# Test comparison and hash of methods testme < 1
class A: self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (testme, 1))])
def __init__(self, x):
self.x = x callLst[:] = []
def f(self): testme > 1
pass self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (testme, 1))])
def g(self):
pass callLst[:] = []
def __eq__(self, other): testme <> 1 # XXX kill this in py3k
return self.x == other.x self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (testme, 1))])
def __hash__(self):
return self.x callLst[:] = []
class B(A): testme != 1
pass self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (testme, 1))])
a1 = A(1) callLst[:] = []
a2 = A(2) 1 == testme
assert a1.f == a1.f self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (1, testme))])
assert a1.f != a2.f
assert a1.f != a1.g callLst[:] = []
assert a1.f == A(1).f 1 < testme
assert hash(a1.f) == hash(a1.f) self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (1, testme))])
assert hash(a1.f) == hash(A(1).f)
callLst[:] = []
assert A.f != a1.f 1 > testme
assert A.f != A.g self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (1, testme))])
assert B.f == A.f
assert hash(B.f) == hash(A.f) callLst[:] = []
1 <> testme
# the following triggers a SystemError in 2.4 self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (1, testme))])
a = A(hash(A.f.im_func)^(-1))
hash(a.f) callLst[:] = []
1 != testme
self.assertCallStack([("__coerce__", (testme, 1)), ('__cmp__', (1, testme))])
def testGetSetAndDel(self):
# Interfering tests
class ExtraTests(AllTests):
@trackCall
def __getattr__(self, *args):
return "SomeVal"
@trackCall
def __setattr__(self, *args):
pass
@trackCall
def __delattr__(self, *args):
pass
testme = ExtraTests()
callLst[:] = []
testme.spam
self.assertCallStack([('__getattr__', (testme, "spam"))])
callLst[:] = []
testme.eggs = "spam, spam, spam and ham"
self.assertCallStack([('__setattr__', (testme, "eggs",
"spam, spam, spam and ham"))])
callLst[:] = []
del testme.cardinal
self.assertCallStack([('__delattr__', (testme, "cardinal"))])
def testDel(self):
x = []
class DelTest:
def __del__(self):
x.append("crab people, crab people")
testme = DelTest()
del testme
import gc
gc.collect()
self.assertEquals(["crab people, crab people"], x)
def testBadTypeReturned(self):
# return values of some method are type-checked
class BadTypeClass:
def __int__(self):
return None
__float__ = __int__
__long__ = __int__
__str__ = __int__
__repr__ = __int__
__oct__ = __int__
__hex__ = __int__
for f in [int, float, long, str, repr, oct, hex]:
self.assertRaises(TypeError, f, BadTypeClass())
def testMixIntsAndLongs(self):
# mixing up ints and longs is okay
class IntLongMixClass:
@trackCall
def __int__(self):
return 42L
@trackCall
def __long__(self):
return 64
mixIntAndLong = IntLongMixClass()
callLst[:] = []
as_int = int(mixIntAndLong)
self.assertEquals(type(as_int), long)
self.assertEquals(as_int, 42L)
self.assertCallStack([('__int__', (mixIntAndLong,))])
callLst[:] = []
as_long = long(mixIntAndLong)
self.assertEquals(type(as_long), int)
self.assertEquals(as_long, 64)
self.assertCallStack([('__long__', (mixIntAndLong,))])
def testHashStuff(self):
# Test correct errors from hash() on objects with comparisons but
# no __hash__
class C0:
pass
hash(C0()) # This should work; the next two should raise TypeError
class C1:
def __cmp__(self, other): return 0
self.assertRaises(TypeError, hash, C1())
class C2:
def __eq__(self, other): return 1
self.assertRaises(TypeError, hash, C2())
def testSFBug532646(self):
# Test for SF bug 532646
class A:
pass
A.__call__ = A()
a = A()
try:
a() # This should not segfault
except RuntimeError:
pass
else:
self.fail("Failed to raise RuntimeError")
def testForExceptionsRaisedInInstanceGetattr2(self):
# Tests for exceptions raised in instance_getattr2().
def booh(self):
raise AttributeError("booh")
class A:
a = property(booh)
try:
A().a # Raised AttributeError: A instance has no attribute 'a'
except AttributeError, x:
if str(x) != "booh":
self.fail("attribute error for A().a got masked: %s" % x)
class E:
__eq__ = property(booh)
E() == E() # In debug mode, caused a C-level assert() to fail
class I:
__init__ = property(booh)
try:
# In debug mode, printed XXX undetected error and
# raises AttributeError
I()
except AttributeError, x:
pass
else:
self.fail("attribute error for I.__init__ got masked")
def testHashComparisonOfMethods(self):
# Test comparison and hash of methods
class A:
def __init__(self, x):
self.x = x
def f(self):
pass
def g(self):
pass
def __eq__(self, other):
return self.x == other.x
def __hash__(self):
return self.x
class B(A):
pass
a1 = A(1)
a2 = A(2)
self.assertEquals(a1.f, a1.f)
self.assertNotEquals(a1.f, a2.f)
self.assertNotEquals(a1.f, a1.g)
self.assertEquals(a1.f, A(1).f)
self.assertEquals(hash(a1.f), hash(a1.f))
self.assertEquals(hash(a1.f), hash(A(1).f))
self.assertNotEquals(A.f, a1.f)
self.assertNotEquals(A.f, A.g)
self.assertEquals(B.f, A.f)
self.assertEquals(hash(B.f), hash(A.f))
# the following triggers a SystemError in 2.4
a = A(hash(A.f.im_func)^(-1))
hash(a.f)
def test_main():
test_support.run_unittest(ClassTests)
if __name__=='__main__':
test_main()