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_
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