wassname/IndicoIo-python
1
0
Fork 0
mirror of https://github.com/wassname/IndicoIo-python.git synced 2026-06-27 16:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

ADD: default apis for text/image with tests

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
indicoio/utils/multi.py
+2 -2
View File
@@ -46,7 +46,7 @@ def handle_response(result):
raise Exception("Sorry, %s API returned an unexpected response:\n%s" % (api, result[api]))
def predict_text(input_text, apis, cloud=None, batch=False, api_key=None, **kwargs):
def predict_text(input_text, apis=TEXT_APIS, cloud=None, batch=False, api_key=None, **kwargs):
"""
Given input text, returns the results of specified text apis. Possible apis
include: [ 'text_tags', 'political', 'sentiment', 'language' ]
@@ -80,7 +80,7 @@ def predict_text(input_text, apis, cloud=None, batch=False, api_key=None, **kwar
**kwargs)
def predict_image(image, apis, cloud=None, batch=False, api_key=None, **kwargs):
def predict_image(image, apis=IMAGE_APIS, cloud=None, batch=False, api_key=None, **kwargs):
"""
Given input image, returns the results of specified image apis. Possible apis
include: ['fer', 'facial_features', 'image_features']
tests/test_remote.py
+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,