From c91a4bc8a75cb671349fc48685415ebc3e984457 Mon Sep 17 00:00:00 2001 From: kngwyu Date: Thu, 4 Jun 2020 14:38:15 +0900 Subject: [PATCH] More detailed tests --- tests/test_envs.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_envs.py b/tests/test_envs.py index 209ab24..9a5400f 100644 --- a/tests/test_envs.py +++ b/tests/test_envs.py @@ -7,9 +7,13 @@ import pytest def test_ant_maze(maze_id): env = gym.make("Ant{}-v0".format(maze_id)) assert env.reset().shape == (30,) + s, _, _, _ = env.step(env.action_space.sample()) + assert s.shape == (30,) @pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS) def test_point_maze(maze_id): env = gym.make("Point{}-v0".format(maze_id)) assert env.reset().shape == (7,) + s, _, _, _ = env.step(env.action_space.sample()) + assert s.shape == (7,)