Swimmer

tests/test_envs.py

@@ -26,6 +26,19 @@ def test_point_maze(maze_id):
             assert s.shape == (7,)
 
 
+@pytest.mark.parametrize("maze_id", mujoco_maze.TaskRegistry.keys())
+def test_swimmer_maze(maze_id):
+    if "Fall" in maze_id or "Push" in maze_id:
+        return
+    for i in range(2):
+        env = gym.make(f"Swimmer{maze_id}-v{i}")
+        s0 = env.reset()
+        s, _, _, _ = env.step(env.action_space.sample())
+        if not env.unwrapped._top_down_view:
+            assert s0.shape == (11,)
+            assert s.shape == (11,)
+
+
 @pytest.mark.parametrize("v", [0, 1])
 def test_maze_args(v):
     env = gym.make(f"PointTRoom-v{v}", task_kwargs={"goals": [(-2.0, -3.0)]})