greentest.py: fix AttributeError in modulename function when using python 2.4/2.5.

3586466d · Ralf Schmitt · b2dac472 · 3586466d
Commit 3586466d authored Oct 14, 2010 by Ralf Schmitt
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

greentest/greentest.py greentest/greentest.py +4 -1

No files found.
--- a/greentest/greentest.py
+++ b/greentest/greentest.py
@@ -95,7 +95,10 @@ class TestCase(BaseTestCase):
    @property
    def modulename(self):
        test_method = getattr(self, self.testname)
-        return test_method.__func__.func_code.co_filename
+        try:
+            return test_method.__func__.func_code.co_filename
+        except AttributeError:
+            return test_method.im_func.func_code.co_filename

    @property
    def fullname(self):