diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py index 09555c406d7..9b5e2dd1c80 100644 --- a/dask/tests/test_base.py +++ b/dask/tests/test_base.py @@ -654,6 +654,7 @@ def test_visualize(): @pytest.mark.skipif(sys.flags.optimize, reason="graphviz exception with Python -OO flag") def test_visualize_lists(tmpdir): + pytest.importorskip('graphviz') fn = os.path.join(str(tmpdir), 'myfile.dot') dask.visualize([{'abc-xyz': (add, 1, 2)}], filename=fn) with open(fn) as f: