ENH: New baseline images and updates to plot test tolerances (#801)

* ENH: New baseline images and updates to plot test tolerances * ENH: ruff updates * ENH: Updating tolerance to 20 for test_xsection_plot * ENH: New baseline file updates and tests
ARM-DOE · Feb 16, 2024 · c15802b · c15802b
1 parent 707c3bc
commit c15802b
Show file tree

Hide file tree

Showing 65 changed files with 68 additions and 63 deletions.
diff --git a/tests/plotting/baseline/test_2D_timeseries_plot.png b/tests/plotting/baseline/test_2D_timeseries_plot.png
diff --git a/tests/plotting/baseline/test_2d_as_1d.png b/tests/plotting/baseline/test_2d_as_1d.png
diff --git a/tests/plotting/baseline/test_add_nan_line.png b/tests/plotting/baseline/test_add_nan_line.png
diff --git a/tests/plotting/baseline/test_assessment_overplot.png b/tests/plotting/baseline/test_assessment_overplot.png
diff --git a/tests/plotting/baseline/test_assessment_overplot_multi.png b/tests/plotting/baseline/test_assessment_overplot_multi.png
diff --git a/tests/plotting/baseline/test_barb_sounding_plot.png b/tests/plotting/baseline/test_barb_sounding_plot.png
diff --git a/tests/plotting/baseline/test_colorbar_labels.png b/tests/plotting/baseline/test_colorbar_labels.png
diff --git a/tests/plotting/baseline/test_contour.png b/tests/plotting/baseline/test_contour.png
diff --git a/tests/plotting/baseline/test_contour2.png b/tests/plotting/baseline/test_contour2.png
diff --git a/tests/plotting/baseline/test_contour_stamp.png b/tests/plotting/baseline/test_contour_stamp.png
diff --git a/tests/plotting/baseline/test_contourf.png b/tests/plotting/baseline/test_contourf.png
diff --git a/tests/plotting/baseline/test_contourf2.png b/tests/plotting/baseline/test_contourf2.png
diff --git a/tests/plotting/baseline/test_enhanced_skewt_plot.png b/tests/plotting/baseline/test_enhanced_skewt_plot.png
diff --git a/tests/plotting/baseline/test_enhanced_skewt_plot_2.png b/tests/plotting/baseline/test_enhanced_skewt_plot_2.png
diff --git a/tests/plotting/baseline/test_fill_between.png b/tests/plotting/baseline/test_fill_between.png
diff --git a/tests/plotting/baseline/test_geoplot.png b/tests/plotting/baseline/test_geoplot.png
diff --git a/tests/plotting/baseline/test_geoplot_tile.png b/tests/plotting/baseline/test_geoplot_tile.png
diff --git a/tests/plotting/baseline/test_groupby_plot.png b/tests/plotting/baseline/test_groupby_plot.png
diff --git a/tests/plotting/baseline/test_heatmap.png b/tests/plotting/baseline/test_heatmap.png
diff --git a/tests/plotting/baseline/test_heatmap2.png b/tests/plotting/baseline/test_heatmap2.png
diff --git a/tests/plotting/baseline/test_heatmap3.png b/tests/plotting/baseline/test_heatmap3.png
diff --git a/tests/plotting/baseline/test_match_ylimits_plot.png b/tests/plotting/baseline/test_match_ylimits_plot.png
diff --git a/tests/plotting/baseline/test_multi_skewt_plot.png b/tests/plotting/baseline/test_multi_skewt_plot.png
diff --git a/tests/plotting/baseline/test_multidataset_plot_dict.png b/tests/plotting/baseline/test_multidataset_plot_dict.png
diff --git a/tests/plotting/baseline/test_multidataset_plot_tuple.png b/tests/plotting/baseline/test_multidataset_plot_tuple.png
diff --git a/tests/plotting/baseline/test_plot.png b/tests/plotting/baseline/test_plot.png
diff --git a/tests/plotting/baseline/test_plot_barbs_from_u_v.png b/tests/plotting/baseline/test_plot_barbs_from_u_v.png
diff --git a/tests/plotting/baseline/test_plot_barbs_from_u_v2.png b/tests/plotting/baseline/test_plot_barbs_from_u_v2.png
diff --git a/tests/plotting/baseline/test_plot_datarose.png b/tests/plotting/baseline/test_plot_datarose.png
diff --git a/tests/plotting/baseline/test_qc_bar_plot.png b/tests/plotting/baseline/test_qc_bar_plot.png
diff --git a/tests/plotting/baseline/test_qc_flag_block_plot.png b/tests/plotting/baseline/test_qc_flag_block_plot.png
diff --git a/tests/plotting/baseline/test_scatter.png b/tests/plotting/baseline/test_scatter.png
diff --git a/tests/plotting/baseline/test_scatter2.png b/tests/plotting/baseline/test_scatter2.png
diff --git a/tests/plotting/baseline/test_size_distribution.png b/tests/plotting/baseline/test_size_distribution.png
diff --git a/tests/plotting/baseline/test_size_distribution2.png b/tests/plotting/baseline/test_size_distribution2.png
diff --git a/tests/plotting/baseline/test_skewt_options.png b/tests/plotting/baseline/test_skewt_options.png
diff --git a/tests/plotting/baseline/test_skewt_plot.png b/tests/plotting/baseline/test_skewt_plot.png
diff --git a/tests/plotting/baseline/test_skewt_plot_spd_dir.png b/tests/plotting/baseline/test_skewt_plot_spd_dir.png
diff --git a/tests/plotting/baseline/test_stacked_bar_graph.png b/tests/plotting/baseline/test_stacked_bar_graph.png
diff --git a/tests/plotting/baseline/test_stacked_bar_graph2.png b/tests/plotting/baseline/test_stacked_bar_graph2.png
diff --git a/tests/plotting/baseline/test_stacked_bar_graph3.png b/tests/plotting/baseline/test_stacked_bar_graph3.png
diff --git a/tests/plotting/baseline/test_stacked_bar_graph_sorted.png b/tests/plotting/baseline/test_stacked_bar_graph_sorted.png
diff --git a/tests/plotting/baseline/test_stair_graph.png b/tests/plotting/baseline/test_stair_graph.png
diff --git a/tests/plotting/baseline/test_stair_graph2.png b/tests/plotting/baseline/test_stair_graph2.png
diff --git a/tests/plotting/baseline/test_stair_graph_sorted.png b/tests/plotting/baseline/test_stair_graph_sorted.png
diff --git a/tests/plotting/baseline/test_time_height_scatter.png b/tests/plotting/baseline/test_time_height_scatter.png
diff --git a/tests/plotting/baseline/test_time_height_scatter2.png b/tests/plotting/baseline/test_time_height_scatter2.png
diff --git a/tests/plotting/baseline/test_time_plot.png b/tests/plotting/baseline/test_time_plot.png
diff --git a/tests/plotting/baseline/test_time_plot2.png b/tests/plotting/baseline/test_time_plot2.png
diff --git a/tests/plotting/baseline/test_time_plot_match_color_ylabel.png b/tests/plotting/baseline/test_time_plot_match_color_ylabel.png
diff --git a/tests/plotting/baseline/test_timeseries_invert.png b/tests/plotting/baseline/test_timeseries_invert.png
diff --git a/tests/plotting/baseline/test_violin.png b/tests/plotting/baseline/test_violin.png
diff --git a/tests/plotting/baseline/test_violin2.png b/tests/plotting/baseline/test_violin2.png
diff --git a/tests/plotting/baseline/test_wind_rose.png b/tests/plotting/baseline/test_wind_rose.png
diff --git a/tests/plotting/baseline/test_xlim_correction_plot.png b/tests/plotting/baseline/test_xlim_correction_plot.png
diff --git a/tests/plotting/baseline/test_xsection_plot.png b/tests/plotting/baseline/test_xsection_plot.png
diff --git a/tests/plotting/baseline/test_xsection_plot_map.png b/tests/plotting/baseline/test_xsection_plot_map.png
diff --git a/tests/plotting/baseline/test_y_axis_flag_meanings.png b/tests/plotting/baseline/test_y_axis_flag_meanings.png
diff --git a/tests/plotting/test_contourdisplay.py b/tests/plotting/test_contourdisplay.py
@@ -8,7 +8,7 @@
 matplotlib.use('Agg')
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_contour():
     files = sample_files.EXAMPLE_MET_CONTOUR
     time = '2019-05-08T04:00:00.000000000'
@@ -36,7 +36,7 @@ def test_contour():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_contour_stamp():
     files = sample_files.EXAMPLE_STAMP_WILDCARD
     test = {}
@@ -58,7 +58,7 @@ def test_contour_stamp():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_contour2():
     files = sample_files.EXAMPLE_MET_CONTOUR
     time = '2019-05-08T04:00:00.000000000'
@@ -86,7 +86,7 @@ def test_contour2():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_contourf():
     files = sample_files.EXAMPLE_MET_CONTOUR
     time = '2019-05-08T04:00:00.000000000'
@@ -126,7 +126,7 @@ def test_contourf():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_contourf2():
     files = sample_files.EXAMPLE_MET_CONTOUR
     time = '2019-05-08T04:00:00.000000000'

diff --git a/tests/plotting/test_distributiondisplay.py b/tests/plotting/test_distributiondisplay.py
@@ -80,7 +80,7 @@ def test_distribution_errors():
     matplotlib.pyplot.close(fig=histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stair_graph():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -94,7 +94,7 @@ def test_stair_graph():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stair_graph2():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     del sonde_ds['tdry'].attrs['units']
@@ -109,7 +109,7 @@ def test_stair_graph2():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stair_graph_sorted():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -128,7 +128,7 @@ def test_stair_graph_sorted():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stacked_bar_graph():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -142,7 +142,7 @@ def test_stacked_bar_graph():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stacked_bar_graph2():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -158,7 +158,7 @@ def test_stacked_bar_graph2():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stacked_bar_graph3():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     del sonde_ds['tdry'].attrs['units']
@@ -173,7 +173,7 @@ def test_stacked_bar_graph3():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_stacked_bar_graph_sorted():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -192,7 +192,7 @@ def test_stacked_bar_graph_sorted():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_heatmap():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
 
@@ -212,7 +212,7 @@ def test_heatmap():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_heatmap2():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     del sonde_ds['tdry'].attrs['units']
@@ -233,7 +233,7 @@ def test_heatmap2():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_heatmap3():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     del sonde_ds['tdry'].attrs['units']
@@ -253,7 +253,7 @@ def test_heatmap3():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_size_distribution():
     sigma = 10
     mu = 50
@@ -270,7 +270,7 @@ def test_size_distribution():
         matplotlib.pyplot.close(histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_size_distribution2():
     sigma = 10
     mu = 50
@@ -334,7 +334,7 @@ def test_histogram_kwargs():
     matplotlib.pyplot.close(fig=histdisplay.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_violin():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_MET1)
 
@@ -352,7 +352,7 @@ def test_violin():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_violin2():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_MET1)
     del ds['temp_mean'].attrs['units']
@@ -371,7 +371,7 @@ def test_violin2():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_scatter():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_MET1)
     # Create a DistributionDisplay object to compare fields
@@ -394,7 +394,7 @@ def test_scatter():
         matplotlib.pyplot.close(display.fig)
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_scatter2():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_MET1)
     del ds['wspd_arith_mean'].attrs['units']

