Cyber_analyst-round1 / tests /test_models.py
Humanlearning's picture
feat: implement core RL training infrastructure, including GRPO training, evaluation utilities, custom environments, and Modal-based execution scripts.
3807ea3
raw
history blame
498 Bytes
from CyberSecurity_OWASP import (
CyberSecurityOWASPAction,
CyberSecurityOWASPObservation,
CyberSecurityOWASPState,
)
def test_models_serialize():
action = CyberSecurityOWASPAction(tool_name="noop")
assert action.model_dump()["tool_name"] == "noop"
obs = CyberSecurityOWASPObservation(phase="discover", message="ok")
assert obs.model_dump()["phase"] == "discover"
state = CyberSecurityOWASPState(episode_id="e1", seed=1)
assert state.model_dump()["seed"] == 1