diff options
Diffstat (limited to 'yaksh')
| -rw-r--r-- | yaksh/python_assertion_evaluator.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/yaksh/python_assertion_evaluator.py b/yaksh/python_assertion_evaluator.py index 11fa101..9e1a6ac 100644 --- a/yaksh/python_assertion_evaluator.py +++ b/yaksh/python_assertion_evaluator.py @@ -2,6 +2,7 @@ import sys import traceback import os +import re from os.path import join import importlib @@ -84,9 +85,17 @@ class PythonAssertionEvaluator(BaseEvaluator): ) except TimeoutException: raise - except Exception: + except RecursionError: msg = traceback.format_exc(limit=0) - err = "Error in Test case: {0}".format(msg) + err = "Error Traceback: {0}".format(msg) + except Exception: + err_tb_lines = traceback.format_exc().splitlines() + stripped_tb_lines = [] + for line in err_tb_lines: + if '.py' not in line: + stripped_tb_lines.append(line) + stripped_tb = '\n'.join(stripped_tb_lines) + err = "Error Traceback:\n{0}".format(stripped_tb) else: success = True err = None |