def test_days_slice():
rate = Rate('rate(2 days)')
rate.now = datetime(year=2018, month=2, day=11, hour=12)
expected = [
datetime(year=2018, month=2, day=13),
datetime(year=2018, month=2, day=15),
]
assert rate.get_next(count=2) == expected
def test_day():
rate = Rate('rate(1 day)')
rate.now = datetime(year=2018, month=2, day=11, hour=12)
expected = [datetime(year=2018, month=2, day=12)]
assert rate.get_next() == expected
def test_invalid_day():
rate = Rate('rate(1 day)')
rate.now = datetime(year=2018, month=2, day=11, hour=12)
with pytest.raises(ValueError):
result = rate.get_next(1, 2)