File: test_example_10.py

package info (click to toggle)
python-polyfactory 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 1,976 kB
  • sloc: python: 11,113; makefile: 102; sh: 34
file content (17 lines) | stat: -rw-r--r-- 418 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 
from pydantic import BaseModel, Field

from polyfactory.factories.pydantic_factory import ModelFactory


class Payment(BaseModel):
    amount: int = Field(0)
    currency: str = Field(examples=["USD", "EUR", "INR"])


class PaymentFactory(ModelFactory[Payment]):
    __use_examples__ = True


def test_use_examples() -> None:
    instance = PaymentFactory.build()
    assert instance.currency in ["USD", "EUR", "INR"]