tests/test_agents.py

from app.agents.orchestrator import Orchestrator
from app.env.env_core import PolyGuardEnv


def test_orchestrator_run_step() -> None:
    env = PolyGuardEnv()
    env.reset(seed=55, difficulty="easy")
    orchestrator = Orchestrator(env)
    out = orchestrator.run_step()
    assert "final_action" in out
    assert "reward" in out