gjm
e8140f2bc5
发现的缺口：maximized_drawing_area 原先只包住取证循环，审查阶段没被包住 → LLM 在受控子循环里 zoom + 截图拿到的是未最大化的 2110x462（正是上一轮刚解决的 低分辨率问题，在最需要放大看细节的审查阶段又出现了）；且审查中的 zoom 不复位视口， 跑完把用户的 CAD 留在最后一次放大处。 改动： 1. __main__.run_review：把 maximized_drawing_area 从「只包取证」扩到「包取证 + 审查」 （两者都要截图）；report/annotate 留在 with 之外。 2. phases/review.py：新增 _VIEW_TOOLS；review_frame 拆成外壳 + _review_frame， 外壳 try/finally 里仅当确实动过视口才复位到该图框整幅视图 （zoom_to_window(win, margin=0.02)）。纯查结构化数据的路径不受影响。 3. checklist.build_system_prompt：把「可调用只读工具」改写成明确的放大取证协议—— 截图看不清时用 zoom_to_window（坐标由图框/实体 bbox 算）或 zoom_to_entity， 再 get_viewport_screenshot 复核；并约束不要为看图而反复放大。 次生问题（同轮修复）：_MAX_OBS_IMAGES 原为硬编码 3，而模型一轮内会发 8 次截图 → 拍了 8 张只回灌 3 张。改为配置项 REVIEW_OBS_IMAGES（默认 6），并把预算耗尽的占位 文案改成明确告知模型「不要再请求截图，请给出结论」。 实测（真实图纸 1 图框，新增 verify_review_maximized.py）：取证与审查两阶段窗口均为 [-4,-2,3444,1442]（最大化态），结束精确还原 [68,0,2316,960]，三项全 PASS； 子循环里 LLM 连发 8 组 zoom_to_window + get_viewport_screenshot，提示词生效。 pytest 304 passed。
145 lines
5.3 KiB
Python
145 lines
5.3 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_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") == ""
|