1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
from typing import Any, Dict, Generic, Tuple, TypeVar
import pytest
from pydantic import BaseModel
from litestar.plugins.pydantic.utils import pydantic_get_type_hints_with_generics_resolved
T = TypeVar("T")
class GenericPydanticModel(BaseModel, Generic[T]):
foo: T
@pytest.mark.parametrize(
("annotation", "expected_type_hints"),
(
(GenericPydanticModel, {"foo": T}),
(GenericPydanticModel[int], {"foo": int}),
),
)
def test_get_pydantic_type_hints_with_generics_resolved(annotation: Any, expected_type_hints: Dict[str, Any]) -> None:
type_hints = pydantic_get_type_hints_with_generics_resolved(annotation)
# In Python 3.12 and Pydantic V1, `__slots__` is returned in `get_type_hints`.
slots_type = type_hints.pop("__slots__", None)
if slots_type is not None:
assert slots_type == Tuple[str, ...]
assert type_hints == expected_type_hints
|
