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)