def test_valid_input(self): """extract_keywords should return keywords from string.""" for inp, out in self.DATA: res = extract_keywords(inp, p=self.p, step=self.step, desired_words=self.dw) self.assertEqual(2, len(res)) for x in out: self.assertEqual(True, x in res)
def test_invalid_string(self): """extract_keywords should return empty list on invalid string.""" self.assertEqual([], extract_keywords(STR["PUNCTUATION"])) self.assertEqual([], extract_keywords(STR["INVALID"]))
def test_empty_input(self): """extract_keywords should return empty list on empty input.""" self.assertEqual([], extract_keywords(""))