Method test_isin

xarray/tests/test_units.py:2867–2884

(self, unit, dtype)

        ),
    )
    def test_isin(self, unit, dtype):
        array = (
            create_nan_array([1.4, np.nan, 2.3, np.nan, np.nan, 9.1], dtype)
            * unit_registry.m
        )
        data_array = xr.DataArray(data=array, dims="x")

        raw_values = create_nan_array([1.4, np.nan, 2.3], dtype)
        values = raw_values * unit

        units = {None: unit_registry.m if array.check(unit) else None}
        expected = strip_units(data_array).isin(
            strip_units(convert_units(values, units))
        ) & array.check(unit)
        actual = data_array.isin(values)

        assert_units_equal(expected, actual)
        assert_identical(expected, actual)

    @pytest.mark.parametrize(
        "variant", ("masking", "replacing_scalar", "replacing_array", "dropping")