pass 3

Author: KRRT7

tests/test_instrument_codeflash_capture.py

@@ -22,7 +22,7 @@ def target_function(self):
 
 class MyClass:
 
-    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=True)
+    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file("test_return_values").as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=True)
     def __init__(self):
         self.x = 1
 
@@ -86,7 +86,7 @@ def target_function(self):
 
 class MyClass(ParentClass):
 
-    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=True)
+    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=True)
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
@@ -128,7 +128,7 @@ def helper(self):
 
 class MyClass:
 
-    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=True)
+    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=True)
     def __init__(self):
         self.x = 1
 
@@ -184,7 +184,7 @@ def helper(self):
 
 class MyClass:
 
-    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=True)
+    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=True)
     def __init__(self):
         self.x = 1
 
@@ -197,7 +197,7 @@ def target_function(self):
 
 class HelperClass:
 
-    @codeflash_capture(function_name='HelperClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=False)
+    @codeflash_capture(function_name='HelperClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=False)
     def __init__(self):
         self.y = 1
 
@@ -271,7 +271,7 @@ def another_helper(self):
 
 class MyClass:
 
-    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=True)
+    @codeflash_capture(function_name='MyClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=True)
     def __init__(self):
         self.x = 1
 
@@ -289,7 +289,7 @@ def target_function(self):
 
 class HelperClass1:
 
-    @codeflash_capture(function_name='HelperClass1.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=False)
+    @codeflash_capture(function_name='HelperClass1.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=False)
     def __init__(self):
         self.y = 1
 
@@ -304,7 +304,7 @@ def helper1(self):
 
 class HelperClass2:
 
-    @codeflash_capture(function_name='HelperClass2.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=False)
+    @codeflash_capture(function_name='HelperClass2.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=False)
     def __init__(self):
         self.z = 2
 
@@ -313,7 +313,7 @@ def helper2(self):
 
 class AnotherHelperClass:
 
-    @codeflash_capture(function_name='AnotherHelperClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values"))!s}', tests_root='{test_path.parent!s}', is_fto=False)
+    @codeflash_capture(function_name='AnotherHelperClass.__init__', tmp_dir_path='{get_run_tmp_file(Path("test_return_values")).as_posix()}', tests_root='{test_path.parent.as_posix()}', is_fto=False)
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 

tests/test_instrument_tests.py

@@ -266,20 +266,21 @@ def test_prepare_image_for_yolo():
         assert compare_results(return_val_1, ret)
     codeflash_con.close()
 """
-    with tempfile.NamedTemporaryFile(mode="w") as f:
-        f.write(code)
-        f.flush()
-        func = FunctionToOptimize(function_name="prepare_image_for_yolo", parents=[], file_path=Path("module.py"))
+    with tempfile.TemporaryDirectory() as tempdir:
+        temp_file_path = Path(tempdir) / "temp_test.py"
+        with temp_file_path.open("w") as f:
+            f.write(code)
+        func = FunctionToOptimize(function_name="prepare_image_for_yolo", parents=[], file_path=temp_file_path)
         original_cwd = Path.cwd()
         run_cwd = Path(__file__).parent.parent.resolve()
         os.chdir(run_cwd)
         success, new_test = inject_profiling_into_existing_test(
-            Path(f.name), [CodePosition(10, 14)], func, Path(f.name).parent, "pytest"
+            temp_file_path, [CodePosition(10, 14)], func, temp_file_path.parent, "pytest"
         )
         os.chdir(original_cwd)
     assert success
     assert new_test == 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()
     )
 
 
@@ -379,7 +380,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('"', "'")
 
         success, new_perf_test = inject_profiling_into_existing_test(