CoolFace
Modelpublic

dougvk/Unlimited-OCR-RDNA4

sourceHugging Facemitupdated 2mo agoView on Hugging Face
0likes
test_cli.py34 linesDownload Raw Back to tests
1import json2from pathlib import Path3 4from unlimited_ocr_rdna4.cli import run5from unlimited_ocr_rdna4.infer import default_output_path6 7 8def test_prepare_dry_run_json_after_subcommand(capsys, tmp_path) -> None:9    code = run(10        [11            "prepare",12            "--model-dir",13            str(tmp_path / "model"),14            "--cache-dir",15            str(tmp_path / "cache"),16            "--dry-run",17            "--json",18        ]19    )20    assert code == 021    payload = json.loads(capsys.readouterr().out)22    assert payload["schema_version"] == 123    assert payload["status"] == "dry-run"24    assert payload["prepared"] is False25 26 27def test_no_command_returns_usage_error(capsys) -> None:28    assert run([]) == 229    assert "usage:" in capsys.readouterr().err30 31 32def test_default_output_path() -> None:33    assert default_output_path(Path("invoice.pdf")) == Path("invoice.pdf.ocr.md")34