mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 05:06:59 +00:00
d288935865
* fix(skills): preserve Markdown table cells in data-analysis exports * docs(skills): document Markdown cell escaping contract Clarify handling of Markdown exports and escaping rules. * docs: describe data-analysis Markdown export escaping Clarify data-analysis skill features and behavior in README. * test(skills): cover Markdown export headers and multiline cells Add a test for markdown export functionality in data analysis.
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""Markdown exports preserve the shape of real DuckDB query results."""
|
|
|
|
import runpy
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
duckdb = pytest.importorskip("duckdb")
|
|
pytest.importorskip("openpyxl")
|
|
SCRIPT = (
|
|
Path(__file__).resolve().parents[2]
|
|
/ "skills/public/data-analysis/scripts/analyze.py"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"header, value, expected_header, expected_value",
|
|
[
|
|
("left|right", "a|b", r"left\|right", r"a\|b"),
|
|
("notes", "first\nsecond", "notes", "first<br>second"),
|
|
("first\r\nsecond", "a\rb\r\nc", "first<br>second", "a<br>b<br>c"),
|
|
("path", r"a\|b", "path", r"a\\\|b"),
|
|
("数量", 12, "数量", "12"),
|
|
],
|
|
)
|
|
def test_query_markdown_export_keeps_each_cell_on_one_line(
|
|
tmp_path, header, value, expected_header, expected_value
|
|
):
|
|
analysis = runpy.run_path(str(SCRIPT))
|
|
output = tmp_path / "result.md"
|
|
with duckdb.connect() as con:
|
|
con.execute('CREATE TABLE input ("' + header.replace('"', '""') + '" VARCHAR)')
|
|
con.execute("INSERT INTO input VALUES (?)", [value])
|
|
analysis["action_query"](con, "SELECT * FROM input", {}, str(output))
|
|
assert output.read_text(encoding="utf-8").splitlines() == [
|
|
f"| {expected_header} |",
|
|
"| --- |",
|
|
f"| {expected_value} |",
|
|
]
|