def test_keyword_counts(): cli = CLI() document = "This is a test document." keyword_counts = {"this": 1, "is": 1, "a": 1, "test": 1, "document": 1} cli._process_document(document) assert cli.keyword_counts == keyword_counts
def test_set_document(): cli = CLI() document = "This is a test document." cli._process_document(document) assert cli.document == document