Function test_array_to_df_conversion

dask/dataframe/tests/test_dataframe.py:5343–5350

()

def test_array_to_df_conversion():
    def foo(arr):
        return pd.DataFrame(arr)

    arr = da.random.random((100, 10), chunks=(10, 10))
    result = arr.map_blocks(foo, meta=pd.DataFrame(np.random.random((1, 10))))
    expected = pd.DataFrame(arr.compute())
    assert_eq(result, expected, check_index=False)


@pytest.mark.parametrize("npartitions", [1, 5])