def test_get_surround_idxs_cube():
"""
CubeGridRemap.get_surround_idxs(): ne=30
"""
from cube_remap import CubeGridRemap
from util.convert_coord.cs_ll import abp2latlon
ne, ngq = 30, 4
rotated = False
cube = CubeGridRemap(ne, ngq, rotated)
td = (np.pi / 2) / ne / 3 / 2 # tiny delta
ij = (1, 2, 1, 1, 1)
gid = cube.ij2gid[ij]
alpha, beta = cube.alpha_betas[gid]
lat, lon = abp2latlon(alpha + td, beta + td, ij[0])
ret_uids = cube.get_surround_idxs(lat, lon)
a_equal(ret_uids, [3, 16, 7, 19])
ij = (1, 2, 1, 2, 3)
gid = cube.ij2gid[ij]
alpha, beta = cube.alpha_betas[gid]
lat, lon = abp2latlon(alpha + td, beta + td, ij[0])
ret_uids = cube.get_surround_idxs(lat, lon)
a_equal(ret_uids, [22, 23, 25, 26])
"""
