def test_weekday_start_stop_time(self):
"""
Test grouping arrival and departure times by weekday.
"""
data = utils.get_data()
weekdays = utils.weekday_start_stop_time(data[10])
self.assertIsInstance(weekdays, dict)
self.assertEqual(len(weekdays), 7)
self.assertEqual(len(weekdays[0]), 2)
self.assertItemsEqual(weekdays.keys(), [i for i in range(7)])
self.assertIn([[34745], [64792]], weekdays.values())
def presence_start_end_view(user_id):
"""
Returns mean arrival and departure time for each weekday.
"""
data = get_data()
if user_id not in data:
log.debug('User %s not found!', user_id)
return []
weekdays = weekday_start_stop_time(data[user_id])
result = [(calendar.day_abbr[weekday], mean(times[0]), mean(times[1]))
for weekday, times in weekdays.items()]
return result
