def test_friends_with_incomplete_bdate_field(self): response = [ {'id': 1, 'first_name': '', 'last_name': '', 'bdate': '15.1', 'online': 0}, {'id': 2, 'first_name': '', 'last_name': '', 'bdate': '3.9', 'online': 0}, ] with patch('age.get_friends', return_value=response): predicted_age = age_predict(user_id=1) self.assertEqual(predicted_age, None)
def test_friends_with_bdate_field(self): today = dt.date.today() yesterday = dt.date(day=today.day-1, month=today.month, year=1996).strftime('%d.%m.%Y') tomorrow = dt.date(day=today.day+1, month=today.month + 1, year=1992).strftime('%d.%m.%Y') response = [ {'id': 1, 'first_name': '', 'last_name': '', 'bdate': yesterday, 'online': 0}, {'id': 2, 'first_name': '', 'last_name': '', 'bdate': tomorrow, 'online': 0}, {'id': 3, 'first_name': '', 'last_name': '', 'bdate': '11.11', 'online': 0}, {'id': 4, 'first_name': '', 'last_name': '', 'online': 0}, ] with patch('age.get_friends', return_value=response): predicted_age = age_predict(user_id=1) self.assertEqual(predicted_age, 23.5)
def test_empty_response(self): response = [] with patch('age.get_friends', return_value=response): predicted_age = age_predict(user_id=1) self.assertEqual(predicted_age, None)
from network import get_network, plot_graph from api import get_friends, get_wall from age import age_predict from api_models import normalize print('test age_predict') print(age_predict(125483792)) print('test graph') friends = get_friends(125483792, 'id')['response'] users_ids = [] names = [] k = 0 for friend in friends['items']: users_ids.append(friend['id']) names.append(friend['first_name'] + ' ' + friend['last_name']) k += 1 if k == 50: break edges = get_network(users_ids, as_edgelist=True) plot_graph(edges, names) ''' with open('1.txt', 'w') as f: post_list = get_wall(domain='pn6', count=5) f.writelines(normalize(post_list)) '''
from age import age_predict from api import messages_get_history, get_friends from messages import count_dates_from_messages, plotly_messages_freq from api_models import Message from network import get_network, plot_graph print(age_predict(155393609)) history = messages_get_history(155393609, count=300) mlist = [Message(**mes) for mes in history] dates = count_dates_from_messages(mlist) plotly_messages_freq(dates[0], dates[1]) friends = get_friends(155393609, fields='bdate')['response']['items'] ids = [] names = [] for friend in friends: ids.append(friend['id']) names.append(friend['first_name'] + ' ' + friend['last_name']) edges = get_network(ids) plot_graph(edges, names)