from app.catalog import SEGMODEL_ARCHS, TASK_TYPES, YOLO_MODELS


def test_catalog_contains_required_capabilities():
    assert "Unet" in SEGMODEL_ARCHS
    assert "YOLOv9e-seg" in YOLO_MODELS
    for task in [
        "dataset.video_frames",
        "segmodel.train",
        "yolo.train_custom",
        "yolo.predict",
        "mmseg.flops_fps",
        "analysis.all",
    ]:
        assert task in TASK_TYPES