Make it work

tests/test_envs.py

@@ -0,0 +1,15 @@
+import gym
+import mujoco_maze
+import pytest
+
+
+@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
+def test_ant_maze(maze_id):
+    env = gym.make("AntMaze{}-v0".format(maze_id))
+    assert env.reset().shape == (30,)
+
+
+@pytest.mark.parametrize("maze_id", mujoco_maze.MAZE_IDS)
+def test_point_maze(maze_id):
+    env = gym.make("PointMaze{}-v0".format(maze_id))
+    assert env.reset().shape == (7,)