37 lines
877 B
Python
37 lines
877 B
Python
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")
|