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)