Add dataset bench and validation agents
This commit is contained in:
13
backend/tests/test_dataset_service.py
Normal file
13
backend/tests/test_dataset_service.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from app.modules.dataset.service import create_dataset, describe_dataset
|
||||
|
||||
|
||||
def test_create_dataset_layout(tmp_path, monkeypatch):
|
||||
from types import SimpleNamespace
|
||||
from app.modules.dataset import service
|
||||
|
||||
monkeypatch.setattr(service, "settings", SimpleNamespace(project_root=tmp_path))
|
||||
created = create_dataset("case 01", "demo")
|
||||
assert created["name"] == "case_01"
|
||||
assert created["counts"] == {"images": 0, "labels": 0, "masks": 0}
|
||||
described = describe_dataset("case_01")
|
||||
assert described["layout"]["images"].endswith("images")
|
||||
Reference in New Issue
Block a user