"""Tests for dataset and model YAML registries."""

from pathlib import Path

from forja_core.registry.dataset_store import FileSystemDatasetStore
from forja_core.registry.model_store import FileSystemModelStore

FIXTURES = Path(__file__).parent.parent / "fixtures"


def test_get_dataset_active_version() -> None:
    store = FileSystemDatasetStore(FIXTURES / "datasets")
    ds = store.get_dataset("incident_sft")
    assert ds.name == "incident_sft"
    assert ds.version == "v1"
    assert ds.format == "jsonl"


def test_get_model_active_version() -> None:
    store = FileSystemModelStore(FIXTURES / "models")
    m = store.get_model("gpt4o_lora_base")
    assert m.adaptation == "lora"
    assert m.base_model_id == "gpt-4o"