def testLikelihoodProbabilities(self):
filename = "../train-test-likelihood.pos"
hmm = HiddenMarkovModel(filename=filename)
expected = {('<s>', '<s>'):1.0, ('Pierre', 'NNP'):.5, (',',','):1.0, ('years', 'CD'): .666}
for key in expected:
self.assertAlmostEqual(hmm.getLikelihoodProbability(*key), expected[key], delta=.01)
def testPriorProbabilities(self):
filename = "../train-test-prior.pos"
hmm = HiddenMarkovModel(filename=filename)
self.assertAlmostEqual(hmm.getPriorProbability((',', 'CD')), .5, delta=.1)
