def test_saveClassifier_loadClassifier(self): temp = Classifier.train(self.infos) c = temp["classifier"] Classifier.saveClassifier(c, "whentest") c2 = Classifier.loadClassifier("whentest") assert isinstance(c2, nltk.classify.naivebayes.NaiveBayesClassifier) os.remove("whentest.classifier")
def test_train(self): temp = Classifier.train(self.infos) assert "classifier" in temp assert "fiturs" in temp