mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-12 18:13:35 +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()
|