add test

Author: willschlitzer

pygmt/datasets/samples.py

@@ -4,6 +4,7 @@
 import warnings
 
 import pandas as pd
+import xarray as xr
 from pygmt.exceptions import GMTInvalidInput
 from pygmt.src import which
 
@@ -351,15 +352,11 @@ def _load_static_earth_relief(**kwargs):
     """
     Load the static_earth_relief file for internal testing.
 
-    Parameters
-    ----------
-    kwargs
-
     Returns
     -------
+    data : xarray.DataArray
+        A grid of Earth relief for internal tests.
     """
-    fname = which("@static_earth_relief", download="c")
-    data = pd.read_csv(
-        fname, sep="\t", header=None, names=["longitude", "latitude", "bathymetry"]
-    )
+    fname = which("@static_earth_relief.nc", download="c")
+    data = xr.open_dataarray(fname)
     return data

pygmt/tests/test_datasets_samples.py

@@ -1,8 +1,10 @@
 """
 Test basic functionality for loading sample datasets.
 """
+import numpy.testing as npt
 import pandas as pd
 import pytest
+import xarray as xr
 from pygmt.datasets import (
     load_fractures_compilation,
     load_hotspots,
@@ -145,3 +147,17 @@ def test_hotspots():
         "place_name",
     ]
     assert isinstance(data, pd.DataFrame)
+
+
+def test_load_static_earth_relief():
+    """
+    Check that @static_earth_relief.nc loads without errors.
+    """
+    data = load_sample_data("static_earth_relief")
+    assert data.dims == ("lat", "lon")
+    assert data.shape == (14, 8)
+    assert data.min() == 190
+    assert data.max() == 981
+    assert data.median() == 467
+    npt.assert_allclose(data.mean(), 490.87946)
+    assert isinstance(data, xr.DataArray)