numpy 2.4.1__cp314-cp314t-win_arm64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (932) hide show
  1. numpy/__config__.py +170 -0
  2. numpy/__config__.pyi +108 -0
  3. numpy/__init__.cython-30.pxd +1242 -0
  4. numpy/__init__.pxd +1155 -0
  5. numpy/__init__.py +955 -0
  6. numpy/__init__.pyi +6202 -0
  7. numpy/_array_api_info.py +346 -0
  8. numpy/_array_api_info.pyi +206 -0
  9. numpy/_configtool.py +39 -0
  10. numpy/_configtool.pyi +1 -0
  11. numpy/_core/__init__.py +201 -0
  12. numpy/_core/__init__.pyi +666 -0
  13. numpy/_core/_add_newdocs.py +7151 -0
  14. numpy/_core/_add_newdocs.pyi +2 -0
  15. numpy/_core/_add_newdocs_scalars.py +381 -0
  16. numpy/_core/_add_newdocs_scalars.pyi +16 -0
  17. numpy/_core/_asarray.py +130 -0
  18. numpy/_core/_asarray.pyi +43 -0
  19. numpy/_core/_dtype.py +366 -0
  20. numpy/_core/_dtype.pyi +56 -0
  21. numpy/_core/_dtype_ctypes.py +120 -0
  22. numpy/_core/_dtype_ctypes.pyi +83 -0
  23. numpy/_core/_exceptions.py +162 -0
  24. numpy/_core/_exceptions.pyi +54 -0
  25. numpy/_core/_internal.py +968 -0
  26. numpy/_core/_internal.pyi +61 -0
  27. numpy/_core/_methods.py +252 -0
  28. numpy/_core/_methods.pyi +22 -0
  29. numpy/_core/_multiarray_tests.cp314t-win_arm64.lib +0 -0
  30. numpy/_core/_multiarray_tests.cp314t-win_arm64.pyd +0 -0
  31. numpy/_core/_multiarray_umath.cp314t-win_arm64.lib +0 -0
  32. numpy/_core/_multiarray_umath.cp314t-win_arm64.pyd +0 -0
  33. numpy/_core/_operand_flag_tests.cp314t-win_arm64.lib +0 -0
  34. numpy/_core/_operand_flag_tests.cp314t-win_arm64.pyd +0 -0
  35. numpy/_core/_rational_tests.cp314t-win_arm64.lib +0 -0
  36. numpy/_core/_rational_tests.cp314t-win_arm64.pyd +0 -0
  37. numpy/_core/_simd.cp314t-win_arm64.lib +0 -0
  38. numpy/_core/_simd.cp314t-win_arm64.pyd +0 -0
  39. numpy/_core/_simd.pyi +35 -0
  40. numpy/_core/_string_helpers.py +100 -0
  41. numpy/_core/_string_helpers.pyi +12 -0
  42. numpy/_core/_struct_ufunc_tests.cp314t-win_arm64.lib +0 -0
  43. numpy/_core/_struct_ufunc_tests.cp314t-win_arm64.pyd +0 -0
  44. numpy/_core/_type_aliases.py +131 -0
  45. numpy/_core/_type_aliases.pyi +86 -0
  46. numpy/_core/_ufunc_config.py +515 -0
  47. numpy/_core/_ufunc_config.pyi +69 -0
  48. numpy/_core/_umath_tests.cp314t-win_arm64.lib +0 -0
  49. numpy/_core/_umath_tests.cp314t-win_arm64.pyd +0 -0
  50. numpy/_core/_umath_tests.pyi +47 -0
  51. numpy/_core/arrayprint.py +1779 -0
  52. numpy/_core/arrayprint.pyi +158 -0
  53. numpy/_core/cversions.py +13 -0
  54. numpy/_core/defchararray.py +1414 -0
  55. numpy/_core/defchararray.pyi +1150 -0
  56. numpy/_core/einsumfunc.py +1650 -0
  57. numpy/_core/einsumfunc.pyi +184 -0
  58. numpy/_core/fromnumeric.py +4233 -0
  59. numpy/_core/fromnumeric.pyi +1735 -0
  60. numpy/_core/function_base.py +547 -0
  61. numpy/_core/function_base.pyi +276 -0
  62. numpy/_core/getlimits.py +462 -0
  63. numpy/_core/getlimits.pyi +124 -0
  64. numpy/_core/include/numpy/__multiarray_api.c +376 -0
  65. numpy/_core/include/numpy/__multiarray_api.h +1628 -0
  66. numpy/_core/include/numpy/__ufunc_api.c +55 -0
  67. numpy/_core/include/numpy/__ufunc_api.h +349 -0
  68. numpy/_core/include/numpy/_neighborhood_iterator_imp.h +90 -0
  69. numpy/_core/include/numpy/_numpyconfig.h +33 -0
  70. numpy/_core/include/numpy/_public_dtype_api_table.h +86 -0
  71. numpy/_core/include/numpy/arrayobject.h +7 -0
  72. numpy/_core/include/numpy/arrayscalars.h +198 -0
  73. numpy/_core/include/numpy/dtype_api.h +547 -0
  74. numpy/_core/include/numpy/halffloat.h +70 -0
  75. numpy/_core/include/numpy/ndarrayobject.h +304 -0
  76. numpy/_core/include/numpy/ndarraytypes.h +1982 -0
  77. numpy/_core/include/numpy/npy_2_compat.h +249 -0
  78. numpy/_core/include/numpy/npy_2_complexcompat.h +28 -0
  79. numpy/_core/include/numpy/npy_3kcompat.h +374 -0
  80. numpy/_core/include/numpy/npy_common.h +989 -0
  81. numpy/_core/include/numpy/npy_cpu.h +126 -0
  82. numpy/_core/include/numpy/npy_endian.h +79 -0
  83. numpy/_core/include/numpy/npy_math.h +602 -0
  84. numpy/_core/include/numpy/npy_no_deprecated_api.h +20 -0
  85. numpy/_core/include/numpy/npy_os.h +42 -0
  86. numpy/_core/include/numpy/numpyconfig.h +185 -0
  87. numpy/_core/include/numpy/random/LICENSE.txt +21 -0
  88. numpy/_core/include/numpy/random/bitgen.h +20 -0
  89. numpy/_core/include/numpy/random/distributions.h +209 -0
  90. numpy/_core/include/numpy/random/libdivide.h +2079 -0
  91. numpy/_core/include/numpy/ufuncobject.h +343 -0
  92. numpy/_core/include/numpy/utils.h +37 -0
  93. numpy/_core/lib/npy-pkg-config/mlib.ini +12 -0
  94. numpy/_core/lib/npy-pkg-config/npymath.ini +20 -0
  95. numpy/_core/lib/npymath.lib +0 -0
  96. numpy/_core/lib/pkgconfig/numpy.pc +7 -0
  97. numpy/_core/memmap.py +363 -0
  98. numpy/_core/memmap.pyi +3 -0
  99. numpy/_core/multiarray.py +1740 -0
  100. numpy/_core/multiarray.pyi +1316 -0
  101. numpy/_core/numeric.py +2758 -0
  102. numpy/_core/numeric.pyi +1276 -0
  103. numpy/_core/numerictypes.py +633 -0
  104. numpy/_core/numerictypes.pyi +196 -0
  105. numpy/_core/overrides.py +188 -0
  106. numpy/_core/overrides.pyi +47 -0
  107. numpy/_core/printoptions.py +32 -0
  108. numpy/_core/printoptions.pyi +28 -0
  109. numpy/_core/records.py +1088 -0
  110. numpy/_core/records.pyi +340 -0
  111. numpy/_core/shape_base.py +996 -0
  112. numpy/_core/shape_base.pyi +182 -0
  113. numpy/_core/strings.py +1813 -0
  114. numpy/_core/strings.pyi +536 -0
  115. numpy/_core/tests/_locales.py +72 -0
  116. numpy/_core/tests/_natype.py +144 -0
  117. numpy/_core/tests/data/astype_copy.pkl +0 -0
  118. numpy/_core/tests/data/generate_umath_validation_data.cpp +170 -0
  119. numpy/_core/tests/data/recarray_from_file.fits +0 -0
  120. numpy/_core/tests/data/umath-validation-set-README.txt +15 -0
  121. numpy/_core/tests/data/umath-validation-set-arccos.csv +1429 -0
  122. numpy/_core/tests/data/umath-validation-set-arccosh.csv +1429 -0
  123. numpy/_core/tests/data/umath-validation-set-arcsin.csv +1429 -0
  124. numpy/_core/tests/data/umath-validation-set-arcsinh.csv +1429 -0
  125. numpy/_core/tests/data/umath-validation-set-arctan.csv +1429 -0
  126. numpy/_core/tests/data/umath-validation-set-arctanh.csv +1429 -0
  127. numpy/_core/tests/data/umath-validation-set-cbrt.csv +1429 -0
  128. numpy/_core/tests/data/umath-validation-set-cos.csv +1375 -0
  129. numpy/_core/tests/data/umath-validation-set-cosh.csv +1429 -0
  130. numpy/_core/tests/data/umath-validation-set-exp.csv +412 -0
  131. numpy/_core/tests/data/umath-validation-set-exp2.csv +1429 -0
  132. numpy/_core/tests/data/umath-validation-set-expm1.csv +1429 -0
  133. numpy/_core/tests/data/umath-validation-set-log.csv +271 -0
  134. numpy/_core/tests/data/umath-validation-set-log10.csv +1629 -0
  135. numpy/_core/tests/data/umath-validation-set-log1p.csv +1429 -0
  136. numpy/_core/tests/data/umath-validation-set-log2.csv +1629 -0
  137. numpy/_core/tests/data/umath-validation-set-sin.csv +1370 -0
  138. numpy/_core/tests/data/umath-validation-set-sinh.csv +1429 -0
  139. numpy/_core/tests/data/umath-validation-set-tan.csv +1429 -0
  140. numpy/_core/tests/data/umath-validation-set-tanh.csv +1429 -0
  141. numpy/_core/tests/examples/cython/checks.pyx +373 -0
  142. numpy/_core/tests/examples/cython/meson.build +43 -0
  143. numpy/_core/tests/examples/cython/setup.py +39 -0
  144. numpy/_core/tests/examples/limited_api/limited_api1.c +17 -0
  145. numpy/_core/tests/examples/limited_api/limited_api2.pyx +11 -0
  146. numpy/_core/tests/examples/limited_api/limited_api_latest.c +19 -0
  147. numpy/_core/tests/examples/limited_api/meson.build +59 -0
  148. numpy/_core/tests/examples/limited_api/setup.py +24 -0
  149. numpy/_core/tests/test__exceptions.py +90 -0
  150. numpy/_core/tests/test_abc.py +54 -0
  151. numpy/_core/tests/test_api.py +655 -0
  152. numpy/_core/tests/test_argparse.py +90 -0
  153. numpy/_core/tests/test_array_api_info.py +113 -0
  154. numpy/_core/tests/test_array_coercion.py +928 -0
  155. numpy/_core/tests/test_array_interface.py +222 -0
  156. numpy/_core/tests/test_arraymethod.py +84 -0
  157. numpy/_core/tests/test_arrayobject.py +75 -0
  158. numpy/_core/tests/test_arrayprint.py +1324 -0
  159. numpy/_core/tests/test_casting_floatingpoint_errors.py +154 -0
  160. numpy/_core/tests/test_casting_unittests.py +955 -0
  161. numpy/_core/tests/test_conversion_utils.py +209 -0
  162. numpy/_core/tests/test_cpu_dispatcher.py +48 -0
  163. numpy/_core/tests/test_cpu_features.py +450 -0
  164. numpy/_core/tests/test_custom_dtypes.py +393 -0
  165. numpy/_core/tests/test_cython.py +352 -0
  166. numpy/_core/tests/test_datetime.py +2792 -0
  167. numpy/_core/tests/test_defchararray.py +858 -0
  168. numpy/_core/tests/test_deprecations.py +460 -0
  169. numpy/_core/tests/test_dlpack.py +190 -0
  170. numpy/_core/tests/test_dtype.py +2110 -0
  171. numpy/_core/tests/test_einsum.py +1351 -0
  172. numpy/_core/tests/test_errstate.py +131 -0
  173. numpy/_core/tests/test_extint128.py +217 -0
  174. numpy/_core/tests/test_finfo.py +86 -0
  175. numpy/_core/tests/test_function_base.py +504 -0
  176. numpy/_core/tests/test_getlimits.py +171 -0
  177. numpy/_core/tests/test_half.py +593 -0
  178. numpy/_core/tests/test_hashtable.py +36 -0
  179. numpy/_core/tests/test_indexerrors.py +122 -0
  180. numpy/_core/tests/test_indexing.py +1692 -0
  181. numpy/_core/tests/test_item_selection.py +167 -0
  182. numpy/_core/tests/test_limited_api.py +102 -0
  183. numpy/_core/tests/test_longdouble.py +370 -0
  184. numpy/_core/tests/test_mem_overlap.py +933 -0
  185. numpy/_core/tests/test_mem_policy.py +453 -0
  186. numpy/_core/tests/test_memmap.py +248 -0
  187. numpy/_core/tests/test_multiarray.py +11008 -0
  188. numpy/_core/tests/test_multiprocessing.py +55 -0
  189. numpy/_core/tests/test_multithreading.py +377 -0
  190. numpy/_core/tests/test_nditer.py +3533 -0
  191. numpy/_core/tests/test_nep50_promotions.py +287 -0
  192. numpy/_core/tests/test_numeric.py +4295 -0
  193. numpy/_core/tests/test_numerictypes.py +650 -0
  194. numpy/_core/tests/test_overrides.py +800 -0
  195. numpy/_core/tests/test_print.py +202 -0
  196. numpy/_core/tests/test_protocols.py +46 -0
  197. numpy/_core/tests/test_records.py +544 -0
  198. numpy/_core/tests/test_regression.py +2677 -0
  199. numpy/_core/tests/test_scalar_ctors.py +203 -0
  200. numpy/_core/tests/test_scalar_methods.py +328 -0
  201. numpy/_core/tests/test_scalarbuffer.py +153 -0
  202. numpy/_core/tests/test_scalarinherit.py +105 -0
  203. numpy/_core/tests/test_scalarmath.py +1168 -0
  204. numpy/_core/tests/test_scalarprint.py +403 -0
  205. numpy/_core/tests/test_shape_base.py +904 -0
  206. numpy/_core/tests/test_simd.py +1345 -0
  207. numpy/_core/tests/test_simd_module.py +105 -0
  208. numpy/_core/tests/test_stringdtype.py +1855 -0
  209. numpy/_core/tests/test_strings.py +1523 -0
  210. numpy/_core/tests/test_ufunc.py +3405 -0
  211. numpy/_core/tests/test_umath.py +4962 -0
  212. numpy/_core/tests/test_umath_accuracy.py +132 -0
  213. numpy/_core/tests/test_umath_complex.py +631 -0
  214. numpy/_core/tests/test_unicode.py +369 -0
  215. numpy/_core/umath.py +60 -0
  216. numpy/_core/umath.pyi +232 -0
  217. numpy/_distributor_init.py +15 -0
  218. numpy/_distributor_init.pyi +1 -0
  219. numpy/_expired_attrs_2_0.py +78 -0
  220. numpy/_expired_attrs_2_0.pyi +61 -0
  221. numpy/_globals.py +121 -0
  222. numpy/_globals.pyi +17 -0
  223. numpy/_pyinstaller/__init__.py +0 -0
  224. numpy/_pyinstaller/__init__.pyi +0 -0
  225. numpy/_pyinstaller/hook-numpy.py +36 -0
  226. numpy/_pyinstaller/hook-numpy.pyi +6 -0
  227. numpy/_pyinstaller/tests/__init__.py +16 -0
  228. numpy/_pyinstaller/tests/pyinstaller-smoke.py +32 -0
  229. numpy/_pyinstaller/tests/test_pyinstaller.py +35 -0
  230. numpy/_pytesttester.py +201 -0
  231. numpy/_pytesttester.pyi +18 -0
  232. numpy/_typing/__init__.py +173 -0
  233. numpy/_typing/_add_docstring.py +153 -0
  234. numpy/_typing/_array_like.py +106 -0
  235. numpy/_typing/_char_codes.py +213 -0
  236. numpy/_typing/_dtype_like.py +114 -0
  237. numpy/_typing/_extended_precision.py +15 -0
  238. numpy/_typing/_nbit.py +19 -0
  239. numpy/_typing/_nbit_base.py +94 -0
  240. numpy/_typing/_nbit_base.pyi +39 -0
  241. numpy/_typing/_nested_sequence.py +79 -0
  242. numpy/_typing/_scalars.py +20 -0
  243. numpy/_typing/_shape.py +8 -0
  244. numpy/_typing/_ufunc.py +7 -0
  245. numpy/_typing/_ufunc.pyi +975 -0
  246. numpy/_utils/__init__.py +95 -0
  247. numpy/_utils/__init__.pyi +28 -0
  248. numpy/_utils/_convertions.py +18 -0
  249. numpy/_utils/_convertions.pyi +4 -0
  250. numpy/_utils/_inspect.py +192 -0
  251. numpy/_utils/_inspect.pyi +70 -0
  252. numpy/_utils/_pep440.py +486 -0
  253. numpy/_utils/_pep440.pyi +118 -0
  254. numpy/char/__init__.py +2 -0
  255. numpy/char/__init__.pyi +111 -0
  256. numpy/conftest.py +248 -0
  257. numpy/core/__init__.py +33 -0
  258. numpy/core/__init__.pyi +0 -0
  259. numpy/core/_dtype.py +10 -0
  260. numpy/core/_dtype.pyi +0 -0
  261. numpy/core/_dtype_ctypes.py +10 -0
  262. numpy/core/_dtype_ctypes.pyi +0 -0
  263. numpy/core/_internal.py +27 -0
  264. numpy/core/_multiarray_umath.py +57 -0
  265. numpy/core/_utils.py +21 -0
  266. numpy/core/arrayprint.py +10 -0
  267. numpy/core/defchararray.py +10 -0
  268. numpy/core/einsumfunc.py +10 -0
  269. numpy/core/fromnumeric.py +10 -0
  270. numpy/core/function_base.py +10 -0
  271. numpy/core/getlimits.py +10 -0
  272. numpy/core/multiarray.py +25 -0
  273. numpy/core/numeric.py +12 -0
  274. numpy/core/numerictypes.py +10 -0
  275. numpy/core/overrides.py +10 -0
  276. numpy/core/overrides.pyi +7 -0
  277. numpy/core/records.py +10 -0
  278. numpy/core/shape_base.py +10 -0
  279. numpy/core/umath.py +10 -0
  280. numpy/ctypeslib/__init__.py +13 -0
  281. numpy/ctypeslib/__init__.pyi +15 -0
  282. numpy/ctypeslib/_ctypeslib.py +603 -0
  283. numpy/ctypeslib/_ctypeslib.pyi +236 -0
  284. numpy/doc/ufuncs.py +138 -0
  285. numpy/dtypes.py +41 -0
  286. numpy/dtypes.pyi +630 -0
  287. numpy/exceptions.py +246 -0
  288. numpy/exceptions.pyi +27 -0
  289. numpy/f2py/__init__.py +86 -0
  290. numpy/f2py/__init__.pyi +5 -0
  291. numpy/f2py/__main__.py +5 -0
  292. numpy/f2py/__version__.py +1 -0
  293. numpy/f2py/__version__.pyi +1 -0
  294. numpy/f2py/_backends/__init__.py +9 -0
  295. numpy/f2py/_backends/__init__.pyi +5 -0
  296. numpy/f2py/_backends/_backend.py +44 -0
  297. numpy/f2py/_backends/_backend.pyi +46 -0
  298. numpy/f2py/_backends/_distutils.py +76 -0
  299. numpy/f2py/_backends/_distutils.pyi +13 -0
  300. numpy/f2py/_backends/_meson.py +244 -0
  301. numpy/f2py/_backends/_meson.pyi +62 -0
  302. numpy/f2py/_backends/meson.build.template +58 -0
  303. numpy/f2py/_isocbind.py +62 -0
  304. numpy/f2py/_isocbind.pyi +13 -0
  305. numpy/f2py/_src_pyf.py +247 -0
  306. numpy/f2py/_src_pyf.pyi +28 -0
  307. numpy/f2py/auxfuncs.py +1004 -0
  308. numpy/f2py/auxfuncs.pyi +262 -0
  309. numpy/f2py/capi_maps.py +811 -0
  310. numpy/f2py/capi_maps.pyi +33 -0
  311. numpy/f2py/cb_rules.py +665 -0
  312. numpy/f2py/cb_rules.pyi +17 -0
  313. numpy/f2py/cfuncs.py +1563 -0
  314. numpy/f2py/cfuncs.pyi +31 -0
  315. numpy/f2py/common_rules.py +143 -0
  316. numpy/f2py/common_rules.pyi +9 -0
  317. numpy/f2py/crackfortran.py +3725 -0
  318. numpy/f2py/crackfortran.pyi +266 -0
  319. numpy/f2py/diagnose.py +149 -0
  320. numpy/f2py/diagnose.pyi +1 -0
  321. numpy/f2py/f2py2e.py +788 -0
  322. numpy/f2py/f2py2e.pyi +74 -0
  323. numpy/f2py/f90mod_rules.py +269 -0
  324. numpy/f2py/f90mod_rules.pyi +16 -0
  325. numpy/f2py/func2subr.py +329 -0
  326. numpy/f2py/func2subr.pyi +7 -0
  327. numpy/f2py/rules.py +1629 -0
  328. numpy/f2py/rules.pyi +41 -0
  329. numpy/f2py/setup.cfg +3 -0
  330. numpy/f2py/src/fortranobject.c +1436 -0
  331. numpy/f2py/src/fortranobject.h +173 -0
  332. numpy/f2py/symbolic.py +1518 -0
  333. numpy/f2py/symbolic.pyi +219 -0
  334. numpy/f2py/tests/__init__.py +16 -0
  335. numpy/f2py/tests/src/abstract_interface/foo.f90 +34 -0
  336. numpy/f2py/tests/src/abstract_interface/gh18403_mod.f90 +6 -0
  337. numpy/f2py/tests/src/array_from_pyobj/wrapmodule.c +235 -0
  338. numpy/f2py/tests/src/assumed_shape/.f2py_f2cmap +1 -0
  339. numpy/f2py/tests/src/assumed_shape/foo_free.f90 +34 -0
  340. numpy/f2py/tests/src/assumed_shape/foo_mod.f90 +41 -0
  341. numpy/f2py/tests/src/assumed_shape/foo_use.f90 +19 -0
  342. numpy/f2py/tests/src/assumed_shape/precision.f90 +4 -0
  343. numpy/f2py/tests/src/block_docstring/foo.f +6 -0
  344. numpy/f2py/tests/src/callback/foo.f +62 -0
  345. numpy/f2py/tests/src/callback/gh17797.f90 +7 -0
  346. numpy/f2py/tests/src/callback/gh18335.f90 +17 -0
  347. numpy/f2py/tests/src/callback/gh25211.f +10 -0
  348. numpy/f2py/tests/src/callback/gh25211.pyf +18 -0
  349. numpy/f2py/tests/src/callback/gh26681.f90 +18 -0
  350. numpy/f2py/tests/src/cli/gh_22819.pyf +6 -0
  351. numpy/f2py/tests/src/cli/hi77.f +3 -0
  352. numpy/f2py/tests/src/cli/hiworld.f90 +3 -0
  353. numpy/f2py/tests/src/common/block.f +11 -0
  354. numpy/f2py/tests/src/common/gh19161.f90 +10 -0
  355. numpy/f2py/tests/src/crackfortran/accesstype.f90 +13 -0
  356. numpy/f2py/tests/src/crackfortran/common_with_division.f +17 -0
  357. numpy/f2py/tests/src/crackfortran/data_common.f +8 -0
  358. numpy/f2py/tests/src/crackfortran/data_multiplier.f +5 -0
  359. numpy/f2py/tests/src/crackfortran/data_stmts.f90 +20 -0
  360. numpy/f2py/tests/src/crackfortran/data_with_comments.f +8 -0
  361. numpy/f2py/tests/src/crackfortran/foo_deps.f90 +6 -0
  362. numpy/f2py/tests/src/crackfortran/gh15035.f +16 -0
  363. numpy/f2py/tests/src/crackfortran/gh17859.f +12 -0
  364. numpy/f2py/tests/src/crackfortran/gh22648.pyf +7 -0
  365. numpy/f2py/tests/src/crackfortran/gh23533.f +5 -0
  366. numpy/f2py/tests/src/crackfortran/gh23598.f90 +4 -0
  367. numpy/f2py/tests/src/crackfortran/gh23598Warn.f90 +11 -0
  368. numpy/f2py/tests/src/crackfortran/gh23879.f90 +20 -0
  369. numpy/f2py/tests/src/crackfortran/gh27697.f90 +12 -0
  370. numpy/f2py/tests/src/crackfortran/gh2848.f90 +13 -0
  371. numpy/f2py/tests/src/crackfortran/operators.f90 +49 -0
  372. numpy/f2py/tests/src/crackfortran/privatemod.f90 +11 -0
  373. numpy/f2py/tests/src/crackfortran/publicmod.f90 +10 -0
  374. numpy/f2py/tests/src/crackfortran/pubprivmod.f90 +10 -0
  375. numpy/f2py/tests/src/crackfortran/unicode_comment.f90 +4 -0
  376. numpy/f2py/tests/src/f2cmap/.f2py_f2cmap +1 -0
  377. numpy/f2py/tests/src/f2cmap/isoFortranEnvMap.f90 +9 -0
  378. numpy/f2py/tests/src/isocintrin/isoCtests.f90 +34 -0
  379. numpy/f2py/tests/src/kind/foo.f90 +20 -0
  380. numpy/f2py/tests/src/mixed/foo.f +5 -0
  381. numpy/f2py/tests/src/mixed/foo_fixed.f90 +8 -0
  382. numpy/f2py/tests/src/mixed/foo_free.f90 +8 -0
  383. numpy/f2py/tests/src/modules/gh25337/data.f90 +8 -0
  384. numpy/f2py/tests/src/modules/gh25337/use_data.f90 +6 -0
  385. numpy/f2py/tests/src/modules/gh26920/two_mods_with_no_public_entities.f90 +21 -0
  386. numpy/f2py/tests/src/modules/gh26920/two_mods_with_one_public_routine.f90 +21 -0
  387. numpy/f2py/tests/src/modules/module_data_docstring.f90 +12 -0
  388. numpy/f2py/tests/src/modules/use_modules.f90 +20 -0
  389. numpy/f2py/tests/src/negative_bounds/issue_20853.f90 +7 -0
  390. numpy/f2py/tests/src/parameter/constant_array.f90 +45 -0
  391. numpy/f2py/tests/src/parameter/constant_both.f90 +57 -0
  392. numpy/f2py/tests/src/parameter/constant_compound.f90 +15 -0
  393. numpy/f2py/tests/src/parameter/constant_integer.f90 +22 -0
  394. numpy/f2py/tests/src/parameter/constant_non_compound.f90 +23 -0
  395. numpy/f2py/tests/src/parameter/constant_real.f90 +23 -0
  396. numpy/f2py/tests/src/quoted_character/foo.f +14 -0
  397. numpy/f2py/tests/src/regression/AB.inc +1 -0
  398. numpy/f2py/tests/src/regression/assignOnlyModule.f90 +25 -0
  399. numpy/f2py/tests/src/regression/datonly.f90 +17 -0
  400. numpy/f2py/tests/src/regression/f77comments.f +26 -0
  401. numpy/f2py/tests/src/regression/f77fixedform.f95 +5 -0
  402. numpy/f2py/tests/src/regression/f90continuation.f90 +9 -0
  403. numpy/f2py/tests/src/regression/incfile.f90 +5 -0
  404. numpy/f2py/tests/src/regression/inout.f90 +9 -0
  405. numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 +5 -0
  406. numpy/f2py/tests/src/regression/mod_derived_types.f90 +23 -0
  407. numpy/f2py/tests/src/return_character/foo77.f +45 -0
  408. numpy/f2py/tests/src/return_character/foo90.f90 +48 -0
  409. numpy/f2py/tests/src/return_complex/foo77.f +45 -0
  410. numpy/f2py/tests/src/return_complex/foo90.f90 +48 -0
  411. numpy/f2py/tests/src/return_integer/foo77.f +56 -0
  412. numpy/f2py/tests/src/return_integer/foo90.f90 +59 -0
  413. numpy/f2py/tests/src/return_logical/foo77.f +56 -0
  414. numpy/f2py/tests/src/return_logical/foo90.f90 +59 -0
  415. numpy/f2py/tests/src/return_real/foo77.f +45 -0
  416. numpy/f2py/tests/src/return_real/foo90.f90 +48 -0
  417. numpy/f2py/tests/src/routines/funcfortranname.f +5 -0
  418. numpy/f2py/tests/src/routines/funcfortranname.pyf +11 -0
  419. numpy/f2py/tests/src/routines/subrout.f +4 -0
  420. numpy/f2py/tests/src/routines/subrout.pyf +10 -0
  421. numpy/f2py/tests/src/size/foo.f90 +44 -0
  422. numpy/f2py/tests/src/string/char.f90 +29 -0
  423. numpy/f2py/tests/src/string/fixed_string.f90 +34 -0
  424. numpy/f2py/tests/src/string/gh24008.f +8 -0
  425. numpy/f2py/tests/src/string/gh24662.f90 +7 -0
  426. numpy/f2py/tests/src/string/gh25286.f90 +14 -0
  427. numpy/f2py/tests/src/string/gh25286.pyf +12 -0
  428. numpy/f2py/tests/src/string/gh25286_bc.pyf +12 -0
  429. numpy/f2py/tests/src/string/scalar_string.f90 +9 -0
  430. numpy/f2py/tests/src/string/string.f +12 -0
  431. numpy/f2py/tests/src/value_attrspec/gh21665.f90 +9 -0
  432. numpy/f2py/tests/test_abstract_interface.py +26 -0
  433. numpy/f2py/tests/test_array_from_pyobj.py +678 -0
  434. numpy/f2py/tests/test_assumed_shape.py +50 -0
  435. numpy/f2py/tests/test_block_docstring.py +20 -0
  436. numpy/f2py/tests/test_callback.py +263 -0
  437. numpy/f2py/tests/test_character.py +641 -0
  438. numpy/f2py/tests/test_common.py +23 -0
  439. numpy/f2py/tests/test_crackfortran.py +421 -0
  440. numpy/f2py/tests/test_data.py +71 -0
  441. numpy/f2py/tests/test_docs.py +66 -0
  442. numpy/f2py/tests/test_f2cmap.py +17 -0
  443. numpy/f2py/tests/test_f2py2e.py +983 -0
  444. numpy/f2py/tests/test_isoc.py +56 -0
  445. numpy/f2py/tests/test_kind.py +52 -0
  446. numpy/f2py/tests/test_mixed.py +35 -0
  447. numpy/f2py/tests/test_modules.py +83 -0
  448. numpy/f2py/tests/test_parameter.py +129 -0
  449. numpy/f2py/tests/test_pyf_src.py +43 -0
  450. numpy/f2py/tests/test_quoted_character.py +18 -0
  451. numpy/f2py/tests/test_regression.py +187 -0
  452. numpy/f2py/tests/test_return_character.py +48 -0
  453. numpy/f2py/tests/test_return_complex.py +67 -0
  454. numpy/f2py/tests/test_return_integer.py +55 -0
  455. numpy/f2py/tests/test_return_logical.py +65 -0
  456. numpy/f2py/tests/test_return_real.py +109 -0
  457. numpy/f2py/tests/test_routines.py +29 -0
  458. numpy/f2py/tests/test_semicolon_split.py +75 -0
  459. numpy/f2py/tests/test_size.py +45 -0
  460. numpy/f2py/tests/test_string.py +100 -0
  461. numpy/f2py/tests/test_symbolic.py +500 -0
  462. numpy/f2py/tests/test_value_attrspec.py +15 -0
  463. numpy/f2py/tests/util.py +442 -0
  464. numpy/f2py/use_rules.py +99 -0
  465. numpy/f2py/use_rules.pyi +9 -0
  466. numpy/fft/__init__.py +213 -0
  467. numpy/fft/__init__.pyi +38 -0
  468. numpy/fft/_helper.py +235 -0
  469. numpy/fft/_helper.pyi +44 -0
  470. numpy/fft/_pocketfft.py +1693 -0
  471. numpy/fft/_pocketfft.pyi +137 -0
  472. numpy/fft/_pocketfft_umath.cp314t-win_arm64.lib +0 -0
  473. numpy/fft/_pocketfft_umath.cp314t-win_arm64.pyd +0 -0
  474. numpy/fft/tests/__init__.py +0 -0
  475. numpy/fft/tests/test_helper.py +167 -0
  476. numpy/fft/tests/test_pocketfft.py +589 -0
  477. numpy/lib/__init__.py +97 -0
  478. numpy/lib/__init__.pyi +52 -0
  479. numpy/lib/_array_utils_impl.py +62 -0
  480. numpy/lib/_array_utils_impl.pyi +10 -0
  481. numpy/lib/_arraypad_impl.py +926 -0
  482. numpy/lib/_arraypad_impl.pyi +88 -0
  483. numpy/lib/_arraysetops_impl.py +1158 -0
  484. numpy/lib/_arraysetops_impl.pyi +462 -0
  485. numpy/lib/_arrayterator_impl.py +224 -0
  486. numpy/lib/_arrayterator_impl.pyi +45 -0
  487. numpy/lib/_datasource.py +700 -0
  488. numpy/lib/_datasource.pyi +30 -0
  489. numpy/lib/_format_impl.py +1036 -0
  490. numpy/lib/_format_impl.pyi +56 -0
  491. numpy/lib/_function_base_impl.py +5760 -0
  492. numpy/lib/_function_base_impl.pyi +2324 -0
  493. numpy/lib/_histograms_impl.py +1085 -0
  494. numpy/lib/_histograms_impl.pyi +40 -0
  495. numpy/lib/_index_tricks_impl.py +1048 -0
  496. numpy/lib/_index_tricks_impl.pyi +267 -0
  497. numpy/lib/_iotools.py +900 -0
  498. numpy/lib/_iotools.pyi +116 -0
  499. numpy/lib/_nanfunctions_impl.py +2006 -0
  500. numpy/lib/_nanfunctions_impl.pyi +48 -0
  501. numpy/lib/_npyio_impl.py +2583 -0
  502. numpy/lib/_npyio_impl.pyi +299 -0
  503. numpy/lib/_polynomial_impl.py +1465 -0
  504. numpy/lib/_polynomial_impl.pyi +338 -0
  505. numpy/lib/_scimath_impl.py +642 -0
  506. numpy/lib/_scimath_impl.pyi +93 -0
  507. numpy/lib/_shape_base_impl.py +1289 -0
  508. numpy/lib/_shape_base_impl.pyi +236 -0
  509. numpy/lib/_stride_tricks_impl.py +582 -0
  510. numpy/lib/_stride_tricks_impl.pyi +73 -0
  511. numpy/lib/_twodim_base_impl.py +1201 -0
  512. numpy/lib/_twodim_base_impl.pyi +408 -0
  513. numpy/lib/_type_check_impl.py +710 -0
  514. numpy/lib/_type_check_impl.pyi +348 -0
  515. numpy/lib/_ufunclike_impl.py +199 -0
  516. numpy/lib/_ufunclike_impl.pyi +60 -0
  517. numpy/lib/_user_array_impl.py +310 -0
  518. numpy/lib/_user_array_impl.pyi +226 -0
  519. numpy/lib/_utils_impl.py +784 -0
  520. numpy/lib/_utils_impl.pyi +22 -0
  521. numpy/lib/_version.py +153 -0
  522. numpy/lib/_version.pyi +17 -0
  523. numpy/lib/array_utils.py +7 -0
  524. numpy/lib/array_utils.pyi +6 -0
  525. numpy/lib/format.py +24 -0
  526. numpy/lib/format.pyi +24 -0
  527. numpy/lib/introspect.py +94 -0
  528. numpy/lib/introspect.pyi +3 -0
  529. numpy/lib/mixins.py +180 -0
  530. numpy/lib/mixins.pyi +78 -0
  531. numpy/lib/npyio.py +1 -0
  532. numpy/lib/npyio.pyi +5 -0
  533. numpy/lib/recfunctions.py +1681 -0
  534. numpy/lib/recfunctions.pyi +444 -0
  535. numpy/lib/scimath.py +13 -0
  536. numpy/lib/scimath.pyi +12 -0
  537. numpy/lib/stride_tricks.py +1 -0
  538. numpy/lib/stride_tricks.pyi +4 -0
  539. numpy/lib/tests/__init__.py +0 -0
  540. numpy/lib/tests/data/py2-np0-objarr.npy +0 -0
  541. numpy/lib/tests/data/py2-objarr.npy +0 -0
  542. numpy/lib/tests/data/py2-objarr.npz +0 -0
  543. numpy/lib/tests/data/py3-objarr.npy +0 -0
  544. numpy/lib/tests/data/py3-objarr.npz +0 -0
  545. numpy/lib/tests/data/python3.npy +0 -0
  546. numpy/lib/tests/data/win64python2.npy +0 -0
  547. numpy/lib/tests/test__datasource.py +328 -0
  548. numpy/lib/tests/test__iotools.py +358 -0
  549. numpy/lib/tests/test__version.py +64 -0
  550. numpy/lib/tests/test_array_utils.py +32 -0
  551. numpy/lib/tests/test_arraypad.py +1427 -0
  552. numpy/lib/tests/test_arraysetops.py +1302 -0
  553. numpy/lib/tests/test_arrayterator.py +45 -0
  554. numpy/lib/tests/test_format.py +1054 -0
  555. numpy/lib/tests/test_function_base.py +4750 -0
  556. numpy/lib/tests/test_histograms.py +855 -0
  557. numpy/lib/tests/test_index_tricks.py +693 -0
  558. numpy/lib/tests/test_io.py +2857 -0
  559. numpy/lib/tests/test_loadtxt.py +1099 -0
  560. numpy/lib/tests/test_mixins.py +215 -0
  561. numpy/lib/tests/test_nanfunctions.py +1438 -0
  562. numpy/lib/tests/test_packbits.py +376 -0
  563. numpy/lib/tests/test_polynomial.py +325 -0
  564. numpy/lib/tests/test_recfunctions.py +1042 -0
  565. numpy/lib/tests/test_regression.py +231 -0
  566. numpy/lib/tests/test_shape_base.py +813 -0
  567. numpy/lib/tests/test_stride_tricks.py +655 -0
  568. numpy/lib/tests/test_twodim_base.py +559 -0
  569. numpy/lib/tests/test_type_check.py +473 -0
  570. numpy/lib/tests/test_ufunclike.py +97 -0
  571. numpy/lib/tests/test_utils.py +80 -0
  572. numpy/lib/user_array.py +1 -0
  573. numpy/lib/user_array.pyi +1 -0
  574. numpy/linalg/__init__.py +95 -0
  575. numpy/linalg/__init__.pyi +71 -0
  576. numpy/linalg/_linalg.py +3657 -0
  577. numpy/linalg/_linalg.pyi +548 -0
  578. numpy/linalg/_umath_linalg.cp314t-win_arm64.lib +0 -0
  579. numpy/linalg/_umath_linalg.cp314t-win_arm64.pyd +0 -0
  580. numpy/linalg/_umath_linalg.pyi +60 -0
  581. numpy/linalg/lapack_lite.cp314t-win_arm64.lib +0 -0
  582. numpy/linalg/lapack_lite.cp314t-win_arm64.pyd +0 -0
  583. numpy/linalg/lapack_lite.pyi +143 -0
  584. numpy/linalg/tests/__init__.py +0 -0
  585. numpy/linalg/tests/test_deprecations.py +21 -0
  586. numpy/linalg/tests/test_linalg.py +2442 -0
  587. numpy/linalg/tests/test_regression.py +182 -0
  588. numpy/ma/API_CHANGES.txt +135 -0
  589. numpy/ma/LICENSE +24 -0
  590. numpy/ma/README.rst +236 -0
  591. numpy/ma/__init__.py +53 -0
  592. numpy/ma/__init__.pyi +458 -0
  593. numpy/ma/core.py +8929 -0
  594. numpy/ma/core.pyi +3720 -0
  595. numpy/ma/extras.py +2266 -0
  596. numpy/ma/extras.pyi +297 -0
  597. numpy/ma/mrecords.py +762 -0
  598. numpy/ma/mrecords.pyi +96 -0
  599. numpy/ma/tests/__init__.py +0 -0
  600. numpy/ma/tests/test_arrayobject.py +40 -0
  601. numpy/ma/tests/test_core.py +6008 -0
  602. numpy/ma/tests/test_deprecations.py +65 -0
  603. numpy/ma/tests/test_extras.py +1945 -0
  604. numpy/ma/tests/test_mrecords.py +495 -0
  605. numpy/ma/tests/test_old_ma.py +939 -0
  606. numpy/ma/tests/test_regression.py +83 -0
  607. numpy/ma/tests/test_subclassing.py +469 -0
  608. numpy/ma/testutils.py +294 -0
  609. numpy/ma/testutils.pyi +69 -0
  610. numpy/matlib.py +380 -0
  611. numpy/matlib.pyi +580 -0
  612. numpy/matrixlib/__init__.py +12 -0
  613. numpy/matrixlib/__init__.pyi +3 -0
  614. numpy/matrixlib/defmatrix.py +1119 -0
  615. numpy/matrixlib/defmatrix.pyi +218 -0
  616. numpy/matrixlib/tests/__init__.py +0 -0
  617. numpy/matrixlib/tests/test_defmatrix.py +455 -0
  618. numpy/matrixlib/tests/test_interaction.py +360 -0
  619. numpy/matrixlib/tests/test_masked_matrix.py +240 -0
  620. numpy/matrixlib/tests/test_matrix_linalg.py +110 -0
  621. numpy/matrixlib/tests/test_multiarray.py +17 -0
  622. numpy/matrixlib/tests/test_numeric.py +18 -0
  623. numpy/matrixlib/tests/test_regression.py +31 -0
  624. numpy/polynomial/__init__.py +187 -0
  625. numpy/polynomial/__init__.pyi +31 -0
  626. numpy/polynomial/_polybase.py +1191 -0
  627. numpy/polynomial/_polybase.pyi +262 -0
  628. numpy/polynomial/_polytypes.pyi +501 -0
  629. numpy/polynomial/chebyshev.py +2001 -0
  630. numpy/polynomial/chebyshev.pyi +180 -0
  631. numpy/polynomial/hermite.py +1738 -0
  632. numpy/polynomial/hermite.pyi +106 -0
  633. numpy/polynomial/hermite_e.py +1640 -0
  634. numpy/polynomial/hermite_e.pyi +106 -0
  635. numpy/polynomial/laguerre.py +1673 -0
  636. numpy/polynomial/laguerre.pyi +100 -0
  637. numpy/polynomial/legendre.py +1603 -0
  638. numpy/polynomial/legendre.pyi +100 -0
  639. numpy/polynomial/polynomial.py +1625 -0
  640. numpy/polynomial/polynomial.pyi +109 -0
  641. numpy/polynomial/polyutils.py +759 -0
  642. numpy/polynomial/polyutils.pyi +307 -0
  643. numpy/polynomial/tests/__init__.py +0 -0
  644. numpy/polynomial/tests/test_chebyshev.py +618 -0
  645. numpy/polynomial/tests/test_classes.py +613 -0
  646. numpy/polynomial/tests/test_hermite.py +553 -0
  647. numpy/polynomial/tests/test_hermite_e.py +554 -0
  648. numpy/polynomial/tests/test_laguerre.py +535 -0
  649. numpy/polynomial/tests/test_legendre.py +566 -0
  650. numpy/polynomial/tests/test_polynomial.py +691 -0
  651. numpy/polynomial/tests/test_polyutils.py +123 -0
  652. numpy/polynomial/tests/test_printing.py +557 -0
  653. numpy/polynomial/tests/test_symbol.py +217 -0
  654. numpy/py.typed +0 -0
  655. numpy/random/LICENSE.md +71 -0
  656. numpy/random/__init__.pxd +14 -0
  657. numpy/random/__init__.py +213 -0
  658. numpy/random/__init__.pyi +124 -0
  659. numpy/random/_bounded_integers.cp314t-win_arm64.lib +0 -0
  660. numpy/random/_bounded_integers.cp314t-win_arm64.pyd +0 -0
  661. numpy/random/_bounded_integers.pxd +38 -0
  662. numpy/random/_bounded_integers.pyi +1 -0
  663. numpy/random/_common.cp314t-win_arm64.lib +0 -0
  664. numpy/random/_common.cp314t-win_arm64.pyd +0 -0
  665. numpy/random/_common.pxd +110 -0
  666. numpy/random/_common.pyi +16 -0
  667. numpy/random/_examples/cffi/extending.py +44 -0
  668. numpy/random/_examples/cffi/parse.py +53 -0
  669. numpy/random/_examples/cython/extending.pyx +77 -0
  670. numpy/random/_examples/cython/extending_distributions.pyx +117 -0
  671. numpy/random/_examples/cython/meson.build +53 -0
  672. numpy/random/_examples/numba/extending.py +86 -0
  673. numpy/random/_examples/numba/extending_distributions.py +67 -0
  674. numpy/random/_generator.cp314t-win_arm64.lib +0 -0
  675. numpy/random/_generator.cp314t-win_arm64.pyd +0 -0
  676. numpy/random/_generator.pyi +862 -0
  677. numpy/random/_mt19937.cp314t-win_arm64.lib +0 -0
  678. numpy/random/_mt19937.cp314t-win_arm64.pyd +0 -0
  679. numpy/random/_mt19937.pyi +27 -0
  680. numpy/random/_pcg64.cp314t-win_arm64.lib +0 -0
  681. numpy/random/_pcg64.cp314t-win_arm64.pyd +0 -0
  682. numpy/random/_pcg64.pyi +41 -0
  683. numpy/random/_philox.cp314t-win_arm64.lib +0 -0
  684. numpy/random/_philox.cp314t-win_arm64.pyd +0 -0
  685. numpy/random/_philox.pyi +36 -0
  686. numpy/random/_pickle.py +88 -0
  687. numpy/random/_pickle.pyi +43 -0
  688. numpy/random/_sfc64.cp314t-win_arm64.lib +0 -0
  689. numpy/random/_sfc64.cp314t-win_arm64.pyd +0 -0
  690. numpy/random/_sfc64.pyi +25 -0
  691. numpy/random/bit_generator.cp314t-win_arm64.lib +0 -0
  692. numpy/random/bit_generator.cp314t-win_arm64.pyd +0 -0
  693. numpy/random/bit_generator.pxd +40 -0
  694. numpy/random/bit_generator.pyi +123 -0
  695. numpy/random/c_distributions.pxd +119 -0
  696. numpy/random/lib/npyrandom.lib +0 -0
  697. numpy/random/mtrand.cp314t-win_arm64.lib +0 -0
  698. numpy/random/mtrand.cp314t-win_arm64.pyd +0 -0
  699. numpy/random/mtrand.pyi +759 -0
  700. numpy/random/tests/__init__.py +0 -0
  701. numpy/random/tests/data/__init__.py +0 -0
  702. numpy/random/tests/data/generator_pcg64_np121.pkl.gz +0 -0
  703. numpy/random/tests/data/generator_pcg64_np126.pkl.gz +0 -0
  704. numpy/random/tests/data/mt19937-testset-1.csv +1001 -0
  705. numpy/random/tests/data/mt19937-testset-2.csv +1001 -0
  706. numpy/random/tests/data/pcg64-testset-1.csv +1001 -0
  707. numpy/random/tests/data/pcg64-testset-2.csv +1001 -0
  708. numpy/random/tests/data/pcg64dxsm-testset-1.csv +1001 -0
  709. numpy/random/tests/data/pcg64dxsm-testset-2.csv +1001 -0
  710. numpy/random/tests/data/philox-testset-1.csv +1001 -0
  711. numpy/random/tests/data/philox-testset-2.csv +1001 -0
  712. numpy/random/tests/data/sfc64-testset-1.csv +1001 -0
  713. numpy/random/tests/data/sfc64-testset-2.csv +1001 -0
  714. numpy/random/tests/data/sfc64_np126.pkl.gz +0 -0
  715. numpy/random/tests/test_direct.py +595 -0
  716. numpy/random/tests/test_extending.py +131 -0
  717. numpy/random/tests/test_generator_mt19937.py +2825 -0
  718. numpy/random/tests/test_generator_mt19937_regressions.py +221 -0
  719. numpy/random/tests/test_random.py +1724 -0
  720. numpy/random/tests/test_randomstate.py +2099 -0
  721. numpy/random/tests/test_randomstate_regression.py +213 -0
  722. numpy/random/tests/test_regression.py +175 -0
  723. numpy/random/tests/test_seed_sequence.py +79 -0
  724. numpy/random/tests/test_smoke.py +882 -0
  725. numpy/rec/__init__.py +2 -0
  726. numpy/rec/__init__.pyi +23 -0
  727. numpy/strings/__init__.py +2 -0
  728. numpy/strings/__init__.pyi +97 -0
  729. numpy/testing/__init__.py +22 -0
  730. numpy/testing/__init__.pyi +107 -0
  731. numpy/testing/_private/__init__.py +0 -0
  732. numpy/testing/_private/__init__.pyi +0 -0
  733. numpy/testing/_private/extbuild.py +250 -0
  734. numpy/testing/_private/extbuild.pyi +25 -0
  735. numpy/testing/_private/utils.py +2830 -0
  736. numpy/testing/_private/utils.pyi +505 -0
  737. numpy/testing/overrides.py +84 -0
  738. numpy/testing/overrides.pyi +10 -0
  739. numpy/testing/print_coercion_tables.py +207 -0
  740. numpy/testing/print_coercion_tables.pyi +26 -0
  741. numpy/testing/tests/__init__.py +0 -0
  742. numpy/testing/tests/test_utils.py +2123 -0
  743. numpy/tests/__init__.py +0 -0
  744. numpy/tests/test__all__.py +10 -0
  745. numpy/tests/test_configtool.py +51 -0
  746. numpy/tests/test_ctypeslib.py +383 -0
  747. numpy/tests/test_lazyloading.py +42 -0
  748. numpy/tests/test_matlib.py +59 -0
  749. numpy/tests/test_numpy_config.py +47 -0
  750. numpy/tests/test_numpy_version.py +54 -0
  751. numpy/tests/test_public_api.py +807 -0
  752. numpy/tests/test_reloading.py +76 -0
  753. numpy/tests/test_scripts.py +48 -0
  754. numpy/tests/test_warnings.py +79 -0
  755. numpy/typing/__init__.py +233 -0
  756. numpy/typing/__init__.pyi +3 -0
  757. numpy/typing/mypy_plugin.py +200 -0
  758. numpy/typing/tests/__init__.py +0 -0
  759. numpy/typing/tests/data/fail/arithmetic.pyi +126 -0
  760. numpy/typing/tests/data/fail/array_constructors.pyi +34 -0
  761. numpy/typing/tests/data/fail/array_like.pyi +15 -0
  762. numpy/typing/tests/data/fail/array_pad.pyi +6 -0
  763. numpy/typing/tests/data/fail/arrayprint.pyi +15 -0
  764. numpy/typing/tests/data/fail/arrayterator.pyi +14 -0
  765. numpy/typing/tests/data/fail/bitwise_ops.pyi +17 -0
  766. numpy/typing/tests/data/fail/char.pyi +63 -0
  767. numpy/typing/tests/data/fail/chararray.pyi +61 -0
  768. numpy/typing/tests/data/fail/comparisons.pyi +27 -0
  769. numpy/typing/tests/data/fail/constants.pyi +3 -0
  770. numpy/typing/tests/data/fail/datasource.pyi +16 -0
  771. numpy/typing/tests/data/fail/dtype.pyi +17 -0
  772. numpy/typing/tests/data/fail/einsumfunc.pyi +12 -0
  773. numpy/typing/tests/data/fail/flatiter.pyi +38 -0
  774. numpy/typing/tests/data/fail/fromnumeric.pyi +148 -0
  775. numpy/typing/tests/data/fail/histograms.pyi +12 -0
  776. numpy/typing/tests/data/fail/index_tricks.pyi +14 -0
  777. numpy/typing/tests/data/fail/lib_function_base.pyi +60 -0
  778. numpy/typing/tests/data/fail/lib_polynomial.pyi +29 -0
  779. numpy/typing/tests/data/fail/lib_utils.pyi +3 -0
  780. numpy/typing/tests/data/fail/lib_version.pyi +6 -0
  781. numpy/typing/tests/data/fail/linalg.pyi +52 -0
  782. numpy/typing/tests/data/fail/ma.pyi +155 -0
  783. numpy/typing/tests/data/fail/memmap.pyi +5 -0
  784. numpy/typing/tests/data/fail/modules.pyi +17 -0
  785. numpy/typing/tests/data/fail/multiarray.pyi +52 -0
  786. numpy/typing/tests/data/fail/ndarray.pyi +11 -0
  787. numpy/typing/tests/data/fail/ndarray_misc.pyi +49 -0
  788. numpy/typing/tests/data/fail/nditer.pyi +8 -0
  789. numpy/typing/tests/data/fail/nested_sequence.pyi +17 -0
  790. numpy/typing/tests/data/fail/npyio.pyi +24 -0
  791. numpy/typing/tests/data/fail/numerictypes.pyi +5 -0
  792. numpy/typing/tests/data/fail/random.pyi +62 -0
  793. numpy/typing/tests/data/fail/rec.pyi +17 -0
  794. numpy/typing/tests/data/fail/scalars.pyi +86 -0
  795. numpy/typing/tests/data/fail/shape.pyi +7 -0
  796. numpy/typing/tests/data/fail/shape_base.pyi +8 -0
  797. numpy/typing/tests/data/fail/stride_tricks.pyi +9 -0
  798. numpy/typing/tests/data/fail/strings.pyi +52 -0
  799. numpy/typing/tests/data/fail/testing.pyi +28 -0
  800. numpy/typing/tests/data/fail/twodim_base.pyi +39 -0
  801. numpy/typing/tests/data/fail/type_check.pyi +12 -0
  802. numpy/typing/tests/data/fail/ufunc_config.pyi +21 -0
  803. numpy/typing/tests/data/fail/ufunclike.pyi +21 -0
  804. numpy/typing/tests/data/fail/ufuncs.pyi +17 -0
  805. numpy/typing/tests/data/fail/warnings_and_errors.pyi +5 -0
  806. numpy/typing/tests/data/misc/extended_precision.pyi +9 -0
  807. numpy/typing/tests/data/mypy.ini +8 -0
  808. numpy/typing/tests/data/pass/arithmetic.py +614 -0
  809. numpy/typing/tests/data/pass/array_constructors.py +138 -0
  810. numpy/typing/tests/data/pass/array_like.py +43 -0
  811. numpy/typing/tests/data/pass/arrayprint.py +37 -0
  812. numpy/typing/tests/data/pass/arrayterator.py +28 -0
  813. numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
  814. numpy/typing/tests/data/pass/comparisons.py +316 -0
  815. numpy/typing/tests/data/pass/dtype.py +57 -0
  816. numpy/typing/tests/data/pass/einsumfunc.py +36 -0
  817. numpy/typing/tests/data/pass/flatiter.py +26 -0
  818. numpy/typing/tests/data/pass/fromnumeric.py +272 -0
  819. numpy/typing/tests/data/pass/index_tricks.py +62 -0
  820. numpy/typing/tests/data/pass/lib_user_array.py +22 -0
  821. numpy/typing/tests/data/pass/lib_utils.py +19 -0
  822. numpy/typing/tests/data/pass/lib_version.py +18 -0
  823. numpy/typing/tests/data/pass/literal.py +52 -0
  824. numpy/typing/tests/data/pass/ma.py +199 -0
  825. numpy/typing/tests/data/pass/mod.py +149 -0
  826. numpy/typing/tests/data/pass/modules.py +45 -0
  827. numpy/typing/tests/data/pass/multiarray.py +77 -0
  828. numpy/typing/tests/data/pass/ndarray_conversion.py +81 -0
  829. numpy/typing/tests/data/pass/ndarray_misc.py +199 -0
  830. numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
  831. numpy/typing/tests/data/pass/nditer.py +4 -0
  832. numpy/typing/tests/data/pass/numeric.py +90 -0
  833. numpy/typing/tests/data/pass/numerictypes.py +17 -0
  834. numpy/typing/tests/data/pass/random.py +1498 -0
  835. numpy/typing/tests/data/pass/recfunctions.py +164 -0
  836. numpy/typing/tests/data/pass/scalars.py +249 -0
  837. numpy/typing/tests/data/pass/shape.py +19 -0
  838. numpy/typing/tests/data/pass/simple.py +170 -0
  839. numpy/typing/tests/data/pass/ufunc_config.py +64 -0
  840. numpy/typing/tests/data/pass/ufunclike.py +52 -0
  841. numpy/typing/tests/data/pass/ufuncs.py +16 -0
  842. numpy/typing/tests/data/pass/warnings_and_errors.py +6 -0
  843. numpy/typing/tests/data/reveal/arithmetic.pyi +719 -0
  844. numpy/typing/tests/data/reveal/array_api_info.pyi +70 -0
  845. numpy/typing/tests/data/reveal/array_constructors.pyi +277 -0
  846. numpy/typing/tests/data/reveal/arraypad.pyi +27 -0
  847. numpy/typing/tests/data/reveal/arrayprint.pyi +25 -0
  848. numpy/typing/tests/data/reveal/arraysetops.pyi +74 -0
  849. numpy/typing/tests/data/reveal/arrayterator.pyi +27 -0
  850. numpy/typing/tests/data/reveal/bitwise_ops.pyi +166 -0
  851. numpy/typing/tests/data/reveal/char.pyi +225 -0
  852. numpy/typing/tests/data/reveal/chararray.pyi +138 -0
  853. numpy/typing/tests/data/reveal/comparisons.pyi +264 -0
  854. numpy/typing/tests/data/reveal/constants.pyi +14 -0
  855. numpy/typing/tests/data/reveal/ctypeslib.pyi +81 -0
  856. numpy/typing/tests/data/reveal/datasource.pyi +23 -0
  857. numpy/typing/tests/data/reveal/dtype.pyi +132 -0
  858. numpy/typing/tests/data/reveal/einsumfunc.pyi +39 -0
  859. numpy/typing/tests/data/reveal/emath.pyi +54 -0
  860. numpy/typing/tests/data/reveal/fft.pyi +37 -0
  861. numpy/typing/tests/data/reveal/flatiter.pyi +86 -0
  862. numpy/typing/tests/data/reveal/fromnumeric.pyi +347 -0
  863. numpy/typing/tests/data/reveal/getlimits.pyi +53 -0
  864. numpy/typing/tests/data/reveal/histograms.pyi +25 -0
  865. numpy/typing/tests/data/reveal/index_tricks.pyi +70 -0
  866. numpy/typing/tests/data/reveal/lib_function_base.pyi +409 -0
  867. numpy/typing/tests/data/reveal/lib_polynomial.pyi +147 -0
  868. numpy/typing/tests/data/reveal/lib_utils.pyi +17 -0
  869. numpy/typing/tests/data/reveal/lib_version.pyi +20 -0
  870. numpy/typing/tests/data/reveal/linalg.pyi +154 -0
  871. numpy/typing/tests/data/reveal/ma.pyi +1098 -0
  872. numpy/typing/tests/data/reveal/matrix.pyi +73 -0
  873. numpy/typing/tests/data/reveal/memmap.pyi +19 -0
  874. numpy/typing/tests/data/reveal/mod.pyi +178 -0
  875. numpy/typing/tests/data/reveal/modules.pyi +51 -0
  876. numpy/typing/tests/data/reveal/multiarray.pyi +197 -0
  877. numpy/typing/tests/data/reveal/nbit_base_example.pyi +20 -0
  878. numpy/typing/tests/data/reveal/ndarray_assignability.pyi +82 -0
  879. numpy/typing/tests/data/reveal/ndarray_conversion.pyi +83 -0
  880. numpy/typing/tests/data/reveal/ndarray_misc.pyi +246 -0
  881. numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi +47 -0
  882. numpy/typing/tests/data/reveal/nditer.pyi +49 -0
  883. numpy/typing/tests/data/reveal/nested_sequence.pyi +25 -0
  884. numpy/typing/tests/data/reveal/npyio.pyi +83 -0
  885. numpy/typing/tests/data/reveal/numeric.pyi +170 -0
  886. numpy/typing/tests/data/reveal/numerictypes.pyi +16 -0
  887. numpy/typing/tests/data/reveal/polynomial_polybase.pyi +217 -0
  888. numpy/typing/tests/data/reveal/polynomial_polyutils.pyi +218 -0
  889. numpy/typing/tests/data/reveal/polynomial_series.pyi +138 -0
  890. numpy/typing/tests/data/reveal/random.pyi +1546 -0
  891. numpy/typing/tests/data/reveal/rec.pyi +171 -0
  892. numpy/typing/tests/data/reveal/scalars.pyi +191 -0
  893. numpy/typing/tests/data/reveal/shape.pyi +13 -0
  894. numpy/typing/tests/data/reveal/shape_base.pyi +52 -0
  895. numpy/typing/tests/data/reveal/stride_tricks.pyi +27 -0
  896. numpy/typing/tests/data/reveal/strings.pyi +196 -0
  897. numpy/typing/tests/data/reveal/testing.pyi +198 -0
  898. numpy/typing/tests/data/reveal/twodim_base.pyi +225 -0
  899. numpy/typing/tests/data/reveal/type_check.pyi +67 -0
  900. numpy/typing/tests/data/reveal/ufunc_config.pyi +29 -0
  901. numpy/typing/tests/data/reveal/ufunclike.pyi +31 -0
  902. numpy/typing/tests/data/reveal/ufuncs.pyi +142 -0
  903. numpy/typing/tests/data/reveal/warnings_and_errors.pyi +11 -0
  904. numpy/typing/tests/test_isfile.py +38 -0
  905. numpy/typing/tests/test_runtime.py +110 -0
  906. numpy/typing/tests/test_typing.py +205 -0
  907. numpy/version.py +11 -0
  908. numpy/version.pyi +9 -0
  909. numpy-2.4.1.dist-info/DELVEWHEEL +2 -0
  910. numpy-2.4.1.dist-info/METADATA +139 -0
  911. numpy-2.4.1.dist-info/RECORD +932 -0
  912. numpy-2.4.1.dist-info/WHEEL +4 -0
  913. numpy-2.4.1.dist-info/entry_points.txt +13 -0
  914. numpy-2.4.1.dist-info/licenses/LICENSE.txt +914 -0
  915. numpy-2.4.1.dist-info/licenses/numpy/_core/include/numpy/libdivide/LICENSE.txt +21 -0
  916. numpy-2.4.1.dist-info/licenses/numpy/_core/src/common/pythoncapi-compat/COPYING +14 -0
  917. numpy-2.4.1.dist-info/licenses/numpy/_core/src/highway/LICENSE +371 -0
  918. numpy-2.4.1.dist-info/licenses/numpy/_core/src/multiarray/dragon4_LICENSE.txt +27 -0
  919. numpy-2.4.1.dist-info/licenses/numpy/_core/src/npysort/x86-simd-sort/LICENSE.md +28 -0
  920. numpy-2.4.1.dist-info/licenses/numpy/_core/src/umath/svml/LICENSE +30 -0
  921. numpy-2.4.1.dist-info/licenses/numpy/fft/pocketfft/LICENSE.md +25 -0
  922. numpy-2.4.1.dist-info/licenses/numpy/linalg/lapack_lite/LICENSE.txt +48 -0
  923. numpy-2.4.1.dist-info/licenses/numpy/ma/LICENSE +24 -0
  924. numpy-2.4.1.dist-info/licenses/numpy/random/LICENSE.md +71 -0
  925. numpy-2.4.1.dist-info/licenses/numpy/random/src/distributions/LICENSE.md +61 -0
  926. numpy-2.4.1.dist-info/licenses/numpy/random/src/mt19937/LICENSE.md +61 -0
  927. numpy-2.4.1.dist-info/licenses/numpy/random/src/pcg64/LICENSE.md +22 -0
  928. numpy-2.4.1.dist-info/licenses/numpy/random/src/philox/LICENSE.md +31 -0
  929. numpy-2.4.1.dist-info/licenses/numpy/random/src/sfc64/LICENSE.md +27 -0
  930. numpy-2.4.1.dist-info/licenses/numpy/random/src/splitmix64/LICENSE.md +9 -0
  931. numpy.libs/msvcp140-5f1c5dd31916990d94181e07bc3afb32.dll +0 -0
  932. numpy.libs/scipy_openblas-7b69cbfd2599e6035f1310f2a72d59a6.dll +0 -0
