def test_set_index_reduces_partitions_small(shuffle_method):
    df = pd.DataFrame({'x': np.random.random(100)})
    ddf = dd.from_pandas(df, npartitions=50)
    ddf2 = ddf.set_index('x', shuffle=shuffle_method, npartitions='auto')
    assert (ddf2.npartitions < 10)
def test_set_index_reduces_partitions_small(shuffle_method):
    df = pd.DataFrame({'x': np.random.random(100)})
    ddf = dd.from_pandas(df, npartitions=50)
    ddf2 = ddf.set_index('x', shuffle=shuffle_method, npartitions='auto')
    assert (ddf2.npartitions < 10)