Method test_default_labels

xarray/tests/test_plot.py:2556–2588

(self)

    @pytest.mark.slow
    def test_default_labels(self) -> None:
        g = xplt.FacetGrid(self.darray, col="col", row="row")
        assert (2, 3) == g.axs.shape

        g.map_dataarray(xplt.imshow, "x", "y")

        # Rightmost column should be labeled
        for label, ax in zip(
            self.darray.coords["row"].values, g.axs[:, -1], strict=True
        ):
            assert substring_in_axes(label, ax)

        # Top row should be labeled
        for label, ax in zip(
            self.darray.coords["col"].values, g.axs[0, :], strict=True
        ):
            assert substring_in_axes(label, ax)

        # ensure that row & col labels can be changed
        g.set_titles("abc={value}")
        for label, ax in zip(
            self.darray.coords["row"].values, g.axs[:, -1], strict=True
        ):
            assert substring_in_axes(f"abc={label}", ax)
            # previous labels were "row=row0" etc.
            assert substring_not_in_axes("row=", ax)

        for label, ax in zip(
            self.darray.coords["col"].values, g.axs[0, :], strict=True
        ):
            assert substring_in_axes(f"abc={label}", ax)
            # previous labels were "col=row0" etc.
            assert substring_not_in_axes("col=", ax)


@pytest.mark.filterwarnings("ignore:tight_layout cannot")