File: test_faker_customization.py

package info (click to toggle)
python-polyfactory 2.22.2-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 1,892 kB
  • sloc: python: 11,338; makefile: 103; sh: 37
file content (15 lines) | stat: -rw-r--r-- 394 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
 
from faker import Faker

from polyfactory.factories.pydantic_factory import ModelFactory
from tests.models import Pet


def test_allows_user_to_define_faker_instance() -> None:
    my_faker = Faker()
    setattr(my_faker, "__test__attr__", None)

    class MyFactory(ModelFactory):
        __model__ = Pet
        __faker__ = my_faker

    assert hasattr(MyFactory.__faker__, "__test__attr__")