|
from collections import namedtuple |
|
from datetime import datetime, timedelta |
|
from enum import Enum |
|
from pathlib import Path |
|
from typing import ClassVar, Dict, List, Literal, Optional, Set, Tuple, Union |
|
from uuid import UUID |
|
|
|
import pytest |
|
from gradio_client.utils import json_schema_to_python_type |
|
from pydantic import Field, confloat, conint, conlist |
|
from pydantic.networks import AnyUrl, EmailStr, IPvAnyAddress |
|
|
|
from gradio.data_classes import GradioModel, GradioRootModel |
|
|
|
|
|
class StringModel(GradioModel): |
|
data: str |
|
answer: ClassVar = "Dict(data: str)" |
|
|
|
|
|
class IntegerRootModel(GradioRootModel): |
|
root: int |
|
|
|
answer: ClassVar = "int" |
|
|
|
|
|
class FloatModel(GradioModel): |
|
data: float |
|
|
|
answer: ClassVar = "Dict(data: float)" |
|
|
|
|
|
class ListModel(GradioModel): |
|
items: List[int] |
|
|
|
answer: ClassVar = "Dict(items: List[int])" |
|
|
|
|
|
class DictModel(GradioModel): |
|
data_dict: Dict[str, int] |
|
|
|
answer: ClassVar = "Dict(data_dict: Dict(str, int))" |
|
|
|
|
|
class DictModel2(GradioModel): |
|
data_dict: Dict[str, List[float]] |
|
|
|
answer: ClassVar = "Dict(data_dict: Dict(str, List[float]))" |
|
|
|
|
|
class OptionalModel(GradioModel): |
|
optional_data: Optional[int] |
|
|
|
answer: ClassVar = "Dict(optional_data: int | None)" |
|
|
|
|
|
class ColorEnum(Enum): |
|
RED = "red" |
|
GREEN = "green" |
|
BLUE = "blue" |
|
|
|
|
|
class EnumRootModel(GradioModel): |
|
color: ColorEnum |
|
|
|
answer: ClassVar = "Dict(color: Literal['red', 'green', 'blue'])" |
|
|
|
|
|
class EmailModel(GradioModel): |
|
email: EmailStr |
|
|
|
answer: ClassVar = "Dict(email: str)" |
|
|
|
|
|
class RootWithNestedModel(GradioModel): |
|
nested_int: IntegerRootModel |
|
nested_enum: EnumRootModel |
|
nested_dict: DictModel2 |
|
|
|
answer: ClassVar = "Dict(nested_int: int, nested_enum: Dict(color: Literal['red', 'green', 'blue']), nested_dict: Dict(data_dict: Dict(str, List[float])))" |
|
|
|
|
|
class LessNestedModel(GradioModel): |
|
nested_int: int |
|
nested_enum: ColorEnum |
|
nested_dict: Dict[str, List[Union[int, float]]] |
|
|
|
answer: ClassVar = "Dict(nested_int: int, nested_enum: Literal['red', 'green', 'blue'], nested_dict: Dict(str, List[int | float]))" |
|
|
|
|
|
class StatusModel(GradioModel): |
|
status: Literal["active", "inactive"] |
|
|
|
answer: ClassVar = "Dict(status: Literal['active', 'inactive'])" |
|
|
|
|
|
class PointModel(GradioRootModel): |
|
root: Tuple[float, float] |
|
|
|
answer: ClassVar = "Tuple[float, float]" |
|
|
|
|
|
class UuidModel(GradioModel): |
|
uuid: UUID |
|
|
|
answer: ClassVar = "Dict(uuid: str)" |
|
|
|
|
|
class UrlModel(GradioModel): |
|
url: AnyUrl |
|
|
|
answer: ClassVar = "Dict(url: str)" |
|
|
|
|
|
class CustomFieldModel(GradioModel): |
|
name: str = Field(..., title="Name of the item", max_length=50) |
|
price: float = Field(..., title="Price of the item", gt=0) |
|
|
|
answer: ClassVar = "Dict(name: str, price: float)" |
|
|
|
|
|
class DurationModel(GradioModel): |
|
duration: timedelta |
|
|
|
answer: ClassVar = "Dict(duration: str)" |
|
|
|
|
|
class IPv4Model(GradioModel): |
|
ipv4_address: IPvAnyAddress |
|
|
|
answer: ClassVar = "Dict(ipv4_address: str)" |
|
|
|
|
|
class DateTimeModel(GradioModel): |
|
created_at: datetime |
|
updated_at: datetime |
|
|
|
answer: ClassVar = "Dict(created_at: str, updated_at: str)" |
|
|
|
|
|
class SetModel(GradioModel): |
|
unique_numbers: Set[int] |
|
|
|
answer: ClassVar = "Dict(unique_numbers: List[int])" |
|
|
|
|
|
class ItemModel(GradioModel): |
|
name: str |
|
price: float |
|
|
|
|
|
class OrderModel(GradioModel): |
|
items: List[ItemModel] |
|
|
|
answer: ClassVar = "Dict(items: List[Dict(name: str, price: float)])" |
|
|
|
|
|
class TemperatureUnitEnum(Enum): |
|
CELSIUS = "Celsius" |
|
FAHRENHEIT = "Fahrenheit" |
|
KELVIN = "Kelvin" |
|
|
|
|
|
class TemperatureConversionModel(GradioModel): |
|
temperature: confloat(ge=-273.15, le=1.416808) |
|
from_unit: TemperatureUnitEnum |
|
to_unit: TemperatureUnitEnum = Field(..., title="Target temperature unit") |
|
|
|
answer: ClassVar = "Dict(temperature: float, from_unit: Literal['Celsius', 'Fahrenheit', 'Kelvin'], to_unit: All[Literal['Celsius', 'Fahrenheit', 'Kelvin']])" |
|
|
|
|
|
class CartItemModel(GradioModel): |
|
product_name: str = Field(..., title="Name of the product", max_length=50) |
|
quantity: int = Field(..., title="Quantity of the product", ge=1) |
|
price_per_unit: float = Field(..., title="Price per unit", gt=0) |
|
|
|
|
|
class ShoppingCartModel(GradioModel): |
|
items: List[CartItemModel] |
|
|
|
answer: ClassVar = "Dict(items: List[Dict(product_name: str, quantity: int, price_per_unit: float)])" |
|
|
|
|
|
class CoordinateModel(GradioModel): |
|
latitude: float |
|
longitude: float |
|
|
|
|
|
class PathModel(GradioModel): |
|
coordinates: conlist(CoordinateModel, min_length=2, max_length=2) |
|
|
|
answer: ClassVar = ( |
|
"Dict(coordinates: List[Dict(latitude: float, longitude: float)])" |
|
) |
|
|
|
|
|
class CreditCardModel(GradioModel): |
|
card_number: conint(ge=1, le=9999999999999999) |
|
|
|
answer: ClassVar = "Dict(card_number: int)" |
|
|
|
|
|
class TupleListModel(GradioModel): |
|
data: List[Tuple[int, str]] |
|
|
|
answer: ClassVar = "Dict(data: List[Tuple[int, str]]" |
|
|
|
|
|
class PathListModel(GradioModel): |
|
file_paths: List[Path] |
|
|
|
answer: ClassVar = "Dict(file_paths: List[str])" |
|
|
|
|
|
class PostModel(GradioModel): |
|
author: str |
|
content: str |
|
tags: List[str] |
|
likes: int = 0 |
|
|
|
answer: ClassVar = "Dict(author: str, content: str, tags: List[str], likes: int)" |
|
|
|
|
|
Person = namedtuple("Person", ["name", "age"]) |
|
|
|
|
|
class NamedTupleDictionaryModel(GradioModel): |
|
people: Dict[str, Person] |
|
|
|
answer: ClassVar = "Dict(people: Dict(str, Tuple[Any, Any]))" |
|
|
|
|
|
MODELS = [ |
|
StringModel, |
|
IntegerRootModel, |
|
FloatModel, |
|
ListModel, |
|
DictModel, |
|
DictModel2, |
|
OptionalModel, |
|
EnumRootModel, |
|
EmailModel, |
|
RootWithNestedModel, |
|
LessNestedModel, |
|
StatusModel, |
|
PointModel, |
|
UuidModel, |
|
UrlModel, |
|
CustomFieldModel, |
|
DurationModel, |
|
IPv4Model, |
|
DateTimeModel, |
|
SetModel, |
|
OrderModel, |
|
TemperatureConversionModel, |
|
ShoppingCartModel, |
|
PathModel, |
|
CreditCardModel, |
|
PathListModel, |
|
NamedTupleDictionaryModel, |
|
] |
|
|
|
|
|
@pytest.mark.parametrize("model", MODELS) |
|
def test_api_info_for_model(model): |
|
assert json_schema_to_python_type(model.model_json_schema()) == model.answer |
|
|
