lekss361
9b9c6e85d3
ird_overlay_lookup.py — read-side ИРД-overlay lookup: ST_Intersects участка (WKT 4326) против ird_overlays (м.132) → ограничения + reg_numb_border + zone_index/type + группировка by_kind. Standalone (parcels.py не тронут), GIST idx_ird_overlays_geom, graceful-degrade при отсутствии таблицы. Wiring в analyze деферится в D9b flag-gated. 4 теста. Refs #1067. Co-authored-by: lekss361 <lekss361@gendsgn.local> Co-committed-by: lekss361 <lekss361@gendsgn.local>
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""Тесты parcel_ird_overlaps (#1067 D9) — ИРД-overlay lookup для analyze."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from sqlalchemy.exc import OperationalError
|
|
|
|
from app.services.site_finder.ird_overlay_lookup import parcel_ird_overlaps
|
|
|
|
|
|
class _Result:
|
|
def __init__(self, rows: list[dict[str, Any]]) -> None:
|
|
self._rows = rows
|
|
|
|
def mappings(self) -> _Result:
|
|
return self
|
|
|
|
def all(self) -> list[dict[str, Any]]:
|
|
return self._rows
|
|
|
|
|
|
class _FakeDB:
|
|
def __init__(
|
|
self, rows: list[dict[str, Any]] | None = None, raise_exc: Exception | None = None
|
|
) -> None:
|
|
self._rows = rows or []
|
|
self._raise = raise_exc
|
|
|
|
def execute(self, sql: Any, params: dict[str, Any] | None = None) -> _Result:
|
|
if self._raise is not None:
|
|
raise self._raise
|
|
return _Result(self._rows)
|
|
|
|
|
|
_WKT = "POLYGON((60.5 56.8, 60.6 56.8, 60.6 56.9, 60.5 56.9, 60.5 56.8))"
|
|
|
|
|
|
def test_groups_overlaps_by_kind() -> None:
|
|
rows = [
|
|
{
|
|
"layer_kind": "zouit_okn",
|
|
"reg_numb_border": "66:41-6.41",
|
|
"category": 36940,
|
|
"subcategory": 20,
|
|
"zone_index": None,
|
|
"zone_type_name": None,
|
|
},
|
|
{
|
|
"layer_kind": "zouit_okn",
|
|
"reg_numb_border": "66:41-6.42",
|
|
"category": 36940,
|
|
"subcategory": 20,
|
|
"zone_index": None,
|
|
"zone_type_name": None,
|
|
},
|
|
{
|
|
"layer_kind": "territorial_zones",
|
|
"reg_numb_border": "66:41-7.10",
|
|
"category": 472819,
|
|
"subcategory": 1,
|
|
"zone_index": "Ц-1",
|
|
"zone_type_name": None,
|
|
},
|
|
]
|
|
res = parcel_ird_overlaps(_FakeDB(rows), _WKT)
|
|
assert len(res["ird_overlaps"]) == 3
|
|
assert res["ird_by_kind"] == {"zouit_okn": 2, "territorial_zones": 1}
|
|
tz = next(o for o in res["ird_overlaps"] if o["layer_kind"] == "territorial_zones")
|
|
assert tz["reg_numb_border"] == "66:41-7.10"
|
|
assert tz["zone_index"] == "Ц-1"
|
|
|
|
|
|
def test_empty_when_no_wkt() -> None:
|
|
res = parcel_ird_overlaps(_FakeDB([{"layer_kind": "x"}]), None)
|
|
assert res == {"ird_overlaps": [], "ird_by_kind": {}}
|
|
|
|
|
|
def test_empty_when_no_overlap() -> None:
|
|
res = parcel_ird_overlaps(_FakeDB([]), _WKT)
|
|
assert res == {"ird_overlaps": [], "ird_by_kind": {}}
|
|
|
|
|
|
def test_graceful_when_table_missing() -> None:
|
|
"""ird_overlays ещё не задеплоена → пусто, без падения analyze."""
|
|
db = _FakeDB(raise_exc=OperationalError("stmt", {}, Exception("no such table")))
|
|
res = parcel_ird_overlaps(db, _WKT)
|
|
assert res == {"ird_overlaps": [], "ird_by_kind": {}}
|