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})
