tests/conftest.py

"""Shared fixtures for Latin BERT case study benchmarks."""

import pytest

DEFAULT_MODEL_PATH = "latincy/latin-bert"


def pytest_addoption(parser):
    parser.addoption(
        "--model-path",
        action="store",
        default=DEFAULT_MODEL_PATH,
        help="Path to HF-compatible model directory (default: /tmp/latin-bert-hub)",
    )


@pytest.fixture(scope="session")
def model_path(request):
    """Model path fixture — use --model-path to benchmark different models."""
    return request.config.getoption("--model-path")