124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
|
"""consumption_log.txt 兼容 TSV 格式。"""
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
|
|||
|
|
from app.config import settings
|
|||
|
|
from app.services.consumable_vision_algorithm import ClsTop3
|
|||
|
|
from app.services.consumption_tsv_log import (
|
|||
|
|
HEADER,
|
|||
|
|
_RANGE_SEP,
|
|||
|
|
append_consumption_tsv_line,
|
|||
|
|
build_consumption_markdown,
|
|||
|
|
build_tsv_line,
|
|||
|
|
init_consumption_log_file,
|
|||
|
|
short_camera_label,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_short_camera_label() -> None:
|
|||
|
|
assert short_camera_label("or-cam-01") == "cam01"
|
|||
|
|
assert short_camera_label("or-cam-2") == "cam02"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_build_tsv_line_matches_sample_shape(monkeypatch: pytest.MonkeyPatch) -> None:
|
|||
|
|
monkeypatch.setattr(settings, "consumption_log_timezone", "UTC")
|
|||
|
|
best = ClsTop3(
|
|||
|
|
t1_name="一次性医用灭菌棉签",
|
|||
|
|
t1_conf=0.9997,
|
|||
|
|
t2_name="cls2",
|
|||
|
|
t2_conf=0.0003,
|
|||
|
|
t3_name="cls3",
|
|||
|
|
t3_conf=0.0002,
|
|||
|
|
t1_pid="2237844",
|
|||
|
|
t2_pid="11765-1-101",
|
|||
|
|
t3_pid="21504-1-1",
|
|||
|
|
)
|
|||
|
|
# 墙钟：拉流起点对齐到 2024-01-01T00:00:00Z，时间窗 +0s…+45s
|
|||
|
|
w0 = 1704067200.0
|
|||
|
|
line = build_tsv_line(
|
|||
|
|
name_to_code={},
|
|||
|
|
best=best,
|
|||
|
|
doctor_id="DOCTOR_PLACEHOLDER",
|
|||
|
|
camera_id="or-cam-01",
|
|||
|
|
wall_start_epoch=w0,
|
|||
|
|
wall_end_epoch=w0 + 45.0,
|
|||
|
|
)
|
|||
|
|
parts = line.rstrip("\n").split("\t")
|
|||
|
|
assert len(parts) == 7
|
|||
|
|
assert parts[0] == "2237844"
|
|||
|
|
assert parts[1] == "一次性医用灭菌棉签 0.9997"
|
|||
|
|
assert parts[2] == "cls2"
|
|||
|
|
assert parts[3] == "cls3"
|
|||
|
|
assert parts[4] == "1"
|
|||
|
|
assert parts[5] == "DOCTOR_PLACEHOLDER"
|
|||
|
|
assert (
|
|||
|
|
parts[6]
|
|||
|
|
== "cam01@2024-01-01T00:00:00.000+00:00"
|
|||
|
|
+ _RANGE_SEP
|
|||
|
|
+ "2024-01-01T00:00:45.000+00:00"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_header_columns() -> None:
|
|||
|
|
cols = HEADER.strip().split("\t")
|
|||
|
|
assert cols[0] == "物品id"
|
|||
|
|
assert cols[-1] == "时间戳"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_per_surgery_file_init_and_append(
|
|||
|
|
tmp_path,
|
|||
|
|
monkeypatch: pytest.MonkeyPatch,
|
|||
|
|
) -> None:
|
|||
|
|
monkeypatch.setattr(settings, "consumption_tsv_log_enabled", True)
|
|||
|
|
monkeypatch.setattr(
|
|||
|
|
settings,
|
|||
|
|
"consumption_tsv_log_path",
|
|||
|
|
str(tmp_path / "{surgery_id}.txt"),
|
|||
|
|
)
|
|||
|
|
init_consumption_log_file("or-001")
|
|||
|
|
append_consumption_tsv_line("or-001", "row1\n")
|
|||
|
|
append_consumption_tsv_line("or-001", "row2\n")
|
|||
|
|
p = tmp_path / "or-001.txt"
|
|||
|
|
assert p.read_text(encoding="utf-8") == HEADER + "row1\n" + "row2\n"
|
|||
|
|
init_consumption_log_file("or-001")
|
|||
|
|
assert p.read_text(encoding="utf-8") == HEADER
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_build_consumption_markdown_top123_columns(monkeypatch: pytest.MonkeyPatch) -> None:
|
|||
|
|
monkeypatch.setattr(settings, "consumption_log_timezone", "UTC")
|
|||
|
|
best = ClsTop3(
|
|||
|
|
t1_name="一次性医用灭菌棉签",
|
|||
|
|
t1_conf=0.9997,
|
|||
|
|
t2_name="cls2",
|
|||
|
|
t2_conf=0.0003,
|
|||
|
|
t3_name="cls3",
|
|||
|
|
t3_conf=0.0002,
|
|||
|
|
t1_pid="2237844",
|
|||
|
|
t2_pid="11765-1-101",
|
|||
|
|
t3_pid="21504-1-1",
|
|||
|
|
)
|
|||
|
|
w0 = 1704067200.0
|
|||
|
|
md = build_consumption_markdown(
|
|||
|
|
name_to_code={},
|
|||
|
|
best=best,
|
|||
|
|
doctor_id="DOCTOR_PLACEHOLDER",
|
|||
|
|
camera_id="or-cam-01",
|
|||
|
|
wall_start_epoch=w0,
|
|||
|
|
wall_end_epoch=w0 + 45.0,
|
|||
|
|
)
|
|||
|
|
assert "Top1 物品id" in md and "Top1 物品名称" in md and "Top1 置信度" in md
|
|||
|
|
assert "Top2 物品名称" in md and "Top3 物品名称" in md
|
|||
|
|
assert "Top2 物品id" not in md
|
|||
|
|
assert "2237844" in md
|
|||
|
|
assert "一次性医用灭菌棉签" in md
|
|||
|
|
assert "0.9997" in md
|
|||
|
|
assert "cls2" in md and "cls3" in md
|
|||
|
|
assert "11765-1-101" not in md and "21504-1-1" not in md
|
|||
|
|
assert "0.0003" not in md and "0.0002" not in md
|
|||
|
|
assert "DOCTOR_PLACEHOLDER" in md
|
|||
|
|
assert "| 1 |" in md
|
|||
|
|
# 终端为可读时间戳，非落盘用 ISO@cam
|
|||
|
|
assert "2024-01-01 00:00:00.000" in md and "2024-01-01 00:00:45.000" in md
|
|||
|
|
assert "cam01" in md and " · " in md and _RANGE_SEP in md
|
|||
|
|
assert "cam01@2024-01" not in md
|