test_hmm.py 文件源码

def test_valid_fit(self):
        obs = [np.array([1, 1]), np.array([[1, 1], [2, 2]])]
        called = [False, False]

        def init(x):
            self.assertEqual(len(x), len(obs))
            self.assertEqual(np.ndim(x[0]), 2)
            called[0] = True

        def fit(x):
            self.assertEqual(len(x), len(obs))
            self.assertEqual(np.ndim(x[0]), 2)
            called[1] = True

        self.hmm.init_callback = init
        self.hmm.fit_callback = fit
        self.hmm.fit(obs)
        self.assertEqual(self.hmm.n_features_, 2)
        self.assertTrue(called[0])
        self.assertTrue(called[1])

        called[0], called[1] = False, False
        self.hmm.fit(obs)
        self.assertFalse(called[0])
        self.assertTrue(called[1])