def test_conversation_scores(self):
conversation = Conversation(self.client_context)
pos, sub = conversation.calculate_sentiment_score()
self.assertEqual(0.0, pos)
self.assertEqual(0.5, sub)
question1 = Question.create_from_text(self.client_context,
"Hello There")
conversation.record_dialog(question1)
pos, sub = conversation.calculate_sentiment_score()
self.assertEqual(0.0, pos)
self.assertEqual(0.0, sub)
question2 = Question.create_from_text(self.client_context,
"I really like you")
conversation.record_dialog(question2)
pos, sub = conversation.calculate_sentiment_score()
self.assertEqual(0.1, pos)
self.assertEqual(0.1, sub)
question3 = Question.create_from_text(self.client_context,
"Do you like me")
conversation.record_dialog(question3)
pos, sub = conversation.calculate_sentiment_score()
self.assertEqual(0.06666666666666667, pos)
self.assertEqual(0.06666666666666667, sub)
def test_conversation_sentiment(self):
conversation = Conversation(self.client_context)
question1 = Question.create_from_text(self.client_context, "I am so unhappy")
conversation.record_dialog(question1)
question2 = Question.create_from_text(self.client_context, "I do not like the colour red")
conversation.record_dialog(question2)
question3 = Question.create_from_text(self.client_context, "Custard makes me feel sick")
conversation.record_dialog(question3)
conversation.recalculate_sentiment_score(self.client_context)
positivity, subjectivity = conversation.calculate_sentiment_score()
self.assertEqual(-0.4380952380952381, positivity)
self.assertEqual(0.5857142857142857, subjectivity)
