Strip out wrong info from tracebacks on CPython 2.5

DasIch · Feb 12, 2011 · b7ccbec · b7ccbec
1 parent 07560e7
commit b7ccbec
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/attest/reporters.py b/attest/reporters.py
@@ -13,7 +13,8 @@
     abstractmethod = lambda x: x
 
 from . import statistics
-from .hook import ExpressionEvaluator, TestFailure
+from .hook import (ExpressionEvaluator, TestFailure, COMPILES_AST,
+                   AssertImportHook)
 
 
 class TestResult(object):
@@ -60,6 +61,11 @@ def raw_traceback(self):
 
         """
         tb = traceback.extract_tb(self.exc_info[2])
+        if not COMPILES_AST and AssertImportHook.enabled:
+            newtb = []
+            for filename, lineno, funcname, text in tb:
+                newtb.append((filename, 0, funcname, None))
+            tb = newtb
         clean = []
         thisfile = path.abspath(path.dirname(__file__))
         for item in tb: