Function test_concat_multi_dim_index

xarray/tests/test_concat.py:1594–1638

()

def test_concat_multi_dim_index() -> None:
    ds1 = (
        Dataset(
            {"foo": (("x", "y"), np.random.randn(2, 2))},
            coords={"x": [1, 2], "y": [3, 4]},
        )
        .drop_indexes(["x", "y"])
        .set_xindex(["x", "y"], XYIndex)
    )
    ds2 = (
        Dataset(
            {"foo": (("x", "y"), np.random.randn(2, 2))},
            coords={"x": [1, 2], "y": [5, 6]},
        )
        .drop_indexes(["x", "y"])
        .set_xindex(["x", "y"], XYIndex)
    )

    expected = (
        Dataset(
            {
                "foo": (
                    ("x", "y"),
                    np.concatenate([ds1.foo.data, ds2.foo.data], axis=-1),
                )
            },
            coords={"x": [1, 2], "y": [3, 4, 5, 6]},
        )
        .drop_indexes(["x", "y"])
        .set_xindex(["x", "y"], XYIndex)
    )
    # note: missing 'override'
    joins: list[types.JoinOptions] = ["inner", "outer", "exact", "left", "right"]
    for join in joins:
        actual = concat([ds1, ds2], dim="y", join=join)
        assert_identical(actual, expected, check_default_indexes=False)

    with pytest.raises(AlignmentError):
        actual = concat([ds1, ds2], dim="x", join="exact")

    # TODO: fix these, or raise better error message
    with pytest.raises(AssertionError):
        joins_lr: list[types.JoinOptions] = ["left", "right"]
        for join in joins_lr:
            actual = concat([ds1, ds2], dim="x", join=join)


class TestConcatDataTree: