make easier to navigate

Author: KRRT7

tests/scripts/end_to_end_test_async.py

@@ -1,12 +1,12 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="main.py",
+        file_path=Path("main.py"),
         min_improvement_x=0.1,
         coverage_expectations=[
             CoverageExpectation(
@@ -17,7 +17,7 @@ def run_test(expected_improvement_pct: int) -> bool:
         ],
     )
     cwd = (
-        pathlib.Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "async_e2e"
+        Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "async_e2e"
     ).resolve()
     return run_codeflash_command(cwd, config, expected_improvement_pct)
 

tests/scripts/end_to_end_test_benchmark_sort.py

@@ -1,13 +1,13 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
-    cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
+    cwd = (Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     config = TestConfig(
-        file_path=pathlib.Path("bubble_sort.py"),
+        file_path=Path("bubble_sort.py"),
         function_name="sorter",
         benchmarks_root=cwd / "tests" / "pytest" / "benchmarks",
         min_improvement_x=0.70,

tests/scripts/end_to_end_test_bubblesort_pytest.py

@@ -1,12 +1,13 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="bubble_sort.py",
+        file_path=Path("bubble_sort.py"),
         function_name="sorter",
         min_improvement_x=0.70,
         coverage_expectations=[
@@ -15,7 +16,7 @@ def run_test(expected_improvement_pct: int) -> bool:
             )
         ],
     )
-    cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
+    cwd = (Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     return run_codeflash_command(
         cwd, config, expected_improvement_pct, ['print("codeflash stdout: Sorting list")', 'print(f"result: {arr}")']
     )

tests/scripts/end_to_end_test_bubblesort_unittest.py

@@ -1,14 +1,14 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="bubble_sort.py", function_name="sorter", min_improvement_x=0.30
+        file_path=Path("bubble_sort.py"), function_name="sorter", min_improvement_x=0.30
     )
-    cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
+    cwd = (Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     return run_codeflash_command(cwd, config, expected_improvement_pct)
 
 

tests/scripts/end_to_end_test_coverage.py

@@ -1,19 +1,20 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="bubble_sort.py",
+        file_path=Path("bubble_sort.py"),
         function_name="sorter_one_level_depth",
         coverage_expectations=[
             CoverageExpectation(function_name="sorter_one_level_depth", expected_coverage=100.0, expected_lines=[2])
         ],
     )
     cwd = (
-        pathlib.Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "my-best-repo"
+        Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "my-best-repo"
     ).resolve()
     return run_codeflash_command(cwd, config, expected_improvement_pct)
 

tests/scripts/end_to_end_test_init_optimization.py

@@ -1,12 +1,12 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="remove_control_chars.py",
+        file_path=Path("remove_control_chars.py"),
         function_name="CharacterRemover.remove_control_characters",
         min_improvement_x=0.1,
         coverage_expectations=[
@@ -15,7 +15,7 @@ def run_test(expected_improvement_pct: int) -> bool:
             )
         ],
     )
-    cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
+    cwd = (Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     return run_codeflash_command(cwd, config, expected_improvement_pct)
 
 

tests/scripts/end_to_end_test_topological_sort_worktree.py

@@ -1,12 +1,12 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
     config = TestConfig(
-        file_path="topological_sort.py",
+        file_path=Path("topological_sort.py"),
         function_name="Graph.topologicalSort",
         min_improvement_x=0.05,
         use_worktree=True,
@@ -17,9 +17,9 @@ def run_test(expected_improvement_pct: int) -> bool:
                 expected_lines=[25, 26, 27, 28, 29, 30, 31],
             )
         ],
-        expected_unit_tests=13,
+        expected_unit_tests=1,
     )
-    cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
+    cwd = (Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     return_var = run_codeflash_command(cwd, config, expected_improvement_pct)
     return return_var
 

tests/scripts/end_to_end_test_tracer_replay.py

@@ -1,7 +1,7 @@
 import os
-import pathlib
+from pathlib import Path
 
-from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
+from tests.scripts.end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codeflash_command, run_with_retries
 
 
 def run_test(expected_improvement_pct: int) -> bool:
@@ -14,7 +14,7 @@ def run_test(expected_improvement_pct: int) -> bool:
         ],
     )
     cwd = (
-        pathlib.Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "simple_tracer_e2e"
+       Path(__file__).parent.parent.parent / "code_to_optimize" / "code_directories" / "simple_tracer_e2e"
     ).resolve()
     return run_codeflash_command(cwd, config, expected_improvement_pct)