Add ctypes typehints

numpy · shoyer · Feb 19, 2018 · Feb 15, 2018 · Feb 15, 2018 · Feb 18, 2018
commit f6efbfdffd871959116a70f625ca5b25052f2e53
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
@@ -2,6 +2,8 @@
 #
 from typing import Any, Tuple, Optional
 
+from numpy.core._internal import _ctypes
+
 class dtype: pass
 _dtype = dtype      # for ndarray type
 
@@ -51,7 +53,7 @@ class ndarray:
     ndim: int
     shape: Tuple[int, ...]
     strides: Tuple[int, ...]
-    ctypes: Any
+    ctypes: _ctypes
     base: Optional[ndarray]
 
 

diff --git a/numpy/core/__init__.pyi b/numpy/core/__init__.pyi
diff --git a/numpy/core/_internal.pyi b/numpy/core/_internal.pyi
@@ -0,0 +1,21 @@
+from typing import Any
+
+# TODO: add better annotations when ctypes is stubbed out
+
+class _ctypes:
+    @property
+    def data(self) -> int: ...
+
+    @property
+    def shape(self) -> Any: ...
+
+    @property
+    def strides(self) -> Any: ...
+
+    def data_as(self, obj: Any) -> Any: ...
+    def shape_as(self, obj: Any) -> Any: ...
+    def strides_as(self, obj: Any) -> Any: ...
+    def get_data(self) -> int: ...
+    def get_shape(self) -> Any: ...
+    def get_strides(self) -> Any: ...
+    def get_as_parameter(self) -> Any: ...