try to supress warngins during tests

Author: phobson

probscale/tests/test_viz.py

@@ -744,6 +744,7 @@ def test_probplot_prob(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_qq(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -759,6 +760,7 @@ def test_probplot_qq(plot_data):
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 @pytest.mark.skipif(stats is None, reason="no scipy")
 def test_probplot_qq_dist(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     norm = stats.norm(*stats.norm.fit(plot_data))
     fig = viz.probplot(
@@ -769,6 +771,7 @@ def test_probplot_qq_dist(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_pp(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     scatter_kws = dict(
         color="b", linestyle="--", markeredgecolor="g", markerfacecolor="none"
@@ -791,6 +794,7 @@ def test_probplot_pp(plot_data):
     remove_text=True,
 )
 def test_probplot_prob_bestfit(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -806,6 +810,7 @@ def test_probplot_prob_bestfit(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_prob_bestfit_exceedance(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -822,6 +827,7 @@ def test_probplot_prob_bestfit_exceedance(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_qq_bestfit(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -837,6 +843,7 @@ def test_probplot_qq_bestfit(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_pp_bestfit(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     scatter_kws = {"marker": "s", "color": "red"}
     line_kws = {"linestyle": "--", "linewidth": 3}
@@ -857,6 +864,7 @@ def test_probplot_pp_bestfit(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_prob_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data, ax=ax, datalabel="Test xlabel", datascale="log", probax="y"
@@ -867,6 +875,7 @@ def test_probplot_prob_probax_y(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_qq_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -882,6 +891,7 @@ def test_probplot_qq_probax_y(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_pp_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     scatter_kws = dict(
         color="b", linestyle="--", markeredgecolor="g", markerfacecolor="none"
@@ -901,6 +911,7 @@ def test_probplot_pp_probax_y(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_prob_bestfit_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -917,6 +928,7 @@ def test_probplot_prob_bestfit_probax_y(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_qq_bestfit_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(
         plot_data,
@@ -933,6 +945,7 @@ def test_probplot_qq_bestfit_probax_y(plot_data):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_pp_bestfit_probax_y(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     scatter_kws = {"marker": "s", "color": "red"}
     line_kws = {"linestyle": "--", "linewidth": 3}
@@ -955,6 +968,7 @@ def test_probplot_pp_bestfit_probax_y(plot_data):
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=LOOSE_TOLERANCE)
 @pytest.mark.skipif(stats is None, reason="no scipy")
 def test_probplot_beta_dist_best_fit_y(plot_data):
+    plt.close("all")
     fig, (ax1, ax2) = plt.subplots(ncols=2)
     dist = stats.beta(3, 3)
     fig = viz.probplot(
@@ -990,6 +1004,7 @@ def test_probplot_beta_dist_best_fit_y(plot_data):
 )
 @pytest.mark.skipif(stats is None, reason="no scipy")
 def test_probplot_beta_dist_best_fit_x(plot_data):
+    plt.close("all")
     fig, (ax1, ax2) = plt.subplots(nrows=2)
     dist = stats.beta(3, 3)
     fig = viz.probplot(
@@ -1019,6 +1034,7 @@ def test_probplot_beta_dist_best_fit_x(plot_data):
 
 
 def test_probplot_test_results(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig, results = viz.probplot(plot_data, return_best_fit_results=True)
 
@@ -1046,6 +1062,7 @@ def test__set_prob_limits_x(probax, N, minval, maxval):
 
 @pytest.mark.mpl_image_compare(baseline_dir=BASELINE_DIR, tolerance=TIGHT_TOLERANCE)
 def test_probplot_color_and_label(plot_data):
+    plt.close("all")
     fig, ax = plt.subplots()
     fig = viz.probplot(plot_data, ax=ax, color="pink", label="A Top-Level Label")
     ax.legend(loc="lower right")

setup.cfg

@@ -3,12 +3,5 @@ description-file = readme.md
 license_file = LICENSE
 
 [tool:pytest]
-pep8ignore =
-    E501
-    probscale/tests/test_*.py E241 E131
-    docs/* ALL
-    check_probscale.py ALL
-
 markers =
     mpl_image_compare
-    pep8