26 lines
564 B
Python
26 lines
564 B
Python
from datetime import timedelta
|
|
|
|
import pytest
|
|
|
|
from t_tech.invest import CandleInterval
|
|
from t_tech.invest.utils import (
|
|
candle_interval_to_timedelta,
|
|
ceil_datetime,
|
|
floor_datetime,
|
|
now,
|
|
)
|
|
|
|
|
|
@pytest.fixture(params=[i.value for i in CandleInterval])
|
|
def interval(request) -> timedelta:
|
|
return candle_interval_to_timedelta(request.param)
|
|
|
|
|
|
def test_floor_ceil(interval: timedelta):
|
|
now_ = now()
|
|
|
|
a, b = floor_datetime(now_, interval), ceil_datetime(now_, interval)
|
|
|
|
assert a < b
|
|
assert b - a == interval
|