Method test_stacking_stacked

xarray/tests/test_units.py:3636–3660

(self, func, dtype)

        ids=repr,
    )
    def test_stacking_stacked(self, func, dtype):
        array = (
            np.linspace(0, 10, 5 * 10).reshape(5, 10).astype(dtype) * unit_registry.m
        )
        x = np.arange(array.shape[0])
        y = np.arange(array.shape[1])

        data_array = xr.DataArray(
            name="data", data=array, coords={"x": x, "y": y}, dims=("x", "y")
        )
        stacked = data_array.stack(z=("x", "y"))

        expected = attach_units(func(strip_units(stacked)), {"data": unit_registry.m})
        actual = func(stacked)

        assert_units_equal(expected, actual)
        # TODO: strip_units/attach_units reconstruct DataArrays from scratch,
        # losing index structure (e.g., MultiIndex from stack becomes regular Index).
        # Fix these utilities to preserve indexes, then remove check_indexes=False.
        if func.name == "reset_index":
            assert_identical(
                expected, actual, check_default_indexes=False, check_indexes=False
            )
        else:
            assert_identical(expected, actual, check_indexes=False)

    @pytest.mark.skip(reason="indexes don't support units")
    def test_to_unstacked_dataset(self, dtype):