def test_save_load(self): filename = os.path.join(self.tempdir, 'model.pkl') expected = self.model.model.components_ self.model.save(filename) tmp_model = TopicModel.load(filename) observed = tmp_model.model.components_ self.assertEqual(observed.shape, expected.shape) self.assertTrue(np.equal(observed, expected).all())
def test_save_load(tmpdir, model): filepath = str(tmpdir.join("model.pkl")) expected = model.model.components_ model.save(filepath) tmp_model = TopicModel.load(filepath) observed = tmp_model.model.components_ assert observed.shape == expected.shape assert np.equal(observed, expected).all()