add in pytest.importorskip('graphviz')

mrocklin · Mar 16, 2019 · d65202f · d65202f
1 parent eb53ec3
commit d65202f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: