18 lines
476 B
Python
18 lines
476 B
Python
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_custom",
|
|
"yolo.heatmap_custom",
|
|
"yolo.predict",
|
|
"mmseg.flops_fps",
|
|
"analysis.all",
|
|
]:
|
|
assert task in TASK_TYPES
|