Update end_to_end_test_topological_sort_worktree.py

KRRT7 · KRRT7 · commit 69e5edf8b91e · 2025-12-12T11:06:59.000-05:00
diff --git a/tests/scripts/end_to_end_test_topological_sort_worktree.py b/tests/scripts/end_to_end_test_topological_sort_worktree.py
@@ -17,7 +17,7 @@ def run_test(expected_improvement_pct: int) -> bool:
                 expected_lines=[25, 26, 27, 28, 29, 30, 31],
             )
         ],
-        expected_unit_tests=1,
+        expected_unit_tests=13,
     )
     cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()
     return_var = run_codeflash_command(cwd, config, expected_improvement_pct)

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def run_test(expected_improvement_pct: int) -> bool:`
`17`	`17`	`expected_lines=[25, 26, 27, 28, 29, 30, 31],`
`18`	`18`	`)`
`19`	`19`	`],`
`20`		`- expected_unit_tests=1,`
	`20`	`+ expected_unit_tests=13,`
`21`	`21`	`)`
`22`	`22`	`cwd = (pathlib.Path(__file__).parent.parent.parent / "code_to_optimize").resolve()`
`23`	`23`	`return_var = run_codeflash_command(cwd, config, expected_improvement_pct)`