Merge pull request #825 from codeflash-ai/concolic-error

CF-753 catch all exceptions for concolic cleanup

Author: misrasaurabh1

codeflash/code_utils/concolic_utils.py

@@ -70,10 +70,11 @@ def __init__(self) -> None:
 
 def clean_concolic_tests(test_suite_code: str) -> str:
     try:
-        can_parse = True
         tree = ast.parse(test_suite_code)
-    except SyntaxError:
+        can_parse = True
+    except Exception:
         can_parse = False
+        tree = None
 
     if not can_parse:
         return AssertCleanup().transform_asserts(test_suite_code)
@@ -87,7 +88,6 @@ def clean_concolic_tests(test_suite_code: str) -> str:
                         new_body.append(ast.Expr(value=stmt.test.left))
                     else:
                         new_body.append(stmt)
-
                 else:
                     new_body.append(stmt)
             node.body = new_body

codeflash/tracer.py

@@ -103,6 +103,11 @@ def main(args: Namespace | None = None) -> ArgumentParser:
             replay_test_paths = []
             if parsed_args.module and unknown_args[0] == "pytest":
                 pytest_splits, test_paths = pytest_split(unknown_args[1:])
+                if pytest_splits is None or test_paths is None:
+                    console.print(f"❌ Could not find test files in the specified paths: {unknown_args[1:]}")
+                    console.print(f"Current working directory: {Path.cwd()}")
+                    console.print("Please ensure the test directory exists and contains test files.")
+                    sys.exit(1)
 
             if len(pytest_splits) > 1:
                 processes = []