pass

Author: KRRT7

codeflash/code_utils/instrument_existing_tests.py

@@ -213,7 +213,7 @@ def visit_FunctionDef(self, node: ast.FunctionDef, test_class_name: str | None =
                                     args=[
                                         ast.JoinedStr(
                                             values=[
-                                                ast.Constant(value=get_run_tmp_file("test_return_values_").as_posix()),
+                                                ast.Constant(value=get_run_tmp_file("test_return_values_").resolve().as_posix()),
                                                 ast.FormattedValue(
                                                     value=ast.Name(id="codeflash_iteration", ctx=ast.Load()),
                                                     conversion=-1,

tests/test_instrument_all_and_run.py

@@ -122,7 +122,7 @@ def test_sort():
         assert new_test is not None
         assert new_test.replace('"', "'") == expected.format(
             module_path="code_to_optimize.tests.pytest.test_perfinjector_bubble_sort_results_temp",
-            tmp_dir_path=get_run_tmp_file(Path("test_return_values")),
+            tmp_dir_path=get_run_tmp_file("test_return_values").as_posix(),
         ).replace('"', "'")
 
         with test_path.open("w") as f:
@@ -300,16 +300,16 @@ def test_sort():
     fto = FunctionToOptimize(
         function_name="sorter", parents=[FunctionParent(name="BubbleSorter", type="ClassDef")], file_path=Path(fto_path)
     )
-    with tempfile.NamedTemporaryFile(mode="w") as f:
-        f.write(code)
-        f.flush()
+    with tempfile.TemporaryDirectory() as tempdir:
+        temp_test_path = Path(tempdir) / "temp_test.py"
+        temp_test_path.write_text(code)
 
         success, new_test = inject_profiling_into_existing_test(
-            Path(f.name), [CodePosition(7, 13), CodePosition(12, 13)], fto, Path(f.name).parent, "pytest"
+            temp_test_path, [CodePosition(7, 13), CodePosition(12, 13)], fto, temp_test_path.parent, "pytest"
         )
     assert success
     assert new_test.replace('"', "'") == expected.format(
-        module_path=Path(f.name).name, tmp_dir_path=get_run_tmp_file(Path("test_return_values"))
+        module_path="temp_test", tmp_dir_path=get_run_tmp_file("test_return_values").as_posix()
     ).replace('"', "'")
     tests_root = (Path(__file__).parent.resolve() / "../code_to_optimize/tests/pytest/").resolve()
     test_path = tests_root / "test_class_method_behavior_results_temp.py"
@@ -319,7 +319,7 @@ def test_sort():
     try:
         new_test = expected.format(
             module_path="code_to_optimize.tests.pytest.test_class_method_behavior_results_temp",
-            tmp_dir_path=get_run_tmp_file(Path("test_return_values")),
+            tmp_dir_path=get_run_tmp_file("test_return_values").as_posix(),
         )
 
         with test_path.open("w") as f: