def test_compute_signature():
s = symbol('s', 'int64')
t = symbol('t', 'float32')
d = symbol('d', 'datetime')
assert compute_signature(s + t) == float64(int64, float32)
assert (compute_signature(d.truncate(days=1)) ==
datetime64('D')(datetime64('us')))
assert compute_signature(d.day + 1) == int64(datetime64('us'))
def test_compute_signature():
s = symbol('s', 'int64')
t = symbol('t', 'float32')
d = symbol('d', 'datetime')
assert compute_signature(s + t) == float64(int64, float32)
assert (compute_signature(d.truncate(days=1)) == datetime64('D')(
datetime64('us')))
assert compute_signature(d.day + 1) == int64(datetime64('us'))
def test_get_numba_type():
assert get_numba_type(datashape.bool_) == nb.bool_
assert get_numba_type(datashape.date_) == datetime64('D')
assert get_numba_type(datashape.datetime_) == datetime64('us')
assert get_numba_type(datashape.timedelta_) == timedelta64('us')
assert get_numba_type(datashape.TimeDelta('D')) == timedelta64('D')
assert get_numba_type(datashape.int64) == int64
assert get_numba_type(datashape.String(7, "A")) == char(7)
assert get_numba_type(datashape.String(None, "A")) == nb.types.string
assert get_numba_type(datashape.String(7)) == unichar(7)
def test_get_numba_type():
assert get_numba_type(datashape.bool_) == nb.bool_
assert get_numba_type(datashape.date_) == datetime64('D')
assert get_numba_type(datashape.datetime_) == datetime64('us')
assert get_numba_type(datashape.timedelta_) == timedelta64('us')
assert get_numba_type(datashape.TimeDelta('D')) == timedelta64('D')
assert get_numba_type(datashape.int64) == int64
assert get_numba_type(datashape.String(7, "A")) == char(7)
assert get_numba_type(datashape.String(None, "A")) == nb.types.string
assert get_numba_type(datashape.String(7)) == unichar(7)
