def test_resize_invalid_environment_shape(self):
with self.assertRaises(ValueError):
self.env.observation_space = Box(low=0,
high=255,
shape=(4, ),
dtype=np.uint8)
Resize(self.env, width=self.width, height=self.height)
def setup_method(self):
self.width = 16
self.height = 16
self.env = DummyDiscrete2DEnv()
self.env_r = Resize(DummyDiscrete2DEnv(),
width=self.width,
height=self.height)
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._width = 16
self._height = 16
self.env_r = Resize(self.env, width=self._width, height=self._height)
self.obs = self.env.reset()
self.obs_r = self.env_r.reset()
def test_baseline(self):
"""Test the baseline initialization."""
box_env = TfEnv(DummyBoxEnv())
deterministic_mlp_baseline = DeterministicMLPBaseline(env_spec=box_env)
gaussian_mlp_baseline = GaussianMLPBaseline(env_spec=box_env)
discrete_env = TfEnv(Resize(DummyDiscrete2DEnv(), width=64, height=64))
gaussian_conv_baseline = GaussianConvBaseline(
env_spec=discrete_env,
regressor_args=dict(
conv_filters=[32, 32],
conv_filter_sizes=[1, 1],
conv_strides=[1, 1],
conv_pads=["VALID", "VALID"],
hidden_sizes=(32, 32)))
self.sess.run(tf.global_variables_initializer())
deterministic_mlp_baseline.get_param_values(trainable=True)
gaussian_mlp_baseline.get_param_values(trainable=True)
gaussian_conv_baseline.get_param_values(trainable=True)
def test_resize_invalid_environment_type(self):
with self.assertRaises(ValueError):
self.env.observation_space = Discrete(64)
Resize(self.env, width=self.width, height=self.height)
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))