def test_total_steps(self): from models import UserStats import datetime today = datetime.datetime(2017, 10, 20) monday = today - datetime.timedelta(days=today.weekday()) tuesday = monday + datetime.timedelta(days=1) wednesday = monday + datetime.timedelta(days=2) thursday = monday + datetime.timedelta(days=3) friday = monday + datetime.timedelta(days=4) saturday = monday + datetime.timedelta(days=5) sunday = monday + datetime.timedelta(days=6) s = UserStats() s.add(25, monday) s.add(25, monday) s.add(7, tuesday) s.add(89, wednesday) s.add(2, thursday) s.add(3, friday) s.add(5, saturday) s.add(9, sunday) self.assertEqual(s.total_steps(), 25 + 25 + 7 + 89 + 2 + 3 + 5 + 9)