From 50664d953b0f51a27b46aee9f7ca5b2eab40d214 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 29 Oct 2023 13:55:46 +0100 Subject: [PATCH] Better implementation of blacklist to only test gym builtin --- ...test_all_gym_builtin_envs.py => test_gym_builtin_envs.py} | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) rename test/{test_all_gym_builtin_envs.py => test_gym_builtin_envs.py} (90%) diff --git a/test/test_all_gym_builtin_envs.py b/test/test_gym_builtin_envs.py similarity index 90% rename from test/test_all_gym_builtin_envs.py rename to test/test_gym_builtin_envs.py index f2eeac6..e131c52 100644 --- a/test/test_all_gym_builtin_envs.py +++ b/test/test_gym_builtin_envs.py @@ -12,12 +12,15 @@ GYM_IDS = [spec.id for spec in gym.envs.registry.values() if not isinstance(spec.entry_point, Callable) and "fancy_gym" not in spec.entry_point and 'make_bb_env_helper' not in spec.entry_point and 'jax' not in spec.id.lower() - and 'jax' not in spec.id.lower() + and 'shimmy' not in spec.id.lower() + and 'ale_py' not in spec.id.lower() + and 'tabular' not in spec.id.lower() and not re.match(r'GymV2.Environment', spec.id) ] GYM_MP_IDS = fancy_gym.ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS['all'] SEED = 1 +print(GYM_IDS) @pytest.mark.parametrize('env_id', GYM_IDS) def test_step_gym_functionality(env_id: str):