mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
830de11c66
* fix(uploads): exclude indented bold lines from outlines * refactor(uploads): name the indented-code pattern
52 lines
2.4 KiB
Python
52 lines
2.4 KiB
Python
"""Indented bold examples must not displace document headings in outlines."""
|
|
|
|
import pytest
|
|
|
|
from deerflow.uploads.companions import register_companion
|
|
from deerflow.utils.file_outline import MAX_OUTLINE_ENTRIES, extract_outline, extract_outline_for_file
|
|
|
|
|
|
@pytest.mark.parametrize("indent", [" ", "\t", " \t", " \t", " \t"])
|
|
@pytest.mark.parametrize("text", ["**PART II**", "**1** **Code example**"])
|
|
def test_indented_bold_code_preserves_real_heading_lines(tmp_path, indent, text):
|
|
path = tmp_path / "guide.md"
|
|
path.write_text(f"# Overview\n\n{indent}{text}\n\n## Findings\n", encoding="utf-8")
|
|
|
|
assert extract_outline(path) == [{"title": "Overview", "line": 1}, {"title": "Findings", "line": 5}]
|
|
|
|
|
|
@pytest.mark.parametrize("indent", [" ", "\t"])
|
|
@pytest.mark.parametrize("text", ["**PART II**", "**1** **Code example**"])
|
|
def test_indented_bold_examples_cannot_exhaust_outline_slots(tmp_path, indent, text):
|
|
path = tmp_path / "guide.md"
|
|
path.write_text((indent + text + "\n") * (MAX_OUTLINE_ENTRIES + 1) + "\n# Actual findings\n", encoding="utf-8")
|
|
|
|
assert extract_outline(path) == [{"title": "Actual findings", "line": MAX_OUTLINE_ENTRIES + 3}]
|
|
|
|
|
|
@pytest.mark.parametrize("indent", ["", " ", " ", " "])
|
|
def test_bold_headings_with_up_to_three_spaces_are_preserved(tmp_path, indent):
|
|
path = tmp_path / "report.md"
|
|
path.write_text(f"{indent}**PART II**\n{indent}**1** **概述**\n", encoding="utf-8")
|
|
|
|
assert extract_outline(path) == [{"title": "PART II", "line": 1}, {"title": "1 概述", "line": 2}]
|
|
|
|
|
|
def test_owned_conversion_uses_the_same_indentation_rule(tmp_path):
|
|
uploads = tmp_path / "thread/user-data/uploads"
|
|
uploads.mkdir(parents=True)
|
|
original = uploads / "report.pdf"
|
|
markdown = uploads / "report.md"
|
|
original.write_bytes(b"synthetic source fixture; converter not invoked")
|
|
markdown.write_text(" **PART II**\n\n# Actual section\n", encoding="utf-8")
|
|
register_companion(original, markdown)
|
|
|
|
assert extract_outline_for_file(original) == ([{"title": "Actual section", "line": 3}], [])
|
|
|
|
|
|
def test_existing_fenced_code_and_heading_styles_are_preserved(tmp_path):
|
|
path = tmp_path / "guide.md"
|
|
path.write_text("~~~\n**PART II**\n~~~\n# Section\n**PART III**\n**2** **Details**\n", encoding="utf-8")
|
|
|
|
assert extract_outline(path) == [{"title": "Section", "line": 4}, {"title": "PART III", "line": 5}, {"title": "2 Details", "line": 6}]
|