awesome-fastapi-projects/app/source_graph/factories.py

"""Factories for creating test data."""
from polyfactory.factories.pydantic_factory import ModelFactory
from polyfactory.pytest_plugin import register_fixture

from app.source_graph.models import SourceGraphRepoData


@register_fixture
class SourceGraphRepoDataFactory(ModelFactory[SourceGraphRepoData]):
    """Factory for creating SourceGraphRepoData."""

    __model__ = SourceGraphRepoData