ADD: default apis for text/image with tests

This commit is contained in:
Chris Lee
2015-06-03 15:37:01 -04:00
parent 49a94d2986
commit adf8295f82
2 changed files with 9 additions and 2 deletions
+7
View File
@@ -136,6 +136,13 @@ class BatchAPIRun(unittest.TestCase):
self.assertTrue(isinstance(response, list))
self.assertTrue(set(response[0].keys()) == set(config.TEXT_APIS))
def test_default_multi_api_text(self):
test_data = ['clearly an english sentence']
response = batch_predict_text(test_data, api_key=self.api_key)
self.assertTrue(isinstance(response, list))
self.assertTrue(set(response[0].keys()) == set(config.TEXT_APIS))
def test_multi_api_bad_api(self):
self.assertRaises(ValueError,
batch_predict_text,