refactor: move line_profile_utils, edit_generated_tests to languages/python/static_analysis/

codeflash/code_utils/code_replacer.py

@@ -17,7 +17,7 @@ from codeflash.code_utils.code_extractor import (
 )
 from codeflash.code_utils.config_parser import find_conftest_files
 from codeflash.code_utils.formatter import sort_imports
-from codeflash.code_utils.line_profile_utils import ImportAdder
+from codeflash.languages.python.static_analysis.line_profile_utils import ImportAdder
 from codeflash.languages import is_python
 from codeflash.models.models import FunctionParent
 

codeflash/optimization/function_optimizer.py

@@ -59,7 +59,7 @@ from codeflash.code_utils.config_consts import (
     get_effort_value,
 )
 from codeflash.code_utils.deduplicate_code import normalize_code
-from codeflash.code_utils.edit_generated_tests import (
+from codeflash.languages.python.static_analysis.edit_generated_tests import (
     add_runtime_comments_to_generated_tests,
     disable_ts_check,
     inject_test_globals,
@@ -70,7 +70,7 @@ from codeflash.code_utils.env_utils import get_pr_number
 from codeflash.code_utils.formatter import format_code, format_generated_code, sort_imports
 from codeflash.code_utils.git_utils import git_root_dir
 from codeflash.code_utils.instrument_existing_tests import inject_profiling_into_existing_test
-from codeflash.code_utils.line_profile_utils import add_decorator_imports, contains_jit_decorator
+from codeflash.languages.python.static_analysis.line_profile_utils import add_decorator_imports, contains_jit_decorator
 from codeflash.code_utils.shell_utils import make_env_with_project_root
 from codeflash.languages.python.static_analysis.static_analysis import get_first_top_level_function_or_method_ast
 from codeflash.code_utils.time_utils import humanize_runtime

tests/test_add_runtime_comments.py

@@ -4,7 +4,7 @@ from unittest.mock import Mock
 
 import pytest
 
-from codeflash.code_utils.edit_generated_tests import add_runtime_comments_to_generated_tests
+from codeflash.languages.python.static_analysis.edit_generated_tests import add_runtime_comments_to_generated_tests
 from codeflash.models.models import (
     FunctionTestInvocation,
     GeneratedTests,

tests/test_instrument_line_profiler.py

@@ -2,7 +2,7 @@ import os
 from pathlib import Path
 from tempfile import TemporaryDirectory
 
-from codeflash.code_utils.line_profile_utils import add_decorator_imports, contains_jit_decorator
+from codeflash.languages.python.static_analysis.line_profile_utils import add_decorator_imports, contains_jit_decorator
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.models.models import CodeOptimizationContext
 from codeflash.optimization.function_optimizer import FunctionOptimizer

tests/test_instrument_tests.py

@@ -15,7 +15,7 @@ from codeflash.code_utils.instrument_existing_tests import (
     FunctionImportedAsVisitor,
     inject_profiling_into_existing_test,
 )
-from codeflash.code_utils.line_profile_utils import add_decorator_imports
+from codeflash.languages.python.static_analysis.line_profile_utils import add_decorator_imports
 from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 from codeflash.models.models import (
     CodeOptimizationContext,

tests/test_remove_functions_from_generated_tests.py

@@ -2,7 +2,7 @@ from pathlib import Path
 
 import pytest
 
-from codeflash.code_utils.edit_generated_tests import remove_functions_from_generated_tests
+from codeflash.languages.python.static_analysis.edit_generated_tests import remove_functions_from_generated_tests
 from codeflash.models.models import GeneratedTests, GeneratedTestsList