Explicitly checking array.dtype.type

seisman · seisman · commit 18645561f905 · 2024-10-14T18:50:30.000+08:00
diff --git a/pygmt/tests/test_clib_vectors_to_arrays.py b/pygmt/tests/test_clib_vectors_to_arrays.py
@@ -84,7 +84,7 @@ def test_vectors_to_arrays_pandas_string():
         pd.Series(["abcdef", "123456"], dtype="string"),
     ]
     arrays = vectors_to_arrays(vectors)
-    assert all(isinstance(i.dtype, np.dtypes.StrDType) for i in arrays)
+    assert all(i.dtype.type == np.str_ for i in arrays)
     _check_arrays(arrays)
 
 
@@ -104,5 +104,5 @@ def test_vectors_to_arrays_pyarrow_datetime():
         ),
     ]
     arrays = vectors_to_arrays(vectors)
-    assert all(isinstance(i.dtype, np.dtypes.DateTime64DType) for i in arrays)
+    assert all(i.dtype.type == np.datetime64 for i in arrays)
     _check_arrays(arrays)

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ def test_vectors_to_arrays_pandas_string():`
`84`	`84`	`pd.Series(["abcdef", "123456"], dtype="string"),`
`85`	`85`	`]`
`86`	`86`	`arrays = vectors_to_arrays(vectors)`
`87`		`- assert all(isinstance(i.dtype, np.dtypes.StrDType) for i in arrays)`
	`87`	`+ assert all(i.dtype.type == np.str_ for i in arrays)`
`88`	`88`	`_check_arrays(arrays)`
`89`	`89`
`90`	`90`
`@@ -104,5 +104,5 @@ def test_vectors_to_arrays_pyarrow_datetime():`
`104`	`104`	`),`
`105`	`105`	`]`
`106`	`106`	`arrays = vectors_to_arrays(vectors)`
`107`		`- assert all(isinstance(i.dtype, np.dtypes.DateTime64DType) for i in arrays)`
	`107`	`+ assert all(i.dtype.type == np.datetime64 for i in arrays)`
`108`	`108`	`_check_arrays(arrays)`