def test_kn_produces_expected_values():
lm = LanguageModel(3)
data = open_file('kn_test.txt')
lm.train(data)
assert lm.kn_evaluate(['text', 'shall', 'train']) == -2.0770634192748685
assert lm.kn_evaluate(['this', 'text', 'dog']) == -3.1656313103493887
assert lm.kn_evaluate(['the', 'brown', 'cat']) == -2.4724841297894433
def test_kn_produces_expected_values_n4():
lm = LanguageModel(4)
data = open_file('kn_test.txt')
lm.train(data)
assert lm.kn_evaluate(['shall', 'train', 'text',
'.']) == -0.7742507185722116