mirror of
https://github.com/simonlin1212/TradingAgents-astock.git
synced 2026-10-30 09:29:56 +00:00
waterblueandCodex
7f4431fdb1
Prefer PDF-compatible CJK fonts, reuse single-face fonts for bold fallback, and render Chinese text with compact spacing and character wrapping. 改进中文 PDF 导出：优先使用更稳的 CJK 字体，单字体 fallback 复用粗体，并使用紧凑空白与字符级换行改善排版。 Co-authored-by: Codex <codex@example.com>
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
"""Regression tests for PDF export."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from fpdf.enums import WrapMode
|
|
|
|
from web.pdf_export import (
|
|
PDFExportError,
|
|
_ReportPDF,
|
|
_compact_inline_text,
|
|
_discover_cjk_fonts,
|
|
_find_cjk_fonts,
|
|
_format_table_cells,
|
|
generate_pdf,
|
|
)
|
|
|
|
|
|
def test_generate_pdf_with_chinese_markdown_when_cjk_font_available():
|
|
try:
|
|
_find_cjk_fonts()
|
|
except PDFExportError as exc:
|
|
pytest.skip(str(exc))
|
|
|
|
state = {
|
|
"market_report": "# 技术分析\n- 趋势: 偏强\n- 结论: 中文字体测试通过。",
|
|
"news_report": "| 项目 | 结论 |\n|---|---|\n| 中文 | 可嵌入 |",
|
|
"final_trade_decision": "最终建议: HOLD 观望。",
|
|
}
|
|
|
|
pdf_bytes = generate_pdf(state, "600519", "2026-05-30", "HOLD")
|
|
|
|
assert pdf_bytes.startswith(b"%PDF-")
|
|
assert len(pdf_bytes) > 1000
|
|
|
|
|
|
def test_pdf_text_helpers_compact_alignment_whitespace():
|
|
assert _compact_inline_text("Neutral Analyst: 我会把") == "Neutral Analyst: 我会把"
|
|
assert _format_table_cells(["表面看是放量反弹", "但对连续跌停后的", "ST"]) == "表面看是放量反弹 | 但对连续跌停后的 | ST"
|
|
|
|
|
|
def test_pdf_multicell_defaults_to_left_alignment():
|
|
pdf = object.__new__(_ReportPDF)
|
|
calls = []
|
|
|
|
pdf.l_margin = 10
|
|
pdf.r_margin = 10
|
|
pdf.w = 210
|
|
pdf.set_x = lambda x: None
|
|
pdf.multi_cell = lambda *args, **kwargs: calls.append((args, kwargs))
|
|
|
|
pdf._write_multicell(5.5, "近 5 日均量明显高于近 20 日均量。")
|
|
|
|
assert calls[0][1]["align"] == "L"
|
|
assert calls[0][1]["wrapmode"] == WrapMode.CHAR
|
|
|
|
|
|
def test_wqy_discovery_reuses_same_font_for_bold(monkeypatch, tmp_path):
|
|
wqy = tmp_path / "wqy-microhei.ttc"
|
|
noto_bold = tmp_path / "NotoSansCJK-Bold.ttc"
|
|
wqy.write_bytes(b"font")
|
|
noto_bold.write_bytes(b"font")
|
|
|
|
def fake_find_font_file(pattern: str) -> Path | None:
|
|
return {
|
|
"wqy-microhei.ttc": wqy,
|
|
"NotoSansCJK-Bold.ttc": noto_bold,
|
|
}.get(pattern)
|
|
|
|
monkeypatch.setattr("web.pdf_export._find_font_file", fake_find_font_file)
|
|
|
|
assert _discover_cjk_fonts() == (wqy, wqy)
|