import datetime import pytest from django.contrib.auth.models import User from baby.app.common.calendar import MonthCalendar from baby.app.models.regle import Regle @pytest.mark.django_db class TestCalendar: def test_nominal_case(self): user = User.objects.create_user("user") ref = datetime.date(2023, 9, 2) times = [ -28 - 25 - 26, -26 - 25, -25, 0 ] for i, x in enumerate(times, start=1): Regle.objects.create(user=user, date=ref + datetime.timedelta(days=x)) cal = MonthCalendar(user, 11, 23) data = cal.get_calendar_dict() js = cal.json print(data)