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)
def test_add(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) s = UserStats() self.assertEqual(s.monday_steps, 0) s.add(25, monday) self.assertEqual(s.monday_steps, 25) s.add(25, monday) self.assertEqual(s.monday_steps, 50) s.add(7, tuesday) self.assertEqual(s.tuesday_steps, 7) s.add(89, wednesday) self.assertEqual(s.wednesday_steps, 89) s.add(2, thursday) self.assertEqual(s.thursday_steps, 2) s.add(3, friday) self.assertEqual(s.friday_steps, 3)