def test_process_line_reads_trigrams(): line = 'This is a test line.' trigrams = {} expected = {('This', 'is'): ['a'], ('is', 'a'): ['test'], ('a', 'test'): ['line.']} gen_pac_trigrams.process_name(line, trigrams) assert trigrams == expected
def test_process_ignores_whitespace(): line = '\t\n One \t\n two three\n\t\t\n ' trigrams = {} expected = {('One', 'two'): ['three']} gen_pac_trigrams.process_name(line, trigrams) assert trigrams == expected
def test_process_line_handles_three_words(): line = 'One two three' trigrams = {} expected = {('One', 'two'): ['three']} gen_pac_trigrams.process_name(line, trigrams) assert trigrams == expected
def test_process_line_ignores_less_than_three_words(): line = 'One two' trigrams = {} expected = {} gen_pac_trigrams.process_name(line, trigrams) assert trigrams == expected
def test_process_line_ignores_blank_line(): line = '' trigrams = {} expected = {} gen_pac_trigrams.process_name(line, trigrams) assert trigrams == expected