Function test_like_forgets_graph

dask/array/tests/test_creation.py:1106–1115

Test that array creation functions with like=x do not internally store the graph of x

(func)

    "func", [da.array, da.asarray, da.asanyarray, da.arange, da.tri]
)
def test_like_forgets_graph(func):
    """Test that array creation functions with like=x do not
    internally store the graph of x
    """
    x = da.arange(3).map_blocks(lambda x: x)
    with pytest.raises(Exception, match="local object"):
        pickle.dumps(x)

    a = func(1, like=x)
    pickle.dumps(a)