def test_get_neighbors_latlon():
'''
LatlonGridRemap.get_neighbors(): nlat=180, nlon=360 (regular)
'''
from cube_remap import LatlonGridRemap
nlat, nlon = 180, 360
ll = LatlonGridRemap(nlat, nlon, 'regular')
ret = ll.get_neighbors(nlon+1)
expect = (0, 1, 2, nlon, nlon+2, 2*nlon, 2*nlon+1, 2*nlon+2)
a_equal(expect, ret)
ret = ll.get_neighbors(nlon)
expect = (nlon-1, 0, 1, 2*nlon-1, nlon+1, 3*nlon-1, 2*nlon, 2*nlon+1)
a_equal(expect, ret)
ret = ll.get_neighbors(1)
expect = (0, 2, nlon, nlon+1, nlon+2)
a_equal(expect, ret)
ret = ll.get_neighbors(0)
expect = (nlon-1, 1, 2*nlon-1, nlon, nlon+1)
a_equal(expect, ret)
ret = ll.get_neighbors((nlat-1)*nlon)
expect = ((nlat-1)*nlon-1, (nlat-2)*nlon, (nlat-2)*nlon+1, nlat*nlon-1, (nlat-1)*nlon+1)
a_equal(expect, ret)
