Пример #1
0
 def test_get_params_dict(self):
     params = DiscreteModel(n_actions=10).get_params_dict()
     assert "actions" in params
     assert "dtype" in params["actions"]
     assert params["actions"]["dtype"] == numpy.int_
Пример #2
0
 def test_init(self):
     env_1 = DiscreteModel(n_actions=10)
     assert env_1.n_actions == 10
     env_2 = DiscreteModel(env=DummyEnv())
     assert env_2.n_actions == 10