def test_index_with_dataframe(shuffle_method):
    res1 = shuffle(d, d[['b']], shuffle=shuffle_method).compute()
    res2 = shuffle(d, ['b'], shuffle=shuffle_method).compute()
    res3 = shuffle(d, 'b', shuffle=shuffle_method).compute()
    assert (sorted(res1.values.tolist()) == sorted(res2.values.tolist()))
    assert (sorted(res1.values.tolist()) == sorted(res3.values.tolist()))
def test_index_with_dataframe(shuffle_method):
    res1 = shuffle(d, d[['b']], shuffle=shuffle_method).compute()
    res2 = shuffle(d, ['b'], shuffle=shuffle_method).compute()
    res3 = shuffle(d, 'b', shuffle=shuffle_method).compute()
    assert (sorted(res1.values.tolist()) == sorted(res2.values.tolist()))
    assert (sorted(res1.values.tolist()) == sorted(res3.values.tolist()))