tests/test_render.py

from __future__ import annotations

from hydradeck.core.types import ExtractedFact, Source
from hydradeck.render import (
    build_slide_frames_from_sections,
    build_slide_frames_from_report,
    enforce_slide_density,
    render_beamer_frames,
    render_paper,
    render_report_structured,
)


def test_render_paper_converts_markdown_like_body() -> None:
    sources = [Source(url="https://example.com", title="Example", snippet="snippet")]
    facts = [
        ExtractedFact(
            claim="Claim A",
            evidence="Evidence A",
            url="https://example.com",
            title="Example",
        )
    ]
    body = """## Heading
- bullet 1
- bullet 2
`inline`
```python
print('x')
```
"""

    tex = render_paper(
        topic="demo",
        outline=["背景", "创新"],
        body=body,
        facts=facts,
        sources=sources,
    )

    assert "```" not in tex
    assert "## Heading" not in tex
    assert "Heading" in tex
    assert "\\begin{itemize}" in tex
    assert "bullet 1" in tex
    assert "\\section*{1. Introduction and Background}" in tex


def test_render_templates_use_facts_not_generic_filler() -> None:
    sources = [Source(url="https://example.com", title="Example", snippet="snippet")]
    facts = [
        ExtractedFact(
            claim="RynnBrain released checkpoints on 2026-02-09",
            evidence="project timeline from official repo",
            url="https://example.com",
            title="Example",
        ),
        ExtractedFact(
            claim="Model introduces interleaved reasoning with spatial grounding",
            evidence="technical report description",
            url="https://example.com",
            title="Example",
        ),
    ]
    paper = render_paper(
        topic="demo",
        outline=["背景", "创新", "架构"],
        body="结论段落 [1]",
        facts=facts,
        sources=sources,
    )
    section_blocks = [
        {"name": "背景", "latex": facts[0].claim},
        {"name": "创新", "latex": facts[1].claim},
    ]
    frames = build_slide_frames_from_sections(section_blocks, language="en")
    slides = render_beamer_frames("demo", frames, language="en")

    assert "released checkpoints" in paper
    assert "interleaved reasoning" in paper
    assert "released checkpoints" in slides
    assert "interleaved reasoning" in slides
    section_one = paper.split("\\section*{1. Introduction and Background}", 1)[1]
    section_one = section_one.split("\\section*{2. Logical Outline}", 1)[0]
    assert "\\begin{itemize}" not in section_one


def test_render_beamer_from_report_derives_outline() -> None:
    paper = (
        "\\documentclass{article}\n"
        "\\begin{document}\n"
        "\\section*{Executive Summary}\nAlpha beta gamma.\n\n"
        "\\section*{Methodology}\nMethod details.\n\n"
        "\\section*{Results}\nResult details.\n"
        "\\end{document}\n"
    )
    frames = build_slide_frames_from_report(paper, language="en")
    slides = render_beamer_frames("demo", frames, language="en")
    assert "\\begin{frame}{Agenda}" in slides
    assert "Executive Summary" in slides
    assert "Methodology" in slides


def test_render_beamer_frames_limits_density() -> None:
    report = (
        "\\documentclass{article}\\begin{document}"
        "\\section*{Overview} A long sentence about architecture and implementation details repeated."
        " Another long sentence about evaluation metrics and reproducibility details."
        "\\section*{Results} Multiple findings with evidence and quantitative metrics."
        "\\end{document}"
    )
    frames = build_slide_frames_from_report(report, language="en")
    assert len(frames) >= 2
    tex = render_beamer_frames("demo", frames, language="en")
    assert "\\begin{frame}{Agenda}" in tex
    assert "\\begin{itemize}" in tex


def test_render_report_structured_zh_uses_ctex() -> None:
    section_blocks = [
        {"name": "方法", "latex": "本节给出方法细节与参数说明。"},
        {"name": "结果", "latex": "本节给出结果与证据。"},
    ]
    tex = render_report_structured("中文研究报告", section_blocks, language="zh")
    assert "\\documentclass[11pt]{ctexart}" in tex
    assert "本研究报告聚焦可追溯证据" not in tex
    assert "建议定期刷新证据并进行复跑验证" not in tex
    assert "\\section*{方法}" in tex


def test_render_beamer_frames_zh_uses_ctexbeamer() -> None:
    frames = [
        build_slide_frames_from_report(
            "\\documentclass{ctexart}\\begin{document}\\section*{结果}关键结果一。关键结果二。\\end{document}",
            language="zh",
        )[0]
    ]
    tex = render_beamer_frames("中文主题", frames, language="zh")
    assert "\\documentclass[aspectratio=169]{ctexbeamer}" in tex
    assert "\\begin{frame}{目录}" in tex


def test_build_slide_frames_from_sections_splits_long_section() -> None:
    section_blocks = [
        {
            "name": "Results",
            "latex": (
                "The first finding shows strong improvement in consistency and precision. "
                "The second finding shows stronger robustness under distribution shift. "
                "The third finding indicates cost-performance improvement. "
                "The fourth finding confirms stability across runs. "
                "The fifth finding highlights limitations and guardrails."
            ),
        }
    ]
    frames = build_slide_frames_from_sections(section_blocks, language="en")
    assert len(frames) >= 2


def test_render_report_structured_removes_bracket_refs() -> None:
    section_blocks = [
        {"name": "Evidence", "latex": "Claim [1] with support [2] and \\cite{src1}."}
    ]
    tex = render_report_structured("demo", section_blocks, language="en")
    assert "[1]" not in tex
    assert "[2]" not in tex
    assert "\\cite{" not in tex


def test_enforce_slide_density_splits_large_bullet_groups() -> None:
    frames = [
        {
            "title": "Results",
            "bullets": [
                "point one with enough words to be valid",
                "point two with enough words to be valid",
                "point three with enough words to be valid",
                "point four with enough words to be valid",
                "point five with enough words to be valid",
            ],
        }
    ]
    from hydradeck.render import SlideFrame

    fr = [SlideFrame(title=x["title"], bullets=x["bullets"]) for x in frames]
    out = enforce_slide_density(fr, language="en", max_bullets_per_frame=4)
    assert len(out) == 2
    assert out[0].title == "Results"
    assert "(cont.)" in out[1].title