simon
366e60e8a9
- 新增 report_db 包: MySQL 连接/建表/幂等写入 (news_report/news_event, myquant 库) - 新增 report_import 包: 历史 178 份日报 HTML 解析入库, 表头驱动列映射 - reporter.py 完全切换: generate_report 结构化入库, 不再生成/上传 HTML - CLI: 新增 report-import 子命令 - 依赖: uv add pymysql; 配置: NEWS_DB_* / REPORT_HISTORY_DIR - 文档: docs/report_db_design.md(实现逻辑), docs/db_schema.md(表结构供 API/前端) - 测试: 24 个单测通过 (parser/builder/models/importer)
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
"""日报数据模型单元测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import date, datetime
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from report_db.models import EventRow, ReportData
|
|
|
|
|
|
class TestEventRow:
|
|
def test_minimal(self) -> None:
|
|
ev = EventRow(section="news", rank=1, title="标题")
|
|
assert ev.importance is None
|
|
assert ev.sentiment is None
|
|
|
|
def test_full(self) -> None:
|
|
ev = EventRow(
|
|
section="intl", rank=2, importance=4, event_type="地缘政治",
|
|
title="t", summary="s", sentiment="negative", source="ForexLive",
|
|
url="https://x.com/1",
|
|
)
|
|
assert ev.sentiment == "negative"
|
|
|
|
def test_missing_title_raises(self) -> None:
|
|
with pytest.raises(ValidationError):
|
|
EventRow(section="news", rank=1) # type: ignore[call-arg]
|
|
|
|
|
|
class TestReportData:
|
|
def test_defaults(self) -> None:
|
|
r = ReportData(
|
|
report_date=date(2026, 7, 11),
|
|
report_type="finance",
|
|
generated_at=datetime(2026, 7, 11, 7, 0),
|
|
)
|
|
assert r.file_name == ""
|
|
assert r.stats == {}
|
|
assert r.events == []
|
|
|
|
def test_with_events(self) -> None:
|
|
r = ReportData(
|
|
report_date=date(2026, 7, 11),
|
|
report_type="finance",
|
|
generated_at=datetime(2026, 7, 11, 7, 0),
|
|
ai_summary="摘要",
|
|
events=[EventRow(section="xwlb", rank=1, title="t")],
|
|
)
|
|
assert len(r.events) == 1
|