mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 17:54:15 +08:00
shy3130
697c27bb02
- 市场环境: 新增情绪周期6阶段(冰点/启动/主升/高潮/退潮/修复, 连板梯队驱动, EMA平滑+2日确认+弱档否决, 平均段长9.7天)与概念/行业主线排名(涨停梯队聚合, 可配置宽基/风格标签过滤); 市场环境页重构, regime 透明加列, 与5档state并存 - 挖掘: 因子与策略挖掘全链路(API/worker/进程锁/候选库/前端工作台/文档), 周度调度默认关闭且永不自动发布 - 回测: 财务快照因子(点时口径), 批量回测预计算共享下期收益, 信号路径矩阵列依赖展开修复(consecutive_limit_ups 缺列报错) - 数据/性能: enriched 生成与预热治理, 重任务限流, 行情/K线缓存复用, 时区修复 - 测试: 后端全量 914 通过; GUI 黑盒验证截图存证 gui-test-screenshots/
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
"""storage 统计求和不变量测试 — total_size_mb 必须等于各部分之和, 不得重复累加。
|
|
|
|
历史 bug: other_dirs 循环被复制了两遍, 且 financials 既在 other_dirs 里又有专属
|
|
统计块 —— financials 被计入 3 次、pools/backtest_results/screener_results/ai_cache
|
|
各计入 2 次, total_size_mb 虚高。此测试用精确整 MB 文件锁定总和不变量。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from app.api.data import _compute_storage
|
|
|
|
MB = 1024 * 1024
|
|
|
|
|
|
def _write_mb(path: Path, mb: float) -> None:
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_bytes(b"\0" * int(mb * MB))
|
|
|
|
|
|
def _build_tree(data_dir: Path) -> None:
|
|
_write_mb(data_dir / "kline_daily" / "part.parquet", 1.0)
|
|
_write_mb(data_dir / "financials" / "metrics" / "part.parquet", 2.0)
|
|
_write_mb(data_dir / "pools" / "p.json", 1.0)
|
|
_write_mb(data_dir / "backtest_results" / "r.json", 1.0)
|
|
_write_mb(data_dir / "screener_results" / "s.json", 1.0)
|
|
_write_mb(data_dir / "ai_cache" / "c.json", 1.0)
|
|
_write_mb(data_dir / "capabilities.json", 0.5) # 根目录散文件
|
|
|
|
|
|
def test_total_equals_sum_of_parts(tmp_path):
|
|
_build_tree(tmp_path)
|
|
stats = _compute_storage(tmp_path)
|
|
|
|
parts_mb = (
|
|
stats["daily_size_mb"] # 1.0 (subdirs 表)
|
|
+ stats["financials_size_mb"] # 2.0 (专属统计块)
|
|
+ 1.0 + 1.0 + 1.0 + 1.0 # pools/backtest/screener/ai_cache (单次循环)
|
|
+ 0.5 # 根目录散文件
|
|
)
|
|
assert stats["total_size_mb"] == parts_mb == 7.5
|
|
|
|
|
|
def test_financials_counted_once_not_three_times(tmp_path):
|
|
_build_tree(tmp_path)
|
|
stats = _compute_storage(tmp_path)
|
|
|
|
assert stats["financials_files"] == 1
|
|
assert stats["financials_size_mb"] == 2.0
|
|
# 旧 bug: financials 计 3 次 (6MB) + 其余 4 目录各计 2 次 (8MB) → 15.5
|
|
assert stats["total_size_mb"] != 15.5
|
|
|
|
|
|
def test_missing_dirs_contribute_zero(tmp_path):
|
|
_write_mb(tmp_path / "kline_daily" / "part.parquet", 1.0)
|
|
stats = _compute_storage(tmp_path)
|
|
|
|
# financials 目录不存在时不含其明细键 (既有行为), 且不贡献任何体积
|
|
assert stats.get("financials_files", 0) == 0
|
|
assert stats.get("financials_size_mb", 0.0) == 0.0
|
|
assert stats["total_size_mb"] == 1.0
|