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),
)
)