diff --git a/approve_all.py b/approve_all.py index 7427dc0..b1e1cef 100644 --- a/approve_all.py +++ b/approve_all.py @@ -15,13 +15,13 @@ def move(a: str, b: str) -> None: def approve_all( - failed_comparison_loader: Callable[[], list[str]] = load_failed_comparisons, + failed_comparisons: list[str], mover: Callable[[str, str], None] = move, ) -> None: - for line in failed_comparison_loader(): + for line in failed_comparisons: a, b = line.split(" -> ") mover(a, b) if __name__ == "__main__": - approve_all() + approve_all(load_failed_comparisons()) diff --git a/test_approve_all.py b/test_approve_all.py index 00abd34..fe280ab 100644 --- a/test_approve_all.py +++ b/test_approve_all.py @@ -50,8 +50,7 @@ def render_template(root_dir, template_path): def test__approve_all__with_loader_and_saver(): - def failed_comparison_loader(): - return [ + failed_comparisons = [ "a.received.txt -> a.approved.txt", "b.received.txt -> b.approved.txt", ] @@ -62,6 +61,6 @@ def mover(a, b): nonlocal moves moves.append(f"{a} -> {b}") - approve_all(failed_comparison_loader, mover) + approve_all(failed_comparisons, mover) - assert moves == failed_comparison_loader() + assert moves == failed_comparisons