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)