from app.capabilities import get_capability_matrix


def test_capability_matrix_covers_core_domains():
    matrix = get_capability_matrix()
    domains = {item["id"]: item for item in matrix["domains"]}

    assert {"dataset", "segmodel", "yolo", "mmseg", "visual", "analysis", "system"} <= set(domains)
    assert domains["dataset"]["tasks"]["required_ready"] == domains["dataset"]["tasks"]["required"]
    assert domains["yolo"]["ready"] is True
    assert domains["mmseg"]["ready"] is True


def test_capability_matrix_tracks_user_requirements():
    matrix = get_capability_matrix()
    requirements = {item["id"]: item for item in matrix["requirements"]}

    assert requirements["user_script_mapping"]["passed"] is True
    assert requirements["runtime_readiness"]["passed"] is True
    assert requirements["yolo_heatmap"]["passed"] is True
    assert requirements["training_curves"]["passed"] is True