baby/tests/test_a.py

import datetime

import pytest
from django.contrib.auth.models import User

from baby.app.models.regle import Regle, Cycle


@pytest.mark.django_db
class Test:


    def test_cycle(self):
        today = datetime.date.today()
        user = User.objects.create_user("user")
        times = [
            -29 - 26 -26,
            -26 -26,
            -26
        ]
        for i, x in enumerate(times, start=1):
            Regle.objects.create(user=user, date=today+datetime.timedelta(days=x))

        cycle = Cycle.from_day(user, today)
        print(i)



    def test_cycle2(self):
        today = datetime.date.today()
        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))

        #cycle = Cycle.from_date_range(user, ref+datetime.timedelta(days=30), ref+datetime.timedelta(days=390))
        cycle = Cycle.from_month(user, 11, 2023)
        print(cycle)
        print(i)