mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-26 08:17:18 +00:00
25 lines
828 B
Python
25 lines
828 B
Python
![]() |
import pytest
|
||
|
|
||
|
import gym
|
||
|
from gym.wrappers import RecordEpisodeStatistics
|
||
|
|
||
|
|
||
|
@pytest.mark.parametrize('env_id', ['CartPole-v0', 'Pendulum-v0'])
|
||
|
@pytest.mark.parametrize('deque_size', [2, 5])
|
||
|
def test_record_episode_statistics(env_id, deque_size):
|
||
|
env = gym.make(env_id)
|
||
|
env = RecordEpisodeStatistics(env, deque_size)
|
||
|
|
||
|
for n in range(5):
|
||
|
env.reset()
|
||
|
assert env.episode_return == 0.0
|
||
|
assert env.episode_length == 0
|
||
|
for t in range(env.spec.max_episode_steps):
|
||
|
_, _, done, info = env.step(env.action_space.sample())
|
||
|
if done:
|
||
|
assert 'episode' in info
|
||
|
assert all([item in info['episode'] for item in ['r', 'l', 't']])
|
||
|
break
|
||
|
assert len(env.return_queue) == deque_size
|
||
|
assert len(env.length_queue) == deque_size
|