@pytest.mark.parametrize('compression', [None, 'ZLib'])
def test_disk_shuffle_with_compression_option(compression):
    with dask.config.set({'dataframe.shuffle-compression': compression}):
        test_shuffle('disk')
@pytest.mark.parametrize('compression', [None, 'ZLib'])
def test_disk_shuffle_with_compression_option(compression):
    with dask.config.set({'dataframe.shuffle-compression': compression}):
        test_shuffle('disk')