Check before switching backend

jemrobinson · Jul 2, 2018 · bba617c · bba617c
1 parent c1579ea
commit bba617c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/decorations/test_decorations_atlas_text.py b/tests/decorations/test_decorations_atlas_text.py
@@ -77,9 +77,10 @@ def test_atlas_text_incorrect_alignment():
 
 
 def test_atlas_text_renderer():
-    matplotlib.pyplot.switch_backend("TkAgg")
-    with mATLASplotlib.canvases.Simple() as canvas:
-        canvas.plot_dataset([0, 1], [5, 10], style="scatter")
-        canvas.add_ATLAS_label(0.5, 0.5, plot_type="Internal", anchor_to="upper right")
-    matplotlib.pyplot.switch_backend("pdf")
+    if hasattr(matplotlib.backends, "backend_agg"):
+        matplotlib.pyplot.switch_backend("agg")
+        with mATLASplotlib.canvases.Simple() as canvas:
+            canvas.plot_dataset([0, 1], [5, 10], style="scatter")
+            canvas.add_ATLAS_label(0.5, 0.5, plot_type="Internal", anchor_to="upper right")
+        matplotlib.pyplot.switch_backend("pdf")