mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-27 16:10:34 +08:00
Madison May
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
import unittest
|
|
|
|
import numpy as np
|
|
|
|
from indicoio.local import political, sentiment, fer, facial_features, language, image_features
|
|
|
|
|
|
class FullAPIRun(unittest.TestCase):
|
|
|
|
def test_political(self):
|
|
political_set = set(['Libertarian', 'Liberal', 'Conservative', 'Green'])
|
|
test_string = "Guns don't kill people, people kill people."
|
|
response = political(test_string)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
self.assertEqual(political_set, set(response.keys()))
|
|
|
|
test_string = "Save the whales"
|
|
response = political(test_string)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
assert response['Green'] > 0.5
|
|
|
|
def test_posneg(self):
|
|
test_string = "Worst song ever."
|
|
response = sentiment(test_string)
|
|
|
|
self.assertTrue(isinstance(response, float))
|
|
self.assertTrue(response < 0.5)
|
|
|
|
test_string = "Best song ever."
|
|
response = sentiment(test_string)
|
|
self.assertTrue(isinstance(response, float))
|
|
self.assertTrue(response > 0.5)
|
|
|
|
def test_good_fer(self):
|
|
fer_set = set(['Angry', 'Sad', 'Neutral', 'Surprise', 'Fear', 'Happy'])
|
|
test_face = np.random.rand(48,48).tolist()
|
|
response = fer(test_face)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
self.assertEqual(fer_set, set(response.keys()))
|
|
|
|
def test_bad_fer(self):
|
|
fer_set = set(['Angry', 'Sad', 'Neutral', 'Surprise', 'Fear', 'Happy'])
|
|
test_face = np.random.rand(56,56).tolist()
|
|
response = fer(test_face)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
self.assertEqual(fer_set, set(response.keys()))
|
|
|
|
def test_good_facial_features(self):
|
|
test_face = np.random.rand(48,48).tolist()
|
|
response = facial_features(test_face)
|
|
|
|
self.assertTrue(isinstance(response, list))
|
|
self.assertEqual(len(response), 48)
|
|
|
|
def test_good_image_features_greyscale(self):
|
|
test_image = np.random.rand(64, 64).tolist()
|
|
response = image_features(test_image)
|
|
|
|
self.assertTrue(isinstance(response, list))
|
|
self.assertEqual(len(response), 2048)
|
|
|
|
def test_good_image_features_rgb(self):
|
|
test_image = np.random.rand(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',
|
|
'Spanish',
|
|
'Tagalog',
|
|
'Esperanto',
|
|
'French',
|
|
'Chinese',
|
|
'French',
|
|
'Bulgarian',
|
|
'Latin',
|
|
'Slovak',
|
|
'Hebrew',
|
|
'Russian',
|
|
'German',
|
|
'Japanese',
|
|
'Korean',
|
|
'Portuguese',
|
|
'Italian',
|
|
'Polish',
|
|
'Turkish',
|
|
'Dutch',
|
|
'Arabic',
|
|
'Persian (Farsi)',
|
|
'Czech',
|
|
'Swedish',
|
|
'Indonesian',
|
|
'Vietnamese',
|
|
'Romanian',
|
|
'Greek',
|
|
'Danish',
|
|
'Hungarian',
|
|
'Thai',
|
|
'Finnish',
|
|
'Norwegian',
|
|
'Lithuanian'
|
|
])
|
|
language_dict = language('clearly an english sentence')
|
|
self.assertEqual(language_set, set(language_dict.keys()))
|
|
assert language_dict['English'] > 0.25
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|