dekun
28 lines
780 B
Python
28 lines
780 B
Python
from datetime import datetime
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from packages.domain.buckets import shanghai_bucket, shanghai_bucket_hour, shanghai_day
|
|
|
|
_SH = ZoneInfo("Asia/Shanghai")
|
|
|
|
|
|
def _ms(y, m, d, hh, mm=0):
|
|
return int(datetime(y, m, d, hh, mm, tzinfo=_SH).timestamp() * 1000)
|
|
|
|
|
|
def test_shanghai_day():
|
|
# UTC 2026-07-31 16:00 = 上海 2026-08-01 00:00
|
|
ts = int(datetime(2026, 7, 31, 16, 0, tzinfo=ZoneInfo("UTC")).timestamp() * 1000)
|
|
assert shanghai_day(ts) == "2026-08-01"
|
|
|
|
|
|
def test_bucket_hour():
|
|
ts = _ms(2026, 7, 31, 14, 35)
|
|
assert shanghai_bucket(ts, 60) == 14 * 60
|
|
assert shanghai_bucket_hour(ts) == 14
|
|
|
|
|
|
def test_bucket_30m():
|
|
ts = _ms(2026, 7, 31, 14, 35)
|
|
assert shanghai_bucket(ts, 30) == 14 * 60 + 30
|