42 lines
1.1 KiB
Python
Executable file
42 lines
1.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
from engine import encode, decode, format_location, parse_location, spherical_dist, fetch_data
|
|
from tqdm import tqdm
|
|
import numpy as np
|
|
|
|
DIST_THRESH = 0.1
|
|
|
|
|
|
def gen_locations() -> list[tuple[float, float]]:
|
|
"""Generate a list of locations for testing."""
|
|
# return [(52.210796, 0.091659)]
|
|
return fetch_data("tesco")
|
|
|
|
|
|
def test_encode_decode():
|
|
locations = gen_locations()
|
|
passed = 0
|
|
fails = 0
|
|
dists = []
|
|
|
|
try:
|
|
for loc in tqdm(locations):
|
|
encoded = format_location(encode(loc))
|
|
decoded = decode(parse_location(encoded))
|
|
dist = spherical_dist(np.array(loc), np.array(decoded))
|
|
dists.append(dist)
|
|
|
|
if dist > DIST_THRESH:
|
|
tqdm.write(f"FAIL:\n\tloc: {loc}\n\tdecoded: {decoded}\n\tdist: {dist}")
|
|
fails += 1
|
|
else:
|
|
passed += 1
|
|
except KeyboardInterrupt:
|
|
print("KeyboardInterrupt - halting...")
|
|
|
|
avg_dist = sum(dists) / len(dists)
|
|
print(f"\nDONE. Passed: {passed}. Failed: {fails}. Average error: {avg_dist:.3f}m")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_encode_decode()
|