(self, fill_value, extension_array)
| 5849 | ), |
| 5850 | ) |
| 5851 | def test_fillna_extension_array(self, fill_value, extension_array) -> None: |
| 5852 | srs = pd.DataFrame({"data": extension_array}, index=np.array([1, 2, 3])) |
| 5853 | ds = srs.to_xarray() |
| 5854 | filled = ds.fillna(fill_value) |
| 5855 | assert filled["data"].dtype == extension_array.dtype |
| 5856 | assert ( |
| 5857 | filled["data"].values |
| 5858 | == np.array([fill_value, *srs["data"].values[1:]], dtype="object") |
| 5859 | ).all() |
| 5860 | |
| 5861 | @pytest.mark.parametrize( |
| 5862 | "extension_array", |