def test_save_classifier(self):
mock_fan_yang = Mock(FanYang)
mock_fan_yang.classifier_filename = 'data/test-fan-yang.classifier'
mock_fan_yang.classifier = 'test'
mock_fan_yang.save_classifier = FanYang.save_classifier.__get__(mock_fan_yang)
mock_fan_yang.save_classifier()
self.assertTrue(os.path.isfile(mock_fan_yang.classifier_filename))
os.remove(mock_fan_yang.classifier_filename)
def test_load_classifier(self):
mock_fan_yang = Mock(FanYang)
file_name = 'data/fan-yang.classifier'
with open(file_name, mode='w') as f:
cPickle.dump(file_name, f)
mock_fan_yang.classifier_filename = file_name
mock_fan_yang.classifier = None
mock_fan_yang.load_classifier = FanYang.load_classifier.__get__(mock_fan_yang)
mock_fan_yang.load_classifier()
self.assertIsNotNone(mock_fan_yang.classifier)
os.remove(file_name)
def test_train(self):
train_set_filename = 'data/train-set.txt'
with codecs.open(train_set_filename, mode='wb', encoding='utf-8') as f:
f.write(u'1,1,1,1,0.999,1\n'
u'0,0,0,0,0.111,0\n')
mock_fan_yang = Mock(FanYang)
mock_fan_yang.train_data_filename = train_set_filename
mock_fan_yang.feature_names = FanYang.feature_names
mock_fan_yang.classifier = None
mock_fan_yang.train = FanYang.train.__get__(mock_fan_yang)
mock_fan_yang.train()
self.assertIsNotNone(mock_fan_yang.classifier)
os.remove(train_set_filename)
