Add dataset bench and validation agents
This commit is contained in:
14
backend/tests/test_agents.py
Normal file
14
backend/tests/test_agents.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from app.agents.evaluation_agent import evaluate_project
|
||||
from app.agents.validation_agent import validate_project
|
||||
|
||||
|
||||
def test_evaluation_agent_returns_checks():
|
||||
result = evaluate_project()
|
||||
assert result["agent"] == "evaluation_suggestion_agent"
|
||||
assert result["checks"]
|
||||
|
||||
|
||||
def test_validation_agent_lightweight():
|
||||
result = validate_project(run_build=False)
|
||||
assert result["agent"] == "validation_agent"
|
||||
assert any(item["name"] == "catalog_has_yolo_heatmap" for item in result["checks"])
|
||||
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