simon
65ead54b4f
- 删除 5 个过时/残留文档(project_plan/agent_prompt/optimization_plan/report_db_design/deploy/README) - 新建 docs/architecture.md(项目架构:11 包职责+数据模型+配置+产物) - 重写 docs/user-guide.md(CLI 全量+增量/断点续跑+MCP+FAQ) - 重写 README.md(精简入口+文档索引) - 更新 continuation.md(追加本次记录) - 更新 .gitignore(排除 data/* 运行产物)
59 lines
1.8 KiB
Python
59 lines
1.8 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",
|
|
sources=["ForexLive", "新浪财经"],
|
|
url="https://x.com/1",
|
|
)
|
|
assert ev.sentiment == "negative"
|
|
assert ev.sources == ["ForexLive", "新浪财经"]
|
|
|
|
def test_sources_optional(self) -> None:
|
|
"""sources 为可选项(旧数据无多源记录)。"""
|
|
ev = EventRow(section="news", rank=1, title="t")
|
|
assert ev.sources is None
|
|
|
|
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
|