Method test_get_index

xarray/tests/test_dataarray.py:193–198

(self)

            assert array.indexes[k].equals(expected_indexes[k])

    def test_get_index(self) -> None:
        array = DataArray(np.zeros((2, 3)), coords={"x": ["a", "b"]}, dims=["x", "y"])
        assert array.get_index("x").equals(pd.Index(["a", "b"]))
        assert array.get_index("y").equals(pd.Index([0, 1, 2]))
        with pytest.raises(KeyError):
            array.get_index("z")

    def test_get_index_size_zero(self) -> None:
        array = DataArray(np.zeros((0,)), dims=["x"])