/
githubmirror
/
gym
Обзор
Документация
Войти
/
githubmirror
/
gym
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/spaces/test_multibinary.py
19 строк
603 B
Mark Towers
Update Composite spaces with full coverage (#3047)
04 сен 2022, 01:39
Не верифицирован
04 сен 2022, 01:39
f39747d
Код
Авторство
О чём код?
import numpy as np from gym.spaces import MultiBinary def test_sample(): space = MultiBinary(4) sample = space.sample(mask=np.array([0, 0, 1, 1], dtype=np.int8)) assert np.all(sample == [0, 0, 1, 1]) sample = space.sample(mask=np.array([0, 1, 2, 2], dtype=np.int8)) assert sample[0] == 0 and sample[1] == 1 assert sample[2] == 0 or sample[2] == 1 assert sample[3] == 0 or sample[3] == 1 space = MultiBinary(np.array([2, 3])) sample = space.sample(mask=np.array([[0, 0, 0], [1, 1, 1]], dtype=np.int8)) assert np.all(sample == [[0, 0, 0], [1, 1, 1]]), sample