engine_test: more tests

engine_test.py

@@ -1,8 +1,9 @@
 #!/usr/bin/env python
 
-from engine import encode, decode, format_location, parse_location, spherical_dist, fetch_data
+from engine import encode_greggs, decode_greggs, encode, decode, format_location, parse_location, spherical_dist, fetch_data
 from tqdm import tqdm
 import numpy as np
+import time
 
 DIST_THRESH = 0.1
 
@@ -19,18 +20,32 @@ def test_encode_decode():
     fails = 0
     dists = []
 
+    with open("fails.txt", "w") as f:
+        f.write("lat, lon, guesslat, guesslon\n")
+
+    with open("success.txt", "w") as f:
+        f.write("lat, lon, guesslat, guesslon\n")
+
     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
+                dists.append(dist)
+
+                with open("fails.txt", "a") as f:
+                    f.write(f"{loc[0]}, {loc[1]}, {decoded[0]}, {decoded[1]}\n")
+
             else:
                 passed += 1
+
+                with open("success.txt", "a") as f:
+                    f.write(f"{loc[0]}, {loc[1]}, {decoded[0]}, {decoded[1]}\n")
+
     except KeyboardInterrupt:
         print("KeyboardInterrupt - halting...")
 
@@ -38,5 +53,25 @@ def test_encode_decode():
     print(f"\nDONE. Passed: {passed}. Failed: {fails}. Average error: {avg_dist:.3f}m")
 
 
+def test_encode_decode_greggs():
+    num_greggs = len(fetch_data("greggs"))
+    fails = 0
+
+    for i in tqdm(range(num_greggs)):
+        # t = time.monotonic()
+        encoded = encode_greggs(i)
+        # tqdm.write(f"Encoding took {time.monotonic() - t:.3f}s")
+
+        # t = time.monotonic()
+        decoded = decode_greggs(encoded)
+        # tqdm.write(f"Decoding took {time.monotonic() - t:.3f}s")
+
+        if i != decoded:
+            tqdm.write(f"FAIL: {i} != {decoded}")
+            fails += 1
+
+    print(f"Encode/Decode Greggs: {fails} failures.")
+
+
 if __name__ == "__main__":
     test_encode_decode()