29 lines
1.6 KiB
Python
29 lines
1.6 KiB
Python
import json,re
|
|
from pathlib import Path
|
|
r=Path(__file__).resolve().parents[1]
|
|
states=[line.split('\t') for line in (r/'data/states.tsv').read_text().splitlines()]
|
|
assert len(states)==50 and len({s[0] for s in states})==50
|
|
assert len({s[1] for s in states})==50 and len({s[2] for s in states})==50
|
|
assert dict((s[0],s[2]) for s in states)['AK']=='Juneau'
|
|
assert dict((s[0],s[2]) for s in states)['HI']=='Honolulu'
|
|
assert all(s[0] not in ('DC','PR') for s in states)
|
|
m=json.loads((r/'data/map-manifest.json').read_text())
|
|
s=(r/'src/map_data.c').read_text()
|
|
for view in ('national','northeast','hawaii'):
|
|
data=list(map(int,re.search(view+r'_spans\[\] = \{\s*([\d,\s]+)',s)[1].replace('\n','').split(',')))
|
|
off=list(map(int,re.search(view+r'_offsets\[51\] = \{([\d,]+)',s)[1].split(',')))
|
|
assert len(off)==51 and off[0]==0 and off[-1]==len(data)
|
|
w,h=m[view]['size']; occupied=set()
|
|
for i,state in enumerate(states):
|
|
area=0
|
|
for k in range(off[i],off[i+1],3):
|
|
y,x,n=data[k:k+3];assert 0<=y<h and 0<=x<x+n<=w
|
|
for xx in range(x,x+n):assert (xx,y) not in occupied;occupied.add((xx,y))
|
|
area+=n
|
|
assert area==m[view]['pixels'][state[0]]
|
|
if view=='national':assert area>0
|
|
for code in ('CT','DE','ME','MD','MA','NH','NJ','NY','PA','RI','VT'):assert m['northeast']['pixels'][code]>0
|
|
assert m['hawaii']['pixels']['HI']>400
|
|
assert all(n==0 for code,n in m['hawaii']['pixels'].items() if code!='HI')
|
|
print('PASS: 50 unique pairs and geometry IDs, bounded/disjoint spans, all national masks, Northeast and Hawaii detail coverage.')
|