Method test_dropna_extension_array

xarray/tests/test_dataset.py:5876–5881

(self, extension_array)

        ),
    )
    def test_dropna_extension_array(self, extension_array) -> None:
        srs = pd.DataFrame({"data": extension_array}, index=np.array([1, 2, 3]))
        ds = srs.to_xarray()
        dropped = ds.dropna("index")
        assert dropped["data"].dtype == extension_array.dtype
        assert (dropped["data"].values == srs["data"].values[1:]).all()

    def test_fillna(self) -> None:
        ds = Dataset({"a": ("x", [np.nan, 1, np.nan, 3])}, {"x": [0, 1, 2, 3]})