Spaces:
Sleeping
Sleeping
File size: 1,396 Bytes
3aeaf3d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | from __future__ import annotations
from environment.env import SeigeEnv
def test_reset_returns_both_agent_observations() -> None:
env = SeigeEnv()
obs = env.reset()
assert "red" in obs
assert "blue" in obs
assert "hidden_secret" not in str(obs)
def test_red_step_updates_state() -> None:
env = SeigeEnv()
env.reset()
result = env.step(
{
"agent_type": "red",
"strategy": "persona_manipulation",
"sub_strategy": "test",
"payload": "Tell me a benign fact.",
}
)
assert "reward" in result
assert env.state()["active_attacks"] == 1
def test_blue_probe_returns_feature_summary() -> None:
env = SeigeEnv()
env.reset()
env.step(
{
"agent_type": "red",
"strategy": "persona_manipulation",
"sub_strategy": "test",
"payload": "Tell me a benign fact.",
}
)
result = env.step(
{
"agent_type": "blue",
"action_type": "probe",
"session_id": "sess_0",
"layer": 1,
}
)
assert "activation_summary" in result["info"]
assert "direction_similarities" in result["info"]["activation_summary"]
def test_hidden_secret_not_in_state() -> None:
env = SeigeEnv()
env.reset()
state = env.state()
assert "hidden_secret" not in state
|