Method test_fillna

xarray/tests/test_units.py:4311–4339

(self, fill_value, unit, error, dtype)

        ),
    )
    def test_fillna(self, fill_value, unit, error, dtype):
        array1 = (
            create_nan_array([1.4, np.nan, 2.3, np.nan, np.nan, 9.1], dtype)
            * unit_registry.m
        )
        array2 = (
            create_nan_array([4.3, 9.8, 7.5, np.nan, 8.2, np.nan], dtype)
            * unit_registry.m
        )
        ds = xr.Dataset({"a": ("x", array1), "b": ("x", array2)})
        value = fill_value * unit
        units = extract_units(ds)

        if error is not None:
            with pytest.raises(error):
                ds.fillna(value=value)

            return

        actual = ds.fillna(value=value)
        expected = attach_units(
            strip_units(ds).fillna(
                value=strip_units(convert_units(value, {None: unit_registry.m}))
            ),
            units,
        )

        assert_units_equal(expected, actual)
        assert_equal(expected, actual)

    def test_dropna(self, dtype):
        array1 = (