def test_divide_int(self):
op = batch_operator_manager.get(models.IntegerField, 'div_int')
self.assertEqual(3, op(6, 2))
self.assertEqual(2, op(9, op.cast('4')))
def test_substract_int(self):
op = batch_operator_manager.get(models.IntegerField, 'sub_int')
self.assertEqual(1, op(3, 2))
self.assertEqual(3, op(4, op.cast('1')))
def test_multiply_int(self):
op = batch_operator_manager.get(models.IntegerField, 'mul_int')
self.assertEqual(6, op(3, 2))
self.assertEqual(8, op(2, op.cast('4')))
def test_remove_end(self):
op = batch_operator_manager.get(models.CharField, 'rm_end')
self.assertTrue(op.need_arg)
self.assertEqual('Gally', op('Gallyyy', 2))
self.assertEqual('', op('Gally', op.cast('10')))
def test_add_int(self):
op = batch_operator_manager.get(models.IntegerField, 'add_int')
self.assertEqual(3, op(1, 2))
self.assertEqual(5, op(4, op.cast('1')))
def test_remove_substring(self):
op = batch_operator_manager.get(models.CharField, 'rm_substr')
self.assertTrue(op.need_arg)
fieldval = 'Gally the battle angel'
self.assertEqual('Gally the angel', op(fieldval, 'battle '))
self.assertEqual(fieldval, op(fieldval, 'evil '))
def test_suffix(self):
op = batch_operator_manager.get(models.CharField, 'suffix')
self.assertTrue(op.need_arg)
self.assertEqual('Gally my love', op('Gally', ' my love'))
def test_prefix(self):
op = batch_operator_manager.get(models.CharField, 'prefix')
self.assertTrue(op.need_arg)
self.assertEqual('My Gally', op('Gally', 'My '))
def test_title(self):
op = batch_operator_manager.get(models.CharField, 'title')
self.assertFalse(op.need_arg)
self.assertEqual('Gally', op('gally'))
def test_lower(self):
op = batch_operator_manager.get(models.CharField, 'lower')
self.assertFalse(op.need_arg)
self.assertEqual('gally', op('GALLY'))
