class TestDevRemoteBSTClassifier(AsyncTestCase): """Test the RemoteBSTClassifier against the dev server""" def setUp(self): super().setUp() conf = get_conf() self.model_params = { "hiddenLayers": [5, 6, 7, 8], "learningRate": 0.4, } url = conf['aimetrics']['dev']['hosts']['ai']['base_url'] self.clf = RemoteBSTClassifier(url, "bnn", model_params=self.model_params) self.assertIsNotNone(self.clf) @gen_test def test_async_fit(self): yield self.clf.async_fit(train_X, train_y) # test params m = yield self.clf.get_model() self.assertEqual(2+ len(self.model_params['hiddenLayers']), len(m["model"]["layers"])) def tearDown(self): if self.clf.model_id: self.clf.destroy_model()