def test_schema_type_conversion():
typespec = [
# name, type, nullable
('json', postgresql.JSON, True, dt.JSON),
('jsonb', postgresql.JSONB, True, dt.JSONB),
('uuid', postgresql.UUID, True, dt.UUID),
('macaddr', postgresql.MACADDR, True, dt.MACADDR),
('inet', postgresql.INET, True, dt.INET),
]
sqla_types = []
ibis_types = []
for name, t, nullable, ibis_type in typespec:
sqla_type = sa.Column(name, t, nullable=nullable)
sqla_types.append(sqla_type)
ibis_types.append((name, ibis_type(nullable=nullable)))
# Create a table with placeholder stubs for JSON, JSONB, and UUID.
engine = sa.create_engine('postgresql://')
table = sa.Table('tname', sa.MetaData(bind=engine), *sqla_types)
# Check that we can correctly create a schema with dt.any for the
# missing types.
schema = schema_from_table(table)
expected = ibis.schema(ibis_types)
assert_equal(schema, expected)
def test_sqla_schema_conversion(self):
typespec = [
# name, type, nullable
('smallint', sat.SmallInteger, False, dt.int16),
('int', sat.Integer, True, dt.int32),
('integer', sat.INTEGER(), True, dt.int32),
('bigint', sat.BigInteger, False, dt.int64),
('real', sat.REAL, True, dt.float),
('bool', sat.Boolean, True, dt.boolean),
('timestamp', sat.DateTime, True, dt.timestamp),
]
sqla_types = []
ibis_types = []
for name, t, nullable, ibis_type in typespec:
sqla_type = sa.Column(name, t, nullable=nullable)
sqla_types.append(sqla_type)
ibis_types.append((name, ibis_type(nullable=nullable)))
table = sa.Table('tname', self.meta, *sqla_types)
schema = schema_from_table(table)
expected = ibis.schema(ibis_types)
assert_equal(schema, expected)
def test_sa_default_numeric_precision_and_scale(con, backend, dialects,
default_precisions,
default_scales):
# TODO: find a better way to access ibis.sql.alchemy
from ibis.backends.base.sql.alchemy import schema_from_table
dialect = dialects[backend.name()]
default_precision = default_precisions[backend.name()]
default_scale = default_scales[backend.name()]
typespec = [
# name, sqlalchemy type, ibis type
('n1', dialect.NUMERIC, dt.Decimal(default_precision, default_scale)),
('n2', dialect.NUMERIC(5), dt.Decimal(5, default_scale)),
('n3', dialect.NUMERIC(None, 4), dt.Decimal(default_precision, 4)),
('n4', dialect.NUMERIC(10, 2), dt.Decimal(10, 2)),
]
sqla_types = []
ibis_types = []
for name, t, ibis_type in typespec:
sqla_type = sa.Column(name, t, nullable=True)
sqla_types.append(sqla_type)
ibis_types.append((name, ibis_type(nullable=True)))
# Create a table with the numeric types.
table_name = 'test_sa_default_param_decimal'
engine = con.con
table = sa.Table(table_name, sa.MetaData(bind=engine), *sqla_types)
# Check that we can correctly recover the default precision and scale.
schema = schema_from_table(table)
expected = ibis.schema(ibis_types)
assert_equal(schema, expected)
con.drop_table(table_name, force=True)