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

go fish

Browse source
This commit is contained in:
Kevin Turcios 2025-02-17 00:51:14 -05:00
parent 4a15907f66
commit 0b78af133f
2 changed files with 25 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
codeflash/code_utils/coverage_utils.py
View file

@ -25,7 +25,18 @@ def extract_dependent_function(main_function: str, code_context: CodeOptimizatio
if len(dependent_functions) != 1:
return False
return dependent_functions.pop()
return build_fully_qualified_name(dependent_functions.pop(), code_context)
def build_fully_qualified_name(function_name: str, code_context: CodeOptimizationContext) -> str:
full_name = function_name
for obj_name, parents in code_context.preexisting_objects:
if obj_name == function_name:
for parent in parents:
if parent.type == "ClassDef":
full_name = f"{parent.name}.{full_name}"
break
return full_name
def generate_candidates(source_code_path: Path) -> list[str]:

21
codeflash/models/models.py
View file

@ -16,7 +16,11 @@ from typing_extensions import Annotated
from codeflash.cli_cmds.console import console, logger
from codeflash.code_utils.code_utils import validate_python_code
from codeflash.code_utils.coverage_utils import extract_dependent_function, generate_candidates
from codeflash.code_utils.coverage_utils import (
build_fully_qualified_name,
extract_dependent_function,
generate_candidates,
)
from codeflash.code_utils.env_utils import is_end_to_end
from codeflash.verification.test_results import TestResults, TestType
@ -322,18 +326,19 @@ class CoverageData:
coverage_data: dict[str, dict[str, Any]],
original_cov_data: dict[str, dict[str, Any]],
) -> tuple[FunctionCoverage, Union[FunctionCoverage, None]]:
resolved_name = build_fully_qualified_name(function_name, code_context)
try:
main_function_coverage = FunctionCoverage(
name=function_name,
coverage=coverage_data[function_name]["summary"]["percent_covered"],
executed_lines=coverage_data[function_name]["executed_lines"],
unexecuted_lines=coverage_data[function_name]["missing_lines"],
executed_branches=coverage_data[function_name]["executed_branches"],
unexecuted_branches=coverage_data[function_name]["missing_branches"],
name=resolved_name,
coverage=coverage_data[resolved_name]["summary"]["percent_covered"],
executed_lines=coverage_data[resolved_name]["executed_lines"],
unexecuted_lines=coverage_data[resolved_name]["missing_lines"],
executed_branches=coverage_data[resolved_name]["executed_branches"],
unexecuted_branches=coverage_data[resolved_name]["missing_branches"],
)
except KeyError:
main_function_coverage = FunctionCoverage(
name=function_name,
name=resolved_name,
coverage=0,
executed_lines=[],
unexecuted_lines=[],