背景:逐图框审查固定「整体 + 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 项全绿
127 lines
4.6 KiB
Python
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") == ""
|