Add TRoom

2020-09-17 01:27:38 +09:00
parent 4a195d9848
commit 8f630b58c2
3 changed files with 50 additions and 2 deletions
@@ -20,3 +20,11 @@ def test_point_maze(maze_id):
        assert env.reset().shape == (7,)
        s, _, _, _ = env.step(env.action_space.sample())
        assert s.shape == (7,)
+
+
+@pytest.mark.parametrize("v", [0, 1])
+def test_maze_args(v):
+    env = gym.make(f"PointTRoom-v{v}", task_kwargs={"goals": [(-2.0, 4.0)]})
+    assert env.reset().shape == (7,)
+    s, _, _, _ = env.step(env.action_space.sample())
+    assert s.shape == (7,)