def test_build_with_multi_count(self):
count = 2
models = build(ModelOne, count)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertNotEqual(models[0], models[1])
def test_build_with_multi_model_and_multi_count_and_single_initial(self):
count1 = 2
initial1 = {'decimal_field': Decimal('10.00')}
initial_list1 = list()
for _ in itertools.repeat(None, count1):
initial_list1.append(initial1)
count2 = 3
initial2 = {'char_field': 'test value'}
initial_list2 = list()
for _ in itertools.repeat(None, count2):
initial_list2.append(initial2)
args = ((ModelOne, initial_list1), (ModelTwo, initial_list2))
models = build(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
self.assertEqual(models[2].char_field, initial2['char_field'])
self.assertEqual(models[3].char_field, initial2['char_field'])
self.assertEqual(models[4].char_field, initial2['char_field'])
def test_build_with_multi_model_and_multi_count_and_single_initial(self):
count1 = 2
initial1 = {
'decimal_field': Decimal('10.00')
}
initial_list1 = list()
for _ in itertools.repeat(None, count1):
initial_list1.append(initial1)
count2 = 3
initial2 = {
'char_field': 'test value'
}
initial_list2 = list()
for _ in itertools.repeat(None, count2):
initial_list2.append(initial2)
args = ((ModelOne, initial_list1), (ModelTwo, initial_list2))
models = build(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
self.assertEqual(models[2].char_field, initial2['char_field'])
self.assertEqual(models[3].char_field, initial2['char_field'])
self.assertEqual(models[4].char_field, initial2['char_field'])
def test_build_with_multi_count(self):
count = 2
models = build(ModelOne, count)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertNotEqual(models[0], models[1])
def test_build_with_multi_model_and_multi_count_and_multi_initial(self):
initial1_1 = {
'decimal_field': Decimal('10.00')
}
initial1_2 = {
'decimal_field': Decimal('8.00')
}
initial2_1 = {
'char_field': 'test value 1'
}
initial2_2 = {
'char_field': 'test value 2'
}
args = ((ModelOne, [initial1_1, initial1_2]),
(ModelTwo, [initial2_1, initial2_2]))
models = build(*args)
self.assertEqual(len(models), 4)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1_1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1_2['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertEqual(models[2].char_field, initial2_1['char_field'])
self.assertEqual(models[3].char_field, initial2_2['char_field'])
def test_build_with_multi_model_and_single_count(self):
count1 = 1
count2 = 1
args = ((ModelOne, count1), (ModelTwo, count2))
models = build(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
def test_build_with_count_and_initial(self):
initial = {
'decimal_field': Decimal('10.00')
}
args = (ModelOne, initial)
model = build(*args)
self.assertIsInstance(model, ModelOne)
self.assertEqual(model.decimal_field, initial['decimal_field'])
def test_build_with_multi_model_and_single_count(self):
count1 = 1
count2 = 1
args = ((ModelOne, count1), (ModelTwo, count2))
models = build(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
def test_build_with_multi_model_and_single_count_and_single_initial(self):
initial1 = {'decimal_field': Decimal('10.00')}
initial2 = {'char_field': 'test value'}
args = ((ModelOne, initial1), (ModelTwo, initial2))
models = build(*args)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelTwo)
self.assertEqual(models[1].char_field, initial2['char_field'])
def test_build_with_multi_count_and_multi_initial(self):
count = 2
initial1 = {'decimal_field': Decimal('10.00')}
initial2 = {'decimal_field': Decimal('8.00')}
args = (ModelOne, [initial1, initial2])
models = build(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial2['decimal_field'])
def test_build_with_multi_model_and_multi_count(self):
count1 = 2
count2 = 3
args = ((ModelOne, count1), (ModelTwo, count2))
models = build(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
def test_build_with_multi_model_and_multi_count(self):
count1 = 2
count2 = 3
args = ((ModelOne, count1), (ModelTwo, count2))
models = build(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
def test_build_with_multi_count_and_single_initial(self):
count = 2
initial1 = {'decimal_field': Decimal('10.00')}
initial_list = list()
for _ in itertools.repeat(None, count):
initial_list.append(initial1)
args = (ModelOne, initial_list)
models = build(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
def test_build_with_multi_model_and_single_count_and_single_initial(self):
initial1 = {
'decimal_field': Decimal('10.00')
}
initial2 = {
'char_field': 'test value'
}
args = ((ModelOne, initial1), (ModelTwo, initial2))
models = build(*args)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelTwo)
self.assertEqual(models[1].char_field, initial2['char_field'])
def test_build_with_multi_count_and_single_initial(self):
count = 2
initial1 = {
'decimal_field': Decimal('10.00')
}
initial_list = list()
for _ in itertools.repeat(None, count):
initial_list.append(initial1)
args = (ModelOne, initial_list)
models = build(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
def test_build_with_multi_count_and_multi_initial(self):
count = 2
initial1 = {
'decimal_field': Decimal('10.00')
}
initial2 = {
'decimal_field': Decimal('8.00')
}
args = (ModelOne, [initial1, initial2])
models = build(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial2['decimal_field'])
def test_build_with_multi_model_and_multi_count_and_multi_initial(self):
initial1_1 = {'decimal_field': Decimal('10.00')}
initial1_2 = {'decimal_field': Decimal('8.00')}
initial2_1 = {'char_field': 'test value 1'}
initial2_2 = {'char_field': 'test value 2'}
args = ((ModelOne, [initial1_1,
initial1_2]), (ModelTwo, [initial2_1, initial2_2]))
models = build(*args)
self.assertEqual(len(models), 4)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1_1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1_2['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertEqual(models[2].char_field, initial2_1['char_field'])
self.assertEqual(models[3].char_field, initial2_2['char_field'])
def test_build_with_multi_model_trivial(self):
models = build((ModelOne, ), (ModelTwo, ))
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
def test_build_trivial(self):
model = build(ModelOne)
self.assertIsInstance(model, ModelOne)
def test_build_with_count_and_initial(self):
initial = {'decimal_field': Decimal('10.00')}
args = (ModelOne, initial)
model = build(*args)
self.assertIsInstance(model, ModelOne)
self.assertEqual(model.decimal_field, initial['decimal_field'])
def test_build_with_single_count(self):
count = 1
args = (ModelOne, count)
model = build(*args)
self.assertIsInstance(model, ModelOne)
def test_build_with_single_count(self):
count = 1
args = (ModelOne, count)
model = build(*args)
self.assertIsInstance(model, ModelOne)
def test_build_trivial(self):
model = build(ModelOne)
self.assertIsInstance(model, ModelOne)
def test_build_with_multi_model_trivial(self):
models = build((ModelOne, ), (ModelTwo, ))
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
