python - Django 固定装置具有所有可能的组合？

Question

我正在尝试测试统计函数，该函数会计算数据库中的对象类型。为此，我想为模型字段的每种可能组合至少创建一个实例。随机测试数据需要大量的测试对象来确保满足所有可能的组合。

这是我的一个模型的简短示例:

class Member(models.Model)
    is_active = models.BooleanField()
    name = models.CharField()
    balance = models.IntegerField()
    division = models.ForeignKey(Division, on_delete=models.CASCADE)

Class Division(models.Model)
    name = models.CharField()

这就是我现在使用 django_dynamic_fixture 所做的事情:

from django.test import TestCase
from django_dynamic_fixture import G
from members.models import Member

class StatisticsTestCase(TestCase):
    def setUp(self):
        for is_active in [True, False]:
            for balance in [-100, 0, 100]:
                for division in Division.objects.all()
                    G(Member, is_active=is_active, balance=balance, division=division)

但这看起来很糟糕并且很难阅读。是否有一种更简单、更容易阅读代码的方法来创建对象属性的所有可能组合？

我对不同的测试数据生成器持开放态度，只要它们能够与 Python 3 和 Django 很好地配合即可。

Answer 1

import itertools

combinations = list(itertools.product(*[[True, False], [-100, 0, 100]]))
for division in Division.objects.all():
    for is_active, balance in combinations:
        G(Member, is_active=is_active, balance=balance, division=division)

<小时/>

更新:

import itertools

combinations = list(itertools.product([True, False], [-100, 0, 100], Division.objects.all()))
for is_active, balance, division in combinations:
    G(Member, is_active=is_active, balance=balance, division=division)