diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2021-07-09 16:09:24 +0200 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2021-07-09 16:46:31 +0200 |
| commit | a8699e28d348dfeea54f319cca29aff6ff26b0ef (patch) | |
| tree | 7b82dbf4106fe9f6bc56ccd6e58d4d7eb335f671 /numpy | |
| parent | 6a8dc45c2a777e8b5ca7def6fbc41e41ee0b67eb (diff) | |
| download | numpy-a8699e28d348dfeea54f319cca29aff6ff26b0ef.tar.gz | |
TST: Add typing tests for `__array_prepare__` and `__array_wrap__`
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/typing/tests/data/reveal/ndarray_misc.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/numpy/typing/tests/data/reveal/ndarray_misc.py b/numpy/typing/tests/data/reveal/ndarray_misc.py index 2e198eb6f..050b82cdc 100644 --- a/numpy/typing/tests/data/reveal/ndarray_misc.py +++ b/numpy/typing/tests/data/reveal/ndarray_misc.py @@ -11,14 +11,15 @@ import ctypes as ct from typing import Any import numpy as np +from numpy.typing import NDArray -class SubClass(np.ndarray): ... +class SubClass(NDArray[np.object_]): ... f8: np.float64 B: SubClass -AR_f8: np.ndarray[Any, np.dtype[np.float64]] -AR_i8: np.ndarray[Any, np.dtype[np.int64]] -AR_U: np.ndarray[Any, np.dtype[np.str_]] +AR_f8: NDArray[np.float64] +AR_i8: NDArray[np.int64] +AR_U: NDArray[np.str_] ctypes_obj = AR_f8.ctypes @@ -126,7 +127,7 @@ reveal_type(AR_f8.round(out=B)) # E: SubClass reveal_type(f8.repeat(1)) # E: numpy.ndarray[Any, numpy.dtype[{float64}]] reveal_type(AR_f8.repeat(1)) # E: numpy.ndarray[Any, numpy.dtype[{float64}]] -reveal_type(B.repeat(1)) # E: numpy.ndarray[Any, Any] +reveal_type(B.repeat(1)) # E: numpy.ndarray[Any, numpy.dtype[numpy.object_]] reveal_type(f8.std()) # E: Any reveal_type(AR_f8.std()) # E: Any @@ -189,3 +190,6 @@ reveal_type(float(AR_U)) # E: float reveal_type(complex(AR_f8)) # E: complex reveal_type(operator.index(AR_i8)) # E: int + +reveal_type(AR_f8.__array_prepare__(B)) # E: numpy.ndarray[Any, numpy.dtype[numpy.object_]] +reveal_type(AR_f8.__array_wrap__(B)) # E: numpy.ndarray[Any, numpy.dtype[numpy.object_]] |