Change test_makecpt.py to use colorbar instead of grdimage (#1748)

willschlitzer · weiji14 · seisman · web-flow · commit ab23aa887bd7 · 2022-02-26T15:10:59.000+08:00
Co-authored-by: Wei Ji &lt;23487320+weiji14@users.noreply.github.com&gt;
Co-authored-by: Dongdong Tian &lt;seisman.info@gmail.com&gt;
diff --git a/pygmt/tests/baseline/test_makecpt_categorical.png.dvc b/pygmt/tests/baseline/test_makecpt_categorical.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 9a8c5ecec5f8962987635057f577a01c
-  size: 6029
+- md5: 49bc3322fbb073626fe5f61b92a473fb
+  size: 4692
   path: test_makecpt_categorical.png
diff --git a/pygmt/tests/baseline/test_makecpt_continuous.png.dvc b/pygmt/tests/baseline/test_makecpt_continuous.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 9a18c81caa92725aa9de5c2dd7d74e46
-  size: 75856
+- md5: 28eec5c7c5b76ba0659c20dd60ed8108
+  size: 2861
   path: test_makecpt_continuous.png
diff --git a/pygmt/tests/baseline/test_makecpt_cyclic.png.dvc b/pygmt/tests/baseline/test_makecpt_cyclic.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 4ad410eb2843b3b47818121563c79254
-  size: 4628
+- md5: 150c12b8d8959561bdf36d380dc23d10
+  size: 3747
   path: test_makecpt_cyclic.png
diff --git a/pygmt/tests/baseline/test_makecpt_plot_colorbar.png.dvc b/pygmt/tests/baseline/test_makecpt_plot_colorbar.png.dvc
@@ -0,0 +1,4 @@
+outs:
+- md5: 989d9bf57f716f4721f14992bf39a359
+  size: 2529
+  path: test_makecpt_plot_colorbar.png
diff --git a/pygmt/tests/baseline/test_makecpt_plot_colorbar_scaled_with_series.png.dvc b/pygmt/tests/baseline/test_makecpt_plot_colorbar_scaled_with_series.png.dvc
@@ -0,0 +1,4 @@
+outs:
+- md5: dbf2dcd2c30bdb921599a09b0ba258f8
+  size: 3386
+  path: test_makecpt_plot_colorbar_scaled_with_series.png
diff --git a/pygmt/tests/baseline/test_makecpt_plot_grid.png.dvc b/pygmt/tests/baseline/test_makecpt_plot_grid.png.dvc
diff --git a/pygmt/tests/baseline/test_makecpt_plot_grid_scaled_with_series.png.dvc b/pygmt/tests/baseline/test_makecpt_plot_grid_scaled_with_series.png.dvc
diff --git a/pygmt/tests/baseline/test_makecpt_plot_points.png.dvc b/pygmt/tests/baseline/test_makecpt_plot_points.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: b28ff0dad316d0b6a860b8af7e402a56
+- md5: 7944289487b6efd9755a931e371d8022
   size: 26601
   path: test_makecpt_plot_points.png
diff --git a/pygmt/tests/baseline/test_makecpt_reverse_color_and_zsign.png.dvc b/pygmt/tests/baseline/test_makecpt_reverse_color_and_zsign.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 6a5ca01aebcc23b00dc884fcd9c927d8
-  size: 141567
+- md5: 412c8de63e53ebac483e0fd42b79cad6
+  size: 3620
   path: test_makecpt_reverse_color_and_zsign.png
diff --git a/pygmt/tests/baseline/test_makecpt_reverse_color_only.png.dvc b/pygmt/tests/baseline/test_makecpt_reverse_color_only.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 5ad9b6aab72e45be8a1adc20f9e1c0e9
-  size: 71377
+- md5: 3d919f2db5ae1302f38a0178224074c6
+  size: 3964
   path: test_makecpt_reverse_color_only.png
diff --git a/pygmt/tests/baseline/test_makecpt_truncated_zlow_zhigh.png.dvc b/pygmt/tests/baseline/test_makecpt_truncated_zlow_zhigh.png.dvc
@@ -1,4 +1,4 @@
 outs:
-- md5: 26f3d0d95a68799ddd4eb8a130bc1445
-  size: 97790
+- md5: 8235d66d1606fce2380f7fd034e874c1
+  size: 2907
   path: test_makecpt_truncated_zlow_zhigh.png
diff --git a/pygmt/tests/test_makecpt.py b/pygmt/tests/test_makecpt.py
@@ -6,7 +6,6 @@
 import numpy as np
 import pytest
 from pygmt import Figure, makecpt
-from pygmt.datasets import load_earth_relief
 from pygmt.exceptions import GMTInvalidInput
 from pygmt.helpers import GMTTempFile
 
@@ -22,24 +21,16 @@ def fixture_points():
     return np.loadtxt(POINTS_DATA)
 
 
-@pytest.fixture(scope="module", name="region")
-def fixture_region():
+@pytest.fixture(scope="module", name="position")
+def fixture_position():
     """
-    The data region.
+    Return a standard position argument for the colorbar.
     """
-    return [10, 70, -5, 10]
-
-
-@pytest.fixture(scope="module", name="grid")
-def fixture_grid():
-    """
-    Load the grid data from the sample earth_relief file.
-    """
-    return load_earth_relief(registration="gridline")
+    return "x0c/0c+w10c+h"
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_plot_points(points, region):
+def test_makecpt_plot_points(points):
     """
     Use static color palette table to change color of points.
     """
@@ -49,33 +40,33 @@ def test_makecpt_plot_points(points, region):
         x=points[:, 0],
         y=points[:, 1],
         color=points[:, 2],
-        region=region,
+        region=[10, 70, -5, 10],
         style="c1c",
         cmap=True,
     )
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_plot_grid(grid):
+def test_makecpt_plot_colorbar(position):
     """
-    Use static color palette table to change color of grid.
+    Use static color palette table to plot a colorbar.
     """
     fig = Figure()
     makecpt(cmap="relief")
