def test_neighbours(bpc, prec):
for i_ in range(100):
code = encode(rand_lng(), rand_lat(), bits_per_char=bpc, precision=prec)
lng, lat, lng_err, lat_err = decode_exactly(code, bits_per_char=bpc)
neighbours = utils.neighbours(code, bpc)
directions = {'north', 'north-east', 'north-west', 'east',
'west', 'south', 'south-east', 'south-west'}
assert directions == set(neighbours.keys())
# no duplicates (depends on level)
assert len(neighbours) == len(set(neighbours.values()))
for k, v in neighbours.items():
n_lng, n_lat, n_lng_err, n_lat_err = decode_exactly(v, bits_per_char=bpc)
# same level
assert len(code) == len(v)
assert lng_err == n_lng_err
assert lat_err == n_lat_err
# neighbour is in disc 4x error
assert (lng == n_lng or # east / west
lng - 2 * lng_err == n_lng or lng - 2 * lng_err + 360 == n_lng or # south
lng + 2 * lng_err == n_lng or lng + 2 * lng_err - 360 == n_lng) # north
assert (lat == n_lat or # north / south
lat - 2 * lat_err == n_lat or lat - 2 * lat_err + 180 == n_lat or # west
lat + 2 * lat_err == n_lat or lat + 2 * lat_err - 180 == n_lat) # east
def test_rectangle(bpc, prec):
code = encode(rand_lng(), rand_lat(), bits_per_char=bpc, precision=prec)
lng, lat, lng_err, lat_err = decode_exactly(code, bits_per_char=bpc)
rect = utils.rectangle(code, bits_per_char=bpc)
assert isinstance(rect, dict)
assert rect['type'] == 'Feature'
assert rect['geometry']['type'] == 'Polygon'
assert rect['bbox'] == (lng - lng_err, lat - lat_err, lng + lng_err,
lat + lat_err)
assert rect['properties'] == dict(
code=code,
lng=lng,
lat=lat,
lng_err=lng_err,
lat_err=lat_err,
bits_per_char=bpc,
)
coords = rect['geometry']['coordinates']
assert 1 == len(coords) # one external ring
assert 5 == len(coords[0]) # rectangle has 5 coordinates
# ccw
assert (lng - lng_err, lat - lat_err) == coords[0][0] # lower left
assert (lng + lng_err, lat - lat_err) == coords[0][1] # lower right
assert (lng + lng_err, lat + lat_err) == coords[0][2] # upper right
assert (lng - lng_err, lat + lat_err) == coords[0][3] # upper left
assert (lng - lng_err, lat - lat_err) == coords[0][4] # lower left