def test_get_week_dates_for(self): school_year = SchoolYearFactory() week = Week(datetime.date(2020, 1, 20)) assert school_year.get_week_dates_for(week) == [ datetime.date(2020, 1, 20), datetime.date(2020, 1, 21), datetime.date(2020, 1, 22), datetime.date(2020, 1, 23), datetime.date(2020, 1, 24), ]
def test_get_week_dates_for(self): school_year = SchoolYearFactory() monday = datetime.date(2020, 1, 20) sunday = datetime.date(2020, 1, 26) week = (monday, sunday) self.assertEqual( school_year.get_week_dates_for(week), [ datetime.date(2020, 1, 20), datetime.date(2020, 1, 21), datetime.date(2020, 1, 22), datetime.date(2020, 1, 23), datetime.date(2020, 1, 24), ], )