def assert_type(type_string, type_expected, generic_type_expected): if type_expected is None: type_assigned = MssqlEngineSpec.get_sqla_column_type(type_string) self.assertIsNone(type_assigned) else: column_spec = MssqlEngineSpec.get_column_spec(type_string) if column_spec != None: self.assertIsInstance(column_spec.sqla_type, type_expected) self.assertEquals(column_spec.generic_type, generic_type_expected)
def test_mssql_column_types( type_string: str, type_expected: TypeEngine, generic_type_expected: GenericDataType, ) -> None: from superset.db_engine_specs.mssql import MssqlEngineSpec if type_expected is None: type_assigned = MssqlEngineSpec.get_sqla_column_type(type_string) assert type_assigned is None else: column_spec = MssqlEngineSpec.get_column_spec(type_string) if column_spec is not None: assert isinstance(column_spec.sqla_type, type_expected) assert column_spec.generic_type == generic_type_expected