diff --git a/tests/test_remote.py b/tests/test_remote.py index ce4c707..2d8c9b7 100644 --- a/tests/test_remote.py +++ b/tests/test_remote.py @@ -134,6 +134,8 @@ class BatchAPIRun(unittest.TestCase): self.assertTrue(isinstance(response, dict)) self.assertTrue(set(response.keys()) == set(config.IMAGE_APIS)) + self.assertTrue("results" in response["fer"]) + def test_batch_multi_api_text(self): test_data = ['clearly an english sentence'] @@ -141,6 +143,7 @@ class BatchAPIRun(unittest.TestCase): self.assertTrue(isinstance(response, dict)) self.assertTrue(set(response.keys()) == set(config.TEXT_APIS)) + self.assertTrue("results" in response["sentiment"]) def test_default_multi_api_text(self): test_data = ['clearly an english sentence']