Files
cad-agent/tests/test_config.py
T
gjm b757892c67 feat(review): agent 侧虚拟工具 inspect_region —— 按可读分辨率放大指定区域
解决三个缺口:① 放大没复用标定/瓦片数学(LLM 一次 zoom 只出一张,宽区域仍读不清);
②「看清没」由 VLM 自评(易过度自信/幻觉);③ 只能按 handle / 手算 bbox 指认,
图上看到但没 handle 的东西指认不了。

核心思路:把「放大」升级成一个语义化工具——LLM 只指认「要看哪里」,由 agent 复用
tiling.py 的标定数学算出该切几张、每张覆盖多大,用构造保证可读。

新增 dwg_review/inspect.py(agent 侧,不编 DLL):
- TOOL_SCHEMA:inspect_region(target, purpose);target 三形态 {handle} / {window} /
  {label,cx,cy,span},purpose = text(8px) / line(6px)。
- InspectState:单图框内「截图标签 → 世界可见范围」登记表,支持逐级下钻
  (返回的 inspect-N 可继续作为 label 指认)。
- resolve_target / _point_box:label+cx+cy → WCS。cy 是图像坐标(上→下),
  世界 y 必须翻转(y = y2 - cy*H)——单测专门钉住。
- plan_tiles:复用 world_per_px + tiles_for_box + subdivide;超 INSPECT_MAX_TILES(4)
  则硬失败并给出「缩小 target / 分多次请求」的指引(不静默降级);拿不到比例/像素
  则退回单张并如实告知。

关键几何:CAD 的 _.ZOOM _W 是等比适配(不拉伸),故「截图上的归一化坐标」必须按
max(W/ew, H/eh) 反算真实可见范围(新增 tiling.visible_box)——直接用送进去的窗口会偏。
zoom_to_window 返回的 view 是「加过 margin 的窗口」,不是可见范围,不能混用。

证据层配套:evidence_model.make_image(..., window=) 记录每张图的可见世界范围;
capture_images 用 view_px(overall 图原始像素)+ visible_box 逐张登记。

子循环接入:_run_readonly_tool 在只读白名单校验之前拦截 inspect_region(它不在
READ_ONLY_TOOLS 里,内部只调白名单内的只读 MCP 工具);_review_tools 启用时用
inspect_region 取代三个裸视口工具;inspect_region 计入 _VIEW_TOOLS → 结束复位视口。
配置:INSPECT_ENABLED(True) / INSPECT_MAX_TILES(4)。

端到端实测(真实图纸 1 图框):取证与审查两阶段均在最大化态、结束精确还原(4 项 PASS);
LLM 的 11 次取证调用里 2 次是 inspect_region,且都显式给 window + purpose:"text"
(x[20000,115000] y[-6500,3000],即 A0 图框底部横条=标题栏所在)。
验收口径达成:报告报出标题栏乱码 ???????????????,而 get_text_content 对全图返回 0 条文字
(本图文字全在块/自定义实体内)→ 块内文字是被视觉放大读出来的。
pytest 343 passed。
2026-10-16 19:15:32 +08:00

151 lines
5.4 KiB
Python