diff --git a/tests/plotting/test_geodisplay.py b/tests/plotting/test_geodisplay.py
@@ -17,7 +17,7 @@
 
 
 @pytest.mark.skipif(not CARTOPY_AVAILABLE, reason='Cartopy is not installed.')
-@pytest.mark.mpl_image_compare(style='default', tolerance=30)
+@pytest.mark.mpl_image_compare(style='default', tolerance=10)
 def test_geoplot():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     geodisplay = GeographicPlotDisplay({'sgpsondewnpnC1.b1': sonde_ds}, figsize=(15, 8))
@@ -46,7 +46,7 @@ def test_geoplot():
 
 
 @pytest.mark.skipif(not CARTOPY_AVAILABLE, reason='Cartopy is not installed.')
-@pytest.mark.mpl_image_compare(style='default', tolerance=30)
+@pytest.mark.mpl_image_compare(style='default', tolerance=10)
 def test_geoplot_tile():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     geodisplay = GeographicPlotDisplay({'sgpsondewnpnC1.b1': sonde_ds}, figsize=(15, 8))

diff --git a/tests/plotting/test_skewtdisplay.py b/tests/plotting/test_skewtdisplay.py
@@ -1,4 +1,5 @@
 import matplotlib
+import matplotlib.pyplot as plt
 import numpy as np
 import pytest
 
