def test_2d_coordinates_ij_default(self): grid = Rectilinear([1.0, 2.0, 4.0, 8.0], [1.0, 2.0, 3.0], indexing="ij") self.assertArrayEqual( grid.get_point_coordinates(axis=0), np.array([1.0, 1.0, 1.0, 2.0, 2.0, 2.0, 4.0, 4.0, 4.0, 8.0, 8.0, 8.0]), ) self.assertArrayEqual( grid.get_point_coordinates(axis=1), np.array([1.0, 2.0, 3.0] * 4) )
def test_2d_coordinates_xy_default(self): grid = Rectilinear([1.0, 2.0, 4.0, 8.0], [1.0, 2.0, 3.0], indexing="xy") self.assertTrue( np.allclose( grid.get_point_coordinates(axis=0), np.array([1.0, 2.0, 4.0, 8.0] * 3) ) ) self.assertTrue( np.allclose( grid.get_point_coordinates(axis=1), np.tile([1.0, 2.0, 3.0], (4, 1)).T.flat, ) )
def test_2d_coordinates_ij_indexing_with_kwds(self): grid = Rectilinear([1.0, 2.0, 4.0, 8.0], [1.0, 2.0, 3.0], indexing="ij") # self.assertTrue (np.allclose (grid.get_point_coordinates (axis=0, indexing='xy'), # np.array ([1., 2., 3.] * 4))) # self.assertTrue (np.allclose (grid.get_point_coordinates (axis=1, indexing='xy'), # np.tile ([1., 2., 4., 8.], (3, 1)).T.flat)) self.assertTrue( np.allclose( grid.get_point_coordinates(axis=0, indexing="ij"), np.tile([1.0, 2.0, 4.0, 8.0], (3, 1)).T.flat, ) ) self.assertTrue( np.allclose( grid.get_point_coordinates(axis=1, indexing="ij"), np.array([1.0, 2.0, 3.0] * 4), ) )