Files
tick-stock-panel/backend/tests/test_enriched_range_warming_guard.py
T
shy3130 697c27bb02 feat(v0.2): 市场阶段与主线识别 + 因子挖掘全链路 + 数据层完善
- 市场环境: 新增情绪周期6阶段(冰点/启动/主升/高潮/退潮/修复, 连板梯队驱动,
  EMA平滑+2日确认+弱档否决, 平均段长9.7天)与概念/行业主线排名(涨停梯队聚合,
  可配置宽基/风格标签过滤); 市场环境页重构, regime 透明加列, 与5档state并存
- 挖掘: 因子与策略挖掘全链路(API/worker/进程锁/候选库/前端工作台/文档),
  周度调度默认关闭且永不自动发布
- 回测: 财务快照因子(点时口径), 批量回测预计算共享下期收益,
  信号路径矩阵列依赖展开修复(consecutive_limit_ups 缺列报错)
- 数据/性能: enriched 生成与预热治理, 重任务限流, 行情/K线缓存复用, 时区修复
- 测试: 后端全量 914 通过; GUI 黑盒验证截图存证 gui-test-screenshots/
2026-08-16 23:39:07 +08:00

57 lines
1.8 KiB
Python

"""get_enriched_range 的预热守卫测试 — 预热期间不得触发同步全量重算。
启动后台预热线程正在 _refresh_enriched (300 天 scan + compute, 低配机 50s+)
时, 请求线程进入 get_enriched_range 应返回 None (缓存不覆盖语义),
不能在请求线程里并发跑第二次全量重算。与 get_enriched_latest 守卫对齐。
"""
from __future__ import annotations
from datetime import date
import polars as pl
from app.tickflow.repository import KlineRepository
def _bare_repo() -> KlineRepository:
"""跳过 __init__ (避免 DataStore/目录依赖), 只装配守卫涉及的属性。"""
repo = KlineRepository.__new__(KlineRepository)
repo._enriched_history_cache = None
repo._enriched_warming = True
return repo
def test_get_enriched_range_returns_none_while_warming():
repo = _bare_repo()
refresh_calls: list[int] = []
def _spy_refresh():
refresh_calls.append(1)
repo._refresh_enriched = _spy_refresh # type: ignore[method-assign]
result = repo.get_enriched_range(date(2026, 1, 1), date(2026, 8, 14))
assert result is None
assert refresh_calls == [], "预热期间不得触发 _refresh_enriched"
def test_get_enriched_range_rebuilds_when_cold_and_not_warming():
repo = _bare_repo()
repo._enriched_warming = False
built = pl.DataFrame({
"symbol": ["600000.SH", "600000.SH"],
"date": [date(2026, 1, 1), date(2026, 8, 14)],
})
def _fake_refresh():
repo._enriched_history_cache = built
repo._refresh_enriched = _fake_refresh # type: ignore[method-assign]
result = repo.get_enriched_range(date(2026, 1, 1), date(2026, 8, 14))
assert result is not None
assert result.height == 2
assert result["symbol"].unique().to_list() == ["600000.SH"]