def test_stack_frames_axis(self):
env = StackFrames(DummyDiscrete2DEnv(random=False),
n_frames=self.n_frames,
axis=0)
env.reset()
obs, _, _, _ = env.step(1)
assert obs.shape[0] == self.n_frames
env = StackFrames(DummyDiscrete2DEnv(random=False),
n_frames=self.n_frames,
axis=2)
env.reset()
obs, _, _, _ = env.step(1)
assert obs.shape[2] == self.n_frames
def setUp(self):
self.n_frames = 4
self.env = TfEnv(DummyDiscrete2DEnv(random=False))
self.env_s = TfEnv(
StackFrames(
DummyDiscrete2DEnv(random=False), n_frames=self.n_frames))
self.width, self.height = self.env.observation_space.shape
def test_stack_frames_invalid_environment_shape(self):
with pytest.raises(ValueError):
self.env.observation_space = gym.spaces.Box(low=0,
high=255,
shape=(4, ),
dtype=np.uint8)
StackFrames(self.env, n_frames=4)
def setUp(self):
self.shape = (50, 50)
self.env = mock.Mock()
self.env.observation_space = Box(
low=0, high=255, shape=self.shape, dtype=np.uint8)
self.env.reset.return_value = np.zeros(self.shape)
self.env.step.side_effect = self._step
self._n_frames = 4
self.env_s = StackFrames(self.env, n_frames=self._n_frames)
self.obs = self.env.reset()
self.obs_s = self.env_s.reset()
self.frame_width = self.env.observation_space.shape[0]
self.frame_height = self.env.observation_space.shape[1]
def test_stack_frames_invalid_environment_type(self):
with self.assertRaises(ValueError):
self.env.observation_space = Discrete(64)
StackFrames(self.env, n_frames=4)
def test_stack_frames_invalid_environment_type(self):
with pytest.raises(ValueError):
self.env.observation_space = gym.spaces.Discrete(64)
StackFrames(self.env, n_frames=4)
def setup_method(self):
self.n_frames = 4
self.env = DummyDiscrete2DEnv(random=False)
self.env_s = StackFrames(DummyDiscrete2DEnv(random=False),
n_frames=self.n_frames)
self.width, self.height = self.env.observation_space.shape
def test_invalid_axis_raises_error(self):
with pytest.raises(ValueError):
StackFrames(DummyDiscrete2DEnv(random=False),
n_frames=self.n_frames,
axis=5)
