def test_visualize():
pytest.importorskip('graphviz')
with tmpdir() as d:
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, 'mydask'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
x.visualize(filename=os.path.join(d, 'mydask.pdf'))
assert os.path.exists(os.path.join(d, 'mydask.pdf'))
visualize(x, 1, 2, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
dsk = {'a': 1, 'b': (add, 'a', 2), 'c': (mul, 'a', 1)}
visualize(x, dsk, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
def test_visualize():
pytest.importorskip('graphviz')
try:
d = tempfile.mkdtemp()
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, 'mydask'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
x.visualize(filename=os.path.join(d, 'mydask.pdf'))
assert os.path.exists(os.path.join(d, 'mydask.pdf'))
visualize(x, 1, 2, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
finally:
shutil.rmtree(d)
def test_visualize():
pytest.importorskip('graphviz')
with tmpdir() as d:
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, 'mydask'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
x.visualize(filename=os.path.join(d, 'mydask.pdf'))
assert os.path.exists(os.path.join(d, 'mydask.pdf'))
visualize(x, 1, 2, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
dsk = {'a': 1, 'b': (add, 'a', 2), 'c': (mul, 'a', 1)}
visualize(x, dsk, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
def test_visualize():
pytest.importorskip('graphviz')
try:
d = tempfile.mkdtemp()
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, 'mydask'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
x.visualize(filename=os.path.join(d, 'mydask.pdf'))
assert os.path.exists(os.path.join(d, 'mydask.pdf'))
visualize(x, 1, 2, filename=os.path.join(d, 'mydask.png'))
assert os.path.exists(os.path.join(d, 'mydask.png'))
finally:
shutil.rmtree(d)
def visualize(self, raw_dask=True, *args, **kwargs):
dot_status = self.dot_status()
if raw_dask:
dsk = self._dask
else:
dsk = self.dask
return visualize(
dsk,
data_attributes=dot_status,
# function_attributes=dot_status,
*args,
**kwargs)
def test_visualize():
pytest.importorskip("graphviz")
with tmpdir() as d:
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, "mydask"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x.visualize(filename=os.path.join(d, "mydask.pdf"))
assert os.path.exists(os.path.join(d, "mydask.pdf"))
visualize(x, 1, 2, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
dsk = {"a": 1, "b": (add, "a", 2), "c": (mul, "a", 1)}
visualize(x, dsk, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x = Tuple(dsk, ["a", "b", "c"])
visualize(x, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
def test_visualize():
pytest.importorskip("graphviz")
with tmpdir() as d:
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, "mydask"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x.visualize(filename=os.path.join(d, "mydask.pdf"))
assert os.path.exists(os.path.join(d, "mydask.pdf"))
visualize(x, 1, 2, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
dsk = {"a": 1, "b": (add, "a", 2), "c": (mul, "a", 1)}
visualize(x, dsk, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x = Tuple(dsk, ["a", "b", "c"])
visualize(x, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
# To see if visualize() works when the filename parameter is set to None
# If the function raises an error, the test will fail
x.visualize(filename=None)
def test_visualize():
pytest.importorskip("graphviz")
pytest.importorskip("ipycytoscape")
with tmpdir() as d:
x = da.arange(5, chunks=2)
x.visualize(filename=os.path.join(d, "mydask"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x.visualize(filename=os.path.join(d, "mydask.pdf"))
assert os.path.exists(os.path.join(d, "mydask.pdf"))
visualize(x, 1, 2, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
dsk = {"a": 1, "b": (add, "a", 2), "c": (mul, "a", 1)}
visualize(x, dsk, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x = Tuple(dsk, ["a", "b", "c"])
visualize(x, filename=os.path.join(d, "mydask.png"))
assert os.path.exists(os.path.join(d, "mydask.png"))
x = Tuple(dsk, ["a", "b", "c"])
visualize(x, filename=os.path.join(d, "cyt"), engine="cytoscape")
assert os.path.exists(os.path.join(d, "cyt.html"))
visualize(x, filename=os.path.join(d, "cyt2.html"), engine="ipycytoscape")
assert os.path.exists(os.path.join(d, "cyt2.html"))
with dask.config.set(visualization__engine="cytoscape"):
visualize(x, filename=os.path.join(d, "cyt3.html"))
assert os.path.exists(os.path.join(d, "cyt3.html"))
with pytest.raises(ValueError, match="not-real"):
visualize(x, engine="not-real")
# To see if visualize() works when the filename parameter is set to None
# If the function raises an error, the test will fail
x.visualize(filename=None)