Dominik Roth
a559f92562
In prevous gym versions executing a step returned obs, reward, done, info = env.step(...) With the switch to gymnasium this has changed to obs, reward, terminated, truncated, info = env.step(...) We also made the code a bit more self explainatory.
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from itertools import chain
|
|
|
|
import pytest
|
|
from metaworld.envs import ALL_V2_ENVIRONMENTS_GOAL_OBSERVABLE
|
|
|
|
import fancy_gym
|
|
from test.utils import run_env, run_env_determinism
|
|
|
|
METAWORLD_IDS = [f'metaworld:{env.split("-goal-observable")[0]}' for env, _ in
|
|
ALL_V2_ENVIRONMENTS_GOAL_OBSERVABLE.items()]
|
|
METAWORLD_MP_IDS = list(chain(*fancy_gym.ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS.values()))
|
|
SEED = 1
|
|
|
|
|
|
@pytest.mark.parametrize('env_id', METAWORLD_IDS)
|
|
def test_step_metaworld_functionality(env_id: str):
|
|
"""Tests that step environments run without errors using random actions."""
|
|
run_env(env_id)
|
|
|
|
|
|
@pytest.mark.parametrize('env_id', METAWORLD_IDS)
|
|
def test_step_metaworld_determinism(env_id: str):
|
|
"""Tests that for step environments identical seeds produce identical trajectories."""
|
|
run_env_determinism(env_id, SEED)
|
|
|
|
|
|
@pytest.mark.parametrize('env_id', METAWORLD_MP_IDS)
|
|
def test_bb_metaworld_functionality(env_id: str):
|
|
"""Tests that black box environments run without errors using random actions."""
|
|
run_env(env_id)
|
|
|
|
|
|
@pytest.mark.parametrize('env_id', METAWORLD_MP_IDS)
|
|
def test_bb_metaworld_determinism(env_id: str):
|
|
"""Tests that for black box environment identical seeds produce identical trajectories."""
|
|
run_env_determinism(env_id, SEED)
|