Files
ashare-data/tests/test_code_mapping.py
T
曾志威 dbc8107caa P0/P1/P2 一次性落地:补齐文档、修复 main 挂载/笔误、引入 logging+tests
P0:
- 清理 src/fetchers/sector.py 第 225 行起的旧版残留代码
- 修复 src/fetchers/market_daily.py 中 fetch_history -> _fetch_history 笔误
- 在 src/main.py 挂载 --market-daily 子命令

P1:
- 修正 src/db.py docstring(market_breadth -> market_daily 等)
- requirements.txt 补 baostock;pyproject.toml 同步 + 新增 [dev] extras
- README 增加 config.yaml 安全提示,将 git 历史清理升级为高风险 P0 由用户决策

P2:
- 引入 src/log.py 统一 logging(控制台 + logs/ashare.log 按日滚动 7 天)
- 建立 tests/ 框架,4 个测试文件 / 19 个 pytest 用例全部通过
- pyproject.toml 新增 [tool.pytest.ini_options]
- config.example.yaml 补全 workers 字段与多源说明
- README 更新功能概览/参数说明/表结构/项目结构/设计说明全章节
- TODO.md 全面重写,按 P0/P2 整理剩余条目并附变更日志
- .gitignore 新增 .claude/
2026-05-15 13:26:21 +08:00

31 lines
826 B
Python

"""验证 code_to_bs 的代码前缀映射规则"""
from src.baostock_conn import code_to_bs
def test_shanghai_main_board():
assert code_to_bs("600000") == "sh.600000"
assert code_to_bs("601318") == "sh.601318"
def test_shanghai_kechuang():
# 科创板 688 仍以 6 开头
assert code_to_bs("688981") == "sh.688981"
def test_shenzhen_main_and_chinext():
assert code_to_bs("000001") == "sz.000001"
assert code_to_bs("002594") == "sz.002594"
assert code_to_bs("300750") == "sz.300750"
def test_beijing_returns_none():
# BaoStock 不含北交所,应显式返回 None
assert code_to_bs("920001") is None
assert code_to_bs("920999") is None
def test_warrant_or_b_share_prefix_9():
# 沪市 B 股以 9 开头,与 6 同走 sh.
assert code_to_bs("900901") == "sh.900901"