summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2020-10-16 09:43:59 -0600
committerGitHub <noreply@github.com>2020-10-16 09:43:59 -0600
commitaebaa8d2dbff7af620a2fca3b5ca9fbb08d4c60c (patch)
tree8607a29809bac2429cdc15e7657bae6408c6beb2
parentb9dd2be0108cb312b4c34239a1dc8d24ef3a05a9 (diff)
parent50e09dbffba0d8f0b1682c3ed570885cae68fdbb (diff)
downloadnumpy-aebaa8d2dbff7af620a2fca3b5ca9fbb08d4c60c.tar.gz
Merge pull request #17572 from BvB93/constants
ENH: Add annotations for three new constants
-rw-r--r--numpy/__init__.pyi10
-rw-r--r--numpy/typing/tests/data/fail/constants.py6
-rw-r--r--numpy/typing/tests/data/reveal/constants.py7
3 files changed, 18 insertions, 5 deletions
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
index 64e4c75ce..80fb213fd 100644
--- a/numpy/__init__.pyi
+++ b/numpy/__init__.pyi
@@ -220,11 +220,8 @@ __all__ = [
]
DataSource: Any
-False_: Any
MachAr: Any
ScalarType: Any
-True_: Any
-UFUNC_PYVALS_NAME: Any
angle: Any
append: Any
apply_along_axis: Any
@@ -1972,9 +1969,14 @@ SHIFT_OVERFLOW: Final[int]
SHIFT_UNDERFLOW: Final[int]
UFUNC_BUFSIZE_DEFAULT: Final[int]
WRAP: Final[int]
-little_endian: Final[int]
tracemalloc_domain: Final[int]
+little_endian: Final[bool]
+True_: Final[bool_]
+False_: Final[bool_]
+
+UFUNC_PYVALS_NAME: Final[str]
+
class ufunc:
@property
def __name__(self) -> str: ...
diff --git a/numpy/typing/tests/data/fail/constants.py b/numpy/typing/tests/data/fail/constants.py
new file mode 100644
index 000000000..67ee0e0bc
--- /dev/null
+++ b/numpy/typing/tests/data/fail/constants.py
@@ -0,0 +1,6 @@
+import numpy as np
+
+np.Inf = np.Inf # E: Cannot assign to final
+np.ALLOW_THREADS = np.ALLOW_THREADS # E: Cannot assign to final
+np.little_endian = np.little_endian # E: Cannot assign to final
+np.UFUNC_PYVALS_NAME = np.UFUNC_PYVALS_NAME # E: Cannot assign to final
diff --git a/numpy/typing/tests/data/reveal/constants.py b/numpy/typing/tests/data/reveal/constants.py
index 8e00810bd..dc5803974 100644
--- a/numpy/typing/tests/data/reveal/constants.py
+++ b/numpy/typing/tests/data/reveal/constants.py
@@ -40,5 +40,10 @@ reveal_type(np.SHIFT_OVERFLOW) # E: int
reveal_type(np.SHIFT_UNDERFLOW) # E: int
reveal_type(np.UFUNC_BUFSIZE_DEFAULT) # E: int
reveal_type(np.WRAP) # E: int
-reveal_type(np.little_endian) # E: int
reveal_type(np.tracemalloc_domain) # E: int
+
+reveal_type(np.little_endian) # E: bool
+reveal_type(np.True_) # E: numpy.bool_
+reveal_type(np.False_) # E: numpy.bool_
+
+reveal_type(np.UFUNC_PYVALS_NAME) # E: str