Loading tests/borch/test_module.py +9 −0 Original line number Diff line number Diff line Loading @@ -362,3 +362,12 @@ def test_observed_del(observed): def test_forward_on_observed_raises_error(observed): with pytest.raises(RuntimeError): observed() def test_set_rv_overwrites_attr_in_dict(): model = module.Module() model.test = 1 assert model.__dict__['test'] ==1 model.test = dist.Normal(0, 1) assert 'test' not in model.__dict__ Loading
tests/borch/test_module.py +9 −0 Original line number Diff line number Diff line Loading @@ -362,3 +362,12 @@ def test_observed_del(observed): def test_forward_on_observed_raises_error(observed): with pytest.raises(RuntimeError): observed() def test_set_rv_overwrites_attr_in_dict(): model = module.Module() model.test = 1 assert model.__dict__['test'] ==1 model.test = dist.Normal(0, 1) assert 'test' not in model.__dict__