def test_shuffle_empty_partitions(shuffle_method):
    df = pd.DataFrame({'x': ([1, 2, 3] * 10)})
    ddf = dd.from_pandas(df, npartitions=3)
    s = shuffle(ddf, ddf.x, npartitions=6, shuffle=shuffle_method)
    parts = compute_as_if_collection(dd.DataFrame, s.dask, s.__dask_keys__())
    for p in parts:
        assert (s.columns == p.columns)
def test_shuffle_empty_partitions(shuffle_method):
    df = pd.DataFrame({'x': ([1, 2, 3] * 10)})
    ddf = dd.from_pandas(df, npartitions=3)
    s = shuffle(ddf, ddf.x, npartitions=6, shuffle=shuffle_method)
    parts = compute_as_if_collection(dd.DataFrame, s.dask, s.__dask_keys__())
    for p in parts:
        assert (s.columns == p.columns)