modify expectations

This commit is contained in:
Kevin Turcios 2025-12-29 17:39:45 -05:00
parent c47e480106
commit 094e20efcc
3 changed files with 3 additions and 3 deletions

View file

@ -7,7 +7,7 @@ from end_to_end_test_utilities import CoverageExpectation, TestConfig, run_codef
def run_test(expected_improvement_pct: int) -> bool:
config = TestConfig(
file_path="src/aviary/common_tags.py",
expected_unit_tests=2,
expected_unit_tests_count=2,
min_improvement_x=0.05,
coverage_expectations=[
CoverageExpectation(

View file

@ -17,7 +17,7 @@ def run_test(expected_improvement_pct: int) -> bool:
expected_lines=[25, 26, 27, 28, 29, 30, 31],
)
],
expected_unit_tests=13,
expected_unit_test_files=1, # Per-function count
)
cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
return_var = run_codeflash_command(cwd, config, expected_improvement_pct)

View file

@ -8,7 +8,7 @@ def run_test(expected_improvement_pct: int) -> bool:
config = TestConfig(
trace_mode=True,
min_improvement_x=0.1,
expected_unit_tests=None, # Tracer creates replay tests dynamically, skip validation
expected_unit_tests_count=None, # Tracer creates replay tests dynamically, skip validation
coverage_expectations=[
CoverageExpectation(function_name="funcA", expected_coverage=100.0, expected_lines=[6, 7, 8, 9, 11, 14])
],