from kwic import kwic

empty_document = ""
design_words_doc = "Design is hard.\nLet's just implement."
goodbye_buddy_doc = "Hello there.\nHello there, buddy.\nHello and goodbye, buddy.\nHello is like buddy Goodbye!"
hello_buddy_periods = "Hello there.  Hello there, buddy.    Hello and goodbye, buddy. Hello is like buddy Goodbye!"

hello_buddy_periods_output = ["Hello there.", "  Hello there, buddy.", "    Hello and goodbye, buddy.",
                              " Hello is like buddy Goodbye!"]

if __name__ == "__main__":
    # Ensure empty input gives empty output
    assert(kwic(empty_document) == [])

    # Ensure real input does not produce empty output
    assert(kwic(design_words_doc) != [])

    # Make sure it's broken into two lines
    assert(len(kwic(design_words_doc)) == 2)

    # Make sure it's broken into four line
    assert(len(kwic(goodbye_buddy_doc)) == 4)

    # Make sure line with just periods shows up as itself
    assert(kwic(hello_buddy_periods)[0] == hello_buddy_periods)

    # Make sure it's broken into four lines once it's broken by periods instead
    # Also, this time it keeps the ending period like it's supposed to
    assert(kwic(hello_buddy_periods, periodsToBreaks=True) == hello_buddy_periods_output)