@@ -0,0 +1,1168 @@
1
+ import contextlib
2
+ import itertools
3
+ import operator
4
+ import platform
5
+ import sys
6
+ import warnings
7
+
8
+ import pytest
9
+ from hypothesis import given, settings
10
+ from hypothesis.extra import numpy as hynp
11
+ from hypothesis.strategies import sampled_from
12
+
13
+ import numpy as np
14
+ from numpy._core._rational_tests import rational
15
+ from numpy._utils import _pep440
16
+ from numpy.exceptions import ComplexWarning
17
+ from numpy.testing import (
18
+ IS_PYPY,
19
+ _gen_alignment_data,
20
+ assert_,
21
+ assert_almost_equal,
22
+ assert_array_equal,
23
+ assert_equal,
24
+ assert_raises,
25
+ check_support_sve,
26
+ )
27
+
28
+ types = [np.bool, np.byte, np.ubyte, np.short, np.ushort, np.intc, np.uintc,
29
+ np.int_, np.uint, np.longlong, np.ulonglong,
30
+ np.single, np.double, np.longdouble, np.csingle,
31
+ np.cdouble, np.clongdouble]
32
+
33
+ floating_types = np.floating.__subclasses__()
34
+ complex_floating_types = np.complexfloating.__subclasses__()
35
+
36
+ objecty_things = [object(), None, np.array(None, dtype=object)]
37
+
38
+ binary_operators_for_scalars = [
39
+ operator.lt, operator.le, operator.eq, operator.ne, operator.ge,
40
+ operator.gt, operator.add, operator.floordiv, operator.mod,
41
+ operator.mul, operator.pow, operator.sub, operator.truediv
42
+ ]
43
+ binary_operators_for_scalar_ints = binary_operators_for_scalars + [
44
+ operator.xor, operator.or_, operator.and_
45
+ ]
46
+
47
+
48
+ # This compares scalarmath against ufuncs.
49
+
50
+ class TestTypes:
51
+ def test_types(self):
52
+ for atype in types:
53
+ a = atype(1)
54
+ assert_(a == 1, f"error with {atype!r}: got {a!r}")
55
+
56
+ def test_type_add(self):
57
+ # list of types
58
+ for k, atype in enumerate(types):
59
+ a_scalar = atype(3)
60
+ a_array = np.array([3], dtype=atype)
61
+ for l, btype in enumerate(types):
62
+ b_scalar = btype(1)
63
+ b_array = np.array([1], dtype=btype)
64
+ c_scalar = a_scalar + b_scalar
65
+ c_array = a_array + b_array
66
+ # It was comparing the type numbers, but the new ufunc
67
+ # function-finding mechanism finds the lowest function
68
+ # to which both inputs can be cast - which produces 'l'
69
+ # when you do 'q' + 'b'. The old function finding mechanism
70
+ # skipped ahead based on the first argument, but that
71
+ # does not produce properly symmetric results...
72
+ assert_equal(c_scalar.dtype, c_array.dtype,
73
+ "error with types (%d/'%c' + %d/'%c')" %
74
+ (k, np.dtype(atype).char, l, np.dtype(btype).char))
75
+
76
+ def test_type_create(self):
77
+ for atype in types:
78
+ a = np.array([1, 2, 3], atype)
79
+ b = atype([1, 2, 3])
80
+ assert_equal(a, b)
81
+
82
+ def test_leak(self):
83
+ # test leak of scalar objects
84
+ # a leak would show up in valgrind as still-reachable of ~2.6MB
85
+ for i in range(200000):
86
+ np.add(1, 1)
87
+
88
+
89
+ def check_ufunc_scalar_equivalence(op, arr1, arr2):
90
+ scalar1 = arr1[()]
91
+ scalar2 = arr2[()]
92
+ assert isinstance(scalar1, np.generic)
93
+ assert isinstance(scalar2, np.generic)
94
+
95
+ if arr1.dtype.kind == "c" or arr2.dtype.kind == "c":
96
+ comp_ops = {operator.ge, operator.gt, operator.le, operator.lt}
97
+ if op in comp_ops and (np.isnan(scalar1) or np.isnan(scalar2)):
98
+ pytest.xfail("complex comp ufuncs use sort-order, scalars do not.")
99
+ if op == operator.pow and arr2.item() in [-1, 0, 0.5, 1, 2]:
100
+ # array**scalar special case can have different result dtype
101
+ # (Other powers may have issues also, but are not hit here.)
102
+ # TODO: It would be nice to resolve this issue.
103
+ pytest.skip("array**2 can have incorrect/weird result dtype")
104
+
105
+ # ignore fpe's since they may just mismatch for integers anyway.
106
+ with warnings.catch_warnings(), np.errstate(all="ignore"):
107
+ # Comparisons DeprecationWarnings replacing errors (2022-03):
108
+ warnings.simplefilter("error", DeprecationWarning)
109
+ try:
110
+ res = op(arr1, arr2)
111
+ except Exception as e:
112
+ with pytest.raises(type(e)):
113
+ op(scalar1, scalar2)
114
+ else:
115
+ scalar_res = op(scalar1, scalar2)
116
+ assert_array_equal(scalar_res, res, strict=True)
117
+
118
+
119
+ @pytest.mark.slow
120
+ @settings(max_examples=10000, deadline=2000)
121
+ @given(sampled_from(binary_operators_for_scalars),
122
+ hynp.arrays(dtype=hynp.scalar_dtypes(), shape=()),
123
+ hynp.arrays(dtype=hynp.scalar_dtypes(), shape=()))
124
+ def test_array_scalar_ufunc_equivalence(op, arr1, arr2):
125
+ """
126
+ This is a thorough test attempting to cover important promotion paths
127
+ and ensuring that arrays and scalars stay as aligned as possible.
128
+ However, if it creates troubles, it should maybe just be removed.
129
+ """
130
+ check_ufunc_scalar_equivalence(op, arr1, arr2)
131
+
132
+
133
+ @pytest.mark.slow
134
+ @given(sampled_from(binary_operators_for_scalars),
135
+ hynp.scalar_dtypes(), hynp.scalar_dtypes())
136
+ def test_array_scalar_ufunc_dtypes(op, dt1, dt2):
137
+ # Same as above, but don't worry about sampling weird values so that we
138
+ # do not have to sample as much
139
+ arr1 = np.array(2, dtype=dt1)
140
+ arr2 = np.array(3, dtype=dt2) # some power do weird things.
141
+
142
+ check_ufunc_scalar_equivalence(op, arr1, arr2)
143
+
144
+
145
+ @pytest.mark.parametrize("fscalar", [np.float16, np.float32])
146
+ def test_int_float_promotion_truediv(fscalar):
147
+ # Promotion for mixed int and float32/float16 must not go to float64
148
+ i = np.int8(1)
149
+ f = fscalar(1)
150
+ expected = np.result_type(i, f)
151
+ assert (i / f).dtype == expected
152
+ assert (f / i).dtype == expected
153
+ # But normal int / int true division goes to float64:
154
+ assert (i / i).dtype == np.dtype("float64")
155
+ # For int16, result has to be ast least float32 (takes ufunc path):
156
+ assert (np.int16(1) / f).dtype == np.dtype("float32")
157
+
158
+
159
+ class TestBaseMath:
160
+ @pytest.mark.xfail(check_support_sve(), reason="gh-22982")
161
+ def test_blocked(self):
162
+ # test alignments offsets for simd instructions
163
+ # alignments for vz + 2 * (vs - 1) + 1
164
+ for dt, sz in [(np.float32, 11), (np.float64, 7), (np.int32, 11)]:
165
+ for out, inp1, inp2, msg in _gen_alignment_data(dtype=dt,
166
+ type='binary',
167
+ max_size=sz):
168
+ exp1 = np.ones_like(inp1)
169
+ inp1[...] = np.ones_like(inp1)
170
+ inp2[...] = np.zeros_like(inp2)
171
+ assert_almost_equal(np.add(inp1, inp2), exp1, err_msg=msg)
172
+ assert_almost_equal(np.add(inp1, 2), exp1 + 2, err_msg=msg)
173
+ assert_almost_equal(np.add(1, inp2), exp1, err_msg=msg)
174
+
175
+ np.add(inp1, inp2, out=out)
176
+ assert_almost_equal(out, exp1, err_msg=msg)
177
+
178
+ inp2[...] += np.arange(inp2.size, dtype=dt) + 1
179
+ assert_almost_equal(np.square(inp2),
180
+ np.multiply(inp2, inp2), err_msg=msg)
181
+ # skip true divide for ints
182
+ if dt != np.int32:
183
+ assert_almost_equal(np.reciprocal(inp2),
184
+ np.divide(1, inp2), err_msg=msg)
185
+
186
+ inp1[...] = np.ones_like(inp1)
187
+ np.add(inp1, 2, out=out)
188
+ assert_almost_equal(out, exp1 + 2, err_msg=msg)
189
+ inp2[...] = np.ones_like(inp2)
190
+ np.add(2, inp2, out=out)
191
+ assert_almost_equal(out, exp1 + 2, err_msg=msg)
192
+
193
+ def test_lower_align(self):
194
+ # check data that is not aligned to element size
195
+ # i.e doubles are aligned to 4 bytes on i386
196
+ d = np.zeros(23 * 8, dtype=np.int8)[4:-4].view(np.float64)
197
+ o = np.zeros(23 * 8, dtype=np.int8)[4:-4].view(np.float64)
198
+ assert_almost_equal(d + d, d * 2)
199
+ np.add(d, d, out=o)
200
+ np.add(np.ones_like(d), d, out=o)
201
+ np.add(d, np.ones_like(d), out=o)
202
+ np.add(np.ones_like(d), d)
203
+ np.add(d, np.ones_like(d))
204
+
205
+
206
+ class TestPower:
207
+ def test_small_types(self):
208
+ for t in [np.int8, np.int16, np.float16]:
209
+ a = t(3)
210
+ b = a ** 4
211
+ assert_(b == 81, f"error with {t!r}: got {b!r}")
212
+
213
+ def test_large_types(self):
214
+ for t in [np.int32, np.int64, np.float32, np.float64, np.longdouble]:
215
+ a = t(51)
216
+ b = a ** 4
217
+ msg = f"error with {t!r}: got {b!r}"
218
+ if np.issubdtype(t, np.integer):
219
+ assert_(b == 6765201, msg)
220
+ else:
221
+ assert_almost_equal(b, 6765201, err_msg=msg)
222
+
223
+ def test_integers_to_negative_integer_power(self):
224
+ # Note that the combination of uint64 with a signed integer
225
+ # has common type np.float64. The other combinations should all
226
+ # raise a ValueError for integer ** negative integer.
227
+ exp = [np.array(-1, dt)[()] for dt in 'bhilq']
228
+
229
+ # 1 ** -1 possible special case
230
+ base = [np.array(1, dt)[()] for dt in 'bhilqBHILQ']
231
+ for i1, i2 in itertools.product(base, exp):
232
+ if i1.dtype != np.uint64:
233
+ assert_raises(ValueError, operator.pow, i1, i2)
234
+ else:
235
+ res = operator.pow(i1, i2)
236
+ assert_(res.dtype.type is np.float64)
237
+ assert_almost_equal(res, 1.)
238
+
239
+ # -1 ** -1 possible special case
240
+ base = [np.array(-1, dt)[()] for dt in 'bhilq']
241
+ for i1, i2 in itertools.product(base, exp):
242
+ if i1.dtype != np.uint64:
243
+ assert_raises(ValueError, operator.pow, i1, i2)
244
+ else:
245
+ res = operator.pow(i1, i2)
246
+ assert_(res.dtype.type is np.float64)
247
+ assert_almost_equal(res, -1.)
248
+
249
+ # 2 ** -1 perhaps generic
250
+ base = [np.array(2, dt)[()] for dt in 'bhilqBHILQ']
251
+ for i1, i2 in itertools.product(base, exp):
252
+ if i1.dtype != np.uint64:
253
+ assert_raises(ValueError, operator.pow, i1, i2)
254
+ else:
255
+ res = operator.pow(i1, i2)
256
+ assert_(res.dtype.type is np.float64)
257
+ assert_almost_equal(res, .5)
258
+
259
+ def test_mixed_types(self):
260
+ typelist = [np.int8, np.int16, np.float16,
261
+ np.float32, np.float64, np.int8,
262
+ np.int16, np.int32, np.int64]
263
+ for t1 in typelist:
264
+ for t2 in typelist:
265
+ a = t1(3)
266
+ b = t2(2)
267
+ result = a**b
268
+ msg = f"error with {t1!r} and {t2!r}:got {result!r}, expected {9!r}"
269
+ if np.issubdtype(np.dtype(result), np.integer):
270
+ assert_(result == 9, msg)
271
+ else:
272
+ assert_almost_equal(result, 9, err_msg=msg)
273
+
274
+ def test_modular_power(self):
275
+ # modular power is not implemented, so ensure it errors
276
+ a = 5
277
+ b = 4
278
+ c = 10
279
+ expected = pow(a, b, c) # noqa: F841
280
+ for t in (np.int32, np.float32, np.complex64):
281
+ # note that 3-operand power only dispatches on the first argument
282
+ assert_raises(TypeError, operator.pow, t(a), b, c)
283
+ assert_raises(TypeError, operator.pow, np.array(t(a)), b, c)
284
+
285
+
286
+ def floordiv_and_mod(x, y):
287
+ return (x // y, x % y)
288
+
289
+
290
+ def _signs(dt):
291
+ if dt in np.typecodes['UnsignedInteger']:
292
+ return (+1,)
293
+ else:
294
+ return (+1, -1)
295
+
296
+
297
+ class TestModulus:
298
+
299
+ def test_modulus_basic(self):
300
+ dt = np.typecodes['AllInteger'] + np.typecodes['Float']
301
+ for op in [floordiv_and_mod, divmod]:
302
+ for dt1, dt2 in itertools.product(dt, dt):
303
+ for sg1, sg2 in itertools.product(_signs(dt1), _signs(dt2)):
304
+ fmt = 'op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s'
305
+ msg = fmt % (op.__name__, dt1, dt2, sg1, sg2)
306
+ a = np.array(sg1 * 71, dtype=dt1)[()]
307
+ b = np.array(sg2 * 19, dtype=dt2)[()]
308
+ div, rem = op(a, b)
309
+ assert_equal(div * b + rem, a, err_msg=msg)
310
+ if sg2 == -1:
311
+ assert_(b < rem <= 0, msg)
312
+ else:
313
+ assert_(b > rem >= 0, msg)
314
+
315
+ def test_float_modulus_exact(self):
316
+ # test that float results are exact for small integers. This also
317
+ # holds for the same integers scaled by powers of two.
318
+ nlst = list(range(-127, 0))
319
+ plst = list(range(1, 128))
320
+ dividend = nlst + [0] + plst
321
+ divisor = nlst + plst
322
+ arg = list(itertools.product(dividend, divisor))
323
+ tgt = [divmod(*t) for t in arg]
324
+
325
+ a, b = np.array(arg, dtype=int).T
326
+ # convert exact integer results from Python to float so that
327
+ # signed zero can be used, it is checked.
328
+ tgtdiv, tgtrem = np.array(tgt, dtype=float).T
329
+ tgtdiv = np.where((tgtdiv == 0.0) & ((b < 0) ^ (a < 0)), -0.0, tgtdiv)
330
+ tgtrem = np.where((tgtrem == 0.0) & (b < 0), -0.0, tgtrem)
331
+
332
+ for op in [floordiv_and_mod, divmod]:
333
+ for dt in np.typecodes['Float']:
334
+ msg = f'op: {op.__name__}, dtype: {dt}'
335
+ fa = a.astype(dt)
336
+ fb = b.astype(dt)
337
+ # use list comprehension so a_ and b_ are scalars
338
+ div, rem = zip(*[op(a_, b_) for a_, b_ in zip(fa, fb)])
339
+ assert_equal(div, tgtdiv, err_msg=msg)
340
+ assert_equal(rem, tgtrem, err_msg=msg)
341
+
342
+ def test_float_modulus_roundoff(self):
343
+ # gh-6127
344
+ dt = np.typecodes['Float']
345
+ for op in [floordiv_and_mod, divmod]:
346
+ for dt1, dt2 in itertools.product(dt, dt):
347
+ for sg1, sg2 in itertools.product((+1, -1), (+1, -1)):
348
+ fmt = 'op: %s, dt1: %s, dt2: %s, sg1: %s, sg2: %s'
349
+ msg = fmt % (op.__name__, dt1, dt2, sg1, sg2)
350
+ a = np.array(sg1 * 78 * 6e-8, dtype=dt1)[()]
351
+ b = np.array(sg2 * 6e-8, dtype=dt2)[()]
352
+ div, rem = op(a, b)
353
+ # Equal assertion should hold when fmod is used
354
+ assert_equal(div * b + rem, a, err_msg=msg)
355
+ if sg2 == -1:
356
+ assert_(b < rem <= 0, msg)
357
+ else:
358
+ assert_(b > rem >= 0, msg)
359
+
360
+ def test_float_modulus_corner_cases(self):
361
+ # Check remainder magnitude.
362
+ for dt in np.typecodes['Float']:
363
+ b = np.array(1.0, dtype=dt)
364
+ a = np.nextafter(np.array(0.0, dtype=dt), -b)
365
+ rem = operator.mod(a, b)
366
+ assert_(rem <= b, f'dt: {dt}')
367
+ rem = operator.mod(-a, -b)
368
+ assert_(rem >= -b, f'dt: {dt}')
369
+
370
+ # Check nans, inf
371
+ with warnings.catch_warnings(), np.errstate(all='ignore'):
372
+ for dt in np.typecodes['Float']:
373
+ fone = np.array(1.0, dtype=dt)
374
+ fzer = np.array(0.0, dtype=dt)
375
+ finf = np.array(np.inf, dtype=dt)
376
+ fnan = np.array(np.nan, dtype=dt)
377
+ rem = operator.mod(fone, fzer)
378
+ assert_(np.isnan(rem), f'dt: {dt}')
379
+ # MSVC 2008 returns NaN here, so disable the check.
380
+ #rem = operator.mod(fone, finf)
381
+ #assert_(rem == fone, 'dt: %s' % dt)
382
+ rem = operator.mod(fone, fnan)
383
+ assert_(np.isnan(rem), f'dt: {dt}')
384
+ rem = operator.mod(finf, fone)
385
+ assert_(np.isnan(rem), f'dt: {dt}')
386
+ for op in [floordiv_and_mod, divmod]:
387
+ div, mod = op(fone, fzer)
388
+ assert_(np.isinf(div)) and assert_(np.isnan(mod))
389
+
390
+ def test_inplace_floordiv_handling(self):
391
+ # issue gh-12927
392
+ # this only applies to in-place floordiv //=, because the output type
393
+ # promotes to float which does not fit
394
+ a = np.array([1, 2], np.int64)
395
+ b = np.array([1, 2], np.uint64)
396
+ with pytest.raises(TypeError,
397
+ match=r"Cannot cast ufunc 'floor_divide' output from"):
398
+ a //= b
399
+
400
+ class TestComparison:
401
+ def test_comparision_different_types(self):
402
+ x = np.array(1)
403
+ y = np.array('s')
404
+ eq = x == y
405
+ neq = x != y
406
+ assert eq is np.bool_(False)
407
+ assert neq is np.bool_(True)
408
+
409
+
410
+ class TestComplexDivision:
411
+ def test_zero_division(self):
412
+ with np.errstate(all="ignore"):
413
+ for t in [np.complex64, np.complex128]:
414
+ a = t(0.0)
415
+ b = t(1.0)
416
+ assert_(np.isinf(b / a))
417
+ b = t(complex(np.inf, np.inf))
418
+ assert_(np.isinf(b / a))
419
+ b = t(complex(np.inf, np.nan))
420
+ assert_(np.isinf(b / a))
421
+ b = t(complex(np.nan, np.inf))
422
+ assert_(np.isinf(b / a))
423
+ b = t(complex(np.nan, np.nan))
424
+ assert_(np.isnan(b / a))
425
+ b = t(0.)
426
+ assert_(np.isnan(b / a))
427
+
428
+ def test_signed_zeros(self):
429
+ with np.errstate(all="ignore"):
430
+ for t in [np.complex64, np.complex128]:
431
+ # tupled (numerator, denominator, expected)
432
+ # for testing as expected == numerator/denominator
433
+ data = (
434
+ (( 0.0, -1.0), ( 0.0, 1.0), (-1.0, -0.0)),
435
+ (( 0.0, -1.0), ( 0.0, -1.0), ( 1.0, -0.0)),
436
+ (( 0.0, -1.0), (-0.0, -1.0), ( 1.0, 0.0)),
437
+ (( 0.0, -1.0), (-0.0, 1.0), (-1.0, 0.0)),
438
+ (( 0.0, 1.0), ( 0.0, -1.0), (-1.0, 0.0)),
439
+ (( 0.0, -1.0), ( 0.0, -1.0), ( 1.0, -0.0)),
440
+ ((-0.0, -1.0), ( 0.0, -1.0), ( 1.0, -0.0)),
441
+ ((-0.0, 1.0), ( 0.0, -1.0), (-1.0, -0.0))
442
+ )
443
+ for cases in data:
444
+ n = cases[0]
445
+ d = cases[1]
446
+ ex = cases[2]
447
+ result = t(complex(n[0], n[1])) / t(complex(d[0], d[1]))
448
+ # check real and imag parts separately to avoid comparison
449
+ # in array context, which does not account for signed zeros
450
+ assert_equal(result.real, ex[0])
451
+ assert_equal(result.imag, ex[1])
452
+
453
+ def test_branches(self):
454
+ with np.errstate(all="ignore"):
455
+ for t in [np.complex64, np.complex128]:
456
+ # tupled (numerator, denominator, expected)
457
+ # for testing as expected == numerator/denominator
458
+ data = []
459
+
460
+ # trigger branch: real(fabs(denom)) > imag(fabs(denom))
461
+ # followed by else condition as neither are == 0
462
+ data.append((( 2.0, 1.0), ( 2.0, 1.0), (1.0, 0.0)))
463
+
464
+ # trigger branch: real(fabs(denom)) > imag(fabs(denom))
465
+ # followed by if condition as both are == 0
466
+ # is performed in test_zero_division(), so this is skipped
467
+
468
+ # trigger else if branch: real(fabs(denom)) < imag(fabs(denom))
469
+ data.append(((1.0, 2.0), (1.0, 2.0), (1.0, 0.0)))
470
+
471
+ for cases in data:
472
+ n = cases[0]
473
+ d = cases[1]
474
+ ex = cases[2]
475
+ result = t(complex(n[0], n[1])) / t(complex(d[0], d[1]))
476
+ # check real and imag parts separately to avoid comparison
477
+ # in array context, which does not account for signed zeros
478
+ assert_equal(result.real, ex[0])
479
+ assert_equal(result.imag, ex[1])
480
+
481
+
482
+ class TestConversion:
483
+ def test_int_from_long(self):
484
+ l = [1e6, 1e12, 1e18, -1e6, -1e12, -1e18]
485
+ li = [10**6, 10**12, 10**18, -10**6, -10**12, -10**18]
486
+ for T in [None, np.float64, np.int64]:
487
+ a = np.array(l, dtype=T)
488
+ assert_equal([int(_m) for _m in a], li)
489
+
490
+ a = np.array(l[:3], dtype=np.uint64)
491
+ assert_equal([int(_m) for _m in a], li[:3])
492
+
493
+ def test_iinfo_long_values(self):
494
+ for code in 'bBhH':
495
+ with pytest.raises(OverflowError):
496
+ np.array(np.iinfo(code).max + 1, dtype=code)
497
+
498
+ for code in np.typecodes['AllInteger']:
499
+ res = np.array(np.iinfo(code).max, dtype=code)
500
+ tgt = np.iinfo(code).max
501
+ assert_(res == tgt)
502
+
503
+ for code in np.typecodes['AllInteger']:
504
+ res = np.dtype(code).type(np.iinfo(code).max)
505
+ tgt = np.iinfo(code).max
506
+ assert_(res == tgt)
507
+
508
+ def test_int_raise_behaviour(self):
509
+ def overflow_error_func(dtype):
510
+ dtype(np.iinfo(dtype).max + 1)
511
+
512
+ for code in [np.int_, np.uint, np.longlong, np.ulonglong]:
513
+ assert_raises(OverflowError, overflow_error_func, code)
514
+
515
+ def test_int_from_infinite_longdouble(self):
516
+ # gh-627
517
+ x = np.longdouble(np.inf)
518
+ assert_raises(OverflowError, int, x)
519
+ with pytest.warns(ComplexWarning):
520
+ x = np.clongdouble(np.inf)
521
+ assert_raises(OverflowError, int, x)
522
+
523
+ @pytest.mark.skipif(not IS_PYPY, reason="Test is PyPy only (gh-9972)")
524
+ def test_int_from_infinite_longdouble___int__(self):
525
+ x = np.longdouble(np.inf)
526
+ assert_raises(OverflowError, x.__int__)
527
+ with pytest.warns(ComplexWarning):
528
+ x = np.clongdouble(np.inf)
529
+ assert_raises(OverflowError, x.__int__)
530
+
531
+ @pytest.mark.skipif(np.finfo(np.double) == np.finfo(np.longdouble),
532
+ reason="long double is same as double")
533
+ @pytest.mark.skipif(platform.machine().startswith("ppc"),
534
+ reason="IBM double double")
535
+ def test_int_from_huge_longdouble(self):
536
+ # Produce a longdouble that would overflow a double,
537
+ # use exponent that avoids bug in Darwin pow function.
538
+ exp = np.finfo(np.double).maxexp - 1
539
+ huge_ld = 2 * 1234 * np.longdouble(2) ** exp
540
+ huge_i = 2 * 1234 * 2 ** exp
541
+ assert_(huge_ld != np.inf)
542
+ assert_equal(int(huge_ld), huge_i)
543
+
544
+ def test_int_from_longdouble(self):
545
+ x = np.longdouble(1.5)
546
+ assert_equal(int(x), 1)
547
+ x = np.longdouble(-10.5)
548
+ assert_equal(int(x), -10)
549
+
550
+ def test_numpy_scalar_relational_operators(self):
551
+ # All integer
552
+ for dt1 in np.typecodes['AllInteger']:
553
+ assert_(1 > np.array(0, dtype=dt1)[()], f"type {dt1} failed")
554
+ assert_(not 1 < np.array(0, dtype=dt1)[()], f"type {dt1} failed")
555
+
556
+ for dt2 in np.typecodes['AllInteger']:
557
+ assert_(np.array(1, dtype=dt1)[()] > np.array(0, dtype=dt2)[()],
558
+ f"type {dt1} and {dt2} failed")
559
+ assert_(not np.array(1, dtype=dt1)[()] < np.array(0, dtype=dt2)[()],
560
+ f"type {dt1} and {dt2} failed")
561
+
562
+ # Unsigned integers
563
+ for dt1 in 'BHILQP':
564
+ assert_(-1 < np.array(1, dtype=dt1)[()], f"type {dt1} failed")
565
+ assert_(not -1 > np.array(1, dtype=dt1)[()], f"type {dt1} failed")
566
+ assert_(-1 != np.array(1, dtype=dt1)[()], f"type {dt1} failed")
567
+
568
+ # unsigned vs signed
569
+ for dt2 in 'bhilqp':
570
+ assert_(np.array(1, dtype=dt1)[()] > np.array(-1, dtype=dt2)[()],
571
+ f"type {dt1} and {dt2} failed")
572
+ assert_(not np.array(1, dtype=dt1)[()] < np.array(-1, dtype=dt2)[()],
573
+ f"type {dt1} and {dt2} failed")
574
+ assert_(np.array(1, dtype=dt1)[()] != np.array(-1, dtype=dt2)[()],
575
+ f"type {dt1} and {dt2} failed")
576
+
577
+ # Signed integers and floats
578
+ for dt1 in 'bhlqp' + np.typecodes['Float']:
579
+ assert_(1 > np.array(-1, dtype=dt1)[()], f"type {dt1} failed")
580
+ assert_(not 1 < np.array(-1, dtype=dt1)[()], f"type {dt1} failed")
581
+ assert_(-1 == np.array(-1, dtype=dt1)[()], f"type {dt1} failed")
582
+
583
+ for dt2 in 'bhlqp' + np.typecodes['Float']:
584
+ assert_(np.array(1, dtype=dt1)[()] > np.array(-1, dtype=dt2)[()],
585
+ f"type {dt1} and {dt2} failed")
586
+ assert_(not np.array(1, dtype=dt1)[()] < np.array(-1, dtype=dt2)[()],
587
+ f"type {dt1} and {dt2} failed")
588
+ assert_(np.array(-1, dtype=dt1)[()] == np.array(-1, dtype=dt2)[()],
589
+ f"type {dt1} and {dt2} failed")
590
+
591
+ def test_scalar_comparison_to_none(self):
592
+ # Scalars should just return False and not give a warnings.
593
+ # The comparisons are flagged by pep8, ignore that.
594
+ with warnings.catch_warnings(record=True) as w:
595
+ warnings.filterwarnings('always', '', FutureWarning)
596
+ assert_(not np.float32(1) == None) # noqa: E711
597
+ assert_(not np.str_('test') == None) # noqa: E711
598
+ # This is dubious (see below):
599
+ assert_(not np.datetime64('NaT') == None) # noqa: E711
600
+
601
+ assert_(np.float32(1) != None) # noqa: E711
602
+ assert_(np.str_('test') != None) # noqa: E711
603
+ # This is dubious (see below):
604
+ assert_(np.datetime64('NaT') != None) # noqa: E711
605
+ assert_(len(w) == 0)
606
+
607
+ # For documentation purposes, this is why the datetime is dubious.
608
+ # At the time of deprecation this was no behaviour change, but
609
+ # it has to be considered when the deprecations are done.
610
+ assert_(np.equal(np.datetime64('NaT'), None))
611
+
612
+
613
+ #class TestRepr:
614
+ # def test_repr(self):
615
+ # for t in types:
616
+ # val = t(1197346475.0137341)
617
+ # val_repr = repr(val)
618
+ # val2 = eval(val_repr)
619
+ # assert_equal( val, val2 )
620
+
621
+
622
+ class TestRepr:
623
+ def _test_type_repr(self, t):
624
+ finfo = np.finfo(t)
625
+ last_fraction_bit_idx = finfo.nexp + finfo.nmant
626
+ last_exponent_bit_idx = finfo.nexp
627
+ storage_bytes = np.dtype(t).itemsize * 8
628
+ # could add some more types to the list below
629
+ for which in ['small denorm', 'small norm']:
630
+ # Values from https://en.wikipedia.org/wiki/IEEE_754
631
+ constr = np.array([0x00] * storage_bytes, dtype=np.uint8)
632
+ if which == 'small denorm':
633
+ byte = last_fraction_bit_idx // 8
634
+ bytebit = 7 - (last_fraction_bit_idx % 8)
635
+ constr[byte] = 1 << bytebit
636
+ elif which == 'small norm':
637
+ byte = last_exponent_bit_idx // 8
638
+ bytebit = 7 - (last_exponent_bit_idx % 8)
639
+ constr[byte] = 1 << bytebit
640
+ else:
641
+ raise ValueError('hmm')
642
+ val = constr.view(t)[0]
643
+ val_repr = repr(val)
644
+ val2 = t(eval(val_repr))
645
+ if not (val2 == 0 and val < 1e-100):
646
+ assert_equal(val, val2)
647
+
648
+ def test_float_repr(self):
649
+ # long double test cannot work, because eval goes through a python
650
+ # float
651
+ for t in [np.float32, np.float64]:
652
+ self._test_type_repr(t)
653
+
654
+
655
+ if not IS_PYPY:
656
+ # sys.getsizeof() is not valid on PyPy
657
+ class TestSizeOf:
658
+
659
+ def test_equal_nbytes(self):
660
+ for type in types:
661
+ x = type(0)
662
+ assert_(sys.getsizeof(x) > x.nbytes)
663
+
664
+ def test_error(self):
665
+ d = np.float32()
666
+ assert_raises(TypeError, d.__sizeof__, "a")
667
+
668
+
669
+ class TestMultiply:
670
+ def test_seq_repeat(self):
671
+ # Test that basic sequences get repeated when multiplied with
672
+ # numpy integers. And errors are raised when multiplied with others.
673
+ # Some of this behaviour may be controversial and could be open for
674
+ # change.
675
+ accepted_types = set(np.typecodes["AllInteger"])
676
+ deprecated_types = {'?'}
677
+ forbidden_types = (
678
+ set(np.typecodes["All"]) - accepted_types - deprecated_types)
679
+ forbidden_types -= {'V'} # can't default-construct void scalars
680
+
681
+ for seq_type in (list, tuple):
682
+ seq = seq_type([1, 2, 3])
683
+ for numpy_type in accepted_types:
684
+ i = np.dtype(numpy_type).type(2)
685
+ assert_equal(seq * i, seq * int(i))
686
+ assert_equal(i * seq, int(i) * seq)
687
+
688
+ for numpy_type in deprecated_types:
689
+ i = np.dtype(numpy_type).type()
690
+ with assert_raises(TypeError):
691
+ operator.mul(seq, i)
692
+
693
+ for numpy_type in forbidden_types:
694
+ i = np.dtype(numpy_type).type()
695
+ assert_raises(TypeError, operator.mul, seq, i)
696
+ assert_raises(TypeError, operator.mul, i, seq)
697
+
698
+ def test_no_seq_repeat_basic_array_like(self):
699
+ # Test that an array-like which does not know how to be multiplied
700
+ # does not attempt sequence repeat (raise TypeError).
701
+ # See also gh-7428.
702
+ class ArrayLike:
703
+ def __init__(self, arr):
704
+ self.arr = arr
705
+
706
+ def __array__(self, dtype=None, copy=None):
707
+ return self.arr
708
+
709
+ # Test for simple ArrayLike above and memoryviews (original report)
710
+ for arr_like in (ArrayLike(np.ones(3)), memoryview(np.ones(3))):
711
+ assert_array_equal(arr_like * np.float32(3.), np.full(3, 3.))
712
+ assert_array_equal(np.float32(3.) * arr_like, np.full(3, 3.))
713
+ assert_array_equal(arr_like * np.int_(3), np.full(3, 3))
714
+ assert_array_equal(np.int_(3) * arr_like, np.full(3, 3))
715
+
716
+
717
+ class TestNegative:
718
+ def test_exceptions(self):
719
+ a = np.ones((), dtype=np.bool)[()]
720
+ assert_raises(TypeError, operator.neg, a)
721
+
722
+ def test_result(self):
723
+ types = np.typecodes['AllInteger'] + np.typecodes['AllFloat']
724
+ with warnings.catch_warnings():
725
+ warnings.simplefilter('ignore', RuntimeWarning)
726
+ for dt in types:
727
+ a = np.ones((), dtype=dt)[()]
728
+ if dt in np.typecodes['UnsignedInteger']:
729
+ st = np.dtype(dt).type
730
+ max = st(np.iinfo(dt).max)
731
+ assert_equal(operator.neg(a), max)
732
+ else:
733
+ assert_equal(operator.neg(a) + a, 0)
734
+
735
+ class TestSubtract:
736
+ def test_exceptions(self):
737
+ a = np.ones((), dtype=np.bool)[()]
738
+ assert_raises(TypeError, operator.sub, a, a)
739
+
740
+ def test_result(self):
741
+ types = np.typecodes['AllInteger'] + np.typecodes['AllFloat']
742
+ with warnings.catch_warnings():
743
+ warnings.simplefilter('ignore', RuntimeWarning)
744
+ for dt in types:
745
+ a = np.ones((), dtype=dt)[()]
746
+ assert_equal(operator.sub(a, a), 0)
747
+
748
+
749
+ class TestAbs:
750
+ def _test_abs_func(self, absfunc, test_dtype):
751
+ x = test_dtype(-1.5)
752
+ assert_equal(absfunc(x), 1.5)
753
+ x = test_dtype(0.0)
754
+ res = absfunc(x)
755
+ # assert_equal() checks zero signedness
756
+ assert_equal(res, 0.0)
757
+ x = test_dtype(-0.0)
758
+ res = absfunc(x)
759
+ assert_equal(res, 0.0)
760
+
761
+ x = test_dtype(np.finfo(test_dtype).max)
762
+ assert_equal(absfunc(x), x.real)
763
+
764
+ with warnings.catch_warnings():
765
+ warnings.simplefilter('ignore', UserWarning)
766
+ x = test_dtype(np.finfo(test_dtype).tiny)
767
+ assert_equal(absfunc(x), x.real)
768
+
769
+ x = test_dtype(np.finfo(test_dtype).min)
770
+ assert_equal(absfunc(x), -x.real)
771
+
772
+ @pytest.mark.parametrize("dtype", floating_types + complex_floating_types)
773
+ def test_builtin_abs(self, dtype):
774
+ if (
775
+ sys.platform == "cygwin" and dtype == np.clongdouble and
776
+ (
777
+ _pep440.parse(platform.release().split("-")[0])
778
+ < _pep440.Version("3.3.0")
779
+ )
780
+ ):
781
+ pytest.xfail(
782
+ reason="absl is computed in double precision on cygwin < 3.3"
783
+ )
784
+ self._test_abs_func(abs, dtype)
785
+
786
+ @pytest.mark.parametrize("dtype", floating_types + complex_floating_types)
787
+ def test_numpy_abs(self, dtype):
788
+ if (
789
+ sys.platform == "cygwin" and dtype == np.clongdouble and
790
+ (
791
+ _pep440.parse(platform.release().split("-")[0])
792
+ < _pep440.Version("3.3.0")
793
+ )
794
+ ):
795
+ pytest.xfail(
796
+ reason="absl is computed in double precision on cygwin < 3.3"
797
+ )
798
+ self._test_abs_func(np.abs, dtype)
799
+
800
+ class TestBitShifts:
801
+
802
+ @pytest.mark.parametrize('type_code', np.typecodes['AllInteger'])
803
+ @pytest.mark.parametrize('op',
804
+ [operator.rshift, operator.lshift], ids=['>>', '<<'])
805
+ def test_shift_all_bits(self, type_code, op):
806
+ """Shifts where the shift amount is the width of the type or wider """
807
+ # gh-2449
808
+ dt = np.dtype(type_code)
809
+ nbits = dt.itemsize * 8
810
+ for val in [5, -5]:
811
+ for shift in [nbits, nbits + 4]:
812
+ val_scl = np.array(val).astype(dt)[()]
813
+ shift_scl = dt.type(shift)
814
+ res_scl = op(val_scl, shift_scl)
815
+ if val_scl < 0 and op is operator.rshift:
816
+ # sign bit is preserved
817
+ assert_equal(res_scl, -1)
818
+ else:
819
+ assert_equal(res_scl, 0)
820
+
821
+ # Result on scalars should be the same as on arrays
822
+ val_arr = np.array([val_scl] * 32, dtype=dt)
823
+ shift_arr = np.array([shift] * 32, dtype=dt)
824
+ res_arr = op(val_arr, shift_arr)
825
+ assert_equal(res_arr, res_scl)
826
+
827
+
828
+ class TestHash:
829
+ @pytest.mark.parametrize("type_code", np.typecodes['AllInteger'])
830
+ def test_integer_hashes(self, type_code):
831
+ scalar = np.dtype(type_code).type
832
+ for i in range(128):
833
+ assert hash(i) == hash(scalar(i))
834
+
835
+ @pytest.mark.parametrize("type_code", np.typecodes['AllFloat'])
836
+ def test_float_and_complex_hashes(self, type_code):
837
+ scalar = np.dtype(type_code).type
838
+ for val in [np.pi, np.inf, 3, 6.]:
839
+ numpy_val = scalar(val)
840
+ # Cast back to Python, in case the NumPy scalar has less precision
841
+ if numpy_val.dtype.kind == 'c':
842
+ val = complex(numpy_val)
843
+ else:
844
+ val = float(numpy_val)
845
+ assert val == numpy_val
846
+ assert hash(val) == hash(numpy_val)
847
+
848
+ if hash(float(np.nan)) != hash(float(np.nan)):
849
+ # If Python distinguishes different NaNs we do so too (gh-18833)
850
+ assert hash(scalar(np.nan)) != hash(scalar(np.nan))
851
+
852
+ @pytest.mark.parametrize("type_code", np.typecodes['Complex'])
853
+ def test_complex_hashes(self, type_code):
854
+ # Test some complex valued hashes specifically:
855
+ scalar = np.dtype(type_code).type
856
+ for val in [np.pi + 1j, np.inf - 3j, 3j, 6. + 1j]:
857
+ numpy_val = scalar(val)
858
+ assert hash(complex(numpy_val)) == hash(numpy_val)
859
+
860
+
861
+ @contextlib.contextmanager
862
+ def recursionlimit(n):
863
+ o = sys.getrecursionlimit()
864
+ try:
865
+ sys.setrecursionlimit(n)
866
+ yield
867
+ finally:
868
+ sys.setrecursionlimit(o)
869
+
870
+
871
+ @given(sampled_from(objecty_things),
872
+ sampled_from(binary_operators_for_scalar_ints),
873
+ sampled_from(types + [rational]))
874
+ @pytest.mark.thread_unsafe(reason="sets recursion limit globally")
875
+ def test_operator_object_left(o, op, type_):
876
+ try:
877
+ with recursionlimit(200):
878
+ op(o, type_(1))
879
+ except TypeError:
880
+ pass
881
+
882
+
883
+ @given(sampled_from(objecty_things),
884
+ sampled_from(binary_operators_for_scalar_ints),
885
+ sampled_from(types + [rational]))
886
+ @pytest.mark.thread_unsafe(reason="sets recursion limit globally")
887
+ def test_operator_object_right(o, op, type_):
888
+ try:
889
+ with recursionlimit(200):
890
+ op(type_(1), o)
891
+ except TypeError:
892
+ pass
893
+
894
+
895
+ @given(sampled_from(binary_operators_for_scalars),
896
+ sampled_from(types),
897
+ sampled_from(types))
898
+ def test_operator_scalars(op, type1, type2):
899
+ try:
900
+ op(type1(1), type2(1))
901
+ except TypeError:
902
+ pass
903
+
904
+
905
+ @pytest.mark.parametrize("op", binary_operators_for_scalars)
906
+ @pytest.mark.parametrize("sctype", [np.longdouble, np.clongdouble])
907
+ def test_longdouble_operators_with_obj(sctype, op):
908
+ # This is/used to be tricky, because NumPy generally falls back to
909
+ # using the ufunc via `np.asarray()`, this effectively might do:
910
+ # longdouble + None
911
+ # -> asarray(longdouble) + np.array(None, dtype=object)
912
+ # -> asarray(longdouble).astype(object) + np.array(None, dtype=object)
913
+ # And after getting the scalars in the inner loop:
914
+ # -> longdouble + None
915
+ #
916
+ # That would recurse infinitely. Other scalars return the python object
917
+ # on cast, so this type of things works OK.
918
+ #
919
+ # As of NumPy 2.1, this has been consolidated into the np.generic binops
920
+ # and now checks `.item()`. That also allows the below path to work now.
921
+ try:
922
+ op(sctype(3), None)
923
+ except TypeError:
924
+ pass
925
+ try:
926
+ op(None, sctype(3))
927
+ except TypeError:
928
+ pass
929
+
930
+
931
+ @pytest.mark.parametrize("op", [operator.add, operator.pow, operator.sub])
932
+ @pytest.mark.parametrize("sctype", [np.longdouble, np.clongdouble])
933
+ def test_longdouble_with_arrlike(sctype, op):
934
+ # As of NumPy 2.1, longdouble behaves like other types and can coerce
935
+ # e.g. lists. (Not necessarily better, but consistent.)
936
+ assert_array_equal(op(sctype(3), [1, 2]), op(3, np.array([1, 2])))
937
+ assert_array_equal(op([1, 2], sctype(3)), op(np.array([1, 2]), 3))
938
+
939
+
940
+ @pytest.mark.parametrize("op", binary_operators_for_scalars)
941
+ @pytest.mark.parametrize("sctype", [np.longdouble, np.clongdouble])
942
+ @np.errstate(all="ignore")
943
+ def test_longdouble_operators_with_large_int(sctype, op):
944
+ # (See `test_longdouble_operators_with_obj` for why longdouble is special)
945
+ # NEP 50 means that the result is clearly a (c)longdouble here:
946
+ if sctype == np.clongdouble and op in [operator.mod, operator.floordiv]:
947
+ # The above operators are not support for complex though...
948
+ with pytest.raises(TypeError):
949
+ op(sctype(3), 2**64)
950
+ with pytest.raises(TypeError):
951
+ op(sctype(3), 2**64)
952
+ else:
953
+ assert op(sctype(3), -2**64) == op(sctype(3), sctype(-2**64))
954
+ assert op(2**64, sctype(3)) == op(sctype(2**64), sctype(3))
955
+
956
+
957
+ @pytest.mark.parametrize("dtype", np.typecodes["AllInteger"])
958
+ @pytest.mark.parametrize("operation", [
959
+ lambda min, max: max + max,
960
+ lambda min, max: min - max,
961
+ lambda min, max: max * max], ids=["+", "-", "*"])
962
+ def test_scalar_integer_operation_overflow(dtype, operation):
963
+ st = np.dtype(dtype).type
964
+ min = st(np.iinfo(dtype).min)
965
+ max = st(np.iinfo(dtype).max)
966
+
967
+ with pytest.warns(RuntimeWarning, match="overflow encountered"):
968
+ operation(min, max)
969
+
970
+
971
+ @pytest.mark.parametrize("dtype", np.typecodes["Integer"])
972
+ @pytest.mark.parametrize("operation", [
973
+ lambda min, neg_1: -min,
974
+ lambda min, neg_1: abs(min),
975
+ lambda min, neg_1: min * neg_1,
976
+ pytest.param(lambda min, neg_1: min // neg_1,
977
+ marks=pytest.mark.skip(reason="broken on some platforms"))],
978
+ ids=["neg", "abs", "*", "//"])
979
+ def test_scalar_signed_integer_overflow(dtype, operation):
980
+ # The minimum signed integer can "overflow" for some additional operations
981
+ st = np.dtype(dtype).type
982
+ min = st(np.iinfo(dtype).min)
983
+ neg_1 = st(-1)
984
+
985
+ with pytest.warns(RuntimeWarning, match="overflow encountered"):
986
+ operation(min, neg_1)
987
+
988
+
989
+ @pytest.mark.parametrize("dtype", np.typecodes["UnsignedInteger"])
990
+ def test_scalar_unsigned_integer_overflow(dtype):
991
+ val = np.dtype(dtype).type(8)
992
+ with pytest.warns(RuntimeWarning, match="overflow encountered"):
993
+ -val
994
+
995
+ zero = np.dtype(dtype).type(0)
996
+ -zero # does not warn
997
+
998
+ @pytest.mark.parametrize("dtype", np.typecodes["AllInteger"])
999
+ @pytest.mark.parametrize("operation", [
1000
+ lambda val, zero: val // zero,
1001
+ lambda val, zero: val % zero, ], ids=["//", "%"])
1002
+ def test_scalar_integer_operation_divbyzero(dtype, operation):
1003
+ st = np.dtype(dtype).type
1004
+ val = st(100)
1005
+ zero = st(0)
1006
+
1007
+ with pytest.warns(RuntimeWarning, match="divide by zero"):
1008
+ operation(val, zero)
1009
+
1010
+
1011
+ ops_with_names = [
1012
+ ("__lt__", "__gt__", operator.lt, True),
1013
+ ("__le__", "__ge__", operator.le, True),
1014
+ ("__eq__", "__eq__", operator.eq, True),
1015
+ # Note __op__ and __rop__ may be identical here:
1016
+ ("__ne__", "__ne__", operator.ne, True),
1017
+ ("__gt__", "__lt__", operator.gt, True),
1018
+ ("__ge__", "__le__", operator.ge, True),
1019
+ ("__floordiv__", "__rfloordiv__", operator.floordiv, False),
1020
+ ("__truediv__", "__rtruediv__", operator.truediv, False),
1021
+ ("__add__", "__radd__", operator.add, False),
1022
+ ("__mod__", "__rmod__", operator.mod, False),
1023
+ ("__mul__", "__rmul__", operator.mul, False),
1024
+ ("__pow__", "__rpow__", operator.pow, False),
1025
+ ("__sub__", "__rsub__", operator.sub, False),
1026
+ ]
1027
+
1028
+
1029
+ @pytest.mark.parametrize(["__op__", "__rop__", "op", "cmp"], ops_with_names)
1030
+ @pytest.mark.parametrize("sctype", [np.float32, np.float64, np.longdouble])
1031
+ def test_subclass_deferral(sctype, __op__, __rop__, op, cmp):
1032
+ """
1033
+ This test covers scalar subclass deferral. Note that this is exceedingly
1034
+ complicated, especially since it tends to fall back to the array paths and
1035
+ these additionally add the "array priority" mechanism.
1036
+
1037
+ The behaviour was modified subtly in 1.22 (to make it closer to how Python
1038
+ scalars work). Due to its complexity and the fact that subclassing NumPy
1039
+ scalars is probably a bad idea to begin with. There is probably room
1040
+ for adjustments here.
1041
+ """
1042
+ class myf_simple1(sctype):
1043
+ pass
1044
+
1045
+ class myf_simple2(sctype):
1046
+ pass
1047
+
1048
+ def op_func(self, other):
1049
+ return __op__
1050
+
1051
+ def rop_func(self, other):
1052
+ return __rop__
1053
+
1054
+ myf_op = type("myf_op", (sctype,), {__op__: op_func, __rop__: rop_func})
1055
+
1056
+ # inheritance has to override, or this is correctly lost:
1057
+ res = op(myf_simple1(1), myf_simple2(2))
1058
+ assert type(res) == sctype or type(res) == np.bool
1059
+ assert op(myf_simple1(1), myf_simple2(2)) == op(1, 2) # inherited
1060
+
1061
+ # Two independent subclasses do not really define an order. This could
1062
+ # be attempted, but we do not since Python's `int` does neither:
1063
+ assert op(myf_op(1), myf_simple1(2)) == __op__
1064
+ assert op(myf_simple1(1), myf_op(2)) == op(1, 2) # inherited
1065
+
1066
+
1067
+ def test_longdouble_complex():
1068
+ # Simple test to check longdouble and complex combinations, since these
1069
+ # need to go through promotion, which longdouble needs to be careful about.
1070
+ x = np.longdouble(1)
1071
+ assert x + 1j == 1 + 1j
1072
+ assert 1j + x == 1 + 1j
1073
+
1074
+
1075
+ @pytest.mark.parametrize(["__op__", "__rop__", "op", "cmp"], ops_with_names)
1076
+ @pytest.mark.parametrize("subtype", [float, int, complex, np.float16])
1077
+ def test_pyscalar_subclasses(subtype, __op__, __rop__, op, cmp):
1078
+ # This tests that python scalar subclasses behave like a float64 (if they
1079
+ # don't override it).
1080
+ # In an earlier version of NEP 50, they behaved like the Python buildins.
1081
+ def op_func(self, other):
1082
+ return __op__
1083
+
1084
+ def rop_func(self, other):
1085
+ return __rop__
1086
+
1087
+ # Check that deferring is indicated using `__array_ufunc__`:
1088
+ myt = type("myt", (subtype,),
1089
+ {__op__: op_func, __rop__: rop_func, "__array_ufunc__": None})
1090
+
1091
+ # Just like normally, we should never presume we can modify the float.
1092
+ assert op(myt(1), np.float64(2)) == __op__
1093
+ assert op(np.float64(1), myt(2)) == __rop__
1094
+
1095
+ if op in {operator.mod, operator.floordiv} and subtype == complex:
1096
+ return # module is not support for complex. Do not test.
1097
+
1098
+ if __rop__ == __op__:
1099
+ return
1100
+
1101
+ # When no deferring is indicated, subclasses are handled normally.
1102
+ myt = type("myt", (subtype,), {__rop__: rop_func})
1103
+ behaves_like = lambda x: np.array(subtype(x))[()]
1104
+
1105
+ # Check for float32, as a float subclass float64 may behave differently
1106
+ res = op(myt(1), np.float16(2))
1107
+ expected = op(behaves_like(1), np.float16(2))
1108
+ assert res == expected
1109
+ assert type(res) == type(expected)
1110
+ res = op(np.float32(2), myt(1))
1111
+ expected = op(np.float32(2), behaves_like(1))
1112
+ assert res == expected
1113
+ assert type(res) == type(expected)
1114
+
1115
+ # Same check for longdouble (compare via dtype to accept float64 when
1116
+ # longdouble has the identical size), which is currently not perfectly
1117
+ # consistent.
1118
+ res = op(myt(1), np.longdouble(2))
1119
+ expected = op(behaves_like(1), np.longdouble(2))
1120
+ assert res == expected
1121
+ assert np.dtype(type(res)) == np.dtype(type(expected))
1122
+ res = op(np.float32(2), myt(1))
1123
+ expected = op(np.float32(2), behaves_like(1))
1124
+ assert res == expected
1125
+ assert np.dtype(type(res)) == np.dtype(type(expected))
1126
+
1127
+
1128
+ def test_truediv_int():
1129
+ # This should work, as the result is float:
1130
+ assert np.uint8(3) / 123454 == np.float64(3) / 123454
1131
+
1132
+
1133
+ @pytest.mark.slow
1134
+ @pytest.mark.parametrize("op",
1135
+ # TODO: Power is a bit special, but here mostly bools seem to behave oddly
1136
+ [op for op in binary_operators_for_scalars if op is not operator.pow])
1137
+ @pytest.mark.parametrize("sctype", types)
1138
+ @pytest.mark.parametrize("other_type", [float, int, complex])
1139
+ @pytest.mark.parametrize("rop", [True, False])
1140
+ def test_scalar_matches_array_op_with_pyscalar(op, sctype, other_type, rop):
1141
+ # Check that the ufunc path matches by coercing to an array explicitly
1142
+ val1 = sctype(2)
1143
+ val2 = other_type(2)
1144
+
1145
+ if rop:
1146
+ _op = op
1147
+ op = lambda x, y: _op(y, x)
1148
+
1149
+ try:
1150
+ res = op(val1, val2)
1151
+ except TypeError:
1152
+ try:
1153
+ expected = op(np.asarray(val1), val2)
1154
+ raise AssertionError("ufunc didn't raise.")
1155
+ except TypeError:
1156
+ return
1157
+ else:
1158
+ expected = op(np.asarray(val1), val2)
1159
+
1160
+ # Note that we only check dtype equivalency, as ufuncs may pick the lower
1161
+ # dtype if they are equivalent.
1162
+ assert res == expected
1163
+ if isinstance(val1, float) and other_type is complex and rop:
1164
+ # Python complex accepts float subclasses, so we don't get a chance
1165
+ # and the result may be a Python complex (thus, the `np.array()``)
1166
+ assert np.array(res).dtype == expected.dtype
1167
+ else:
1168
+ assert res.dtype == expected.dtype