Fix more tests

Author: seisman

pygmt/tests/test_clib_put_vector.py

@@ -10,7 +10,7 @@
 import pytest
 from pygmt import clib
 from pygmt.clib.session import DTYPES_NUMERIC
-from pygmt.exceptions import GMTCLibError, GMTInvalidInput
+from pygmt.exceptions import GMTCLibError, GMTInvalidInput, GMTTypeError
 from pygmt.helpers import GMTTempFile
 
 
@@ -225,7 +225,7 @@ def test_put_vector_invalid_dtype():
                 dim=[2, 3, 0, 0],  # ncolumns, nrows, dtype, unused
             )
             data = np.array([37, 12, 556], dtype=dtype)
-            with pytest.raises(GMTInvalidInput, match="Unsupported numpy data type"):
+            with pytest.raises(GMTTypeError, match="Invalida data type"):
                 lib.put_vector(dataset, column=0, vector=data)
 
 

pygmt/tests/test_grdcontour.py

@@ -7,7 +7,7 @@
 import numpy as np
 import pytest
 from pygmt import Figure
-from pygmt.exceptions import GMTInvalidInput
+from pygmt.exceptions import GMTTypeError
 from pygmt.helpers.testing import load_static_earth_relief
 
 TEST_CONTOUR_FILE = Path(__file__).parent / "data" / "contours.txt"
@@ -123,5 +123,5 @@ def test_grdcontour_fails():
     Should fail for unrecognized input.
     """
     fig = Figure()
-    with pytest.raises(GMTInvalidInput):
+    with pytest.raises(GMTTypeError):
         fig.grdcontour(np.arange(20).reshape((4, 5)))

pygmt/tests/test_info.py

@@ -11,7 +11,7 @@
 import pytest
 import xarray as xr
 from pygmt import info
-from pygmt.exceptions import GMTInvalidInput
+from pygmt.exceptions import GMTTypeError
 from pygmt.helpers.testing import skip_if_no
 
 POINTS_DATA = Path(__file__).parent / "data" / "points.txt"
@@ -245,7 +245,7 @@ def test_info_fails():
     Make sure info raises an exception if not given either a file name, pandas
     DataFrame, or numpy ndarray.
     """
-    with pytest.raises(GMTInvalidInput):
+    with pytest.raises(GMTTypeError):
         info(data=xr.DataArray(21))
 
 

pygmt/tests/test_nearneighbor.py

@@ -11,7 +11,7 @@
 from pygmt import nearneighbor
 from pygmt.datasets import load_sample_data
 from pygmt.enums import GridRegistration, GridType
-from pygmt.exceptions import GMTInvalidInput
+from pygmt.exceptions import GMTTypeError
 from pygmt.helpers import GMTTempFile
 
 
@@ -62,7 +62,7 @@ def test_nearneighbor_wrong_kind_of_input(ship_data):
     Run nearneighbor using grid input that is not file/matrix/vectors.
     """
     data = ship_data.bathymetry.to_xarray()  # convert pandas.Series to xarray.DataArray
-    with pytest.raises(GMTInvalidInput):
+    with pytest.raises(GMTTypeError):
         nearneighbor(
             data=data, spacing="5m", region=[245, 255, 20, 30], search_radius="10m"
         )

pygmt/tests/test_triangulate.py

@@ -10,7 +10,7 @@
 import xarray as xr
 from pygmt import triangulate, which
 from pygmt.enums import GridRegistration, GridType
-from pygmt.exceptions import GMTInvalidInput
+from pygmt.exceptions import GMTInvalidInput, GMTTypeError
 from pygmt.helpers import GMTTempFile
 
 
@@ -94,7 +94,7 @@ def test_delaunay_triples_wrong_kind_of_input(dataframe):
     Run triangulate.delaunay_triples using grid input that is not file/matrix/vectors.
     """
     data = dataframe.z.to_xarray()  # convert pandas.Series to xarray.DataArray
-    with pytest.raises(GMTInvalidInput):
+    with pytest.raises(GMTTypeError):
         triangulate.delaunay_triples(data=data)