Function test_string_accessor

dask/dataframe/dask_expr/tests/test_string_accessor.py:77–114

(ser, dser, func, kwargs)

    ],
)
def test_string_accessor(ser, dser, func, kwargs):
    if pyarrow_strings_enabled():
        ser = ser.astype("string[pyarrow]")

    assert_eq(getattr(ser.str, func)(**kwargs), getattr(dser.str, func)(**kwargs))

    if func in (
        "contains",
        "endswith",
        "fullmatch",
        "isalnum",
        "isalpha",
        "isdecimal",
        "isdigit",
        "islower",
        "isspace",
        "istitle",
        "isupper",
        "startswith",
        "match",
    ):
        # This returns arrays and doesn't work in dask/dask either
        return

    ser.index = ser.values
    ser = ser.sort_index()
    dser = from_pandas(ser, npartitions=3)
    pdf_result = getattr(ser.index.str, func)(**kwargs)

    if func == "cat" and len(kwargs) > 0:
        # Doesn't work with others on Index
        return
    if isinstance(pdf_result, pd.DataFrame):
        assert_eq(
            getattr(dser.index.str, func)(**kwargs), pdf_result, check_index=False
        )
    else:
        assert_eq(getattr(dser.index.str, func)(**kwargs), pdf_result)


def test_str_accessor_cat(ser, dser):