mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-07 00:11:46 +00:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""Test suite of AddWhiteNoise and ObstructView wrapper."""
|
|
|
|
import gymnasium as gym
|
|
from gymnasium.wrappers import AddWhiteNoise, HumanRendering, ObstructView
|
|
|
|
|
|
def test_white_noise_rendering():
|
|
for mode in ["rgb_array"]:
|
|
env = gym.make("CartPole-v1", render_mode=mode, disable_env_checker=True)
|
|
env = AddWhiteNoise(env, probability_of_noise_per_pixel=0.5)
|
|
env = HumanRendering(env)
|
|
|
|
assert env.render_mode == "human"
|
|
env.reset()
|
|
|
|
for _ in range(75):
|
|
_, _, terminated, truncated, _ = env.step(env.action_space.sample())
|
|
if terminated or truncated:
|
|
env.reset()
|
|
|
|
env.close()
|
|
|
|
env = gym.make("CartPole-v1", render_mode=mode, disable_env_checker=True)
|
|
env = ObstructView(env, obstructed_pixels_ratio=0.5, obstruction_width=100)
|
|
env = HumanRendering(env)
|
|
|
|
assert env.render_mode == "human"
|
|
env.reset()
|
|
|
|
for _ in range(75):
|
|
_, _, terminated, truncated, _ = env.step(env.action_space.sample())
|
|
if terminated or truncated:
|
|
env.reset()
|
|
|
|
env.close()
|