79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
"""근거리 제어점 직결 — 조밀 기준선이 격자 탐색에 안 틀어지는지 검증."""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[2]))
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
import B05_Profile.B05_Profile_Engine_Solver as S
|
|
|
|
|
|
@pytest.fixture()
|
|
def fake_project(tmp_path: Path) -> Path:
|
|
"""1m 해상도 완만한 합성 DTM을 가진 가짜 프로젝트 루트."""
|
|
models = tmp_path / "B04_PreProcess" / "models"
|
|
models.mkdir(parents=True)
|
|
n = 61
|
|
x = np.arange(n, dtype=np.float64)
|
|
y = np.arange(n, dtype=np.float64)
|
|
xx, yy = np.meshgrid(x, y)
|
|
z = 100.0 + 0.02 * xx + 0.01 * yy # 경사 2% — 제약에 안 걸린다
|
|
np.savez_compressed(
|
|
models / "dtm_classification_smooth.npz",
|
|
x=x,
|
|
y=y,
|
|
z=z,
|
|
valid_mask=np.ones_like(z, dtype=bool),
|
|
)
|
|
return tmp_path
|
|
|
|
|
|
def _solve(project_root: Path, pts: list[dict]) -> list[list[float]]:
|
|
points_data = {
|
|
"bp": pts[0],
|
|
"ep": pts[-1],
|
|
"cp": [dict(p, order=i + 1) for i, p in enumerate(pts[1:-1])],
|
|
}
|
|
result = S.solve_optimal_route(project_root, "classification", True, points_data, {}, "dtm")
|
|
return result["polyline"]
|
|
|
|
|
|
def test_dense_baseline_preserved(fake_project: Path) -> None:
|
|
"""3m 간격 조밀 제어점(문턱 4m 이하) — 평면이 원좌표 그대로, 끼어든 점 없음."""
|
|
rng = np.random.default_rng(7)
|
|
ts = np.arange(5.0, 50.0, 2.0)
|
|
pts = [
|
|
{"x": float(t + rng.uniform(-0.3, 0.3)), "y": float(t * 0.7 + rng.uniform(-0.3, 0.3))}
|
|
for t in ts
|
|
]
|
|
poly = _solve(fake_project, pts)
|
|
assert len(poly) == len(pts)
|
|
for got, want in zip(poly, pts):
|
|
assert got[0] == pytest.approx(want["x"], abs=1e-9)
|
|
assert got[1] == pytest.approx(want["y"], abs=1e-9)
|
|
|
|
|
|
def test_same_cell_pair_not_merged(fake_project: Path) -> None:
|
|
"""같은 칸에 스냅되는 초근접 정점 쌍도 안 합쳐지고 둘 다 남는다."""
|
|
pts = [
|
|
{"x": 10.0, "y": 10.0},
|
|
{"x": 10.6, "y": 10.3}, # 10.0과 같은 2m 칸
|
|
{"x": 13.0, "y": 11.8},
|
|
{"x": 16.0, "y": 13.2},
|
|
]
|
|
poly = _solve(fake_project, pts)
|
|
assert len(poly) == len(pts)
|
|
assert [p[:2] for p in poly] == [[p["x"], p["y"]] for p in pts]
|
|
|
|
|
|
def test_sparse_pair_still_searched(fake_project: Path) -> None:
|
|
"""문턱(2×셀=4m)을 넘는 먼 구간은 종전대로 격자 탐색 — 중간점이 생긴다."""
|
|
pts = [{"x": 5.0, "y": 5.0}, {"x": 45.0, "y": 40.0}]
|
|
poly = _solve(fake_project, pts)
|
|
assert len(poly) > 2 # 탐색이 격자 중간점을 채웠다
|
|
assert poly[0][:2] == [5.0, 5.0]
|
|
assert poly[-1][:2] == [45.0, 40.0]
|