dodox/mujoco_maze
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add 2Rooms

Browse Source
This commit is contained in:
kngwyu
2020-06-30 01:38:02 +09:00
parent 767bd1891a
commit 142b42e34f
3 changed files with 66 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_envs.py
+2 -2
View File
@@ -4,7 +4,7 @@ import pytest
import mujoco_maze
@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
@pytest.mark.parametrize("maze_id", mujoco_maze.TaskRegistry.keys())
def test_ant_maze(maze_id):
env = gym.make("Ant{}-v0".format(maze_id))
assert env.reset().shape == (30,)
@@ -12,7 +12,7 @@ def test_ant_maze(maze_id):
assert s.shape == (30,)
@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
@pytest.mark.parametrize("maze_id", mujoco_maze.TaskRegistry.keys())
def test_point_maze(maze_id):
env = gym.make("Point{}-v0".format(maze_id))
assert env.reset().shape == (7,)