Function test_creation_data_producers

dask/array/tests/test_creation.py:567–592

()

def test_creation_data_producers():
    x = np.arange(64).reshape((8, 8))
    d = da.from_array(x, chunks=(4, 4))
    dsk = collections_to_expr([d]).__dask_graph__()
    assert all(v.data_producer for v in dsk.values())

    # blockwise fusion
    x = d.astype("float64")
    dsk = collections_to_expr([x]).__dask_graph__()
    assert sum(v.data_producer for v in dsk.values()) == 4
    assert sum(isinstance(v, Alias) for v in dsk.values()) == 4
    assert len(dsk) == 8

    # linear fusion
    x = d[slice(0, 6), None].astype("float64")
    dsk = collections_to_expr([x]).__dask_graph__()
    assert sum(v.data_producer for v in dsk.values()) == 4
    assert sum(isinstance(v, Alias) for v in dsk.values()) == 4
    assert len(dsk) == 8

    # no fusion
    x = d[[1, 3, 5, 7, 6, 4, 2, 0]].astype("float64")
    dsk = collections_to_expr([x]).__dask_graph__()
    assert sum(v.data_producer and "array-" in k[0] for k, v in dsk.items()) == 4
    assert sum(v.data_producer for v in dsk.values()) == 8  # getitem data nodes
    assert len(dsk) == 24


def test_diagonal():