diff --git a/engine_test.py b/engine_test.py index 8441ef9..70a7b38 100755 --- a/engine_test.py +++ b/engine_test.py @@ -1,6 +1,6 @@ #!/usr/bin/env python -from engine import encode_greggs, decode_greggs, 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, format_dist, parse_dist from tqdm import tqdm import numpy as np import time @@ -62,8 +62,11 @@ def test_encode_decode_greggs(): encoded = encode_greggs(i) # tqdm.write(f"Encoding took {time.monotonic() - t:.3f}s") + stringified = ",".join(map(format_dist, encoded)) + parsed = [parse_dist(s.strip()) for s in stringified.split(",")] + # t = time.monotonic() - decoded = decode_greggs(encoded) + decoded = decode_greggs(parsed) # tqdm.write(f"Decoding took {time.monotonic() - t:.3f}s") if i != decoded: