63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from pathlib import Path
|
|
|
|
import numpy as np
|
|
import rasterio
|
|
from rasterio.transform import from_origin
|
|
|
|
from app.core.geo import PathPoint
|
|
from app.core.landcover import landcover_path, worldcover_class
|
|
|
|
|
|
def write_raster(path: Path, data: np.ndarray, nodata: float = -9999.0) -> None:
|
|
transform = from_origin(30.0, 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=nodata,
|
|
) as dataset:
|
|
dataset.write(data, 1)
|
|
|
|
|
|
def test_worldcover_class_mapping() -> None:
|
|
assert worldcover_class(10) == "tree_cover"
|
|
assert worldcover_class(50) == "built_up"
|
|
assert worldcover_class(255) == "unknown"
|
|
|
|
|
|
def test_landcover_path_segments_and_vegetation_depth(tmp_path: Path) -> None:
|
|
landcover_dir = tmp_path / "landcover"
|
|
canopy_dir = tmp_path / "canopy"
|
|
landcover_dir.mkdir()
|
|
canopy_dir.mkdir()
|
|
write_raster(
|
|
landcover_dir / "worldcover.tif",
|
|
np.array([[10, 10, 50, 50]], dtype="uint8"),
|
|
nodata=0,
|
|
)
|
|
write_raster(
|
|
canopy_dir / "canopy.tif",
|
|
np.array([[12.0, 14.0, 0.0, 0.0]], dtype="float32"),
|
|
)
|
|
points = [
|
|
PathPoint(lat=59.995, lon=30.005, distance_m=0),
|
|
PathPoint(lat=59.995, lon=30.015, distance_m=10),
|
|
PathPoint(lat=59.995, lon=30.025, distance_m=20),
|
|
PathPoint(lat=59.995, lon=30.035, distance_m=30),
|
|
]
|
|
|
|
result = landcover_path(points, landcover_dir, canopy_dir)
|
|
|
|
assert result.vegetation_depth_m == 20
|
|
assert len(result.segments) == 2
|
|
assert result.segments[0].class_name == "tree_cover"
|
|
assert result.segments[0].forest_type == "unknown"
|
|
assert result.segments[0].canopy_height_m == 13.0
|
|
assert result.segments[1].class_name == "built_up"
|