"""`dwg_review.config` 单元测试(只验证纯函数与常量类型,不依赖真实环境变量)。"""
import os
from dwg_review import config
def test_resolve_path_relative_uses_project_root():
assert config._resolve_path("reports") == config.PROJECT_ROOT / "reports"
def test_resolve_path_absolute_unchanged(tmp_path):
assert config._resolve_path(str(tmp_path)) == tmp_path
def test_project_root_is_repo_root():
assert (config.PROJECT_ROOT / "dwg_review" / "config.py").is_file()
def test_read_env_prefers_environ(monkeypatch):
monkeypatch.setenv("CAD_TEST_ENV_PROBE", "hello")
assert config._read_env("CAD_TEST_ENV_PROBE") == "hello"
def test_read_env_missing_returns_empty(monkeypatch):
monkeypatch.delenv("CAD_TEST_DEFINITELY_MISSING_XYZ", raising=False)
assert config._read_env("CAD_TEST_DEFINITELY_MISSING_XYZ") == ""
def test_read_bool_defaults_when_unset(monkeypatch):
monkeypatch.delenv("CAD_TEST_BOOL_UNSET", raising=False)
assert config._read_bool("CAD_TEST_BOOL_UNSET", True) is True
assert config._read_bool("CAD_TEST_BOOL_UNSET", False) is False
def test_read_bool_empty_string_uses_default(monkeypatch):
monkeypatch.setenv("CAD_TEST_BOOL_EMPTY", " ")
assert config._read_bool("CAD_TEST_BOOL_EMPTY", True) is True
def test_read_bool_truthy_and_falsy(monkeypatch):
for raw in ("1", "true", "TRUE", "Yes", "on"):
monkeypatch.setenv("CAD_TEST_BOOL", raw)
assert config._read_bool("CAD_TEST_BOOL", False) is True
for raw in ("0", "false", "no", "off", "nope"):
monkeypatch.setenv("CAD_TEST_BOOL", raw)
assert config._read_bool("CAD_TEST_BOOL", True) is False
def test_has_api_key_reflects_module_value(monkeypatch):
monkeypatch.setattr(config, "API_KEY", "")
assert config.has_api_key() is False
monkeypatch.setattr(config, "API_KEY", "sk-x")
assert config.has_api_key() is True
def test_has_api_key_ignores_whitespace(monkeypatch):
monkeypatch.setattr(config, "API_KEY", " ")
assert config.has_api_key() is False
def test_constants_have_expected_types():
assert isinstance(config.HOST, str)
assert isinstance(config.PORT, int)
assert isinstance(config.CALL_TIMEOUT, float)
assert isinstance(config.CALL_RETRIES, int)
assert isinstance(config.MAX_FRAME_BYTES, int)
assert isinstance(config.IMAGE_MAX_EDGE, int)
assert isinstance(config.IMAGE_FORMAT, str)
assert isinstance(config.CHECKLIST_PATH, type(config.PROJECT_ROOT))
def test_credibility_constants_have_expected_types():
assert isinstance(config.DETERMINISTIC_CHECKS, bool)
assert isinstance(config.VALIDATE_HANDLES, bool)
assert isinstance(config.CHECK_UNCLOSED_POLYLINES, bool)
assert isinstance(config.ZERO_LENGTH_TOLERANCE, float)
assert isinstance(config.DUP_TOLERANCE, float)
assert isinstance(config.DUP_GROUP_LIMIT, int)
def test_tiling_constants_have_expected_types():
assert isinstance(config.TILING_ENABLED, bool)
assert isinstance(config.TEXT_HEIGHT_MM, float)
assert isinstance(config.TEXT_READABLE_PX, int)
assert isinstance(config.LINE_READABLE_PX, int)
assert isinstance(config.TILE_MAX_COUNT, int)
assert isinstance(config.TILE_OVERLAP, float)
assert isinstance(config.HEATMAP_CELL_DIV, int)
assert isinstance(config.HEATMAP_CLUSTER_DILATE, int)
assert isinstance(config.HEATMAP_MAX_AREA_FRAC, float)
def test_tiling_defaults_are_sane():
# 默认启用瓦片规划;可读下限为正;成本上界为正
assert config.TILING_ENABLED is True
assert config.TEXT_HEIGHT_MM > 0
assert config.TEXT_READABLE_PX > 0
assert config.LINE_READABLE_PX > 0
assert config.TILE_MAX_COUNT > 0
assert config.HEATMAP_CELL_DIV > 0
def test_review_obs_images_default_is_sane():
# 审查子循环回灌截图上限:须为正,且不低于旧硬编码值 3
assert config.REVIEW_OBS_IMAGES >= 3
def test_inspect_defaults_are_sane():
# agent 侧虚拟工具默认启用;单次瓦片上限为正
assert config.INSPECT_ENABLED is True
assert config.INSPECT_MAX_TILES >= 1
def test_drawing_area_constants_have_expected_types():
assert isinstance(config.DRAWING_AREA_MAXIMIZE, bool)
assert isinstance(config.DRAWING_AREA_ENTER_WAIT, float)
assert isinstance(config.DRAWING_AREA_RESTORE_WAIT, float)
def test_drawing_area_defaults_are_sane():
# 默认开启绘图区最大化;还原等待须盖过插件内 700ms 的延迟还原(实测 1.0s 才收敛)
assert config.DRAWING_AREA_MAXIMIZE is True
assert config.DRAWING_AREA_ENTER_WAIT >= 0
assert config.DRAWING_AREA_RESTORE_WAIT >= 1.2
def test_credibility_defaults_are_on():
# 默认开启,保证默认拿到可信度收益
assert config.DETERMINISTIC_CHECKS is True
assert config.VALIDATE_HANDLES is True
assert config.CHECK_UNCLOSED_POLYLINES is True
def test_checklist_path_resolves_relative_to_root():
assert config.CHECKLIST_PATH.name == "review_checklist.json"
assert config.CHECKLIST_PATH.parent == config.PROJECT_ROOT
def test_port_default_matches_baseline():
# 基座固定监听 127.0.0.1:8080;若此值被改动,请同步更新基座与文档。
assert config.HOST == "127.0.0.1"
assert config.PORT == 8080
def test_os_name_guard_in_read_env(monkeypatch):
# 非 Windows 平台不应触碰注册表
monkeypatch.setattr(os, "name", "posix")
monkeypatch.delenv("CAD_TEST_ENV_PROBE2", raising=False)
assert config._read_env("CAD_TEST_ENV_PROBE2") == ""