shorten error string, more enhancements needed

Author: Codeflash Bot

codeflash/code_utils/code_utils.py

@@ -362,6 +362,10 @@ def exit_with_message(message: str, *, error_on_exit: bool = False) -> None:
     sys.exit(1 if error_on_exit else 0)
 
 
+def shorten_pytest_error(pytest_error_string: str) -> str:
+    return "\n".join(re.findall(r"^[E>] +(.*)$", pytest_error_string, re.MULTILINE))
+
+
 def extract_unique_errors(pytest_output: str) -> set[str]:
     unique_errors = set()
 

codeflash/verification/equivalence.py

@@ -4,6 +4,7 @@
 from typing import TYPE_CHECKING
 
 from codeflash.cli_cmds.console import logger
+from codeflash.code_utils.code_utils import shorten_pytest_error
 from codeflash.models.models import TestDiff, TestDiffScope, TestResults, TestType, VerificationType
 from codeflash.verification.comparator import comparator
 
@@ -57,12 +58,15 @@ def compare_test_results(original_results: TestResults, candidate_results: TestR
             if candidate_test_failures
             else ""
         )
+        if cdd_pytest_error:
+            cdd_pytest_error = shorten_pytest_error(cdd_pytest_error)
         original_pytest_error = (
             original_test_failures.get(original_test_result.id.test_fn_qualified_name(), "")
             if original_test_failures
             else ""
         )
-
+        if original_pytest_error:
+            original_pytest_error = shorten_pytest_error(original_pytest_error)
         test_src_code = original_test_result.id.get_src_code(original_test_result.file_name)
         test_diff = TestDiff(
             scope=TestDiffScope.RETURN_VALUE,