@@ -9,7 +10,7 @@
 matplotlib.use('Agg')
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_skewt_plot():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     skewt = SkewTDisplay(sonde_ds)
@@ -18,7 +19,7 @@ def test_skewt_plot():
     return skewt.fig
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_skewt_plot_spd_dir():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     skewt = SkewTDisplay(sonde_ds, ds_name='act_datastream')
@@ -27,7 +28,7 @@ def test_skewt_plot_spd_dir():
     return skewt.fig
 
 
-@pytest.mark.mpl_image_compare(tolerance=81)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_multi_skewt_plot():
     files = sample_files.EXAMPLE_TWP_SONDE_20060121
     test = {}
@@ -37,7 +38,7 @@ def test_multi_skewt_plot():
         sonde_ds = sonde_ds.resample(time='30s').nearest()
         test.update({time: sonde_ds})
 
-    skewt = SkewTDisplay(test, subplot_shape=(2, 2))
+    skewt = SkewTDisplay(test, subplot_shape=(2, 2), figsize=(12, 14))
     i = 0
     j = 0
     for f in files:
@@ -52,15 +53,17 @@ def test_multi_skewt_plot():
             dsname=time,
             p_levels_to_plot=np.arange(10.0, 1000.0, 25),
         )
+        skewt.axes[j, i].set_ylim([1000, 10])
         if j == 1:
             i += 1
             j = 0
         elif j == 0:
             j += 1
+    plt.tight_layout()
     return skewt.fig
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_enhanced_skewt_plot():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     display = act.plotting.SkewTDisplay(ds)
@@ -69,7 +72,7 @@ def test_enhanced_skewt_plot():
     return display.fig
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_enhanced_skewt_plot_2():
     ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     display = act.plotting.SkewTDisplay(ds)
@@ -87,7 +90,7 @@ def test_enhanced_skewt_plot_2():
     return display.fig
 
 
-@pytest.mark.mpl_image_compare(tolerance=30)
+@pytest.mark.mpl_image_compare(tolerance=10)
 def test_skewt_options():
     sonde_ds = act.io.arm.read_arm_netcdf(sample_files.EXAMPLE_SONDE1)
     skewt = SkewTDisplay(sonde_ds)