def test_3d_grid_xy_indexing(self): grid = UniformRectilinear((4, 3, 2), (1, 2, 3), (-1, 0, 1), indexing='xy') self.assert_point_count(grid, 24) self.assert_cell_count(grid, 6) self.assert_shape(grid, (2, 3, 4)) self.assert_spacing(grid, (3., 2., 1.)) self.assert_origin(grid, (1., 0., -1.)) self.assert_x(grid, [ -1., 0., 1., 2., -1., 0., 1., 2., -1., 0., 1., 2., -1., 0., 1., 2., -1., 0., 1., 2., -1., 0., 1., 2. ]) self.assert_y(grid, [ 0., 0., 0., 0., 2., 2., 2., 2., 4., 4., 4., 4., 0., 0., 0., 0., 2., 2., 2., 2., 4., 4., 4., 4. ]) self.assert_z(grid, [ 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4. ]) self.assert_offset(grid, 8. * np.arange(1, grid.get_cell_count() + 1))
def test_3d_grid_ij_indexing(self): grid = UniformRectilinear((4, 3, 2), (1, 2, 3), (-1, 0, 1), indexing="ij") self.assert_point_count(grid, 24) self.assert_cell_count(grid, 6) self.assert_shape(grid, (4, 3, 2)) self.assert_spacing(grid, (1.0, 2.0, 3.0)) self.assert_origin(grid, (-1.0, 0.0, 1.0)) self.assert_x( grid, [ 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, 1.0, 4.0, ], ) self.assert_y( grid, [ 0.0, 0.0, 2.0, 2.0, 4.0, 4.0, 0.0, 0.0, 2.0, 2.0, 4.0, 4.0, 0.0, 0.0, 2.0, 2.0, 4.0, 4.0, 0.0, 0.0, 2.0, 2.0, 4.0, 4.0, ], ) self.assert_z( grid, [ -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, ], ) self.assert_offset(grid, 8.0 * np.arange(1, grid.get_cell_count() + 1))
def test_3d_grid_xy_indexing(self): grid = UniformRectilinear((4, 3, 2), (1, 2, 3), (-1, 0, 1), indexing="xy") self.assert_point_count(grid, 24) self.assert_cell_count(grid, 6) self.assert_shape(grid, (2, 3, 4)) self.assert_spacing(grid, (3.0, 2.0, 1.0)) self.assert_origin(grid, (1.0, 0.0, -1.0)) self.assert_x( grid, [ -1.0, 0.0, 1.0, 2.0, -1.0, 0.0, 1.0, 2.0, -1.0, 0.0, 1.0, 2.0, -1.0, 0.0, 1.0, 2.0, -1.0, 0.0, 1.0, 2.0, -1.0, 0.0, 1.0, 2.0, ], ) self.assert_y( grid, [ 0.0, 0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 4.0, 4.0, 4.0, 4.0, 0.0, 0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 2.0, 4.0, 4.0, 4.0, 4.0, ], ) self.assert_z( grid, [ 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, ], ) self.assert_offset(grid, 8.0 * np.arange(1, grid.get_cell_count() + 1))