/
githubmirror
/
gym
Обзор
Документация
Войти
/
githubmirror
/
gym
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/wrappers/test_human_rendering.py
32 строки
828 B
Mark Towers
Rename "rgb_array" to "rgb_array_list" and "single_rgb_array" to "rgb_array" (#3040)
01 сен 2022, 16:06
Не верифицирован
01 сен 2022, 16:06
799c8d2
Код
Авторство
О чём код?
import re import pytest import gym from gym.wrappers import HumanRendering def test_human_rendering(): for mode in ["rgb_array", "rgb_array_list"]: env = HumanRendering( gym.make("CartPole-v1", render_mode=mode, disable_env_checker=True) ) 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="human") with pytest.raises( AssertionError, match=re.escape( "Expected env.render_mode to be one of 'rgb_array' or 'rgb_array_list' but got 'human'" ), ): HumanRendering(env) env.close()