MCPcopy
hub / github.com/dask/dask / test_shuffle

Function test_shuffle

dask/array/tests/test_shuffle.py:32–37  ·  view source on GitHub ↗
(arr, darr, indexer, chunks, other_chunks)

Source from the content-addressed store, hash-verified

30 ],
31)
32def test_shuffle(arr, darr, indexer, chunks, other_chunks):
33 result = darr.shuffle(indexer, axis=1)
34 expected = arr[:, list(flatten(indexer))]
35 assert_eq(result, expected)
36 assert result.chunks[0] == other_chunks
37 assert result.chunks[1] == chunks
38
39
40@pytest.mark.parametrize("tol, chunks", ((1, (3, 2, 3)), (1.4, (5, 3))))

Callers

nothing calls this directly

Calls 3

flattenFunction · 0.90
assert_eqFunction · 0.90
shuffleMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…