def test_min_max_length(self):
v = ArrayField(IntegerField(), min_length=5, max_length=10)
generated = v.generate()
self.assertTrue(5 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), min_length=0, max_length=0)
generated = v.generate()
self.assertEqual(generated, [])
v = ArrayField(IntegerField(), min_length=10, max_length=5)
with self.assertRaises(ValueError):
generated = v.generate()
def test_min_max_length(self):
v = ArrayField(IntegerField(), min_length=5, max_length=10)
generated = v.generate()
self.assertTrue(5 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), min_length=0, max_length=0)
generated = v.generate()
self.assertEqual(generated, [])
v = ArrayField(IntegerField(), min_length=10, max_length=5)
with self.assertRaises(ValueError):
generated = v.generate()
def test_blankable(self):
v = ArrayField(IntegerField(), blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 1
mocked_random.choice = random.choice
mocked_random.randint = random.randint
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
mocked_random.choice = random.choice
mocked_random.randint = random.randint
self.assertEqual(v.generate(), [])
def test_blankable(self):
v = ArrayField(IntegerField(), blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 1
mocked_random.choice = random.choice
mocked_random.randint = random.randint
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), blank=True, blank_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
mocked_random.choice = random.choice
mocked_random.randint = random.randint
self.assertEqual(v.generate(), [])
def test_default(self):
v = ArrayField(IntegerField())
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
int_field = mock.Mock(wraps=IntegerField())
int_field.generate = lambda: 0
str_field = mock.Mock(wraps=StringField())
str_field.generate = lambda: 'string'
v = ArrayField([int_field, str_field])
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.randint.return_value = 2
mocked_random.choice.side_effect = [int_field, str_field]
generated = v.generate()
self.assertEqual(len(generated), 2)
self.assertEqual(generated[0], 0)
self.assertTrue(generated[1], 'string')
def test_default(self):
v = ArrayField(IntegerField())
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
int_field = mock.Mock(wraps=IntegerField())
int_field.generate = lambda: 0
str_field = mock.Mock(wraps=StringField())
str_field.generate = lambda: "string"
v = ArrayField([int_field, str_field])
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.randint.return_value = 2
mocked_random.choice.side_effect = [int_field, str_field]
generated = v.generate()
self.assertEqual(len(generated), 2)
self.assertEqual(generated[0], 0)
self.assertTrue(generated[1], "string")
def test_not_required(self):
v = ArrayField(IntegerField(), required=False, not_present_prob=0)
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), required=False, not_present_prob=1)
with self.assertRaises(NotGeneratedException):
v.generate()
v = ArrayField(IntegerField(), required=False, not_present_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
with self.assertRaises(NotGeneratedException):
v.generate()
v = ArrayField(IntegerField(), required=False, not_present_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 1
mocked_random.choice = random.choice
mocked_random.randint = random.randint
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
def test_not_required(self):
v = ArrayField(IntegerField(), required=False, not_present_prob=0)
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
v = ArrayField(IntegerField(), required=False, not_present_prob=1)
with self.assertRaises(NotGeneratedException):
v.generate()
v = ArrayField(IntegerField(), required=False, not_present_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 0
with self.assertRaises(NotGeneratedException):
v.generate()
v = ArrayField(IntegerField(), required=False, not_present_prob=0.5)
with mock.patch(FIELDS_RANDOM_MODULE) as mocked_random:
mocked_random.random.return_value = 1
mocked_random.choice = random.choice
mocked_random.randint = random.randint
generated = v.generate()
self.assertTrue(1 <= len(generated) <= 10)
self.assertTrue(all([isinstance(i, int) for i in generated]))
