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

duplicate global statements need to be handled

Browse source
This commit is contained in:
aseembits93 2025-04-29 18:54:54 -07:00
parent 08c8067630
commit 38f0253c41
1 changed files with 1 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_code_replacement.py
View file

@ -789,23 +789,7 @@ class MainClass:
def test_code_replacement10() -> None:
get_code_output = """from __future__ import annotations
class HelperClass:
def __init__(self, name):
self.name = name
def helper_method(self):
return self.name
class MainClass:
def __init__(self, name):
self.name = name
def main_method(self):
return HelperClass(self.name).helper_method()
"""
get_code_output = 'from __future__ import annotations\nimport os\n\nos.environ["CODEFLASH_API_KEY"] = "cf-test-key"\nclass HelperClass:\n def __init__(self, name):\n self.name = name\n\n def helper_method(self):\n return self.name\n\n\nclass MainClass:\n def __init__(self, name):\n self.name = name\n\n def main_method(self):\n return HelperClass(self.name).helper_method()\n'
file_path = Path(__file__).resolve()
func_top_optimize = FunctionToOptimize(
function_name="main_method", file_path=file_path, parents=[FunctionParent("MainClass", "ClassDef")]