Function test_fuse_getter_with_asarray

dask/array/tests/test_optimization.py:139–151

(chunks)

@pytest.mark.parametrize("chunks", [10, 5, 3])
def test_fuse_getter_with_asarray(chunks):
    x = np.ones(10) * 1234567890
    y = da.ones(10, chunks=chunks)
    z = x + y
    dsk = z.__dask_optimize__(z.dask, z.__dask_keys__())
    if chunks == 10:
        assert len(dsk) == 2 and any(isinstance(t, Alias) for t in dsk.values())
    else:
        assert any(
            isinstance(v, DataNode) and isinstance(v.value, np.ndarray)
            for v in dsk.values()
        )
    assert_eq(z, x + 1)


@pytest.mark.xfail(reason="blockwise fusion does not respect this, which is ok")