def test_reshape(self): lat = np.linspace(0, 1, 12).reshape((3, 4)) lon = np.linspace(10, 20, 12).reshape((3, 4)) c = StackedCoordinates([lat, lon]) assert c.reshape((4, 3)) == StackedCoordinates([lat.reshape((4, 3)), lon.reshape((4, 3))]) assert c.flatten().reshape((3, 4)) == c
def test_unque_shaped(self): lat = ArrayCoordinates1d([[0, 1, 2], [1, 0, 5]], name="lat") lon = ArrayCoordinates1d([[10, 20, 20], [20, 10, 60]], name="lon") c = StackedCoordinates([lat, lon]) # flattens c2 = c.unique() assert_equal(c2["lat"].coordinates, [0, 1, 2, 5]) assert_equal(c2["lon"].coordinates, [10, 20, 20, 60]) c2, I = c.unique(return_index=True) assert_equal(c2["lat"].coordinates, [0, 1, 2, 5]) assert_equal(c2["lon"].coordinates, [10, 20, 20, 60]) assert c.flatten()[I] == c2