multi-agent-train-lab / tests /test_smoke.py
3v324v23's picture
feat: 实现多智能体协作的机器学习训练与调试闭环
11ac7be
import tempfile
import unittest
from pathlib import Path
from multi_agent_lab.config import TrainConfig
from multi_agent_lab.orchestrator import Orchestrator
class SmokeTest(unittest.TestCase):
def test_stable_run_completes(self) -> None:
with tempfile.TemporaryDirectory() as td:
runs_dir = Path(td) / "runs"
orch = Orchestrator(base_runs_dir=runs_dir)
cfg = TrainConfig(seed=123, n_samples=600, n_features=10, epochs=10, lr=0.3)
store, summary = orch.run(cfg=cfg, scenario="stable")
self.assertTrue(store.path("report.md").exists())
ev = summary.get("eval", {})
self.assertGreaterEqual(float(ev.get("val_acc", 0.0)), 0.75)
if __name__ == "__main__":
unittest.main()