From fa37838aeee1ae2325eb2b48f20a564f913d7a42 Mon Sep 17 00:00:00 2001
From: Dongdong Tian <seisman.info@gmail.com>
Date: Mon, 8 Jul 2024 21:56:58 +0800
Subject: [PATCH] Add tests for reading images

---
 pygmt/tests/test_clib_read_data.py | 80 +++++++++++++++++++++++++++++-
 1 file changed, 79 insertions(+), 1 deletion(-)

diff --git a/pygmt/tests/test_clib_read_data.py b/pygmt/tests/test_clib_read_data.py
index 934e22bf536..8349ffd1d4c 100644
--- a/pygmt/tests/test_clib_read_data.py
+++ b/pygmt/tests/test_clib_read_data.py
@@ -7,7 +7,7 @@
 
 import numpy as np
 from pygmt.clib import Session
-from pygmt.datatypes import _GMT_DATASET, _GMT_GRID
+from pygmt.datatypes import _GMT_DATASET, _GMT_GRID, _GMT_IMAGE
 from pygmt.helpers import GMTTempFile
 
 
@@ -143,3 +143,81 @@ def test_clib_read_data_image_as_grid():
         assert header.wesn[:] == [-180.0, 180.0, -90.0, 90.0]
 
         assert image.data
+
+
+def test_clib_read_data_image():
+    """
+    Test the Session.read_data method for images.
+    """
+    with Session() as lib:
+        data_ptr = lib.read_data(
+            "GMT_IS_IMAGE",
+            "GMT_IS_SURFACE",
+            "GMT_CONTAINER_AND_DATA",
+            None,
+            "@earth_day_01d_p",
+            None,
+        )
+        image = ctp.cast(data_ptr, ctp.POINTER(_GMT_IMAGE)).contents
+        header = image.header.contents
+        assert header.n_rows == 180
+        assert header.n_columns == 360
+        assert header.n_bands == 3
+        assert header.wesn[:] == [-180.0, 180.0, -90.0, 90.0]
+
+        assert image.data
+
+
+def test_clib_read_data_image_two_steps():
+    """
+    Test the Session.read_data method for images in two steps, first reading the header
+    and then the data.
+    """
+    family, geometry = "GMT_IS_IMAGE", "GMT_IS_SURFACE"
+    infile = "@earth_day_01d_p"
+    with Session() as lib:
+        # Read the header first
+        data_ptr = lib.read_data(
+            family, geometry, "GMT_CONTAINER_ONLY", None, infile, None
+        )
+        image = ctp.cast(data_ptr, ctp.POINTER(_GMT_IMAGE)).contents
+        header = image.header.contents
+        assert header.n_rows == 180
+        assert header.n_columns == 360
+        assert header.n_bands == 3
+        assert header.wesn[:] == [-180.0, 180.0, -90.0, 90.0]
+
+        assert not image.data  # The data is not read yet
+
+        # Read the data
+        data_ptr = lib.read_data(
+            family, geometry, "GMT_DATA_ONLY", None, infile, data_ptr
+        )
+        image = ctp.cast(data_ptr, ctp.POINTER(_GMT_GRID)).contents
+
+        assert image.data is not None  # The data is read
+
+
+def test_clib_read_data_grid_as_image():
+    """
+    Test the Session.read_data method for reading a grid as an image.
+
+    Same as test_clib_read_data_image() but with a grid file.
+    """
+    with Session() as lib:
+        data_ptr = lib.read_data(
+            "GMT_IS_IMAGE",
+            "GMT_IS_SURFACE",
+            "GMT_CONTAINER_AND_DATA",
+            None,
+            "@earth_relief_01d_p",
+            None,
+        )
+        image = ctp.cast(data_ptr, ctp.POINTER(_GMT_IMAGE)).contents
+        header = image.header.contents
+        assert header.n_rows == 180
+        assert header.n_columns == 360
+        assert header.n_bands == 1
+        assert header.wesn[:] == [-179.5, 179.5, -89.5, 89.5]
+
+        assert image.data