wassname/IndicoIo-python
1
0
Fork 0
mirror of https://github.com/wassname/IndicoIo-python.git synced 2026-07-15 01:10:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

ADD: content filtering

Browse Source
This commit is contained in:
jagreene
2015-07-10 02:43:28 -04:00
committed by Madison May
parent 024e4a0432
commit d1f2c37a39
5 changed files with 47 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_remote.py
+13 -2
View File
@@ -6,8 +6,8 @@ from requests import ConnectionError
from nose.plugins.skip import Skip, SkipTest
from indicoio import config
from indicoio import political, sentiment, fer, facial_features, language, image_features, text_tags
from indicoio import batch_political, batch_sentiment, batch_fer, batch_facial_features
from indicoio import political, sentiment, fer, facial_features, content_filtering, language, image_features, text_tags
from indicoio import batch_political, batch_sentiment, batch_fer, batch_content_filtering, batch_facial_features
from indicoio import batch_language, batch_image_features, batch_text_tags
from indicoio import keywords, batch_keywords
from indicoio import sentiment_hq, batch_sentiment_hq
@@ -64,6 +64,12 @@ class BatchAPIRun(unittest.TestCase):
self.assertTrue(isinstance(response, list))
self.assertTrue(isinstance(response[0], dict))
def test_batch_content_filtering(self):
test_data = [generate_array((48,48))]
response = batch_content_filtering(test_data, api_key=self.api_key)
self.assertTrue(isinstance(response, list))
self.assertTrue(isinstance(response[0], float))
def test_batch_fer_bad_b64(self):
test_data = ["$bad#FI jeaf9(#0"]
self.assertRaises(IndicoError, batch_fer, test_data, api_key=self.api_key)
@@ -324,6 +330,11 @@ class FullAPIRun(unittest.TestCase):
self.assertTrue(isinstance(response, dict))
self.assertEqual(fer_set, set(response.keys()))
def test_safe_content_filtering(self):
test_face = self.load_image("data/happy.png", as_grey=True)
response = content_filtering(test_face)
self.assertTrue(response < 0.5)
def test_good_facial_features(self):
test_face = generate_array((48,48))
response = facial_features(test_face)