mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-27 16:10:34 +08:00
Chris Lee
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
import unittest, os
|
|
|
|
from indicoio import config
|
|
from indicoio import image_recognition
|
|
|
|
DIR = os.path.dirname(os.path.realpath(__file__))
|
|
class TestVersioning(unittest.TestCase):
|
|
def setUp(self):
|
|
self.api_key = config.api_key
|
|
|
|
def test_single_image_recognition(self):
|
|
test_data = os.path.normpath(os.path.join(DIR, "data", "fear.png"))
|
|
response = image_recognition(test_data, api_key = self.api_key, top_n=3)
|
|
self.assertIsInstance(response, dict)
|
|
self.assertEqual(len(response), 3)
|
|
self.assertIsInstance(response.values()[0], float)
|
|
|
|
def test_batch_image_recognition(self):
|
|
test_data = os.path.normpath(os.path.join(DIR, "data", "fear.png"))
|
|
response = image_recognition([test_data, test_data], api_key = self.api_key, top_n=3)
|
|
self.assertIsInstance(response, list)
|
|
self.assertIsInstance(response[0], dict)
|
|
self.assertEqual(len(response[0]), 3)
|
|
self.assertIsInstance(response[0].values()[0], float)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|