Add dataset bench and validation agents

This commit is contained in:
2026-06-30 12:38:25 +08:00
parent 69f9a8e29b
commit dd7b7384ec
16 changed files with 853 additions and 24 deletions

View 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"])

View 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")