File size: 270 Bytes
fd0c71a
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from app.env.env_core import PolyGuardEnv


def test_postsave_inference_path() -> None:
    env = PolyGuardEnv()
    env.reset(seed=77, difficulty="easy")
    action = env.get_legal_actions()[0]
    _, reward, _, _ = env.step(action)
    assert 0.001 <= reward <= 0.999