89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
import httpx
|
|
import pytest
|
|
|
|
from pipeline.download import noise
|
|
|
|
|
|
def test_download_tile_splits_after_retries(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(noise, "MAX_RETRIES", 1)
|
|
monkeypatch.setattr(noise, "MIN_TILE_SIZE", 50)
|
|
|
|
def fake_fetch_tile_bytes(
|
|
wcs_base,
|
|
coverage_id,
|
|
min_e,
|
|
min_n,
|
|
max_e,
|
|
max_n,
|
|
wcs_version="1.0.0",
|
|
):
|
|
if max_e - min_e > 50 or max_n - min_n > 50:
|
|
raise httpx.TimeoutException("too large")
|
|
return b"II*\x00fake-tiff"
|
|
|
|
monkeypatch.setattr(noise, "_fetch_tile_bytes", fake_fetch_tile_bytes)
|
|
|
|
paths, failures = noise._download_tile("base", "coverage", 0, 0, 100, 100, tmp_path)
|
|
|
|
assert failures == []
|
|
assert len(paths) == 4
|
|
assert sorted(path.name for path in paths) == [
|
|
"tile_0_0_50_50.tif",
|
|
"tile_0_50_50_100.tif",
|
|
"tile_50_0_100_50.tif",
|
|
"tile_50_50_100_100.tif",
|
|
]
|
|
|
|
|
|
def test_download_tile_reports_unsplittable_failure(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(noise, "MAX_RETRIES", 1)
|
|
monkeypatch.setattr(noise, "MIN_TILE_SIZE", 100)
|
|
|
|
def fake_fetch_tile_bytes(*args, **kwargs):
|
|
raise httpx.ConnectError("offline")
|
|
|
|
monkeypatch.setattr(noise, "_fetch_tile_bytes", fake_fetch_tile_bytes)
|
|
|
|
paths, failures = noise._download_tile("base", "coverage", 0, 0, 100, 100, tmp_path)
|
|
|
|
assert paths == []
|
|
assert failures == [(0, 0, 100, 100)]
|
|
|
|
|
|
def test_download_raster_tolerates_missing_tiles_when_allowed(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(noise, "BNG_MIN_E", 0)
|
|
monkeypatch.setattr(noise, "BNG_MAX_E", 100)
|
|
monkeypatch.setattr(noise, "BNG_MIN_N", 0)
|
|
monkeypatch.setattr(noise, "BNG_MAX_N", 100)
|
|
monkeypatch.setattr(noise, "TILE_SIZE", 100)
|
|
|
|
def fake_download_tile(*args, **kwargs):
|
|
return [], [(0, 0, 100, 100)]
|
|
|
|
monkeypatch.setattr(noise, "_download_tile", fake_download_tile)
|
|
|
|
paths = noise.download_raster(
|
|
tmp_path,
|
|
"base",
|
|
"coverage",
|
|
"Airport",
|
|
allow_missing_tiles=True,
|
|
)
|
|
|
|
assert paths == []
|
|
|
|
|
|
def test_download_raster_raises_on_missing_strict_tiles(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(noise, "BNG_MIN_E", 0)
|
|
monkeypatch.setattr(noise, "BNG_MAX_E", 100)
|
|
monkeypatch.setattr(noise, "BNG_MIN_N", 0)
|
|
monkeypatch.setattr(noise, "BNG_MAX_N", 100)
|
|
monkeypatch.setattr(noise, "TILE_SIZE", 100)
|
|
|
|
def fake_download_tile(*args, **kwargs):
|
|
return [], [(0, 0, 100, 100)]
|
|
|
|
monkeypatch.setattr(noise, "_download_tile", fake_download_tile)
|
|
|
|
with pytest.raises(RuntimeError, match=r"\[Road\] Failed to download"):
|
|
noise.download_raster(tmp_path, "base", "coverage", "Road")
|