Image features test

This commit is contained in:
Madison May
2014-10-16 13:05:45 -04:00
parent 29df762b47
commit 42f19db69e
3 changed files with 17 additions and 3 deletions
+1 -1
View File
@@ -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",
],
)
+8 -1
View File
@@ -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',
+8 -1
View File
@@ -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([