ADD: default apis for text/image with tests

This commit is contained in:
Chris Lee committed 2015-06-03 15:37:01 -04:00
1 parent 49a94d2986
commit adf8295f82
2 files changed
+9 -2

No files matched your search

+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,