206 lines
7.3 KiB
Python
206 lines
7.3 KiB
Python
"""Tests pour la carte globale interactive (index.py)."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
|
|
def test_parse_basename_coords_valid():
|
|
"""Parse les coordonnées d'un basename LHD valide."""
|
|
from lidar_pipeline.index import parse_basename_coords
|
|
assert parse_basename_coords("LHD_FXX_1000_6881_PTS_LAMB93_IGN69") == (1000, 6881)
|
|
assert parse_basename_coords("LHD_FXX_1049_6895_PTS_LAMB93_IGN69") == (1049, 6895)
|
|
|
|
|
|
def test_parse_basename_coords_with_res_suffix():
|
|
"""Les noms de dossier avec suffixe résolution sont aussi parsables."""
|
|
from lidar_pipeline.index import parse_basename_coords
|
|
assert parse_basename_coords("LHD_FXX_1000_6881_PTS_LAMB93_IGN69_r0p2") == (1000, 6881)
|
|
|
|
|
|
def test_parse_basename_coords_invalid():
|
|
"""Les noms non-LHD retournent None."""
|
|
from lidar_pipeline.index import parse_basename_coords
|
|
assert parse_basename_coords("random_dir") is None
|
|
assert parse_basename_coords("DTM") is None
|
|
assert parse_basename_coords("") is None
|
|
|
|
|
|
def test_strip_res_suffix_primary():
|
|
"""Dossier sans suffixe = résolution primaire (0.5)."""
|
|
from lidar_pipeline.index import _strip_res_suffix
|
|
base, res = _strip_res_suffix("LHD_FXX_1000_6881_PTS_LAMB93_IGN69")
|
|
assert base == "LHD_FXX_1000_6881_PTS_LAMB93_IGN69"
|
|
assert res == 0.5
|
|
|
|
|
|
def test_strip_res_suffix_multi():
|
|
"""Dossier avec suffixe _r0p2 = résolution 0.2."""
|
|
from lidar_pipeline.index import _strip_res_suffix
|
|
base, res = _strip_res_suffix("LHD_FXX_1000_6881_PTS_LAMB93_IGN69_r0p2")
|
|
assert base == "LHD_FXX_1000_6881_PTS_LAMB93_IGN69"
|
|
assert res == 0.2
|
|
|
|
|
|
def test_compute_bbox():
|
|
"""Calcule la bounding box d'un ensemble de tuiles."""
|
|
from lidar_pipeline.index import compute_bbox
|
|
tiles = [
|
|
{'col': 1000, 'row': 6881},
|
|
{'col': 1001, 'row': 6882},
|
|
{'col': 1002, 'row': 6880},
|
|
]
|
|
bbox = compute_bbox(tiles)
|
|
assert bbox == {'min_col': 1000, 'max_col': 1002, 'min_row': 6880, 'max_row': 6882}
|
|
|
|
|
|
def test_compute_bbox_empty():
|
|
"""Aucune tuile → None."""
|
|
from lidar_pipeline.index import compute_bbox
|
|
assert compute_bbox([]) is None
|
|
|
|
|
|
def _make_fake_viz_dir(vis_dir, basename, col, row, viz_keys=('hillshade_multi', 'svf'), ext='webp', res_suffix=''):
|
|
"""Crée un faux dossier de visualisations avec de petites images."""
|
|
from PIL import Image as PILImage
|
|
import numpy as np
|
|
|
|
dir_name = f"LHD_FXX_{col}_{row}_PTS_LAMB93_IGN69{res_suffix}"
|
|
tile_dir = Path(vis_dir) / dir_name
|
|
tile_dir.mkdir(parents=True, exist_ok=True)
|
|
for v in viz_keys:
|
|
arr = np.random.randint(0, 255, (50, 50, 3), dtype=np.uint8)
|
|
img = PILImage.fromarray(arr)
|
|
fname = f"LHD_FXX_{col}_{row}_PTS_LAMB93_IGN69{res_suffix}_{v}.{ext}"
|
|
img.save(str(tile_dir / fname), format='WEBP', quality=80)
|
|
return tile_dir
|
|
|
|
|
|
def test_scan_tiles(tmp_path):
|
|
"""scan_tiles détecte les dossiers de tuiles et leurs visualisations."""
|
|
from lidar_pipeline.index import scan_tiles
|
|
|
|
vis_dir = tmp_path / "visualisations"
|
|
vis_dir.mkdir()
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi', 'svf'))
|
|
_make_fake_viz_dir(vis_dir, "b", 1001, 6881, ('hillshade_multi',))
|
|
|
|
tiles = scan_tiles(vis_dir)
|
|
assert len(tiles) == 2
|
|
names = sorted(t['dir_name'] for t in tiles)
|
|
assert "LHD_FXX_1000_6881_PTS_LAMB93_IGN69" in names
|
|
assert "LHD_FXX_1001_6881_PTS_LAMB93_IGN69" in names
|
|
# Vérifie que les viz sont détectées
|
|
t0 = next(t for t in tiles if t['col'] == 1000)
|
|
assert 'hillshade_multi' in t0['viz']
|
|
assert 'svf' in t0['viz']
|
|
|
|
|
|
def test_scan_tiles_ignores_non_lhd(tmp_path):
|
|
"""Les dossiers non-LHD (ex: temp, DTM) sont ignorés."""
|
|
from lidar_pipeline.index import scan_tiles
|
|
|
|
vis_dir = tmp_path / "visualisations"
|
|
vis_dir.mkdir()
|
|
(vis_dir / "random_folder").mkdir()
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881)
|
|
|
|
tiles = scan_tiles(vis_dir)
|
|
assert len(tiles) == 1
|
|
assert tiles[0]['col'] == 1000
|
|
|
|
|
|
def test_scan_tiles_multi_resolution(tmp_path):
|
|
"""Les dossiers avec suffixe résolution sont correctement décodés."""
|
|
from lidar_pipeline.index import scan_tiles
|
|
|
|
vis_dir = tmp_path / "visualisations"
|
|
vis_dir.mkdir()
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881, res_suffix='')
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881, res_suffix='_r0p2')
|
|
|
|
tiles = scan_tiles(vis_dir)
|
|
assert len(tiles) == 2
|
|
resolutions = sorted(t['resolution'] for t in tiles)
|
|
assert resolutions == [0.2, 0.5]
|
|
|
|
|
|
def test_build_index_generates_html(tmp_path):
|
|
"""build_index génère index.html et les vignettes."""
|
|
from lidar_pipeline.index import build_index
|
|
|
|
output_dir = tmp_path / "output"
|
|
vis_dir = output_dir / "visualisations"
|
|
vis_dir.mkdir(parents=True)
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi', 'svf'))
|
|
_make_fake_viz_dir(vis_dir, "b", 1001, 6881, ('hillshade_multi',))
|
|
|
|
result = build_index(output_dir)
|
|
assert result is not None
|
|
html_path = Path(result)
|
|
assert html_path.exists()
|
|
assert html_path.name == "index.html"
|
|
|
|
content = html_path.read_text(encoding='utf-8')
|
|
# Vérifie la présence des éléments clés
|
|
assert "Carte globale LiDAR" in content
|
|
assert "LHD_FXX_1000_6881" in content
|
|
assert "LHD_FXX_1001_6881" in content
|
|
# Vérifie que le JSON intégré est valide
|
|
assert "const DATA" in content
|
|
# Vérifie les vignettes générées
|
|
thumb_dir = output_dir / "index_thumbs"
|
|
assert thumb_dir.is_dir()
|
|
thumbs = list(thumb_dir.glob("*.jpg"))
|
|
assert len(thumbs) >= 2 # au moins hillshade pour chaque tuile
|
|
|
|
|
|
def test_build_index_empty_returns_none(tmp_path):
|
|
"""Aucune tuile → build_index retourne None sans crash."""
|
|
from lidar_pipeline.index import build_index
|
|
|
|
output_dir = tmp_path / "output"
|
|
(output_dir / "visualisations").mkdir(parents=True)
|
|
result = build_index(output_dir)
|
|
assert result is None
|
|
|
|
|
|
def test_build_index_embeds_valid_json(tmp_path):
|
|
"""Le JSON embarqué dans le HTML est valide et contient les tuiles."""
|
|
from lidar_pipeline.index import build_index
|
|
|
|
output_dir = tmp_path / "output"
|
|
vis_dir = output_dir / "visualisations"
|
|
vis_dir.mkdir(parents=True)
|
|
_make_fake_viz_dir(vis_dir, "a", 1000, 6881, ('hillshade_multi',))
|
|
|
|
build_index(output_dir)
|
|
content = (output_dir / "index.html").read_text(encoding='utf-8')
|
|
# Extrait le JSON entre "const DATA = " et ";"
|
|
start = content.index("const DATA = ") + len("const DATA = ")
|
|
# Trouve le ; de fin de déclaration
|
|
depth = 0
|
|
end = start
|
|
for i, ch in enumerate(content[start:], start):
|
|
if ch == '{':
|
|
depth += 1
|
|
elif ch == '}':
|
|
depth -= 1
|
|
if depth == 0:
|
|
end = i + 1
|
|
break
|
|
data = json.loads(content[start:end])
|
|
assert 'tiles' in data
|
|
assert 'bbox' in data
|
|
assert 'vizList' in data
|
|
assert len(data['tiles']) == 1
|
|
assert data['tiles'][0]['col'] == 1000
|
|
assert data['tiles'][0]['row'] == 6881
|
|
|
|
|
|
def test_pick_display_viz_prefers_hillshade():
|
|
"""Le choix de viz par défaut privilégie hillshade_multi."""
|
|
from lidar_pipeline.index import _pick_display_viz
|
|
assert _pick_display_viz(['svf', 'hillshade_multi', 'slope']) == 'hillshade_multi'
|
|
assert _pick_display_viz(['svf', 'slope']) == 'svf'
|
|
assert _pick_display_viz(['topo']) == 'topo'
|