def test_predict_next_sentence_more_words_ideal(self):
ngram = NGramTrie(2)
ngram.gram_log_probabilities = {(1, 2): -0.18, (1, 3): -1.79, (2, 3): -1, (2, 4): -2, (3, 4): -0.1,
(3, 5): -1.8}
actual_res = ngram.predict_next_sentence((1,))
expected_res = [1, 2, 3, 4]
self.assertEqual(actual_res, expected_res)
def test_predict_next_sentence_wrong_size(self):
ngram = NGramTrie(2)
actual_res = ngram.predict_next_sentence((1, 2))
expected_res = []
self.assertEqual(actual_res, expected_res)
def test_predict_next_sentence_no_match(self):
ngram = NGramTrie(2)
ngram.gram_log_probabilities = {(4, 2): -0.18, (4, 3): -1.79}
actual_res = ngram.predict_next_sentence((1,))
expected_res = [1]
self.assertEqual(actual_res, expected_res)
def test_predict_next_sentence_simple_ideal(self):
ngram = NGramTrie(2)
ngram.gram_log_probabilities = {(1, 2): -0.18, (1, 3): -1.79}
actual_res = ngram.predict_next_sentence((1,))
expected_res = [1, 2]
self.assertEqual(actual_res, expected_res)
def test_predict_next_sentence_not_a_tuple(self):
ngram = NGramTrie(2)
actual_res = ngram.predict_next_sentence([1])
expected_res = []
self.assertEqual(actual_res, expected_res)