class TestMockRemoteBSTClassifier(AsyncHTTPTestCase):
"""Test the RemoteBSTClassifier against a mock server"""
def get_app(self):
app = Application([
(r'/classifier/create/(\w+)', MockAICreateHandler),
(r'/classifier/(\w+)', MockAIObjectHandler),
(r'/classifier/(\w+)/train', MockAITrainHandler),
(r'/classifier/(\w+)/predict', MockAIPredictHandler),
])
self.clf = RemoteBSTClassifier(self.get_url('/'), "bnn")
self.assertIsNotNone(self.clf)
return app
@gen_test
def test_mock_create_model(self):
clf_id = yield self.clf._create_model()
self.assertEqual(clf_model_id, clf_id)
@gen_test
def test_mock_async_fit(self):
yield self.clf.async_fit(train_X, train_y)
self.assertIsNotNone(self.clf.training_error)
def test_mock_fit(self):
self.clf.fit(train_X, train_y)
