def test_raster_field_3d (self): g = RasterField ((4, 3, 2), (1.5, 1., 3), (0.5, 0, -.5 ), indexing='ij') self.assert_point_count (g, 24) self.assert_cell_count (g, 6) g.add_field ('Point Data', g.get_x (), centering='point') cell_data = np.arange (6.)*10. g.add_field ('Cell Data', cell_data, centering='zonal') g.get_field ('Cell Data') self.assert_field_values (g, 'Cell Data', [0., 10., 20., 30., 40., 50.]) self.assert_field_values (g, 'Point Data', [-0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5]) self.assert_shape (g, (4, 3, 2)) x = g.get_x () data = g.get_field ('Point Data') self.assertEqual (x.size, data.size)
def test_raster_field_3d(self): g = RasterField((4, 3, 2), (1.5, 1.0, 3), (0.5, 0, -0.5), indexing="ij") self.assert_point_count(g, 24) self.assert_cell_count(g, 6) g.add_field("Point Data", g.get_x(), centering="point") cell_data = np.arange(6.0) * 10.0 g.add_field("Cell Data", cell_data, centering="zonal") g.get_field("Cell Data") self.assert_field_values(g, "Cell Data", [0.0, 10.0, 20.0, 30.0, 40.0, 50.0]) self.assert_field_values( g, "Point Data", [ -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, -0.5, 2.5, ], ) self.assert_shape(g, (4, 3, 2)) x = g.get_x() data = g.get_field("Point Data") self.assertEqual(x.size, data.size)