code/tests/test_state.py

"""Tests for the OpenEnv State interface."""

from cropRL.models import CroprlAction
from cropRL.server.cropRL_environment import CroprlEnvironment


class TestState:
    def test_state_step_count(self, env):
        assert env.state.step_count == 0
        env.step(CroprlAction(action_id=0))
        assert env.state.step_count == 1

    def test_state_episode_id(self, env):
        assert env.state.episode_id is not None
        assert len(env.state.episode_id) > 0

    def test_state_task_id(self):
        e = CroprlEnvironment(task_id="hard")
        e.reset(seed=42)
        assert e.state.task_id == "hard"