def test_split_data_type_empty_string(self):
db_type = ''
raw_type, raw_arguments = TypeEngine.split_data_type(db_type)
self.assertEqual(raw_type, '')
self.assertEqual(len(raw_arguments), 0)
def test_split_data_type_without_argument(self):
db_type = 'varchar'
raw_type, raw_arguments = TypeEngine.split_data_type(db_type)
self.assertEqual(raw_type, 'varchar')
self.assertEqual(len(raw_arguments), 0)
def test_split_data_type_with_single_argument(self):
db_type = 'varchar(255)'
raw_type, raw_arguments = TypeEngine.split_data_type(db_type)
self.assertEqual(raw_type, 'varchar')
self.assertEqual(len(raw_arguments), 1)
self.assertEqual(raw_arguments[0], '255')
def test_split_data_type_with_multiple_arguments(self):
db_type = 'numeric(10, 5)'
raw_type, raw_arguments = TypeEngine.split_data_type(db_type)
self.assertEqual(raw_type, 'numeric')
self.assertEqual(len(raw_arguments), 2)
self.assertEqual(raw_arguments[0], '10')
self.assertEqual(raw_arguments[1], '5')
def setUpClass(cls):
cls.type_engine = TypeEngine(db_to_ansi_mapper={
'char': Char,
'int': Integer,
},
ansi_to_db_mapper={
'CHAR': 'char',
'INTEGER': 'integer',
})