def test_timedelta_strings():
inputs = [
"1 day", "-2 hours", "3 seconds", "1 microsecond", "1003 milliseconds"
]
for ts in inputs:
assert discover(ts) == TimeDelta(unit=ts.split()[1])
with pytest.raises(ValueError):
TimeDelta(unit='buzz light-years')
def test_timedelta():
assert_dshape_equal(TimeDelta(), TimeDelta())
assert_dshape_equal(TimeDelta('ns'), TimeDelta('ns'))
with pytest.raises(AssertionError) as e:
assert_dshape_equal(TimeDelta('us'), TimeDelta('ns'))
assert "'us' != 'ns'" in str(e.value)
assert '_.unit' in str(e.value)
assert_dshape_equal(
TimeDelta('us'),
TimeDelta('ns'),
check_timedelta_unit=False,
)
def test_timedelta_str(unit):
assert str(TimeDelta(unit=unit)) == 'timedelta[unit=%r]' % unit
def test_timedelta_repr(unit):
assert repr(TimeDelta(unit=unit)) == 'TimeDelta(unit=%r)' % unit
def test_option_timedelta_to_numpy(unit):
assert (Option(TimeDelta(unit=unit)).to_numpy_dtype() == np.dtype(
'timedelta64[%s]' % unit))
