85 lines
3.0 KiB
Rust
85 lines
3.0 KiB
Rust
use wifi_densepose_geo::*;
|
|
use wifi_densepose_geo::coord;
|
|
|
|
#[test]
|
|
fn test_haversine() {
|
|
let toronto = GeoPoint { lat: 43.6532, lon: -79.3832, alt: 0.0 };
|
|
let ottawa = GeoPoint { lat: 45.4215, lon: -75.6972, alt: 0.0 };
|
|
let dist = coord::haversine(&toronto, &ottawa);
|
|
assert!((dist - 353_000.0).abs() < 5_000.0, "Toronto-Ottawa ~353km, got {:.0}m", dist);
|
|
}
|
|
|
|
#[test]
|
|
fn test_wgs84_to_enu() {
|
|
let origin = GeoPoint { lat: 43.0, lon: -79.0, alt: 100.0 };
|
|
let point = GeoPoint { lat: 43.001, lon: -79.0, alt: 100.0 };
|
|
let enu = coord::wgs84_to_enu(&point, &origin);
|
|
assert!((enu[1] - 111.0).abs() < 5.0, "0.001 deg lat ~111m north, got {:.1}m", enu[1]);
|
|
assert!(enu[0].abs() < 1.0, "same longitude should have ~0 east, got {:.1}m", enu[0]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_enu_roundtrip() {
|
|
let origin = GeoPoint { lat: 43.6532, lon: -79.3832, alt: 76.0 };
|
|
let local = [100.0, 200.0, 5.0]; // 100m east, 200m north, 5m up
|
|
let geo = coord::enu_to_wgs84(&local, &origin);
|
|
let back = coord::wgs84_to_enu(&geo, &origin);
|
|
assert!((back[0] - local[0]).abs() < 0.01);
|
|
assert!((back[1] - local[1]).abs() < 0.01);
|
|
assert!((back[2] - local[2]).abs() < 0.01);
|
|
}
|
|
|
|
#[test]
|
|
fn test_tile_coords() {
|
|
let tile = coord::wgs84_to_tile(43.6532, -79.3832, 16);
|
|
assert!(tile.x > 0 && tile.y > 0);
|
|
assert_eq!(tile.z, 16);
|
|
let bounds = coord::tile_bounds(&tile);
|
|
assert!(bounds.south < 43.66 && bounds.north > 43.64);
|
|
}
|
|
|
|
#[test]
|
|
fn test_tiles_for_bbox() {
|
|
let bbox = GeoBBox::from_center(
|
|
&GeoPoint { lat: 43.6532, lon: -79.3832, alt: 0.0 },
|
|
500.0,
|
|
);
|
|
let tiles = coord::tiles_for_bbox(&bbox, 16);
|
|
assert!(tiles.len() >= 4 && tiles.len() <= 25, "500m radius should need 4-25 tiles, got {}", tiles.len());
|
|
}
|
|
|
|
#[test]
|
|
fn test_geo_bbox_from_center() {
|
|
let center = GeoPoint { lat: 43.0, lon: -79.0, alt: 0.0 };
|
|
let bbox = GeoBBox::from_center(¢er, 1000.0);
|
|
assert!(bbox.south < 43.0 && bbox.north > 43.0);
|
|
assert!(bbox.west < -79.0 && bbox.east > -79.0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_hgt_parse() {
|
|
// Create minimal 3x3 HGT data (big-endian i16)
|
|
let mut data = Vec::new();
|
|
for h in [100i16, 110, 120, 105, 115, 125, 110, 120, 130] {
|
|
data.extend_from_slice(&h.to_be_bytes());
|
|
}
|
|
let grid = wifi_densepose_geo::terrain::parse_hgt(&data, 43.0, -79.0).unwrap();
|
|
assert_eq!(grid.heights[0], 100.0);
|
|
assert_eq!(grid.heights[4], 115.0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_registration() {
|
|
let origin = GeoPoint { lat: 43.6532, lon: -79.3832, alt: 76.0 };
|
|
let reg = wifi_densepose_geo::register::auto_register(&origin);
|
|
|
|
let local = [10.0f32, 0.0, 20.0]; // 10m east, 20m forward
|
|
let geo = wifi_densepose_geo::register::local_to_wgs84(®, &local);
|
|
assert!((geo.lat - origin.lat).abs() < 0.001);
|
|
assert!((geo.lon - origin.lon).abs() < 0.001);
|
|
|
|
let back = wifi_densepose_geo::register::wgs84_to_local(®, &geo);
|
|
assert!((back[0] - local[0]).abs() < 0.1);
|
|
assert!((back[2] - local[2]).abs() < 0.1);
|
|
}
|