| from llmserve_env.models import default_action | |
| from server.llmserve_environment import LLMServeEnvironment | |
| def test_environment_reset_and_step() -> None: | |
| env = LLMServeEnvironment(seed=7) | |
| obs = env.reset(task_id="static_workload", seed=7) | |
| assert obs.task_id == "static_workload" | |
| next_obs = env.step(default_action()) | |
| assert next_obs.step_index == 1 | |
| assert isinstance(next_obs.reward, float) | |
| assert "phase" in next_obs.metadata | |
| assert next_obs.done is False | |