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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
"""Module exists only to test generic boundaries.
Filename should not start with "test_".
"""
import dataclasses
from typing import TypedDict
import pydantic.dataclasses
from pydantic import BaseModel
from polyfactory.factories import DataclassFactory, TypedDictFactory
from polyfactory.factories.pydantic_factory import ModelFactory
class PydanticClass(BaseModel):
field: str
class PydanticClassFactory(ModelFactory[PydanticClass]):
__model__ = PydanticClass
@pydantic.dataclasses.dataclass
class PydanticDataClass:
field: str
class PydanticDataClassFactory(DataclassFactory[PydanticDataClass]):
__model__ = PydanticDataClass
@dataclasses.dataclass()
class PythonDataClass:
field: str
class PythonDataClassFactory(DataclassFactory[PythonDataClass]):
__model__ = PythonDataClass
class TypedDictClass(TypedDict):
field: str
class TypedDictClassFactory(TypedDictFactory[TypedDictClass]):
__model__ = TypedDictClass
|
