Add tests for Python built-in types

GenericMappingTools · Nov 6, 2024 · 6f966db · 6f966db
1 parent a9635b5
commit 6f966db
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/pygmt/tests/test_clib_to_numpy.py b/pygmt/tests/test_clib_to_numpy.py
@@ -21,6 +21,25 @@ def _check_result(result, supported):
     assert (result.dtype.type in DTYPES) == supported
 
 
+########################################################################################
+# Test the _to_numpy function with Python built-in types.
+########################################################################################
+@pytest.mark.parametrize(
+    ("data", "supported"),
+    [
+        pytest.param([1, 2, 3], True, id="int"),
+        pytest.param([1.0, 2.0, 3.0], True, id="float"),
+    ],
+)
+def test_to_numpy_python_types_numeric(data, supported):
+    """
+    Test the _to_numpy function with Python built-in numeric types.
+    """
+    result = _to_numpy(data)
+    _check_result(result, supported)
+    npt.assert_array_equal(result, data)
+
+
 ########################################################################################
 # Test the _to_numpy function with NumPy arrays.
 #