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))
