Merge pull request #247 from Farama-Foundation/othertests

reenable environment specific tests
Farama-Foundation · Oct 14, 2024 · a65445e · a65445e
2 parents ce1aa31 + f1ec4d7
commit a65445e
Show file tree

Hide file tree

Showing 6 changed files with 22 additions and 0 deletions.
diff --git a/tests/envs/franka_kitchen/test_kitchen_env.py b/tests/envs/franka_kitchen/test_kitchen_env.py
@@ -3,11 +3,14 @@
 import gymnasium as gym
 import pytest
 
+import gymnasium_robotics
 from gymnasium_robotics.envs.franka_kitchen.kitchen_env import (
     OBS_ELEMENT_GOALS,
     OBS_ELEMENT_INDICES,
 )
 
+gym.register_envs(gymnasium_robotics)
+
 TASKS = ["microwave", "kettle"]
 
 

diff --git a/tests/envs/hand/test_manipulate.py b/tests/envs/hand/test_manipulate.py
@@ -3,6 +3,9 @@
 import gymnasium as gym
 import pytest
 
+import gymnasium_robotics
+
+gym.register_envs(gymnasium_robotics)
 ENVIRONMENT_IDS = (
     #  "HandManipulateEgg-v0",
     #  "HandManipulatePen-v0",

diff --git a/tests/envs/hand/test_manipulate_touch_sensors.py b/tests/envs/hand/test_manipulate_touch_sensors.py
@@ -3,6 +3,10 @@
 import gymnasium as gym
 import pytest
 
+import gymnasium_robotics
+
+gym.register_envs(gymnasium_robotics)
+
 ENVIRONMENT_IDS = (
     "HandManipulateEgg_ContinuousTouchSensors-v1",
     "HandManipulatePen_BooleanTouchSensors-v1",

diff --git a/tests/envs/hand/test_reach.py b/tests/envs/hand/test_reach.py
@@ -2,6 +2,10 @@
 
 import gymnasium as gym
 
+import gymnasium_robotics
+
+gym.register_envs(gymnasium_robotics)
+
 
 def test_serialize_deserialize():
     env1 = gym.make("HandReach-v2", distance_threshold=1e-6)

diff --git a/tests/envs/maze/test_ant_maze.py b/tests/envs/maze/test_ant_maze.py
@@ -2,6 +2,10 @@
 import numpy as np
 import pytest
 
+import gymnasium_robotics
+
+gym.register_envs(gymnasium_robotics)
+
 
 @pytest.mark.parametrize("version", ["v4", "v5"])
 def test_reset(version):

diff --git a/tests/envs/maze/test_point_maze.py b/tests/envs/maze/test_point_maze.py
@@ -1,6 +1,10 @@
 import gymnasium as gym
 import numpy as np
 
+import gymnasium_robotics
+
+gym.register_envs(gymnasium_robotics)
+
 
 def test_reset():
     """Check that PointMaze does not reset into a success state."""