def test_word_filter(self): # print '== test_word_filter ==' res = s_text_filter.word_filter(self.short_text) self.assertEqual(res, ['week', 'ago', 'ordered', 'kindle', 'and', 'when', 'was', 'delivered', 'the', 'battery', 'was', 'critical', 'stage', 'and', 'would', 'not', 'power'])
def test_suffix_stem_dont_lose_words(self): terms = s_text_filter.word_filter(self.long_text) fixed_terms = s_text_filter.suffix_stem(terms) self.assertEqual(len(terms), len(fixed_terms))