mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-27 16:10:34 +08:00
Slater-Victoroff
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
import unittest
|
|
|
|
import numpy as np
|
|
|
|
from IndicoIo import political, spam, posneg, fer, facial_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()))
|
|
|
|
def test_spam(self):
|
|
spam_set = set(['Spam', 'Ham'])
|
|
test_string = "Buy a new car!!"
|
|
response = spam(test_string)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
self.assertEqual(spam_set, set(response.keys()))
|
|
|
|
def test_posneg(self):
|
|
posneg_set = set(['Positive', 'Negative'])
|
|
test_string = "Worst song ever."
|
|
response = posneg(test_string)
|
|
|
|
self.assertTrue(isinstance(response, dict))
|
|
self.assertEqual(posneg_set, set(response.keys()))
|
|
|
|
def test_good_fer(self):
|
|
fer_set = set(['Angry', 'Sad', 'Neutral', 'Surprise', 'Fear', 'Happy'])
|
|
test_face = np.linspace(0,50,48*48).reshape(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.linspace(0,50,56*56).reshape(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.linspace(0,50,48*48).reshape(48,48).tolist()
|
|
response = facial_features(test_face)
|
|
|
|
self.assertTrue(isinstance(response, list))
|
|
self.assertEqual(len(response), 48)
|
|
|
|
def test_full_facial_features(self):
|
|
features_set = set(['feature_vector', 'warnings'])
|
|
test_face = np.linspace(0,50,56*56).reshape(56,56).tolist()
|
|
response = facial_features(test_face, True)
|
|
|
|
self.assertEqual(set(response.keys()), features_set)
|
|
self.assertEqual(response['warnings'], 'Using a 48x48 array will produce the best results')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|