def test_shuffle_from_one_partition_to_one_other(shuffle_method):
    df = pd.DataFrame({'x': [1, 2, 3]})
    a = dd.from_pandas(df, 1)
    for i in [1, 2]:
        b = shuffle(a, 'x', npartitions=i, shuffle=shuffle_method)
        assert (len(a.compute(scheduler='sync')) == len(b.compute(scheduler='sync')))
def test_shuffle_from_one_partition_to_one_other(shuffle_method):
    df = pd.DataFrame({'x': [1, 2, 3]})
    a = dd.from_pandas(df, 1)
    for i in [1, 2]:
        b = shuffle(a, 'x', npartitions=i, shuffle=shuffle_method)
        assert (len(a.compute(scheduler='sync')) == len(b.compute(scheduler='sync')))