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
|
from dataclasses import dataclass
from datetime import date, datetime
from typing import List, Optional, Union
from uuid import UUID
from polyfactory.factories import DataclassFactory
from polyfactory.pytest_plugin import register_fixture
@dataclass
class Person:
id: UUID
name: str
hobbies: Optional[List[str]]
nicks: List[str]
age: Union[float, int]
birthday: Union[datetime, date]
@register_fixture(name="aliased_person_factory")
class PersonFactory(DataclassFactory[Person]): ...
def test_person_factory(aliased_person_factory: PersonFactory) -> None:
person_instance = aliased_person_factory.build()
assert isinstance(person_instance, Person)
|