def test_get_interpolated_value_gc_coord_outside(self):
grid1 = grids.Grid(origin=[1.0, 2.0],
basis=[[1.0, 0.0], [1.0, 1.0]],
ranges=[[-3, 4], [-2, 8]])
data1 = np.arange(7 * 10).reshape((7, 10))
griddata = grids.GridData(grid1, data1)
gridcoords = np.array([[-3, -2], [3, 8]])
data = griddata.get_interpolated_value_gc(gridcoords)
true_data = np.array([0, 0])
self.assertTrue(np.all(data == true_data))
def test_get_subgrid_dataview_modifying_data(self):
grid1 = grids.Grid(origin=[1.0, 2.0],
basis=[[1.0, 0.0], [1.0, 1.0]],
ranges=[[-3, 4], [-2, 8]])
data1 = np.arange(7 * 10).reshape((7, 10))
griddata = grids.GridData(grid1, data1)
grid2 = grids.Grid(origin=[4.0, 3.0],
basis=[[1.0, 0.0], [1.0, 1.0]],
ranges=[[-2, 2], [-2, 2]])
data2 = griddata.get_subgrid_dataview(grid2)
data2 *= 2
diff = np.max(np.abs(data2 - griddata.data[3:7, 1:5]))
self.assertLess(diff, FLOAT_TOLERANCE)
def test_get_interpolated_value_cartesian(self):
grid1 = grids.Grid(origin=[1.0, 2.0],
basis=[[1.0, 0.0], [1.0, 1.0]],
ranges=[[-3, 4], [-2, 8]])
data1 = np.arange(7 * 10).reshape((7, 10))
griddata = grids.GridData(grid1, data1)
cartesians = grid1.gridcoord_to_cartesian([[-3, -2], [3, 8]])
# Perturb cartesians. Obtained value must be the same as we would have
# get for the unperturbed ones.
cartesians += 0.1
data = griddata.get_interpolated_value(cartesians,
coordtype=grids.CARTESIAN_COORD)
true_data = np.array([0, 0])
self.assertTrue(np.all(data == true_data))
