Expand existing tests to test int type colors

GenericMappingTools · Oct 4, 2021 · 5bc6a4c · 5bc6a4c
1 parent f2ad5e5
commit 5bc6a4c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/pygmt/tests/test_plot.py b/pygmt/tests/test_plot.py
@@ -302,12 +302,13 @@ def test_plot_sizes_colors_transparencies():
     return fig
 
 
-@pytest.mark.mpl_image_compare
+@pytest.mark.mpl_image_compare(filename="test_plot_matrix.png")
+@pytest.mark.parametrize("color", ["#aaaaaa", 170])
 @pytest.mark.xfail(
     condition=gmt_version <= Version("6.2.0"),
     reason="Upstream bug fixed in https://github.com/GenericMappingTools/gmt/pull/5799.",
 )
-def test_plot_matrix(data):
+def test_plot_matrix(data, color):
     """
     Plot the data passing in a matrix and specifying columns.
     """
@@ -317,7 +318,7 @@ def test_plot_matrix(data):
         region=[10, 70, -5, 10],
         projection="M15c",
         style="cc",
-        color="#aaaaaa",
+        color=color,
         frame="a",
         incols="0,1,2+s0.5",
     )

diff --git a/pygmt/tests/test_plot3d.py b/pygmt/tests/test_plot3d.py
@@ -313,7 +313,9 @@ def test_plot3d_sizes_colors_transparencies():
 
 
 @pytest.mark.mpl_image_compare
-def test_plot3d_matrix(data, region):
+@pytest.mark.mpl_image_compare(filename="test_plot3d_matrix.png")
+@pytest.mark.parametrize("color", ["#aaaaaa", 170])
+def test_plot3d_matrix(data, region, color):
     """
     Plot the data passing in a matrix and specifying incols.
     """
@@ -325,7 +327,7 @@ def test_plot3d_matrix(data, region):
         region=region,
         projection="M20c",
         style="c1c",
-        color="#aaaaaa",
+        color=color,
         frame=["a", "za"],
         incols="0,1,2",
     )