def test_msg_sentiment_analysis(self): """Test sentiment analysis of a message""" pos_messages = Message.query.filter_by(message_id=101).all() analyzed_messages = sentiment_analysis.analyze_messages(pos_messages) self.assertTrue(analyzed_messages[0].sentiment.polarity > 0.3) neg_messages = Message.query.filter_by(message_id=104).all() analyzed_messages = sentiment_analysis.analyze_messages(neg_messages) self.assertTrue(analyzed_messages[0].sentiment.polarity < -0.3)
def messages_types_data(user_id): """Return data about messages emotions.""" if 'user_id' not in session: return redirect("/") elif session['user_id'] != user_id: return redirect("/users/%s" % session['user_id']) messages = sentiment_analysis.get_messages(user_id) sentiment_list = sentiment_analysis.analyze_messages(messages) arranged_list = sentiment_analysis.categorize_messages(sentiment_list) data_list_of_dicts = { 'arranged_list': [ { "value": arranged_list[0], "color": "#F7464A", "highlight": "#FF5A5E", "label": "Positive messages" }, { "value": arranged_list[1], "color": "#46BFBD", "highlight": "#5AD3D1", "label": "Negative messages" }, { "value": arranged_list[2], "color": "#FDB45C", "highlight": "#FFC870", "label": "Neutral messages" } ] } return jsonify(data_list_of_dicts)