Function test_contains

xarray/tests/test_accessor_str.py:103–144

(dtype)

def test_contains(dtype) -> None:
    values = xr.DataArray(["Foo", "xYz", "fOOomMm__fOo", "MMM_"]).astype(dtype)

    # case insensitive using regex
    pat = values.dtype.type("FOO|mmm")
    result = values.str.contains(pat, case=False)
    expected = xr.DataArray([True, False, True, True])
    assert result.dtype == expected.dtype
    assert_equal(result, expected)
    result = values.str.contains(re.compile(pat, flags=re.IGNORECASE))
    assert result.dtype == expected.dtype
    assert_equal(result, expected)

    # case sensitive using regex
    pat = values.dtype.type("Foo|mMm")
    result = values.str.contains(pat)
    expected = xr.DataArray([True, False, True, False])
    assert result.dtype == expected.dtype
    assert_equal(result, expected)
    result = values.str.contains(re.compile(pat))
    assert result.dtype == expected.dtype
    assert_equal(result, expected)

    # case insensitive without regex
    result = values.str.contains("foo", regex=False, case=False)
    expected = xr.DataArray([True, False, True, False])
    assert result.dtype == expected.dtype
    assert_equal(result, expected)

    # case sensitive without regex
    result = values.str.contains("fO", regex=False, case=True)
    expected = xr.DataArray([False, False, True, False])
    assert result.dtype == expected.dtype
    assert_equal(result, expected)

    # regex regex=False
    pat_re = re.compile("(/w+)")
    with pytest.raises(
        ValueError,
        match=r"Must use regular expression matching for regular expression object.",
    ):
        values.str.contains(pat_re, regex=False)


def test_contains_broadcast(dtype) -> None: