def test_make_ngram(): '''Test the make ngram method''' ngram_freq = NgramFrequencies(1, 'test_file.txt') ngram_freq_2 = NgramFrequencies(2, 'test_file.txt') ngram_freq.make_ngram() ngram_freq_2.make_ngram() assert list(ngram_freq.dic_ngram.keys()) == [ 'ab', "dr", 'd', 'gh', 'COMMA', "ab's" ] assert list(ngram_freq_2.dic_ngram.keys()) == [ 'ab_dr', "dr_d", 'gh_COMMA', "COMMA_ab's" ]
def main(): file_name = input('enter file name: ') unigrams = NgramFrequencies(1, file_name) bigrams = NgramFrequencies(2, file_name) trigrams = NgramFrequencies(3, file_name) print("\n Top 10 unigrams:") unigrams.make_ngram() print("", *unigrams.top_n_freqs(10), sep="\n ") print("\n Top 10 bigrams:") bigrams.make_ngram() print("", *bigrams.top_n_freqs(10), sep="\n ") print("\n Top 10 trigrams:") trigrams.make_ngram() print("", *trigrams.top_n_freqs(10), sep="\n ")