Skip to content

Commit 5cd13ad

Browse files
new line
1 parent c75bbf6 commit 5cd13ad

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/test_formatter.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
from codeflash.discovery.functions_to_optimize import FunctionToOptimize
1313
from codeflash.optimization.function_optimizer import FunctionOptimizer
14+
from codeflash.optimization.function_optimizer import FunctionSource
1415
from codeflash.verification.verification_utils import TestConfig
1516

1617
def test_remove_duplicate_imports():
@@ -257,7 +258,9 @@ def _run_formatting_test(source_filename: str, should_content_change: bool):
257258
)
258259

259260
optimizer.reformat_code_and_helpers(
260-
helper_functions=[],
261+
helper_functions=[
262+
FunctionSource()
263+
],
261264
path=target_path,
262265
original_code=optimizer.function_to_optimize_source_code,
263266
)
@@ -277,4 +280,4 @@ def test_formatting_file_with_many_diffs():
277280

278281
def test_formatting_file_with_few_diffs():
279282
"""Test that files with few formatting errors are formatted (content changed)."""
280-
_run_formatting_test("few_formatting_errors.py", should_content_change=True)
283+
_run_formatting_test("few_formatting_errors.py", should_content_change=True)

0 commit comments

Comments
 (0)