File size: 727 Bytes
f340984 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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")
|