def test_group_top5_weeks(self): data = utils.get_data() top5_data = utils.group_top5_weeks(data[10]) self.assertEqual(top5_data, {'2013-09-09 - 2013-09-15': 78217}) top5_data = utils.group_top5_weeks(data[11]) self.assertEqual(top5_data, { '2013-09-02 - 2013-09-08': 22999, '2013-09-09 - 2013-09-15': 95403 })
def top5_weeks_view(user_id): """ Returns top 5 weeks in work. """ data = get_data() if user_id not in data: log.debug('User %s not found!', user_id) abort(404) weekdays = group_top5_weeks(data[user_id]) result = sorted(weekdays.items(), key=lambda x: x[1], reverse=True) return result[0:5]