File size: 333 Bytes
21c7db9 | 1 2 3 4 5 6 7 8 9 10 11 12 | 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
|