update to match new def naming

Author: KRRT7

tests/test_trace_benchmarks.py

@@ -33,7 +33,7 @@ def test_trace_benchmarks() -> None:
         function_calls = cursor.fetchall()
 
         # Assert the length of function calls
-        assert len(function_calls) == 7, f"Expected 7 function calls, but got {len(function_calls)}"
+        assert len(function_calls) == 8, f"Expected 8 function calls, but got {len(function_calls)}"
 
         bubble_sort_path = (project_root / "bubble_sort_codeflash_trace.py").as_posix()
         process_and_bubble_sort_path = (project_root / "process_and_bubble_sort_codeflash_trace.py").as_posix()
@@ -82,7 +82,7 @@ def test_trace_benchmarks() -> None:
         # Close connection
         conn.close()
         generate_replay_test(output_file, replay_tests_dir)
-        test_class_sort_path = replay_tests_dir/ Path("test_tests_pytest_benchmarks_test_test_benchmark_bubble_sort_example__replay_test_0.py")
+        test_class_sort_path = replay_tests_dir/ Path("test_code_to_optimize_tests_pytest_benchmarks_test_test_benchmark_bubble_sort_example__replay_test_0.py")
         assert test_class_sort_path.exists()
         test_class_sort_code = f"""
 from code_to_optimize.bubble_sort_codeflash_trace import \\
@@ -95,13 +95,13 @@ def test_trace_benchmarks() -> None:
 functions = ['sort_class', 'sort_static', 'sorter']
 trace_file_path = r"{output_file.as_posix()}"
 
-def test_code_to_optimize_bubble_sort_codeflash_trace_sorter():
+def test_code_to_optimize_bubble_sort_codeflash_trace_sorter_test_sort():
     for args_pkl, kwargs_pkl in get_next_arg_and_return(trace_file=trace_file_path, benchmark_function_name="test_sort", function_name="sorter", file_path=r"{bubble_sort_path}", num_to_get=100):
         args = pickle.loads(args_pkl)
         kwargs = pickle.loads(kwargs_pkl)
         ret = code_to_optimize_bubble_sort_codeflash_trace_sorter(*args, **kwargs)
 
-def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sorter():
+def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sorter_test_class_sort():
     for args_pkl, kwargs_pkl in get_next_arg_and_return(trace_file=trace_file_path, benchmark_function_name="test_class_sort", function_name="sorter", file_path=r"{bubble_sort_path}", class_name="Sorter", num_to_get=100):
         args = pickle.loads(args_pkl)
         kwargs = pickle.loads(kwargs_pkl)
@@ -113,21 +113,21 @@ def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sorter():
         else:
             ret = code_to_optimize_bubble_sort_codeflash_trace_Sorter.sorter(*args, **kwargs)
 
-def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sort_class():
+def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sort_class_test_class_sort2():
     for args_pkl, kwargs_pkl in get_next_arg_and_return(trace_file=trace_file_path, benchmark_function_name="test_class_sort2", function_name="sort_class", file_path=r"{bubble_sort_path}", class_name="Sorter", num_to_get=100):
         args = pickle.loads(args_pkl)
         kwargs = pickle.loads(kwargs_pkl)
         if not args:
             raise ValueError("No arguments provided for the method.")
         ret = code_to_optimize_bubble_sort_codeflash_trace_Sorter.sort_class(*args[1:], **kwargs)
 
-def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sort_static():
+def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter_sort_static_test_class_sort3():
     for args_pkl, kwargs_pkl in get_next_arg_and_return(trace_file=trace_file_path, benchmark_function_name="test_class_sort3", function_name="sort_static", file_path=r"{bubble_sort_path}", class_name="Sorter", num_to_get=100):
         args = pickle.loads(args_pkl)
         kwargs = pickle.loads(kwargs_pkl)
         ret = code_to_optimize_bubble_sort_codeflash_trace_Sorter.sort_static(*args, **kwargs)
 
-def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter___init__():
+def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter___init___test_class_sort4():
     for args_pkl, kwargs_pkl in get_next_arg_and_return(trace_file=trace_file_path, benchmark_function_name="test_class_sort4", function_name="__init__", file_path=r"{bubble_sort_path}", class_name="Sorter", num_to_get=100):
         args = pickle.loads(args_pkl)
         kwargs = pickle.loads(kwargs_pkl)
@@ -142,7 +142,7 @@ def test_code_to_optimize_bubble_sort_codeflash_trace_Sorter___init__():
 """
         assert test_class_sort_path.read_text("utf-8").strip()==test_class_sort_code.strip()
 
-        test_sort_path = replay_tests_dir / Path("test_tests_pytest_benchmarks_test_test_process_and_sort_example__replay_test_0.py")
+        test_sort_path = replay_tests_dir / Path("test_code_to_optimize_tests_pytest_benchmarks_test_test_process_and_sort_example__replay_test_0.py")
         assert test_sort_path.exists()
         test_sort_code = f"""
 from code_to_optimize.bubble_sort_codeflash_trace import \\