def infer_literal_type(value):
import ibis.expr.rules as rules
if value is None or value is null:
return dt.null
elif isinstance(value, bool):
return dt.boolean
elif isinstance(value, compat.integer_types):
return rules.int_literal_class(value)
elif isinstance(value, float):
return dt.double
elif isinstance(value, six.string_types):
return dt.string
elif isinstance(value, datetime.datetime):
return dt.timestamp
elif isinstance(value, datetime.date):
return dt.date
elif isinstance(value, datetime.time):
return dt.time
elif isinstance(value, list):
if not value:
return dt.Array(dt.any)
return dt.Array(
rules.highest_precedence_type(list(map(literal, value))))
raise com.InputTypeError(value)
def output_type(self):
import ibis.expr.rules as rules
value = self.value
if isinstance(value, bool):
return BooleanScalar
elif isinstance(value, compat.integer_types):
return rules.int_literal_class(value).scalar_type()
elif isinstance(value, float):
return DoubleScalar
elif isinstance(value, six.string_types):
return StringScalar
elif isinstance(value, datetime.datetime):
return TimestampScalar
elif isinstance(value, datetime.date):
return DateScalar
elif isinstance(value, list):
value_type = rules.highest_precedence_type(
list(map(literal, value))
)
return lambda value, value_type=value_type: ArrayScalar(
value, dt.Array(value_type)
)
raise com.InputTypeError(value)
def infer_literal_type(value):
import ibis.expr.rules as rules
if value is None or value is null:
return dt.null
elif isinstance(value, bool):
return dt.boolean
elif isinstance(value, compat.integer_types):
return rules.int_literal_class(value)
elif isinstance(value, float):
return dt.double
elif isinstance(value, six.string_types):
return dt.string
elif isinstance(value, datetime.timedelta):
return dt.interval
elif isinstance(value, datetime.datetime):
return dt.timestamp
elif isinstance(value, datetime.date):
return dt.date
elif isinstance(value, datetime.time):
return dt.time
elif isinstance(value, list):
if not value:
return dt.Array(dt.null)
return dt.Array(
rules.highest_precedence_type(list(map(literal, value))))
elif isinstance(value, collections.OrderedDict):
if not value:
raise TypeError('Empty struct type not supported')
return dt.Struct(
list(value.keys()),
[literal(element).type() for element in value.values()],
)
elif isinstance(value, dict):
if not value:
return dt.Map(dt.null, dt.null)
return dt.Map(
rules.highest_precedence_type(list(map(literal, value.keys()))),
rules.highest_precedence_type(list(map(literal, value.values()))),
)
raise com.InputTypeError(value)
def test_timestamp_precedence():
ts = ibis.literal(datetime.now())
null_ts = ibis.NA
highest_type = highest_precedence_type([ts, null_ts])
assert highest_type == 'timestamp'
def output_type(self):
cases, results, default = self.args
out_exprs = results + [default]
typename = rules.highest_precedence_type(out_exprs)
return rules.shape_like_args(cases, typename)
def output_type(self):
cases, results, default = self.args
out_exprs = results + [default]
typename = rules.highest_precedence_type(out_exprs)
return rules.shape_like_args(cases, typename)
def test_precedence_with_no_arguments():
with pytest.raises(ValueError) as e:
highest_precedence_type([])
assert str(e.value) == 'Must pass at least one expression'
def test_timestamp_precedence():
ts = ibis.literal(datetime.now())
null_ts = ibis.NA
highest_type = highest_precedence_type([ts, null_ts])
assert highest_type == 'timestamp'
def type(self):
from ibis.expr import rules
return rules.highest_precedence_type(self.values)
def test_precedence_with_no_arguments():
with pytest.raises(ValueError) as e:
highest_precedence_type([])
assert str(e.value) == 'Must pass at least one expression'
