Method test_coords

xarray/tests/test_dataarray.py:1520–1568

(self)

        assert_array_equal(da.coords["time.dayofyear"], da.values)

    def test_coords(self) -> None:
        # use int64 to ensure repr() consistency on windows
        coords = [
            IndexVariable("x", np.array([-1, -2], "int64")),
            IndexVariable("y", np.array([0, 1, 2], "int64")),
        ]
        da = DataArray(np.random.randn(2, 3), coords, name="foo")

        # len
        assert len(da.coords) == 2

        # iter
        assert list(da.coords) == ["x", "y"]

        assert coords[0].identical(da.coords["x"])
        assert coords[1].identical(da.coords["y"])

        assert "x" in da.coords
        assert 0 not in da.coords
        assert "foo" not in da.coords

        with pytest.raises(KeyError):
            da.coords[0]
        with pytest.raises(KeyError):
            da.coords["foo"]

        # repr
        expected_repr = dedent(
            """\
        Coordinates:
          * x        (x) int64 16B -1 -2
          * y        (y) int64 24B 0 1 2"""
        )
        actual = repr(da.coords)
        assert expected_repr == actual

        # dtypes
        assert da.coords.dtypes == {"x": np.dtype("int64"), "y": np.dtype("int64")}

        del da.coords["x"]
        da._indexes = filter_indexes_from_coords(da.xindexes, set(da.coords))
        expected = DataArray(da.values, {"y": [0, 1, 2]}, dims=["x", "y"], name="foo")
        assert_identical(da, expected)

        with pytest.raises(
            ValueError, match=r"cannot drop or update coordinate.*corrupt.*index "
        ):
            self.mda["level_1"] = ("x", np.arange(4))
            self.mda.coords["level_1"] = ("x", np.arange(4))

    def test_coords_to_index(self) -> None:
        da = DataArray(np.zeros((2, 3)), [("x", [1, 2]), ("y", list("abc"))])