-    fig.grdimage(grid, projection="W0/10c")
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_plot_grid_scaled_with_series(grid):
+def test_makecpt_plot_colorbar_scaled_with_series(position):
     """
-    Use static color palette table scaled to a min/max series to change color
-    of grid.
+    Use static color palette table scaled to a min/max series and plot it on a
+    colorbar.
     """
     fig = Figure()
-    makecpt(cmap="oleron", series=[-4500, 4500])
-    fig.grdimage(grid, projection="W0/10c")
+    makecpt(cmap="oleron", series=[0, 1000])
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
@@ -105,70 +96,70 @@ def test_makecpt_invalid_output():
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_truncated_zlow_zhigh(grid):
+def test_makecpt_truncated_zlow_zhigh(position):
     """
     Use static color palette table that is truncated to z-low and z-high.
     """
     fig = Figure()
-    makecpt(cmap="rainbow", truncate=[0.15, 0.85], series=[-4500, 4500])
-    fig.grdimage(grid, projection="W0/10c")
+    makecpt(cmap="rainbow", truncate=[0.15, 0.85], series=[0, 1000])
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_reverse_color_only(grid):
+def test_makecpt_reverse_color_only(position):
     """
     Use static color palette table with its colors reversed.
     """
     fig = Figure()
-    makecpt(cmap="earth", reverse=True)
-    fig.grdimage(grid, projection="W0/10c")
+    makecpt(cmap="earth", reverse=True, series=[0, 1000])
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_reverse_color_and_zsign(grid):
+def test_makecpt_reverse_color_and_zsign(position):
     """
     Use static color palette table with both its colors and z-value sign
     reversed.
     """
     fig = Figure()
-    makecpt(cmap="earth", reverse="cz")
-    fig.grdimage(grid, projection="W0/10c")
+    makecpt(cmap="earth", reverse="cz", series=[0, 1000])
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_continuous(grid):
+def test_makecpt_continuous(position):
     """
     Use static color palette table that is continuous from blue to white and
-    scaled from -4500 to 4500m.
+    scaled from 0 to 1000 m.
     """
     fig = Figure()
-    makecpt(cmap="blue,white", continuous=True, series=[-4500, 4500])
-    fig.grdimage(grid, projection="W0/10c")
+    makecpt(cmap="blue,white", continuous=True, series=[0, 1000])
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_categorical(region):
+def test_makecpt_categorical(position):
     """
     Use static color palette table that is categorical.
     """
     fig = Figure()
     makecpt(cmap="categorical", categorical=True)
-    fig.colorbar(cmap=True, region=region, frame=True, position="JBC")
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
 @pytest.mark.mpl_image_compare
-def test_makecpt_cyclic(region):
+def test_makecpt_cyclic(position):
     """
     Use static color palette table that is cyclic.
     """
     fig = Figure()
     makecpt(cmap="cork", cyclic=True)
-    fig.colorbar(cmap=True, region=region, frame=True, position="JBC")
+    fig.colorbar(cmap=True, frame=True, position=position)
     return fig
 
 
