"""visualize_pipeline TSV 解析单元测试（无需 GPU）。"""
from __future__ import annotations

import sys
from pathlib import Path

PACK_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(PACK_ROOT / "scripts"))

from visualize_tsv import parse_result_tsv  # noqa: E402


def test_parse_offline_12col_with_doctor_summary(tmp_path: Path) -> None:
    tsv = tmp_path / "r.txt"
    tsv.write_text(
        "rank\tstart_sec\tend_sec\tproduct_id_top1\ttop1_name\ttop1_conf\t"
        "product_id_top2\ttop2_name\ttop2_conf\tproduct_id_top3\ttop3_name\ttop3_conf\n"
        "1\t1.0\t5.0\tP1\t手套\t0.9\t\t\t\t\t\t\n"
        "医生信息：张三 (id=D01, conf=0.91)\n",
        encoding="utf-8",
    )
    segs, doc = parse_result_tsv(tsv)
    assert len(segs) == 1
    assert segs[0].row.n1 == "手套"
    assert doc is not None and "张三" in doc


def test_parse_stream_15col(tmp_path: Path) -> None:
    tsv = tmp_path / "s.txt"
    header = "\t".join(
        [
            "rank", "start_sec", "end_sec", "product_id_top1", "top1_name", "top1_conf",
            "product_id_top2", "top2_name", "top2_conf", "product_id_top3", "top3_name", "top3_conf",
            "doctor_id", "doctor_name", "doctor_conf",
        ]
    )
    row = [
        "1", "2.0", "8.0", "",
        "（无有效耗材帧：好帧/白名单/耗材置信度未全部满足）", "",
        "", "", "", "", "", "", "D2", "李四", "0.77",
    ]
    tsv.write_text(header + "\n" + "\t".join(row) + "\n", encoding="utf-8")
    segs, doc = parse_result_tsv(tsv)
    assert len(segs) == 1
    assert segs[0].is_failure()
    assert segs[0].doctor_name == "李四"
    assert doc is None


def test_parse_failure_hud_text(tmp_path: Path) -> None:
    tsv = tmp_path / "f.txt"
    tsv.write_text(
        "rank\tstart_sec\tend_sec\tproduct_id_top1\ttop1_name\ttop1_conf\t"
        "product_id_top2\ttop2_name\ttop2_conf\tproduct_id_top3\ttop3_name\ttop3_conf\n"
        "1\t0.5\t3.0\t\t（无有效耗材帧：好帧/白名单/耗材置信度未全部满足）\t\t"
        "\t\t\t\t\t\t\n",
        encoding="utf-8",
    )
    segs, _ = parse_result_tsv(tsv)
    assert segs[0].is_failure()
    assert "无有效耗材帧" in segs[0].row.n1