Mark unit tests with @pytest.mark.benchmark part 1

.github/workflows/benchmarks.yml

@@ -61,7 +61,7 @@ jobs:
           echo $CONDA/bin >> $GITHUB_PATH
           conda install --solver=libmamba gmt=6.4.0 python=3.12 \
                         numpy pandas xarray netCDF4 packaging \
-                        pytest pytest-benchmark pytest-mpl
+                        geopandas pytest pytest-benchmark pytest-mpl
           python -m pip install -U pytest-codspeed setuptools
 
       # Install the package that we want to test

pygmt/tests/test_geopandas.py

@@ -61,6 +61,7 @@ def fixture_gdf_ridge():
     return gdf
 
 
+@pytest.mark.benchmark
 def test_geopandas_info_geodataframe(gdf):
     """
     Check that info can return the bounding box region from a

pygmt/tests/test_grd2cpt.py

@@ -18,6 +18,7 @@ def fixture_grid():
     return load_static_earth_relief()
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_grd2cpt(grid):
     """

pygmt/tests/test_grd2xyz.py

@@ -20,6 +20,7 @@ def fixture_grid():
     return load_static_earth_relief()
 
 
+@pytest.mark.benchmark
 def test_grd2xyz(grid):
     """
     Make sure grd2xyz works as expected.

pygmt/tests/test_grdclip.py

@@ -55,6 +55,7 @@ def test_grdclip_outgrid(grid, expected_grid):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_grdclip_no_outgrid(grid, expected_grid):
     """
     Test the below and above parameters for grdclip with no set outgrid.

pygmt/tests/test_grdcontour.py

@@ -33,6 +33,7 @@ def test_grdcontour(grid):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_grdcontour_labels(grid):
     """

pygmt/tests/test_grdcut.py

@@ -53,6 +53,7 @@ def test_grdcut_dataarray_in_file_out(grid, expected_grid, region):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_grdcut_dataarray_in_dataarray_out(grid, expected_grid, region):
     """
     Test grdcut on an input DataArray, and output as DataArray.

pygmt/tests/test_grdfill.py

@@ -71,6 +71,7 @@ def fixture_expected_grid():
     )
 
 
+@pytest.mark.benchmark
 def test_grdfill_dataarray_out(grid, expected_grid):
     """
     Test grdfill with a DataArray output.

pygmt/tests/test_grdfilter.py

@@ -39,6 +39,7 @@ def fixture_expected_grid():
     )
 
 
+@pytest.mark.benchmark
 def test_grdfilter_dataarray_in_dataarray_out(grid, expected_grid):
     """
     Test grdfilter with an input DataArray, and output as DataArray.

pygmt/tests/test_grdgradient.py

@@ -53,6 +53,7 @@ def test_grdgradient_outgrid(grid, expected_grid):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_grdgradient_no_outgrid(grid, expected_grid):
     """
     Test the azimuth and direction parameters for grdgradient with no set

pygmt/tests/test_grdhisteq.py

@@ -69,7 +69,8 @@ def test_equalize_grid_outgrid_file(grid, expected_grid, region):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
-def test_equalize_grid_outgrid(grid, expected_grid, region):
+@pytest.mark.benchmark
+def test_equalize_grid_no_outgrid(grid, expected_grid, region):
     """
     Test grdhisteq.equalize_grid with ``outgrid=None``.
     """
@@ -81,6 +82,7 @@ def test_equalize_grid_outgrid(grid, expected_grid, region):
     xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_compute_bins_no_outfile(grid, expected_df, region):
     """
     Test grdhisteq.compute_bins with no ``outfile``.

pygmt/tests/test_grdimage.py

@@ -131,6 +131,7 @@ def test_grdimage_shading_xarray(grid, shading):
     return fig_ref, fig_test
 
 
+@pytest.mark.benchmark
 @check_figures_equal()
 def test_grdimage_grid_and_shading_with_xarray(grid, xrgrid):
     """

pygmt/tests/test_grdimage_image.py

@@ -53,6 +53,7 @@ def test_grdimage_image():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare(filename="test_grdimage_image.png")
 def test_grdimage_image_dataarray(xr_image):
     """

pygmt/tests/test_grdinfo.py

@@ -16,6 +16,7 @@ def fixture_grid():
     return load_static_earth_relief()
 
 
