Expaned unittest

Author: cvanelteren

tests/baseline/test_colorbar.png

@@ -2,12 +2,6 @@
 from matplotlib.testing import setup
 
 
-@pytest.fixture()
-def setup_mpl():
-    setup()
-    plt.clf()
-
-
 @pytest.mark.mpl_image_compare
 def test_standardized_inputs_1d():
     N = 5
@@ -37,10 +31,3 @@ def test_standardized_inputs_1d():
         ax.scatter(y + 2, marker="s", markersize=5**2)
         fig.format(xlabel="xlabel", ylabel="ylabel")
     return fig
-
-
-if __name__ == "__main__":
-    for func in [
-        test_standardized_inputs_1d,
-    ]:
-        func().savefig(f"./tests/baseline/{func.__name__}.png")

tests/baseline/test_inbounds_data.png

@@ -108,14 +108,3 @@ def test_colorbar():
     axs[2].format(title="Imshow plot\ndiscrete=False (default)", yformatter="auto")
     fig.show()
     return fig
-
-
-if __name__ == "__main__":
-    for func in [
-        test_standardized_input,
-        test_inbounds_data,
-        test_colorbar,
-    ]:
-        fig = func()
-        fig.savefig(f"./tests/baseline/{func.__name__}.png")
-    plt.show(block=1)