diff --git a/tests/local/test_local.py b/tests/local/test_local.py index 241d7da..a4365c1 100644 --- a/tests/local/test_local.py +++ b/tests/local/test_local.py @@ -16,11 +16,11 @@ class FullAPIRun(unittest.TestCase): self.assertEqual(political_set, set(response.keys())) def test_posneg(self): - posneg_set = set(['Sentiment']) test_string = "Worst 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']) diff --git a/tests/remote/test_remote.py b/tests/remote/test_remote.py index d5169cb..64d3b45 100644 --- a/tests/remote/test_remote.py +++ b/tests/remote/test_remote.py @@ -16,11 +16,11 @@ class FullAPIRun(unittest.TestCase): self.assertEqual(political_set, set(response.keys())) def test_posneg(self): - posneg_set = set(['Sentiment']) test_string = "Worst 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'])