+@pytest.mark.benchmark
 def test_grdinfo(grid):
     """
     Make sure grdinfo works as expected.

pygmt/tests/test_grdlandmask.py

@@ -44,6 +44,7 @@ def test_grdlandmask_outgrid(expected_grid):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_grdlandmask_no_outgrid(expected_grid):
     """
     Test grdlandmask with no set outgrid.

pygmt/tests/test_grdproject.py

@@ -58,6 +58,7 @@ def test_grdproject_file_out(grid, expected_grid):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 @pytest.mark.parametrize(
     "projection",
     ["M10c", "EPSG:3395 +width=10", "+proj=merc +ellps=WGS84 +units=m +width=10"],

pygmt/tests/test_grdsample.py

@@ -69,6 +69,7 @@ def test_grdsample_file_out(grid, expected_grid, region, spacing):
         xr.testing.assert_allclose(a=temp_grid, b=expected_grid)
 
 
+@pytest.mark.benchmark
 def test_grdsample_dataarray_out(grid, expected_grid, region, spacing):
     """
     Test grdsample with no outgrid set and the spacing is changed.

pygmt/tests/test_grdtrack.py

@@ -55,6 +55,7 @@ def fixture_dataframe():
     )
 
 
+@pytest.mark.benchmark
 def test_grdtrack_input_dataframe_and_dataarray(dataarray, dataframe, expected_array):
     """
     Run grdtrack by passing in a pandas.DataFrame and xarray.DataArray as

pygmt/tests/test_grdview.py

@@ -214,6 +214,7 @@ def test_grdview_on_a_plane_styled_with_facadepen(xrgrid):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_grdview_drapegrid_dataarray(xrgrid):
     """

pygmt/tests/test_grdvolume.py

@@ -75,6 +75,7 @@ def test_grdvolume_no_outfile(grid):
         grdvolume(grid=grid, output_type="file")
 
 
+@pytest.mark.benchmark
 def test_grdvolume_no_outgrid(grid, data, region):
     """
     Test the expected output of grdvolume with no output file set.

pygmt/tests/test_histogram.py

@@ -15,6 +15,7 @@ def fixture_data(request):
     return request.param(data)
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare(filename="test_histogram.png")
 def test_histogram(data):
     """

pygmt/tests/test_info.py

@@ -119,6 +119,7 @@ def test_info_numpy_array_time_column():
     assert output == expected_output
 
 
+@pytest.mark.benchmark
 @pytest.mark.parametrize(
     "dtype",
     [
@@ -144,6 +145,7 @@ def test_info_pandas_dataframe_date_column(dtype):
     assert output == expected_output
 
 
+@pytest.mark.benchmark
 def test_info_xarray_dataset_time_column():
     """
     Make sure info works on xarray.Dataset 1-D inputs with a time column.

pygmt/tests/test_inset.py

@@ -5,6 +5,7 @@
 from pygmt import Figure
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_inset_aliases():
     """

pygmt/tests/test_io.py

@@ -8,6 +8,7 @@
 from pygmt.io import load_dataarray
 
 
+@pytest.mark.benchmark
 def test_io_load_dataarray():
     """
     Check that load_dataarray works to read a netCDF grid with

pygmt/tests/test_legend.py

@@ -38,6 +38,7 @@ def test_legend_default_position():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_legend_entries():
     """

pygmt/tests/test_logo.py

@@ -5,6 +5,7 @@
 from pygmt import Figure
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_logo():
     """

pygmt/tests/test_makecpt.py

@@ -71,6 +71,7 @@ def test_makecpt_plot_colorbar_scaled_with_series(position):
     return fig
 
 
+@pytest.mark.benchmark
 def test_makecpt_output_cpt_file():
     """
     Save the generated static color palette table to a .cpt file.

pygmt/tests/test_meca.py

@@ -81,6 +81,7 @@ def test_meca_spec_single_focalmecha_file():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare(filename="test_meca_spec_multiple_focalmecha.png")
 @pytest.mark.parametrize(
     "inputtype", ["dict_mecha", "dict_mecha_mixed", "dataframe", "array2d"]

pygmt/tests/test_nearneighbor.py

@@ -37,6 +37,7 @@ def test_nearneighbor_input_data(array_func, ship_data):
     npt.assert_allclose(output.mean(), -2378.2385)
 
 
+@pytest.mark.benchmark
 def test_nearneighbor_input_xyz(ship_data):
     """
     Run nearneighbor by passing in x, y, z numpy.ndarrays individually.

pygmt/tests/test_plot.py

@@ -352,6 +352,7 @@ def test_plot_from_file(region):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_plot_vectors():
     """

pygmt/tests/test_plot3d.py

@@ -378,6 +378,7 @@ def test_plot3d_from_file(region):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_plot3d_vectors():
     """

pygmt/tests/test_project.py

@@ -34,6 +34,7 @@ def test_project_generate():
     )
 
 
