added conopy

api/tests/test_viewshed_and_canopy.py

@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+import sys
+from pathlib import Path
+
+import numpy as np
+import rasterio
+from rasterio.transform import from_origin
+
+from app.core.geo import GeoPoint
+from app.core.viewshed import _find_dem_tiles, _radius_bbox
+
+sys.path.append(str(Path(__file__).resolve().parents[2] / "scripts"))
+import bootstrap_canopy  # noqa: E402
+
+
+def _write_dem(path: Path, west: float) -> None:
+    data = np.ones((10, 10), dtype="float32")
+    transform = from_origin(west, 60.0, 0.01, 0.01)
+    with rasterio.open(
+        path,
+        "w",
+        driver="GTiff",
+        height=data.shape[0],
+        width=data.shape[1],
+        count=1,
+        dtype=data.dtype,
+        crs="EPSG:4326",
+        transform=transform,
+        nodata=-9999.0,
+    ) as dataset:
+        dataset.write(data, 1)
+
+
+def test_viewshed_dem_search_uses_radius_bbox(tmp_path: Path) -> None:
+    _write_dem(tmp_path / "west.tif", 30.0)
+    _write_dem(tmp_path / "east.tif", 30.1)
+    bbox = _radius_bbox(GeoPoint(lat=59.95, lon=30.095), 2_000)
+
+    tiles = _find_dem_tiles(tmp_path, bbox)
+
+    assert {tile.name for tile in tiles} == {"west.tif", "east.tif"}
+
+
+def test_canopy_selects_intersecting_tiles_from_geojson_index() -> None:
+    index = {
+        "type": "FeatureCollection",
+        "features": [
+            {
+                "type": "Feature",
+                "bbox": [30.0, 59.0, 31.0, 60.0],
+                "properties": {"filename": "tile_a.tif"},
+            },
+            {
+                "type": "Feature",
+                "bbox": [40.0, 59.0, 41.0, 60.0],
+                "properties": {"filename": "tile_b.tif"},
+            },
+        ],
+    }
+
+    urls = bootstrap_canopy.select_tile_urls(index, (30.2, 59.2, 30.4, 59.4))
+
+    assert len(urls) == 1
+    assert urls[0].endswith("/tile_a.tif")
+
+
+def test_canopy_urls_file_ignores_comments(tmp_path: Path) -> None:
+    urls_file = tmp_path / "urls.txt"
+    urls_file.write_text(
+        "# comment\n"
+        "https://example.com/a.tif\n"
+        "\n"
+        "https://example.com/b.tif\n",
+        encoding="utf-8",
+    )
+
+    assert bootstrap_canopy.urls_from_file(urls_file) == [
+        "https://example.com/a.tif",
+        "https://example.com/b.tif",
+    ]