Files
Gymnasium/tests/experimental/wrappers/test_lambda_action.py
2022-12-10 22:04:14 +00:00

20 lines
760 B
Python

"""Test suite for LambdaActionV0."""
from gymnasium.experimental.wrappers import LambdaActionV0
from gymnasium.spaces import Box
from tests.experimental.wrappers.utils import record_action_step
from tests.testing_env import GenericTestEnv
def test_lambda_action_wrapper():
"""Tests LambdaAction through checking that the action taken is transformed by function."""
env = GenericTestEnv(step_func=record_action_step)
wrapped_env = LambdaActionV0(env, lambda action: action - 2, Box(2, 3))
sampled_action = wrapped_env.action_space.sample()
assert sampled_action not in env.action_space
_, _, _, _, info = wrapped_env.step(sampled_action)
assert info["action"] in env.action_space
assert sampled_action - 2 == info["action"]