use correct attribute to find the instance of a bound method

--HG--
branch : trunk
This commit is contained in:
Benjamin Peterson 2009-08-29 16:07:48 -05:00
parent a051eb1a05
commit ee86950af4
1 changed files with 6 additions and 1 deletions

View File

@ -2,13 +2,18 @@ import py
from py.test.collect import Function from py.test.collect import Function
if py.std.sys.version_info > (3, 0):
_self = "__self__"
else:
_self = "im_self"
class TestCaseUnit(Function): class TestCaseUnit(Function):
""" compatibility Unit executor for TestCase methods """ compatibility Unit executor for TestCase methods
honouring setUp and tearDown semantics. honouring setUp and tearDown semantics.
""" """
def runtest(self, _deprecated=None): def runtest(self, _deprecated=None):
boundmethod = self.obj boundmethod = self.obj
instance = boundmethod.im_self instance = getattr(boundmethod, _self)
instance.setUp() instance.setUp()
try: try:
boundmethod() boundmethod()