From 9bedcd25c823561f9a499f47fabd5f7c436be6e6 Mon Sep 17 00:00:00 2001 From: ankitjavalkar Date: Fri, 17 Feb 2017 17:33:25 +0530 Subject: Remove filename strings from error traceback in python assertion --- yaksh/python_assertion_evaluator.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'yaksh') diff --git a/yaksh/python_assertion_evaluator.py b/yaksh/python_assertion_evaluator.py index 77efbc1..c8f2dd0 100644 --- a/yaksh/python_assertion_evaluator.py +++ b/yaksh/python_assertion_evaluator.py @@ -83,12 +83,15 @@ class PythonAssertionEvaluator(BaseEvaluator): 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) + line = re.sub(r'File\s+".*?",\s+line', + 'File , line', + line + ) + stripped_tb_lines.append(line) stripped_tb = '\n'.join(stripped_tb_lines[-10::]) err = "Expected Test Case:\n{0}\n" \ - "Error Traceback- {1} {2} in:\n {3}\n{4}".format( + "Error Traceback - {1} {2} in:\n {3}\n{4}".format( self.test_case, type.__name__, str(value), -- cgit