numpy 2.4.1__pp311-pypy311_pp73-macosx_14_0_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 (1039) 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 +942 -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.pypy311-pp73-darwin.so +0 -0
  30. numpy/_core/_multiarray_umath.pypy311-pp73-darwin.so +0 -0
  31. numpy/_core/_operand_flag_tests.pypy311-pp73-darwin.so +0 -0
  32. numpy/_core/_rational_tests.pypy311-pp73-darwin.so +0 -0
  33. numpy/_core/_simd.pyi +35 -0
  34. numpy/_core/_simd.pypy311-pp73-darwin.so +0 -0
  35. numpy/_core/_string_helpers.py +100 -0
  36. numpy/_core/_string_helpers.pyi +12 -0
  37. numpy/_core/_struct_ufunc_tests.pypy311-pp73-darwin.so +0 -0
  38. numpy/_core/_type_aliases.py +131 -0
  39. numpy/_core/_type_aliases.pyi +86 -0
  40. numpy/_core/_ufunc_config.py +515 -0
  41. numpy/_core/_ufunc_config.pyi +69 -0
  42. numpy/_core/_umath_tests.pyi +47 -0
  43. numpy/_core/_umath_tests.pypy311-pp73-darwin.so +0 -0
  44. numpy/_core/arrayprint.py +1779 -0
  45. numpy/_core/arrayprint.pyi +158 -0
  46. numpy/_core/cversions.py +13 -0
  47. numpy/_core/defchararray.py +1414 -0
  48. numpy/_core/defchararray.pyi +1150 -0
  49. numpy/_core/einsumfunc.py +1650 -0
  50. numpy/_core/einsumfunc.pyi +184 -0
  51. numpy/_core/fromnumeric.py +4233 -0
  52. numpy/_core/fromnumeric.pyi +1735 -0
  53. numpy/_core/function_base.py +547 -0
  54. numpy/_core/function_base.pyi +276 -0
  55. numpy/_core/getlimits.py +462 -0
  56. numpy/_core/getlimits.pyi +124 -0
  57. numpy/_core/include/numpy/__multiarray_api.c +376 -0
  58. numpy/_core/include/numpy/__multiarray_api.h +1628 -0
  59. numpy/_core/include/numpy/__ufunc_api.c +55 -0
  60. numpy/_core/include/numpy/__ufunc_api.h +349 -0
  61. numpy/_core/include/numpy/_neighborhood_iterator_imp.h +90 -0
  62. numpy/_core/include/numpy/_numpyconfig.h +33 -0
  63. numpy/_core/include/numpy/_public_dtype_api_table.h +86 -0
  64. numpy/_core/include/numpy/arrayobject.h +7 -0
  65. numpy/_core/include/numpy/arrayscalars.h +198 -0
  66. numpy/_core/include/numpy/dtype_api.h +547 -0
  67. numpy/_core/include/numpy/halffloat.h +70 -0
  68. numpy/_core/include/numpy/ndarrayobject.h +304 -0
  69. numpy/_core/include/numpy/ndarraytypes.h +1982 -0
  70. numpy/_core/include/numpy/npy_2_compat.h +249 -0
  71. numpy/_core/include/numpy/npy_2_complexcompat.h +28 -0
  72. numpy/_core/include/numpy/npy_3kcompat.h +374 -0
  73. numpy/_core/include/numpy/npy_common.h +989 -0
  74. numpy/_core/include/numpy/npy_cpu.h +126 -0
  75. numpy/_core/include/numpy/npy_endian.h +79 -0
  76. numpy/_core/include/numpy/npy_math.h +602 -0
  77. numpy/_core/include/numpy/npy_no_deprecated_api.h +20 -0
  78. numpy/_core/include/numpy/npy_os.h +42 -0
  79. numpy/_core/include/numpy/numpyconfig.h +185 -0
  80. numpy/_core/include/numpy/random/LICENSE.txt +21 -0
  81. numpy/_core/include/numpy/random/bitgen.h +20 -0
  82. numpy/_core/include/numpy/random/distributions.h +209 -0
  83. numpy/_core/include/numpy/random/libdivide.h +2079 -0
  84. numpy/_core/include/numpy/ufuncobject.h +343 -0
  85. numpy/_core/include/numpy/utils.h +37 -0
  86. numpy/_core/lib/libnpymath.a +0 -0
  87. numpy/_core/lib/npy-pkg-config/mlib.ini +12 -0
  88. numpy/_core/lib/npy-pkg-config/npymath.ini +20 -0
  89. numpy/_core/lib/pkgconfig/numpy.pc +7 -0
  90. numpy/_core/memmap.py +363 -0
  91. numpy/_core/memmap.pyi +3 -0
  92. numpy/_core/multiarray.py +1740 -0
  93. numpy/_core/multiarray.pyi +1316 -0
  94. numpy/_core/numeric.py +2758 -0
  95. numpy/_core/numeric.pyi +1276 -0
  96. numpy/_core/numerictypes.py +633 -0
  97. numpy/_core/numerictypes.pyi +196 -0
  98. numpy/_core/overrides.py +188 -0
  99. numpy/_core/overrides.pyi +47 -0
  100. numpy/_core/printoptions.py +32 -0
  101. numpy/_core/printoptions.pyi +28 -0
  102. numpy/_core/records.py +1088 -0
  103. numpy/_core/records.pyi +340 -0
  104. numpy/_core/shape_base.py +996 -0
  105. numpy/_core/shape_base.pyi +182 -0
  106. numpy/_core/strings.py +1813 -0
  107. numpy/_core/strings.pyi +536 -0
  108. numpy/_core/tests/_locales.py +72 -0
  109. numpy/_core/tests/_natype.py +144 -0
  110. numpy/_core/tests/data/astype_copy.pkl +0 -0
  111. numpy/_core/tests/data/generate_umath_validation_data.cpp +170 -0
  112. numpy/_core/tests/data/recarray_from_file.fits +0 -0
  113. numpy/_core/tests/data/umath-validation-set-README.txt +15 -0
  114. numpy/_core/tests/data/umath-validation-set-arccos.csv +1429 -0
  115. numpy/_core/tests/data/umath-validation-set-arccosh.csv +1429 -0
  116. numpy/_core/tests/data/umath-validation-set-arcsin.csv +1429 -0
  117. numpy/_core/tests/data/umath-validation-set-arcsinh.csv +1429 -0
  118. numpy/_core/tests/data/umath-validation-set-arctan.csv +1429 -0
  119. numpy/_core/tests/data/umath-validation-set-arctanh.csv +1429 -0
  120. numpy/_core/tests/data/umath-validation-set-cbrt.csv +1429 -0
  121. numpy/_core/tests/data/umath-validation-set-cos.csv +1375 -0
  122. numpy/_core/tests/data/umath-validation-set-cosh.csv +1429 -0
  123. numpy/_core/tests/data/umath-validation-set-exp.csv +412 -0
  124. numpy/_core/tests/data/umath-validation-set-exp2.csv +1429 -0
  125. numpy/_core/tests/data/umath-validation-set-expm1.csv +1429 -0
  126. numpy/_core/tests/data/umath-validation-set-log.csv +271 -0
  127. numpy/_core/tests/data/umath-validation-set-log10.csv +1629 -0
  128. numpy/_core/tests/data/umath-validation-set-log1p.csv +1429 -0
  129. numpy/_core/tests/data/umath-validation-set-log2.csv +1629 -0
  130. numpy/_core/tests/data/umath-validation-set-sin.csv +1370 -0
  131. numpy/_core/tests/data/umath-validation-set-sinh.csv +1429 -0
  132. numpy/_core/tests/data/umath-validation-set-tan.csv +1429 -0
  133. numpy/_core/tests/data/umath-validation-set-tanh.csv +1429 -0
  134. numpy/_core/tests/examples/cython/checks.pyx +373 -0
  135. numpy/_core/tests/examples/cython/meson.build +43 -0
  136. numpy/_core/tests/examples/cython/setup.py +39 -0
  137. numpy/_core/tests/examples/limited_api/limited_api1.c +17 -0
  138. numpy/_core/tests/examples/limited_api/limited_api2.pyx +11 -0
  139. numpy/_core/tests/examples/limited_api/limited_api_latest.c +19 -0
  140. numpy/_core/tests/examples/limited_api/meson.build +59 -0
  141. numpy/_core/tests/examples/limited_api/setup.py +24 -0
  142. numpy/_core/tests/test__exceptions.py +90 -0
  143. numpy/_core/tests/test_abc.py +54 -0
  144. numpy/_core/tests/test_api.py +655 -0
  145. numpy/_core/tests/test_argparse.py +90 -0
  146. numpy/_core/tests/test_array_api_info.py +113 -0
  147. numpy/_core/tests/test_array_coercion.py +928 -0
  148. numpy/_core/tests/test_array_interface.py +222 -0
  149. numpy/_core/tests/test_arraymethod.py +84 -0
  150. numpy/_core/tests/test_arrayobject.py +75 -0
  151. numpy/_core/tests/test_arrayprint.py +1324 -0
  152. numpy/_core/tests/test_casting_floatingpoint_errors.py +154 -0
  153. numpy/_core/tests/test_casting_unittests.py +955 -0
  154. numpy/_core/tests/test_conversion_utils.py +209 -0
  155. numpy/_core/tests/test_cpu_dispatcher.py +48 -0
  156. numpy/_core/tests/test_cpu_features.py +450 -0
  157. numpy/_core/tests/test_custom_dtypes.py +393 -0
  158. numpy/_core/tests/test_cython.py +352 -0
  159. numpy/_core/tests/test_datetime.py +2792 -0
  160. numpy/_core/tests/test_defchararray.py +858 -0
  161. numpy/_core/tests/test_deprecations.py +460 -0
  162. numpy/_core/tests/test_dlpack.py +190 -0
  163. numpy/_core/tests/test_dtype.py +2110 -0
  164. numpy/_core/tests/test_einsum.py +1351 -0
  165. numpy/_core/tests/test_errstate.py +131 -0
  166. numpy/_core/tests/test_extint128.py +217 -0
  167. numpy/_core/tests/test_finfo.py +86 -0
  168. numpy/_core/tests/test_function_base.py +504 -0
  169. numpy/_core/tests/test_getlimits.py +171 -0
  170. numpy/_core/tests/test_half.py +593 -0
  171. numpy/_core/tests/test_hashtable.py +36 -0
  172. numpy/_core/tests/test_indexerrors.py +122 -0
  173. numpy/_core/tests/test_indexing.py +1692 -0
  174. numpy/_core/tests/test_item_selection.py +167 -0
  175. numpy/_core/tests/test_limited_api.py +102 -0
  176. numpy/_core/tests/test_longdouble.py +370 -0
  177. numpy/_core/tests/test_mem_overlap.py +933 -0
  178. numpy/_core/tests/test_mem_policy.py +453 -0
  179. numpy/_core/tests/test_memmap.py +248 -0
  180. numpy/_core/tests/test_multiarray.py +11008 -0
  181. numpy/_core/tests/test_multiprocessing.py +55 -0
  182. numpy/_core/tests/test_multithreading.py +377 -0
  183. numpy/_core/tests/test_nditer.py +3533 -0
  184. numpy/_core/tests/test_nep50_promotions.py +287 -0
  185. numpy/_core/tests/test_numeric.py +4295 -0
  186. numpy/_core/tests/test_numerictypes.py +650 -0
  187. numpy/_core/tests/test_overrides.py +800 -0
  188. numpy/_core/tests/test_print.py +202 -0
  189. numpy/_core/tests/test_protocols.py +46 -0
  190. numpy/_core/tests/test_records.py +544 -0
  191. numpy/_core/tests/test_regression.py +2677 -0
  192. numpy/_core/tests/test_scalar_ctors.py +203 -0
  193. numpy/_core/tests/test_scalar_methods.py +328 -0
  194. numpy/_core/tests/test_scalarbuffer.py +153 -0
  195. numpy/_core/tests/test_scalarinherit.py +105 -0
  196. numpy/_core/tests/test_scalarmath.py +1168 -0
  197. numpy/_core/tests/test_scalarprint.py +403 -0
  198. numpy/_core/tests/test_shape_base.py +904 -0
  199. numpy/_core/tests/test_simd.py +1345 -0
  200. numpy/_core/tests/test_simd_module.py +105 -0
  201. numpy/_core/tests/test_stringdtype.py +1855 -0
  202. numpy/_core/tests/test_strings.py +1523 -0
  203. numpy/_core/tests/test_ufunc.py +3405 -0
  204. numpy/_core/tests/test_umath.py +4962 -0
  205. numpy/_core/tests/test_umath_accuracy.py +132 -0
  206. numpy/_core/tests/test_umath_complex.py +631 -0
  207. numpy/_core/tests/test_unicode.py +369 -0
  208. numpy/_core/umath.py +60 -0
  209. numpy/_core/umath.pyi +232 -0
  210. numpy/_distributor_init.py +15 -0
  211. numpy/_distributor_init.pyi +1 -0
  212. numpy/_expired_attrs_2_0.py +78 -0
  213. numpy/_expired_attrs_2_0.pyi +61 -0
  214. numpy/_globals.py +121 -0
  215. numpy/_globals.pyi +17 -0
  216. numpy/_pyinstaller/__init__.py +0 -0
  217. numpy/_pyinstaller/__init__.pyi +0 -0
  218. numpy/_pyinstaller/hook-numpy.py +36 -0
  219. numpy/_pyinstaller/hook-numpy.pyi +6 -0
  220. numpy/_pyinstaller/tests/__init__.py +16 -0
  221. numpy/_pyinstaller/tests/pyinstaller-smoke.py +32 -0
  222. numpy/_pyinstaller/tests/test_pyinstaller.py +35 -0
  223. numpy/_pytesttester.py +201 -0
  224. numpy/_pytesttester.pyi +18 -0
  225. numpy/_typing/__init__.py +173 -0
  226. numpy/_typing/_add_docstring.py +153 -0
  227. numpy/_typing/_array_like.py +106 -0
  228. numpy/_typing/_char_codes.py +213 -0
  229. numpy/_typing/_dtype_like.py +114 -0
  230. numpy/_typing/_extended_precision.py +15 -0
  231. numpy/_typing/_nbit.py +19 -0
  232. numpy/_typing/_nbit_base.py +94 -0
  233. numpy/_typing/_nbit_base.pyi +39 -0
  234. numpy/_typing/_nested_sequence.py +79 -0
  235. numpy/_typing/_scalars.py +20 -0
  236. numpy/_typing/_shape.py +8 -0
  237. numpy/_typing/_ufunc.py +7 -0
  238. numpy/_typing/_ufunc.pyi +975 -0
  239. numpy/_utils/__init__.py +95 -0
  240. numpy/_utils/__init__.pyi +28 -0
  241. numpy/_utils/_convertions.py +18 -0
  242. numpy/_utils/_convertions.pyi +4 -0
  243. numpy/_utils/_inspect.py +192 -0
  244. numpy/_utils/_inspect.pyi +70 -0
  245. numpy/_utils/_pep440.py +486 -0
  246. numpy/_utils/_pep440.pyi +118 -0
  247. numpy/char/__init__.py +2 -0
  248. numpy/char/__init__.pyi +111 -0
  249. numpy/conftest.py +248 -0
  250. numpy/core/__init__.py +33 -0
  251. numpy/core/__init__.pyi +0 -0
  252. numpy/core/_dtype.py +10 -0
  253. numpy/core/_dtype.pyi +0 -0
  254. numpy/core/_dtype_ctypes.py +10 -0
  255. numpy/core/_dtype_ctypes.pyi +0 -0
  256. numpy/core/_internal.py +27 -0
  257. numpy/core/_multiarray_umath.py +57 -0
  258. numpy/core/_utils.py +21 -0
  259. numpy/core/arrayprint.py +10 -0
  260. numpy/core/defchararray.py +10 -0
  261. numpy/core/einsumfunc.py +10 -0
  262. numpy/core/fromnumeric.py +10 -0
  263. numpy/core/function_base.py +10 -0
  264. numpy/core/getlimits.py +10 -0
  265. numpy/core/multiarray.py +25 -0
  266. numpy/core/numeric.py +12 -0
  267. numpy/core/numerictypes.py +10 -0
  268. numpy/core/overrides.py +10 -0
  269. numpy/core/overrides.pyi +7 -0
  270. numpy/core/records.py +10 -0
  271. numpy/core/shape_base.py +10 -0
  272. numpy/core/umath.py +10 -0
  273. numpy/ctypeslib/__init__.py +13 -0
  274. numpy/ctypeslib/__init__.pyi +15 -0
  275. numpy/ctypeslib/_ctypeslib.py +603 -0
  276. numpy/ctypeslib/_ctypeslib.pyi +236 -0
  277. numpy/distutils/__init__.py +64 -0
  278. numpy/distutils/__init__.pyi +4 -0
  279. numpy/distutils/__pycache__/conv_template.pypy311.pyc +0 -0
  280. numpy/distutils/_shell_utils.py +87 -0
  281. numpy/distutils/armccompiler.py +26 -0
  282. numpy/distutils/ccompiler.py +826 -0
  283. numpy/distutils/ccompiler_opt.py +2668 -0
  284. numpy/distutils/checks/cpu_asimd.c +27 -0
  285. numpy/distutils/checks/cpu_asimddp.c +16 -0
  286. numpy/distutils/checks/cpu_asimdfhm.c +19 -0
  287. numpy/distutils/checks/cpu_asimdhp.c +15 -0
  288. numpy/distutils/checks/cpu_avx.c +20 -0
  289. numpy/distutils/checks/cpu_avx2.c +20 -0
  290. numpy/distutils/checks/cpu_avx512_clx.c +22 -0
  291. numpy/distutils/checks/cpu_avx512_cnl.c +24 -0
  292. numpy/distutils/checks/cpu_avx512_icl.c +26 -0
  293. numpy/distutils/checks/cpu_avx512_knl.c +25 -0
  294. numpy/distutils/checks/cpu_avx512_knm.c +30 -0
  295. numpy/distutils/checks/cpu_avx512_skx.c +26 -0
  296. numpy/distutils/checks/cpu_avx512_spr.c +26 -0
  297. numpy/distutils/checks/cpu_avx512cd.c +20 -0
  298. numpy/distutils/checks/cpu_avx512f.c +20 -0
  299. numpy/distutils/checks/cpu_f16c.c +22 -0
  300. numpy/distutils/checks/cpu_fma3.c +22 -0
  301. numpy/distutils/checks/cpu_fma4.c +13 -0
  302. numpy/distutils/checks/cpu_lsx.c +11 -0
  303. numpy/distutils/checks/cpu_neon.c +19 -0
  304. numpy/distutils/checks/cpu_neon_fp16.c +11 -0
  305. numpy/distutils/checks/cpu_neon_vfpv4.c +21 -0
  306. numpy/distutils/checks/cpu_popcnt.c +32 -0
  307. numpy/distutils/checks/cpu_rvv.c +13 -0
  308. numpy/distutils/checks/cpu_sse.c +20 -0
  309. numpy/distutils/checks/cpu_sse2.c +20 -0
  310. numpy/distutils/checks/cpu_sse3.c +20 -0
  311. numpy/distutils/checks/cpu_sse41.c +20 -0
  312. numpy/distutils/checks/cpu_sse42.c +20 -0
  313. numpy/distutils/checks/cpu_ssse3.c +20 -0
  314. numpy/distutils/checks/cpu_sve.c +14 -0
  315. numpy/distutils/checks/cpu_vsx.c +21 -0
  316. numpy/distutils/checks/cpu_vsx2.c +13 -0
  317. numpy/distutils/checks/cpu_vsx3.c +13 -0
  318. numpy/distutils/checks/cpu_vsx4.c +14 -0
  319. numpy/distutils/checks/cpu_vx.c +16 -0
  320. numpy/distutils/checks/cpu_vxe.c +25 -0
  321. numpy/distutils/checks/cpu_vxe2.c +21 -0
  322. numpy/distutils/checks/cpu_xop.c +12 -0
  323. numpy/distutils/checks/extra_avx512bw_mask.c +18 -0
  324. numpy/distutils/checks/extra_avx512dq_mask.c +16 -0
  325. numpy/distutils/checks/extra_avx512f_reduce.c +41 -0
  326. numpy/distutils/checks/extra_vsx3_half_double.c +12 -0
  327. numpy/distutils/checks/extra_vsx4_mma.c +21 -0
  328. numpy/distutils/checks/extra_vsx_asm.c +36 -0
  329. numpy/distutils/checks/test_flags.c +1 -0
  330. numpy/distutils/command/__init__.py +41 -0
  331. numpy/distutils/command/autodist.py +148 -0
  332. numpy/distutils/command/bdist_rpm.py +22 -0
  333. numpy/distutils/command/build.py +62 -0
  334. numpy/distutils/command/build_clib.py +469 -0
  335. numpy/distutils/command/build_ext.py +752 -0
  336. numpy/distutils/command/build_py.py +31 -0
  337. numpy/distutils/command/build_scripts.py +49 -0
  338. numpy/distutils/command/build_src.py +773 -0
  339. numpy/distutils/command/config.py +516 -0
  340. numpy/distutils/command/config_compiler.py +126 -0
  341. numpy/distutils/command/develop.py +15 -0
  342. numpy/distutils/command/egg_info.py +25 -0
  343. numpy/distutils/command/install.py +79 -0
  344. numpy/distutils/command/install_clib.py +40 -0
  345. numpy/distutils/command/install_data.py +24 -0
  346. numpy/distutils/command/install_headers.py +25 -0
  347. numpy/distutils/command/sdist.py +27 -0
  348. numpy/distutils/conv_template.py +329 -0
  349. numpy/distutils/core.py +215 -0
  350. numpy/distutils/cpuinfo.py +683 -0
  351. numpy/distutils/exec_command.py +315 -0
  352. numpy/distutils/extension.py +101 -0
  353. numpy/distutils/fcompiler/__init__.py +1035 -0
  354. numpy/distutils/fcompiler/absoft.py +158 -0
  355. numpy/distutils/fcompiler/arm.py +71 -0
  356. numpy/distutils/fcompiler/compaq.py +120 -0
  357. numpy/distutils/fcompiler/environment.py +88 -0
  358. numpy/distutils/fcompiler/fujitsu.py +46 -0
  359. numpy/distutils/fcompiler/g95.py +42 -0
  360. numpy/distutils/fcompiler/gnu.py +555 -0
  361. numpy/distutils/fcompiler/hpux.py +41 -0
  362. numpy/distutils/fcompiler/ibm.py +97 -0
  363. numpy/distutils/fcompiler/intel.py +211 -0
  364. numpy/distutils/fcompiler/lahey.py +45 -0
  365. numpy/distutils/fcompiler/mips.py +54 -0
  366. numpy/distutils/fcompiler/nag.py +87 -0
  367. numpy/distutils/fcompiler/none.py +28 -0
  368. numpy/distutils/fcompiler/nv.py +53 -0
  369. numpy/distutils/fcompiler/pathf95.py +33 -0
  370. numpy/distutils/fcompiler/pg.py +128 -0
  371. numpy/distutils/fcompiler/sun.py +51 -0
  372. numpy/distutils/fcompiler/vast.py +52 -0
  373. numpy/distutils/from_template.py +261 -0
  374. numpy/distutils/fujitsuccompiler.py +28 -0
  375. numpy/distutils/intelccompiler.py +106 -0
  376. numpy/distutils/lib2def.py +116 -0
  377. numpy/distutils/line_endings.py +77 -0
  378. numpy/distutils/log.py +111 -0
  379. numpy/distutils/mingw/gfortran_vs2003_hack.c +6 -0
  380. numpy/distutils/mingw32ccompiler.py +620 -0
  381. numpy/distutils/misc_util.py +2484 -0
  382. numpy/distutils/msvc9compiler.py +63 -0
  383. numpy/distutils/msvccompiler.py +76 -0
  384. numpy/distutils/npy_pkg_config.py +441 -0
  385. numpy/distutils/numpy_distribution.py +17 -0
  386. numpy/distutils/pathccompiler.py +21 -0
  387. numpy/distutils/system_info.py +3267 -0
  388. numpy/distutils/tests/__init__.py +0 -0
  389. numpy/distutils/tests/test_build_ext.py +74 -0
  390. numpy/distutils/tests/test_ccompiler_opt.py +808 -0
  391. numpy/distutils/tests/test_ccompiler_opt_conf.py +176 -0
  392. numpy/distutils/tests/test_exec_command.py +217 -0
  393. numpy/distutils/tests/test_fcompiler.py +43 -0
  394. numpy/distutils/tests/test_fcompiler_gnu.py +55 -0
  395. numpy/distutils/tests/test_fcompiler_intel.py +30 -0
  396. numpy/distutils/tests/test_fcompiler_nagfor.py +22 -0
  397. numpy/distutils/tests/test_from_template.py +44 -0
  398. numpy/distutils/tests/test_log.py +34 -0
  399. numpy/distutils/tests/test_mingw32ccompiler.py +47 -0
  400. numpy/distutils/tests/test_misc_util.py +88 -0
  401. numpy/distutils/tests/test_npy_pkg_config.py +84 -0
  402. numpy/distutils/tests/test_shell_utils.py +79 -0
  403. numpy/distutils/tests/test_system_info.py +334 -0
  404. numpy/distutils/tests/utilities.py +90 -0
  405. numpy/distutils/unixccompiler.py +141 -0
  406. numpy/doc/ufuncs.py +138 -0
  407. numpy/dtypes.py +41 -0
  408. numpy/dtypes.pyi +630 -0
  409. numpy/exceptions.py +246 -0
  410. numpy/exceptions.pyi +27 -0
  411. numpy/f2py/__init__.py +86 -0
  412. numpy/f2py/__init__.pyi +5 -0
  413. numpy/f2py/__main__.py +5 -0
  414. numpy/f2py/__version__.py +1 -0
  415. numpy/f2py/__version__.pyi +1 -0
  416. numpy/f2py/_backends/__init__.py +9 -0
  417. numpy/f2py/_backends/__init__.pyi +5 -0
  418. numpy/f2py/_backends/_backend.py +44 -0
  419. numpy/f2py/_backends/_backend.pyi +46 -0
  420. numpy/f2py/_backends/_distutils.py +76 -0
  421. numpy/f2py/_backends/_distutils.pyi +13 -0
  422. numpy/f2py/_backends/_meson.py +244 -0
  423. numpy/f2py/_backends/_meson.pyi +62 -0
  424. numpy/f2py/_backends/meson.build.template +58 -0
  425. numpy/f2py/_isocbind.py +62 -0
  426. numpy/f2py/_isocbind.pyi +13 -0
  427. numpy/f2py/_src_pyf.py +247 -0
  428. numpy/f2py/_src_pyf.pyi +28 -0
  429. numpy/f2py/auxfuncs.py +1004 -0
  430. numpy/f2py/auxfuncs.pyi +262 -0
  431. numpy/f2py/capi_maps.py +811 -0
  432. numpy/f2py/capi_maps.pyi +33 -0
  433. numpy/f2py/cb_rules.py +665 -0
  434. numpy/f2py/cb_rules.pyi +17 -0
  435. numpy/f2py/cfuncs.py +1563 -0
  436. numpy/f2py/cfuncs.pyi +31 -0
  437. numpy/f2py/common_rules.py +143 -0
  438. numpy/f2py/common_rules.pyi +9 -0
  439. numpy/f2py/crackfortran.py +3725 -0
  440. numpy/f2py/crackfortran.pyi +266 -0
  441. numpy/f2py/diagnose.py +149 -0
  442. numpy/f2py/diagnose.pyi +1 -0
  443. numpy/f2py/f2py2e.py +788 -0
  444. numpy/f2py/f2py2e.pyi +74 -0
  445. numpy/f2py/f90mod_rules.py +269 -0
  446. numpy/f2py/f90mod_rules.pyi +16 -0
  447. numpy/f2py/func2subr.py +329 -0
  448. numpy/f2py/func2subr.pyi +7 -0
  449. numpy/f2py/rules.py +1629 -0
  450. numpy/f2py/rules.pyi +41 -0
  451. numpy/f2py/setup.cfg +3 -0
  452. numpy/f2py/src/fortranobject.c +1436 -0
  453. numpy/f2py/src/fortranobject.h +173 -0
  454. numpy/f2py/symbolic.py +1518 -0
  455. numpy/f2py/symbolic.pyi +219 -0
  456. numpy/f2py/tests/__init__.py +16 -0
  457. numpy/f2py/tests/src/abstract_interface/foo.f90 +34 -0
  458. numpy/f2py/tests/src/abstract_interface/gh18403_mod.f90 +6 -0
  459. numpy/f2py/tests/src/array_from_pyobj/wrapmodule.c +235 -0
  460. numpy/f2py/tests/src/assumed_shape/.f2py_f2cmap +1 -0
  461. numpy/f2py/tests/src/assumed_shape/foo_free.f90 +34 -0
  462. numpy/f2py/tests/src/assumed_shape/foo_mod.f90 +41 -0
  463. numpy/f2py/tests/src/assumed_shape/foo_use.f90 +19 -0
  464. numpy/f2py/tests/src/assumed_shape/precision.f90 +4 -0
  465. numpy/f2py/tests/src/block_docstring/foo.f +6 -0
  466. numpy/f2py/tests/src/callback/foo.f +62 -0
  467. numpy/f2py/tests/src/callback/gh17797.f90 +7 -0
  468. numpy/f2py/tests/src/callback/gh18335.f90 +17 -0
  469. numpy/f2py/tests/src/callback/gh25211.f +10 -0
  470. numpy/f2py/tests/src/callback/gh25211.pyf +18 -0
  471. numpy/f2py/tests/src/callback/gh26681.f90 +18 -0
  472. numpy/f2py/tests/src/cli/gh_22819.pyf +6 -0
  473. numpy/f2py/tests/src/cli/hi77.f +3 -0
  474. numpy/f2py/tests/src/cli/hiworld.f90 +3 -0
  475. numpy/f2py/tests/src/common/block.f +11 -0
  476. numpy/f2py/tests/src/common/gh19161.f90 +10 -0
  477. numpy/f2py/tests/src/crackfortran/accesstype.f90 +13 -0
  478. numpy/f2py/tests/src/crackfortran/common_with_division.f +17 -0
  479. numpy/f2py/tests/src/crackfortran/data_common.f +8 -0
  480. numpy/f2py/tests/src/crackfortran/data_multiplier.f +5 -0
  481. numpy/f2py/tests/src/crackfortran/data_stmts.f90 +20 -0
  482. numpy/f2py/tests/src/crackfortran/data_with_comments.f +8 -0
  483. numpy/f2py/tests/src/crackfortran/foo_deps.f90 +6 -0
  484. numpy/f2py/tests/src/crackfortran/gh15035.f +16 -0
  485. numpy/f2py/tests/src/crackfortran/gh17859.f +12 -0
  486. numpy/f2py/tests/src/crackfortran/gh22648.pyf +7 -0
  487. numpy/f2py/tests/src/crackfortran/gh23533.f +5 -0
  488. numpy/f2py/tests/src/crackfortran/gh23598.f90 +4 -0
  489. numpy/f2py/tests/src/crackfortran/gh23598Warn.f90 +11 -0
  490. numpy/f2py/tests/src/crackfortran/gh23879.f90 +20 -0
  491. numpy/f2py/tests/src/crackfortran/gh27697.f90 +12 -0
  492. numpy/f2py/tests/src/crackfortran/gh2848.f90 +13 -0
  493. numpy/f2py/tests/src/crackfortran/operators.f90 +49 -0
  494. numpy/f2py/tests/src/crackfortran/privatemod.f90 +11 -0
  495. numpy/f2py/tests/src/crackfortran/publicmod.f90 +10 -0
  496. numpy/f2py/tests/src/crackfortran/pubprivmod.f90 +10 -0
  497. numpy/f2py/tests/src/crackfortran/unicode_comment.f90 +4 -0
  498. numpy/f2py/tests/src/f2cmap/.f2py_f2cmap +1 -0
  499. numpy/f2py/tests/src/f2cmap/isoFortranEnvMap.f90 +9 -0
  500. numpy/f2py/tests/src/isocintrin/isoCtests.f90 +34 -0
  501. numpy/f2py/tests/src/kind/foo.f90 +20 -0
  502. numpy/f2py/tests/src/mixed/foo.f +5 -0
  503. numpy/f2py/tests/src/mixed/foo_fixed.f90 +8 -0
  504. numpy/f2py/tests/src/mixed/foo_free.f90 +8 -0
  505. numpy/f2py/tests/src/modules/gh25337/data.f90 +8 -0
  506. numpy/f2py/tests/src/modules/gh25337/use_data.f90 +6 -0
  507. numpy/f2py/tests/src/modules/gh26920/two_mods_with_no_public_entities.f90 +21 -0
  508. numpy/f2py/tests/src/modules/gh26920/two_mods_with_one_public_routine.f90 +21 -0
  509. numpy/f2py/tests/src/modules/module_data_docstring.f90 +12 -0
  510. numpy/f2py/tests/src/modules/use_modules.f90 +20 -0
  511. numpy/f2py/tests/src/negative_bounds/issue_20853.f90 +7 -0
  512. numpy/f2py/tests/src/parameter/constant_array.f90 +45 -0
  513. numpy/f2py/tests/src/parameter/constant_both.f90 +57 -0
  514. numpy/f2py/tests/src/parameter/constant_compound.f90 +15 -0
  515. numpy/f2py/tests/src/parameter/constant_integer.f90 +22 -0
  516. numpy/f2py/tests/src/parameter/constant_non_compound.f90 +23 -0
  517. numpy/f2py/tests/src/parameter/constant_real.f90 +23 -0
  518. numpy/f2py/tests/src/quoted_character/foo.f +14 -0
  519. numpy/f2py/tests/src/regression/AB.inc +1 -0
  520. numpy/f2py/tests/src/regression/assignOnlyModule.f90 +25 -0
  521. numpy/f2py/tests/src/regression/datonly.f90 +17 -0
  522. numpy/f2py/tests/src/regression/f77comments.f +26 -0
  523. numpy/f2py/tests/src/regression/f77fixedform.f95 +5 -0
  524. numpy/f2py/tests/src/regression/f90continuation.f90 +9 -0
  525. numpy/f2py/tests/src/regression/incfile.f90 +5 -0
  526. numpy/f2py/tests/src/regression/inout.f90 +9 -0
  527. numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 +5 -0
  528. numpy/f2py/tests/src/regression/mod_derived_types.f90 +23 -0
  529. numpy/f2py/tests/src/return_character/foo77.f +45 -0
  530. numpy/f2py/tests/src/return_character/foo90.f90 +48 -0
  531. numpy/f2py/tests/src/return_complex/foo77.f +45 -0
  532. numpy/f2py/tests/src/return_complex/foo90.f90 +48 -0
  533. numpy/f2py/tests/src/return_integer/foo77.f +56 -0
  534. numpy/f2py/tests/src/return_integer/foo90.f90 +59 -0
  535. numpy/f2py/tests/src/return_logical/foo77.f +56 -0
  536. numpy/f2py/tests/src/return_logical/foo90.f90 +59 -0
  537. numpy/f2py/tests/src/return_real/foo77.f +45 -0
  538. numpy/f2py/tests/src/return_real/foo90.f90 +48 -0
  539. numpy/f2py/tests/src/routines/funcfortranname.f +5 -0
  540. numpy/f2py/tests/src/routines/funcfortranname.pyf +11 -0
  541. numpy/f2py/tests/src/routines/subrout.f +4 -0
  542. numpy/f2py/tests/src/routines/subrout.pyf +10 -0
  543. numpy/f2py/tests/src/size/foo.f90 +44 -0
  544. numpy/f2py/tests/src/string/char.f90 +29 -0
  545. numpy/f2py/tests/src/string/fixed_string.f90 +34 -0
  546. numpy/f2py/tests/src/string/gh24008.f +8 -0
  547. numpy/f2py/tests/src/string/gh24662.f90 +7 -0
  548. numpy/f2py/tests/src/string/gh25286.f90 +14 -0
  549. numpy/f2py/tests/src/string/gh25286.pyf +12 -0
  550. numpy/f2py/tests/src/string/gh25286_bc.pyf +12 -0
  551. numpy/f2py/tests/src/string/scalar_string.f90 +9 -0
  552. numpy/f2py/tests/src/string/string.f +12 -0
  553. numpy/f2py/tests/src/value_attrspec/gh21665.f90 +9 -0
  554. numpy/f2py/tests/test_abstract_interface.py +26 -0
  555. numpy/f2py/tests/test_array_from_pyobj.py +678 -0
  556. numpy/f2py/tests/test_assumed_shape.py +50 -0
  557. numpy/f2py/tests/test_block_docstring.py +20 -0
  558. numpy/f2py/tests/test_callback.py +263 -0
  559. numpy/f2py/tests/test_character.py +641 -0
  560. numpy/f2py/tests/test_common.py +23 -0
  561. numpy/f2py/tests/test_crackfortran.py +421 -0
  562. numpy/f2py/tests/test_data.py +71 -0
  563. numpy/f2py/tests/test_docs.py +66 -0
  564. numpy/f2py/tests/test_f2cmap.py +17 -0
  565. numpy/f2py/tests/test_f2py2e.py +983 -0
  566. numpy/f2py/tests/test_isoc.py +56 -0
  567. numpy/f2py/tests/test_kind.py +52 -0
  568. numpy/f2py/tests/test_mixed.py +35 -0
  569. numpy/f2py/tests/test_modules.py +83 -0
  570. numpy/f2py/tests/test_parameter.py +129 -0
  571. numpy/f2py/tests/test_pyf_src.py +43 -0
  572. numpy/f2py/tests/test_quoted_character.py +18 -0
  573. numpy/f2py/tests/test_regression.py +187 -0
  574. numpy/f2py/tests/test_return_character.py +48 -0
  575. numpy/f2py/tests/test_return_complex.py +67 -0
  576. numpy/f2py/tests/test_return_integer.py +55 -0
  577. numpy/f2py/tests/test_return_logical.py +65 -0
  578. numpy/f2py/tests/test_return_real.py +109 -0
  579. numpy/f2py/tests/test_routines.py +29 -0
  580. numpy/f2py/tests/test_semicolon_split.py +75 -0
  581. numpy/f2py/tests/test_size.py +45 -0
  582. numpy/f2py/tests/test_string.py +100 -0
  583. numpy/f2py/tests/test_symbolic.py +500 -0
  584. numpy/f2py/tests/test_value_attrspec.py +15 -0
  585. numpy/f2py/tests/util.py +442 -0
  586. numpy/f2py/use_rules.py +99 -0
  587. numpy/f2py/use_rules.pyi +9 -0
  588. numpy/fft/__init__.py +213 -0
  589. numpy/fft/__init__.pyi +38 -0
  590. numpy/fft/_helper.py +235 -0
  591. numpy/fft/_helper.pyi +44 -0
  592. numpy/fft/_pocketfft.py +1693 -0
  593. numpy/fft/_pocketfft.pyi +137 -0
  594. numpy/fft/_pocketfft_umath.pypy311-pp73-darwin.so +0 -0
  595. numpy/fft/tests/__init__.py +0 -0
  596. numpy/fft/tests/test_helper.py +167 -0
  597. numpy/fft/tests/test_pocketfft.py +589 -0
  598. numpy/lib/__init__.py +97 -0
  599. numpy/lib/__init__.pyi +52 -0
  600. numpy/lib/_array_utils_impl.py +62 -0
  601. numpy/lib/_array_utils_impl.pyi +10 -0
  602. numpy/lib/_arraypad_impl.py +926 -0
  603. numpy/lib/_arraypad_impl.pyi +88 -0
  604. numpy/lib/_arraysetops_impl.py +1158 -0
  605. numpy/lib/_arraysetops_impl.pyi +462 -0
  606. numpy/lib/_arrayterator_impl.py +224 -0
  607. numpy/lib/_arrayterator_impl.pyi +45 -0
  608. numpy/lib/_datasource.py +700 -0
  609. numpy/lib/_datasource.pyi +30 -0
  610. numpy/lib/_format_impl.py +1036 -0
  611. numpy/lib/_format_impl.pyi +56 -0
  612. numpy/lib/_function_base_impl.py +5760 -0
  613. numpy/lib/_function_base_impl.pyi +2324 -0
  614. numpy/lib/_histograms_impl.py +1085 -0
  615. numpy/lib/_histograms_impl.pyi +40 -0
  616. numpy/lib/_index_tricks_impl.py +1048 -0
  617. numpy/lib/_index_tricks_impl.pyi +267 -0
  618. numpy/lib/_iotools.py +900 -0
  619. numpy/lib/_iotools.pyi +116 -0
  620. numpy/lib/_nanfunctions_impl.py +2006 -0
  621. numpy/lib/_nanfunctions_impl.pyi +48 -0
  622. numpy/lib/_npyio_impl.py +2583 -0
  623. numpy/lib/_npyio_impl.pyi +299 -0
  624. numpy/lib/_polynomial_impl.py +1465 -0
  625. numpy/lib/_polynomial_impl.pyi +338 -0
  626. numpy/lib/_scimath_impl.py +642 -0
  627. numpy/lib/_scimath_impl.pyi +93 -0
  628. numpy/lib/_shape_base_impl.py +1289 -0
  629. numpy/lib/_shape_base_impl.pyi +236 -0
  630. numpy/lib/_stride_tricks_impl.py +582 -0
  631. numpy/lib/_stride_tricks_impl.pyi +73 -0
  632. numpy/lib/_twodim_base_impl.py +1201 -0
  633. numpy/lib/_twodim_base_impl.pyi +408 -0
  634. numpy/lib/_type_check_impl.py +710 -0
  635. numpy/lib/_type_check_impl.pyi +348 -0
  636. numpy/lib/_ufunclike_impl.py +199 -0
  637. numpy/lib/_ufunclike_impl.pyi +60 -0
  638. numpy/lib/_user_array_impl.py +310 -0
  639. numpy/lib/_user_array_impl.pyi +226 -0
  640. numpy/lib/_utils_impl.py +784 -0
  641. numpy/lib/_utils_impl.pyi +22 -0
  642. numpy/lib/_version.py +153 -0
  643. numpy/lib/_version.pyi +17 -0
  644. numpy/lib/array_utils.py +7 -0
  645. numpy/lib/array_utils.pyi +6 -0
  646. numpy/lib/format.py +24 -0
  647. numpy/lib/format.pyi +24 -0
  648. numpy/lib/introspect.py +94 -0
  649. numpy/lib/introspect.pyi +3 -0
  650. numpy/lib/mixins.py +180 -0
  651. numpy/lib/mixins.pyi +78 -0
  652. numpy/lib/npyio.py +1 -0
  653. numpy/lib/npyio.pyi +5 -0
  654. numpy/lib/recfunctions.py +1681 -0
  655. numpy/lib/recfunctions.pyi +444 -0
  656. numpy/lib/scimath.py +13 -0
  657. numpy/lib/scimath.pyi +12 -0
  658. numpy/lib/stride_tricks.py +1 -0
  659. numpy/lib/stride_tricks.pyi +4 -0
  660. numpy/lib/tests/__init__.py +0 -0
  661. numpy/lib/tests/data/py2-np0-objarr.npy +0 -0
  662. numpy/lib/tests/data/py2-objarr.npy +0 -0
  663. numpy/lib/tests/data/py2-objarr.npz +0 -0
  664. numpy/lib/tests/data/py3-objarr.npy +0 -0
  665. numpy/lib/tests/data/py3-objarr.npz +0 -0
  666. numpy/lib/tests/data/python3.npy +0 -0
  667. numpy/lib/tests/data/win64python2.npy +0 -0
  668. numpy/lib/tests/test__datasource.py +328 -0
  669. numpy/lib/tests/test__iotools.py +358 -0
  670. numpy/lib/tests/test__version.py +64 -0
  671. numpy/lib/tests/test_array_utils.py +32 -0
  672. numpy/lib/tests/test_arraypad.py +1427 -0
  673. numpy/lib/tests/test_arraysetops.py +1302 -0
  674. numpy/lib/tests/test_arrayterator.py +45 -0
  675. numpy/lib/tests/test_format.py +1054 -0
  676. numpy/lib/tests/test_function_base.py +4750 -0
  677. numpy/lib/tests/test_histograms.py +855 -0
  678. numpy/lib/tests/test_index_tricks.py +693 -0
  679. numpy/lib/tests/test_io.py +2857 -0
  680. numpy/lib/tests/test_loadtxt.py +1099 -0
  681. numpy/lib/tests/test_mixins.py +215 -0
  682. numpy/lib/tests/test_nanfunctions.py +1438 -0
  683. numpy/lib/tests/test_packbits.py +376 -0
  684. numpy/lib/tests/test_polynomial.py +325 -0
  685. numpy/lib/tests/test_recfunctions.py +1042 -0
  686. numpy/lib/tests/test_regression.py +231 -0
  687. numpy/lib/tests/test_shape_base.py +813 -0
  688. numpy/lib/tests/test_stride_tricks.py +655 -0
  689. numpy/lib/tests/test_twodim_base.py +559 -0
  690. numpy/lib/tests/test_type_check.py +473 -0
  691. numpy/lib/tests/test_ufunclike.py +97 -0
  692. numpy/lib/tests/test_utils.py +80 -0
  693. numpy/lib/user_array.py +1 -0
  694. numpy/lib/user_array.pyi +1 -0
  695. numpy/linalg/__init__.py +95 -0
  696. numpy/linalg/__init__.pyi +71 -0
  697. numpy/linalg/_linalg.py +3657 -0
  698. numpy/linalg/_linalg.pyi +548 -0
  699. numpy/linalg/_umath_linalg.pyi +60 -0
  700. numpy/linalg/_umath_linalg.pypy311-pp73-darwin.so +0 -0
  701. numpy/linalg/lapack_lite.pyi +143 -0
  702. numpy/linalg/lapack_lite.pypy311-pp73-darwin.so +0 -0
  703. numpy/linalg/tests/__init__.py +0 -0
  704. numpy/linalg/tests/test_deprecations.py +21 -0
  705. numpy/linalg/tests/test_linalg.py +2442 -0
  706. numpy/linalg/tests/test_regression.py +182 -0
  707. numpy/ma/API_CHANGES.txt +135 -0
  708. numpy/ma/LICENSE +24 -0
  709. numpy/ma/README.rst +236 -0
  710. numpy/ma/__init__.py +53 -0
  711. numpy/ma/__init__.pyi +458 -0
  712. numpy/ma/core.py +8929 -0
  713. numpy/ma/core.pyi +3720 -0
  714. numpy/ma/extras.py +2266 -0
  715. numpy/ma/extras.pyi +297 -0
  716. numpy/ma/mrecords.py +762 -0
  717. numpy/ma/mrecords.pyi +96 -0
  718. numpy/ma/tests/__init__.py +0 -0
  719. numpy/ma/tests/test_arrayobject.py +40 -0
  720. numpy/ma/tests/test_core.py +6008 -0
  721. numpy/ma/tests/test_deprecations.py +65 -0
  722. numpy/ma/tests/test_extras.py +1945 -0
  723. numpy/ma/tests/test_mrecords.py +495 -0
  724. numpy/ma/tests/test_old_ma.py +939 -0
  725. numpy/ma/tests/test_regression.py +83 -0
  726. numpy/ma/tests/test_subclassing.py +469 -0
  727. numpy/ma/testutils.py +294 -0
  728. numpy/ma/testutils.pyi +69 -0
  729. numpy/matlib.py +380 -0
  730. numpy/matlib.pyi +580 -0
  731. numpy/matrixlib/__init__.py +12 -0
  732. numpy/matrixlib/__init__.pyi +3 -0
  733. numpy/matrixlib/defmatrix.py +1119 -0
  734. numpy/matrixlib/defmatrix.pyi +218 -0
  735. numpy/matrixlib/tests/__init__.py +0 -0
  736. numpy/matrixlib/tests/test_defmatrix.py +455 -0
  737. numpy/matrixlib/tests/test_interaction.py +360 -0
  738. numpy/matrixlib/tests/test_masked_matrix.py +240 -0
  739. numpy/matrixlib/tests/test_matrix_linalg.py +110 -0
  740. numpy/matrixlib/tests/test_multiarray.py +17 -0
  741. numpy/matrixlib/tests/test_numeric.py +18 -0
  742. numpy/matrixlib/tests/test_regression.py +31 -0
  743. numpy/polynomial/__init__.py +187 -0
  744. numpy/polynomial/__init__.pyi +31 -0
  745. numpy/polynomial/_polybase.py +1191 -0
  746. numpy/polynomial/_polybase.pyi +262 -0
  747. numpy/polynomial/_polytypes.pyi +501 -0
  748. numpy/polynomial/chebyshev.py +2001 -0
  749. numpy/polynomial/chebyshev.pyi +180 -0
  750. numpy/polynomial/hermite.py +1738 -0
  751. numpy/polynomial/hermite.pyi +106 -0
  752. numpy/polynomial/hermite_e.py +1640 -0
  753. numpy/polynomial/hermite_e.pyi +106 -0
  754. numpy/polynomial/laguerre.py +1673 -0
  755. numpy/polynomial/laguerre.pyi +100 -0
  756. numpy/polynomial/legendre.py +1603 -0
  757. numpy/polynomial/legendre.pyi +100 -0
  758. numpy/polynomial/polynomial.py +1625 -0
  759. numpy/polynomial/polynomial.pyi +109 -0
  760. numpy/polynomial/polyutils.py +759 -0
  761. numpy/polynomial/polyutils.pyi +307 -0
  762. numpy/polynomial/tests/__init__.py +0 -0
  763. numpy/polynomial/tests/test_chebyshev.py +618 -0
  764. numpy/polynomial/tests/test_classes.py +613 -0
  765. numpy/polynomial/tests/test_hermite.py +553 -0
  766. numpy/polynomial/tests/test_hermite_e.py +554 -0
  767. numpy/polynomial/tests/test_laguerre.py +535 -0
  768. numpy/polynomial/tests/test_legendre.py +566 -0
  769. numpy/polynomial/tests/test_polynomial.py +691 -0
  770. numpy/polynomial/tests/test_polyutils.py +123 -0
  771. numpy/polynomial/tests/test_printing.py +557 -0
  772. numpy/polynomial/tests/test_symbol.py +217 -0
  773. numpy/py.typed +0 -0
  774. numpy/random/LICENSE.md +71 -0
  775. numpy/random/__init__.pxd +14 -0
  776. numpy/random/__init__.py +213 -0
  777. numpy/random/__init__.pyi +124 -0
  778. numpy/random/_bounded_integers.pxd +29 -0
  779. numpy/random/_bounded_integers.pyi +1 -0
  780. numpy/random/_bounded_integers.pypy311-pp73-darwin.so +0 -0
  781. numpy/random/_common.pxd +110 -0
  782. numpy/random/_common.pyi +16 -0
  783. numpy/random/_common.pypy311-pp73-darwin.so +0 -0
  784. numpy/random/_examples/cffi/extending.py +44 -0
  785. numpy/random/_examples/cffi/parse.py +53 -0
  786. numpy/random/_examples/cython/extending.pyx +77 -0
  787. numpy/random/_examples/cython/extending_distributions.pyx +117 -0
  788. numpy/random/_examples/cython/meson.build +53 -0
  789. numpy/random/_examples/numba/extending.py +86 -0
  790. numpy/random/_examples/numba/extending_distributions.py +67 -0
  791. numpy/random/_generator.pyi +862 -0
  792. numpy/random/_generator.pypy311-pp73-darwin.so +0 -0
  793. numpy/random/_mt19937.pyi +27 -0
  794. numpy/random/_mt19937.pypy311-pp73-darwin.so +0 -0
  795. numpy/random/_pcg64.pyi +41 -0
  796. numpy/random/_pcg64.pypy311-pp73-darwin.so +0 -0
  797. numpy/random/_philox.pyi +36 -0
  798. numpy/random/_philox.pypy311-pp73-darwin.so +0 -0
  799. numpy/random/_pickle.py +88 -0
  800. numpy/random/_pickle.pyi +43 -0
  801. numpy/random/_sfc64.pyi +25 -0
  802. numpy/random/_sfc64.pypy311-pp73-darwin.so +0 -0
  803. numpy/random/bit_generator.pxd +40 -0
  804. numpy/random/bit_generator.pyi +123 -0
  805. numpy/random/bit_generator.pypy311-pp73-darwin.so +0 -0
  806. numpy/random/c_distributions.pxd +119 -0
  807. numpy/random/lib/libnpyrandom.a +0 -0
  808. numpy/random/mtrand.pyi +759 -0
  809. numpy/random/mtrand.pypy311-pp73-darwin.so +0 -0
  810. numpy/random/tests/__init__.py +0 -0
  811. numpy/random/tests/data/__init__.py +0 -0
  812. numpy/random/tests/data/generator_pcg64_np121.pkl.gz +0 -0
  813. numpy/random/tests/data/generator_pcg64_np126.pkl.gz +0 -0
  814. numpy/random/tests/data/mt19937-testset-1.csv +1001 -0
  815. numpy/random/tests/data/mt19937-testset-2.csv +1001 -0
  816. numpy/random/tests/data/pcg64-testset-1.csv +1001 -0
  817. numpy/random/tests/data/pcg64-testset-2.csv +1001 -0
  818. numpy/random/tests/data/pcg64dxsm-testset-1.csv +1001 -0
  819. numpy/random/tests/data/pcg64dxsm-testset-2.csv +1001 -0
  820. numpy/random/tests/data/philox-testset-1.csv +1001 -0
  821. numpy/random/tests/data/philox-testset-2.csv +1001 -0
  822. numpy/random/tests/data/sfc64-testset-1.csv +1001 -0
  823. numpy/random/tests/data/sfc64-testset-2.csv +1001 -0
  824. numpy/random/tests/data/sfc64_np126.pkl.gz +0 -0
  825. numpy/random/tests/test_direct.py +595 -0
  826. numpy/random/tests/test_extending.py +131 -0
  827. numpy/random/tests/test_generator_mt19937.py +2825 -0
  828. numpy/random/tests/test_generator_mt19937_regressions.py +221 -0
  829. numpy/random/tests/test_random.py +1724 -0
  830. numpy/random/tests/test_randomstate.py +2099 -0
  831. numpy/random/tests/test_randomstate_regression.py +213 -0
  832. numpy/random/tests/test_regression.py +175 -0
  833. numpy/random/tests/test_seed_sequence.py +79 -0
  834. numpy/random/tests/test_smoke.py +882 -0
  835. numpy/rec/__init__.py +2 -0
  836. numpy/rec/__init__.pyi +23 -0
  837. numpy/strings/__init__.py +2 -0
  838. numpy/strings/__init__.pyi +97 -0
  839. numpy/testing/__init__.py +22 -0
  840. numpy/testing/__init__.pyi +107 -0
  841. numpy/testing/_private/__init__.py +0 -0
  842. numpy/testing/_private/__init__.pyi +0 -0
  843. numpy/testing/_private/extbuild.py +250 -0
  844. numpy/testing/_private/extbuild.pyi +25 -0
  845. numpy/testing/_private/utils.py +2830 -0
  846. numpy/testing/_private/utils.pyi +505 -0
  847. numpy/testing/overrides.py +84 -0
  848. numpy/testing/overrides.pyi +10 -0
  849. numpy/testing/print_coercion_tables.py +207 -0
  850. numpy/testing/print_coercion_tables.pyi +26 -0
  851. numpy/testing/tests/__init__.py +0 -0
  852. numpy/testing/tests/test_utils.py +2123 -0
  853. numpy/tests/__init__.py +0 -0
  854. numpy/tests/test__all__.py +10 -0
  855. numpy/tests/test_configtool.py +51 -0
  856. numpy/tests/test_ctypeslib.py +383 -0
  857. numpy/tests/test_lazyloading.py +42 -0
  858. numpy/tests/test_matlib.py +59 -0
  859. numpy/tests/test_numpy_config.py +47 -0
  860. numpy/tests/test_numpy_version.py +54 -0
  861. numpy/tests/test_public_api.py +807 -0
  862. numpy/tests/test_reloading.py +76 -0
  863. numpy/tests/test_scripts.py +48 -0
  864. numpy/tests/test_warnings.py +79 -0
  865. numpy/typing/__init__.py +233 -0
  866. numpy/typing/__init__.pyi +3 -0
  867. numpy/typing/mypy_plugin.py +200 -0
  868. numpy/typing/tests/__init__.py +0 -0
  869. numpy/typing/tests/data/fail/arithmetic.pyi +126 -0
  870. numpy/typing/tests/data/fail/array_constructors.pyi +34 -0
  871. numpy/typing/tests/data/fail/array_like.pyi +15 -0
  872. numpy/typing/tests/data/fail/array_pad.pyi +6 -0
  873. numpy/typing/tests/data/fail/arrayprint.pyi +15 -0
  874. numpy/typing/tests/data/fail/arrayterator.pyi +14 -0
  875. numpy/typing/tests/data/fail/bitwise_ops.pyi +17 -0
  876. numpy/typing/tests/data/fail/char.pyi +63 -0
  877. numpy/typing/tests/data/fail/chararray.pyi +61 -0
  878. numpy/typing/tests/data/fail/comparisons.pyi +27 -0
  879. numpy/typing/tests/data/fail/constants.pyi +3 -0
  880. numpy/typing/tests/data/fail/datasource.pyi +16 -0
  881. numpy/typing/tests/data/fail/dtype.pyi +17 -0
  882. numpy/typing/tests/data/fail/einsumfunc.pyi +12 -0
  883. numpy/typing/tests/data/fail/flatiter.pyi +38 -0
  884. numpy/typing/tests/data/fail/fromnumeric.pyi +148 -0
  885. numpy/typing/tests/data/fail/histograms.pyi +12 -0
  886. numpy/typing/tests/data/fail/index_tricks.pyi +14 -0
  887. numpy/typing/tests/data/fail/lib_function_base.pyi +60 -0
  888. numpy/typing/tests/data/fail/lib_polynomial.pyi +29 -0
  889. numpy/typing/tests/data/fail/lib_utils.pyi +3 -0
  890. numpy/typing/tests/data/fail/lib_version.pyi +6 -0
  891. numpy/typing/tests/data/fail/linalg.pyi +52 -0
  892. numpy/typing/tests/data/fail/ma.pyi +155 -0
  893. numpy/typing/tests/data/fail/memmap.pyi +5 -0
  894. numpy/typing/tests/data/fail/modules.pyi +17 -0
  895. numpy/typing/tests/data/fail/multiarray.pyi +52 -0
  896. numpy/typing/tests/data/fail/ndarray.pyi +11 -0
  897. numpy/typing/tests/data/fail/ndarray_misc.pyi +49 -0
  898. numpy/typing/tests/data/fail/nditer.pyi +8 -0
  899. numpy/typing/tests/data/fail/nested_sequence.pyi +17 -0
  900. numpy/typing/tests/data/fail/npyio.pyi +24 -0
  901. numpy/typing/tests/data/fail/numerictypes.pyi +5 -0
  902. numpy/typing/tests/data/fail/random.pyi +62 -0
  903. numpy/typing/tests/data/fail/rec.pyi +17 -0
  904. numpy/typing/tests/data/fail/scalars.pyi +86 -0
  905. numpy/typing/tests/data/fail/shape.pyi +7 -0
  906. numpy/typing/tests/data/fail/shape_base.pyi +8 -0
  907. numpy/typing/tests/data/fail/stride_tricks.pyi +9 -0
  908. numpy/typing/tests/data/fail/strings.pyi +52 -0
  909. numpy/typing/tests/data/fail/testing.pyi +28 -0
  910. numpy/typing/tests/data/fail/twodim_base.pyi +39 -0
  911. numpy/typing/tests/data/fail/type_check.pyi +12 -0
  912. numpy/typing/tests/data/fail/ufunc_config.pyi +21 -0
  913. numpy/typing/tests/data/fail/ufunclike.pyi +21 -0
  914. numpy/typing/tests/data/fail/ufuncs.pyi +17 -0
  915. numpy/typing/tests/data/fail/warnings_and_errors.pyi +5 -0
  916. numpy/typing/tests/data/misc/extended_precision.pyi +9 -0
  917. numpy/typing/tests/data/mypy.ini +8 -0
  918. numpy/typing/tests/data/pass/arithmetic.py +614 -0
  919. numpy/typing/tests/data/pass/array_constructors.py +138 -0
  920. numpy/typing/tests/data/pass/array_like.py +43 -0
  921. numpy/typing/tests/data/pass/arrayprint.py +37 -0
  922. numpy/typing/tests/data/pass/arrayterator.py +28 -0
  923. numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
  924. numpy/typing/tests/data/pass/comparisons.py +316 -0
  925. numpy/typing/tests/data/pass/dtype.py +57 -0
  926. numpy/typing/tests/data/pass/einsumfunc.py +36 -0
  927. numpy/typing/tests/data/pass/flatiter.py +26 -0
  928. numpy/typing/tests/data/pass/fromnumeric.py +272 -0
  929. numpy/typing/tests/data/pass/index_tricks.py +62 -0
  930. numpy/typing/tests/data/pass/lib_user_array.py +22 -0
  931. numpy/typing/tests/data/pass/lib_utils.py +19 -0
  932. numpy/typing/tests/data/pass/lib_version.py +18 -0
  933. numpy/typing/tests/data/pass/literal.py +52 -0
  934. numpy/typing/tests/data/pass/ma.py +199 -0
  935. numpy/typing/tests/data/pass/mod.py +149 -0
  936. numpy/typing/tests/data/pass/modules.py +45 -0
  937. numpy/typing/tests/data/pass/multiarray.py +77 -0
  938. numpy/typing/tests/data/pass/ndarray_conversion.py +81 -0
  939. numpy/typing/tests/data/pass/ndarray_misc.py +199 -0
  940. numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
  941. numpy/typing/tests/data/pass/nditer.py +4 -0
  942. numpy/typing/tests/data/pass/numeric.py +90 -0
  943. numpy/typing/tests/data/pass/numerictypes.py +17 -0
  944. numpy/typing/tests/data/pass/random.py +1498 -0
  945. numpy/typing/tests/data/pass/recfunctions.py +164 -0
  946. numpy/typing/tests/data/pass/scalars.py +249 -0
  947. numpy/typing/tests/data/pass/shape.py +19 -0
  948. numpy/typing/tests/data/pass/simple.py +170 -0
  949. numpy/typing/tests/data/pass/ufunc_config.py +64 -0
  950. numpy/typing/tests/data/pass/ufunclike.py +52 -0
  951. numpy/typing/tests/data/pass/ufuncs.py +16 -0
  952. numpy/typing/tests/data/pass/warnings_and_errors.py +6 -0
  953. numpy/typing/tests/data/reveal/arithmetic.pyi +719 -0
  954. numpy/typing/tests/data/reveal/array_api_info.pyi +70 -0
  955. numpy/typing/tests/data/reveal/array_constructors.pyi +277 -0
  956. numpy/typing/tests/data/reveal/arraypad.pyi +27 -0
  957. numpy/typing/tests/data/reveal/arrayprint.pyi +25 -0
  958. numpy/typing/tests/data/reveal/arraysetops.pyi +74 -0
  959. numpy/typing/tests/data/reveal/arrayterator.pyi +27 -0
  960. numpy/typing/tests/data/reveal/bitwise_ops.pyi +166 -0
  961. numpy/typing/tests/data/reveal/char.pyi +225 -0
  962. numpy/typing/tests/data/reveal/chararray.pyi +138 -0
  963. numpy/typing/tests/data/reveal/comparisons.pyi +264 -0
  964. numpy/typing/tests/data/reveal/constants.pyi +14 -0
  965. numpy/typing/tests/data/reveal/ctypeslib.pyi +81 -0
  966. numpy/typing/tests/data/reveal/datasource.pyi +23 -0
  967. numpy/typing/tests/data/reveal/dtype.pyi +132 -0
  968. numpy/typing/tests/data/reveal/einsumfunc.pyi +39 -0
  969. numpy/typing/tests/data/reveal/emath.pyi +54 -0
  970. numpy/typing/tests/data/reveal/fft.pyi +37 -0
  971. numpy/typing/tests/data/reveal/flatiter.pyi +86 -0
  972. numpy/typing/tests/data/reveal/fromnumeric.pyi +347 -0
  973. numpy/typing/tests/data/reveal/getlimits.pyi +53 -0
  974. numpy/typing/tests/data/reveal/histograms.pyi +25 -0
  975. numpy/typing/tests/data/reveal/index_tricks.pyi +70 -0
  976. numpy/typing/tests/data/reveal/lib_function_base.pyi +409 -0
  977. numpy/typing/tests/data/reveal/lib_polynomial.pyi +147 -0
  978. numpy/typing/tests/data/reveal/lib_utils.pyi +17 -0
  979. numpy/typing/tests/data/reveal/lib_version.pyi +20 -0
  980. numpy/typing/tests/data/reveal/linalg.pyi +154 -0
  981. numpy/typing/tests/data/reveal/ma.pyi +1098 -0
  982. numpy/typing/tests/data/reveal/matrix.pyi +73 -0
  983. numpy/typing/tests/data/reveal/memmap.pyi +19 -0
  984. numpy/typing/tests/data/reveal/mod.pyi +178 -0
  985. numpy/typing/tests/data/reveal/modules.pyi +51 -0
  986. numpy/typing/tests/data/reveal/multiarray.pyi +197 -0
  987. numpy/typing/tests/data/reveal/nbit_base_example.pyi +20 -0
  988. numpy/typing/tests/data/reveal/ndarray_assignability.pyi +82 -0
  989. numpy/typing/tests/data/reveal/ndarray_conversion.pyi +83 -0
  990. numpy/typing/tests/data/reveal/ndarray_misc.pyi +246 -0
  991. numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi +47 -0
  992. numpy/typing/tests/data/reveal/nditer.pyi +49 -0
  993. numpy/typing/tests/data/reveal/nested_sequence.pyi +25 -0
  994. numpy/typing/tests/data/reveal/npyio.pyi +83 -0
  995. numpy/typing/tests/data/reveal/numeric.pyi +170 -0
  996. numpy/typing/tests/data/reveal/numerictypes.pyi +16 -0
  997. numpy/typing/tests/data/reveal/polynomial_polybase.pyi +217 -0
  998. numpy/typing/tests/data/reveal/polynomial_polyutils.pyi +218 -0
  999. numpy/typing/tests/data/reveal/polynomial_series.pyi +138 -0
  1000. numpy/typing/tests/data/reveal/random.pyi +1546 -0
  1001. numpy/typing/tests/data/reveal/rec.pyi +171 -0
  1002. numpy/typing/tests/data/reveal/scalars.pyi +191 -0
  1003. numpy/typing/tests/data/reveal/shape.pyi +13 -0
  1004. numpy/typing/tests/data/reveal/shape_base.pyi +52 -0
  1005. numpy/typing/tests/data/reveal/stride_tricks.pyi +27 -0
  1006. numpy/typing/tests/data/reveal/strings.pyi +196 -0
  1007. numpy/typing/tests/data/reveal/testing.pyi +198 -0
  1008. numpy/typing/tests/data/reveal/twodim_base.pyi +225 -0
  1009. numpy/typing/tests/data/reveal/type_check.pyi +67 -0
  1010. numpy/typing/tests/data/reveal/ufunc_config.pyi +29 -0
  1011. numpy/typing/tests/data/reveal/ufunclike.pyi +31 -0
  1012. numpy/typing/tests/data/reveal/ufuncs.pyi +142 -0
  1013. numpy/typing/tests/data/reveal/warnings_and_errors.pyi +11 -0
  1014. numpy/typing/tests/test_isfile.py +38 -0
  1015. numpy/typing/tests/test_runtime.py +110 -0
  1016. numpy/typing/tests/test_typing.py +205 -0
  1017. numpy/version.py +11 -0
  1018. numpy/version.pyi +9 -0
  1019. numpy-2.4.1.dist-info/METADATA +139 -0
  1020. numpy-2.4.1.dist-info/RECORD +1039 -0
  1021. numpy-2.4.1.dist-info/WHEEL +6 -0
  1022. numpy-2.4.1.dist-info/entry_points.txt +13 -0
  1023. numpy-2.4.1.dist-info/licenses/LICENSE.txt +935 -0
  1024. numpy-2.4.1.dist-info/licenses/numpy/_core/include/numpy/libdivide/LICENSE.txt +21 -0
  1025. numpy-2.4.1.dist-info/licenses/numpy/_core/src/common/pythoncapi-compat/COPYING +14 -0
  1026. numpy-2.4.1.dist-info/licenses/numpy/_core/src/highway/LICENSE +371 -0
  1027. numpy-2.4.1.dist-info/licenses/numpy/_core/src/multiarray/dragon4_LICENSE.txt +27 -0
  1028. numpy-2.4.1.dist-info/licenses/numpy/_core/src/npysort/x86-simd-sort/LICENSE.md +28 -0
  1029. numpy-2.4.1.dist-info/licenses/numpy/_core/src/umath/svml/LICENSE +30 -0
  1030. numpy-2.4.1.dist-info/licenses/numpy/fft/pocketfft/LICENSE.md +25 -0
  1031. numpy-2.4.1.dist-info/licenses/numpy/linalg/lapack_lite/LICENSE.txt +48 -0
  1032. numpy-2.4.1.dist-info/licenses/numpy/ma/LICENSE +24 -0
  1033. numpy-2.4.1.dist-info/licenses/numpy/random/LICENSE.md +71 -0
  1034. numpy-2.4.1.dist-info/licenses/numpy/random/src/distributions/LICENSE.md +61 -0
  1035. numpy-2.4.1.dist-info/licenses/numpy/random/src/mt19937/LICENSE.md +61 -0
  1036. numpy-2.4.1.dist-info/licenses/numpy/random/src/pcg64/LICENSE.md +22 -0
  1037. numpy-2.4.1.dist-info/licenses/numpy/random/src/philox/LICENSE.md +31 -0
  1038. numpy-2.4.1.dist-info/licenses/numpy/random/src/sfc64/LICENSE.md +27 -0
  1039. numpy-2.4.1.dist-info/licenses/numpy/random/src/splitmix64/LICENSE.md +9 -0
