Method test_map_coords_attrs

xarray/tests/test_dataset.py:6542–6572

(self)

        assert_equal(expected, actual)

    def test_map_coords_attrs(self) -> None:
        ds = xr.Dataset(
            {
                "a": (
                    ["x", "y", "z"],
                    np.arange(24).reshape(3, 4, 2),
                    {"attr1": "value1"},
                ),
                "b": ("y", np.arange(4), {"attr2": "value2"}),
            },
            coords={
                "x": ("x", np.array([-1, 0, 1]), {"attr3": "value3"}),
                "z": ("z", list("ab"), {"attr4": "value4"}),
            },
        )

        def func(arr):
            if "y" not in arr.dims:
                return arr

            # drop attrs from coords
            return arr.mean(dim="y").drop_attrs()

        expected = ds.mean(dim="y", keep_attrs=True)
        actual = ds.map(func, keep_attrs=True)

        assert_identical(actual, expected)
        assert actual["x"].attrs

        ds["x"].attrs["y"] = "x"
        assert ds["x"].attrs != actual["x"].attrs

    def test_map_non_dataarray_outputs(self) -> None:
        # Test that map handles non-DataArray outputs by converting them