codeflash-ai/codeflash
1
0
Fork 0
Code Issues 18 Pull requests 33 Projects Releases Packages Wiki Activity Actions 3.1k

Preliminary refactoring.

Browse source
This commit is contained in:
renaud 2024-02-06 17:35:13 -08:00
parent 5b29ac901d
commit 32c09ff6f6
1 changed files with 8 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_code_replacement.py
View file

@ -1,8 +1,8 @@
import os
from tempfile import NamedTemporaryFile
from codeflash.code_utils.code_replacer import replace_function_in_file
os.environ["CODEFLASH_API_KEY"] = "cf-test-key"
from codeflash.code_utils.code_replacer import replace_function_in_file
def test_test_libcst_code_replacement():
@ -45,15 +45,9 @@ def totally_new_function(value):
print("Hello world")
"""
with NamedTemporaryFile(mode="w") as f:
f.write(original_code)
f.flush()
path = f.name
function_name = "NewClass.new_function"
preexisting_functions = ["NewClass.new_function"]
optimized_code = optim_code
new_code = replace_function_in_file(
path, function_name, optimized_code, preexisting_functions
function_name: str = "NewClass.new_function"
preexisting_functions: list[str] = ["NewClass.new_function"]
new_code: str = replace_function_in_file(
original_code, function_name, optim_code, preexisting_functions
)
assert new_code == expected