| from pathlib import Path | |
| from unlimited_ocr_rdna4.constants import MODEL_REVISION | |
| from unlimited_ocr_rdna4.paths import default_cache_dir, default_model_dir | |
| def test_xdg_defaults(monkeypatch) -> None: | |
| monkeypatch.delenv("UNLIMITED_OCR_MODEL_DIR", raising=False) | |
| monkeypatch.setenv("XDG_DATA_HOME", "/data-home") | |
| monkeypatch.setenv("XDG_CACHE_HOME", "/cache-home") | |
| assert default_model_dir() == Path("/data-home/unlimited-ocr-rdna4/models") / MODEL_REVISION | |
| assert default_cache_dir() == Path("/cache-home/unlimited-ocr-rdna4") | |
| def test_model_dir_override(monkeypatch) -> None: | |
| monkeypatch.setenv("UNLIMITED_OCR_MODEL_DIR", "/models/pinned") | |
| assert default_model_dir() == Path("/models/pinned") | |