DI-engine/dizoo/mujoco/envs/test/test_mujoco_gym_env.py

import pytest
import gym


@pytest.mark.envtest
def test_shapes():
    from dizoo.mujoco.envs import mujoco_gym_env
    ant = gym.make('AntTruncatedObs-v2')
    assert ant.observation_space.shape == (27, )
    assert ant.action_space.shape == (8, )
    humanoid = gym.make('HumanoidTruncatedObs-v2')
    assert humanoid.observation_space.shape == (45, )
    assert humanoid.action_space.shape == (17, )