remove TestsCache usage

Author: KRRT7

codeflash/discovery/discover_unit_tests.py

@@ -297,7 +297,7 @@ def filter_test_files_by_imports(
         else:
             logger.debug(f"Skipping {test_file} - no relevant imports found")
 
-    logger.info(f"Import filter: Processing {len(filtered_map)}/{len(file_to_test_map)} test files")
+    logger.debug(f"Import filter: Processing {len(filtered_map)}/{len(file_to_test_map)} test files")
     return filtered_map, import_results
 
 
@@ -491,27 +491,12 @@ def process_test_files(
 
     function_to_test_map = defaultdict(set)
     jedi_project = jedi.Project(path=project_root_path)
-    tests_cache = TestsCache()
 
     with test_files_progress_bar(total=len(file_to_test_map), description="Processing test files") as (
         progress,
         task_id,
     ):
         for test_file, functions in file_to_test_map.items():
-            file_hash = TestsCache.compute_file_hash(test_file)
-            cached_tests = tests_cache.get_tests_for_file(str(test_file), file_hash)
-            if cached_tests:
-                self_cur = tests_cache.cur
-                self_cur.execute(
-                    "SELECT qualified_name_with_modules_from_root FROM discovered_tests WHERE file_path = ? AND file_hash = ?",
-                    (str(test_file), file_hash),
-                )
-                qualified_names = [row[0] for row in self_cur.fetchall()]
-                for cached, qualified_name in zip(cached_tests, qualified_names):
-                    function_to_test_map[qualified_name].add(cached)
-                progress.advance(task_id)
-                continue
-
             try:
                 script = jedi.Script(path=test_file, project=jedi_project)
                 test_functions = set()
@@ -636,18 +621,6 @@ def process_test_files(
                         )
                         qualified_name_with_modules_from_root = f"{module_name_from_file_path(definition[0].module_path, project_root_path)}.{full_name_without_module_prefix}"
 
-                        tests_cache.insert_test(
-                            file_path=str(test_file),
-                            file_hash=file_hash,
-                            qualified_name_with_modules_from_root=qualified_name_with_modules_from_root,
-                            function_name=scope,
-                            test_class=scope_test_class,
-                            test_function=scope_test_function,
-                            test_type=test_type,
-                            line_number=name.line,
-                            col_number=name.column,
-                        )
-
                         function_to_test_map[qualified_name_with_modules_from_root].add(
                             FunctionCalledInTest(
                                 tests_in_file=TestsInFile(
@@ -662,5 +635,4 @@ def process_test_files(
 
             progress.advance(task_id)
 
-    tests_cache.close()
     return {function: list(tests) for function, tests in function_to_test_map.items()}

codeflash/optimization/optimizer.py

@@ -162,9 +162,7 @@ def run(self) -> None:
 
             console.rule()
             start_time = time.time()
-            function_to_tests: dict[str, list[FunctionCalledInTest]] = discover_unit_tests(
-                self.test_cfg, file_to_funcs_to_optimize=file_to_funcs_to_optimize
-            )
+            function_to_tests = discover_unit_tests(self.test_cfg, file_to_funcs_to_optimize=file_to_funcs_to_optimize)
             num_discovered_tests: int = sum([len(value) for value in function_to_tests.values()])
             console.rule()
             logger.info(

tests/test_static_analysis.py

@@ -1,4 +1,4 @@
-import ast
+import ast
 from pathlib import Path
 
 from codeflash.code_utils.static_analysis import (