def test_valid_interval(units, value, expected):
result = rlz.interval(value, units=units)
assert result.equals(expected)
def test_invalid_interval(units, value, expected):
with pytest.raises(expected):
rlz.interval(value, units=units)
class TimestampSub(Binary):
left = rlz.timestamp
right = rlz.interval(
units={'Y', 'Q', 'M', 'W', 'D', 'h', 'm', 's', 'ms', 'us', 'ns'}
)
output_dtype = rlz.dtype_like('left')
class TimeSub(Binary):
left = rlz.time
right = rlz.interval(units={'h', 'm', 's', 'ms', 'us', 'ns'})
output_dtype = rlz.dtype_like('left')
class DateSub(Binary):
left = rlz.date
right = rlz.interval(units={'Y', 'Q', 'M', 'W', 'D'})
output_dtype = rlz.dtype_like('left')
def test_invalid_interval(units, value, expected):
with pytest.raises(expected):
rlz.interval(value, units=units)
def test_valid_interval(units, value, expected):
result = rlz.interval(value, units=units)
assert result.equals(expected)
