ADD: ImageRecognition

tests/test_imagerecognition.py

@@ -0,0 +1,29 @@
+#!/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()

tests/test_remote.py

@@ -209,13 +209,6 @@ class FullAPIRun(unittest.TestCase):
     def setUp(self):
         self.api_key = config.api_key
 
-    def load_image(self, relpath, as_grey=False):
-        im = Image.open(os.path.normpath(os.path.join(DIR, relpath))).convert('L');
-        pixels = list(im.getdata())
-        width, height = im.size
-        pixels = [pixels[i * width:(i + 1) * width] for i in xrange(height)]
-        return pixels
-
     def check_range(self, _list, minimum=0.9, maximum=0.1, span=0.5):
         vector = list(flatten(_list))
         _max = max(vector)
@@ -253,7 +246,7 @@ class FullAPIRun(unittest.TestCase):
             assert v >= .1
 
     def test_keywords_language_detect(self):
-        text = "La semaine suivante, il remporte sa premiere victoire, dans la descente de Val Gardena en Italie, près de cinq ans après la dernière victoire en Coupe du monde d'un Français dans cette discipline, avec le succès de Nicolas Burtin à Kvitfjell"
+        text = "il a remporté sa première victoire dans la descente de Val Gardena en Italie"
         words = set(text.lower().split())
 
         results = keywords(text, language = 'detect')
@@ -269,7 +262,7 @@ class FullAPIRun(unittest.TestCase):
             assert v >= .1
 
     def test_keywords_language(self):
-        text = "La semaine suivante, il remporte sa premiere victoire, dans la descente de Val Gardena en Italie, près de cinq ans après la dernière victoire en Coupe du monde d'un Français dans cette discipline, avec le succès de Nicolas Burtin à Kvitfjell"
+        text = "il a remporté sa première victoire dans la descente de Val Gardena en Italie"
         words = set(text.lower().split())
 
         results = keywords(text, language = 'French')