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

Fix the actual bug

Browse source
This commit is contained in:
Saurabh Misra 2024-07-06 15:26:13 -07:00
parent 0b63da2e2a
commit d28396b52d
2 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
codeflash/optimization/function_context.py
View file

@ -236,7 +236,7 @@ def get_function_variables_definitions(
source_code=source_code[0],
file_path=definition_path,
qualified_name=definition.full_name.removeprefix(
name.module_name + ".",
definition.module_name + ".",
),
only_function_name=definition.name,
),

5
tests/test_function_dependencies.py
View file

@ -102,6 +102,7 @@ def test_recursive_dependency() -> None:
)[0]
assert len(helper_functions) == 1
assert helper_functions[0].jedi_definition.full_name == "test_function_dependencies.calculate_something"
assert helper_functions[0].fully_qualified_name == "test_function_dependencies.calculate_something"
@dataclass
@ -198,6 +199,10 @@ def test_class_method_dependencies() -> None:
== "test_function_dependencies.Graph.topologicalSortUtil"
)
assert code_context.helper_functions[0].jedi_definition.name == "topologicalSortUtil"
assert (
code_context.helper_functions[0].fully_qualified_name
== "test_function_dependencies.Graph.topologicalSortUtil"
)
assert code_context.helper_functions[0].qualified_name == "Graph.topologicalSortUtil"
assert code_context.contextual_dunder_methods == {("Graph", "__init__")}
assert (