Files
cad-agent/tests/test_config.py
T
gjm cba05b0c65 feat(tiling): 自适应瓦片规划器 + VLM 分辨率离线标定
背景:逐图框审查固定「整体 + 2x2 象限」截图,大图幅(A0/A1)读不清细节。
改为按**实体分布**自适应决定截几张、截哪里。

- 新增 dwg_review/tiling.py(纯函数):
  SHEET_SIZES_MM / frame_world_size / frame_scale(sheet_scale>0 用之,否则由
  bbox/纸张反算,否则 default)/ effective_px(与 downsample 同规则)/
  world_per_px / is_text_type(标准名 + 自定义类名含 TEXT/LABEL/DIM/NOTE/TAG/
  ANNO/TITLE)/ cluster_text_flags / tiles_for_box / subdivide / plan_regions
  (文字簇优先、空白区不出图、max_tiles 截断、窗口夹到图框范围)
- 新增 verify_vlm_resolution_offline.py:PIL 合成类图纸图(精确像素字高 +
  可控线间距/线宽/虚线)→ 问 VLM 强约束 JSON → 定 R_text / R_line。
  实测:R_text=12px(黑底;白底 10px);R_line=6px。
  另:实测该模型**无法可靠计数**(6 条线在 2~10px 间距下答 4/10/4/5/4/4),
  故线条判据改用「1 条 vs 2 条」二选一。
- config 新增 TILING_ENABLED / TEXT_HEIGHT_MM / TEXT_READABLE_PX=12 /
  LINE_READABLE_PX=6 / TILE_MAX_COUNT / TILE_OVERLAP / HEATMAP_CELL_DIV /
  HEATMAP_CLUSTER_DILATE / HEATMAP_MAX_AREA_FRAC;EVIDENCE_DETAIL_TILES 与
  EVIDENCE_MAX_IMAGES 保留作回退路径参数
- llm_client 新增 image_size_b64(测量绘图区实际像素,决定单张覆盖范围)
- phases/evidence.capture_images 支持瓦片规划:先截整体图并测像素 →
  plan_regions 成功用规划窗口,否则退回 sub_windows(行为零变化);
  末尾始终复位整体视口。collect_frame 传实体/文字 bbox;任一信封
  total>returned(截断)→ 退回固定策略,避免误删有内容的瓦片
- 测试:tests/test_tiling.py(28 项)、tests/test_evidence.py 新增规划路径与
  降级用例、test_config / test_llm_client 补新常量与 image_size_b64;
  全量 283 项全绿
2026-10-15 23:03:27 +08:00

127 lines
4.6 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_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") == ""