def test_cut_verbs_empty(): doc = get_doc_example("He") span = doc[2:4] verbs_extractor_instance = VerbsExtractor() cut_verbs_result = verbs_extractor_instance.cut_verbs({}, span, doc) assert not cut_verbs_result
def test_cut_verbs_dict(): doc = get_doc_example("He is a well trained athlete.") span = doc[3:5] verbs_extractor_instance = VerbsExtractor() cut_verbs_result = verbs_extractor_instance.cut_verbs({}, span, doc) assert isinstance(cut_verbs_result, dict)