wassname/IndicoIo-python
1
0
Fork 0
mirror of https://github.com/wassname/IndicoIo-python.git synced 2026-08-03 12:30:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated format for batch

Browse Source
This commit is contained in:
Madison May
2015-06-05 04:06:14 -04:00
parent adf8295f82
commit 5623f6de90
2 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
indicoio/utils/multi.py
-4
View File
@@ -28,10 +28,6 @@ def multi(data, type, apis, available, batch=False, **kwargs):
# Convert client api names to server names before sending request
apis = map(CLIENT_SERVER_MAP.get, apis)
result = api_handler(data, apis=apis, batch=batch, **kwargs)
if batch:
return [handle_response(each) for each in result]
return handle_response(result)
def handle_response(result):
tests/test_remote.py
+6 -6
View File
@@ -126,22 +126,22 @@ class BatchAPIRun(unittest.TestCase):
test_data = [generate_array((48,48))]
response = batch_predict_image(test_data, apis=config.IMAGE_APIS, api_key=self.api_key)
self.assertTrue(isinstance(response, list))
self.assertTrue(set(response[0].keys()) == set(config.IMAGE_APIS))
self.assertTrue(isinstance(response, dict))
self.assertTrue(set(response.keys()) == set(config.IMAGE_APIS))
def test_batch_multi_api_text(self):
test_data = ['clearly an english sentence']
response = batch_predict_text(test_data, apis=config.TEXT_APIS, api_key=self.api_key)
self.assertTrue(isinstance(response, list))
self.assertTrue(set(response[0].keys()) == set(config.TEXT_APIS))
self.assertTrue(isinstance(response, dict))
self.assertTrue(set(response.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))
self.assertTrue(isinstance(response, dict))
self.assertTrue(set(response.keys()) == set(config.TEXT_APIS))
def test_multi_api_bad_api(self):
self.assertRaises(ValueError,