factores.py

import factory
from faker import Faker

from core.tests.utils import create_datetime_range
from .. import models

faker = Faker()


class PricingBlockFactory(factory.django.DjangoModelFactory):
    name = factory.LazyAttribute(lambda a: faker.first_name())
    date_range = factory.LazyAttribute(lambda a: create_datetime_range())
    amount = 25.0
    prepaid_amount = 23.0
    set_to_repeat = False

    class Meta:
        model = models.PriceBlock