()
| 589 | bool(sys.flags.optimize), reason="graphviz exception with Python -OO flag" |
| 590 | ) |
| 591 | def test_visualize_highlevelgraph(): |
| 592 | graphviz = pytest.importorskip("graphviz") |
| 593 | with tmpdir() as d: |
| 594 | x = da.arange(5, chunks=2) |
| 595 | viz = x.dask.visualize(filename=os.path.join(d, "mydask.png")) |
| 596 | # check visualization will automatically render in the jupyter notebook |
| 597 | assert isinstance(viz, graphviz.Digraph) |
| 598 | |
| 599 | |
| 600 | @pytest.mark.skipif("not da") |