mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-13 14:31:01 +02:00
Leo Thomas
8f0239c1c9
* Add initial NAIP fetcher * Swap to Element84's EarthSearch API for NAIP STAC search and download * clip to bounds of aoi * Swap to Element84's EarthSearch API for NAIP STAC search and download * rename bands and remove dask chunking * Add DS_Store to .gitignore * restrict date range for naip test * Adjust timerange for tests * Add xarray to pyproj * Reduce aoi size * revert test to use tmp path * Update return types for tool to ensure state gets updated * Mark naip test as xfail * Fix geom creation --------- Co-authored-by: lillythomas <lillyelizathomas@gmail.com> Co-authored-by: Daniel Wiesmann <yellowcap@users.noreply.github.com>
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
import pytest
|
||
from pathlib import Path
|
||
from shapely.geometry import box, mapping
|
||
from langchain_core.tools.base import ToolCall
|
||
|
||
from geo_assistant.tools.naip import fetch_naip_img
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
@pytest.mark.xfail
|
||
async def test_fetch_naip(tmp_path):
|
||
"""
|
||
Integration test: hit MPC STAC for NAIP around Union Market (DC),
|
||
load imagery via odc-stac, and save an RGB PNG.
|
||
|
||
NOTE: This test requires:
|
||
- Internet access (to reach Planetary Computer STAC + blobs)
|
||
- Planetary Computer / NAIP service to be up
|
||
"""
|
||
|
||
# Union Market coordinates from GeoNames: 38.90789, -76.99831
|
||
# N 38°54′28″ W 76°59′54″
|
||
# We'll use a small neighborhood AOI around that point.
|
||
# :contentReference[oaicite:0]{index=0}
|
||
lat = 38.90789
|
||
lon = -76.99831
|
||
|
||
# ~0.01 degrees (~1.1 km) buffer in each direction
|
||
aoi = box(lon - 0.0001, lat - 0.0001, lon + 0.0001, lat + 0.0001)
|
||
aoi_geojson = mapping(aoi)
|
||
|
||
out_png = tmp_path / "naip_test_img.png"
|
||
|
||
tool_call = ToolCall(
|
||
name="fetch_naip_img",
|
||
args={
|
||
"aoi_geojson": aoi_geojson,
|
||
"start_date": "2021-01-01",
|
||
"end_date": "2021-12-31",
|
||
"out_png_path": str(out_png),
|
||
"resolution": 1.0,
|
||
},
|
||
type="tool_call",
|
||
id="test_tool_call_id",
|
||
)
|
||
|
||
# Call the actual tool – no STAC / odc-stac mocking
|
||
result = await fetch_naip_img.ainvoke(tool_call["args"])
|
||
|
||
# Basic sanity checks on result
|
||
assert result["stac_item_count"] > 0, "Expected at least one NAIP item"
|
||
assert "time" in result["dataset_dims"]
|
||
assert result["dataset_dims"]["time"] >= 1
|
||
|
||
# PNG should have been written to disk
|
||
png_path = Path(result["png_path"])
|
||
assert png_path == out_png
|
||
assert png_path.is_file(), f"PNG was not created at {png_path}"
|