def test_top_n_grams_inappropriate(self): ngram = NGramTrie(2) bad_inputs = [[], (), {}, None, True, '', -1, 0, 9.22] expected = () for bad_input in bad_inputs: actual = ngram.top_n_grams(bad_input) self.assertEqual(expected, actual)
def test_top_n_grams_more(self): ngram = NGramTrie(2) top_n = 2000000 ngram.n_gram_frequencies = {(1, 2): 100, (2, 3): 123, (3, 4): 12345} expected = ((3, 4), (2, 3), (1, 2)) actual = ngram.top_n_grams(top_n) self.assertEqual(expected, actual)