def test_MockModel(self): mp = MockPosterior() mm = MockModel(mp) X = torch.empty(0) self.assertEqual(mm.posterior(X), mp) self.assertEqual(mm.num_outputs, 0) mm.state_dict() mm.load_state_dict()