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_record_type(): get_numba_type(datashape.dshape('10 * {a: int64}'))
def test_get_numba_type_failures(): get_numba_type(datashape.string)
def test_fail_on_object_type(): with pytest.raises(TypeError): get_numba_type(datashape.object_)