Function test_sentinel_multispectral

scripts/api_test.py:126–195

()

def test_sentinel_multispectral() -> None:
    band_sets: List[List[str]] = [
        ["red", "green", "blue"],
        ["nir", "red", "green"],
        ["swir22", "nir", "green"],
        ["swir22", "swir16", "red"],
        ["rededge1", "rededge2", "rededge3"],
    ]

    # Freeze location (and timestamp when available) once so every band set
    # is requested at the exact same satellite coordinates.
    position_response = _request_get("/data/current/position")
    if position_response is None:
        return
    try:
        position_data = position_response.json()
    except ValueError:
        print("Invalid JSON body for /data/current/position.")
        return

    lon_lat_alt = position_data.get("lon-lat-alt")
    timestamp = position_data.get("timestamp")

    images = []
    for spectral_bands in band_sets:
        params = {
            "lon": lon_lat_alt[0],
            "lat": lon_lat_alt[1],
            "timestamp": timestamp,
            "spectral_bands": spectral_bands,
            "size_km": 5.0,
            "return_type": "png",
            #"window_seconds": 60.0
        }
        response = _request_get("/data/image/sentinel", params=params)
        if response is None:
            continue

        try:
            metadata = json.loads(response.headers.get("sentinel_metadata", ""))
        except json.JSONDecodeError:
            metadata = None

        if not isinstance(metadata, dict) or not metadata.get("image_available"):
            print(f"No image available for bands {spectral_bands}")
            continue

        try:
            image = mpimg.imread(io.BytesIO(response.content), format="PNG")
        except Exception as exc:
            print(f"Could not decode image for bands {spectral_bands}: {exc}")
            continue

        images.append((image, "_".join(spectral_bands)))

    if not images:
        print("No hyperspectral images available to display.")
        return