mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-19 13:32:03 +00:00
947b857bd4
* Move tests to root with automatic PyCharm import refactoring. This will likely fail some tests * Changed entry point for a registration test env. * Move a stray lunar_lander test to tests/envs/... * black * Change the version from which importlib_metadata is replaced with importlib.metadata. Also requiring installing importlib_metadata for python 3.8 now. ??????????? * Undo last commit
33 lines
768 B
Python
33 lines
768 B
Python
import pytest
|
|
|
|
import numpy as np
|
|
|
|
import gym
|
|
from gym.wrappers import RescaleAction
|
|
|
|
|
|
def test_rescale_action():
|
|
env = gym.make("CartPole-v1")
|
|
with pytest.raises(AssertionError):
|
|
env = RescaleAction(env, -1, 1)
|
|
del env
|
|
|
|
env = gym.make("Pendulum-v1")
|
|
wrapped_env = RescaleAction(gym.make("Pendulum-v1"), -1, 1)
|
|
|
|
seed = 0
|
|
env.seed(seed)
|
|
wrapped_env.seed(seed)
|
|
|
|
obs = env.reset()
|
|
wrapped_obs = wrapped_env.reset()
|
|
assert np.allclose(obs, wrapped_obs)
|
|
|
|
obs, reward, _, _ = env.step([1.5])
|
|
with pytest.raises(AssertionError):
|
|
wrapped_env.step([1.5])
|
|
wrapped_obs, wrapped_reward, _, _ = wrapped_env.step([0.75])
|
|
|
|
assert np.allclose(obs, wrapped_obs)
|
|
assert np.allclose(reward, wrapped_reward)
|