Fix a mpl dep warning with test figures (#205)

DKISTDC · Sep 20, 2022 · fb896ca · fb896ca
1 parent d19d2dc
commit fb896ca
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dkist/dataset/tests/test_plotting.py b/dkist/dataset/tests/test_plotting.py
@@ -9,18 +9,22 @@
 def test_dataset_projection(dataset_3d, aslice):
     pytest.importorskip("ndcube", "2.0.2")  # https://github.com/sunpy/ndcube/pull/509
     ds = dataset_3d[aslice]
+    fig = plt.figure()
     ax = plt.subplot(projection=ds)
     assert isinstance(ax, WCSAxes)
+    return fig
 
 
 # @pytest.mark.mpl_image_compare
 @pytest.mark.parametrize("aslice", (np.s_[0, :, :], np.s_[:, 0, :], np.s_[:, :, 0]))
 def test_2d_plot(dataset_3d, aslice):
+    fig = plt.figure()
     dataset_3d[aslice].plot()
-    return plt.gcf()
+    return fig
 
 
 # @pytest.mark.mpl_image_compare
 def test_2d_plot2(dataset_3d):
+    fig = plt.figure()
     dataset_3d[:, :, 0].plot(axes_units=["Angstrom", "deg", "deg"])
-    return plt.gcf()
+    return fig