File size: 403 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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, )
|