Initial Seg Data Server Net platform
This commit is contained in:
15
backend/tests/test_catalog.py
Normal file
15
backend/tests/test_catalog.py
Normal file
@@ -0,0 +1,15 @@
|
||||
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.predict",
|
||||
"mmseg.flops_fps",
|
||||
"analysis.all",
|
||||
]:
|
||||
assert task in TASK_TYPES
|
||||
|
||||
Reference in New Issue
Block a user