def test_addition_for_ended_in_k(self): self.assertEquals(morphology.addition_for_ended_in_k( u'майк'), set([u'маечк', u'маек'])) self.assertEquals(morphology.addition_for_ended_in_k( u'рубашк'), set([u'рубашечк', u'рубашек'])) self.assertEquals(morphology.addition_for_ended_in_k( u'футболк'), set([u'футболочк', u'футболок']))
def get_all_stems(self, word): stem = self.stemmer.stemWord(word) all_stems = set([stem,]) addition = morphology.addition_for_ended_in_k(stem) if addition: all_stems = all_stems | addition return all_stems