koichi12 commited on
Commit
ddb1ec8
·
verified ·
1 Parent(s): bffb9d4

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +3 -0
  2. .venv/lib/python3.11/site-packages/numpy/linalg/__pycache__/linalg.cpython-311.pyc +3 -0
  3. .venv/lib/python3.11/site-packages/numpy/linalg/_umath_linalg.cpython-311-x86_64-linux-gnu.so +3 -0
  4. .venv/lib/python3.11/site-packages/numpy/linalg/tests/__pycache__/test_linalg.cpython-311.pyc +3 -0
  5. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arithmetic.cpython-311.pyc +0 -0
  6. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_constructors.cpython-311.pyc +0 -0
  7. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/array_like.cpython-311.pyc +0 -0
  8. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayprint.cpython-311.pyc +0 -0
  9. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/arrayterator.cpython-311.pyc +0 -0
  10. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/bitwise_ops.cpython-311.pyc +0 -0
  11. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/comparisons.cpython-311.pyc +0 -0
  12. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/dtype.cpython-311.pyc +0 -0
  13. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/einsumfunc.cpython-311.pyc +0 -0
  14. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/flatiter.cpython-311.pyc +0 -0
  15. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/fromnumeric.cpython-311.pyc +0 -0
  16. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/index_tricks.cpython-311.pyc +0 -0
  17. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_utils.cpython-311.pyc +0 -0
  18. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/lib_version.cpython-311.pyc +0 -0
  19. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/literal.cpython-311.pyc +0 -0
  20. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/mod.cpython-311.pyc +0 -0
  21. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/modules.cpython-311.pyc +0 -0
  22. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/multiarray.cpython-311.pyc +0 -0
  23. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_conversion.cpython-311.pyc +0 -0
  24. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_misc.cpython-311.pyc +0 -0
  25. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ndarray_shape_manipulation.cpython-311.pyc +0 -0
  26. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numeric.cpython-311.pyc +0 -0
  27. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/numerictypes.cpython-311.pyc +0 -0
  28. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/scalars.cpython-311.pyc +0 -0
  29. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple.cpython-311.pyc +0 -0
  30. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/simple_py3.cpython-311.pyc +0 -0
  31. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunc_config.cpython-311.pyc +0 -0
  32. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufunclike.cpython-311.pyc +0 -0
  33. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/ufuncs.cpython-311.pyc +0 -0
  34. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/__pycache__/warnings_and_errors.cpython-311.pyc +0 -0
  35. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arithmetic.py +594 -0
  36. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/array_constructors.py +137 -0
  37. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/arrayterator.py +27 -0
  38. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
  39. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/comparisons.py +301 -0
  40. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/dtype.py +57 -0
  41. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/flatiter.py +16 -0
  42. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/index_tricks.py +64 -0
  43. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/lib_version.py +18 -0
  44. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/literal.py +47 -0
  45. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/mod.py +149 -0
  46. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/multiarray.py +76 -0
  47. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_conversion.py +94 -0
  48. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_misc.py +185 -0
  49. .venv/lib/python3.11/site-packages/numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
  50. .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)