Exemplo n.º 1
0
 def test_ratings_sum(self):
     "sum ratings for some words"
     afinn = Afinn("AFINN-111.txt")
     sentence = "Rainy day but still in a good mood"
     util = Util()
     aggregate_sentiment = 0
     for word in util.get_tokens(sentence):
         if afinn.has_rating(word):
             aggregate_sentiment += afinn.get_rating(word)
     self.assertTrue(aggregate_sentiment == 2)