62 lines
2.2 KiB
Python
62 lines
2.2 KiB
Python
"""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
|