charlotte/CamHack25
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

engine_test: improve _greggs test

Browse source
This commit is contained in:
Oliver Gaskell 2025-11-02 15:47:47 +00:00
parent 2edce7c73a
commit 1b280827ea
No known key found for this signature in database
GPG key ID: F971A08925FCC0AD
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
engine_test.py
View file

@ -1,6 +1,6 @@
#!/usr/bin/env python #!/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 from tqdm import tqdm
import numpy as np import numpy as np
import time import time
@ -62,8 +62,11 @@ def test_encode_decode_greggs():
encoded = encode_greggs(i) encoded = encode_greggs(i)
# tqdm.write(f"Encoding took {time.monotonic() - t:.3f}s") # 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() # t = time.monotonic()
decoded = decode_greggs(encoded) decoded = decode_greggs(parsed)
# tqdm.write(f"Decoding took {time.monotonic() - t:.3f}s") # tqdm.write(f"Decoding took {time.monotonic() - t:.3f}s")
if i != decoded: if i != decoded: