def test_update_empty_prefix(self): prefix = markov._update_prefix(2, "", "I") self.assertEqual("I", prefix)
def test_update_prefix_strip_end_space(self): prefix = markov._update_prefix(2, "I have ", "a") self.assertEqual("have a", prefix)
def test_update_short_prefix(self): prefix = markov._update_prefix(2, "I", "am") self.assertEqual("I am", prefix)