mirror of
https://github.com/azaion/gps-denied-onboard.git
synced 2026-04-22 21:46:36 +00:00
Yuzviak
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
"""Tests for SatelliteDataManager (F04) and mercator utils (H06)."""
|
|
|
|
import asyncio
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from gps_denied.core.satellite import SatelliteDataManager
|
|
from gps_denied.schemas import GPSPoint
|
|
from gps_denied.utils import mercator
|
|
|
|
|
|
def test_latlon_to_tile():
|
|
# Kyiv coordinates
|
|
lat = 50.4501
|
|
lon = 30.5234
|
|
zoom = 15
|
|
|
|
coords = mercator.latlon_to_tile(lat, lon, zoom)
|
|
assert coords.zoom == 15
|
|
assert coords.x > 0
|
|
assert coords.y > 0
|
|
|
|
|
|
def test_tile_to_latlon():
|
|
x, y, zoom = 19131, 10927, 15
|
|
gps = mercator.tile_to_latlon(x, y, zoom)
|
|
|
|
assert 50.0 < gps.lat < 52.0
|
|
assert 30.0 < gps.lon < 31.0
|
|
|
|
|
|
def test_tile_bounds():
|
|
coords = mercator.TileCoords(x=19131, y=10927, zoom=15)
|
|
bounds = mercator.compute_tile_bounds(coords)
|
|
|
|
# Northwest should be "higher" lat and "lower" lon than Southeast
|
|
assert bounds.nw.lat > bounds.se.lat
|
|
assert bounds.nw.lon < bounds.se.lon
|
|
assert bounds.gsd > 0
|
|
|
|
|
|
@pytest.fixture
|
|
def satellite_manager(tmp_path):
|
|
# Use tmp_path for cache so we don't pollute workspace
|
|
sm = SatelliteDataManager(cache_dir=str(tmp_path / "cache"), max_size_gb=0.1)
|
|
yield sm
|
|
sm.cache.close()
|
|
asyncio.run(sm.http_client.aclose())
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_satellite_fetch_and_cache(satellite_manager):
|
|
lat = 48.0
|
|
lon = 37.0
|
|
zoom = 12
|
|
flight_id = "test_flight"
|
|
|
|
# We won't test the actual HTTP Google API in CI to avoid blocks/bans,
|
|
# but we can test the cache mechanism directly.
|
|
coords = satellite_manager.compute_tile_coords(lat, lon, zoom)
|
|
|
|
# Create a fake image (blue square 256x256)
|
|
fake_img = np.zeros((256, 256, 3), dtype=np.uint8)
|
|
fake_img[:] = [255, 0, 0] # BGR
|
|
|
|
# Save to cache
|
|
success = satellite_manager.cache_tile(flight_id, coords, fake_img)
|
|
assert success is True
|
|
|
|
# Read from cache
|
|
cached = satellite_manager.get_cached_tile(flight_id, coords)
|
|
assert cached is not None
|
|
assert cached.shape == (256, 256, 3)
|
|
|
|
# Clear cache
|
|
satellite_manager.clear_flight_cache(flight_id)
|
|
assert satellite_manager.get_cached_tile(flight_id, coords) is None
|
|
|
|
|
|
def test_grid_calculations(satellite_manager):
|
|
# Test 3x3 grid (9 tiles)
|
|
center = mercator.TileCoords(x=100, y=100, zoom=15)
|
|
grid = satellite_manager.get_tile_grid(center, 9)
|
|
assert len(grid) == 9
|
|
|
|
# Ensure center is in grid
|
|
assert any(c.x == 100 and c.y == 100 for c in grid)
|
|
|
|
# Test expansion 9 -> 25
|
|
new_tiles = satellite_manager.expand_search_grid(center, 9, 25)
|
|
assert len(new_tiles) == 16 # 25 - 9
|