mirror of
https://ghfast.top/https://github.com/aeroxw/easy_tdx_max.git
synced 2026-09-12 21:34:21 +08:00
毛利哥andGitHub
12d308e1a4
read_ex_daily_bars 原把第7槽（成交量）同时赋给 amount 与 vol（amount=vol）， 真实成交额藏在第6槽 float32 重解释值里、以 hk_stock_amount 之名暴露。 实测对照：47#IF300 2023-09-11 第6槽 float32=186871758848（元）、第7槽 uint32=105358016（手），与标准市场 sh000300.day 同日 amount/vol 完全一致。 - _EX_DAILY_FMT: <IffffIIf → <IfffffIf，第6槽正名为 float32 成交额 - ExDailyBar.amount: int → float；移除语义错误的 hk_stock_amount 字段 - encode 端沿新 fmt 自动正确：大成交额按 float32 编码，不再溢出 uint32 - CLI offline ex-daily 输出新增 amount 列 - 新增真实样本回归测试（47#IF300 原始记录）+ 补齐缺失的 amount 往返断言 Co-authored-by: GitHub <action@github.com>
229 lines
7.6 KiB
Python
229 lines
7.6 KiB
Python
"""离线扩展市场日线写入测试（纯离线，无网络）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import struct
|
|
from pathlib import Path
|
|
|
|
from easy_tdx.offline.ex_daily_bar import _EX_DAILY_FMT, ExDailyBar, read_ex_daily_bars
|
|
from easy_tdx.offline.write_ex_daily import (
|
|
append_ex_daily_bars,
|
|
encode_ex_daily_bar,
|
|
get_last_ex_bar_date,
|
|
sync_ex_daily_bars,
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# helpers
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def _make_ex_bar(
|
|
year: int = 2026,
|
|
month: int = 6,
|
|
day: int = 6,
|
|
open_: float = 3500.0,
|
|
high: float = 3550.0,
|
|
low: float = 3480.0,
|
|
close: float = 3520.0,
|
|
amount: float = 123456.0,
|
|
vol: int = 98765,
|
|
settlement: float = 3510.0,
|
|
) -> ExDailyBar:
|
|
return ExDailyBar(
|
|
open=open_,
|
|
close=close,
|
|
high=high,
|
|
low=low,
|
|
amount=amount,
|
|
vol=vol,
|
|
settlement=settlement,
|
|
year=year,
|
|
month=month,
|
|
day=day,
|
|
)
|
|
|
|
|
|
def _make_raw_ex_bar(
|
|
year: int = 2026,
|
|
month: int = 6,
|
|
day: int = 6,
|
|
open_: float = 3500.0,
|
|
high: float = 3550.0,
|
|
low: float = 3480.0,
|
|
close: float = 3520.0,
|
|
amount: float = 123456.0,
|
|
vol: int = 98765,
|
|
settlement: float = 3510.0,
|
|
) -> bytes:
|
|
date_int = year * 10000 + month * 100 + day
|
|
return _EX_DAILY_FMT.pack(date_int, open_, high, low, close, amount, vol, settlement)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# encode_ex_daily_bar
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestEncodeExDailyBar:
|
|
def test_output_length(self) -> None:
|
|
bar = _make_ex_bar()
|
|
result = encode_ex_daily_bar(bar)
|
|
assert len(result) == 32
|
|
|
|
def test_date_encoding(self) -> None:
|
|
bar = _make_ex_bar(year=2025, month=12, day=25)
|
|
result = encode_ex_daily_bar(bar)
|
|
date_int = struct.unpack_from("<I", result, 0)[0]
|
|
assert date_int == 20251225
|
|
|
|
def test_float_prices_preserved(self) -> None:
|
|
bar = _make_ex_bar(open_=3500.5, high=3560.25, low=3479.75, close=3520.0)
|
|
result = encode_ex_daily_bar(bar)
|
|
op, hi, lo, cl = struct.unpack_from("<ffff", result, 4)
|
|
assert abs(op - 3500.5) < 0.01
|
|
assert abs(hi - 3560.25) < 0.01
|
|
assert abs(lo - 3479.75) < 0.01
|
|
assert abs(cl - 3520.0) < 0.01
|
|
|
|
def test_settlement_encoding(self) -> None:
|
|
bar = _make_ex_bar(settlement=3510.5)
|
|
result = encode_ex_daily_bar(bar)
|
|
(sett,) = struct.unpack_from("<f", result, 28)
|
|
assert abs(sett - 3510.5) < 0.01
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# round-trip
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestExRoundTrip:
|
|
def test_single_bar_round_trip(self, tmp_path: Path) -> None:
|
|
bar = _make_ex_bar(open_=3500.0, close=3520.0, amount=123456.0, vol=98765)
|
|
encoded = encode_ex_daily_bar(bar)
|
|
|
|
filepath = tmp_path / "29#A1801.day"
|
|
filepath.write_bytes(encoded)
|
|
|
|
bars = read_ex_daily_bars(filepath)
|
|
assert len(bars) == 1
|
|
b = bars[0]
|
|
assert b.year == 2026 and b.month == 6 and b.day == 6
|
|
assert abs(b.open - 3500.0) < 0.01
|
|
assert abs(b.close - 3520.0) < 0.01
|
|
assert abs(b.amount - 123456.0) < 0.01
|
|
assert b.vol == 98765
|
|
|
|
def test_multiple_bars_round_trip(self, tmp_path: Path) -> None:
|
|
bars_in = [
|
|
_make_ex_bar(year=2026, month=6, day=4, open_=3400.0),
|
|
_make_ex_bar(year=2026, month=6, day=5, open_=3450.0),
|
|
_make_ex_bar(year=2026, month=6, day=6, open_=3500.0),
|
|
]
|
|
encoded = b"".join(encode_ex_daily_bar(b) for b in bars_in)
|
|
|
|
filepath = tmp_path / "29#A1801.day"
|
|
filepath.write_bytes(encoded)
|
|
|
|
bars_out = read_ex_daily_bars(filepath)
|
|
assert len(bars_out) == 3
|
|
assert bars_out[0].day == 4
|
|
assert bars_out[2].day == 6
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# get_last_ex_bar_date
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestGetLastExBarDate:
|
|
def test_returns_last_date(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "test.day"
|
|
filepath.write_bytes(_make_raw_ex_bar(year=2026, month=6, day=5))
|
|
assert get_last_ex_bar_date(filepath) == 20260605
|
|
|
|
def test_returns_none_for_empty(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "test.day"
|
|
filepath.write_bytes(b"")
|
|
assert get_last_ex_bar_date(filepath) is None
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# append & sync
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestAppendExDailyBars:
|
|
def test_append_to_existing(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "test.day"
|
|
filepath.write_bytes(_make_raw_ex_bar(year=2026, month=6, day=5))
|
|
|
|
new_bar = _make_ex_bar(year=2026, month=6, day=6)
|
|
written = append_ex_daily_bars(filepath, [new_bar])
|
|
assert written == 1
|
|
|
|
bars = read_ex_daily_bars(filepath)
|
|
assert len(bars) == 2
|
|
|
|
def test_skips_duplicate(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "test.day"
|
|
filepath.write_bytes(_make_raw_ex_bar(year=2026, month=6, day=6))
|
|
|
|
new_bar = _make_ex_bar(year=2026, month=6, day=6)
|
|
written = append_ex_daily_bars(filepath, [new_bar])
|
|
assert written == 0
|
|
|
|
def test_sync_filters_correctly(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "test.day"
|
|
filepath.write_bytes(_make_raw_ex_bar(year=2026, month=6, day=5))
|
|
|
|
server_bars = [
|
|
_make_ex_bar(year=2026, month=6, day=4),
|
|
_make_ex_bar(year=2026, month=6, day=5),
|
|
_make_ex_bar(year=2026, month=6, day=6),
|
|
]
|
|
written = sync_ex_daily_bars(filepath, server_bars)
|
|
assert written == 1
|
|
|
|
bars = read_ex_daily_bars(filepath)
|
|
assert len(bars) == 2
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# 真实样本回归（issue #57）：槽位语义
|
|
# ---------------------------------------------------------------------------
|
|
|
|
# 47#IF300（扩展市场沪深300 指数）2023-09-11 的真实记录。
|
|
# 与标准市场 sh000300.day 同日对照：vol=105358016（手）、amount=186871758848.0（元）完全一致，
|
|
# 证明第 6 槽是 float32 成交额，第 7 槽是 uint32 成交量。
|
|
_REAL_IF300_20230911 = bytes.fromhex(
|
|
"ffb2340148c969451fc56c45856b6945a4786b45b3092e52c0a2470600000000"
|
|
)
|
|
|
|
|
|
class TestRealSampleSlots:
|
|
def test_amount_is_float32_turnover(self, tmp_path: Path) -> None:
|
|
filepath = tmp_path / "47#IF300.day"
|
|
filepath.write_bytes(_REAL_IF300_20230911)
|
|
|
|
bars = read_ex_daily_bars(filepath)
|
|
assert len(bars) == 1
|
|
b = bars[0]
|
|
assert (b.year, b.month, b.day) == (2023, 9, 11)
|
|
assert b.amount == 186871758848.0
|
|
assert b.vol == 105358016
|
|
assert abs(b.close - 3767.54) < 0.01
|
|
# amount 与 vol 不得串槽（回归：旧实现 amount=vol）
|
|
assert b.amount != b.vol
|
|
|
|
def test_large_amount_does_not_overflow(self, tmp_path: Path) -> None:
|
|
# 真实成交额动辄超 uint32 上限，第 6 槽必须按 float32 编码
|
|
bar = _make_ex_bar(amount=186871758848.0)
|
|
encoded = encode_ex_daily_bar(bar)
|
|
filepath = tmp_path / "47#IF300.day"
|
|
filepath.write_bytes(encoded)
|
|
|
|
(b,) = read_ex_daily_bars(filepath)
|
|
assert b.amount == 186871758848.0
|