Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- .gitattributes +3 -0
- .venv/lib/python3.11/site-packages/numpy/linalg/__pycache__/linalg.cpython-311.pyc +3 -0
- .venv/lib/python3.11/site-packages/numpy/linalg/_umath_linalg.cpython-311-x86_64-linux-gnu.so +3 -0
- .venv/lib/python3.11/site-packages/numpy/linalg/tests/__pycache__/test_linalg.cpython-311.pyc +3 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arithmetic.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_constructors.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_like.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayprint.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayterator.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/bitwise_ops.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/comparisons.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/dtype.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/einsumfunc.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/flatiter.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/fromnumeric.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/index_tricks.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_utils.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_version.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/literal.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/mod.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/modules.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/multiarray.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_conversion.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_misc.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_shape_manipulation.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numeric.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numerictypes.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/scalars.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple_py3.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunc_config.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunclike.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufuncs.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/warnings_and_errors.cpython-311.pyc +0 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arithmetic.py +594 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/array_constructors.py +137 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arrayterator.py +27 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/comparisons.py +301 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/dtype.py +57 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/flatiter.py +16 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/index_tricks.py +64 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/lib_version.py +18 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/literal.py +47 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/mod.py +149 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/multiarray.py +76 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py +94 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_misc.py +185 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
- .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/numeric.py +90 -0
.gitattributes
CHANGED
|
@@ -354,3 +354,6 @@ tuning-competition-baseline/.venv/lib/python3.11/site-packages/nvidia/cudnn/lib/
|
|
| 354 |
.venv/lib/python3.11/site-packages/psutil/tests/__pycache__/__init__.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 355 |
.venv/lib/python3.11/site-packages/psutil/__pycache__/_pslinux.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 356 |
.venv/lib/python3.11/site-packages/psutil/tests/__pycache__/test_process.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
| 354 |
.venv/lib/python3.11/site-packages/psutil/tests/__pycache__/__init__.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 355 |
.venv/lib/python3.11/site-packages/psutil/__pycache__/_pslinux.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 356 |
.venv/lib/python3.11/site-packages/psutil/tests/__pycache__/test_process.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 357 |
+
.venv/lib/python3.11/site-packages/numpy/linalg/_umath_linalg.cpython-311-x86_64-linux-gnu.so filter=lfs diff=lfs merge=lfs -text
|
| 358 |
+
.venv/lib/python3.11/site-packages/numpy/linalg/__pycache__/linalg.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
| 359 |
+
.venv/lib/python3.11/site-packages/numpy/linalg/tests/__pycache__/test_linalg.cpython-311.pyc filter=lfs diff=lfs merge=lfs -text
|
.venv/lib/python3.11/site-packages/numpy/linalg/__pycache__/linalg.cpython-311.pyc
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:244c9ef38c6e481f6880650f2192ae6e704af10a334517af030bba8bd461c8e0
|
| 3 |
+
size 107372
|
.venv/lib/python3.11/site-packages/numpy/linalg/_umath_linalg.cpython-311-x86_64-linux-gnu.so
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0587df84ed9751090867a4a1441d2c394c39735420aa388b8bf698b7255f24bd
|
| 3 |
+
size 216793
|
.venv/lib/python3.11/site-packages/numpy/linalg/tests/__pycache__/test_linalg.cpython-311.pyc
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7c507761b4c9f5dc6fd28ae2e10f612c9fc4e33633e5bd89b8cef66635443a5a
|
| 3 |
+
size 148841
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arithmetic.cpython-311.pyc
ADDED
|
Binary file (13.3 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_constructors.cpython-311.pyc
ADDED
|
Binary file (7.15 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_like.cpython-311.pyc
ADDED
|
Binary file (2.63 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayprint.cpython-311.pyc
ADDED
|
Binary file (1.53 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayterator.cpython-311.pyc
ADDED
|
Binary file (1.07 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/bitwise_ops.cpython-311.pyc
ADDED
|
Binary file (2.6 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/comparisons.cpython-311.pyc
ADDED
|
Binary file (7.16 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/dtype.cpython-311.pyc
ADDED
|
Binary file (2.32 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/einsumfunc.cpython-311.pyc
ADDED
|
Binary file (2.33 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/flatiter.cpython-311.pyc
ADDED
|
Binary file (871 Bytes). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/fromnumeric.cpython-311.pyc
ADDED
|
Binary file (12.2 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/index_tricks.cpython-311.pyc
ADDED
|
Binary file (3.61 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_utils.cpython-311.pyc
ADDED
|
Binary file (1.29 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_version.cpython-311.pyc
ADDED
|
Binary file (665 Bytes). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/literal.cpython-311.pyc
ADDED
|
Binary file (2.66 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/mod.cpython-311.pyc
ADDED
|
Binary file (4.24 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/modules.cpython-311.pyc
ADDED
|
Binary file (1.51 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/multiarray.cpython-311.pyc
ADDED
|
Binary file (3.36 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_conversion.cpython-311.pyc
ADDED
|
Binary file (4.21 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_misc.cpython-311.pyc
ADDED
|
Binary file (9.84 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_shape_manipulation.cpython-311.pyc
ADDED
|
Binary file (2.03 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numeric.cpython-311.pyc
ADDED
|
Binary file (4.57 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numerictypes.cpython-311.pyc
ADDED
|
Binary file (2.07 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/scalars.cpython-311.pyc
ADDED
|
Binary file (12.3 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple.cpython-311.pyc
ADDED
|
Binary file (4.95 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple_py3.cpython-311.pyc
ADDED
|
Binary file (325 Bytes). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunc_config.cpython-311.pyc
ADDED
|
Binary file (4.06 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunclike.cpython-311.pyc
ADDED
|
Binary file (2.8 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufuncs.cpython-311.pyc
ADDED
|
Binary file (1.19 kB). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/warnings_and_errors.cpython-311.pyc
ADDED
|
Binary file (537 Bytes). View file
|
|
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arithmetic.py
ADDED
|
@@ -0,0 +1,594 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from typing import Any
|
| 4 |
+
import numpy as np
|
| 5 |
+
import pytest
|
| 6 |
+
|
| 7 |
+
c16 = np.complex128(1)
|
| 8 |
+
f8 = np.float64(1)
|
| 9 |
+
i8 = np.int64(1)
|
| 10 |
+
u8 = np.uint64(1)
|
| 11 |
+
|
| 12 |
+
c8 = np.complex64(1)
|
| 13 |
+
f4 = np.float32(1)
|
| 14 |
+
i4 = np.int32(1)
|
| 15 |
+
u4 = np.uint32(1)
|
| 16 |
+
|
| 17 |
+
dt = np.datetime64(1, "D")
|
| 18 |
+
td = np.timedelta64(1, "D")
|
| 19 |
+
|
| 20 |
+
b_ = np.bool_(1)
|
| 21 |
+
|
| 22 |
+
b = bool(1)
|
| 23 |
+
c = complex(1)
|
| 24 |
+
f = float(1)
|
| 25 |
+
i = int(1)
|
| 26 |
+
|
| 27 |
+
|
| 28 |
+
class Object:
|
| 29 |
+
def __array__(self) -> np.ndarray[Any, np.dtype[np.object_]]:
|
| 30 |
+
ret = np.empty((), dtype=object)
|
| 31 |
+
ret[()] = self
|
| 32 |
+
return ret
|
| 33 |
+
|
| 34 |
+
def __sub__(self, value: Any) -> Object:
|
| 35 |
+
return self
|
| 36 |
+
|
| 37 |
+
def __rsub__(self, value: Any) -> Object:
|
| 38 |
+
return self
|
| 39 |
+
|
| 40 |
+
def __floordiv__(self, value: Any) -> Object:
|
| 41 |
+
return self
|
| 42 |
+
|
| 43 |
+
def __rfloordiv__(self, value: Any) -> Object:
|
| 44 |
+
return self
|
| 45 |
+
|
| 46 |
+
def __mul__(self, value: Any) -> Object:
|
| 47 |
+
return self
|
| 48 |
+
|
| 49 |
+
def __rmul__(self, value: Any) -> Object:
|
| 50 |
+
return self
|
| 51 |
+
|
| 52 |
+
def __pow__(self, value: Any) -> Object:
|
| 53 |
+
return self
|
| 54 |
+
|
| 55 |
+
def __rpow__(self, value: Any) -> Object:
|
| 56 |
+
return self
|
| 57 |
+
|
| 58 |
+
|
| 59 |
+
AR_b: np.ndarray[Any, np.dtype[np.bool_]] = np.array([True])
|
| 60 |
+
AR_u: np.ndarray[Any, np.dtype[np.uint32]] = np.array([1], dtype=np.uint32)
|
| 61 |
+
AR_i: np.ndarray[Any, np.dtype[np.int64]] = np.array([1])
|
| 62 |
+
AR_f: np.ndarray[Any, np.dtype[np.float64]] = np.array([1.0])
|
| 63 |
+
AR_c: np.ndarray[Any, np.dtype[np.complex128]] = np.array([1j])
|
| 64 |
+
AR_m: np.ndarray[Any, np.dtype[np.timedelta64]] = np.array([np.timedelta64(1, "D")])
|
| 65 |
+
AR_M: np.ndarray[Any, np.dtype[np.datetime64]] = np.array([np.datetime64(1, "D")])
|
| 66 |
+
AR_O: np.ndarray[Any, np.dtype[np.object_]] = np.array([Object()])
|
| 67 |
+
|
| 68 |
+
AR_LIKE_b = [True]
|
| 69 |
+
AR_LIKE_u = [np.uint32(1)]
|
| 70 |
+
AR_LIKE_i = [1]
|
| 71 |
+
AR_LIKE_f = [1.0]
|
| 72 |
+
AR_LIKE_c = [1j]
|
| 73 |
+
AR_LIKE_m = [np.timedelta64(1, "D")]
|
| 74 |
+
AR_LIKE_M = [np.datetime64(1, "D")]
|
| 75 |
+
AR_LIKE_O = [Object()]
|
| 76 |
+
|
| 77 |
+
# Array subtractions
|
| 78 |
+
|
| 79 |
+
AR_b - AR_LIKE_u
|
| 80 |
+
AR_b - AR_LIKE_i
|
| 81 |
+
AR_b - AR_LIKE_f
|
| 82 |
+
AR_b - AR_LIKE_c
|
| 83 |
+
AR_b - AR_LIKE_m
|
| 84 |
+
AR_b - AR_LIKE_O
|
| 85 |
+
|
| 86 |
+
AR_LIKE_u - AR_b
|
| 87 |
+
AR_LIKE_i - AR_b
|
| 88 |
+
AR_LIKE_f - AR_b
|
| 89 |
+
AR_LIKE_c - AR_b
|
| 90 |
+
AR_LIKE_m - AR_b
|
| 91 |
+
AR_LIKE_M - AR_b
|
| 92 |
+
AR_LIKE_O - AR_b
|
| 93 |
+
|
| 94 |
+
AR_u - AR_LIKE_b
|
| 95 |
+
AR_u - AR_LIKE_u
|
| 96 |
+
AR_u - AR_LIKE_i
|
| 97 |
+
AR_u - AR_LIKE_f
|
| 98 |
+
AR_u - AR_LIKE_c
|
| 99 |
+
AR_u - AR_LIKE_m
|
| 100 |
+
AR_u - AR_LIKE_O
|
| 101 |
+
|
| 102 |
+
AR_LIKE_b - AR_u
|
| 103 |
+
AR_LIKE_u - AR_u
|
| 104 |
+
AR_LIKE_i - AR_u
|
| 105 |
+
AR_LIKE_f - AR_u
|
| 106 |
+
AR_LIKE_c - AR_u
|
| 107 |
+
AR_LIKE_m - AR_u
|
| 108 |
+
AR_LIKE_M - AR_u
|
| 109 |
+
AR_LIKE_O - AR_u
|
| 110 |
+
|
| 111 |
+
AR_i - AR_LIKE_b
|
| 112 |
+
AR_i - AR_LIKE_u
|
| 113 |
+
AR_i - AR_LIKE_i
|
| 114 |
+
AR_i - AR_LIKE_f
|
| 115 |
+
AR_i - AR_LIKE_c
|
| 116 |
+
AR_i - AR_LIKE_m
|
| 117 |
+
AR_i - AR_LIKE_O
|
| 118 |
+
|
| 119 |
+
AR_LIKE_b - AR_i
|
| 120 |
+
AR_LIKE_u - AR_i
|
| 121 |
+
AR_LIKE_i - AR_i
|
| 122 |
+
AR_LIKE_f - AR_i
|
| 123 |
+
AR_LIKE_c - AR_i
|
| 124 |
+
AR_LIKE_m - AR_i
|
| 125 |
+
AR_LIKE_M - AR_i
|
| 126 |
+
AR_LIKE_O - AR_i
|
| 127 |
+
|
| 128 |
+
AR_f - AR_LIKE_b
|
| 129 |
+
AR_f - AR_LIKE_u
|
| 130 |
+
AR_f - AR_LIKE_i
|
| 131 |
+
AR_f - AR_LIKE_f
|
| 132 |
+
AR_f - AR_LIKE_c
|
| 133 |
+
AR_f - AR_LIKE_O
|
| 134 |
+
|
| 135 |
+
AR_LIKE_b - AR_f
|
| 136 |
+
AR_LIKE_u - AR_f
|
| 137 |
+
AR_LIKE_i - AR_f
|
| 138 |
+
AR_LIKE_f - AR_f
|
| 139 |
+
AR_LIKE_c - AR_f
|
| 140 |
+
AR_LIKE_O - AR_f
|
| 141 |
+
|
| 142 |
+
AR_c - AR_LIKE_b
|
| 143 |
+
AR_c - AR_LIKE_u
|
| 144 |
+
AR_c - AR_LIKE_i
|
| 145 |
+
AR_c - AR_LIKE_f
|
| 146 |
+
AR_c - AR_LIKE_c
|
| 147 |
+
AR_c - AR_LIKE_O
|
| 148 |
+
|
| 149 |
+
AR_LIKE_b - AR_c
|
| 150 |
+
AR_LIKE_u - AR_c
|
| 151 |
+
AR_LIKE_i - AR_c
|
| 152 |
+
AR_LIKE_f - AR_c
|
| 153 |
+
AR_LIKE_c - AR_c
|
| 154 |
+
AR_LIKE_O - AR_c
|
| 155 |
+
|
| 156 |
+
AR_m - AR_LIKE_b
|
| 157 |
+
AR_m - AR_LIKE_u
|
| 158 |
+
AR_m - AR_LIKE_i
|
| 159 |
+
AR_m - AR_LIKE_m
|
| 160 |
+
|
| 161 |
+
AR_LIKE_b - AR_m
|
| 162 |
+
AR_LIKE_u - AR_m
|
| 163 |
+
AR_LIKE_i - AR_m
|
| 164 |
+
AR_LIKE_m - AR_m
|
| 165 |
+
AR_LIKE_M - AR_m
|
| 166 |
+
|
| 167 |
+
AR_M - AR_LIKE_b
|
| 168 |
+
AR_M - AR_LIKE_u
|
| 169 |
+
AR_M - AR_LIKE_i
|
| 170 |
+
AR_M - AR_LIKE_m
|
| 171 |
+
AR_M - AR_LIKE_M
|
| 172 |
+
|
| 173 |
+
AR_LIKE_M - AR_M
|
| 174 |
+
|
| 175 |
+
AR_O - AR_LIKE_b
|
| 176 |
+
AR_O - AR_LIKE_u
|
| 177 |
+
AR_O - AR_LIKE_i
|
| 178 |
+
AR_O - AR_LIKE_f
|
| 179 |
+
AR_O - AR_LIKE_c
|
| 180 |
+
AR_O - AR_LIKE_O
|
| 181 |
+
|
| 182 |
+
AR_LIKE_b - AR_O
|
| 183 |
+
AR_LIKE_u - AR_O
|
| 184 |
+
AR_LIKE_i - AR_O
|
| 185 |
+
AR_LIKE_f - AR_O
|
| 186 |
+
AR_LIKE_c - AR_O
|
| 187 |
+
AR_LIKE_O - AR_O
|
| 188 |
+
|
| 189 |
+
AR_u += AR_b
|
| 190 |
+
AR_u += AR_u
|
| 191 |
+
AR_u += 1 # Allowed during runtime as long as the object is 0D and >=0
|
| 192 |
+
|
| 193 |
+
# Array floor division
|
| 194 |
+
|
| 195 |
+
AR_b // AR_LIKE_b
|
| 196 |
+
AR_b // AR_LIKE_u
|
| 197 |
+
AR_b // AR_LIKE_i
|
| 198 |
+
AR_b // AR_LIKE_f
|
| 199 |
+
AR_b // AR_LIKE_O
|
| 200 |
+
|
| 201 |
+
AR_LIKE_b // AR_b
|
| 202 |
+
AR_LIKE_u // AR_b
|
| 203 |
+
AR_LIKE_i // AR_b
|
| 204 |
+
AR_LIKE_f // AR_b
|
| 205 |
+
AR_LIKE_O // AR_b
|
| 206 |
+
|
| 207 |
+
AR_u // AR_LIKE_b
|
| 208 |
+
AR_u // AR_LIKE_u
|
| 209 |
+
AR_u // AR_LIKE_i
|
| 210 |
+
AR_u // AR_LIKE_f
|
| 211 |
+
AR_u // AR_LIKE_O
|
| 212 |
+
|
| 213 |
+
AR_LIKE_b // AR_u
|
| 214 |
+
AR_LIKE_u // AR_u
|
| 215 |
+
AR_LIKE_i // AR_u
|
| 216 |
+
AR_LIKE_f // AR_u
|
| 217 |
+
AR_LIKE_m // AR_u
|
| 218 |
+
AR_LIKE_O // AR_u
|
| 219 |
+
|
| 220 |
+
AR_i // AR_LIKE_b
|
| 221 |
+
AR_i // AR_LIKE_u
|
| 222 |
+
AR_i // AR_LIKE_i
|
| 223 |
+
AR_i // AR_LIKE_f
|
| 224 |
+
AR_i // AR_LIKE_O
|
| 225 |
+
|
| 226 |
+
AR_LIKE_b // AR_i
|
| 227 |
+
AR_LIKE_u // AR_i
|
| 228 |
+
AR_LIKE_i // AR_i
|
| 229 |
+
AR_LIKE_f // AR_i
|
| 230 |
+
AR_LIKE_m // AR_i
|
| 231 |
+
AR_LIKE_O // AR_i
|
| 232 |
+
|
| 233 |
+
AR_f // AR_LIKE_b
|
| 234 |
+
AR_f // AR_LIKE_u
|
| 235 |
+
AR_f // AR_LIKE_i
|
| 236 |
+
AR_f // AR_LIKE_f
|
| 237 |
+
AR_f // AR_LIKE_O
|
| 238 |
+
|
| 239 |
+
AR_LIKE_b // AR_f
|
| 240 |
+
AR_LIKE_u // AR_f
|
| 241 |
+
AR_LIKE_i // AR_f
|
| 242 |
+
AR_LIKE_f // AR_f
|
| 243 |
+
AR_LIKE_m // AR_f
|
| 244 |
+
AR_LIKE_O // AR_f
|
| 245 |
+
|
| 246 |
+
AR_m // AR_LIKE_u
|
| 247 |
+
AR_m // AR_LIKE_i
|
| 248 |
+
AR_m // AR_LIKE_f
|
| 249 |
+
AR_m // AR_LIKE_m
|
| 250 |
+
|
| 251 |
+
AR_LIKE_m // AR_m
|
| 252 |
+
|
| 253 |
+
AR_O // AR_LIKE_b
|
| 254 |
+
AR_O // AR_LIKE_u
|
| 255 |
+
AR_O // AR_LIKE_i
|
| 256 |
+
AR_O // AR_LIKE_f
|
| 257 |
+
AR_O // AR_LIKE_O
|
| 258 |
+
|
| 259 |
+
AR_LIKE_b // AR_O
|
| 260 |
+
AR_LIKE_u // AR_O
|
| 261 |
+
AR_LIKE_i // AR_O
|
| 262 |
+
AR_LIKE_f // AR_O
|
| 263 |
+
AR_LIKE_O // AR_O
|
| 264 |
+
|
| 265 |
+
# Inplace multiplication
|
| 266 |
+
|
| 267 |
+
AR_b *= AR_LIKE_b
|
| 268 |
+
|
| 269 |
+
AR_u *= AR_LIKE_b
|
| 270 |
+
AR_u *= AR_LIKE_u
|
| 271 |
+
|
| 272 |
+
AR_i *= AR_LIKE_b
|
| 273 |
+
AR_i *= AR_LIKE_u
|
| 274 |
+
AR_i *= AR_LIKE_i
|
| 275 |
+
|
| 276 |
+
AR_f *= AR_LIKE_b
|
| 277 |
+
AR_f *= AR_LIKE_u
|
| 278 |
+
AR_f *= AR_LIKE_i
|
| 279 |
+
AR_f *= AR_LIKE_f
|
| 280 |
+
|
| 281 |
+
AR_c *= AR_LIKE_b
|
| 282 |
+
AR_c *= AR_LIKE_u
|
| 283 |
+
AR_c *= AR_LIKE_i
|
| 284 |
+
AR_c *= AR_LIKE_f
|
| 285 |
+
AR_c *= AR_LIKE_c
|
| 286 |
+
|
| 287 |
+
AR_m *= AR_LIKE_b
|
| 288 |
+
AR_m *= AR_LIKE_u
|
| 289 |
+
AR_m *= AR_LIKE_i
|
| 290 |
+
AR_m *= AR_LIKE_f
|
| 291 |
+
|
| 292 |
+
AR_O *= AR_LIKE_b
|
| 293 |
+
AR_O *= AR_LIKE_u
|
| 294 |
+
AR_O *= AR_LIKE_i
|
| 295 |
+
AR_O *= AR_LIKE_f
|
| 296 |
+
AR_O *= AR_LIKE_c
|
| 297 |
+
AR_O *= AR_LIKE_O
|
| 298 |
+
|
| 299 |
+
# Inplace power
|
| 300 |
+
|
| 301 |
+
AR_u **= AR_LIKE_b
|
| 302 |
+
AR_u **= AR_LIKE_u
|
| 303 |
+
|
| 304 |
+
AR_i **= AR_LIKE_b
|
| 305 |
+
AR_i **= AR_LIKE_u
|
| 306 |
+
AR_i **= AR_LIKE_i
|
| 307 |
+
|
| 308 |
+
AR_f **= AR_LIKE_b
|
| 309 |
+
AR_f **= AR_LIKE_u
|
| 310 |
+
AR_f **= AR_LIKE_i
|
| 311 |
+
AR_f **= AR_LIKE_f
|
| 312 |
+
|
| 313 |
+
AR_c **= AR_LIKE_b
|
| 314 |
+
AR_c **= AR_LIKE_u
|
| 315 |
+
AR_c **= AR_LIKE_i
|
| 316 |
+
AR_c **= AR_LIKE_f
|
| 317 |
+
AR_c **= AR_LIKE_c
|
| 318 |
+
|
| 319 |
+
AR_O **= AR_LIKE_b
|
| 320 |
+
AR_O **= AR_LIKE_u
|
| 321 |
+
AR_O **= AR_LIKE_i
|
| 322 |
+
AR_O **= AR_LIKE_f
|
| 323 |
+
AR_O **= AR_LIKE_c
|
| 324 |
+
AR_O **= AR_LIKE_O
|
| 325 |
+
|
| 326 |
+
# unary ops
|
| 327 |
+
|
| 328 |
+
-c16
|
| 329 |
+
-c8
|
| 330 |
+
-f8
|
| 331 |
+
-f4
|
| 332 |
+
-i8
|
| 333 |
+
-i4
|
| 334 |
+
with pytest.warns(RuntimeWarning):
|
| 335 |
+
-u8
|
| 336 |
+
-u4
|
| 337 |
+
-td
|
| 338 |
+
-AR_f
|
| 339 |
+
|
| 340 |
+
+c16
|
| 341 |
+
+c8
|
| 342 |
+
+f8
|
| 343 |
+
+f4
|
| 344 |
+
+i8
|
| 345 |
+
+i4
|
| 346 |
+
+u8
|
| 347 |
+
+u4
|
| 348 |
+
+td
|
| 349 |
+
+AR_f
|
| 350 |
+
|
| 351 |
+
abs(c16)
|
| 352 |
+
abs(c8)
|
| 353 |
+
abs(f8)
|
| 354 |
+
abs(f4)
|
| 355 |
+
abs(i8)
|
| 356 |
+
abs(i4)
|
| 357 |
+
abs(u8)
|
| 358 |
+
abs(u4)
|
| 359 |
+
abs(td)
|
| 360 |
+
abs(b_)
|
| 361 |
+
abs(AR_f)
|
| 362 |
+
|
| 363 |
+
# Time structures
|
| 364 |
+
|
| 365 |
+
dt + td
|
| 366 |
+
dt + i
|
| 367 |
+
dt + i4
|
| 368 |
+
dt + i8
|
| 369 |
+
dt - dt
|
| 370 |
+
dt - i
|
| 371 |
+
dt - i4
|
| 372 |
+
dt - i8
|
| 373 |
+
|
| 374 |
+
td + td
|
| 375 |
+
td + i
|
| 376 |
+
td + i4
|
| 377 |
+
td + i8
|
| 378 |
+
td - td
|
| 379 |
+
td - i
|
| 380 |
+
td - i4
|
| 381 |
+
td - i8
|
| 382 |
+
td / f
|
| 383 |
+
td / f4
|
| 384 |
+
td / f8
|
| 385 |
+
td / td
|
| 386 |
+
td // td
|
| 387 |
+
td % td
|
| 388 |
+
|
| 389 |
+
|
| 390 |
+
# boolean
|
| 391 |
+
|
| 392 |
+
b_ / b
|
| 393 |
+
b_ / b_
|
| 394 |
+
b_ / i
|
| 395 |
+
b_ / i8
|
| 396 |
+
b_ / i4
|
| 397 |
+
b_ / u8
|
| 398 |
+
b_ / u4
|
| 399 |
+
b_ / f
|
| 400 |
+
b_ / f8
|
| 401 |
+
b_ / f4
|
| 402 |
+
b_ / c
|
| 403 |
+
b_ / c16
|
| 404 |
+
b_ / c8
|
| 405 |
+
|
| 406 |
+
b / b_
|
| 407 |
+
b_ / b_
|
| 408 |
+
i / b_
|
| 409 |
+
i8 / b_
|
| 410 |
+
i4 / b_
|
| 411 |
+
u8 / b_
|
| 412 |
+
u4 / b_
|
| 413 |
+
f / b_
|
| 414 |
+
f8 / b_
|
| 415 |
+
f4 / b_
|
| 416 |
+
c / b_
|
| 417 |
+
c16 / b_
|
| 418 |
+
c8 / b_
|
| 419 |
+
|
| 420 |
+
# Complex
|
| 421 |
+
|
| 422 |
+
c16 + c16
|
| 423 |
+
c16 + f8
|
| 424 |
+
c16 + i8
|
| 425 |
+
c16 + c8
|
| 426 |
+
c16 + f4
|
| 427 |
+
c16 + i4
|
| 428 |
+
c16 + b_
|
| 429 |
+
c16 + b
|
| 430 |
+
c16 + c
|
| 431 |
+
c16 + f
|
| 432 |
+
c16 + i
|
| 433 |
+
c16 + AR_f
|
| 434 |
+
|
| 435 |
+
c16 + c16
|
| 436 |
+
f8 + c16
|
| 437 |
+
i8 + c16
|
| 438 |
+
c8 + c16
|
| 439 |
+
f4 + c16
|
| 440 |
+
i4 + c16
|
| 441 |
+
b_ + c16
|
| 442 |
+
b + c16
|
| 443 |
+
c + c16
|
| 444 |
+
f + c16
|
| 445 |
+
i + c16
|
| 446 |
+
AR_f + c16
|
| 447 |
+
|
| 448 |
+
c8 + c16
|
| 449 |
+
c8 + f8
|
| 450 |
+
c8 + i8
|
| 451 |
+
c8 + c8
|
| 452 |
+
c8 + f4
|
| 453 |
+
c8 + i4
|
| 454 |
+
c8 + b_
|
| 455 |
+
c8 + b
|
| 456 |
+
c8 + c
|
| 457 |
+
c8 + f
|
| 458 |
+
c8 + i
|
| 459 |
+
c8 + AR_f
|
| 460 |
+
|
| 461 |
+
c16 + c8
|
| 462 |
+
f8 + c8
|
| 463 |
+
i8 + c8
|
| 464 |
+
c8 + c8
|
| 465 |
+
f4 + c8
|
| 466 |
+
i4 + c8
|
| 467 |
+
b_ + c8
|
| 468 |
+
b + c8
|
| 469 |
+
c + c8
|
| 470 |
+
f + c8
|
| 471 |
+
i + c8
|
| 472 |
+
AR_f + c8
|
| 473 |
+
|
| 474 |
+
# Float
|
| 475 |
+
|
| 476 |
+
f8 + f8
|
| 477 |
+
f8 + i8
|
| 478 |
+
f8 + f4
|
| 479 |
+
f8 + i4
|
| 480 |
+
f8 + b_
|
| 481 |
+
f8 + b
|
| 482 |
+
f8 + c
|
| 483 |
+
f8 + f
|
| 484 |
+
f8 + i
|
| 485 |
+
f8 + AR_f
|
| 486 |
+
|
| 487 |
+
f8 + f8
|
| 488 |
+
i8 + f8
|
| 489 |
+
f4 + f8
|
| 490 |
+
i4 + f8
|
| 491 |
+
b_ + f8
|
| 492 |
+
b + f8
|
| 493 |
+
c + f8
|
| 494 |
+
f + f8
|
| 495 |
+
i + f8
|
| 496 |
+
AR_f + f8
|
| 497 |
+
|
| 498 |
+
f4 + f8
|
| 499 |
+
f4 + i8
|
| 500 |
+
f4 + f4
|
| 501 |
+
f4 + i4
|
| 502 |
+
f4 + b_
|
| 503 |
+
f4 + b
|
| 504 |
+
f4 + c
|
| 505 |
+
f4 + f
|
| 506 |
+
f4 + i
|
| 507 |
+
f4 + AR_f
|
| 508 |
+
|
| 509 |
+
f8 + f4
|
| 510 |
+
i8 + f4
|
| 511 |
+
f4 + f4
|
| 512 |
+
i4 + f4
|
| 513 |
+
b_ + f4
|
| 514 |
+
b + f4
|
| 515 |
+
c + f4
|
| 516 |
+
f + f4
|
| 517 |
+
i + f4
|
| 518 |
+
AR_f + f4
|
| 519 |
+
|
| 520 |
+
# Int
|
| 521 |
+
|
| 522 |
+
i8 + i8
|
| 523 |
+
i8 + u8
|
| 524 |
+
i8 + i4
|
| 525 |
+
i8 + u4
|
| 526 |
+
i8 + b_
|
| 527 |
+
i8 + b
|
| 528 |
+
i8 + c
|
| 529 |
+
i8 + f
|
| 530 |
+
i8 + i
|
| 531 |
+
i8 + AR_f
|
| 532 |
+
|
| 533 |
+
u8 + u8
|
| 534 |
+
u8 + i4
|
| 535 |
+
u8 + u4
|
| 536 |
+
u8 + b_
|
| 537 |
+
u8 + b
|
| 538 |
+
u8 + c
|
| 539 |
+
u8 + f
|
| 540 |
+
u8 + i
|
| 541 |
+
u8 + AR_f
|
| 542 |
+
|
| 543 |
+
i8 + i8
|
| 544 |
+
u8 + i8
|
| 545 |
+
i4 + i8
|
| 546 |
+
u4 + i8
|
| 547 |
+
b_ + i8
|
| 548 |
+
b + i8
|
| 549 |
+
c + i8
|
| 550 |
+
f + i8
|
| 551 |
+
i + i8
|
| 552 |
+
AR_f + i8
|
| 553 |
+
|
| 554 |
+
u8 + u8
|
| 555 |
+
i4 + u8
|
| 556 |
+
u4 + u8
|
| 557 |
+
b_ + u8
|
| 558 |
+
b + u8
|
| 559 |
+
c + u8
|
| 560 |
+
f + u8
|
| 561 |
+
i + u8
|
| 562 |
+
AR_f + u8
|
| 563 |
+
|
| 564 |
+
i4 + i8
|
| 565 |
+
i4 + i4
|
| 566 |
+
i4 + i
|
| 567 |
+
i4 + b_
|
| 568 |
+
i4 + b
|
| 569 |
+
i4 + AR_f
|
| 570 |
+
|
| 571 |
+
u4 + i8
|
| 572 |
+
u4 + i4
|
| 573 |
+
u4 + u8
|
| 574 |
+
u4 + u4
|
| 575 |
+
u4 + i
|
| 576 |
+
u4 + b_
|
| 577 |
+
u4 + b
|
| 578 |
+
u4 + AR_f
|
| 579 |
+
|
| 580 |
+
i8 + i4
|
| 581 |
+
i4 + i4
|
| 582 |
+
i + i4
|
| 583 |
+
b_ + i4
|
| 584 |
+
b + i4
|
| 585 |
+
AR_f + i4
|
| 586 |
+
|
| 587 |
+
i8 + u4
|
| 588 |
+
i4 + u4
|
| 589 |
+
u8 + u4
|
| 590 |
+
u4 + u4
|
| 591 |
+
b_ + u4
|
| 592 |
+
b + u4
|
| 593 |
+
i + u4
|
| 594 |
+
AR_f + u4
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/array_constructors.py
ADDED
|
@@ -0,0 +1,137 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import sys
|
| 2 |
+
from typing import Any
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
class Index:
|
| 7 |
+
def __index__(self) -> int:
|
| 8 |
+
return 0
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
class SubClass(np.ndarray):
|
| 12 |
+
pass
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
def func(i: int, j: int, **kwargs: Any) -> SubClass:
|
| 16 |
+
return B
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
i8 = np.int64(1)
|
| 20 |
+
|
| 21 |
+
A = np.array([1])
|
| 22 |
+
B = A.view(SubClass).copy()
|
| 23 |
+
B_stack = np.array([[1], [1]]).view(SubClass)
|
| 24 |
+
C = [1]
|
| 25 |
+
|
| 26 |
+
np.ndarray(Index())
|
| 27 |
+
np.ndarray([Index()])
|
| 28 |
+
|
| 29 |
+
np.array(1, dtype=float)
|
| 30 |
+
np.array(1, copy=False)
|
| 31 |
+
np.array(1, order='F')
|
| 32 |
+
np.array(1, order=None)
|
| 33 |
+
np.array(1, subok=True)
|
| 34 |
+
np.array(1, ndmin=3)
|
| 35 |
+
np.array(1, str, copy=True, order='C', subok=False, ndmin=2)
|
| 36 |
+
|
| 37 |
+
np.asarray(A)
|
| 38 |
+
np.asarray(B)
|
| 39 |
+
np.asarray(C)
|
| 40 |
+
|
| 41 |
+
np.asanyarray(A)
|
| 42 |
+
np.asanyarray(B)
|
| 43 |
+
np.asanyarray(B, dtype=int)
|
| 44 |
+
np.asanyarray(C)
|
| 45 |
+
|
| 46 |
+
np.ascontiguousarray(A)
|
| 47 |
+
np.ascontiguousarray(B)
|
| 48 |
+
np.ascontiguousarray(C)
|
| 49 |
+
|
| 50 |
+
np.asfortranarray(A)
|
| 51 |
+
np.asfortranarray(B)
|
| 52 |
+
np.asfortranarray(C)
|
| 53 |
+
|
| 54 |
+
np.require(A)
|
| 55 |
+
np.require(B)
|
| 56 |
+
np.require(B, dtype=int)
|
| 57 |
+
np.require(B, requirements=None)
|
| 58 |
+
np.require(B, requirements="E")
|
| 59 |
+
np.require(B, requirements=["ENSUREARRAY"])
|
| 60 |
+
np.require(B, requirements={"F", "E"})
|
| 61 |
+
np.require(B, requirements=["C", "OWNDATA"])
|
| 62 |
+
np.require(B, requirements="W")
|
| 63 |
+
np.require(B, requirements="A")
|
| 64 |
+
np.require(C)
|
| 65 |
+
|
| 66 |
+
np.linspace(0, 2)
|
| 67 |
+
np.linspace(0.5, [0, 1, 2])
|
| 68 |
+
np.linspace([0, 1, 2], 3)
|
| 69 |
+
np.linspace(0j, 2)
|
| 70 |
+
np.linspace(0, 2, num=10)
|
| 71 |
+
np.linspace(0, 2, endpoint=True)
|
| 72 |
+
np.linspace(0, 2, retstep=True)
|
| 73 |
+
np.linspace(0j, 2j, retstep=True)
|
| 74 |
+
np.linspace(0, 2, dtype=bool)
|
| 75 |
+
np.linspace([0, 1], [2, 3], axis=Index())
|
| 76 |
+
|
| 77 |
+
np.logspace(0, 2, base=2)
|
| 78 |
+
np.logspace(0, 2, base=2)
|
| 79 |
+
np.logspace(0, 2, base=[1j, 2j], num=2)
|
| 80 |
+
|
| 81 |
+
np.geomspace(1, 2)
|
| 82 |
+
|
| 83 |
+
np.zeros_like(A)
|
| 84 |
+
np.zeros_like(C)
|
| 85 |
+
np.zeros_like(B)
|
| 86 |
+
np.zeros_like(B, dtype=np.int64)
|
| 87 |
+
|
| 88 |
+
np.ones_like(A)
|
| 89 |
+
np.ones_like(C)
|
| 90 |
+
np.ones_like(B)
|
| 91 |
+
np.ones_like(B, dtype=np.int64)
|
| 92 |
+
|
| 93 |
+
np.empty_like(A)
|
| 94 |
+
np.empty_like(C)
|
| 95 |
+
np.empty_like(B)
|
| 96 |
+
np.empty_like(B, dtype=np.int64)
|
| 97 |
+
|
| 98 |
+
np.full_like(A, i8)
|
| 99 |
+
np.full_like(C, i8)
|
| 100 |
+
np.full_like(B, i8)
|
| 101 |
+
np.full_like(B, i8, dtype=np.int64)
|
| 102 |
+
|
| 103 |
+
np.ones(1)
|
| 104 |
+
np.ones([1, 1, 1])
|
| 105 |
+
|
| 106 |
+
np.full(1, i8)
|
| 107 |
+
np.full([1, 1, 1], i8)
|
| 108 |
+
|
| 109 |
+
np.indices([1, 2, 3])
|
| 110 |
+
np.indices([1, 2, 3], sparse=True)
|
| 111 |
+
|
| 112 |
+
np.fromfunction(func, (3, 5))
|
| 113 |
+
|
| 114 |
+
np.identity(10)
|
| 115 |
+
|
| 116 |
+
np.atleast_1d(C)
|
| 117 |
+
np.atleast_1d(A)
|
| 118 |
+
np.atleast_1d(C, C)
|
| 119 |
+
np.atleast_1d(C, A)
|
| 120 |
+
np.atleast_1d(A, A)
|
| 121 |
+
|
| 122 |
+
np.atleast_2d(C)
|
| 123 |
+
|
| 124 |
+
np.atleast_3d(C)
|
| 125 |
+
|
| 126 |
+
np.vstack([C, C])
|
| 127 |
+
np.vstack([C, A])
|
| 128 |
+
np.vstack([A, A])
|
| 129 |
+
|
| 130 |
+
np.hstack([C, C])
|
| 131 |
+
|
| 132 |
+
np.stack([C, C])
|
| 133 |
+
np.stack([C, C], axis=0)
|
| 134 |
+
np.stack([C, C], out=B_stack)
|
| 135 |
+
|
| 136 |
+
np.block([[C, C], [C, C]])
|
| 137 |
+
np.block(A)
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arrayterator.py
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
from __future__ import annotations
|
| 3 |
+
|
| 4 |
+
from typing import Any
|
| 5 |
+
import numpy as np
|
| 6 |
+
|
| 7 |
+
AR_i8: np.ndarray[Any, np.dtype[np.int_]] = np.arange(10)
|
| 8 |
+
ar_iter = np.lib.Arrayterator(AR_i8)
|
| 9 |
+
|
| 10 |
+
ar_iter.var
|
| 11 |
+
ar_iter.buf_size
|
| 12 |
+
ar_iter.start
|
| 13 |
+
ar_iter.stop
|
| 14 |
+
ar_iter.step
|
| 15 |
+
ar_iter.shape
|
| 16 |
+
ar_iter.flat
|
| 17 |
+
|
| 18 |
+
ar_iter.__array__()
|
| 19 |
+
|
| 20 |
+
for i in ar_iter:
|
| 21 |
+
pass
|
| 22 |
+
|
| 23 |
+
ar_iter[0]
|
| 24 |
+
ar_iter[...]
|
| 25 |
+
ar_iter[:]
|
| 26 |
+
ar_iter[0, 0, 0]
|
| 27 |
+
ar_iter[..., 0, :]
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/bitwise_ops.py
ADDED
|
@@ -0,0 +1,131 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
i8 = np.int64(1)
|
| 4 |
+
u8 = np.uint64(1)
|
| 5 |
+
|
| 6 |
+
i4 = np.int32(1)
|
| 7 |
+
u4 = np.uint32(1)
|
| 8 |
+
|
| 9 |
+
b_ = np.bool_(1)
|
| 10 |
+
|
| 11 |
+
b = bool(1)
|
| 12 |
+
i = int(1)
|
| 13 |
+
|
| 14 |
+
AR = np.array([0, 1, 2], dtype=np.int32)
|
| 15 |
+
AR.setflags(write=False)
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
i8 << i8
|
| 19 |
+
i8 >> i8
|
| 20 |
+
i8 | i8
|
| 21 |
+
i8 ^ i8
|
| 22 |
+
i8 & i8
|
| 23 |
+
|
| 24 |
+
i8 << AR
|
| 25 |
+
i8 >> AR
|
| 26 |
+
i8 | AR
|
| 27 |
+
i8 ^ AR
|
| 28 |
+
i8 & AR
|
| 29 |
+
|
| 30 |
+
i4 << i4
|
| 31 |
+
i4 >> i4
|
| 32 |
+
i4 | i4
|
| 33 |
+
i4 ^ i4
|
| 34 |
+
i4 & i4
|
| 35 |
+
|
| 36 |
+
i8 << i4
|
| 37 |
+
i8 >> i4
|
| 38 |
+
i8 | i4
|
| 39 |
+
i8 ^ i4
|
| 40 |
+
i8 & i4
|
| 41 |
+
|
| 42 |
+
i8 << i
|
| 43 |
+
i8 >> i
|
| 44 |
+
i8 | i
|
| 45 |
+
i8 ^ i
|
| 46 |
+
i8 & i
|
| 47 |
+
|
| 48 |
+
i8 << b_
|
| 49 |
+
i8 >> b_
|
| 50 |
+
i8 | b_
|
| 51 |
+
i8 ^ b_
|
| 52 |
+
i8 & b_
|
| 53 |
+
|
| 54 |
+
i8 << b
|
| 55 |
+
i8 >> b
|
| 56 |
+
i8 | b
|
| 57 |
+
i8 ^ b
|
| 58 |
+
i8 & b
|
| 59 |
+
|
| 60 |
+
u8 << u8
|
| 61 |
+
u8 >> u8
|
| 62 |
+
u8 | u8
|
| 63 |
+
u8 ^ u8
|
| 64 |
+
u8 & u8
|
| 65 |
+
|
| 66 |
+
u8 << AR
|
| 67 |
+
u8 >> AR
|
| 68 |
+
u8 | AR
|
| 69 |
+
u8 ^ AR
|
| 70 |
+
u8 & AR
|
| 71 |
+
|
| 72 |
+
u4 << u4
|
| 73 |
+
u4 >> u4
|
| 74 |
+
u4 | u4
|
| 75 |
+
u4 ^ u4
|
| 76 |
+
u4 & u4
|
| 77 |
+
|
| 78 |
+
u4 << i4
|
| 79 |
+
u4 >> i4
|
| 80 |
+
u4 | i4
|
| 81 |
+
u4 ^ i4
|
| 82 |
+
u4 & i4
|
| 83 |
+
|
| 84 |
+
u4 << i
|
| 85 |
+
u4 >> i
|
| 86 |
+
u4 | i
|
| 87 |
+
u4 ^ i
|
| 88 |
+
u4 & i
|
| 89 |
+
|
| 90 |
+
u8 << b_
|
| 91 |
+
u8 >> b_
|
| 92 |
+
u8 | b_
|
| 93 |
+
u8 ^ b_
|
| 94 |
+
u8 & b_
|
| 95 |
+
|
| 96 |
+
u8 << b
|
| 97 |
+
u8 >> b
|
| 98 |
+
u8 | b
|
| 99 |
+
u8 ^ b
|
| 100 |
+
u8 & b
|
| 101 |
+
|
| 102 |
+
b_ << b_
|
| 103 |
+
b_ >> b_
|
| 104 |
+
b_ | b_
|
| 105 |
+
b_ ^ b_
|
| 106 |
+
b_ & b_
|
| 107 |
+
|
| 108 |
+
b_ << AR
|
| 109 |
+
b_ >> AR
|
| 110 |
+
b_ | AR
|
| 111 |
+
b_ ^ AR
|
| 112 |
+
b_ & AR
|
| 113 |
+
|
| 114 |
+
b_ << b
|
| 115 |
+
b_ >> b
|
| 116 |
+
b_ | b
|
| 117 |
+
b_ ^ b
|
| 118 |
+
b_ & b
|
| 119 |
+
|
| 120 |
+
b_ << i
|
| 121 |
+
b_ >> i
|
| 122 |
+
b_ | i
|
| 123 |
+
b_ ^ i
|
| 124 |
+
b_ & i
|
| 125 |
+
|
| 126 |
+
~i8
|
| 127 |
+
~i4
|
| 128 |
+
~u8
|
| 129 |
+
~u4
|
| 130 |
+
~b_
|
| 131 |
+
~AR
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/comparisons.py
ADDED
|
@@ -0,0 +1,301 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from typing import Any
|
| 4 |
+
import numpy as np
|
| 5 |
+
|
| 6 |
+
c16 = np.complex128()
|
| 7 |
+
f8 = np.float64()
|
| 8 |
+
i8 = np.int64()
|
| 9 |
+
u8 = np.uint64()
|
| 10 |
+
|
| 11 |
+
c8 = np.complex64()
|
| 12 |
+
f4 = np.float32()
|
| 13 |
+
i4 = np.int32()
|
| 14 |
+
u4 = np.uint32()
|
| 15 |
+
|
| 16 |
+
dt = np.datetime64(0, "D")
|
| 17 |
+
td = np.timedelta64(0, "D")
|
| 18 |
+
|
| 19 |
+
b_ = np.bool_()
|
| 20 |
+
|
| 21 |
+
b = bool()
|
| 22 |
+
c = complex()
|
| 23 |
+
f = float()
|
| 24 |
+
i = int()
|
| 25 |
+
|
| 26 |
+
SEQ = (0, 1, 2, 3, 4)
|
| 27 |
+
|
| 28 |
+
AR_b: np.ndarray[Any, np.dtype[np.bool_]] = np.array([True])
|
| 29 |
+
AR_u: np.ndarray[Any, np.dtype[np.uint32]] = np.array([1], dtype=np.uint32)
|
| 30 |
+
AR_i: np.ndarray[Any, np.dtype[np.int_]] = np.array([1])
|
| 31 |
+
AR_f: np.ndarray[Any, np.dtype[np.float_]] = np.array([1.0])
|
| 32 |
+
AR_c: np.ndarray[Any, np.dtype[np.complex_]] = np.array([1.0j])
|
| 33 |
+
AR_m: np.ndarray[Any, np.dtype[np.timedelta64]] = np.array([np.timedelta64("1")])
|
| 34 |
+
AR_M: np.ndarray[Any, np.dtype[np.datetime64]] = np.array([np.datetime64("1")])
|
| 35 |
+
AR_O: np.ndarray[Any, np.dtype[np.object_]] = np.array([1], dtype=object)
|
| 36 |
+
|
| 37 |
+
# Arrays
|
| 38 |
+
|
| 39 |
+
AR_b > AR_b
|
| 40 |
+
AR_b > AR_u
|
| 41 |
+
AR_b > AR_i
|
| 42 |
+
AR_b > AR_f
|
| 43 |
+
AR_b > AR_c
|
| 44 |
+
|
| 45 |
+
AR_u > AR_b
|
| 46 |
+
AR_u > AR_u
|
| 47 |
+
AR_u > AR_i
|
| 48 |
+
AR_u > AR_f
|
| 49 |
+
AR_u > AR_c
|
| 50 |
+
|
| 51 |
+
AR_i > AR_b
|
| 52 |
+
AR_i > AR_u
|
| 53 |
+
AR_i > AR_i
|
| 54 |
+
AR_i > AR_f
|
| 55 |
+
AR_i > AR_c
|
| 56 |
+
|
| 57 |
+
AR_f > AR_b
|
| 58 |
+
AR_f > AR_u
|
| 59 |
+
AR_f > AR_i
|
| 60 |
+
AR_f > AR_f
|
| 61 |
+
AR_f > AR_c
|
| 62 |
+
|
| 63 |
+
AR_c > AR_b
|
| 64 |
+
AR_c > AR_u
|
| 65 |
+
AR_c > AR_i
|
| 66 |
+
AR_c > AR_f
|
| 67 |
+
AR_c > AR_c
|
| 68 |
+
|
| 69 |
+
AR_m > AR_b
|
| 70 |
+
AR_m > AR_u
|
| 71 |
+
AR_m > AR_i
|
| 72 |
+
AR_b > AR_m
|
| 73 |
+
AR_u > AR_m
|
| 74 |
+
AR_i > AR_m
|
| 75 |
+
|
| 76 |
+
AR_M > AR_M
|
| 77 |
+
|
| 78 |
+
AR_O > AR_O
|
| 79 |
+
1 > AR_O
|
| 80 |
+
AR_O > 1
|
| 81 |
+
|
| 82 |
+
# Time structures
|
| 83 |
+
|
| 84 |
+
dt > dt
|
| 85 |
+
|
| 86 |
+
td > td
|
| 87 |
+
td > i
|
| 88 |
+
td > i4
|
| 89 |
+
td > i8
|
| 90 |
+
td > AR_i
|
| 91 |
+
td > SEQ
|
| 92 |
+
|
| 93 |
+
# boolean
|
| 94 |
+
|
| 95 |
+
b_ > b
|
| 96 |
+
b_ > b_
|
| 97 |
+
b_ > i
|
| 98 |
+
b_ > i8
|
| 99 |
+
b_ > i4
|
| 100 |
+
b_ > u8
|
| 101 |
+
b_ > u4
|
| 102 |
+
b_ > f
|
| 103 |
+
b_ > f8
|
| 104 |
+
b_ > f4
|
| 105 |
+
b_ > c
|
| 106 |
+
b_ > c16
|
| 107 |
+
b_ > c8
|
| 108 |
+
b_ > AR_i
|
| 109 |
+
b_ > SEQ
|
| 110 |
+
|
| 111 |
+
# Complex
|
| 112 |
+
|
| 113 |
+
c16 > c16
|
| 114 |
+
c16 > f8
|
| 115 |
+
c16 > i8
|
| 116 |
+
c16 > c8
|
| 117 |
+
c16 > f4
|
| 118 |
+
c16 > i4
|
| 119 |
+
c16 > b_
|
| 120 |
+
c16 > b
|
| 121 |
+
c16 > c
|
| 122 |
+
c16 > f
|
| 123 |
+
c16 > i
|
| 124 |
+
c16 > AR_i
|
| 125 |
+
c16 > SEQ
|
| 126 |
+
|
| 127 |
+
c16 > c16
|
| 128 |
+
f8 > c16
|
| 129 |
+
i8 > c16
|
| 130 |
+
c8 > c16
|
| 131 |
+
f4 > c16
|
| 132 |
+
i4 > c16
|
| 133 |
+
b_ > c16
|
| 134 |
+
b > c16
|
| 135 |
+
c > c16
|
| 136 |
+
f > c16
|
| 137 |
+
i > c16
|
| 138 |
+
AR_i > c16
|
| 139 |
+
SEQ > c16
|
| 140 |
+
|
| 141 |
+
c8 > c16
|
| 142 |
+
c8 > f8
|
| 143 |
+
c8 > i8
|
| 144 |
+
c8 > c8
|
| 145 |
+
c8 > f4
|
| 146 |
+
c8 > i4
|
| 147 |
+
c8 > b_
|
| 148 |
+
c8 > b
|
| 149 |
+
c8 > c
|
| 150 |
+
c8 > f
|
| 151 |
+
c8 > i
|
| 152 |
+
c8 > AR_i
|
| 153 |
+
c8 > SEQ
|
| 154 |
+
|
| 155 |
+
c16 > c8
|
| 156 |
+
f8 > c8
|
| 157 |
+
i8 > c8
|
| 158 |
+
c8 > c8
|
| 159 |
+
f4 > c8
|
| 160 |
+
i4 > c8
|
| 161 |
+
b_ > c8
|
| 162 |
+
b > c8
|
| 163 |
+
c > c8
|
| 164 |
+
f > c8
|
| 165 |
+
i > c8
|
| 166 |
+
AR_i > c8
|
| 167 |
+
SEQ > c8
|
| 168 |
+
|
| 169 |
+
# Float
|
| 170 |
+
|
| 171 |
+
f8 > f8
|
| 172 |
+
f8 > i8
|
| 173 |
+
f8 > f4
|
| 174 |
+
f8 > i4
|
| 175 |
+
f8 > b_
|
| 176 |
+
f8 > b
|
| 177 |
+
f8 > c
|
| 178 |
+
f8 > f
|
| 179 |
+
f8 > i
|
| 180 |
+
f8 > AR_i
|
| 181 |
+
f8 > SEQ
|
| 182 |
+
|
| 183 |
+
f8 > f8
|
| 184 |
+
i8 > f8
|
| 185 |
+
f4 > f8
|
| 186 |
+
i4 > f8
|
| 187 |
+
b_ > f8
|
| 188 |
+
b > f8
|
| 189 |
+
c > f8
|
| 190 |
+
f > f8
|
| 191 |
+
i > f8
|
| 192 |
+
AR_i > f8
|
| 193 |
+
SEQ > f8
|
| 194 |
+
|
| 195 |
+
f4 > f8
|
| 196 |
+
f4 > i8
|
| 197 |
+
f4 > f4
|
| 198 |
+
f4 > i4
|
| 199 |
+
f4 > b_
|
| 200 |
+
f4 > b
|
| 201 |
+
f4 > c
|
| 202 |
+
f4 > f
|
| 203 |
+
f4 > i
|
| 204 |
+
f4 > AR_i
|
| 205 |
+
f4 > SEQ
|
| 206 |
+
|
| 207 |
+
f8 > f4
|
| 208 |
+
i8 > f4
|
| 209 |
+
f4 > f4
|
| 210 |
+
i4 > f4
|
| 211 |
+
b_ > f4
|
| 212 |
+
b > f4
|
| 213 |
+
c > f4
|
| 214 |
+
f > f4
|
| 215 |
+
i > f4
|
| 216 |
+
AR_i > f4
|
| 217 |
+
SEQ > f4
|
| 218 |
+
|
| 219 |
+
# Int
|
| 220 |
+
|
| 221 |
+
i8 > i8
|
| 222 |
+
i8 > u8
|
| 223 |
+
i8 > i4
|
| 224 |
+
i8 > u4
|
| 225 |
+
i8 > b_
|
| 226 |
+
i8 > b
|
| 227 |
+
i8 > c
|
| 228 |
+
i8 > f
|
| 229 |
+
i8 > i
|
| 230 |
+
i8 > AR_i
|
| 231 |
+
i8 > SEQ
|
| 232 |
+
|
| 233 |
+
u8 > u8
|
| 234 |
+
u8 > i4
|
| 235 |
+
u8 > u4
|
| 236 |
+
u8 > b_
|
| 237 |
+
u8 > b
|
| 238 |
+
u8 > c
|
| 239 |
+
u8 > f
|
| 240 |
+
u8 > i
|
| 241 |
+
u8 > AR_i
|
| 242 |
+
u8 > SEQ
|
| 243 |
+
|
| 244 |
+
i8 > i8
|
| 245 |
+
u8 > i8
|
| 246 |
+
i4 > i8
|
| 247 |
+
u4 > i8
|
| 248 |
+
b_ > i8
|
| 249 |
+
b > i8
|
| 250 |
+
c > i8
|
| 251 |
+
f > i8
|
| 252 |
+
i > i8
|
| 253 |
+
AR_i > i8
|
| 254 |
+
SEQ > i8
|
| 255 |
+
|
| 256 |
+
u8 > u8
|
| 257 |
+
i4 > u8
|
| 258 |
+
u4 > u8
|
| 259 |
+
b_ > u8
|
| 260 |
+
b > u8
|
| 261 |
+
c > u8
|
| 262 |
+
f > u8
|
| 263 |
+
i > u8
|
| 264 |
+
AR_i > u8
|
| 265 |
+
SEQ > u8
|
| 266 |
+
|
| 267 |
+
i4 > i8
|
| 268 |
+
i4 > i4
|
| 269 |
+
i4 > i
|
| 270 |
+
i4 > b_
|
| 271 |
+
i4 > b
|
| 272 |
+
i4 > AR_i
|
| 273 |
+
i4 > SEQ
|
| 274 |
+
|
| 275 |
+
u4 > i8
|
| 276 |
+
u4 > i4
|
| 277 |
+
u4 > u8
|
| 278 |
+
u4 > u4
|
| 279 |
+
u4 > i
|
| 280 |
+
u4 > b_
|
| 281 |
+
u4 > b
|
| 282 |
+
u4 > AR_i
|
| 283 |
+
u4 > SEQ
|
| 284 |
+
|
| 285 |
+
i8 > i4
|
| 286 |
+
i4 > i4
|
| 287 |
+
i > i4
|
| 288 |
+
b_ > i4
|
| 289 |
+
b > i4
|
| 290 |
+
AR_i > i4
|
| 291 |
+
SEQ > i4
|
| 292 |
+
|
| 293 |
+
i8 > u4
|
| 294 |
+
i4 > u4
|
| 295 |
+
u8 > u4
|
| 296 |
+
u4 > u4
|
| 297 |
+
b_ > u4
|
| 298 |
+
b > u4
|
| 299 |
+
i > u4
|
| 300 |
+
AR_i > u4
|
| 301 |
+
SEQ > u4
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/dtype.py
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
dtype_obj = np.dtype(np.str_)
|
| 4 |
+
void_dtype_obj = np.dtype([("f0", np.float64), ("f1", np.float32)])
|
| 5 |
+
|
| 6 |
+
np.dtype(dtype=np.int64)
|
| 7 |
+
np.dtype(int)
|
| 8 |
+
np.dtype("int")
|
| 9 |
+
np.dtype(None)
|
| 10 |
+
|
| 11 |
+
np.dtype((int, 2))
|
| 12 |
+
np.dtype((int, (1,)))
|
| 13 |
+
|
| 14 |
+
np.dtype({"names": ["a", "b"], "formats": [int, float]})
|
| 15 |
+
np.dtype({"names": ["a"], "formats": [int], "titles": [object]})
|
| 16 |
+
np.dtype({"names": ["a"], "formats": [int], "titles": [object()]})
|
| 17 |
+
|
| 18 |
+
np.dtype([("name", np.str_, 16), ("grades", np.float64, (2,)), ("age", "int32")])
|
| 19 |
+
|
| 20 |
+
np.dtype(
|
| 21 |
+
{
|
| 22 |
+
"names": ["a", "b"],
|
| 23 |
+
"formats": [int, float],
|
| 24 |
+
"itemsize": 9,
|
| 25 |
+
"aligned": False,
|
| 26 |
+
"titles": ["x", "y"],
|
| 27 |
+
"offsets": [0, 1],
|
| 28 |
+
}
|
| 29 |
+
)
|
| 30 |
+
|
| 31 |
+
np.dtype((np.float_, float))
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
class Test:
|
| 35 |
+
dtype = np.dtype(float)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
np.dtype(Test())
|
| 39 |
+
|
| 40 |
+
# Methods and attributes
|
| 41 |
+
dtype_obj.base
|
| 42 |
+
dtype_obj.subdtype
|
| 43 |
+
dtype_obj.newbyteorder()
|
| 44 |
+
dtype_obj.type
|
| 45 |
+
dtype_obj.name
|
| 46 |
+
dtype_obj.names
|
| 47 |
+
|
| 48 |
+
dtype_obj * 0
|
| 49 |
+
dtype_obj * 2
|
| 50 |
+
|
| 51 |
+
0 * dtype_obj
|
| 52 |
+
2 * dtype_obj
|
| 53 |
+
|
| 54 |
+
void_dtype_obj["f0"]
|
| 55 |
+
void_dtype_obj[0]
|
| 56 |
+
void_dtype_obj[["f0", "f1"]]
|
| 57 |
+
void_dtype_obj[["f0"]]
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/flatiter.py
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
a = np.empty((2, 2)).flat
|
| 4 |
+
|
| 5 |
+
a.base
|
| 6 |
+
a.copy()
|
| 7 |
+
a.coords
|
| 8 |
+
a.index
|
| 9 |
+
iter(a)
|
| 10 |
+
next(a)
|
| 11 |
+
a[0]
|
| 12 |
+
a[[0, 1, 2]]
|
| 13 |
+
a[...]
|
| 14 |
+
a[:]
|
| 15 |
+
a.__array__()
|
| 16 |
+
a.__array__(np.dtype(np.float64))
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/index_tricks.py
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
from typing import Any
|
| 3 |
+
import numpy as np
|
| 4 |
+
|
| 5 |
+
AR_LIKE_b = [[True, True], [True, True]]
|
| 6 |
+
AR_LIKE_i = [[1, 2], [3, 4]]
|
| 7 |
+
AR_LIKE_f = [[1.0, 2.0], [3.0, 4.0]]
|
| 8 |
+
AR_LIKE_U = [["1", "2"], ["3", "4"]]
|
| 9 |
+
|
| 10 |
+
AR_i8: np.ndarray[Any, np.dtype[np.int64]] = np.array(AR_LIKE_i, dtype=np.int64)
|
| 11 |
+
|
| 12 |
+
np.ndenumerate(AR_i8)
|
| 13 |
+
np.ndenumerate(AR_LIKE_f)
|
| 14 |
+
np.ndenumerate(AR_LIKE_U)
|
| 15 |
+
|
| 16 |
+
np.ndenumerate(AR_i8).iter
|
| 17 |
+
np.ndenumerate(AR_LIKE_f).iter
|
| 18 |
+
np.ndenumerate(AR_LIKE_U).iter
|
| 19 |
+
|
| 20 |
+
next(np.ndenumerate(AR_i8))
|
| 21 |
+
next(np.ndenumerate(AR_LIKE_f))
|
| 22 |
+
next(np.ndenumerate(AR_LIKE_U))
|
| 23 |
+
|
| 24 |
+
iter(np.ndenumerate(AR_i8))
|
| 25 |
+
iter(np.ndenumerate(AR_LIKE_f))
|
| 26 |
+
iter(np.ndenumerate(AR_LIKE_U))
|
| 27 |
+
|
| 28 |
+
iter(np.ndindex(1, 2, 3))
|
| 29 |
+
next(np.ndindex(1, 2, 3))
|
| 30 |
+
|
| 31 |
+
np.unravel_index([22, 41, 37], (7, 6))
|
| 32 |
+
np.unravel_index([31, 41, 13], (7, 6), order='F')
|
| 33 |
+
np.unravel_index(1621, (6, 7, 8, 9))
|
| 34 |
+
|
| 35 |
+
np.ravel_multi_index(AR_LIKE_i, (7, 6))
|
| 36 |
+
np.ravel_multi_index(AR_LIKE_i, (7, 6), order='F')
|
| 37 |
+
np.ravel_multi_index(AR_LIKE_i, (4, 6), mode='clip')
|
| 38 |
+
np.ravel_multi_index(AR_LIKE_i, (4, 4), mode=('clip', 'wrap'))
|
| 39 |
+
np.ravel_multi_index((3, 1, 4, 1), (6, 7, 8, 9))
|
| 40 |
+
|
| 41 |
+
np.mgrid[1:1:2]
|
| 42 |
+
np.mgrid[1:1:2, None:10]
|
| 43 |
+
|
| 44 |
+
np.ogrid[1:1:2]
|
| 45 |
+
np.ogrid[1:1:2, None:10]
|
| 46 |
+
|
| 47 |
+
np.index_exp[0:1]
|
| 48 |
+
np.index_exp[0:1, None:3]
|
| 49 |
+
np.index_exp[0, 0:1, ..., [0, 1, 3]]
|
| 50 |
+
|
| 51 |
+
np.s_[0:1]
|
| 52 |
+
np.s_[0:1, None:3]
|
| 53 |
+
np.s_[0, 0:1, ..., [0, 1, 3]]
|
| 54 |
+
|
| 55 |
+
np.ix_(AR_LIKE_b[0])
|
| 56 |
+
np.ix_(AR_LIKE_i[0], AR_LIKE_f[0])
|
| 57 |
+
np.ix_(AR_i8[0])
|
| 58 |
+
|
| 59 |
+
np.fill_diagonal(AR_i8, 5)
|
| 60 |
+
|
| 61 |
+
np.diag_indices(4)
|
| 62 |
+
np.diag_indices(2, 3)
|
| 63 |
+
|
| 64 |
+
np.diag_indices_from(AR_i8)
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/lib_version.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from numpy.lib import NumpyVersion
|
| 2 |
+
|
| 3 |
+
version = NumpyVersion("1.8.0")
|
| 4 |
+
|
| 5 |
+
version.vstring
|
| 6 |
+
version.version
|
| 7 |
+
version.major
|
| 8 |
+
version.minor
|
| 9 |
+
version.bugfix
|
| 10 |
+
version.pre_release
|
| 11 |
+
version.is_devversion
|
| 12 |
+
|
| 13 |
+
version == version
|
| 14 |
+
version != version
|
| 15 |
+
version < "1.8.0"
|
| 16 |
+
version <= version
|
| 17 |
+
version > version
|
| 18 |
+
version >= "1.8.0"
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/literal.py
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from __future__ import annotations
|
| 2 |
+
|
| 3 |
+
from functools import partial
|
| 4 |
+
from collections.abc import Callable
|
| 5 |
+
|
| 6 |
+
import pytest # type: ignore
|
| 7 |
+
import numpy as np
|
| 8 |
+
|
| 9 |
+
AR = np.array(0)
|
| 10 |
+
AR.setflags(write=False)
|
| 11 |
+
|
| 12 |
+
KACF = frozenset({None, "K", "A", "C", "F"})
|
| 13 |
+
ACF = frozenset({None, "A", "C", "F"})
|
| 14 |
+
CF = frozenset({None, "C", "F"})
|
| 15 |
+
|
| 16 |
+
order_list: list[tuple[frozenset, Callable]] = [
|
| 17 |
+
(KACF, partial(np.ndarray, 1)),
|
| 18 |
+
(KACF, AR.tobytes),
|
| 19 |
+
(KACF, partial(AR.astype, int)),
|
| 20 |
+
(KACF, AR.copy),
|
| 21 |
+
(ACF, partial(AR.reshape, 1)),
|
| 22 |
+
(KACF, AR.flatten),
|
| 23 |
+
(KACF, AR.ravel),
|
| 24 |
+
(KACF, partial(np.array, 1)),
|
| 25 |
+
(CF, partial(np.zeros, 1)),
|
| 26 |
+
(CF, partial(np.ones, 1)),
|
| 27 |
+
(CF, partial(np.empty, 1)),
|
| 28 |
+
(CF, partial(np.full, 1, 1)),
|
| 29 |
+
(KACF, partial(np.zeros_like, AR)),
|
| 30 |
+
(KACF, partial(np.ones_like, AR)),
|
| 31 |
+
(KACF, partial(np.empty_like, AR)),
|
| 32 |
+
(KACF, partial(np.full_like, AR, 1)),
|
| 33 |
+
(KACF, partial(np.add, 1, 1)), # i.e. np.ufunc.__call__
|
| 34 |
+
(ACF, partial(np.reshape, AR, 1)),
|
| 35 |
+
(KACF, partial(np.ravel, AR)),
|
| 36 |
+
(KACF, partial(np.asarray, 1)),
|
| 37 |
+
(KACF, partial(np.asanyarray, 1)),
|
| 38 |
+
]
|
| 39 |
+
|
| 40 |
+
for order_set, func in order_list:
|
| 41 |
+
for order in order_set:
|
| 42 |
+
func(order=order)
|
| 43 |
+
|
| 44 |
+
invalid_orders = KACF - order_set
|
| 45 |
+
for order in invalid_orders:
|
| 46 |
+
with pytest.raises(ValueError):
|
| 47 |
+
func(order=order)
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/mod.py
ADDED
|
@@ -0,0 +1,149 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
f8 = np.float64(1)
|
| 4 |
+
i8 = np.int64(1)
|
| 5 |
+
u8 = np.uint64(1)
|
| 6 |
+
|
| 7 |
+
f4 = np.float32(1)
|
| 8 |
+
i4 = np.int32(1)
|
| 9 |
+
u4 = np.uint32(1)
|
| 10 |
+
|
| 11 |
+
td = np.timedelta64(1, "D")
|
| 12 |
+
b_ = np.bool_(1)
|
| 13 |
+
|
| 14 |
+
b = bool(1)
|
| 15 |
+
f = float(1)
|
| 16 |
+
i = int(1)
|
| 17 |
+
|
| 18 |
+
AR = np.array([1], dtype=np.bool_)
|
| 19 |
+
AR.setflags(write=False)
|
| 20 |
+
|
| 21 |
+
AR2 = np.array([1], dtype=np.timedelta64)
|
| 22 |
+
AR2.setflags(write=False)
|
| 23 |
+
|
| 24 |
+
# Time structures
|
| 25 |
+
|
| 26 |
+
td % td
|
| 27 |
+
td % AR2
|
| 28 |
+
AR2 % td
|
| 29 |
+
|
| 30 |
+
divmod(td, td)
|
| 31 |
+
divmod(td, AR2)
|
| 32 |
+
divmod(AR2, td)
|
| 33 |
+
|
| 34 |
+
# Bool
|
| 35 |
+
|
| 36 |
+
b_ % b
|
| 37 |
+
b_ % i
|
| 38 |
+
b_ % f
|
| 39 |
+
b_ % b_
|
| 40 |
+
b_ % i8
|
| 41 |
+
b_ % u8
|
| 42 |
+
b_ % f8
|
| 43 |
+
b_ % AR
|
| 44 |
+
|
| 45 |
+
divmod(b_, b)
|
| 46 |
+
divmod(b_, i)
|
| 47 |
+
divmod(b_, f)
|
| 48 |
+
divmod(b_, b_)
|
| 49 |
+
divmod(b_, i8)
|
| 50 |
+
divmod(b_, u8)
|
| 51 |
+
divmod(b_, f8)
|
| 52 |
+
divmod(b_, AR)
|
| 53 |
+
|
| 54 |
+
b % b_
|
| 55 |
+
i % b_
|
| 56 |
+
f % b_
|
| 57 |
+
b_ % b_
|
| 58 |
+
i8 % b_
|
| 59 |
+
u8 % b_
|
| 60 |
+
f8 % b_
|
| 61 |
+
AR % b_
|
| 62 |
+
|
| 63 |
+
divmod(b, b_)
|
| 64 |
+
divmod(i, b_)
|
| 65 |
+
divmod(f, b_)
|
| 66 |
+
divmod(b_, b_)
|
| 67 |
+
divmod(i8, b_)
|
| 68 |
+
divmod(u8, b_)
|
| 69 |
+
divmod(f8, b_)
|
| 70 |
+
divmod(AR, b_)
|
| 71 |
+
|
| 72 |
+
# int
|
| 73 |
+
|
| 74 |
+
i8 % b
|
| 75 |
+
i8 % i
|
| 76 |
+
i8 % f
|
| 77 |
+
i8 % i8
|
| 78 |
+
i8 % f8
|
| 79 |
+
i4 % i8
|
| 80 |
+
i4 % f8
|
| 81 |
+
i4 % i4
|
| 82 |
+
i4 % f4
|
| 83 |
+
i8 % AR
|
| 84 |
+
|
| 85 |
+
divmod(i8, b)
|
| 86 |
+
divmod(i8, i)
|
| 87 |
+
divmod(i8, f)
|
| 88 |
+
divmod(i8, i8)
|
| 89 |
+
divmod(i8, f8)
|
| 90 |
+
divmod(i8, i4)
|
| 91 |
+
divmod(i8, f4)
|
| 92 |
+
divmod(i4, i4)
|
| 93 |
+
divmod(i4, f4)
|
| 94 |
+
divmod(i8, AR)
|
| 95 |
+
|
| 96 |
+
b % i8
|
| 97 |
+
i % i8
|
| 98 |
+
f % i8
|
| 99 |
+
i8 % i8
|
| 100 |
+
f8 % i8
|
| 101 |
+
i8 % i4
|
| 102 |
+
f8 % i4
|
| 103 |
+
i4 % i4
|
| 104 |
+
f4 % i4
|
| 105 |
+
AR % i8
|
| 106 |
+
|
| 107 |
+
divmod(b, i8)
|
| 108 |
+
divmod(i, i8)
|
| 109 |
+
divmod(f, i8)
|
| 110 |
+
divmod(i8, i8)
|
| 111 |
+
divmod(f8, i8)
|
| 112 |
+
divmod(i4, i8)
|
| 113 |
+
divmod(f4, i8)
|
| 114 |
+
divmod(i4, i4)
|
| 115 |
+
divmod(f4, i4)
|
| 116 |
+
divmod(AR, i8)
|
| 117 |
+
|
| 118 |
+
# float
|
| 119 |
+
|
| 120 |
+
f8 % b
|
| 121 |
+
f8 % i
|
| 122 |
+
f8 % f
|
| 123 |
+
i8 % f4
|
| 124 |
+
f4 % f4
|
| 125 |
+
f8 % AR
|
| 126 |
+
|
| 127 |
+
divmod(f8, b)
|
| 128 |
+
divmod(f8, i)
|
| 129 |
+
divmod(f8, f)
|
| 130 |
+
divmod(f8, f8)
|
| 131 |
+
divmod(f8, f4)
|
| 132 |
+
divmod(f4, f4)
|
| 133 |
+
divmod(f8, AR)
|
| 134 |
+
|
| 135 |
+
b % f8
|
| 136 |
+
i % f8
|
| 137 |
+
f % f8
|
| 138 |
+
f8 % f8
|
| 139 |
+
f8 % f8
|
| 140 |
+
f4 % f4
|
| 141 |
+
AR % f8
|
| 142 |
+
|
| 143 |
+
divmod(b, f8)
|
| 144 |
+
divmod(i, f8)
|
| 145 |
+
divmod(f, f8)
|
| 146 |
+
divmod(f8, f8)
|
| 147 |
+
divmod(f4, f8)
|
| 148 |
+
divmod(f4, f4)
|
| 149 |
+
divmod(AR, f8)
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/multiarray.py
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
import numpy.typing as npt
|
| 3 |
+
|
| 4 |
+
AR_f8: npt.NDArray[np.float64] = np.array([1.0])
|
| 5 |
+
AR_i4 = np.array([1], dtype=np.int32)
|
| 6 |
+
AR_u1 = np.array([1], dtype=np.uint8)
|
| 7 |
+
|
| 8 |
+
AR_LIKE_f = [1.5]
|
| 9 |
+
AR_LIKE_i = [1]
|
| 10 |
+
|
| 11 |
+
b_f8 = np.broadcast(AR_f8)
|
| 12 |
+
b_i4_f8_f8 = np.broadcast(AR_i4, AR_f8, AR_f8)
|
| 13 |
+
|
| 14 |
+
next(b_f8)
|
| 15 |
+
b_f8.reset()
|
| 16 |
+
b_f8.index
|
| 17 |
+
b_f8.iters
|
| 18 |
+
b_f8.nd
|
| 19 |
+
b_f8.ndim
|
| 20 |
+
b_f8.numiter
|
| 21 |
+
b_f8.shape
|
| 22 |
+
b_f8.size
|
| 23 |
+
|
| 24 |
+
next(b_i4_f8_f8)
|
| 25 |
+
b_i4_f8_f8.reset()
|
| 26 |
+
b_i4_f8_f8.ndim
|
| 27 |
+
b_i4_f8_f8.index
|
| 28 |
+
b_i4_f8_f8.iters
|
| 29 |
+
b_i4_f8_f8.nd
|
| 30 |
+
b_i4_f8_f8.numiter
|
| 31 |
+
b_i4_f8_f8.shape
|
| 32 |
+
b_i4_f8_f8.size
|
| 33 |
+
|
| 34 |
+
np.inner(AR_f8, AR_i4)
|
| 35 |
+
|
| 36 |
+
np.where([True, True, False])
|
| 37 |
+
np.where([True, True, False], 1, 0)
|
| 38 |
+
|
| 39 |
+
np.lexsort([0, 1, 2])
|
| 40 |
+
|
| 41 |
+
np.can_cast(np.dtype("i8"), int)
|
| 42 |
+
np.can_cast(AR_f8, "f8")
|
| 43 |
+
np.can_cast(AR_f8, np.complex128, casting="unsafe")
|
| 44 |
+
|
| 45 |
+
np.min_scalar_type([1])
|
| 46 |
+
np.min_scalar_type(AR_f8)
|
| 47 |
+
|
| 48 |
+
np.result_type(int, AR_i4)
|
| 49 |
+
np.result_type(AR_f8, AR_u1)
|
| 50 |
+
np.result_type(AR_f8, np.complex128)
|
| 51 |
+
|
| 52 |
+
np.dot(AR_LIKE_f, AR_i4)
|
| 53 |
+
np.dot(AR_u1, 1)
|
| 54 |
+
np.dot(1.5j, 1)
|
| 55 |
+
np.dot(AR_u1, 1, out=AR_f8)
|
| 56 |
+
|
| 57 |
+
np.vdot(AR_LIKE_f, AR_i4)
|
| 58 |
+
np.vdot(AR_u1, 1)
|
| 59 |
+
np.vdot(1.5j, 1)
|
| 60 |
+
|
| 61 |
+
np.bincount(AR_i4)
|
| 62 |
+
|
| 63 |
+
np.copyto(AR_f8, [1.6])
|
| 64 |
+
|
| 65 |
+
np.putmask(AR_f8, [True], 1.5)
|
| 66 |
+
|
| 67 |
+
np.packbits(AR_i4)
|
| 68 |
+
np.packbits(AR_u1)
|
| 69 |
+
|
| 70 |
+
np.unpackbits(AR_u1)
|
| 71 |
+
|
| 72 |
+
np.shares_memory(1, 2)
|
| 73 |
+
np.shares_memory(AR_f8, AR_f8, max_work=1)
|
| 74 |
+
|
| 75 |
+
np.may_share_memory(1, 2)
|
| 76 |
+
np.may_share_memory(AR_f8, AR_f8, max_work=1)
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py
ADDED
|
@@ -0,0 +1,94 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import tempfile
|
| 3 |
+
|
| 4 |
+
import numpy as np
|
| 5 |
+
|
| 6 |
+
nd = np.array([[1, 2], [3, 4]])
|
| 7 |
+
scalar_array = np.array(1)
|
| 8 |
+
|
| 9 |
+
# item
|
| 10 |
+
scalar_array.item()
|
| 11 |
+
nd.item(1)
|
| 12 |
+
nd.item(0, 1)
|
| 13 |
+
nd.item((0, 1))
|
| 14 |
+
|
| 15 |
+
# tolist is pretty simple
|
| 16 |
+
|
| 17 |
+
# itemset
|
| 18 |
+
scalar_array.itemset(3)
|
| 19 |
+
nd.itemset(3, 0)
|
| 20 |
+
nd.itemset((0, 0), 3)
|
| 21 |
+
|
| 22 |
+
# tobytes
|
| 23 |
+
nd.tobytes()
|
| 24 |
+
nd.tobytes("C")
|
| 25 |
+
nd.tobytes(None)
|
| 26 |
+
|
| 27 |
+
# tofile
|
| 28 |
+
if os.name != "nt":
|
| 29 |
+
with tempfile.NamedTemporaryFile(suffix=".txt") as tmp:
|
| 30 |
+
nd.tofile(tmp.name)
|
| 31 |
+
nd.tofile(tmp.name, "")
|
| 32 |
+
nd.tofile(tmp.name, sep="")
|
| 33 |
+
|
| 34 |
+
nd.tofile(tmp.name, "", "%s")
|
| 35 |
+
nd.tofile(tmp.name, format="%s")
|
| 36 |
+
|
| 37 |
+
nd.tofile(tmp)
|
| 38 |
+
|
| 39 |
+
# dump is pretty simple
|
| 40 |
+
# dumps is pretty simple
|
| 41 |
+
|
| 42 |
+
# astype
|
| 43 |
+
nd.astype("float")
|
| 44 |
+
nd.astype(float)
|
| 45 |
+
|
| 46 |
+
nd.astype(float, "K")
|
| 47 |
+
nd.astype(float, order="K")
|
| 48 |
+
|
| 49 |
+
nd.astype(float, "K", "unsafe")
|
| 50 |
+
nd.astype(float, casting="unsafe")
|
| 51 |
+
|
| 52 |
+
nd.astype(float, "K", "unsafe", True)
|
| 53 |
+
nd.astype(float, subok=True)
|
| 54 |
+
|
| 55 |
+
nd.astype(float, "K", "unsafe", True, True)
|
| 56 |
+
nd.astype(float, copy=True)
|
| 57 |
+
|
| 58 |
+
# byteswap
|
| 59 |
+
nd.byteswap()
|
| 60 |
+
nd.byteswap(True)
|
| 61 |
+
|
| 62 |
+
# copy
|
| 63 |
+
nd.copy()
|
| 64 |
+
nd.copy("C")
|
| 65 |
+
|
| 66 |
+
# view
|
| 67 |
+
nd.view()
|
| 68 |
+
nd.view(np.int64)
|
| 69 |
+
nd.view(dtype=np.int64)
|
| 70 |
+
nd.view(np.int64, np.matrix)
|
| 71 |
+
nd.view(type=np.matrix)
|
| 72 |
+
|
| 73 |
+
# getfield
|
| 74 |
+
complex_array = np.array([[1 + 1j, 0], [0, 1 - 1j]], dtype=np.complex128)
|
| 75 |
+
|
| 76 |
+
complex_array.getfield("float")
|
| 77 |
+
complex_array.getfield(float)
|
| 78 |
+
|
| 79 |
+
complex_array.getfield("float", 8)
|
| 80 |
+
complex_array.getfield(float, offset=8)
|
| 81 |
+
|
| 82 |
+
# setflags
|
| 83 |
+
nd.setflags()
|
| 84 |
+
|
| 85 |
+
nd.setflags(True)
|
| 86 |
+
nd.setflags(write=True)
|
| 87 |
+
|
| 88 |
+
nd.setflags(True, True)
|
| 89 |
+
nd.setflags(write=True, align=True)
|
| 90 |
+
|
| 91 |
+
nd.setflags(True, True, False)
|
| 92 |
+
nd.setflags(write=True, align=True, uic=False)
|
| 93 |
+
|
| 94 |
+
# fill is pretty simple
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_misc.py
ADDED
|
@@ -0,0 +1,185 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Tests for miscellaneous (non-magic) ``np.ndarray``/``np.generic`` methods.
|
| 3 |
+
|
| 4 |
+
More extensive tests are performed for the methods'
|
| 5 |
+
function-based counterpart in `../from_numeric.py`.
|
| 6 |
+
|
| 7 |
+
"""
|
| 8 |
+
|
| 9 |
+
from __future__ import annotations
|
| 10 |
+
|
| 11 |
+
import operator
|
| 12 |
+
from typing import cast, Any
|
| 13 |
+
|
| 14 |
+
import numpy as np
|
| 15 |
+
|
| 16 |
+
class SubClass(np.ndarray): ...
|
| 17 |
+
|
| 18 |
+
i4 = np.int32(1)
|
| 19 |
+
A: np.ndarray[Any, np.dtype[np.int32]] = np.array([[1]], dtype=np.int32)
|
| 20 |
+
B0 = np.empty((), dtype=np.int32).view(SubClass)
|
| 21 |
+
B1 = np.empty((1,), dtype=np.int32).view(SubClass)
|
| 22 |
+
B2 = np.empty((1, 1), dtype=np.int32).view(SubClass)
|
| 23 |
+
C: np.ndarray[Any, np.dtype[np.int32]] = np.array([0, 1, 2], dtype=np.int32)
|
| 24 |
+
D = np.ones(3).view(SubClass)
|
| 25 |
+
|
| 26 |
+
i4.all()
|
| 27 |
+
A.all()
|
| 28 |
+
A.all(axis=0)
|
| 29 |
+
A.all(keepdims=True)
|
| 30 |
+
A.all(out=B0)
|
| 31 |
+
|
| 32 |
+
i4.any()
|
| 33 |
+
A.any()
|
| 34 |
+
A.any(axis=0)
|
| 35 |
+
A.any(keepdims=True)
|
| 36 |
+
A.any(out=B0)
|
| 37 |
+
|
| 38 |
+
i4.argmax()
|
| 39 |
+
A.argmax()
|
| 40 |
+
A.argmax(axis=0)
|
| 41 |
+
A.argmax(out=B0)
|
| 42 |
+
|
| 43 |
+
i4.argmin()
|
| 44 |
+
A.argmin()
|
| 45 |
+
A.argmin(axis=0)
|
| 46 |
+
A.argmin(out=B0)
|
| 47 |
+
|
| 48 |
+
i4.argsort()
|
| 49 |
+
A.argsort()
|
| 50 |
+
|
| 51 |
+
i4.choose([()])
|
| 52 |
+
_choices = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]], dtype=np.int32)
|
| 53 |
+
C.choose(_choices)
|
| 54 |
+
C.choose(_choices, out=D)
|
| 55 |
+
|
| 56 |
+
i4.clip(1)
|
| 57 |
+
A.clip(1)
|
| 58 |
+
A.clip(None, 1)
|
| 59 |
+
A.clip(1, out=B2)
|
| 60 |
+
A.clip(None, 1, out=B2)
|
| 61 |
+
|
| 62 |
+
i4.compress([1])
|
| 63 |
+
A.compress([1])
|
| 64 |
+
A.compress([1], out=B1)
|
| 65 |
+
|
| 66 |
+
i4.conj()
|
| 67 |
+
A.conj()
|
| 68 |
+
B0.conj()
|
| 69 |
+
|
| 70 |
+
i4.conjugate()
|
| 71 |
+
A.conjugate()
|
| 72 |
+
B0.conjugate()
|
| 73 |
+
|
| 74 |
+
i4.cumprod()
|
| 75 |
+
A.cumprod()
|
| 76 |
+
A.cumprod(out=B1)
|
| 77 |
+
|
| 78 |
+
i4.cumsum()
|
| 79 |
+
A.cumsum()
|
| 80 |
+
A.cumsum(out=B1)
|
| 81 |
+
|
| 82 |
+
i4.max()
|
| 83 |
+
A.max()
|
| 84 |
+
A.max(axis=0)
|
| 85 |
+
A.max(keepdims=True)
|
| 86 |
+
A.max(out=B0)
|
| 87 |
+
|
| 88 |
+
i4.mean()
|
| 89 |
+
A.mean()
|
| 90 |
+
A.mean(axis=0)
|
| 91 |
+
A.mean(keepdims=True)
|
| 92 |
+
A.mean(out=B0)
|
| 93 |
+
|
| 94 |
+
i4.min()
|
| 95 |
+
A.min()
|
| 96 |
+
A.min(axis=0)
|
| 97 |
+
A.min(keepdims=True)
|
| 98 |
+
A.min(out=B0)
|
| 99 |
+
|
| 100 |
+
i4.newbyteorder()
|
| 101 |
+
A.newbyteorder()
|
| 102 |
+
B0.newbyteorder('|')
|
| 103 |
+
|
| 104 |
+
i4.prod()
|
| 105 |
+
A.prod()
|
| 106 |
+
A.prod(axis=0)
|
| 107 |
+
A.prod(keepdims=True)
|
| 108 |
+
A.prod(out=B0)
|
| 109 |
+
|
| 110 |
+
i4.ptp()
|
| 111 |
+
A.ptp()
|
| 112 |
+
A.ptp(axis=0)
|
| 113 |
+
A.ptp(keepdims=True)
|
| 114 |
+
A.astype(int).ptp(out=B0)
|
| 115 |
+
|
| 116 |
+
i4.round()
|
| 117 |
+
A.round()
|
| 118 |
+
A.round(out=B2)
|
| 119 |
+
|
| 120 |
+
i4.repeat(1)
|
| 121 |
+
A.repeat(1)
|
| 122 |
+
B0.repeat(1)
|
| 123 |
+
|
| 124 |
+
i4.std()
|
| 125 |
+
A.std()
|
| 126 |
+
A.std(axis=0)
|
| 127 |
+
A.std(keepdims=True)
|
| 128 |
+
A.std(out=B0.astype(np.float64))
|
| 129 |
+
|
| 130 |
+
i4.sum()
|
| 131 |
+
A.sum()
|
| 132 |
+
A.sum(axis=0)
|
| 133 |
+
A.sum(keepdims=True)
|
| 134 |
+
A.sum(out=B0)
|
| 135 |
+
|
| 136 |
+
i4.take(0)
|
| 137 |
+
A.take(0)
|
| 138 |
+
A.take([0])
|
| 139 |
+
A.take(0, out=B0)
|
| 140 |
+
A.take([0], out=B1)
|
| 141 |
+
|
| 142 |
+
i4.var()
|
| 143 |
+
A.var()
|
| 144 |
+
A.var(axis=0)
|
| 145 |
+
A.var(keepdims=True)
|
| 146 |
+
A.var(out=B0)
|
| 147 |
+
|
| 148 |
+
A.argpartition([0])
|
| 149 |
+
|
| 150 |
+
A.diagonal()
|
| 151 |
+
|
| 152 |
+
A.dot(1)
|
| 153 |
+
A.dot(1, out=B2)
|
| 154 |
+
|
| 155 |
+
A.nonzero()
|
| 156 |
+
|
| 157 |
+
C.searchsorted(1)
|
| 158 |
+
|
| 159 |
+
A.trace()
|
| 160 |
+
A.trace(out=B0)
|
| 161 |
+
|
| 162 |
+
void = cast(np.void, np.array(1, dtype=[("f", np.float64)]).take(0))
|
| 163 |
+
void.setfield(10, np.float64)
|
| 164 |
+
|
| 165 |
+
A.item(0)
|
| 166 |
+
C.item(0)
|
| 167 |
+
|
| 168 |
+
A.ravel()
|
| 169 |
+
C.ravel()
|
| 170 |
+
|
| 171 |
+
A.flatten()
|
| 172 |
+
C.flatten()
|
| 173 |
+
|
| 174 |
+
A.reshape(1)
|
| 175 |
+
C.reshape(3)
|
| 176 |
+
|
| 177 |
+
int(np.array(1.0, dtype=np.float64))
|
| 178 |
+
int(np.array("1", dtype=np.str_))
|
| 179 |
+
|
| 180 |
+
float(np.array(1.0, dtype=np.float64))
|
| 181 |
+
float(np.array("1", dtype=np.str_))
|
| 182 |
+
|
| 183 |
+
complex(np.array(1.0, dtype=np.float64))
|
| 184 |
+
|
| 185 |
+
operator.index(np.array(1, dtype=np.int64))
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_shape_manipulation.py
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import numpy as np
|
| 2 |
+
|
| 3 |
+
nd1 = np.array([[1, 2], [3, 4]])
|
| 4 |
+
|
| 5 |
+
# reshape
|
| 6 |
+
nd1.reshape(4)
|
| 7 |
+
nd1.reshape(2, 2)
|
| 8 |
+
nd1.reshape((2, 2))
|
| 9 |
+
|
| 10 |
+
nd1.reshape((2, 2), order="C")
|
| 11 |
+
nd1.reshape(4, order="C")
|
| 12 |
+
|
| 13 |
+
# resize
|
| 14 |
+
nd1.resize()
|
| 15 |
+
nd1.resize(4)
|
| 16 |
+
nd1.resize(2, 2)
|
| 17 |
+
nd1.resize((2, 2))
|
| 18 |
+
|
| 19 |
+
nd1.resize((2, 2), refcheck=True)
|
| 20 |
+
nd1.resize(4, refcheck=True)
|
| 21 |
+
|
| 22 |
+
nd2 = np.array([[1, 2], [3, 4]])
|
| 23 |
+
|
| 24 |
+
# transpose
|
| 25 |
+
nd2.transpose()
|
| 26 |
+
nd2.transpose(1, 0)
|
| 27 |
+
nd2.transpose((1, 0))
|
| 28 |
+
|
| 29 |
+
# swapaxes
|
| 30 |
+
nd2.swapaxes(0, 1)
|
| 31 |
+
|
| 32 |
+
# flatten
|
| 33 |
+
nd2.flatten()
|
| 34 |
+
nd2.flatten("C")
|
| 35 |
+
|
| 36 |
+
# ravel
|
| 37 |
+
nd2.ravel()
|
| 38 |
+
nd2.ravel("C")
|
| 39 |
+
|
| 40 |
+
# squeeze
|
| 41 |
+
nd2.squeeze()
|
| 42 |
+
|
| 43 |
+
nd3 = np.array([[1, 2]])
|
| 44 |
+
nd3.squeeze(0)
|
| 45 |
+
|
| 46 |
+
nd4 = np.array([[[1, 2]]])
|
| 47 |
+
nd4.squeeze((0, 1))
|
.venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/numeric.py
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Tests for :mod:`numpy.core.numeric`.
|
| 3 |
+
|
| 4 |
+
Does not include tests which fall under ``array_constructors``.
|
| 5 |
+
|
| 6 |
+
"""
|
| 7 |
+
|
| 8 |
+
from __future__ import annotations
|
| 9 |
+
|
| 10 |
+
import numpy as np
|
| 11 |
+
|
| 12 |
+
class SubClass(np.ndarray):
|
| 13 |
+
...
|
| 14 |
+
|
| 15 |
+
i8 = np.int64(1)
|
| 16 |
+
|
| 17 |
+
A = np.arange(27).reshape(3, 3, 3)
|
| 18 |
+
B: list[list[list[int]]] = A.tolist()
|
| 19 |
+
C = np.empty((27, 27)).view(SubClass)
|
| 20 |
+
|
| 21 |
+
np.count_nonzero(i8)
|
| 22 |
+
np.count_nonzero(A)
|
| 23 |
+
np.count_nonzero(B)
|
| 24 |
+
np.count_nonzero(A, keepdims=True)
|
| 25 |
+
np.count_nonzero(A, axis=0)
|
| 26 |
+
|
| 27 |
+
np.isfortran(i8)
|
| 28 |
+
np.isfortran(A)
|
| 29 |
+
|
| 30 |
+
np.argwhere(i8)
|
| 31 |
+
np.argwhere(A)
|
| 32 |
+
|
| 33 |
+
np.flatnonzero(i8)
|
| 34 |
+
np.flatnonzero(A)
|
| 35 |
+
|
| 36 |
+
np.correlate(B[0][0], A.ravel(), mode="valid")
|
| 37 |
+
np.correlate(A.ravel(), A.ravel(), mode="same")
|
| 38 |
+
|
| 39 |
+
np.convolve(B[0][0], A.ravel(), mode="valid")
|
| 40 |
+
np.convolve(A.ravel(), A.ravel(), mode="same")
|
| 41 |
+
|
| 42 |
+
np.outer(i8, A)
|
| 43 |
+
np.outer(B, A)
|
| 44 |
+
np.outer(A, A)
|
| 45 |
+
np.outer(A, A, out=C)
|
| 46 |
+
|
| 47 |
+
np.tensordot(B, A)
|
| 48 |
+
np.tensordot(A, A)
|
| 49 |
+
np.tensordot(A, A, axes=0)
|
| 50 |
+
np.tensordot(A, A, axes=(0, 1))
|
| 51 |
+
|
| 52 |
+
np.isscalar(i8)
|
| 53 |
+
np.isscalar(A)
|
| 54 |
+
np.isscalar(B)
|
| 55 |
+
|
| 56 |
+
np.roll(A, 1)
|
| 57 |
+
np.roll(A, (1, 2))
|
| 58 |
+
np.roll(B, 1)
|
| 59 |
+
|
| 60 |
+
np.rollaxis(A, 0, 1)
|
| 61 |
+
|
| 62 |
+
np.moveaxis(A, 0, 1)
|
| 63 |
+
np.moveaxis(A, (0, 1), (1, 2))
|
| 64 |
+
|
| 65 |
+
np.cross(B, A)
|
| 66 |
+
np.cross(A, A)
|
| 67 |
+
|
| 68 |
+
np.indices([0, 1, 2])
|
| 69 |
+
np.indices([0, 1, 2], sparse=False)
|
| 70 |
+
np.indices([0, 1, 2], sparse=True)
|
| 71 |
+
|
| 72 |
+
np.binary_repr(1)
|
| 73 |
+
|
| 74 |
+
np.base_repr(1)
|
| 75 |
+
|
| 76 |
+
np.allclose(i8, A)
|
| 77 |
+
np.allclose(B, A)
|
| 78 |
+
np.allclose(A, A)
|
| 79 |
+
|
| 80 |
+
np.isclose(i8, A)
|
| 81 |
+
np.isclose(B, A)
|
| 82 |
+
np.isclose(A, A)
|
| 83 |
+
|
| 84 |
+
np.array_equal(i8, A)
|
| 85 |
+
np.array_equal(B, A)
|
| 86 |
+
np.array_equal(A, A)
|
| 87 |
+
|
| 88 |
+
np.array_equiv(i8, A)
|
| 89 |
+
np.array_equiv(B, A)
|
| 90 |
+
np.array_equiv(A, A)
|