File size: 2,713 Bytes
0e6887b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | """DocumentProfiler 测试（intent-understanding-layer 任务 4）。
覆盖需求 2.1–2.4 / 2.6：表类型识别、稳定性时序标注、多文档聚合、保留时间防护。
"""
from __future__ import annotations
from kernel.task_sheet import TableType
from kernel.understanding import DocumentProfiler
class _MockLLM:
def __init__(self, out):
self._out = out
def profile_and_extract(self, text, goal):
return self._out
SL0010_TEXT = """=== File: SL-0010-3.xlsx ===
有关物质检测结果
样品名称 名称 保留时间min RRT 总杂%
SL-0010-20μg 右美托咪定 25.440 1.00 0.00
含量均匀度
含量%
99.19
"""
def test_heuristic_profile_no_llm_sl0010():
profiler = DocumentProfiler(llm=None)
profile = profiler.profile(SL0010_TEXT, ["SL-0010-3.xlsx"])
types = {t.table_type for t in profile.tables}
assert TableType.RELATED_SUBSTANCES in types
assert TableType.CONTENT_UNIFORMITY in types
assert profile.has_stability_time_series is False
assert any("未检测到稳定性" in n for n in profile.notes)
def test_llm_profile_respects_retention_guard():
# LLM 错误地把含保留时间列的表标为稳定性时序 → 防护必须强制改回 False。
llm = _MockLLM({
"tables": [{
"table_type": "related_substances",
"title": "有关物质",
"source_document": "SL-0010-3.xlsx",
"is_stability_time_series": True,
"columns": ["保留时间min", "总杂%"],
}],
"intent": "descriptive_summary",
"items": [],
})
profile = DocumentProfiler(llm).profile(SL0010_TEXT, ["SL-0010-3.xlsx"])
assert profile.tables[0].is_stability_time_series is False
assert profile.has_stability_time_series is False
def test_multi_document_aggregation():
text = (
"=== File: a.xlsx ===\n有关物质检测结果\n总杂%\n0.1\n"
"=== File: b.xlsx ===\n溶出曲线\n累积溶出度%\n95\n"
)
profile = DocumentProfiler(llm=None).profile(text, ["a.xlsx", "b.xlsx"])
srcs = {t.source_document for t in profile.tables}
assert "a.xlsx" in srcs
assert "b.xlsx" in srcs
def test_stability_time_series_detected_from_llm():
llm = _MockLLM({
"tables": [{
"table_type": "stability_time_series",
"title": "长期稳定性",
"is_stability_time_series": True,
"columns": ["时间点（月）", "总杂%"],
}],
"intent": "shelf_life_extrapolation",
"items": [],
})
profile = DocumentProfiler(llm).profile("=== File: s.xlsx ===\n长期稳定性\n", ["s.xlsx"])
assert profile.has_stability_time_series is True
|