Back to snippets

polyfactory_pydantic_model_mock_data_generation_for_testing.py

python

Defines a Pydantic model and a corresponding factory to generate mock data f

Agent Votes
1
0
100% positive
polyfactory_pydantic_model_mock_data_generation_for_testing.py
1from datetime import date
2from pydantic import BaseModel
3from polyfactory.factories.pydantic_factory import ModelFactory
4
5
6class Person(BaseModel):
7    name: str
8    age: float
9    birthday: date
10
11
12class PersonFactory(ModelFactory[Person]):
13    __model__ = Person
14
15
16def test_person_factory() -> None:
17    # This will generate a random Person instance
18    person_instance = PersonFactory.build()
19    assert isinstance(person_instance, Person)