dougvk/Unlimited-OCR-RDNA4

1from pathlib import Path2 3from unlimited_ocr_rdna4.constants import MODEL_REVISION4from unlimited_ocr_rdna4.paths import default_cache_dir, default_model_dir5 6 7def test_xdg_defaults(monkeypatch) -> None:8    monkeypatch.delenv("UNLIMITED_OCR_MODEL_DIR", raising=False)9    monkeypatch.setenv("XDG_DATA_HOME", "/data-home")10    monkeypatch.setenv("XDG_CACHE_HOME", "/cache-home")11    assert default_model_dir() == Path("/data-home/unlimited-ocr-rdna4/models") / MODEL_REVISION12    assert default_cache_dir() == Path("/cache-home/unlimited-ocr-rdna4")13 14 15def test_model_dir_override(monkeypatch) -> None:16    monkeypatch.setenv("UNLIMITED_OCR_MODEL_DIR", "/models/pinned")17    assert default_model_dir() == Path("/models/pinned")18