import os, glob, pytest

def _model_dir():
    env = os.environ.get("TITAN_MODEL_DIR")
    if env:
        return env
    hits = glob.glob(os.path.expanduser("~/.cache/huggingface/hub/models--mlx-community--OLMoE-1B-7B-0125-Instruct-4bit/snapshots/*"))
    return hits[0] if hits else None

def _model_dirs():
    env = os.environ.get("TITAN_MODEL_DIRS")
    if env:
        return [d for d in env.split(":") if d]
    d = _model_dir()
    return [d] if d else []

def pytest_generate_tests(metafunc):
    if "model_dir" in metafunc.fixturenames:
        dirs = _model_dirs()
        metafunc.parametrize("model_dir", dirs or [None], ids=[os.path.basename(os.path.dirname(os.path.dirname(d))) if d else "none" for d in dirs] or ["none"], scope="module")

@pytest.fixture(autouse=True)
def _skip_if_no_model(request):
    if "model_dir" in request.fixturenames and request.getfixturevalue("model_dir") is None:
        pytest.skip("no MoE checkpoint in the HF cache (set TITAN_MODEL_DIRS)")