pygmt.which: Fix the bug when passing multiple files (#2726)

Author: seisman

pygmt/src/which.py

@@ -6,14 +6,13 @@
     GMTTempFile,
     build_arg_string,
     fmt_docstring,
-    kwargs_to_strings,
+    is_nonstr_iter,
     use_alias,
 )
 
 
 @fmt_docstring
 @use_alias(G="download", V="verbose")
-@kwargs_to_strings(fname="sequence_space")
 def which(fname, **kwargs):
     r"""
     Find the full path to specified files.
@@ -63,6 +62,9 @@ def which(fname, **kwargs):
     FileNotFoundError
         If the file is not found.
     """
+    if is_nonstr_iter(fname):  # Got a list of files
+        fname = " ".join(fname)
+
     with GMTTempFile() as tmpfile:
         with Session() as lib:
             lib.call_module(

pygmt/tests/test_which.py

@@ -1,7 +1,7 @@
 """
 Test pygmt.which.
 """
-import os
+from pathlib import Path
 
 import pytest
 from pygmt import which
@@ -13,20 +13,20 @@ def test_which():
     Make sure `which` returns file paths for @files correctly without errors.
     """
     for fname in ["tut_quakes.ngdc", "tut_bathy.nc"]:
-        cached_file = which(f"@{fname}", download="c")
-        assert os.path.exists(cached_file)
-        assert os.path.basename(cached_file) == fname
+        cached_file = which(fname=f"@{fname}", download="c")
+        assert Path(cached_file).exists()
+        assert Path(cached_file).name == fname
 
 
 def test_which_multiple():
     """
     Make sure `which` returns file paths for multiple @files correctly.
     """
     filenames = ["ridge.txt", "tut_ship.xyz"]
-    cached_files = which(fname=[f"@{fname}" for fname in filenames], download="c")
+    cached_files = which([f"@{fname}" for fname in filenames], download="c")
     for cached_file in cached_files:
-        assert os.path.exists(cached_file)
-        assert os.path.basename(cached_file) in filenames
+        assert Path(cached_file).exists()
+        assert Path(cached_file).name in filenames
 
 
 def test_which_fails():