mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 17:54:15 +08:00
intfoo
865e75fc8b
阻断项1: 前端 TypeScript 类型扩展 - MonitorRule.asset_type 加 'index' (api.ts:517) - screenerStrategies 参数加 'index' (api.ts:1412) - klineMinute 响应 asset_type 去重 (api.ts:1296) 阻断项2: 指数监控独立评估 - _evaluate_monitors 股票早期 return 降级为 stock_ready 标志 仅跳过股票轮, ETF/指数轮独立判断数据新鲜度 - 纯指数行情/自选场景下指数规则可正常触发 阻断项3: 核心指数模式不截断分区 - _process_full_market_records 按 index_mode 条件分支: mode=all (完整 CN_Index) → flush 覆盖; mode=core (部分标的) → merge 不截断 - merge_live_enriched_asset 对 index 正确更新 _index_enriched_cache 阻断项4: Free 档额度分批 - _fetch_watchlist_quotes 用 resolve_limit + chunked 按 capability batch 上限分批 - 失败批次跳过不整轮退出, 已有股票实时刷新不受影响 - 复用进程级共享限速器 sleep_between_batches 测试: +7 测试覆盖 4 个阻断项核心场景
79 lines
3.2 KiB
Python
79 lines
3.2 KiB
Python
"""回归测试: 实时指数 merge 不截断盘后管道写入的全量分区 (PR #46 问题 3)。"""
|
|
from datetime import date
|
|
|
|
import polars as pl
|
|
|
|
from app.tickflow.repository import DataStore, KlineRepository
|
|
|
|
|
|
def _enriched_row(symbol: str, close: float, dt: date) -> dict:
|
|
return {
|
|
"symbol": symbol, "date": dt,
|
|
"open": close, "high": close, "low": close, "close": close,
|
|
"volume": 1000, "amount": 10000.0,
|
|
"quote_ts": 1753700400000,
|
|
}
|
|
|
|
|
|
def test_merge_live_enriched_preserves_full_index_partition(tmp_path):
|
|
"""盘后管道 flush 写入全量指数后, 实时 merge 部分指数不丢已有数据。"""
|
|
repo = KlineRepository(DataStore(tmp_path))
|
|
dt = date(2026, 7, 28)
|
|
|
|
# 模拟盘后管道: flush 写入全量 3 只指数
|
|
full_df = pl.DataFrame([
|
|
_enriched_row("000001.SH", 3000.0, dt),
|
|
_enriched_row("399001.SZ", 10000.0, dt),
|
|
_enriched_row("399006.SZ", 2000.0, dt),
|
|
])
|
|
repo.flush_live_enriched_asset("index", full_df)
|
|
|
|
# 模拟实时刷新: 只 merge 核心指数 1 只 (价格更新)
|
|
partial_df = pl.DataFrame([
|
|
_enriched_row("000001.SH", 3001.0, dt),
|
|
])
|
|
repo.merge_live_enriched_asset("index", partial_df)
|
|
|
|
# 验证: 分区文件仍有 3 只指数, 000001.SH 价格已更新, 其他指数未丢失
|
|
out = tmp_path / "kline_index_enriched" / f"date={dt.isoformat()}" / "part.parquet"
|
|
result = pl.read_parquet(out)
|
|
assert len(result) == 3, f"merge 后分区应有 3 只指数, 实际 {len(result)}"
|
|
|
|
sh = result.filter(pl.col("symbol") == "000001.SH")
|
|
assert sh["close"][0] == 3001.0, "merge 应更新 000001.SH 价格"
|
|
|
|
sz = result.filter(pl.col("symbol") == "399001.SZ")
|
|
assert sz["close"][0] == 10000.0, "399001.SZ 不应被 merge 覆盖"
|
|
|
|
cyb = result.filter(pl.col("symbol") == "399006.SZ")
|
|
assert cyb["close"][0] == 2000.0, "399006.SZ 不应被 merge 覆盖"
|
|
|
|
|
|
def test_merge_live_daily_preserves_full_index_partition(tmp_path):
|
|
"""日K merge 同样不截断全量分区。"""
|
|
repo = KlineRepository(DataStore(tmp_path))
|
|
dt = date(2026, 7, 28)
|
|
|
|
# 盘后管道 flush 写入全量 3 只指数日K
|
|
full_df = pl.DataFrame([
|
|
{"symbol": "000001.SH", "date": dt, "open": 3000.0, "high": 3010.0,
|
|
"low": 2990.0, "close": 3000.0, "volume": 1000, "amount": 10000.0},
|
|
{"symbol": "399001.SZ", "date": dt, "open": 10000.0, "high": 10010.0,
|
|
"low": 9990.0, "close": 10000.0, "volume": 2000, "amount": 20000.0},
|
|
{"symbol": "399006.SZ", "date": dt, "open": 2000.0, "high": 2010.0,
|
|
"low": 1990.0, "close": 2000.0, "volume": 3000, "amount": 30000.0},
|
|
])
|
|
repo.flush_live_daily_asset("index", full_df)
|
|
|
|
# 实时 merge 部分指数
|
|
partial_df = pl.DataFrame([
|
|
{"symbol": "000001.SH", "date": dt, "open": 3000.0, "high": 3010.0,
|
|
"low": 2990.0, "close": 3001.0, "volume": 1000, "amount": 10000.0},
|
|
])
|
|
repo.merge_live_daily_asset("index", partial_df)
|
|
|
|
out = tmp_path / "kline_index_daily" / f"date={dt.isoformat()}" / "part.parquet"
|
|
result = pl.read_parquet(out)
|
|
assert len(result) == 3, f"merge 后分区应有 3 只指数, 实际 {len(result)}"
|
|
assert result.filter(pl.col("symbol") == "000001.SH")["close"][0] == 3001.0
|