one models for testing

This commit is contained in:
Waylon Walker 2023-04-28 21:13:46 -05:00
parent ab0cd3e664
commit 0d5771fd57
No known key found for this signature in database
GPG key ID: 66E2BF2B4190EFE4
3 changed files with 2 additions and 96 deletions

View file

@ -1,94 +0,0 @@
"""Models defines a set of classes for representing people and their hair.
Classes:
* `Alpha`: A class for representing an alpha value.
* `Color`: A class for representing a color.
* `Hair`: A class for representing hair.
* `Person`: A class for representing a person.
"""
from typing import Optional
from pydantic import BaseModel, Field
class Alpha(BaseModel):
"""A class for representing an alpha value."""
a: int = Field(
...,
description="The alpha value.",
)
class Color(BaseModel):
"""A class for representing a color."""
r: int = Field(
...,
description="The red component of the color.",
)
g: int = Field(
...,
description="The green component of the color.",
)
b: int = Field(
...,
description="The blue component of the color.",
)
alpha: Alpha = Field(
...,
description="The alpha value of the color.",
)
class Hair(BaseModel):
"""A class for representing hair."""
color: Color = Field(
...,
description="The color of the hair.",
)
length: int = Field(
...,
description="The length of the hair.",
)
class Person(BaseModel):
"""A class for representing a person."""
name: str = Field(
...,
description="The name of the person.",
)
other_name: Optional[str] = Field(
None,
description="An optional other name for the person.",
)
age: int = Field(
...,
description="The age of the person.",
)
email: Optional[str] = Field(
None,
description="An optional email address for the person.",
)
pet: str = Field(
"dog",
description="The person's pet.",
)
address: str = Field(
"123 Main St",
description="Where the person calls home.",
)
hair: Hair = Field(
...,
description="The person's hair.",
)

View file

@ -6,8 +6,8 @@ SPDX-License-Identifier: MIT
"""
from pyannotate_runtime import collect_types
from examples.models import Person
from pydantic_typer import expand_pydantic_args
from tests.models import Person
@expand_pydantic_args()

View file

@ -6,8 +6,8 @@ SPDX-License-Identifier: MIT
"""
import typer
from examples.models import Person
from pydantic_typer import expand_pydantic_args
from tests.models import Person
app = typer.Typer(
name="pydantic_typer",