Try passing "GMT_IS_VECTOR" family type to put_strings

Author: weiji14

pygmt/clib/session.py

@@ -56,7 +56,6 @@
     np.uint64: "GMT_ULONG",
     np.uint32: "GMT_UINT",
     np.datetime64: "GMT_DATETIME",
-    np.str_: "GMT_TEXT",
 }
 
 
@@ -792,15 +791,15 @@ def put_vector(self, dataset, column, vector):
                 )
             )
 
-    def put_strings(self, dataset, strings):
+    def put_strings(self, dataset, family, strings):
         """
         Attach a numpy 1D array of dtype str as a column on a GMT dataset.
 
         Use this function to attach string type numpy array data to a GMT
         dataset and pass it to GMT modules. Wraps ``GMT_Put_Strings``.
 
         The dataset must be created by :meth:`~gmt.clib.Session.create_data`
-        first. Use ``family='GMT_IS_DATASET|GMT_VIA_VECTOR'``.
+        first.
 
         .. warning::
             The numpy array must be C contiguous in memory. If it comes from a
@@ -813,6 +812,9 @@ def put_strings(self, dataset, strings):
         dataset : :class:`ctypes.c_void_p`
             The ctypes void pointer to a ``GMT_Dataset``. Create it with
             :meth:`~gmt.clib.Session.create_data`.
+        family : str
+            The family type of the dataset. Can be either ``GMT_IS_VECTOR`` or
+            ``GMT_IS_MATRIX``.
         strings : numpy 1d-array
             The array that will be attached to the dataset. Must be a 1d C
             contiguous array.
@@ -830,9 +832,10 @@ def put_strings(self, dataset, strings):
             restype=ctp.c_int,
         )
 
-        gmt_type = self._check_dtype_and_dim(strings, ndim=1)
-        strings_pointer = strings.ctypes.data_as(ctp.c_void_p)
-        status = c_put_strings(self.session_pointer, dataset, gmt_type, strings_pointer)
+        strings_pointer = strings.ctypes.data_as(ctp.c_char_p)
+        status = c_put_strings(
+            self.session_pointer, self[family], dataset, strings_pointer
+        )
         if status != 0:
             raise GMTCLibError(
                 f"Failed to put strings of type {strings.dtype} into dataset"

pygmt/tests/test_clib_put.py

@@ -21,7 +21,7 @@ def test_put_strings():
             dim=[1, 5, 1, 0],  # columns, rows, layers, dtype
         )
         s = np.array(["a", "b", "c", "d", "e"], dtype=np.str)
-        lib.put_strings(dataset, strings=s)
+        lib.put_strings(dataset, family="GMT_IS_VECTOR", strings=s)
         # Turns out wesn doesn't matter for Datasets
         wesn = [0] * 6
         # Save the data to a file to see if it's being accessed correctly