Function test_dataframe_split_every

dask/dataframe/dask_expr/tests/test_reductions.py:139–145

(pdf, df, split_every, expect_tasks, reduction)

    "split_every,expect_tasks", [(False, 21), (None, 23), (5, 23), (2, 31)]
)
def test_dataframe_split_every(pdf, df, split_every, expect_tasks, reduction):
    assert_eq(
        getattr(df, reduction)(split_every=split_every),
        getattr(pdf, reduction)(),
    )
    q = getattr(df, reduction)(split_every=split_every).optimize(fuse=False)
    assert len(q.__dask_graph__()) == expect_tasks


@pytest.mark.parametrize(