mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-30 01:50:19 +00:00
32 lines
831 B
Python
32 lines
831 B
Python
import numpy as np
|
|
import pytest
|
|
|
|
import gym
|
|
from gym.wrappers import RescaleAction
|
|
|
|
|
|
def test_rescale_action():
|
|
env = gym.make("CartPole-v1", disable_env_checker=True)
|
|
with pytest.raises(AssertionError):
|
|
env = RescaleAction(env, -1, 1)
|
|
del env
|
|
|
|
env = gym.make("Pendulum-v1", disable_env_checker=True)
|
|
wrapped_env = RescaleAction(
|
|
gym.make("Pendulum-v1", disable_env_checker=True), -1, 1
|
|
)
|
|
|
|
seed = 0
|
|
|
|
obs = env.reset(seed=seed)
|
|
wrapped_obs = wrapped_env.reset(seed=seed)
|
|
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)
|