diff --git a/setup.py b/setup.py index b0d3a21..331195d 100644 --- a/setup.py +++ b/setup.py @@ -34,6 +34,6 @@ setup( "requests >= 1.2.3", "six >= 1.3.0", "numpy >= 1.8.1", - "scikit-image >= 0.10.1" + "scikit-image >= 0.10.1", ], ) diff --git a/tests/local/test_local.py b/tests/local/test_local.py index a4365c1..23dad4e 100644 --- a/tests/local/test_local.py +++ b/tests/local/test_local.py @@ -2,7 +2,7 @@ import unittest import numpy as np -from indicoio.local import political, sentiment, fer, facial_features, language +from indicoio.local import political, sentiment, fer, facial_features, language, image_features class FullAPIRun(unittest.TestCase): @@ -45,6 +45,13 @@ class FullAPIRun(unittest.TestCase): self.assertTrue(isinstance(response, list)) self.assertEqual(len(response), 48) + def test_good_image_features(self): + test_image = np.linspace(0,1,64*64*3).reshape(64,64,3).tolist() + response = image_features(test_image) + + self.assertTrue(isinstance(response, list)) + self.assertEqual(len(response), 2048) + def test_language(self): language_set = set([ 'English', diff --git a/tests/remote/test_remote.py b/tests/remote/test_remote.py index 64d3b45..c4fe600 100644 --- a/tests/remote/test_remote.py +++ b/tests/remote/test_remote.py @@ -2,7 +2,7 @@ import unittest import numpy as np -from indicoio import political, sentiment, fer, facial_features, language +from indicoio import political, sentiment, fer, facial_features, language, image_features class FullAPIRun(unittest.TestCase): @@ -44,6 +44,13 @@ class FullAPIRun(unittest.TestCase): self.assertTrue(isinstance(response, list)) self.assertEqual(len(response), 48) + + def test_good_image_features(self): + test_image = np.linspace(0,1,64*64*3).reshape(64,64,3).tolist() + response = image_features(test_image) + + self.assertTrue(isinstance(response, list)) + self.assertEqual(len(response), 2048) def test_language(self): language_set = set([