Python assert_generic_types Examples

Example #1

 def test_generic_type(self):
     type_expectations = (
         # Numeric
         ("TINYINT", GenericDataType.NUMERIC),
         ("SMALLINT", GenericDataType.NUMERIC),
         ("MEDIUMINT", GenericDataType.NUMERIC),
         ("INT", GenericDataType.NUMERIC),
         ("BIGINT", GenericDataType.NUMERIC),
         ("DECIMAL", GenericDataType.NUMERIC),
         ("FLOAT", GenericDataType.NUMERIC),
         ("DOUBLE", GenericDataType.NUMERIC),
         ("BIT", GenericDataType.NUMERIC),
         # String
         ("CHAR", GenericDataType.STRING),
         ("VARCHAR", GenericDataType.STRING),
         ("TINYTEXT", GenericDataType.STRING),
         ("MEDIUMTEXT", GenericDataType.STRING),
         ("LONGTEXT", GenericDataType.STRING),
         # Temporal
         ("DATE", GenericDataType.TEMPORAL),
         ("DATETIME", GenericDataType.TEMPORAL),
         ("TIMESTAMP", GenericDataType.TEMPORAL),
         ("TIME", GenericDataType.TEMPORAL),
     )
     assert_generic_types(MySQLEngineSpec, type_expectations)

Example #2

File: postgres_tests.py Project: cancanmobile/superset

def test_generic_type():
    type_expectations = (
        # Numeric
        ("SMALLINT", GenericDataType.NUMERIC),
        ("INTEGER", GenericDataType.NUMERIC),
        ("BIGINT", GenericDataType.NUMERIC),
        ("DECIMAL", GenericDataType.NUMERIC),
        ("NUMERIC", GenericDataType.NUMERIC),
        ("REAL", GenericDataType.NUMERIC),
        ("DOUBLE PRECISION", GenericDataType.NUMERIC),
        ("MONEY", GenericDataType.NUMERIC),
        # String
        ("CHAR", GenericDataType.STRING),
        ("VARCHAR", GenericDataType.STRING),
        ("TEXT", GenericDataType.STRING),
        # Temporal
        ("DATE", GenericDataType.TEMPORAL),
        ("TIMESTAMP", GenericDataType.TEMPORAL),
        ("TIME", GenericDataType.TEMPORAL),
        # Boolean
        ("BOOLEAN", GenericDataType.BOOLEAN),
    )
    assert_generic_types(PostgresEngineSpec, type_expectations)