#!/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()