if e.match(space, space.w_OverflowError):
if space.is_true(space.le(w_obj, space.wrap(0))):
return int64_dtype
return uint64_dtype
raise
return int64_dtype
elif space.isinstance_w(w_obj, space.w_float):
return float_dtype
elif space.isinstance_w(w_obj, space.w_complex):
return complex_dtype
elif space.isinstance_w(w_obj, space.w_str):
return variable_dtype(space, 'S%d' % space.len_w(w_obj))
return object_dtype
@signature(ann.instance(W_Dtype), ann.instance(W_Dtype), returns=ann.bool())
def can_cast_to(dt1, dt2):
"""Return whether dtype `dt1` can be cast safely to `dt2`"""
# equivalent to PyArray_CanCastTo
from .casting import can_cast_itemtype
result = can_cast_itemtype(dt1.itemtype, dt2.itemtype)
if result:
if dt1.num == NPY.STRING:
if dt2.num == NPY.STRING:
return dt1.elsize <= dt2.elsize
elif dt2.num == NPY.UNICODE:
return dt1.elsize * 4 <= dt2.elsize
elif dt1.num == NPY.UNICODE and dt2.num == NPY.UNICODE:
return dt1.elsize <= dt2.elsize
elif dt2.num in (NPY.STRING, NPY.UNICODE):
if dt2.num == NPY.STRING:
except OperationError, e:
if e.match(space, space.w_OverflowError):
if space.is_true(space.le(w_obj, space.wrap(0))):
return int64_dtype
return uint64_dtype
raise
return int64_dtype
elif space.isinstance_w(w_obj, space.w_float):
return float_dtype
elif space.isinstance_w(w_obj, space.w_complex):
return complex_dtype
elif space.isinstance_w(w_obj, space.w_str):
return variable_dtype(space, 'S%d' % space.len_w(w_obj))
return object_dtype
@signature(ann.instance(W_Dtype), ann.instance(W_Dtype), returns=ann.bool())
def can_cast_to(dt1, dt2):
"""Return whether dtype `dt1` can be cast safely to `dt2`"""
# equivalent to PyArray_CanCastTo
from .casting import can_cast_itemtype
result = can_cast_itemtype(dt1.itemtype, dt2.itemtype)
if result:
if dt1.num == NPY.STRING:
if dt2.num == NPY.STRING:
return dt1.elsize <= dt2.elsize
elif dt2.num == NPY.UNICODE:
return dt1.elsize * 4 <= dt2.elsize
elif dt1.num == NPY.UNICODE and dt2.num == NPY.UNICODE:
return dt1.elsize <= dt2.elsize
elif dt2.num in (NPY.STRING, NPY.UNICODE):
if dt2.num == NPY.STRING: