def test_get_contact_msgs(self):
"""Test getting contact's msgs"""
contact_msgs = sentiment_analysis.get_contacts_msgs(7)
self.assertEqual(contact_msgs[14][0].message_text,
u'This is an english test message')
def contact_messages_data(user_id):
"""Return contact's messages data."""
if 'user_id' not in session:
return redirect("/")
elif session['user_id'] != user_id:
return redirect("/users/%s" % session['user_id'])
contact_msgs = sentiment_analysis.get_contacts_msgs(user_id)
sentiment_list = sentiment_analysis.get_contacts(contact_msgs)
arranged_list = sentiment_analysis.break_list(sentiment_list)
print arranged_list
data_dict = {
"labels": arranged_list[0],
"datasets": [
{
"label": "Positive messages",
"fillColor": "#F7464A",
"strokeColor": "rgba(220,220,220,0.8)",
"highlightFill": "rgba(151,187,205,0.75)",
"highlightStroke": "rgba(220,220,220,1)",
"data": arranged_list[1]
},
{
"label": "Negative messages",
"fillColor": "#46BFBD",
"strokeColor": "rgba(151,187,205,0.8)",
"highlightFill": "rgba(151,187,205,0.75)",
"highlightStroke": "#46BFBD",
"data": arranged_list[2]
},
{
"label": "Neutral messages",
"fillColor": "#FDB45C",
"strokeColor": "rgba(151,187,205,0.8)",
"highlightFill": "rgba(151,187,205,0.75)",
"highlightStroke": "#C0C0C0",
"data": arranged_list[3]
}
]
};
return jsonify(data_dict)
def test_get_contact_msgs(self):
"""Test getting contact's msgs"""
contact_msgs = sentiment_analysis.get_contacts_msgs(7)
self.assertEqual(contact_msgs[14][0].message_text, u"This is an english test message")
