mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-13 14:31:01 +02:00
Wei Ji
e3373026d6
* Use pydocstyle (D) rule with google convention Add a ruff rule to catch missing documentation. Using google convention so that undocumented-param (D417) rule is enabled to catch missing params, xref https://docs.astral.sh/ruff/rules/undocumented-param. Extended to include D213 (instead of D212) and D410 rules too. * Fix D100 Missing docstring in public module * Fix D101 Missing docstring in public class * Fix D103 Missing docstring in public function Also ignore rule D205 to allow first sentence of docstring to wrap to multiple lines. * Fix D417 Missing argument description in the docstring * Update indent in pyproject.toml file
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
"""Tests for NAIP tool."""
|
|
|
|
from types import NoneType
|
|
|
|
import pytest
|
|
from geojson_pydantic import Feature
|
|
from langchain_core.tools.base import ToolCall
|
|
from shapely.geometry import box, mapping
|
|
|
|
from geo_assistant.agent.state import GeoAssistantState
|
|
from geo_assistant.tools.naip import fetch_naip_img
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_naip():
|
|
"""
|
|
Integration test: hit MPC STAC for NAIP around Union Market (DC),
|
|
load imagery via odc-stac, and save an RGB JPEG.
|
|
|
|
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.
|
|
lat = 38.90789
|
|
lon = -76.99831
|
|
|
|
# ~0.0001 degrees buffer in each direction
|
|
aoi = box(lon - 0.0001, lat - 0.0001, lon + 0.0001, lat + 0.0001)
|
|
aoi_geojson = mapping(aoi)
|
|
aoi_feature = Feature(type="Feature", geometry=aoi_geojson, properties={})
|
|
tool_call = ToolCall(
|
|
name="fetch_naip_img",
|
|
args={
|
|
"start_date": "2021-01-01",
|
|
"end_date": "2021-12-31",
|
|
"state": GeoAssistantState(search_area=aoi_feature, messages=[]),
|
|
},
|
|
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)
|
|
assert "naip_img_bytes" in result.update
|
|
assert result.update["naip_img_bytes"] is not None, "Expected JPEG bytes in result"
|
|
assert isinstance(result.update["naip_img_bytes"], str)
|
|
assert len(result.update["naip_img_bytes"]) > 1, "Expected non-empty JPEG bytes"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_naip_too_large():
|
|
"""
|
|
Integration test: request a larger AOI that should produce an image
|
|
exceeding the 512x512 pixel limit. The tool should return no image
|
|
bytes and include a message indicating it skipped output due to size.
|
|
|
|
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.
|
|
lat = 38.90789
|
|
lon = -76.99831
|
|
|
|
# ~0.003 degrees buffer in each direction
|
|
aoi = box(lon - 0.003, lat - 0.003, lon + 0.003, lat + 0.003)
|
|
aoi_geojson = mapping(aoi)
|
|
aoi_feature = Feature(type="Feature", geometry=aoi_geojson, properties={})
|
|
tool_call = ToolCall(
|
|
name="fetch_naip_img",
|
|
args={
|
|
"start_date": "2021-01-01",
|
|
"end_date": "2021-12-31",
|
|
"state": GeoAssistantState(search_area=aoi_feature, messages=[]),
|
|
},
|
|
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)
|
|
assert "naip_img_bytes" in result.update
|
|
assert result.update["naip_img_bytes"] is None, "Expected no JPEG bytes in result"
|
|
assert isinstance(result.update["naip_img_bytes"], NoneType)
|