def test_return_adolescent_from_12_years_to_16_years():
birth_date = current_date - relativedelta(years=12)
assert calculate(birth_date,
current_date) == ChildClassification.ADOLESCENT
birth_date = current_date - relativedelta(years=16) + relativedelta(days=1)
assert calculate(birth_date,
current_date) == ChildClassification.ADOLESCENT
def test_return_neonate_up_to_30_days_old():
assert calculate(current_date, current_date) == ChildClassification.NEONATE
birth_date = current_date - timedelta(days=30)
assert calculate(birth_date, current_date)
def test_return_undefined_if_birthdate_in_future():
birth_date = current_date + relativedelta(days=1)
assert calculate(birth_date, current_date) == ChildClassification.UNDEFINED
def test_return_undefined_after_16_years():
birth_date = current_date - relativedelta(years=16)
assert calculate(birth_date, current_date) == ChildClassification.UNDEFINED
birth_date = current_date - relativedelta(years=80)
assert calculate(birth_date, current_date) == ChildClassification.UNDEFINED
def test_return_child_from_2_years_to_12_years():
birth_date = current_date - relativedelta(years=2)
assert calculate(birth_date, current_date) == ChildClassification.CHILD
birth_date = current_date - relativedelta(years=12) + relativedelta(days=1)
assert calculate(birth_date, current_date) == ChildClassification.CHILD
def test_return_infant_from_30_days_to_2_years():
birth_date = current_date - relativedelta(days=30)
assert calculate(birth_date, current_date) == ChildClassification.INFANT
birth_date = current_date - relativedelta(years=2) + relativedelta(days=1)
assert calculate(birth_date, current_date) == ChildClassification.INFANT
def get_child_classification(self):
return calculate(self.birthdate, date.today())