def test_title_method_without_title(): document = build_document( ("This is sentence", "This is another one",), ("And some next sentence but no heading",) ) summarizer = EdmundsonSummarizer() summarizer.null_words = ("this", "is", "some", "and",) sentences = summarizer.title_method(document, 10) assert list(map(to_unicode, sentences)) == [ "This is sentence", "This is another one", "And some next sentence but no heading", ]
def test_title_method_1(): document = build_document_from_string(""" # This is cool heading Because I am sentence I like words And because I am string I like characters # blank and heading This is next paragraph because of blank line above Here is the winner because contains words like cool and heading """) summarizer = EdmundsonSummarizer() summarizer.null_words = ("this", "is", "I", "am", "and",) sentences = summarizer.title_method(document, 1) assert list(map(to_unicode, sentences)) == [ "Here is the winner because contains words like cool and heading", ]
def test_title_method_without_null_words(): summarizer = EdmundsonSummarizer() with pytest.raises(ValueError): summarizer.title_method(build_document(), 10)
def test_title_method_with_empty_document(): summarizer = EdmundsonSummarizer() summarizer.null_words = ("ba", "bb", "bc",) sentences = summarizer.title_method(build_document(), 10) assert list(map(to_unicode, sentences)) == []