from __future__ import annotations

from pathlib import Path

import pytest

REPO_ROOT = Path(__file__).resolve().parent.parent
DATA_DIR = REPO_ROOT / "data"


@pytest.fixture(scope="session")
def train_csv() -> Path:
    path = DATA_DIR / "mouse_data.csv"
    if not path.is_file():
        pytest.skip(f"missing training csv: {path}")
    return path


@pytest.fixture(scope="session")
def test_csv() -> Path:
    path = DATA_DIR / "mouse_data_test.csv"
    if not path.is_file():
        pytest.skip(f"missing test csv: {path}")
    return path


@pytest.fixture(scope="session")
def bundled_session():
    """Shared inference session for the bundled model (loading is slow)."""
    from mouse_control import load_model

    return load_model()