Showing wrong order and specialized models in jsons

last_custom_root_examples.py

from pydantic import BaseModel


class Age(BaseModel):
    __root__: int


def divide_age(x: Age, y: int):  # 1
    '''implementation irrelevant'''


# mypy will generate type errors if we accidentally invoke divide_age
# in the wrong order
# divide_age(4, 5)  # 2
# divide_age(Age.parse_obj(4), Age.parse_obj(5))  # 2


class Person(BaseModel):
    age: Age
    name: str


def test_age_is_converted_to_primitive_when_being_jsoned():
    person = Person.parse_obj({'age': 37, 'name': 'John'})
    assert person.json() == '{"age": 37, "name": "John"}'  # 3