+@pytest.mark.benchmark
 @pytest.mark.parametrize("array_func", [np.array, pd.DataFrame, xr.Dataset])
 def test_project_input_matrix(array_func, dataframe):
     """

pygmt/tests/test_psconvert.py

@@ -8,6 +8,7 @@
 from pygmt.exceptions import GMTInvalidInput
 
 
+@pytest.mark.benchmark
 def test_psconvert():
     """
     Check that psconvert creates a figure in the current directory.

pygmt/tests/test_rose.py

@@ -91,6 +91,7 @@ def test_rose_2d_array_multiple(data):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_rose_plot_data_using_cpt(data):
     """

pygmt/tests/test_select.py

@@ -19,6 +19,7 @@ def fixture_dataframe():
     return load_sample_data(name="bathymetry")
 
 
+@pytest.mark.benchmark
 def test_select_input_dataframe(dataframe):
     """
     Run select by passing in a pandas.DataFrame as input.

pygmt/tests/test_session_management.py

@@ -3,10 +3,12 @@
 """
 import os
 
+import pytest
 from pygmt.clib import Session
 from pygmt.session_management import begin, end
 
 
+@pytest.mark.benchmark
 def test_begin_end():
     """
     Run a command inside a begin-end modern mode block.

pygmt/tests/test_solar.py

@@ -39,6 +39,7 @@ def test_solar_terminators():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare(filename="test_solar_set_terminator_datetime.png")
 @pytest.mark.parametrize(
     "terminator_datetime",

pygmt/tests/test_sph2grd.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 
 import numpy.testing as npt
+import pytest
 from pygmt import sph2grd
 from pygmt.helpers import GMTTempFile
 
@@ -20,6 +21,7 @@ def test_sph2grd_outgrid():
         assert Path(tmpfile.name).stat().st_size > 0  # check that outgrid exists
 
 
+@pytest.mark.benchmark
 def test_sph2grd_no_outgrid():
     """
     Test sph2grd with no set outgrid.

pygmt/tests/test_sphdistance.py

@@ -48,6 +48,7 @@ def test_sphdistance_outgrid(array):
         assert Path(tmpfile.name).stat().st_size > 0  # check that outgrid exists
 
 
+@pytest.mark.benchmark
 def test_sphdistance_no_outgrid(array):
     """
     Test sphdistance with no set outgrid.

pygmt/tests/test_sphinterpolate.py

@@ -28,6 +28,7 @@ def test_sphinterpolate_outgrid(mars):
         assert Path(tmpfile.name).stat().st_size > 0  # check that outgrid exists
 
 
+@pytest.mark.benchmark
 def test_sphinterpolate_no_outgrid(mars):
     """
     Test sphinterpolate with no set outgrid.

pygmt/tests/test_subplot.py

@@ -6,6 +6,7 @@
 from pygmt.exceptions import GMTInvalidInput
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_subplot_basic_frame():
     """

pygmt/tests/test_surface.py

@@ -103,6 +103,7 @@ def test_surface_input_data_array(data, region, spacing, expected_grid):
     check_values(output, expected_grid)
 
 
+@pytest.mark.benchmark
 def test_surface_input_xyz(data, region, spacing, expected_grid):
     """
     Run surface by passing in x, y, z numpy.ndarrays individually.

pygmt/tests/test_ternary.py

@@ -64,6 +64,7 @@ def test_ternary(array):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_ternary_3_labels(array):
     """

pygmt/tests/test_text.py

@@ -46,6 +46,7 @@ def test_text_single_line_of_text(region, projection):
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_text_multiple_lines_of_text(region, projection):
     """

pygmt/tests/test_tilemap.py

@@ -24,6 +24,7 @@ def test_tilemap_web_mercator():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_tilemap_ogc_wgs84():
     """

pygmt/tests/test_timestamp.py

@@ -61,6 +61,7 @@ def test_timestamp_offset():
     return fig
 
 
+@pytest.mark.benchmark
 @pytest.mark.mpl_image_compare
 def test_timestamp_font(faketime):
     """