test: Search for jdrones envs rather than manually specify them

iwishiwasaneagle · Mar 7, 2023 · 4198604 · 4198604
1 parent 0408efe
commit 4198604
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/tests/test_gym_api.py b/tests/test_gym_api.py
@@ -7,18 +7,14 @@
 
 @pytest.mark.integration
 @pytest.mark.parametrize(
-    "env,kwargs",
-    [
-        ("PyBulletDroneEnv-v0", {}),
-        ("NonLinearDynamicModelDroneEnv-v0", {}),
-        ("LinearDynamicModelDroneEnv-v0", {}),
-        ("LQRDroneEnv-v0", {}),
-        ("LQRPositionDroneEnv-v0", {}),
-        ("PolyPositionDroneEnv-v0", {}),
-    ],
+    "env",
+    filter(
+        lambda item: "jdrones" in str(item.entry_point),
+        tuple(gymnasium.registry.values()),
+    ),
 )
-def test_make(env, kwargs):
-    assert isinstance(gymnasium.make(env, **kwargs), gymnasium.Env)
+def test_make(env):
+    assert isinstance(gymnasium.make(env), gymnasium.Env)
 
 
 @pytest.mark.integration