from datetime import date
from .models import Gymnast
from datetime import datetime
import pytest


def test_gymnast_tostring():
    g = Gymnast(last_name="Pauchou", first_name="Fred")
    assert str(g) == "Pauchou, Fred"


def test_gymnaste_get_age():
    g = Gymnast(
        last_name="Pauchou",
        first_name="Fred",
        birthdate=datetime.strptime("03/07/1985", "%d/%m/%Y"),
    )
    assert g.age == 35


def test_gymnaste_get_next_age():
    g = Gymnast(
        last_name="Pauchou",
        first_name="Fred",
        birthdate=datetime.strptime("03/07/1985", "%d/%m/%Y"),
    )
    assert g.next_age == 36


def test_gymnaste_next_birthday():
    g = Gymnast(
        last_name="Pauchou",
        first_name="Fred",
        birthdate=datetime.strptime("03/07/1985", "%d/%m/%Y"),
    )
    assert g.next_birthday == datetime.strptime("03/07/2021", "%d/%m/%Y")