def test_get_messages(self):
"""Test if you get messages"""
messages = sentiment_analysis.get_messages(3)
self.assertEqual(messages[0].message_text, "This is a russian test message")
self.assertEqual(messages[0].user_id, 3)
self.assertEqual(messages[0].original_lang_id, 5)
def test_get_messages(self):
"""Test if you get messages"""
messages = sentiment_analysis.get_messages(3)
self.assertEqual(messages[0].message_text,
'This is a russian test message')
self.assertEqual(messages[0].user_id, 3)
self.assertEqual(messages[0].original_lang_id, 5)
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)
