codeflash-admin/codeflash
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash.git synced 2026-05-04 18:25:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix failing unit tests with recent refactoring

Browse source
This commit is contained in:
Sarthak Agarwal 2026-03-07 01:48:41 +05:30
parent 5d872e845d
commit 3367383824
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_mock_candidate_replacement.py
View file

@ -9,7 +9,7 @@ from codeflash.discovery.functions_to_optimize import FunctionToOptimize
from codeflash.languages.python.context.unused_definition_remover import detect_unused_helper_functions
from codeflash.models.function_types import FunctionParent
from codeflash.models.models import CodeStringsMarkdown
from codeflash.optimization.function_optimizer import FunctionOptimizer
from codeflash.languages.python.function_optimizer import PythonFunctionOptimizer
from codeflash.verification.verification_utils import TestConfig
ORIGINAL_SOURCE = '''\
@ -649,7 +649,7 @@ def run_replacement(temp_project):
parents=[FunctionParent(name="PSBaseParser", type="ClassDef")],
)
optimizer = FunctionOptimizer(
optimizer = PythonFunctionOptimizer(
function_to_optimize=function_to_optimize,
test_cfg=test_cfg,
function_to_optimize_source_code=source_file.read_text(encoding="utf-8"),
@ -715,7 +715,7 @@ def test_detect_unused_helpers_handles_attribute_refs(temp_project):
parents=[FunctionParent(name="PSBaseParser", type="ClassDef")],
)
optimizer = FunctionOptimizer(
optimizer = PythonFunctionOptimizer(
function_to_optimize=function_to_optimize,
test_cfg=test_cfg,
function_to_optimize_source_code=source_file.read_text(encoding="utf-8"),