65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
import pendulum
|
|
from django.test import TestCase
|
|
from jarvis.people.models import Gymnast
|
|
from jarvis.followup.models import (
|
|
Chrono,
|
|
ChronoDetails,
|
|
Injury,
|
|
LearnedSkill,
|
|
Plan,
|
|
Point,
|
|
WellBeing,
|
|
GymnastHasRoutine,
|
|
NumberOfRoutineDone,
|
|
HeightWeight,
|
|
Note,
|
|
Intensity,
|
|
SeasonInformation,
|
|
CompetitivePointsStats,
|
|
)
|
|
|
|
from jarvis.followup.models import (
|
|
CHRONO_TYPE_CHOICE,
|
|
SCORE_TYPE_CHOICE,
|
|
)
|
|
|
|
|
|
class TestModels(TestCase):
|
|
def setUp(self):
|
|
"""Mise en place des variables pour les tests."""
|
|
gymnast = Gymnast.objects.create(
|
|
last_name="Pauchou", first_name="Fred", birthdate="1987-07-03", gender=0
|
|
)
|
|
Chrono.objects.create(
|
|
gymnast=gymnast, chrono_type=0, score_type=0, score=15, tof=13
|
|
)
|
|
Injury.objects.create(
|
|
gymnast=gymnast, location=0, injury_type=0, body_side=0, mechanism=0
|
|
)
|
|
|
|
def test_chrono_to_string(self):
|
|
gymnast = Gymnast.objects.get(last_name="Pauchou")
|
|
chrono = Chrono.objects.get(gymnast=gymnast)
|
|
today = pendulum.now().to_date_string()
|
|
print(today)
|
|
self.assertEqual(str(chrono), f"Fred Pauchou - 13.000 ({today} - 0)")
|
|
|
|
# def test_chrono_timeline_representation(self):
|
|
# gymnast = Gymnast.objects.get(last_name="Pauchou")
|
|
# chrono = Chrono.objects.get(gymnast=gymnast)
|
|
# today = pendulum.now().date()
|
|
# self.assertEqual(
|
|
# chrono.timeline_representation,
|
|
# f"<li>{today.to_date_string()} - New personel best {CHRONO_TYPE_CHOICE[chrono.chrono_type][1]}: 15.000' (13.000')</li>", # pylint: disable=line-too-long
|
|
# )
|
|
|
|
def test_compute_tof(self):
|
|
res = Chrono.compute_tof(15)
|
|
self.assertEqual(res, 13)
|
|
|
|
def test_injury_to_string(self):
|
|
gymnast = Gymnast.objects.get(last_name="Pauchou")
|
|
injury = Injury.objects.get(gymnast=gymnast)
|
|
today = pendulum.now().date()
|
|
self.assertEqual(str(injury), f"Fred Pauchou ({today})")
|