Method test_reset_coords

xarray/tests/test_dataarray.py:1615–1676

(self)

        assert_identical(expected, actual)

    def test_reset_coords(self) -> None:
        data = DataArray(
            np.zeros((3, 4)),
            {"bar": ("x", ["a", "b", "c"]), "baz": ("y", range(4)), "y": range(4)},
            dims=["x", "y"],
            name="foo",
        )

        actual1 = data.reset_coords()
        expected1 = Dataset(
            {
                "foo": (["x", "y"], np.zeros((3, 4))),
                "bar": ("x", ["a", "b", "c"]),
                "baz": ("y", range(4)),
                "y": range(4),
            }
        )
        assert_identical(actual1, expected1)

        actual2 = data.reset_coords(["bar", "baz"])
        assert_identical(actual2, expected1)

        actual3 = data.reset_coords("bar")
        expected3 = Dataset(
            {"foo": (["x", "y"], np.zeros((3, 4))), "bar": ("x", ["a", "b", "c"])},
            {"baz": ("y", range(4)), "y": range(4)},
        )
        assert_identical(actual3, expected3)

        actual4 = data.reset_coords(["bar"])
        assert_identical(actual4, expected3)

        actual5 = data.reset_coords(drop=True)
        expected5 = DataArray(
            np.zeros((3, 4)), coords={"y": range(4)}, dims=["x", "y"], name="foo"
        )
        assert_identical(actual5, expected5)

        actual6 = data.copy().reset_coords(drop=True)
        assert_identical(actual6, expected5)

        actual7 = data.reset_coords("bar", drop=True)
        expected7 = DataArray(
            np.zeros((3, 4)),
            {"baz": ("y", range(4)), "y": range(4)},
            dims=["x", "y"],
            name="foo",
        )
        assert_identical(actual7, expected7)

        with pytest.raises(ValueError, match=r"cannot be found"):
            data.reset_coords("foo", drop=True)
        with pytest.raises(ValueError, match=r"cannot be found"):
            data.reset_coords("not_found")
        with pytest.raises(ValueError, match=r"cannot remove index"):
            data.reset_coords("y")

        # non-dimension index coordinate