File size: 793 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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()
|