def test_interpolator_2d(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0], lonbounds[1], latbounds[0], latbounds[1], nx=50, ny=50)
data = np.random.rand(50, 50)
i = CfGeoInterpolator(data, lon, lat, method='nearest')
data2 = i.interpgrid(lon, lat)
assert np.all(data==data2)
def test_interpolator_3d_3dz_1dll(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0], lonbounds[1], latbounds[0], latbounds[1], nx=50, ny=50)
data = np.random.rand(10, 50, 50)
z, dummy, dummy2 = np.meshgrid(np.arange(10), lon, lat, indexing='ij')
i = CfGeoInterpolator(data, lon, lat, z=z, method='nearest')
data2 = i.interpgrid(lon, lat, z=z)
assert np.all(data==data2)
def test_interpolator_3d_1dt_1dz_1dll(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0], lonbounds[1], latbounds[0], latbounds[1], nx=50, ny=50)
data = np.random.rand(9, 10, 50, 50)
z = np.arange(10)
t = np.arange(9)
i = CfGeoInterpolator(data, lon, lat, z=z, t=t, method='nearest')
data2 = i.interpgrid(lon, lat, z=z, t=t)
assert np.all(data==data2)
def test_create_grid(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0], lonbounds[1], latbounds[0], latbounds[1], nx=50, ny=50)
assert lon[0] == lonbounds[0]
assert lon[-1] == lonbounds[1]
assert lat[0] == latbounds[0]
assert lat[-1] == latbounds[1]
assert lon.shape[0] == nx
assert lat.shape[0] == ny
def test_interpolator_2d(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0],
lonbounds[1],
latbounds[0],
latbounds[1],
nx=50,
ny=50)
data = np.random.rand(50, 50)
i = CfGeoInterpolator(data, lon, lat, method='nearest')
data2 = i.interpgrid(lon, lat)
assert np.all(data == data2)
def test_interpolator_3d_3dz_1dll(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0],
lonbounds[1],
latbounds[0],
latbounds[1],
nx=50,
ny=50)
data = np.random.rand(10, 50, 50)
z, dummy, dummy2 = np.meshgrid(np.arange(10), lon, lat, indexing='ij')
i = CfGeoInterpolator(data, lon, lat, z=z, method='nearest')
data2 = i.interpgrid(lon, lat, z=z)
assert np.all(data == data2)
def test_create_grid(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0],
lonbounds[1],
latbounds[0],
latbounds[1],
nx=50,
ny=50)
assert lon[0] == lonbounds[0]
assert lon[-1] == lonbounds[1]
assert lat[0] == latbounds[0]
assert lat[-1] == latbounds[1]
assert lon.shape[0] == nx
assert lat.shape[0] == ny
def test_interpolator_3d_1dt_1dz_1dll(self):
lonbounds = [-70, -60]
latbounds = [40, 50]
nx, ny = 50, 50
lon, lat = create_grid(lonbounds[0],
lonbounds[1],
latbounds[0],
latbounds[1],
nx=50,
ny=50)
data = np.random.rand(9, 10, 50, 50)
z = np.arange(10)
t = np.arange(9)
i = CfGeoInterpolator(data, lon, lat, z=z, t=t, method='nearest')
data2 = i.interpgrid(lon, lat, z=z, t=t)
assert np.all(data == data2)
