Use virtualfile_from_matrix on int/uint/float types and add a test

Co-Authored-By: Dongdong Tian <[email protected]>

Author: weiji14

pygmt/clib/session.py

@@ -1380,7 +1380,7 @@ def virtualfile_from_data(
             raster grid, a vector matrix/arrays, or other supported data input.
         x/y/z : 1d arrays or None
             x, y and z columns as numpy arrays.
-        extra_arrays : list
+        extra_arrays : list of 1d arrays
             Optional. A list of numpy arrays in addition to x, y and z. All
             of these arrays must be of the same size as the x/y/z arrays.
 
@@ -1446,9 +1446,10 @@ def virtualfile_from_data(
                 _data = [array for _, array in data.items()]
             except AttributeError:
                 try:
-                    # Just use virtualfile_from_matrix for 2D
-                    # numpy.ndarray which are not datetime (M) types
-                    assert data.ndim == 2 and not data.dtype.kind == "M"
+                    # Just use virtualfile_from_matrix for 2D numpy.ndarray
+                    # which are signed integer (i), unsigned integer (u) or
+                    # floating point (f) types
+                    assert data.ndim == 2 and data.dtype.kind in "iuf"
                     _virtualfile_from = self.virtualfile_from_matrix
                     _data = (data,)
                 except (AssertionError, AttributeError):

pygmt/tests/test_info.py

@@ -28,6 +28,14 @@ def test_info():
     )
     assert output == expected_output
 
+def test_info_2d_list():
+    """
+    Make sure info works on a 2d list.
+    """
+    output = info(table=[[0, 8], [3, 5], [6, 2]])
+    expected_output = "<vector memory>: N = 3 <0/6> <2/8>\n"
+    assert output == expected_output
+
 
 def test_info_dataframe():
     """