Implemented Warning and Exception annotations

See https://github.com/numpy/numpy-stubs/issues/54. Added annotations and tests for: * np.warnings * np.ModuleDeprecationWarning * np.VisibleDeprecationWarning * np.ComplexWarning * np.RankWarning * np.TooHardError * np.AxisError
numpy · person142 · Apr 18, 2020 · Apr 18, 2020 · Apr 18, 2020 · Apr 18, 2020
commit 2a1c59cd0cddee45a0746e77a28b38e76dd6838d
diff --git a/numpy-stubs/__init__.pyi b/numpy-stubs/__init__.pyi
@@ -1,6 +1,7 @@
 import builtins
 import sys
 import datetime as dt
+import warnings as _warnings
 
 from numpy.core._internal import _ctypes
 from typing import (
@@ -778,3 +779,17 @@ trunc: ufunc
 
 # TODO(shoyer): remove when the full numpy namespace is defined
 def __getattr__(name: str) -> Any: ...
+
+# Warnings
+warnings = _warnings
+class ModuleDeprecationWarning(DeprecationWarning): ...
+class VisibleDeprecationWarning(UserWarning): ...
+class ComplexWarning(RuntimeWarning): ...
+class RankWarning(UserWarning): ...
+
+# Errors
+class TooHardError(RuntimeError): ...
+class AxisError(ValueError, IndexError):
+    # Note that all parameters for __init__ (except self) can technically be Any.
+    # So it's a question here whether or not we should allow what's possible or what's intended
+    def __init__(self, axis: int, ndim: Optional[int] = ..., msg_prefix: Optional[str] = ...) -> None: ...
diff --git a/tests/pass/warnings_and_errors.py b/tests/pass/warnings_and_errors.py
@@ -0,0 +1,18 @@
+import types
+from typing import Type
+
+import numpy as np
+
+warnings: types.ModuleType = np.warnings
+ModuleDeprecationWarning: Type[DeprecationWarning] = np.ModuleDeprecationWarning
+VisibleDeprecationWarning: Type[UserWarning] = np.VisibleDeprecationWarning
+ComplexWarning: Type[RuntimeWarning] = np.ComplexWarning
+RankWarning: Type[UserWarning] = np.RankWarning
+
+TooHardError: Type[RuntimeError] = np.TooHardError
+AxisError1: Type[ValueError] = np.AxisError
+AxisError2: Type[IndexError] = np.AxisError
+
+np.AxisError(1)
+np.AxisError(1, ndim=2)
+np.AxisError(1, ndim=2, msg_prefix='error')