def test_laplace_produces_expected_values():
lm = LanguageModel(3)
data = open_file('kn_test.txt')
lm.train(data)
assert lm.laplace_evaluate(['this', 'shall', 'train', 'PAD']) \
== -2.890371757896165
assert lm.laplace_evaluate(['dog', 'text', '.', 'PAD']) \
== (math.log(1 / 9) + math.log(1 / 2))
def test_laplace_produces_expected_values2():
lm = LanguageModel(1)
data = open_file('kn_test.txt')
lm.train(data)
assert lm.laplace_evaluate(['text']) == math.log(3 / 12)
assert lm.laplace_evaluate(['dog']) == math.log(1 / 12)
