Function test_partition

dask/dataframe/dask_expr/tests/test_string_accessor.py:153–159

()

def test_partition():
    df = DataFrame.from_dict({"A": ["A|B", "C|D"]}, npartitions=2)["A"].str.partition(
        "|"
    )
    result = df[1]
    expected = pd.DataFrame.from_dict({"A": ["A|B", "C|D"]})["A"].str.partition("|")[1]
    assert_eq(result, expected)