def test_set_params(self): nn = ZeroInput(num_iid=2, seed=42) UPDATE = {"seed": 635} nn.update_params(UPDATE) params = nn.get_params() params.pop("type") self.assertDictEqual(params, {"num_iid": 2, "seed": 42, **UPDATE})
def test_get_params(self): nn = ZeroInput(num_iid=2, seed=42) params = nn.get_params() params.pop("type") self.assertDictEqual(params, {"num_iid": 2, "seed": 42})