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
Beispiel #2
0
 def setup_method(self):
     self.width = 16
     self.height = 16
     self.env = DummyDiscrete2DEnv()
     self.env_r = Resize(
         DummyDiscrete2DEnv(), width=self.width, height=self.height)
Beispiel #3
0
 def setUp(self):
     self.width = 16
     self.height = 16
     self.env = TfEnv(DummyDiscrete2DEnv())
     self.env_r = TfEnv(
         Resize(DummyDiscrete2DEnv(), width=self.width, height=self.height))
 def test_invalid_axis_raises_error(self):
     with pytest.raises(ValueError):
         StackFrames(DummyDiscrete2DEnv(random=False),
                     n_frames=self.n_frames,
                     axis=5)