Esempio n. 1
0
 def test_fits_sum_to_one(self):
     mc = MarkovChain(self.corpus, 1)
     mc.fit()
     p = mc.matrix_list[0].p
     self.assertEqual(sum(p[mc.convert_word_to_number('alpha')]),1)
     self.assertEqual(sum(p[mc.convert_word_to_number('beta')]),1)
     self.assertEqual(sum(p[mc.convert_word_to_number('gamma')]),1)
     self.assertEqual(sum(p[mc.convert_word_to_number('delta')]),1)