mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-30 16:40:38 +08:00
Image features test
This commit is contained in:
@@ -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",
|
||||
],
|
||||
)
|
||||
|
||||
@@ -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',
|
||||
|
||||
@@ -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([
|
||||
|
||||
Reference in New Issue
Block a user