fix more tests

Author: misrasaurabh1

tests/test_instrument_all_and_run.py

@@ -247,8 +247,9 @@ def test_sort():
 import time
 
 import dill as pickle
-
 from code_to_optimize.bubble_sort_method import BubbleSorter
+
+
 """
         + codeflash_wrap_string
         + """

tests/test_instrument_tests.py

@@ -307,10 +307,12 @@ def test_prepare_image_for_yolo():
 def test_perfinjector_bubble_sort_results() -> None:
     computed_fn_opt = False
     code = """from code_to_optimize.bubble_sort import sorter
+import datetime
 
 
 def test_sort():
     input = [5, 4, 3, 2, 1, 0]
+    print(datetime.datetime.now().isoformat())
     output = sorter(input)
     assert output == [0, 1, 2, 3, 4, 5]
 
@@ -319,7 +321,8 @@ def test_sort():
     assert output == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]"""
 
     expected = (
-        """import gc
+        """import datetime
+import gc
 import os
 import sqlite3
 import time
@@ -339,17 +342,19 @@ def test_sort():
     codeflash_cur = codeflash_con.cursor()
     codeflash_cur.execute('CREATE TABLE IF NOT EXISTS test_results (test_module_path TEXT, test_class_name TEXT, test_function_name TEXT, function_getting_tested TEXT, loop_index INTEGER, iteration_id TEXT, runtime INTEGER, return_value BLOB, verification_type TEXT)')
     input = [5, 4, 3, 2, 1, 0]
-    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '1', codeflash_loop_index, codeflash_cur, codeflash_con, input)
+    print(datetime.datetime.now().isoformat())
+    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '2', codeflash_loop_index, codeflash_cur, codeflash_con, input)
     assert output == [0, 1, 2, 3, 4, 5]
     input = [5.0, 4.0, 3.0, 2.0, 1.0, 0.0]
-    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '4', codeflash_loop_index, codeflash_cur, codeflash_con, input)
+    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '5', codeflash_loop_index, codeflash_cur, codeflash_con, input)
     assert output == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]
     codeflash_con.close()
 """
     )
 
     expected_perfonly = (
-        """import gc
+        """import datetime
+import gc
 import os
 import time
 
@@ -362,10 +367,10 @@ def test_sort():
 def test_sort():
     codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
     input = [5, 4, 3, 2, 1, 0]
-    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '1', codeflash_loop_index, input)
+    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '2', codeflash_loop_index, input)
     assert output == [0, 1, 2, 3, 4, 5]
     input = [5.0, 4.0, 3.0, 2.0, 1.0, 0.0]
-    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '4', codeflash_loop_index, input)
+    output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort', 'sorter', '5', codeflash_loop_index, input)
     assert output == [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]
 """
     )
@@ -390,7 +395,7 @@ def test_sort():
         os.chdir(run_cwd)
         success, new_test = inject_profiling_into_existing_test(
             test_path,
-            [CodePosition(6, 13), CodePosition(10, 13)],
+            [CodePosition(8, 13), CodePosition(12, 13)],
             func,
             project_root_path,
             "pytest",
@@ -560,6 +565,7 @@ def test_perfinjector_bubble_sort_parametrized_results() -> None:
     computed_fn_opt = False
     code = """from code_to_optimize.bubble_sort import sorter
 import pytest
+import datetime
 
 
 @pytest.mark.parametrize(
@@ -571,6 +577,7 @@ def test_perfinjector_bubble_sort_parametrized_results() -> None:
     ],
 )
 def test_sort_parametrized(input, expected_output):
+    print(datetime.datetime.now().isoformat())
     output = sorter(input)
     assert output == expected_output
 """
@@ -606,6 +613,7 @@ def test_sort_parametrized(input, expected_output):
         """import gc
 import os
 import time
+import datetime
 
 import pytest
 
@@ -618,6 +626,7 @@ def test_sort_parametrized(input, expected_output):
 @pytest.mark.parametrize('input, expected_output', [([5, 4, 3, 2, 1, 0], [0, 1, 2, 3, 4, 5]), ([5.0, 4.0, 3.0, 2.0, 1.0, 0.0], [0.0, 1.0, 2.0, 3.0, 4.0, 5.0]), (list(reversed(range(50))), list(range(50)))])
 def test_sort_parametrized(input, expected_output):
     codeflash_loop_index = int(os.environ['CODEFLASH_LOOP_INDEX'])
+    print(datetime.datetime.now().isoformat())
     output = codeflash_wrap(sorter, '{module_path}', None, 'test_sort_parametrized', 'sorter', '0', codeflash_loop_index, input)
     assert output == expected_output
 """
@@ -643,11 +652,11 @@ def test_sort_parametrized(input, expected_output):
         func = FunctionToOptimize(function_name="sorter", parents=[], file_path=code_path)
         os.chdir(run_cwd)
         success, new_test = inject_profiling_into_existing_test(
-            test_path, [CodePosition(14, 13)], func, project_root_path, "pytest", mode=TestingMode.BEHAVIOR
+            test_path, [CodePosition(16, 13)], func, project_root_path, "pytest", mode=TestingMode.BEHAVIOR
         )
         assert success
         success, new_test_perf = inject_profiling_into_existing_test(
-            test_path, [CodePosition(14, 13)], func, project_root_path, "pytest", mode=TestingMode.PERFORMANCE
+            test_path, [CodePosition(16, 13)], func, project_root_path, "pytest", mode=TestingMode.PERFORMANCE
         )
 
         os.chdir(original_cwd)