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)