DI-engine/dizoo/maze/envs/test_maze_env.py

import pytest
import os
import numpy as np
from dizoo.maze.envs.maze_env import Maze
from easydict import EasyDict
import copy


@pytest.mark.envtest
class TestMazeEnv:

    def test_maze(self):
        env = Maze(EasyDict({'size': 16}))
        env.seed(314)
        assert env._seed == 314
        obs = env.reset()
        assert obs.shape == (16, 16, 3)
        min_val, max_val = 0, 3
        for i in range(100):
            random_action = np.random.randint(min_val, max_val, size=(1, ))
            timestep = env.step(random_action)
            print(timestep)
            print(timestep.obs.max())
            assert isinstance(timestep.obs, np.ndarray)
            assert isinstance(timestep.done, bool)
            if timestep.done:
                env.reset()
        env.close()