重组为 backend/clients/docs 三层结构,并清理 git 污染。
将后端迁入 backend/,完善根目录 .gitignore,删除误提交的 .mypy_cache 缓存文件。 Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
70
backend/tests/test_effective_candidate_consumables.py
Normal file
70
backend/tests/test_effective_candidate_consumables.py
Normal file
@@ -0,0 +1,70 @@
|
||||
"""effective_candidate_consumables / build_name_mapping:仅 YAML,无分类模型回退。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from app.consumable_catalog import (
|
||||
build_name_mapping,
|
||||
effective_candidate_consumables,
|
||||
normalize_candidate_consumables_raw,
|
||||
)
|
||||
|
||||
|
||||
def test_normalize_candidate_consumables_raw_strings_and_export_objects() -> None:
|
||||
assert normalize_candidate_consumables_raw([" a ", "b"]) == ["a", "b"]
|
||||
raw = [
|
||||
{"消耗品编号": "14764-2-4", "名称": "一次性使用手术单"},
|
||||
{"name": "缝线"},
|
||||
]
|
||||
assert normalize_candidate_consumables_raw(raw) == ["一次性使用手术单", "缝线"]
|
||||
assert normalize_candidate_consumables_raw([{}]) == []
|
||||
assert normalize_candidate_consumables_raw("not-a-list") == []
|
||||
|
||||
|
||||
def test_effective_preserves_non_empty_request() -> None:
|
||||
got = effective_candidate_consumables([" 纱布 ", "缝线", "纱布"])
|
||||
assert got == ["纱布", "缝线"]
|
||||
|
||||
|
||||
def test_effective_empty_yaml_no_names_returns_empty(tmp_path: Path) -> None:
|
||||
yml = tmp_path / "empty.yaml"
|
||||
yml.write_text("names: {}\nlabel_id: {}\n", encoding="utf-8")
|
||||
assert effective_candidate_consumables([], labels_yaml_path=yml) == []
|
||||
|
||||
|
||||
def test_effective_empty_prefers_yaml_class_names(tmp_path: Path) -> None:
|
||||
yml = tmp_path / "lab.yaml"
|
||||
yml.write_text(
|
||||
"names:\n 0: 商品甲\n 1: 商品乙\nlabel_id:\n 0: a\n 1: b\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
assert effective_candidate_consumables([], labels_yaml_path=yml) == ["商品甲", "商品乙"]
|
||||
|
||||
|
||||
def test_effective_whitespace_only_requests_empty_yaml_returns_empty(tmp_path: Path) -> None:
|
||||
yml = tmp_path / "empty.yaml"
|
||||
yml.write_text("names: {}\nlabel_id: {}\n", encoding="utf-8")
|
||||
assert effective_candidate_consumables(["", " "], labels_yaml_path=yml) == []
|
||||
|
||||
|
||||
def test_build_name_mapping_from_label_id(tmp_path: Path) -> None:
|
||||
yml = tmp_path / "lab.yaml"
|
||||
yml.write_text(
|
||||
"names:\n 0: 商品A\nlabel_id:\n 0: y1/y2\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
m = build_name_mapping(["商品A"], labels_yaml_path=yml)
|
||||
assert m["商品A"] == "y1/y2"
|
||||
|
||||
|
||||
def test_build_name_mapping_uses_name_when_no_id_in_yaml(tmp_path: Path) -> None:
|
||||
yml = tmp_path / "lab.yaml"
|
||||
yml.write_text(
|
||||
"names:\n 0: 仅表内有的\nlabel_id: {}\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
m = build_name_mapping(["仅表内有的"], labels_yaml_path=yml)
|
||||
assert m["仅表内有的"] == "仅表内有的"
|
||||
Reference in New Issue
Block a user