Method test_inherited_join

xarray/tests/test_merge.py:907–940

(self)

        assert_equal(merged, expected)

    def test_inherited_join(self) -> None:
        tree1 = xr.DataTree.from_dict({"/a/b": ("x", [0, 1])}, coords={"x": [0, 1]})
        tree2 = xr.DataTree.from_dict({"/a/c": ("x", [1, 2])}, coords={"x": [1, 2]})

        expected = xr.DataTree.from_dict(
            {"/a/b": ("x", [0, 1]), "a/c": ("x", [np.nan, 1])}, coords={"x": [0, 1]}
        )
        merged = xr.merge([tree1, tree2], join="left")
        assert_equal(merged, expected)

        expected = xr.DataTree.from_dict(
            {"/a/b": ("x", [1, np.nan]), "a/c": ("x", [1, 2])}, coords={"x": [1, 2]}
        )
        merged = xr.merge([tree1, tree2], join="right")
        assert_equal(merged, expected)

        expected = xr.DataTree.from_dict(
            {"/a/b": ("x", [1]), "a/c": ("x", [1])}, coords={"x": [1]}
        )
        merged = xr.merge([tree1, tree2], join="inner")
        assert_equal(merged, expected)

        expected = xr.DataTree.from_dict(
            {"/a/b": ("x", [0, 1, np.nan]), "a/c": ("x", [np.nan, 1, 2])},
            coords={"x": [0, 1, 2]},
        )
        merged = xr.merge([tree1, tree2], join="outer")
        assert_equal(merged, expected)

        with pytest.raises(
            xr.AlignmentError,
            match=re.escape("cannot align objects with join='exact'"),
        ):
            xr.merge([tree1, tree2], join="exact")

    def test_merge_error_includes_path(self) -> None:
        tree1 = xr.DataTree.from_dict({"/a/b": ("x", [0, 1])})