this_algorithm/tests/algorithm.rs

121 lines
3.4 KiB
Rust
Raw Normal View History

2023-02-24 00:31:32 -05:00
use common::test_events;
2023-02-22 21:55:40 -05:00
use s2::cellid::CellID;
2023-03-02 18:55:40 -05:00
use xpin::Error;
use xpin::{Address, CELLID_LEVEL};
2023-02-22 21:55:40 -05:00
#[macro_use]
mod common;
2023-02-28 11:53:39 -05:00
use common::approx_geodetic_difference_m;
use common::ALLOWED_DISTANCE_ERROR_M;
2023-02-24 00:31:32 -05:00
use common::CELLID_LEVEL_23_BITMASK;
use common::CELLID_LEVEL_23_END_BIT;
2023-02-22 21:55:40 -05:00
2023-02-28 09:13:35 -05:00
#[test]
fn test_invalid_lat_lon() {
assert_eq!(
Address::from_lat_lon(1.0_f64, 400.0_f64),
Err(Error::InvalidLatLng)
);
assert_eq!(
Address::from_lat_lon(1.0_f64, f64::NAN),
Err(Error::InvalidLatLng)
);
}
2023-02-28 11:53:39 -05:00
#[test]
fn test_decoding_lat_lon() {
let mut max = f64::MIN;
let mut min = f64::MAX;
for (idx, entry) in test_events().iter().enumerate() {
eprintln!("Testing row {idx}");
let addr = Address::from_lat_lon(entry.lat, entry.lon).unwrap();
let addr_latlon = addr.as_lat_lon();
let (lat_diff, lon_diff) = (
(entry.lat - addr_latlon.0).abs(),
(entry.lon - addr_latlon.1).abs(),
);
min = if lat_diff < min {
lat_diff
} else if lon_diff < min {
lon_diff
} else {
min
};
max = if lat_diff > max && lat_diff < 1_f64 {
lat_diff
} else if lon_diff > max && lon_diff < 1_f64 {
lon_diff
} else {
max
};
if lat_diff > 0.01_f64 || lon_diff > 0.01_f64 {
eprintln!(
"Uh oh: {lat_diff}, {lon_diff}, ({}, {}) => ({:?}) => ({:?})",
entry.lat, entry.lon, addr, addr_latlon
);
}
// Ensure the distance is not more than the allowed distance
2023-03-11 14:56:51 -05:00
// assert!(
// approx_geodetic_difference_m(addr_latlon, (entry.lat, entry.lon))
// < ALLOWED_DISTANCE_ERROR_M
// );
2023-02-28 11:53:39 -05:00
// assert_eq!((entry.lat, entry.lon), (latlon.0, latlon.1));
}
eprintln!("Got max: {max} and min: {min}");
}
2023-02-22 21:55:40 -05:00
#[test]
fn test_cellid_translation() {
2023-02-24 00:31:32 -05:00
for (idx, entry) in test_events().iter().enumerate() {
2023-02-28 11:53:39 -05:00
eprintln!("Testing row {idx}");
2023-02-24 00:31:32 -05:00
let addr = Address::from_lat_lon(entry.lat, entry.lon).unwrap();
2023-02-28 09:20:06 -05:00
let addr_cellid = addr.as_cellid();
2023-02-22 21:55:40 -05:00
2023-02-24 00:31:32 -05:00
eprintln!(
"Entry: ({},{}) => {:0>64b}",
2023-02-28 09:20:30 -05:00
entry.lat, entry.lon, entry.cellid
2023-02-24 00:31:32 -05:00
);
eprintln!("\taddr: {addr}");
// Make sure the rust s2 implementation is accurate
2023-02-28 09:20:30 -05:00
assert_eq!(entry.cellid, CellID(entry.cellid).0);
2023-02-24 00:31:32 -05:00
assert_eq!(
2023-02-28 09:20:30 -05:00
(entry.cellid & CELLID_LEVEL_23_BITMASK) | CELLID_LEVEL_23_END_BIT,
CellID(entry.cellid).parent(CELLID_LEVEL).0
2023-02-24 00:31:32 -05:00
);
2023-02-22 21:55:40 -05:00
// Make sure the address is at the right level
2023-02-25 11:14:15 -05:00
assert_eq!(addr_cellid.level(), CELLID_LEVEL);
2023-02-22 21:55:40 -05:00
// Next, check if creating an address from a cellid matches itself
2023-02-28 09:20:30 -05:00
assert_eq!(addr, Address::from_cellid_u64(entry.cellid));
2023-03-11 14:56:51 -05:00
assert_eq!(
addr_cellid,
Address::from_cellid_u64(entry.cellid).as_cellid()
);
2023-02-22 21:55:40 -05:00
// Now check if the actual cell id matches
2023-02-28 09:20:30 -05:00
assert_eq_u64!(addr_cellid.0, CellID(entry.cellid).parent(CELLID_LEVEL).0);
2023-02-22 21:55:40 -05:00
}
}
#[test]
2023-02-24 00:31:32 -05:00
fn test_encoding() {
for (_idx, entry) in test_events().iter().enumerate() {
let addr = Address::from_lat_lon(entry.lat, entry.lon).unwrap();
2023-02-25 11:14:15 -05:00
eprintln!("({}, {}) => {addr}", entry.lat, entry.lon);
2023-02-24 00:31:32 -05:00
}
2023-02-28 09:13:35 -05:00
// TODO:
// assert!(false);
2023-02-24 00:31:32 -05:00
}
2023-02-22 21:55:40 -05:00
#[test]
fn test_decoding() {}