Add ternary sample dataset

pygmt/datasets/samples.py

@@ -24,6 +24,7 @@ def list_sample_data():
     """
     names = {
         "bathymetry": "Table of ship bathymetric observations off Baja California",
+        "rock_compositions": "Table of rock sample compositions",
         "earth_relief_holes": "Regional 20 arc-minute Earth relief grid with holes",
         "fractures": "Table of hypothetical fracture lengths and azimuths",
         "hotspots": "Table of locations, names, and symbol sizes of hotpots from "
@@ -80,6 +81,7 @@ def load_sample_data(name):
 
     # Dictionary of private load functions
     load_func = {
+        "rock_compositions": _load_rock_sample_compositions,
         "earth_relief_holes": _load_earth_relief_holes,
         "maunaloa_co2": _load_maunaloa_co2,
         "notre_dame_topography": _load_notre_dame_topography,
@@ -360,6 +362,21 @@ def load_mars_shape(**kwargs):
     return data
 
 
+def _load_rock_sample_compositions():
+    """
+    Loads the remote file @ternary.txt.
+
+    Returns
+    -------
+    data : pandas.DataFrame
+        The data table with columns "water", "air", "limestone", "z".
+    """
+
+    fname = which("@ternary.txt", download="c")
+    data = pd.read_csv(fname, sep="\t", header=None, names=["water", "air", "limestone", "z"])
+    return data
+
+
 def _load_notre_dame_topography():
     """
     Load Table 5.11 in Davis: Statistics and Data Analysis in Geology.

pygmt/tests/test_datasets_samples.py

@@ -186,3 +186,20 @@ def test_maunaloa_co2():
     assert summary.loc["max", "date"] == 2019.3699
     assert summary.loc["min", "co2_ppm"] == 313.2
     assert summary.loc["max", "co2_ppm"] == 414.83
+
+
+def test_ternary():
+    """
+    Check that the @ternary.txt dataset loads without errors.
+    """
+    data = load_sample_data(name="rock_compositions")
+    assert data.shape == (1000, 4)
+    summary = data.describe()
+    assert summary.loc["min", "water"] == 0
+    assert summary.loc["max", "water"] == 1
+    assert summary.loc["min", "air"] == 0
+    assert summary.loc["max", "air"] == 0.921
+    assert summary.loc["min", "limestone"] == 0
+    assert summary.loc["max", "limestone"] == 0.981
+    assert summary.loc["min", "z"] == 1.041
+    assert summary.loc["max", "z"] == 70.844