tests/test_postsave_inference.py

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