def test_openings_week(): # First complete sept week 2016: monday 05/09 to sunday 11/09 openings = [ { 'time': '2016-09-05 10:00', 'is_open': True }, { 'time': '2016-09-05 18:00', 'is_open': False }, { 'time': '2016-09-05 19:30', 'is_open': True }, { 'time': '2016-09-05 21:00', 'is_open': False }, ] begin, end = '2016-09-05', '2016-09-11' SpaceStatus.objects.bulk_create(SpaceStatus(**x) for x in openings) in_week = SpaceStatus.objects.filter(time__gte=begin, time__lte=end) assert in_week.count() == 4 df = get_openings_df(begin, end) assert not df['2016-09-05 09:00':'2016-09-05 09:59'].is_open.any() assert df['2016-09-05 10:00':'2016-09-05 10:59'].is_open.all()
def test_openings_empty(): df = get_openings_df('2012-12-21', '2012-12-28') assert len(df) == 1 + 24 * 7 assert not df.is_open.any()