TST: Add typing tests for np.ctypeslib.c_intp and its mypy plugin

numpy · May 21, 2021 · 5eca599 · 5eca599
1 parent 869243e
commit 5eca599
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/numpy/typing/tests/data/reveal/ctypeslib.py b/numpy/typing/tests/data/reveal/ctypeslib.py
@@ -0,0 +1,3 @@
+import numpy as np
+
+reveal_type(np.ctypeslib.c_intp())  # E: {c_intp}
diff --git a/numpy/typing/tests/data/reveal/ndarray_misc.py b/numpy/typing/tests/data/reveal/ndarray_misc.py
@@ -23,8 +23,8 @@ class SubClass(np.ndarray): ...
 ctypes_obj = AR_f8.ctypes
 
 reveal_type(ctypes_obj.data)  # E: int
-reveal_type(ctypes_obj.shape)  # E: ctypes.Array[ctypes.c_int64]
-reveal_type(ctypes_obj.strides)  # E: ctypes.Array[ctypes.c_int64]
+reveal_type(ctypes_obj.shape)  # E: ctypes.Array[{c_intp}]
+reveal_type(ctypes_obj.strides)  # E: ctypes.Array[{c_intp}]
 reveal_type(ctypes_obj._as_parameter_)  # E: ctypes.c_void_p
 
 reveal_type(ctypes_obj.data_as(ct.c_void_p))  # E: ctypes.c_void_p

diff --git a/numpy/typing/tests/test_typing.py b/numpy/typing/tests/test_typing.py
@@ -8,7 +8,11 @@
 
 import pytest
 import numpy as np
-from numpy.typing.mypy_plugin import _PRECISION_DICT, _EXTENDED_PRECISION_LIST
+from numpy.typing.mypy_plugin import (
+    _PRECISION_DICT,
+    _EXTENDED_PRECISION_LIST,
+    _C_INTP,
+)
 
 try:
     from mypy import api
@@ -219,6 +223,9 @@ def _construct_format_dict():
 
         # numpy.typing
         "_NBitInt": dct['_NBitInt'],
+
+        # numpy.ctypeslib
+        "c_intp": f"ctypes.{_C_INTP}"
     }