mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-28 09:17:18 +00:00
3a8daafce1
* removed return_info, made info dict mandatory in reset * tenatively removed deprecated seed api for environments * added more info type checks to wrapper tests * formatting/style compliance * addressed some comments * polish to address review * fixed tests after merge, and added a test of the return_info deprecation assertion if found in reset signature * some organization of env_checker tests, reverted a probably merge error * added deprecation check for seed function in env * updated docstring * removed debug prints, tweaked test_check_seed_deprecation * changed return_info deprecation check from assertion to warning * fixes to vector envs, now should be correctly structured * added some explanation and typehints for mockup depcreated return info reset function * re-removed seed function from vector envs * added explanation to _reset_return_info_type and changed the return statement
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
import gym
|
|
from gym.wrappers import FrameStack
|
|
|
|
try:
|
|
import lz4
|
|
except ImportError:
|
|
lz4 = None
|
|
|
|
|
|
@pytest.mark.parametrize("env_id", ["CartPole-v1", "Pendulum-v1", "CarRacing-v2"])
|
|
@pytest.mark.parametrize("num_stack", [2, 3, 4])
|
|
@pytest.mark.parametrize(
|
|
"lz4_compress",
|
|
[
|
|
pytest.param(
|
|
True,
|
|
marks=pytest.mark.skipif(
|
|
lz4 is None, reason="Need lz4 to run tests with compression"
|
|
),
|
|
),
|
|
False,
|
|
],
|
|
)
|
|
def test_frame_stack(env_id, num_stack, lz4_compress):
|
|
env = gym.make(env_id, disable_env_checker=True)
|
|
shape = env.observation_space.shape
|
|
env = FrameStack(env, num_stack, lz4_compress)
|
|
assert env.observation_space.shape == (num_stack,) + shape
|
|
assert env.observation_space.dtype == env.env.observation_space.dtype
|
|
|
|
dup = gym.make(env_id, disable_env_checker=True)
|
|
|
|
obs, _ = env.reset(seed=0)
|
|
dup_obs, _ = dup.reset(seed=0)
|
|
assert np.allclose(obs[-1], dup_obs)
|
|
|
|
for _ in range(num_stack**2):
|
|
action = env.action_space.sample()
|
|
dup_obs, _, dup_done, _ = dup.step(action)
|
|
obs, _, done, _ = env.step(action)
|
|
|
|
assert dup_done == done
|
|
assert np.allclose(obs[-1], dup_obs)
|
|
|
|
if done:
|
|
break
|
|
|
|
assert len(obs) == num_stack
|