dekun
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from datetime import datetime
|
|
|
|
from hub_volume_rank_lib import (
|
|
cache_needs_refresh,
|
|
format_volume_quote,
|
|
merge_exchange_rank,
|
|
rank_date_label,
|
|
)
|
|
|
|
|
|
def test_rank_date_label_after_reset():
|
|
# 2026-06-08 09:00 北京时间 → 昨日交易日 2026-06-07
|
|
dt = datetime(2026, 6, 8, 9, 0, 0)
|
|
assert rank_date_label(now=dt, reset_hour=8) == "2026-06-07"
|
|
|
|
|
|
def test_rank_date_label_before_reset():
|
|
# 2026-06-08 07:00 → 当前交易日仍算 2026-06-07，昨日为 2026-06-06
|
|
dt = datetime(2026, 6, 8, 7, 0, 0)
|
|
assert rank_date_label(now=dt, reset_hour=8) == "2026-06-06"
|
|
|
|
|
|
def test_format_volume_quote():
|
|
assert format_volume_quote(1_500_000_000) == "1.50B"
|
|
assert format_volume_quote(2_300_000) == "2.30M"
|
|
assert format_volume_quote(4500) == "4.50K"
|
|
|
|
|
|
def test_cache_needs_refresh_and_merge():
|
|
cache = {"rank_date": "2026-06-05", "exchanges": {}}
|
|
assert cache_needs_refresh(cache, expected_rank_date="2026-06-07") is True
|
|
merged = merge_exchange_rank(
|
|
cache,
|
|
"binance",
|
|
{
|
|
"ok": True,
|
|
"rank_date": "2026-06-07",
|
|
"items": [{"rank": 1, "symbol": "BTC/USDT", "volume_quote": 1.0}],
|
|
"total_symbols": 100,
|
|
},
|
|
)
|
|
assert merged["exchanges"]["binance"]["items"][0]["symbol"] == "BTC/USDT"
|
|
assert merged["rank_date"] == "2026-06-07"
|