@@ -0,0 +1,928 @@
1
+ """
2
+ Tests for array coercion, mainly through testing `np.array` results directly.
3
+ Note that other such tests exist, e.g., in `test_api.py` and many corner-cases
4
+ are tested (sometimes indirectly) elsewhere.
5
+ """
6
+
7
+ from itertools import permutations, product
8
+
9
+ import pytest
10
+ from pytest import param
11
+
12
+ import numpy as np
13
+ import numpy._core._multiarray_umath as ncu
14
+ from numpy._core._rational_tests import rational
15
+ from numpy.testing import IS_64BIT, IS_PYPY, assert_array_equal
16
+
17
+
18
+ def arraylikes():
19
+ """
20
+ Generator for functions converting an array into various array-likes.
21
+ If full is True (default) it includes array-likes not capable of handling
22
+ all dtypes.
23
+ """
24
+ # base array:
25
+ def ndarray(a):
26
+ return a
27
+
28
+ yield param(ndarray, id="ndarray")
29
+
30
+ # subclass:
31
+ class MyArr(np.ndarray):
32
+ pass
33
+
34
+ def subclass(a):
35
+ return a.view(MyArr)
36
+
37
+ yield subclass
38
+
39
+ class _SequenceLike:
40
+ # Older NumPy versions, sometimes cared whether a protocol array was
41
+ # also _SequenceLike. This shouldn't matter, but keep it for now
42
+ # for __array__ and not the others.
43
+ def __len__(self):
44
+ raise TypeError
45
+
46
+ def __getitem__(self, _, /):
47
+ raise TypeError
48
+
49
+ # Array-interface
50
+ class ArrayDunder(_SequenceLike):
51
+ def __init__(self, a):
52
+ self.a = a
53
+
54
+ def __array__(self, dtype=None, copy=None):
55
+ if dtype is None:
56
+ return self.a
57
+ return self.a.astype(dtype)
58
+
59
+ yield param(ArrayDunder, id="__array__")
60
+
61
+ # memory-view
62
+ yield param(memoryview, id="memoryview")
63
+
64
+ # Array-interface
65
+ class ArrayInterface:
66
+ def __init__(self, a):
67
+ self.a = a # need to hold on to keep interface valid
68
+ self.__array_interface__ = a.__array_interface__
69
+
70
+ yield param(ArrayInterface, id="__array_interface__")
71
+
72
+ # Array-Struct
73
+ class ArrayStruct:
74
+ def __init__(self, a):
75
+ self.a = a # need to hold on to keep struct valid
76
+ self.__array_struct__ = a.__array_struct__
77
+
78
+ yield param(ArrayStruct, id="__array_struct__")
79
+
80
+
81
+ def scalar_instances(times=True, extended_precision=True, user_dtype=True):
82
+ # Hard-coded list of scalar instances.
83
+ # Floats:
84
+ yield param(np.sqrt(np.float16(5)), id="float16")
85
+ yield param(np.sqrt(np.float32(5)), id="float32")
86
+ yield param(np.sqrt(np.float64(5)), id="float64")
87
+ if extended_precision:
88
+ yield param(np.sqrt(np.longdouble(5)), id="longdouble")
89
+
90
+ # Complex:
91
+ yield param(np.sqrt(np.complex64(2 + 3j)), id="complex64")
92
+ yield param(np.sqrt(np.complex128(2 + 3j)), id="complex128")
93
+ if extended_precision:
94
+ yield param(np.sqrt(np.clongdouble(2 + 3j)), id="clongdouble")
95
+
96
+ # Bool:
97
+ # XFAIL: Bool should be added, but has some bad properties when it
98
+ # comes to strings, see also gh-9875
99
+ # yield param(np.bool(0), id="bool")
100
+
101
+ # Integers:
102
+ yield param(np.int8(2), id="int8")
103
+ yield param(np.int16(2), id="int16")
104
+ yield param(np.int32(2), id="int32")
105
+ yield param(np.int64(2), id="int64")
106
+
107
+ yield param(np.uint8(2), id="uint8")
108
+ yield param(np.uint16(2), id="uint16")
109
+ yield param(np.uint32(2), id="uint32")
110
+ yield param(np.uint64(2), id="uint64")
111
+
112
+ # Rational:
113
+ if user_dtype:
114
+ yield param(rational(1, 2), id="rational")
115
+
116
+ # Cannot create a structured void scalar directly:
117
+ structured = np.array([(1, 3)], "i,i")[0]
118
+ assert isinstance(structured, np.void)
119
+ assert structured.dtype == np.dtype("i,i")
120
+ yield param(structured, id="structured")
121
+
122
+ if times:
123
+ # Datetimes and timedelta
124
+ yield param(np.timedelta64(2), id="timedelta64[generic]")
125
+ yield param(np.timedelta64(23, "s"), id="timedelta64[s]")
126
+ yield param(np.timedelta64("NaT", "s"), id="timedelta64[s](NaT)")
127
+
128
+ yield param(np.datetime64("NaT"), id="datetime64[generic](NaT)")
129
+ yield param(np.datetime64("2020-06-07 12:43", "ms"), id="datetime64[ms]")
130
+
131
+ # Strings and unstructured void:
132
+ yield param(np.bytes_(b"1234"), id="bytes")
133
+ yield param(np.str_("2345"), id="unicode")
134
+ yield param(np.void(b"4321"), id="unstructured_void")
135
+
136
+
137
+ def is_parametric_dtype(dtype):
138
+ """Returns True if the dtype is a parametric legacy dtype (itemsize
139
+ is 0, or a datetime without units)
140
+ """
141
+ if dtype.itemsize == 0:
142
+ return True
143
+ if issubclass(dtype.type, (np.datetime64, np.timedelta64)):
144
+ if dtype.name.endswith("64"):
145
+ # Generic time units
146
+ return True
147
+ return False
148
+
149
+
150
+ class TestStringDiscovery:
151
+ @pytest.mark.parametrize("obj",
152
+ [object(), 1.2, 10**43, None, "string"],
153
+ ids=["object", "1.2", "10**43", "None", "string"])
154
+ def test_basic_stringlength(self, obj):
155
+ length = len(str(obj))
156
+ expected = np.dtype(f"S{length}")
157
+
158
+ assert np.array(obj, dtype="S").dtype == expected
159
+ assert np.array([obj], dtype="S").dtype == expected
160
+
161
+ # A nested array is also discovered correctly
162
+ arr = np.array(obj, dtype="O")
163
+ assert np.array(arr, dtype="S").dtype == expected
164
+ # Also if we use the dtype class
165
+ assert np.array(arr, dtype=type(expected)).dtype == expected
166
+ # Check that .astype() behaves identical
167
+ assert arr.astype("S").dtype == expected
168
+ # The DType class is accepted by `.astype()`
169
+ assert arr.astype(type(np.dtype("S"))).dtype == expected
170
+
171
+ @pytest.mark.parametrize("obj",
172
+ [object(), 1.2, 10**43, None, "string"],
173
+ ids=["object", "1.2", "10**43", "None", "string"])
174
+ def test_nested_arrays_stringlength(self, obj):
175
+ length = len(str(obj))
176
+ expected = np.dtype(f"S{length}")
177
+ arr = np.array(obj, dtype="O")
178
+ assert np.array([arr, arr], dtype="S").dtype == expected
179
+
180
+ @pytest.mark.parametrize("arraylike", arraylikes())
181
+ def test_unpack_first_level(self, arraylike):
182
+ # We unpack exactly one level of array likes
183
+ obj = np.array([None])
184
+ obj[0] = np.array(1.2)
185
+ # the length of the included item, not of the float dtype
186
+ length = len(str(obj[0]))
187
+ expected = np.dtype(f"S{length}")
188
+
189
+ obj = arraylike(obj)
190
+ # casting to string usually calls str(obj)
191
+ arr = np.array([obj], dtype="S")
192
+ assert arr.shape == (1, 1)
193
+ assert arr.dtype == expected
194
+
195
+
196
+ class TestScalarDiscovery:
197
+ def test_void_special_case(self):
198
+ # Void dtypes with structures discover tuples as elements
199
+ arr = np.array((1, 2, 3), dtype="i,i,i")
200
+ assert arr.shape == ()
201
+ arr = np.array([(1, 2, 3)], dtype="i,i,i")
202
+ assert arr.shape == (1,)
203
+
204
+ def test_char_special_case(self):
205
+ arr = np.array("string", dtype="c")
206
+ assert arr.shape == (6,)
207
+ assert arr.dtype.char == "c"
208
+ arr = np.array(["string"], dtype="c")
209
+ assert arr.shape == (1, 6)
210
+ assert arr.dtype.char == "c"
211
+
212
+ def test_char_special_case_deep(self):
213
+ # Check that the character special case errors correctly if the
214
+ # array is too deep:
215
+ nested = ["string"] # 2 dimensions (due to string being sequence)
216
+ for i in range(ncu.MAXDIMS - 2):
217
+ nested = [nested]
218
+
219
+ arr = np.array(nested, dtype='c')
220
+ assert arr.shape == (1,) * (ncu.MAXDIMS - 1) + (6,)
221
+ with pytest.raises(ValueError):
222
+ np.array([nested], dtype="c")
223
+
224
+ def test_unknown_object(self):
225
+ arr = np.array(object())
226
+ assert arr.shape == ()
227
+ assert arr.dtype == np.dtype("O")
228
+
229
+ @pytest.mark.parametrize("scalar", scalar_instances())
230
+ def test_scalar(self, scalar):
231
+ arr = np.array(scalar)
232
+ assert arr.shape == ()
233
+ assert arr.dtype == scalar.dtype
234
+
235
+ arr = np.array([[scalar, scalar]])
236
+ assert arr.shape == (1, 2)
237
+ assert arr.dtype == scalar.dtype
238
+
239
+ # Additionally to string this test also runs into a corner case
240
+ # with datetime promotion (the difference is the promotion order).
241
+ @pytest.mark.filterwarnings("ignore:Promotion of numbers:FutureWarning")
242
+ def test_scalar_promotion(self):
243
+ for sc1, sc2 in product(scalar_instances(), scalar_instances()):
244
+ sc1, sc2 = sc1.values[0], sc2.values[0]
245
+ # test all combinations:
246
+ try:
247
+ arr = np.array([sc1, sc2])
248
+ except (TypeError, ValueError):
249
+ # The promotion between two times can fail
250
+ # XFAIL (ValueError): Some object casts are currently undefined
251
+ continue
252
+ assert arr.shape == (2,)
253
+ try:
254
+ dt1, dt2 = sc1.dtype, sc2.dtype
255
+ expected_dtype = np.promote_types(dt1, dt2)
256
+ assert arr.dtype == expected_dtype
257
+ except TypeError as e:
258
+ # Will currently always go to object dtype
259
+ assert arr.dtype == np.dtype("O")
260
+
261
+ @pytest.mark.parametrize("scalar", scalar_instances())
262
+ def test_scalar_coercion(self, scalar):
263
+ # This tests various scalar coercion paths, mainly for the numerical
264
+ # types. It includes some paths not directly related to `np.array`.
265
+ if isinstance(scalar, np.inexact):
266
+ # Ensure we have a full-precision number if available
267
+ scalar = type(scalar)((scalar * 2)**0.5)
268
+
269
+ # Use casting from object:
270
+ arr = np.array(scalar, dtype=object).astype(scalar.dtype)
271
+
272
+ # Test various ways to create an array containing this scalar:
273
+ arr1 = np.array(scalar).reshape(1)
274
+ arr2 = np.array([scalar])
275
+ arr3 = np.empty(1, dtype=scalar.dtype)
276
+ arr3[0] = scalar
277
+ arr4 = np.empty(1, dtype=scalar.dtype)
278
+ arr4[:] = [scalar]
279
+ # All of these methods should yield the same results
280
+ assert_array_equal(arr, arr1)
281
+ assert_array_equal(arr, arr2)
282
+ assert_array_equal(arr, arr3)
283
+ assert_array_equal(arr, arr4)
284
+
285
+ @pytest.mark.xfail(IS_PYPY, reason="`int(np.complex128(3))` fails on PyPy")
286
+ @pytest.mark.filterwarnings("ignore::numpy.exceptions.ComplexWarning")
287
+ @pytest.mark.parametrize("cast_to", scalar_instances())
288
+ def test_scalar_coercion_same_as_cast_and_assignment(self, cast_to):
289
+ """
290
+ Test that in most cases:
291
+ * `np.array(scalar, dtype=dtype)`
292
+ * `np.empty((), dtype=dtype)[()] = scalar`
293
+ * `np.array(scalar).astype(dtype)`
294
+ should behave the same. The only exceptions are parametric dtypes
295
+ (mainly datetime/timedelta without unit) and void without fields.
296
+ """
297
+ dtype = cast_to.dtype # use to parametrize only the target dtype
298
+
299
+ for scalar in scalar_instances(times=False):
300
+ scalar = scalar.values[0]
301
+
302
+ if dtype.type == np.void:
303
+ if scalar.dtype.fields is not None and dtype.fields is None:
304
+ # Here, coercion to "V6" works, but the cast fails.
305
+ # Since the types are identical, SETITEM takes care of
306
+ # this, but has different rules than the cast.
307
+ with pytest.raises(TypeError):
308
+ np.array(scalar).astype(dtype)
309
+ np.array(scalar, dtype=dtype)
310
+ np.array([scalar], dtype=dtype)
311
+ continue
312
+
313
+ # The main test, we first try to use casting and if it succeeds
314
+ # continue below testing that things are the same, otherwise
315
+ # test that the alternative paths at least also fail.
316
+ try:
317
+ cast = np.array(scalar).astype(dtype)
318
+ except (TypeError, ValueError, RuntimeError):
319
+ # coercion should also raise (error type may change)
320
+ with pytest.raises(Exception): # noqa: B017
321
+ np.array(scalar, dtype=dtype)
322
+
323
+ if (isinstance(scalar, rational) and
324
+ np.issubdtype(dtype, np.signedinteger)):
325
+ return
326
+
327
+ with pytest.raises(Exception): # noqa: B017
328
+ np.array([scalar], dtype=dtype)
329
+ # assignment should also raise
330
+ res = np.zeros((), dtype=dtype)
331
+ with pytest.raises(Exception): # noqa: B017
332
+ res[()] = scalar
333
+
334
+ return
335
+
336
+ # Non error path:
337
+ arr = np.array(scalar, dtype=dtype)
338
+ assert_array_equal(arr, cast)
339
+ # assignment behaves the same
340
+ ass = np.zeros((), dtype=dtype)
341
+ ass[()] = scalar
342
+ assert_array_equal(ass, cast)
343
+
344
+ @pytest.mark.parametrize("pyscalar", [10, 10.32, 10.14j, 10**100])
345
+ def test_pyscalar_subclasses(self, pyscalar):
346
+ """NumPy arrays are read/write which means that anything but invariant
347
+ behaviour is on thin ice. However, we currently are happy to discover
348
+ subclasses of Python float, int, complex the same as the base classes.
349
+ This should potentially be deprecated.
350
+ """
351
+ class MyScalar(type(pyscalar)):
352
+ pass
353
+
354
+ res = np.array(MyScalar(pyscalar))
355
+ expected = np.array(pyscalar)
356
+ assert_array_equal(res, expected)
357
+
358
+ @pytest.mark.parametrize("dtype_char", np.typecodes["All"])
359
+ def test_default_dtype_instance(self, dtype_char):
360
+ if dtype_char in "SU":
361
+ dtype = np.dtype(dtype_char + "1")
362
+ elif dtype_char == "V":
363
+ # Legacy behaviour was to use V8. The reason was float64 being the
364
+ # default dtype and that having 8 bytes.
365
+ dtype = np.dtype("V8")
366
+ else:
367
+ dtype = np.dtype(dtype_char)
368
+
369
+ discovered_dtype, _ = ncu._discover_array_parameters([], type(dtype))
370
+
371
+ assert discovered_dtype == dtype
372
+ assert discovered_dtype.itemsize == dtype.itemsize
373
+
374
+ @pytest.mark.parametrize("dtype", np.typecodes["Integer"])
375
+ @pytest.mark.parametrize(["scalar", "error"],
376
+ [(np.float64(np.nan), ValueError),
377
+ (np.array(-1).astype(np.ulonglong)[()], OverflowError)])
378
+ def test_scalar_to_int_coerce_does_not_cast(self, dtype, scalar, error):
379
+ """
380
+ Signed integers are currently different in that they do not cast other
381
+ NumPy scalar, but instead use scalar.__int__(). The hardcoded
382
+ exception to this rule is `np.array(scalar, dtype=integer)`.
383
+ """
384
+ dtype = np.dtype(dtype)
385
+
386
+ # This is a special case using casting logic. It warns for the NaN
387
+ # but allows the cast (giving undefined behaviour).
388
+ with np.errstate(invalid="ignore"):
389
+ coerced = np.array(scalar, dtype=dtype)
390
+ cast = np.array(scalar).astype(dtype)
391
+ assert_array_equal(coerced, cast)
392
+
393
+ # However these fail:
394
+ with pytest.raises(error):
395
+ np.array([scalar], dtype=dtype)
396
+ with pytest.raises(error):
397
+ cast[()] = scalar
398
+
399
+
400
+ class TestTimeScalars:
401
+ @pytest.mark.parametrize("dtype", [np.int64, np.float32])
402
+ @pytest.mark.parametrize("scalar",
403
+ [param(np.timedelta64("NaT", "s"), id="timedelta64[s](NaT)"),
404
+ param(np.timedelta64(123, "s"), id="timedelta64[s]"),
405
+ param(np.datetime64("NaT", "generic"), id="datetime64[generic](NaT)"),
406
+ param(np.datetime64(1, "D"), id="datetime64[D]")],)
407
+ def test_coercion_basic(self, dtype, scalar):
408
+ # Note the `[scalar]` is there because np.array(scalar) uses stricter
409
+ # `scalar.__int__()` rules for backward compatibility right now.
410
+ arr = np.array(scalar, dtype=dtype)
411
+ cast = np.array(scalar).astype(dtype)
412
+ assert_array_equal(arr, cast)
413
+
414
+ ass = np.ones((), dtype=dtype)
415
+ if issubclass(dtype, np.integer):
416
+ with pytest.raises(TypeError):
417
+ # raises, as would np.array([scalar], dtype=dtype), this is
418
+ # conversion from times, but behaviour of integers.
419
+ ass[()] = scalar
420
+ else:
421
+ ass[()] = scalar
422
+ assert_array_equal(ass, cast)
423
+
424
+ @pytest.mark.parametrize("dtype", [np.int64, np.float32])
425
+ @pytest.mark.parametrize("scalar",
426
+ [param(np.timedelta64(123, "ns"), id="timedelta64[ns]"),
427
+ param(np.timedelta64(12, "generic"), id="timedelta64[generic]")])
428
+ def test_coercion_timedelta_convert_to_number(self, dtype, scalar):
429
+ # Only "ns" and "generic" timedeltas can be converted to numbers
430
+ # so these are slightly special.
431
+ arr = np.array(scalar, dtype=dtype)
432
+ cast = np.array(scalar).astype(dtype)
433
+ ass = np.ones((), dtype=dtype)
434
+ ass[()] = scalar # raises, as would np.array([scalar], dtype=dtype)
435
+
436
+ assert_array_equal(arr, cast)
437
+ assert_array_equal(cast, cast)
438
+
439
+ @pytest.mark.parametrize("dtype", ["S6", "U6"])
440
+ @pytest.mark.parametrize(["val", "unit"],
441
+ [param(123, "s", id="[s]"), param(123, "D", id="[D]")])
442
+ def test_coercion_assignment_datetime(self, val, unit, dtype):
443
+ # String from datetime64 assignment is currently special cased to
444
+ # never use casting. This is because casting will error in this
445
+ # case, and traditionally in most cases the behaviour is maintained
446
+ # like this. (`np.array(scalar, dtype="U6")` would have failed before)
447
+ # TODO: This discrepancy _should_ be resolved, either by relaxing the
448
+ # cast, or by deprecating the first part.
449
+ scalar = np.datetime64(val, unit)
450
+ dtype = np.dtype(dtype)
451
+ cut_string = dtype.type(str(scalar)[:6])
452
+
453
+ arr = np.array(scalar, dtype=dtype)
454
+ assert arr[()] == cut_string
455
+ ass = np.ones((), dtype=dtype)
456
+ ass[()] = scalar
457
+ assert ass[()] == cut_string
458
+
459
+ with pytest.raises(RuntimeError):
460
+ # However, unlike the above assignment using `str(scalar)[:6]`
461
+ # due to being handled by the string DType and not be casting
462
+ # the explicit cast fails:
463
+ np.array(scalar).astype(dtype)
464
+
465
+ @pytest.mark.parametrize(["val", "unit"],
466
+ [param(123, "s", id="[s]"), param(123, "D", id="[D]")])
467
+ def test_coercion_assignment_timedelta(self, val, unit):
468
+ scalar = np.timedelta64(val, unit)
469
+
470
+ # Unlike datetime64, timedelta allows the unsafe cast:
471
+ np.array(scalar, dtype="S6")
472
+ cast = np.array(scalar).astype("S6")
473
+ ass = np.ones((), dtype="S6")
474
+ ass[()] = scalar
475
+ expected = scalar.astype("S")[:6]
476
+ assert cast[()] == expected
477
+ assert ass[()] == expected
478
+
479
+ class TestNested:
480
+ def test_nested_simple(self):
481
+ initial = [1.2]
482
+ nested = initial
483
+ for i in range(ncu.MAXDIMS - 1):
484
+ nested = [nested]
485
+
486
+ arr = np.array(nested, dtype="float64")
487
+ assert arr.shape == (1,) * ncu.MAXDIMS
488
+ with pytest.raises(ValueError):
489
+ np.array([nested], dtype="float64")
490
+
491
+ with pytest.raises(ValueError, match=".*would exceed the maximum"):
492
+ np.array([nested]) # user must ask for `object` explicitly
493
+
494
+ arr = np.array([nested], dtype=object)
495
+ assert arr.dtype == np.dtype("O")
496
+ assert arr.shape == (1,) * ncu.MAXDIMS
497
+ assert arr.item() is initial
498
+
499
+ def test_pathological_self_containing(self):
500
+ # Test that this also works for two nested sequences
501
+ l = []
502
+ l.append(l)
503
+ arr = np.array([l, l, l], dtype=object)
504
+ assert arr.shape == (3,) + (1,) * (ncu.MAXDIMS - 1)
505
+
506
+ # Also check a ragged case:
507
+ arr = np.array([l, [None], l], dtype=object)
508
+ assert arr.shape == (3, 1)
509
+
510
+ @pytest.mark.parametrize("arraylike", arraylikes())
511
+ def test_nested_arraylikes(self, arraylike):
512
+ # We try storing an array like into an array, but the array-like
513
+ # will have too many dimensions. This means the shape discovery
514
+ # decides that the array-like must be treated as an object (a special
515
+ # case of ragged discovery). The result will be an array with one
516
+ # dimension less than the maximum dimensions, and the array being
517
+ # assigned to it (which does work for object or if `float(arraylike)`
518
+ # works).
519
+ initial = arraylike(np.ones((1, 1)))
520
+
521
+ nested = initial
522
+ for i in range(ncu.MAXDIMS - 1):
523
+ nested = [nested]
524
+
525
+ with pytest.raises(ValueError, match=".*would exceed the maximum"):
526
+ # It will refuse to assign the array into
527
+ np.array(nested, dtype="float64")
528
+
529
+ # If this is object, we end up assigning a (1, 1) array into (1,)
530
+ # (due to running out of dimensions), this is currently supported but
531
+ # a special case which is not ideal.
532
+ arr = np.array(nested, dtype=object)
533
+ assert arr.shape == (1,) * ncu.MAXDIMS
534
+ assert arr.item() == np.array(initial).item()
535
+
536
+ @pytest.mark.parametrize("arraylike", arraylikes())
537
+ def test_uneven_depth_ragged(self, arraylike):
538
+ arr = np.arange(4).reshape((2, 2))
539
+ arr = arraylike(arr)
540
+
541
+ # Array is ragged in the second dimension already:
542
+ out = np.array([arr, [arr]], dtype=object)
543
+ assert out.shape == (2,)
544
+ assert out[0] is arr
545
+ assert type(out[1]) is list
546
+
547
+ # Array is ragged in the third dimension:
548
+ with pytest.raises(ValueError):
549
+ # This is a broadcast error during assignment, because
550
+ # the array shape would be (2, 2, 2) but `arr[0, 0] = arr` fails.
551
+ np.array([arr, [arr, arr]], dtype=object)
552
+
553
+ def test_empty_sequence(self):
554
+ arr = np.array([[], [1], [[1]]], dtype=object)
555
+ assert arr.shape == (3,)
556
+
557
+ # The empty sequence stops further dimension discovery, so the
558
+ # result shape will be (0,) which leads to an error during:
559
+ with pytest.raises(ValueError):
560
+ np.array([[], np.empty((0, 1))], dtype=object)
561
+
562
+ def test_array_of_different_depths(self):
563
+ # When multiple arrays (or array-likes) are included in a
564
+ # sequences and have different depth, we currently discover
565
+ # as many dimensions as they share. (see also gh-17224)
566
+ arr = np.zeros((3, 2))
567
+ mismatch_first_dim = np.zeros((1, 2))
568
+ mismatch_second_dim = np.zeros((3, 3))
569
+
570
+ dtype, shape = ncu._discover_array_parameters(
571
+ [arr, mismatch_second_dim], dtype=np.dtype("O"))
572
+ assert shape == (2, 3)
573
+
574
+ dtype, shape = ncu._discover_array_parameters(
575
+ [arr, mismatch_first_dim], dtype=np.dtype("O"))
576
+ assert shape == (2,)
577
+ # The second case is currently supported because the arrays
578
+ # can be stored as objects:
579
+ res = np.asarray([arr, mismatch_first_dim], dtype=np.dtype("O"))
580
+ assert res[0] is arr
581
+ assert res[1] is mismatch_first_dim
582
+
583
+
584
+ class TestBadSequences:
585
+ # These are tests for bad objects passed into `np.array`, in general
586
+ # these have undefined behaviour. In the old code they partially worked
587
+ # when now they will fail. We could (and maybe should) create a copy
588
+ # of all sequences to be safe against bad-actors.
589
+
590
+ def test_growing_list(self):
591
+ # List to coerce, `mylist` will append to it during coercion
592
+ obj = []
593
+
594
+ class mylist(list):
595
+ def __len__(self):
596
+ obj.append([1, 2])
597
+ return super().__len__()
598
+
599
+ obj.append(mylist([1, 2]))
600
+
601
+ with pytest.raises(RuntimeError):
602
+ np.array(obj)
603
+
604
+ # Note: We do not test a shrinking list. These do very evil things
605
+ # and the only way to fix them would be to copy all sequences.
606
+ # (which may be a real option in the future).
607
+
608
+ def test_mutated_list(self):
609
+ # List to coerce, `mylist` will mutate the first element
610
+ obj = []
611
+
612
+ class mylist(list):
613
+ def __len__(self):
614
+ obj[0] = [2, 3] # replace with a different list.
615
+ return super().__len__()
616
+
617
+ obj.append([2, 3])
618
+ obj.append(mylist([1, 2]))
619
+ # Does not crash:
620
+ np.array(obj)
621
+
622
+ def test_replace_0d_array(self):
623
+ # List to coerce, `mylist` will mutate the first element
624
+ obj = []
625
+
626
+ class baditem:
627
+ def __len__(self):
628
+ obj[0][0] = 2 # replace with a different list.
629
+ raise ValueError("not actually a sequence!")
630
+
631
+ def __getitem__(self, _, /):
632
+ pass
633
+
634
+ # Runs into a corner case in the new code, the `array(2)` is cached
635
+ # so replacing it invalidates the cache.
636
+ obj.append([np.array(2), baditem()])
637
+ with pytest.raises(RuntimeError):
638
+ np.array(obj)
639
+
640
+
641
+ class TestArrayLikes:
642
+ @pytest.mark.parametrize("arraylike", arraylikes())
643
+ def test_0d_object_special_case(self, arraylike):
644
+ arr = np.array(0.)
645
+ obj = arraylike(arr)
646
+ # A single array-like is always converted:
647
+ res = np.array(obj, dtype=object)
648
+ assert_array_equal(arr, res)
649
+
650
+ # But a single 0-D nested array-like never:
651
+ res = np.array([obj], dtype=object)
652
+ assert res[0] is obj
653
+
654
+ @pytest.mark.parametrize("arraylike", arraylikes())
655
+ @pytest.mark.parametrize("arr", [np.array(0.), np.arange(4)])
656
+ def test_object_assignment_special_case(self, arraylike, arr):
657
+ obj = arraylike(arr)
658
+ empty = np.arange(1, dtype=object)
659
+ empty[:] = [obj]
660
+ assert empty[0] is obj
661
+
662
+ def test_0d_generic_special_case(self):
663
+ class ArraySubclass(np.ndarray):
664
+ def __float__(self):
665
+ raise TypeError("e.g. quantities raise on this")
666
+
667
+ arr = np.array(0.)
668
+ obj = arr.view(ArraySubclass)
669
+ res = np.array(obj)
670
+ # The subclass is simply cast:
671
+ assert_array_equal(arr, res)
672
+
673
+ # If the 0-D array-like is included, __float__ is currently
674
+ # guaranteed to be used. We may want to change that, quantities
675
+ # and masked arrays half make use of this.
676
+ with pytest.raises(TypeError):
677
+ np.array([obj])
678
+
679
+ # The same holds for memoryview:
680
+ obj = memoryview(arr)
681
+ res = np.array(obj)
682
+ assert_array_equal(arr, res)
683
+ with pytest.raises(ValueError):
684
+ # The error type does not matter much here.
685
+ np.array([obj])
686
+
687
+ def test_arraylike_classes(self):
688
+ # The classes of array-likes should generally be acceptable to be
689
+ # stored inside a numpy (object) array. This tests all of the
690
+ # special attributes (since all are checked during coercion).
691
+ arr = np.array(np.int64)
692
+ assert arr[()] is np.int64
693
+ arr = np.array([np.int64])
694
+ assert arr[0] is np.int64
695
+
696
+ # This also works for properties/unbound methods:
697
+ class ArrayLike:
698
+ @property
699
+ def __array_interface__(self):
700
+ pass
701
+
702
+ @property
703
+ def __array_struct__(self):
704
+ pass
705
+
706
+ def __array__(self, dtype=None, copy=None):
707
+ pass
708
+
709
+ arr = np.array(ArrayLike)
710
+ assert arr[()] is ArrayLike
711
+ arr = np.array([ArrayLike])
712
+ assert arr[0] is ArrayLike
713
+
714
+ @pytest.mark.skipif(not IS_64BIT, reason="Needs 64bit platform")
715
+ @pytest.mark.thread_unsafe(reason="large slow test in parallel")
716
+ def test_too_large_array_error_paths(self):
717
+ """Test the error paths, including for memory leaks"""
718
+ arr = np.array(0, dtype="uint8")
719
+ # Guarantees that a contiguous copy won't work:
720
+ arr = np.broadcast_to(arr, 2**62)
721
+
722
+ for i in range(5):
723
+ # repeat, to ensure caching cannot have an effect:
724
+ with pytest.raises(MemoryError):
725
+ np.array(arr)
726
+ with pytest.raises(MemoryError):
727
+ np.array([arr])
728
+
729
+ @pytest.mark.parametrize("attribute",
730
+ ["__array_interface__", "__array__", "__array_struct__"])
731
+ @pytest.mark.parametrize("error", [RecursionError, MemoryError])
732
+ def test_bad_array_like_attributes(self, attribute, error):
733
+ # RecursionError and MemoryError are considered fatal. All errors
734
+ # (except AttributeError) should probably be raised in the future,
735
+ # but shapely made use of it, so it will require a deprecation.
736
+
737
+ class BadInterface:
738
+ def __getattr__(self, attr):
739
+ if attr == attribute:
740
+ raise error
741
+ super().__getattr__(attr)
742
+
743
+ with pytest.raises(error):
744
+ np.array(BadInterface())
745
+
746
+ @pytest.mark.parametrize("error", [RecursionError, MemoryError])
747
+ def test_bad_array_like_bad_length(self, error):
748
+ # RecursionError and MemoryError are considered "critical" in
749
+ # sequences. We could expand this more generally though. (NumPy 1.20)
750
+ class BadSequence:
751
+ def __len__(self):
752
+ raise error
753
+
754
+ def __getitem__(self, _, /):
755
+ # must have getitem to be a Sequence
756
+ return 1
757
+
758
+ with pytest.raises(error):
759
+ np.array(BadSequence())
760
+
761
+ def test_array_interface_descr_optional(self):
762
+ # The descr should be optional regression test for gh-27249
763
+ arr = np.ones(10, dtype="V10")
764
+ iface = arr.__array_interface__
765
+ iface.pop("descr")
766
+
767
+ class MyClass:
768
+ __array_interface__ = iface
769
+
770
+ assert_array_equal(np.asarray(MyClass), arr)
771
+
772
+
773
+ class TestAsArray:
774
+ """Test expected behaviors of ``asarray``."""
775
+
776
+ def test_dtype_identity(self):
777
+ """Confirm the intended behavior for *dtype* kwarg.
778
+
779
+ The result of ``asarray()`` should have the dtype provided through the
780
+ keyword argument, when used. This forces unique array handles to be
781
+ produced for unique np.dtype objects, but (for equivalent dtypes), the
782
+ underlying data (the base object) is shared with the original array
783
+ object.
784
+
785
+ Ref https://github.com/numpy/numpy/issues/1468
786
+ """
787
+ int_array = np.array([1, 2, 3], dtype='i')
788
+ assert np.asarray(int_array) is int_array
789
+
790
+ # The character code resolves to the singleton dtype object provided
791
+ # by the numpy package.
792
+ assert np.asarray(int_array, dtype='i') is int_array
793
+
794
+ # Derive a dtype from n.dtype('i'), but add a metadata object to force
795
+ # the dtype to be distinct.
796
+ unequal_type = np.dtype('i', metadata={'spam': True})
797
+ annotated_int_array = np.asarray(int_array, dtype=unequal_type)
798
+ assert annotated_int_array is not int_array
799
+ assert annotated_int_array.base is int_array
800
+ # Create an equivalent descriptor with a new and distinct dtype
801
+ # instance.
802
+ equivalent_requirement = np.dtype('i', metadata={'spam': True})
803
+ annotated_int_array_alt = np.asarray(annotated_int_array,
804
+ dtype=equivalent_requirement)
805
+ assert unequal_type == equivalent_requirement
806
+ assert unequal_type is not equivalent_requirement
807
+ assert annotated_int_array_alt is not annotated_int_array
808
+ assert annotated_int_array_alt.dtype is equivalent_requirement
809
+
810
+ # Check the same logic for a pair of C types whose equivalence may vary
811
+ # between computing environments.
812
+ # Find an equivalent pair.
813
+ integer_type_codes = ('i', 'l', 'q')
814
+ integer_dtypes = [np.dtype(code) for code in integer_type_codes]
815
+ typeA = None
816
+ typeB = None
817
+ for typeA, typeB in permutations(integer_dtypes, r=2):
818
+ if typeA == typeB:
819
+ assert typeA is not typeB
820
+ break
821
+ assert isinstance(typeA, np.dtype) and isinstance(typeB, np.dtype)
822
+
823
+ # These ``asarray()`` calls may produce a new view or a copy,
824
+ # but never the same object.
825
+ long_int_array = np.asarray(int_array, dtype='l')
826
+ long_long_int_array = np.asarray(int_array, dtype='q')
827
+ assert long_int_array is not int_array
828
+ assert long_long_int_array is not int_array
829
+ assert np.asarray(long_int_array, dtype='q') is not long_int_array
830
+ array_a = np.asarray(int_array, dtype=typeA)
831
+ assert typeA == typeB
832
+ assert typeA is not typeB
833
+ assert array_a.dtype is typeA
834
+ assert array_a is not np.asarray(array_a, dtype=typeB)
835
+ assert np.asarray(array_a, dtype=typeB).dtype is typeB
836
+ assert array_a is np.asarray(array_a, dtype=typeB).base
837
+
838
+
839
+ class TestSpecialAttributeLookupFailure:
840
+ # An exception was raised while fetching the attribute
841
+
842
+ class WeirdArrayLike:
843
+ @property
844
+ def __array__(self, dtype=None, copy=None): # noqa: PLR0206
845
+ raise RuntimeError("oops!")
846
+
847
+ class WeirdArrayInterface:
848
+ @property
849
+ def __array_interface__(self):
850
+ raise RuntimeError("oops!")
851
+
852
+ def test_deprecated(self):
853
+ with pytest.raises(RuntimeError):
854
+ np.array(self.WeirdArrayLike())
855
+ with pytest.raises(RuntimeError):
856
+ np.array(self.WeirdArrayInterface())
857
+
858
+
859
+ def test_subarray_from_array_construction():
860
+ # Arrays are more complex, since they "broadcast" on success:
861
+ arr = np.array([1, 2])
862
+
863
+ res = arr.astype("2i")
864
+ assert_array_equal(res, [[1, 1], [2, 2]])
865
+
866
+ res = np.array(arr, dtype="(2,)i")
867
+
868
+ assert_array_equal(res, [[1, 1], [2, 2]])
869
+
870
+ res = np.array([[(1,), (2,)], arr], dtype="2i")
871
+ assert_array_equal(res, [[[1, 1], [2, 2]], [[1, 1], [2, 2]]])
872
+
873
+ # Also try a multi-dimensional example:
874
+ arr = np.arange(5 * 2).reshape(5, 2)
875
+ expected = np.broadcast_to(arr[:, :, np.newaxis, np.newaxis], (5, 2, 2, 2))
876
+
877
+ res = arr.astype("(2,2)f")
878
+ assert_array_equal(res, expected)
879
+
880
+ res = np.array(arr, dtype="(2,2)f")
881
+ assert_array_equal(res, expected)
882
+
883
+
884
+ def test_empty_string():
885
+ # Empty strings are unfortunately often converted to S1 and we need to
886
+ # make sure we are filling the S1 and not the (possibly) detected S0
887
+ # result. This should likely just return S0 and if not maybe the decision
888
+ # to return S1 should be moved.
889
+ res = np.array([""] * 10, dtype="S")
890
+ assert_array_equal(res, np.array("\0", "S1"))
891
+ assert res.dtype == "S1"
892
+
893
+ arr = np.array([""] * 10, dtype=object)
894
+
895
+ res = arr.astype("S")
896
+ assert_array_equal(res, b"")
897
+ assert res.dtype == "S1"
898
+
899
+ res = np.array(arr, dtype="S")
900
+ assert_array_equal(res, b"")
901
+ # TODO: This is arguably weird/wrong, but seems old:
902
+ assert res.dtype == f"S{np.dtype('O').itemsize}"
903
+
904
+ res = np.array([[""] * 10, arr], dtype="S")
905
+ assert_array_equal(res, b"")
906
+ assert res.shape == (2, 10)
907
+ assert res.dtype == "S1"
908
+
909
+
910
+ @pytest.mark.parametrize("dtype", ["S", "U", object])
911
+ @pytest.mark.parametrize("res_dt,hug_val",
912
+ [("float16", "1e30"), ("float32", "1e200")])
913
+ def test_string_to_float_coercion_errors(dtype, res_dt, hug_val):
914
+ # This test primarly tests setitem
915
+ val = np.array(["3M"], dtype=dtype)[0] # use the scalar
916
+
917
+ with pytest.raises(ValueError):
918
+ np.array(val, dtype=res_dt)
919
+
920
+ val = np.array([hug_val], dtype=dtype)[0] # use the scalar
921
+
922
+ with np.errstate(all="warn"):
923
+ with pytest.warns(RuntimeWarning):
924
+ np.array(val, dtype=res_dt)
925
+
926
+ with np.errstate(all="raise"):
927
+ with pytest.raises(FloatingPointError):
928
+ np.array(val, dtype=res_dt)