add pytest.importorskip("matplotlib") to all test_draw_graph

materialsproject · Jul 4, 2024 · 4a1d1ec · 4a1d1ec
1 parent c0bed9e
commit 4a1d1ec
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tests/core/test_flow.py b/tests/core/test_flow.py
@@ -354,6 +354,8 @@ def test_graph():
 
 
 def test_draw_graph():
+    pytest.importorskip("matplotlib")
+
     from jobflow import Flow, JobOrder
 
     # test unconnected graph
@@ -385,6 +387,8 @@ def test_draw_graph():
 
 @pytest.mark.usefixtures("no_pydot")
 def test_draw_graph_nopydot():
+    pytest.importorskip("matplotlib")
+
     from jobflow import Flow, JobOrder
 
     # test unconnected graph

diff --git a/tests/utils/test_graph.py b/tests/utils/test_graph.py
@@ -33,6 +33,8 @@ def test_itergraph():
 
 
 def test_draw_graph():
+    pytest.importorskip("matplotlib")
+
     from networkx import DiGraph, planar_layout
 
     from jobflow.utils.graph import draw_graph
@@ -45,6 +47,8 @@ def test_draw_graph():
 
 @pytest.mark.usefixtures("no_pydot")
 def test_draw_graph_no_pydot():
+    pytest.importorskip("matplotlib")
+
     from networkx import DiGraph
 
     from jobflow.utils.graph import draw_graph