def testTrainClassify(self): model = BayesModel() model.train(Texts.pushkin2, True) model.train(Texts.udaff, False) return model.classify(Texts.udaff2).addCallback(lambda result: self.assert_(not result)). \ addCallback(lambda _: model.classify(Texts.pushkin)).addCallback(lambda result: self.assert_(result))
def testTrainClassifySimple(self): model = BayesModel() model.train(u"мама мыла раму", True) model.train(u"папа пошел гулять", False) return model.classify(u"папа").addCallback(lambda result: self.assert_(not result)). \ addCallback(lambda _: model.classify(u"мама")).addCallback(lambda result: self.assert_(result))