numpy 2.4.0__cp313-cp313t-musllinux_1_2_aarch64.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 (915) 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.cpython-313t-aarch64-linux-musl.so +0 -0
  30. numpy/_core/_multiarray_umath.cpython-313t-aarch64-linux-musl.so +0 -0
  31. numpy/_core/_operand_flag_tests.cpython-313t-aarch64-linux-musl.so +0 -0
  32. numpy/_core/_rational_tests.cpython-313t-aarch64-linux-musl.so +0 -0
  33. numpy/_core/_simd.cpython-313t-aarch64-linux-musl.so +0 -0
  34. numpy/_core/_simd.pyi +35 -0
  35. numpy/_core/_string_helpers.py +100 -0
  36. numpy/_core/_string_helpers.pyi +12 -0
  37. numpy/_core/_struct_ufunc_tests.cpython-313t-aarch64-linux-musl.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.cpython-313t-aarch64-linux-musl.so +0 -0
  43. numpy/_core/_umath_tests.pyi +47 -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 +353 -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 +1515 -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/doc/ufuncs.py +138 -0
  278. numpy/dtypes.py +41 -0
  279. numpy/dtypes.pyi +630 -0
  280. numpy/exceptions.py +246 -0
  281. numpy/exceptions.pyi +27 -0
  282. numpy/f2py/__init__.py +86 -0
  283. numpy/f2py/__init__.pyi +5 -0
  284. numpy/f2py/__main__.py +5 -0
  285. numpy/f2py/__version__.py +1 -0
  286. numpy/f2py/__version__.pyi +1 -0
  287. numpy/f2py/_backends/__init__.py +9 -0
  288. numpy/f2py/_backends/__init__.pyi +5 -0
  289. numpy/f2py/_backends/_backend.py +44 -0
  290. numpy/f2py/_backends/_backend.pyi +46 -0
  291. numpy/f2py/_backends/_distutils.py +76 -0
  292. numpy/f2py/_backends/_distutils.pyi +13 -0
  293. numpy/f2py/_backends/_meson.py +244 -0
  294. numpy/f2py/_backends/_meson.pyi +62 -0
  295. numpy/f2py/_backends/meson.build.template +58 -0
  296. numpy/f2py/_isocbind.py +62 -0
  297. numpy/f2py/_isocbind.pyi +13 -0
  298. numpy/f2py/_src_pyf.py +247 -0
  299. numpy/f2py/_src_pyf.pyi +28 -0
  300. numpy/f2py/auxfuncs.py +1004 -0
  301. numpy/f2py/auxfuncs.pyi +262 -0
  302. numpy/f2py/capi_maps.py +811 -0
  303. numpy/f2py/capi_maps.pyi +33 -0
  304. numpy/f2py/cb_rules.py +665 -0
  305. numpy/f2py/cb_rules.pyi +17 -0
  306. numpy/f2py/cfuncs.py +1563 -0
  307. numpy/f2py/cfuncs.pyi +31 -0
  308. numpy/f2py/common_rules.py +143 -0
  309. numpy/f2py/common_rules.pyi +9 -0
  310. numpy/f2py/crackfortran.py +3725 -0
  311. numpy/f2py/crackfortran.pyi +266 -0
  312. numpy/f2py/diagnose.py +149 -0
  313. numpy/f2py/diagnose.pyi +1 -0
  314. numpy/f2py/f2py2e.py +788 -0
  315. numpy/f2py/f2py2e.pyi +74 -0
  316. numpy/f2py/f90mod_rules.py +269 -0
  317. numpy/f2py/f90mod_rules.pyi +16 -0
  318. numpy/f2py/func2subr.py +329 -0
  319. numpy/f2py/func2subr.pyi +7 -0
  320. numpy/f2py/rules.py +1629 -0
  321. numpy/f2py/rules.pyi +41 -0
  322. numpy/f2py/setup.cfg +3 -0
  323. numpy/f2py/src/fortranobject.c +1436 -0
  324. numpy/f2py/src/fortranobject.h +173 -0
  325. numpy/f2py/symbolic.py +1518 -0
  326. numpy/f2py/symbolic.pyi +219 -0
  327. numpy/f2py/tests/__init__.py +16 -0
  328. numpy/f2py/tests/src/abstract_interface/foo.f90 +34 -0
  329. numpy/f2py/tests/src/abstract_interface/gh18403_mod.f90 +6 -0
  330. numpy/f2py/tests/src/array_from_pyobj/wrapmodule.c +235 -0
  331. numpy/f2py/tests/src/assumed_shape/.f2py_f2cmap +1 -0
  332. numpy/f2py/tests/src/assumed_shape/foo_free.f90 +34 -0
  333. numpy/f2py/tests/src/assumed_shape/foo_mod.f90 +41 -0
  334. numpy/f2py/tests/src/assumed_shape/foo_use.f90 +19 -0
  335. numpy/f2py/tests/src/assumed_shape/precision.f90 +4 -0
  336. numpy/f2py/tests/src/block_docstring/foo.f +6 -0
  337. numpy/f2py/tests/src/callback/foo.f +62 -0
  338. numpy/f2py/tests/src/callback/gh17797.f90 +7 -0
  339. numpy/f2py/tests/src/callback/gh18335.f90 +17 -0
  340. numpy/f2py/tests/src/callback/gh25211.f +10 -0
  341. numpy/f2py/tests/src/callback/gh25211.pyf +18 -0
  342. numpy/f2py/tests/src/callback/gh26681.f90 +18 -0
  343. numpy/f2py/tests/src/cli/gh_22819.pyf +6 -0
  344. numpy/f2py/tests/src/cli/hi77.f +3 -0
  345. numpy/f2py/tests/src/cli/hiworld.f90 +3 -0
  346. numpy/f2py/tests/src/common/block.f +11 -0
  347. numpy/f2py/tests/src/common/gh19161.f90 +10 -0
  348. numpy/f2py/tests/src/crackfortran/accesstype.f90 +13 -0
  349. numpy/f2py/tests/src/crackfortran/common_with_division.f +17 -0
  350. numpy/f2py/tests/src/crackfortran/data_common.f +8 -0
  351. numpy/f2py/tests/src/crackfortran/data_multiplier.f +5 -0
  352. numpy/f2py/tests/src/crackfortran/data_stmts.f90 +20 -0
  353. numpy/f2py/tests/src/crackfortran/data_with_comments.f +8 -0
  354. numpy/f2py/tests/src/crackfortran/foo_deps.f90 +6 -0
  355. numpy/f2py/tests/src/crackfortran/gh15035.f +16 -0
  356. numpy/f2py/tests/src/crackfortran/gh17859.f +12 -0
  357. numpy/f2py/tests/src/crackfortran/gh22648.pyf +7 -0
  358. numpy/f2py/tests/src/crackfortran/gh23533.f +5 -0
  359. numpy/f2py/tests/src/crackfortran/gh23598.f90 +4 -0
  360. numpy/f2py/tests/src/crackfortran/gh23598Warn.f90 +11 -0
  361. numpy/f2py/tests/src/crackfortran/gh23879.f90 +20 -0
  362. numpy/f2py/tests/src/crackfortran/gh27697.f90 +12 -0
  363. numpy/f2py/tests/src/crackfortran/gh2848.f90 +13 -0
  364. numpy/f2py/tests/src/crackfortran/operators.f90 +49 -0
  365. numpy/f2py/tests/src/crackfortran/privatemod.f90 +11 -0
  366. numpy/f2py/tests/src/crackfortran/publicmod.f90 +10 -0
  367. numpy/f2py/tests/src/crackfortran/pubprivmod.f90 +10 -0
  368. numpy/f2py/tests/src/crackfortran/unicode_comment.f90 +4 -0
  369. numpy/f2py/tests/src/f2cmap/.f2py_f2cmap +1 -0
  370. numpy/f2py/tests/src/f2cmap/isoFortranEnvMap.f90 +9 -0
  371. numpy/f2py/tests/src/isocintrin/isoCtests.f90 +34 -0
  372. numpy/f2py/tests/src/kind/foo.f90 +20 -0
  373. numpy/f2py/tests/src/mixed/foo.f +5 -0
  374. numpy/f2py/tests/src/mixed/foo_fixed.f90 +8 -0
  375. numpy/f2py/tests/src/mixed/foo_free.f90 +8 -0
  376. numpy/f2py/tests/src/modules/gh25337/data.f90 +8 -0
  377. numpy/f2py/tests/src/modules/gh25337/use_data.f90 +6 -0
  378. numpy/f2py/tests/src/modules/gh26920/two_mods_with_no_public_entities.f90 +21 -0
  379. numpy/f2py/tests/src/modules/gh26920/two_mods_with_one_public_routine.f90 +21 -0
  380. numpy/f2py/tests/src/modules/module_data_docstring.f90 +12 -0
  381. numpy/f2py/tests/src/modules/use_modules.f90 +20 -0
  382. numpy/f2py/tests/src/negative_bounds/issue_20853.f90 +7 -0
  383. numpy/f2py/tests/src/parameter/constant_array.f90 +45 -0
  384. numpy/f2py/tests/src/parameter/constant_both.f90 +57 -0
  385. numpy/f2py/tests/src/parameter/constant_compound.f90 +15 -0
  386. numpy/f2py/tests/src/parameter/constant_integer.f90 +22 -0
  387. numpy/f2py/tests/src/parameter/constant_non_compound.f90 +23 -0
  388. numpy/f2py/tests/src/parameter/constant_real.f90 +23 -0
  389. numpy/f2py/tests/src/quoted_character/foo.f +14 -0
  390. numpy/f2py/tests/src/regression/AB.inc +1 -0
  391. numpy/f2py/tests/src/regression/assignOnlyModule.f90 +25 -0
  392. numpy/f2py/tests/src/regression/datonly.f90 +17 -0
  393. numpy/f2py/tests/src/regression/f77comments.f +26 -0
  394. numpy/f2py/tests/src/regression/f77fixedform.f95 +5 -0
  395. numpy/f2py/tests/src/regression/f90continuation.f90 +9 -0
  396. numpy/f2py/tests/src/regression/incfile.f90 +5 -0
  397. numpy/f2py/tests/src/regression/inout.f90 +9 -0
  398. numpy/f2py/tests/src/regression/lower_f2py_fortran.f90 +5 -0
  399. numpy/f2py/tests/src/regression/mod_derived_types.f90 +23 -0
  400. numpy/f2py/tests/src/return_character/foo77.f +45 -0
  401. numpy/f2py/tests/src/return_character/foo90.f90 +48 -0
  402. numpy/f2py/tests/src/return_complex/foo77.f +45 -0
  403. numpy/f2py/tests/src/return_complex/foo90.f90 +48 -0
  404. numpy/f2py/tests/src/return_integer/foo77.f +56 -0
  405. numpy/f2py/tests/src/return_integer/foo90.f90 +59 -0
  406. numpy/f2py/tests/src/return_logical/foo77.f +56 -0
  407. numpy/f2py/tests/src/return_logical/foo90.f90 +59 -0
  408. numpy/f2py/tests/src/return_real/foo77.f +45 -0
  409. numpy/f2py/tests/src/return_real/foo90.f90 +48 -0
  410. numpy/f2py/tests/src/routines/funcfortranname.f +5 -0
  411. numpy/f2py/tests/src/routines/funcfortranname.pyf +11 -0
  412. numpy/f2py/tests/src/routines/subrout.f +4 -0
  413. numpy/f2py/tests/src/routines/subrout.pyf +10 -0
  414. numpy/f2py/tests/src/size/foo.f90 +44 -0
  415. numpy/f2py/tests/src/string/char.f90 +29 -0
  416. numpy/f2py/tests/src/string/fixed_string.f90 +34 -0
  417. numpy/f2py/tests/src/string/gh24008.f +8 -0
  418. numpy/f2py/tests/src/string/gh24662.f90 +7 -0
  419. numpy/f2py/tests/src/string/gh25286.f90 +14 -0
  420. numpy/f2py/tests/src/string/gh25286.pyf +12 -0
  421. numpy/f2py/tests/src/string/gh25286_bc.pyf +12 -0
  422. numpy/f2py/tests/src/string/scalar_string.f90 +9 -0
  423. numpy/f2py/tests/src/string/string.f +12 -0
  424. numpy/f2py/tests/src/value_attrspec/gh21665.f90 +9 -0
  425. numpy/f2py/tests/test_abstract_interface.py +26 -0
  426. numpy/f2py/tests/test_array_from_pyobj.py +678 -0
  427. numpy/f2py/tests/test_assumed_shape.py +50 -0
  428. numpy/f2py/tests/test_block_docstring.py +20 -0
  429. numpy/f2py/tests/test_callback.py +263 -0
  430. numpy/f2py/tests/test_character.py +641 -0
  431. numpy/f2py/tests/test_common.py +23 -0
  432. numpy/f2py/tests/test_crackfortran.py +421 -0
  433. numpy/f2py/tests/test_data.py +71 -0
  434. numpy/f2py/tests/test_docs.py +66 -0
  435. numpy/f2py/tests/test_f2cmap.py +17 -0
  436. numpy/f2py/tests/test_f2py2e.py +983 -0
  437. numpy/f2py/tests/test_isoc.py +56 -0
  438. numpy/f2py/tests/test_kind.py +52 -0
  439. numpy/f2py/tests/test_mixed.py +35 -0
  440. numpy/f2py/tests/test_modules.py +83 -0
  441. numpy/f2py/tests/test_parameter.py +129 -0
  442. numpy/f2py/tests/test_pyf_src.py +43 -0
  443. numpy/f2py/tests/test_quoted_character.py +18 -0
  444. numpy/f2py/tests/test_regression.py +187 -0
  445. numpy/f2py/tests/test_return_character.py +48 -0
  446. numpy/f2py/tests/test_return_complex.py +67 -0
  447. numpy/f2py/tests/test_return_integer.py +55 -0
  448. numpy/f2py/tests/test_return_logical.py +65 -0
  449. numpy/f2py/tests/test_return_real.py +109 -0
  450. numpy/f2py/tests/test_routines.py +29 -0
  451. numpy/f2py/tests/test_semicolon_split.py +75 -0
  452. numpy/f2py/tests/test_size.py +45 -0
  453. numpy/f2py/tests/test_string.py +100 -0
  454. numpy/f2py/tests/test_symbolic.py +500 -0
  455. numpy/f2py/tests/test_value_attrspec.py +15 -0
  456. numpy/f2py/tests/util.py +442 -0
  457. numpy/f2py/use_rules.py +99 -0
  458. numpy/f2py/use_rules.pyi +9 -0
  459. numpy/fft/__init__.py +213 -0
  460. numpy/fft/__init__.pyi +38 -0
  461. numpy/fft/_helper.py +235 -0
  462. numpy/fft/_helper.pyi +44 -0
  463. numpy/fft/_pocketfft.py +1693 -0
  464. numpy/fft/_pocketfft.pyi +137 -0
  465. numpy/fft/_pocketfft_umath.cpython-313t-aarch64-linux-musl.so +0 -0
  466. numpy/fft/tests/__init__.py +0 -0
  467. numpy/fft/tests/test_helper.py +167 -0
  468. numpy/fft/tests/test_pocketfft.py +589 -0
  469. numpy/lib/__init__.py +97 -0
  470. numpy/lib/__init__.pyi +52 -0
  471. numpy/lib/_array_utils_impl.py +62 -0
  472. numpy/lib/_array_utils_impl.pyi +10 -0
  473. numpy/lib/_arraypad_impl.py +926 -0
  474. numpy/lib/_arraypad_impl.pyi +88 -0
  475. numpy/lib/_arraysetops_impl.py +1158 -0
  476. numpy/lib/_arraysetops_impl.pyi +462 -0
  477. numpy/lib/_arrayterator_impl.py +224 -0
  478. numpy/lib/_arrayterator_impl.pyi +45 -0
  479. numpy/lib/_datasource.py +700 -0
  480. numpy/lib/_datasource.pyi +30 -0
  481. numpy/lib/_format_impl.py +1036 -0
  482. numpy/lib/_format_impl.pyi +56 -0
  483. numpy/lib/_function_base_impl.py +5758 -0
  484. numpy/lib/_function_base_impl.pyi +2324 -0
  485. numpy/lib/_histograms_impl.py +1085 -0
  486. numpy/lib/_histograms_impl.pyi +40 -0
  487. numpy/lib/_index_tricks_impl.py +1048 -0
  488. numpy/lib/_index_tricks_impl.pyi +267 -0
  489. numpy/lib/_iotools.py +900 -0
  490. numpy/lib/_iotools.pyi +116 -0
  491. numpy/lib/_nanfunctions_impl.py +2001 -0
  492. numpy/lib/_nanfunctions_impl.pyi +48 -0
  493. numpy/lib/_npyio_impl.py +2583 -0
  494. numpy/lib/_npyio_impl.pyi +299 -0
  495. numpy/lib/_polynomial_impl.py +1465 -0
  496. numpy/lib/_polynomial_impl.pyi +338 -0
  497. numpy/lib/_scimath_impl.py +642 -0
  498. numpy/lib/_scimath_impl.pyi +93 -0
  499. numpy/lib/_shape_base_impl.py +1289 -0
  500. numpy/lib/_shape_base_impl.pyi +236 -0
  501. numpy/lib/_stride_tricks_impl.py +582 -0
  502. numpy/lib/_stride_tricks_impl.pyi +73 -0
  503. numpy/lib/_twodim_base_impl.py +1201 -0
  504. numpy/lib/_twodim_base_impl.pyi +408 -0
  505. numpy/lib/_type_check_impl.py +710 -0
  506. numpy/lib/_type_check_impl.pyi +348 -0
  507. numpy/lib/_ufunclike_impl.py +199 -0
  508. numpy/lib/_ufunclike_impl.pyi +60 -0
  509. numpy/lib/_user_array_impl.py +310 -0
  510. numpy/lib/_user_array_impl.pyi +226 -0
  511. numpy/lib/_utils_impl.py +784 -0
  512. numpy/lib/_utils_impl.pyi +22 -0
  513. numpy/lib/_version.py +153 -0
  514. numpy/lib/_version.pyi +17 -0
  515. numpy/lib/array_utils.py +7 -0
  516. numpy/lib/array_utils.pyi +6 -0
  517. numpy/lib/format.py +24 -0
  518. numpy/lib/format.pyi +24 -0
  519. numpy/lib/introspect.py +94 -0
  520. numpy/lib/introspect.pyi +3 -0
  521. numpy/lib/mixins.py +180 -0
  522. numpy/lib/mixins.pyi +78 -0
  523. numpy/lib/npyio.py +1 -0
  524. numpy/lib/npyio.pyi +5 -0
  525. numpy/lib/recfunctions.py +1681 -0
  526. numpy/lib/recfunctions.pyi +444 -0
  527. numpy/lib/scimath.py +13 -0
  528. numpy/lib/scimath.pyi +12 -0
  529. numpy/lib/stride_tricks.py +1 -0
  530. numpy/lib/stride_tricks.pyi +4 -0
  531. numpy/lib/tests/__init__.py +0 -0
  532. numpy/lib/tests/data/py2-np0-objarr.npy +0 -0
  533. numpy/lib/tests/data/py2-objarr.npy +0 -0
  534. numpy/lib/tests/data/py2-objarr.npz +0 -0
  535. numpy/lib/tests/data/py3-objarr.npy +0 -0
  536. numpy/lib/tests/data/py3-objarr.npz +0 -0
  537. numpy/lib/tests/data/python3.npy +0 -0
  538. numpy/lib/tests/data/win64python2.npy +0 -0
  539. numpy/lib/tests/test__datasource.py +328 -0
  540. numpy/lib/tests/test__iotools.py +358 -0
  541. numpy/lib/tests/test__version.py +64 -0
  542. numpy/lib/tests/test_array_utils.py +32 -0
  543. numpy/lib/tests/test_arraypad.py +1427 -0
  544. numpy/lib/tests/test_arraysetops.py +1302 -0
  545. numpy/lib/tests/test_arrayterator.py +45 -0
  546. numpy/lib/tests/test_format.py +1054 -0
  547. numpy/lib/tests/test_function_base.py +4705 -0
  548. numpy/lib/tests/test_histograms.py +855 -0
  549. numpy/lib/tests/test_index_tricks.py +693 -0
  550. numpy/lib/tests/test_io.py +2857 -0
  551. numpy/lib/tests/test_loadtxt.py +1099 -0
  552. numpy/lib/tests/test_mixins.py +215 -0
  553. numpy/lib/tests/test_nanfunctions.py +1438 -0
  554. numpy/lib/tests/test_packbits.py +376 -0
  555. numpy/lib/tests/test_polynomial.py +325 -0
  556. numpy/lib/tests/test_recfunctions.py +1042 -0
  557. numpy/lib/tests/test_regression.py +231 -0
  558. numpy/lib/tests/test_shape_base.py +813 -0
  559. numpy/lib/tests/test_stride_tricks.py +655 -0
  560. numpy/lib/tests/test_twodim_base.py +559 -0
  561. numpy/lib/tests/test_type_check.py +473 -0
  562. numpy/lib/tests/test_ufunclike.py +97 -0
  563. numpy/lib/tests/test_utils.py +80 -0
  564. numpy/lib/user_array.py +1 -0
  565. numpy/lib/user_array.pyi +1 -0
  566. numpy/linalg/__init__.py +95 -0
  567. numpy/linalg/__init__.pyi +71 -0
  568. numpy/linalg/_linalg.py +3657 -0
  569. numpy/linalg/_linalg.pyi +548 -0
  570. numpy/linalg/_umath_linalg.cpython-313t-aarch64-linux-musl.so +0 -0
  571. numpy/linalg/_umath_linalg.pyi +60 -0
  572. numpy/linalg/lapack_lite.cpython-313t-aarch64-linux-musl.so +0 -0
  573. numpy/linalg/lapack_lite.pyi +143 -0
  574. numpy/linalg/tests/__init__.py +0 -0
  575. numpy/linalg/tests/test_deprecations.py +21 -0
  576. numpy/linalg/tests/test_linalg.py +2442 -0
  577. numpy/linalg/tests/test_regression.py +182 -0
  578. numpy/ma/API_CHANGES.txt +135 -0
  579. numpy/ma/LICENSE +24 -0
  580. numpy/ma/README.rst +236 -0
  581. numpy/ma/__init__.py +53 -0
  582. numpy/ma/__init__.pyi +458 -0
  583. numpy/ma/core.py +8929 -0
  584. numpy/ma/core.pyi +3720 -0
  585. numpy/ma/extras.py +2266 -0
  586. numpy/ma/extras.pyi +297 -0
  587. numpy/ma/mrecords.py +762 -0
  588. numpy/ma/mrecords.pyi +96 -0
  589. numpy/ma/tests/__init__.py +0 -0
  590. numpy/ma/tests/test_arrayobject.py +40 -0
  591. numpy/ma/tests/test_core.py +6008 -0
  592. numpy/ma/tests/test_deprecations.py +65 -0
  593. numpy/ma/tests/test_extras.py +1945 -0
  594. numpy/ma/tests/test_mrecords.py +495 -0
  595. numpy/ma/tests/test_old_ma.py +939 -0
  596. numpy/ma/tests/test_regression.py +83 -0
  597. numpy/ma/tests/test_subclassing.py +469 -0
  598. numpy/ma/testutils.py +294 -0
  599. numpy/ma/testutils.pyi +69 -0
  600. numpy/matlib.py +380 -0
  601. numpy/matlib.pyi +580 -0
  602. numpy/matrixlib/__init__.py +12 -0
  603. numpy/matrixlib/__init__.pyi +3 -0
  604. numpy/matrixlib/defmatrix.py +1119 -0
  605. numpy/matrixlib/defmatrix.pyi +218 -0
  606. numpy/matrixlib/tests/__init__.py +0 -0
  607. numpy/matrixlib/tests/test_defmatrix.py +455 -0
  608. numpy/matrixlib/tests/test_interaction.py +360 -0
  609. numpy/matrixlib/tests/test_masked_matrix.py +240 -0
  610. numpy/matrixlib/tests/test_matrix_linalg.py +110 -0
  611. numpy/matrixlib/tests/test_multiarray.py +17 -0
  612. numpy/matrixlib/tests/test_numeric.py +18 -0
  613. numpy/matrixlib/tests/test_regression.py +31 -0
  614. numpy/polynomial/__init__.py +187 -0
  615. numpy/polynomial/__init__.pyi +31 -0
  616. numpy/polynomial/_polybase.py +1191 -0
  617. numpy/polynomial/_polybase.pyi +262 -0
  618. numpy/polynomial/_polytypes.pyi +501 -0
  619. numpy/polynomial/chebyshev.py +2001 -0
  620. numpy/polynomial/chebyshev.pyi +180 -0
  621. numpy/polynomial/hermite.py +1738 -0
  622. numpy/polynomial/hermite.pyi +106 -0
  623. numpy/polynomial/hermite_e.py +1640 -0
  624. numpy/polynomial/hermite_e.pyi +106 -0
  625. numpy/polynomial/laguerre.py +1673 -0
  626. numpy/polynomial/laguerre.pyi +100 -0
  627. numpy/polynomial/legendre.py +1603 -0
  628. numpy/polynomial/legendre.pyi +100 -0
  629. numpy/polynomial/polynomial.py +1625 -0
  630. numpy/polynomial/polynomial.pyi +109 -0
  631. numpy/polynomial/polyutils.py +759 -0
  632. numpy/polynomial/polyutils.pyi +307 -0
  633. numpy/polynomial/tests/__init__.py +0 -0
  634. numpy/polynomial/tests/test_chebyshev.py +618 -0
  635. numpy/polynomial/tests/test_classes.py +613 -0
  636. numpy/polynomial/tests/test_hermite.py +553 -0
  637. numpy/polynomial/tests/test_hermite_e.py +554 -0
  638. numpy/polynomial/tests/test_laguerre.py +535 -0
  639. numpy/polynomial/tests/test_legendre.py +566 -0
  640. numpy/polynomial/tests/test_polynomial.py +691 -0
  641. numpy/polynomial/tests/test_polyutils.py +123 -0
  642. numpy/polynomial/tests/test_printing.py +557 -0
  643. numpy/polynomial/tests/test_symbol.py +217 -0
  644. numpy/py.typed +0 -0
  645. numpy/random/LICENSE.md +71 -0
  646. numpy/random/__init__.pxd +14 -0
  647. numpy/random/__init__.py +213 -0
  648. numpy/random/__init__.pyi +124 -0
  649. numpy/random/_bounded_integers.cpython-313t-aarch64-linux-musl.so +0 -0
  650. numpy/random/_bounded_integers.pxd +29 -0
  651. numpy/random/_bounded_integers.pyi +1 -0
  652. numpy/random/_common.cpython-313t-aarch64-linux-musl.so +0 -0
  653. numpy/random/_common.pxd +107 -0
  654. numpy/random/_common.pyi +16 -0
  655. numpy/random/_examples/cffi/extending.py +44 -0
  656. numpy/random/_examples/cffi/parse.py +53 -0
  657. numpy/random/_examples/cython/extending.pyx +77 -0
  658. numpy/random/_examples/cython/extending_distributions.pyx +117 -0
  659. numpy/random/_examples/cython/meson.build +53 -0
  660. numpy/random/_examples/numba/extending.py +86 -0
  661. numpy/random/_examples/numba/extending_distributions.py +67 -0
  662. numpy/random/_generator.cpython-313t-aarch64-linux-musl.so +0 -0
  663. numpy/random/_generator.pyi +862 -0
  664. numpy/random/_mt19937.cpython-313t-aarch64-linux-musl.so +0 -0
  665. numpy/random/_mt19937.pyi +27 -0
  666. numpy/random/_pcg64.cpython-313t-aarch64-linux-musl.so +0 -0
  667. numpy/random/_pcg64.pyi +41 -0
  668. numpy/random/_philox.cpython-313t-aarch64-linux-musl.so +0 -0
  669. numpy/random/_philox.pyi +36 -0
  670. numpy/random/_pickle.py +88 -0
  671. numpy/random/_pickle.pyi +43 -0
  672. numpy/random/_sfc64.cpython-313t-aarch64-linux-musl.so +0 -0
  673. numpy/random/_sfc64.pyi +25 -0
  674. numpy/random/bit_generator.cpython-313t-aarch64-linux-musl.so +0 -0
  675. numpy/random/bit_generator.pxd +35 -0
  676. numpy/random/bit_generator.pyi +123 -0
  677. numpy/random/c_distributions.pxd +119 -0
  678. numpy/random/lib/libnpyrandom.a +0 -0
  679. numpy/random/mtrand.cpython-313t-aarch64-linux-musl.so +0 -0
  680. numpy/random/mtrand.pyi +759 -0
  681. numpy/random/tests/__init__.py +0 -0
  682. numpy/random/tests/data/__init__.py +0 -0
  683. numpy/random/tests/data/generator_pcg64_np121.pkl.gz +0 -0
  684. numpy/random/tests/data/generator_pcg64_np126.pkl.gz +0 -0
  685. numpy/random/tests/data/mt19937-testset-1.csv +1001 -0
  686. numpy/random/tests/data/mt19937-testset-2.csv +1001 -0
  687. numpy/random/tests/data/pcg64-testset-1.csv +1001 -0
  688. numpy/random/tests/data/pcg64-testset-2.csv +1001 -0
  689. numpy/random/tests/data/pcg64dxsm-testset-1.csv +1001 -0
  690. numpy/random/tests/data/pcg64dxsm-testset-2.csv +1001 -0
  691. numpy/random/tests/data/philox-testset-1.csv +1001 -0
  692. numpy/random/tests/data/philox-testset-2.csv +1001 -0
  693. numpy/random/tests/data/sfc64-testset-1.csv +1001 -0
  694. numpy/random/tests/data/sfc64-testset-2.csv +1001 -0
  695. numpy/random/tests/data/sfc64_np126.pkl.gz +0 -0
  696. numpy/random/tests/test_direct.py +595 -0
  697. numpy/random/tests/test_extending.py +131 -0
  698. numpy/random/tests/test_generator_mt19937.py +2825 -0
  699. numpy/random/tests/test_generator_mt19937_regressions.py +221 -0
  700. numpy/random/tests/test_random.py +1724 -0
  701. numpy/random/tests/test_randomstate.py +2099 -0
  702. numpy/random/tests/test_randomstate_regression.py +213 -0
  703. numpy/random/tests/test_regression.py +175 -0
  704. numpy/random/tests/test_seed_sequence.py +79 -0
  705. numpy/random/tests/test_smoke.py +882 -0
  706. numpy/rec/__init__.py +2 -0
  707. numpy/rec/__init__.pyi +23 -0
  708. numpy/strings/__init__.py +2 -0
  709. numpy/strings/__init__.pyi +97 -0
  710. numpy/testing/__init__.py +22 -0
  711. numpy/testing/__init__.pyi +107 -0
  712. numpy/testing/_private/__init__.py +0 -0
  713. numpy/testing/_private/__init__.pyi +0 -0
  714. numpy/testing/_private/extbuild.py +250 -0
  715. numpy/testing/_private/extbuild.pyi +25 -0
  716. numpy/testing/_private/utils.py +2830 -0
  717. numpy/testing/_private/utils.pyi +505 -0
  718. numpy/testing/overrides.py +84 -0
  719. numpy/testing/overrides.pyi +10 -0
  720. numpy/testing/print_coercion_tables.py +207 -0
  721. numpy/testing/print_coercion_tables.pyi +26 -0
  722. numpy/testing/tests/__init__.py +0 -0
  723. numpy/testing/tests/test_utils.py +2123 -0
  724. numpy/tests/__init__.py +0 -0
  725. numpy/tests/test__all__.py +10 -0
  726. numpy/tests/test_configtool.py +51 -0
  727. numpy/tests/test_ctypeslib.py +383 -0
  728. numpy/tests/test_lazyloading.py +42 -0
  729. numpy/tests/test_matlib.py +59 -0
  730. numpy/tests/test_numpy_config.py +47 -0
  731. numpy/tests/test_numpy_version.py +54 -0
  732. numpy/tests/test_public_api.py +804 -0
  733. numpy/tests/test_reloading.py +76 -0
  734. numpy/tests/test_scripts.py +48 -0
  735. numpy/tests/test_warnings.py +79 -0
  736. numpy/typing/__init__.py +233 -0
  737. numpy/typing/__init__.pyi +3 -0
  738. numpy/typing/mypy_plugin.py +200 -0
  739. numpy/typing/tests/__init__.py +0 -0
  740. numpy/typing/tests/data/fail/arithmetic.pyi +126 -0
  741. numpy/typing/tests/data/fail/array_constructors.pyi +34 -0
  742. numpy/typing/tests/data/fail/array_like.pyi +15 -0
  743. numpy/typing/tests/data/fail/array_pad.pyi +6 -0
  744. numpy/typing/tests/data/fail/arrayprint.pyi +15 -0
  745. numpy/typing/tests/data/fail/arrayterator.pyi +14 -0
  746. numpy/typing/tests/data/fail/bitwise_ops.pyi +17 -0
  747. numpy/typing/tests/data/fail/char.pyi +63 -0
  748. numpy/typing/tests/data/fail/chararray.pyi +61 -0
  749. numpy/typing/tests/data/fail/comparisons.pyi +27 -0
  750. numpy/typing/tests/data/fail/constants.pyi +3 -0
  751. numpy/typing/tests/data/fail/datasource.pyi +16 -0
  752. numpy/typing/tests/data/fail/dtype.pyi +17 -0
  753. numpy/typing/tests/data/fail/einsumfunc.pyi +12 -0
  754. numpy/typing/tests/data/fail/flatiter.pyi +38 -0
  755. numpy/typing/tests/data/fail/fromnumeric.pyi +148 -0
  756. numpy/typing/tests/data/fail/histograms.pyi +12 -0
  757. numpy/typing/tests/data/fail/index_tricks.pyi +14 -0
  758. numpy/typing/tests/data/fail/lib_function_base.pyi +60 -0
  759. numpy/typing/tests/data/fail/lib_polynomial.pyi +29 -0
  760. numpy/typing/tests/data/fail/lib_utils.pyi +3 -0
  761. numpy/typing/tests/data/fail/lib_version.pyi +6 -0
  762. numpy/typing/tests/data/fail/linalg.pyi +52 -0
  763. numpy/typing/tests/data/fail/ma.pyi +155 -0
  764. numpy/typing/tests/data/fail/memmap.pyi +5 -0
  765. numpy/typing/tests/data/fail/modules.pyi +17 -0
  766. numpy/typing/tests/data/fail/multiarray.pyi +52 -0
  767. numpy/typing/tests/data/fail/ndarray.pyi +11 -0
  768. numpy/typing/tests/data/fail/ndarray_misc.pyi +49 -0
  769. numpy/typing/tests/data/fail/nditer.pyi +8 -0
  770. numpy/typing/tests/data/fail/nested_sequence.pyi +17 -0
  771. numpy/typing/tests/data/fail/npyio.pyi +24 -0
  772. numpy/typing/tests/data/fail/numerictypes.pyi +5 -0
  773. numpy/typing/tests/data/fail/random.pyi +62 -0
  774. numpy/typing/tests/data/fail/rec.pyi +17 -0
  775. numpy/typing/tests/data/fail/scalars.pyi +86 -0
  776. numpy/typing/tests/data/fail/shape.pyi +7 -0
  777. numpy/typing/tests/data/fail/shape_base.pyi +8 -0
  778. numpy/typing/tests/data/fail/stride_tricks.pyi +9 -0
  779. numpy/typing/tests/data/fail/strings.pyi +52 -0
  780. numpy/typing/tests/data/fail/testing.pyi +28 -0
  781. numpy/typing/tests/data/fail/twodim_base.pyi +39 -0
  782. numpy/typing/tests/data/fail/type_check.pyi +12 -0
  783. numpy/typing/tests/data/fail/ufunc_config.pyi +21 -0
  784. numpy/typing/tests/data/fail/ufunclike.pyi +21 -0
  785. numpy/typing/tests/data/fail/ufuncs.pyi +17 -0
  786. numpy/typing/tests/data/fail/warnings_and_errors.pyi +5 -0
  787. numpy/typing/tests/data/misc/extended_precision.pyi +9 -0
  788. numpy/typing/tests/data/mypy.ini +8 -0
  789. numpy/typing/tests/data/pass/arithmetic.py +614 -0
  790. numpy/typing/tests/data/pass/array_constructors.py +138 -0
  791. numpy/typing/tests/data/pass/array_like.py +43 -0
  792. numpy/typing/tests/data/pass/arrayprint.py +37 -0
  793. numpy/typing/tests/data/pass/arrayterator.py +28 -0
  794. numpy/typing/tests/data/pass/bitwise_ops.py +131 -0
  795. numpy/typing/tests/data/pass/comparisons.py +316 -0
  796. numpy/typing/tests/data/pass/dtype.py +57 -0
  797. numpy/typing/tests/data/pass/einsumfunc.py +36 -0
  798. numpy/typing/tests/data/pass/flatiter.py +26 -0
  799. numpy/typing/tests/data/pass/fromnumeric.py +272 -0
  800. numpy/typing/tests/data/pass/index_tricks.py +62 -0
  801. numpy/typing/tests/data/pass/lib_user_array.py +22 -0
  802. numpy/typing/tests/data/pass/lib_utils.py +19 -0
  803. numpy/typing/tests/data/pass/lib_version.py +18 -0
  804. numpy/typing/tests/data/pass/literal.py +52 -0
  805. numpy/typing/tests/data/pass/ma.py +199 -0
  806. numpy/typing/tests/data/pass/mod.py +149 -0
  807. numpy/typing/tests/data/pass/modules.py +45 -0
  808. numpy/typing/tests/data/pass/multiarray.py +77 -0
  809. numpy/typing/tests/data/pass/ndarray_conversion.py +81 -0
  810. numpy/typing/tests/data/pass/ndarray_misc.py +199 -0
  811. numpy/typing/tests/data/pass/ndarray_shape_manipulation.py +47 -0
  812. numpy/typing/tests/data/pass/nditer.py +4 -0
  813. numpy/typing/tests/data/pass/numeric.py +90 -0
  814. numpy/typing/tests/data/pass/numerictypes.py +17 -0
  815. numpy/typing/tests/data/pass/random.py +1498 -0
  816. numpy/typing/tests/data/pass/recfunctions.py +164 -0
  817. numpy/typing/tests/data/pass/scalars.py +249 -0
  818. numpy/typing/tests/data/pass/shape.py +19 -0
  819. numpy/typing/tests/data/pass/simple.py +170 -0
  820. numpy/typing/tests/data/pass/ufunc_config.py +64 -0
  821. numpy/typing/tests/data/pass/ufunclike.py +52 -0
  822. numpy/typing/tests/data/pass/ufuncs.py +16 -0
  823. numpy/typing/tests/data/pass/warnings_and_errors.py +6 -0
  824. numpy/typing/tests/data/reveal/arithmetic.pyi +719 -0
  825. numpy/typing/tests/data/reveal/array_api_info.pyi +70 -0
  826. numpy/typing/tests/data/reveal/array_constructors.pyi +277 -0
  827. numpy/typing/tests/data/reveal/arraypad.pyi +27 -0
  828. numpy/typing/tests/data/reveal/arrayprint.pyi +25 -0
  829. numpy/typing/tests/data/reveal/arraysetops.pyi +74 -0
  830. numpy/typing/tests/data/reveal/arrayterator.pyi +27 -0
  831. numpy/typing/tests/data/reveal/bitwise_ops.pyi +166 -0
  832. numpy/typing/tests/data/reveal/char.pyi +225 -0
  833. numpy/typing/tests/data/reveal/chararray.pyi +138 -0
  834. numpy/typing/tests/data/reveal/comparisons.pyi +264 -0
  835. numpy/typing/tests/data/reveal/constants.pyi +14 -0
  836. numpy/typing/tests/data/reveal/ctypeslib.pyi +81 -0
  837. numpy/typing/tests/data/reveal/datasource.pyi +23 -0
  838. numpy/typing/tests/data/reveal/dtype.pyi +132 -0
  839. numpy/typing/tests/data/reveal/einsumfunc.pyi +39 -0
  840. numpy/typing/tests/data/reveal/emath.pyi +54 -0
  841. numpy/typing/tests/data/reveal/fft.pyi +37 -0
  842. numpy/typing/tests/data/reveal/flatiter.pyi +86 -0
  843. numpy/typing/tests/data/reveal/fromnumeric.pyi +347 -0
  844. numpy/typing/tests/data/reveal/getlimits.pyi +53 -0
  845. numpy/typing/tests/data/reveal/histograms.pyi +25 -0
  846. numpy/typing/tests/data/reveal/index_tricks.pyi +70 -0
  847. numpy/typing/tests/data/reveal/lib_function_base.pyi +409 -0
  848. numpy/typing/tests/data/reveal/lib_polynomial.pyi +147 -0
  849. numpy/typing/tests/data/reveal/lib_utils.pyi +17 -0
  850. numpy/typing/tests/data/reveal/lib_version.pyi +20 -0
  851. numpy/typing/tests/data/reveal/linalg.pyi +154 -0
  852. numpy/typing/tests/data/reveal/ma.pyi +1098 -0
  853. numpy/typing/tests/data/reveal/matrix.pyi +73 -0
  854. numpy/typing/tests/data/reveal/memmap.pyi +19 -0
  855. numpy/typing/tests/data/reveal/mod.pyi +178 -0
  856. numpy/typing/tests/data/reveal/modules.pyi +51 -0
  857. numpy/typing/tests/data/reveal/multiarray.pyi +197 -0
  858. numpy/typing/tests/data/reveal/nbit_base_example.pyi +20 -0
  859. numpy/typing/tests/data/reveal/ndarray_assignability.pyi +82 -0
  860. numpy/typing/tests/data/reveal/ndarray_conversion.pyi +83 -0
  861. numpy/typing/tests/data/reveal/ndarray_misc.pyi +246 -0
  862. numpy/typing/tests/data/reveal/ndarray_shape_manipulation.pyi +47 -0
  863. numpy/typing/tests/data/reveal/nditer.pyi +49 -0
  864. numpy/typing/tests/data/reveal/nested_sequence.pyi +25 -0
  865. numpy/typing/tests/data/reveal/npyio.pyi +83 -0
  866. numpy/typing/tests/data/reveal/numeric.pyi +170 -0
  867. numpy/typing/tests/data/reveal/numerictypes.pyi +16 -0
  868. numpy/typing/tests/data/reveal/polynomial_polybase.pyi +217 -0
  869. numpy/typing/tests/data/reveal/polynomial_polyutils.pyi +218 -0
  870. numpy/typing/tests/data/reveal/polynomial_series.pyi +138 -0
  871. numpy/typing/tests/data/reveal/random.pyi +1546 -0
  872. numpy/typing/tests/data/reveal/rec.pyi +171 -0
  873. numpy/typing/tests/data/reveal/scalars.pyi +191 -0
  874. numpy/typing/tests/data/reveal/shape.pyi +13 -0
  875. numpy/typing/tests/data/reveal/shape_base.pyi +52 -0
  876. numpy/typing/tests/data/reveal/stride_tricks.pyi +27 -0
  877. numpy/typing/tests/data/reveal/strings.pyi +196 -0
  878. numpy/typing/tests/data/reveal/testing.pyi +198 -0
  879. numpy/typing/tests/data/reveal/twodim_base.pyi +225 -0
  880. numpy/typing/tests/data/reveal/type_check.pyi +67 -0
  881. numpy/typing/tests/data/reveal/ufunc_config.pyi +29 -0
  882. numpy/typing/tests/data/reveal/ufunclike.pyi +31 -0
  883. numpy/typing/tests/data/reveal/ufuncs.pyi +142 -0
  884. numpy/typing/tests/data/reveal/warnings_and_errors.pyi +11 -0
  885. numpy/typing/tests/test_isfile.py +38 -0
  886. numpy/typing/tests/test_runtime.py +110 -0
  887. numpy/typing/tests/test_typing.py +205 -0
  888. numpy/version.py +11 -0
  889. numpy/version.pyi +9 -0
  890. numpy-2.4.0.dist-info/METADATA +139 -0
  891. numpy-2.4.0.dist-info/RECORD +915 -0
  892. numpy-2.4.0.dist-info/WHEEL +5 -0
  893. numpy-2.4.0.dist-info/entry_points.txt +13 -0
  894. numpy-2.4.0.dist-info/licenses/LICENSE.txt +935 -0
  895. numpy-2.4.0.dist-info/licenses/numpy/_core/include/numpy/libdivide/LICENSE.txt +21 -0
  896. numpy-2.4.0.dist-info/licenses/numpy/_core/src/common/pythoncapi-compat/COPYING +14 -0
  897. numpy-2.4.0.dist-info/licenses/numpy/_core/src/highway/LICENSE +371 -0
  898. numpy-2.4.0.dist-info/licenses/numpy/_core/src/multiarray/dragon4_LICENSE.txt +27 -0
  899. numpy-2.4.0.dist-info/licenses/numpy/_core/src/npysort/x86-simd-sort/LICENSE.md +28 -0
  900. numpy-2.4.0.dist-info/licenses/numpy/_core/src/umath/svml/LICENSE +30 -0
  901. numpy-2.4.0.dist-info/licenses/numpy/fft/pocketfft/LICENSE.md +25 -0
  902. numpy-2.4.0.dist-info/licenses/numpy/linalg/lapack_lite/LICENSE.txt +48 -0
  903. numpy-2.4.0.dist-info/licenses/numpy/ma/LICENSE +24 -0
  904. numpy-2.4.0.dist-info/licenses/numpy/random/LICENSE.md +71 -0
  905. numpy-2.4.0.dist-info/licenses/numpy/random/src/distributions/LICENSE.md +61 -0
  906. numpy-2.4.0.dist-info/licenses/numpy/random/src/mt19937/LICENSE.md +61 -0
  907. numpy-2.4.0.dist-info/licenses/numpy/random/src/pcg64/LICENSE.md +22 -0
  908. numpy-2.4.0.dist-info/licenses/numpy/random/src/philox/LICENSE.md +31 -0
  909. numpy-2.4.0.dist-info/licenses/numpy/random/src/sfc64/LICENSE.md +27 -0
  910. numpy-2.4.0.dist-info/licenses/numpy/random/src/splitmix64/LICENSE.md +9 -0
  911. numpy.libs/libgcc_s-2d945d6c-767fb991.so.1 +0 -0
  912. numpy.libs/libgcc_s-2d945d6c.so.1 +0 -0
  913. numpy.libs/libgfortran-67378ab2-e7e7cfab.so.5.0.0 +0 -0
  914. numpy.libs/libscipy_openblas64_-1fc386ee.so +0 -0
  915. numpy.libs/libstdc++-85f2cd6d.so.6.0.33 +0 -0
@@ -0,0 +1,1099 @@
1
+ """
2
+ Tests specific to `np.loadtxt` added during the move of loadtxt to be backed
3
+ by C code.
4
+ These tests complement those found in `test_io.py`.
5
+ """
6
+
7
+ import os
8
+ import sys
9
+ from io import StringIO
10
+ from tempfile import NamedTemporaryFile, mkstemp
11
+
12
+ import pytest
13
+
14
+ import numpy as np
15
+ from numpy.ma.testutils import assert_equal
16
+ from numpy.testing import HAS_REFCOUNT, IS_PYPY, assert_array_equal
17
+
18
+
19
+ def test_scientific_notation():
20
+ """Test that both 'e' and 'E' are parsed correctly."""
21
+ data = StringIO(
22
+
23
+ "1.0e-1,2.0E1,3.0\n"
24
+ "4.0e-2,5.0E-1,6.0\n"
25
+ "7.0e-3,8.0E1,9.0\n"
26
+ "0.0e-4,1.0E-1,2.0"
27
+
28
+ )
29
+ expected = np.array(
30
+ [[0.1, 20., 3.0], [0.04, 0.5, 6], [0.007, 80., 9], [0, 0.1, 2]]
31
+ )
32
+ assert_array_equal(np.loadtxt(data, delimiter=","), expected)
33
+
34
+
35
+ @pytest.mark.parametrize("comment", ["..", "//", "@-", "this is a comment:"])
36
+ def test_comment_multiple_chars(comment):
37
+ content = "# IGNORE\n1.5, 2.5# ABC\n3.0,4.0# XXX\n5.5,6.0\n"
38
+ txt = StringIO(content.replace("#", comment))
39
+ a = np.loadtxt(txt, delimiter=",", comments=comment)
40
+ assert_equal(a, [[1.5, 2.5], [3.0, 4.0], [5.5, 6.0]])
41
+
42
+
43
+ def mixed_types_structured():
44
+ """
45
+ Function providing heterogeneous input data with a structured dtype, along
46
+ with the associated structured array.
47
+ """
48
+ data = StringIO(
49
+
50
+ "1000;2.4;alpha;-34\n"
51
+ "2000;3.1;beta;29\n"
52
+ "3500;9.9;gamma;120\n"
53
+ "4090;8.1;delta;0\n"
54
+ "5001;4.4;epsilon;-99\n"
55
+ "6543;7.8;omega;-1\n"
56
+
57
+ )
58
+ dtype = np.dtype(
59
+ [('f0', np.uint16), ('f1', np.float64), ('f2', 'S7'), ('f3', np.int8)]
60
+ )
61
+ expected = np.array(
62
+ [
63
+ (1000, 2.4, "alpha", -34),
64
+ (2000, 3.1, "beta", 29),
65
+ (3500, 9.9, "gamma", 120),
66
+ (4090, 8.1, "delta", 0),
67
+ (5001, 4.4, "epsilon", -99),
68
+ (6543, 7.8, "omega", -1)
69
+ ],
70
+ dtype=dtype
71
+ )
72
+ return data, dtype, expected
73
+
74
+
75
+ @pytest.mark.parametrize('skiprows', [0, 1, 2, 3])
76
+ def test_structured_dtype_and_skiprows_no_empty_lines(skiprows):
77
+ data, dtype, expected = mixed_types_structured()
78
+ a = np.loadtxt(data, dtype=dtype, delimiter=";", skiprows=skiprows)
79
+ assert_array_equal(a, expected[skiprows:])
80
+
81
+
82
+ def test_unpack_structured():
83
+ data, dtype, expected = mixed_types_structured()
84
+
85
+ a, b, c, d = np.loadtxt(data, dtype=dtype, delimiter=";", unpack=True)
86
+ assert_array_equal(a, expected["f0"])
87
+ assert_array_equal(b, expected["f1"])
88
+ assert_array_equal(c, expected["f2"])
89
+ assert_array_equal(d, expected["f3"])
90
+
91
+
92
+ def test_structured_dtype_with_shape():
93
+ dtype = np.dtype([("a", "u1", 2), ("b", "u1", 2)])
94
+ data = StringIO("0,1,2,3\n6,7,8,9\n")
95
+ expected = np.array([((0, 1), (2, 3)), ((6, 7), (8, 9))], dtype=dtype)
96
+ assert_array_equal(np.loadtxt(data, delimiter=",", dtype=dtype), expected)
97
+
98
+
99
+ def test_structured_dtype_with_multi_shape():
100
+ dtype = np.dtype([("a", "u1", (2, 2))])
101
+ data = StringIO("0 1 2 3\n")
102
+ expected = np.array([(((0, 1), (2, 3)),)], dtype=dtype)
103
+ assert_array_equal(np.loadtxt(data, dtype=dtype), expected)
104
+
105
+
106
+ def test_nested_structured_subarray():
107
+ # Test from gh-16678
108
+ point = np.dtype([('x', float), ('y', float)])
109
+ dt = np.dtype([('code', int), ('points', point, (2,))])
110
+ data = StringIO("100,1,2,3,4\n200,5,6,7,8\n")
111
+ expected = np.array(
112
+ [
113
+ (100, [(1., 2.), (3., 4.)]),
114
+ (200, [(5., 6.), (7., 8.)]),
115
+ ],
116
+ dtype=dt
117
+ )
118
+ assert_array_equal(np.loadtxt(data, dtype=dt, delimiter=","), expected)
119
+
120
+
121
+ def test_structured_dtype_offsets():
122
+ # An aligned structured dtype will have additional padding
123
+ dt = np.dtype("i1, i4, i1, i4, i1, i4", align=True)
124
+ data = StringIO("1,2,3,4,5,6\n7,8,9,10,11,12\n")
125
+ expected = np.array([(1, 2, 3, 4, 5, 6), (7, 8, 9, 10, 11, 12)], dtype=dt)
126
+ assert_array_equal(np.loadtxt(data, delimiter=",", dtype=dt), expected)
127
+
128
+
129
+ @pytest.mark.parametrize("param", ("skiprows", "max_rows"))
130
+ def test_exception_negative_row_limits(param):
131
+ """skiprows and max_rows should raise for negative parameters."""
132
+ with pytest.raises(ValueError, match="argument must be nonnegative"):
133
+ np.loadtxt("foo.bar", **{param: -3})
134
+
135
+
136
+ @pytest.mark.parametrize("param", ("skiprows", "max_rows"))
137
+ def test_exception_noninteger_row_limits(param):
138
+ with pytest.raises(TypeError, match="argument must be an integer"):
139
+ np.loadtxt("foo.bar", **{param: 1.0})
140
+
141
+
142
+ @pytest.mark.parametrize(
143
+ "data, shape",
144
+ [
145
+ ("1 2 3 4 5\n", (1, 5)), # Single row
146
+ ("1\n2\n3\n4\n5\n", (5, 1)), # Single column
147
+ ]
148
+ )
149
+ def test_ndmin_single_row_or_col(data, shape):
150
+ arr = np.array([1, 2, 3, 4, 5])
151
+ arr2d = arr.reshape(shape)
152
+
153
+ assert_array_equal(np.loadtxt(StringIO(data), dtype=int), arr)
154
+ assert_array_equal(np.loadtxt(StringIO(data), dtype=int, ndmin=0), arr)
155
+ assert_array_equal(np.loadtxt(StringIO(data), dtype=int, ndmin=1), arr)
156
+ assert_array_equal(np.loadtxt(StringIO(data), dtype=int, ndmin=2), arr2d)
157
+
158
+
159
+ @pytest.mark.parametrize("badval", [-1, 3, None, "plate of shrimp"])
160
+ def test_bad_ndmin(badval):
161
+ with pytest.raises(ValueError, match="Illegal value of ndmin keyword"):
162
+ np.loadtxt("foo.bar", ndmin=badval)
163
+
164
+
165
+ @pytest.mark.parametrize(
166
+ "ws",
167
+ (
168
+ " ", # space
169
+ "\t", # tab
170
+ "\u2003", # em
171
+ "\u00A0", # non-break
172
+ "\u3000", # ideographic space
173
+ )
174
+ )
175
+ def test_blank_lines_spaces_delimit(ws):
176
+ txt = StringIO(
177
+ f"1 2{ws}30\n\n{ws}\n"
178
+ f"4 5 60{ws}\n {ws} \n"
179
+ f"7 8 {ws} 90\n # comment\n"
180
+ f"3 2 1"
181
+ )
182
+ # NOTE: It is unclear that the ` # comment` should succeed. Except
183
+ # for delimiter=None, which should use any whitespace (and maybe
184
+ # should just be implemented closer to Python
185
+ expected = np.array([[1, 2, 30], [4, 5, 60], [7, 8, 90], [3, 2, 1]])
186
+ assert_equal(
187
+ np.loadtxt(txt, dtype=int, delimiter=None, comments="#"), expected
188
+ )
189
+
190
+
191
+ def test_blank_lines_normal_delimiter():
192
+ txt = StringIO('1,2,30\n\n4,5,60\n\n7,8,90\n# comment\n3,2,1')
193
+ expected = np.array([[1, 2, 30], [4, 5, 60], [7, 8, 90], [3, 2, 1]])
194
+ assert_equal(
195
+ np.loadtxt(txt, dtype=int, delimiter=',', comments="#"), expected
196
+ )
197
+
198
+
199
+ @pytest.mark.parametrize("dtype", (float, object))
200
+ def test_maxrows_no_blank_lines(dtype):
201
+ txt = StringIO("1.5,2.5\n3.0,4.0\n5.5,6.0")
202
+ res = np.loadtxt(txt, dtype=dtype, delimiter=",", max_rows=2)
203
+ assert_equal(res.dtype, dtype)
204
+ assert_equal(res, np.array([["1.5", "2.5"], ["3.0", "4.0"]], dtype=dtype))
205
+
206
+
207
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
208
+ reason="PyPy bug in error formatting")
209
+ @pytest.mark.parametrize("dtype", (np.dtype("f8"), np.dtype("i2")))
210
+ def test_exception_message_bad_values(dtype):
211
+ txt = StringIO("1,2\n3,XXX\n5,6")
212
+ msg = f"could not convert string 'XXX' to {dtype} at row 1, column 2"
213
+ with pytest.raises(ValueError, match=msg):
214
+ np.loadtxt(txt, dtype=dtype, delimiter=",")
215
+
216
+
217
+ def test_converters_negative_indices():
218
+ txt = StringIO('1.5,2.5\n3.0,XXX\n5.5,6.0')
219
+ conv = {-1: lambda s: np.nan if s == 'XXX' else float(s)}
220
+ expected = np.array([[1.5, 2.5], [3.0, np.nan], [5.5, 6.0]])
221
+ res = np.loadtxt(txt, dtype=np.float64, delimiter=",", converters=conv)
222
+ assert_equal(res, expected)
223
+
224
+
225
+ def test_converters_negative_indices_with_usecols():
226
+ txt = StringIO('1.5,2.5,3.5\n3.0,4.0,XXX\n5.5,6.0,7.5\n')
227
+ conv = {-1: lambda s: np.nan if s == 'XXX' else float(s)}
228
+ expected = np.array([[1.5, 3.5], [3.0, np.nan], [5.5, 7.5]])
229
+ res = np.loadtxt(
230
+ txt,
231
+ dtype=np.float64,
232
+ delimiter=",",
233
+ converters=conv,
234
+ usecols=[0, -1],
235
+ )
236
+ assert_equal(res, expected)
237
+
238
+ # Second test with variable number of rows:
239
+ res = np.loadtxt(StringIO('''0,1,2\n0,1,2,3,4'''), delimiter=",",
240
+ usecols=[0, -1], converters={-1: (lambda x: -1)})
241
+ assert_array_equal(res, [[0, -1], [0, -1]])
242
+
243
+
244
+ def test_ragged_error():
245
+ rows = ["1,2,3", "1,2,3", "4,3,2,1"]
246
+ with pytest.raises(ValueError,
247
+ match="the number of columns changed from 3 to 4 at row 3"):
248
+ np.loadtxt(rows, delimiter=",")
249
+
250
+
251
+ def test_ragged_usecols():
252
+ # usecols, and negative ones, work even with varying number of columns.
253
+ txt = StringIO("0,0,XXX\n0,XXX,0,XXX\n0,XXX,XXX,0,XXX\n")
254
+ expected = np.array([[0, 0], [0, 0], [0, 0]])
255
+ res = np.loadtxt(txt, dtype=float, delimiter=",", usecols=[0, -2])
256
+ assert_equal(res, expected)
257
+
258
+ txt = StringIO("0,0,XXX\n0\n0,XXX,XXX,0,XXX\n")
259
+ with pytest.raises(ValueError,
260
+ match="invalid column index -2 at row 2 with 1 columns"):
261
+ # There is no -2 column in the second row:
262
+ np.loadtxt(txt, dtype=float, delimiter=",", usecols=[0, -2])
263
+
264
+
265
+ def test_empty_usecols():
266
+ txt = StringIO("0,0,XXX\n0,XXX,0,XXX\n0,XXX,XXX,0,XXX\n")
267
+ res = np.loadtxt(txt, dtype=np.dtype([]), delimiter=",", usecols=[])
268
+ assert res.shape == (3,)
269
+ assert res.dtype == np.dtype([])
270
+
271
+
272
+ @pytest.mark.parametrize("c1", ["a", "の", "🫕"])
273
+ @pytest.mark.parametrize("c2", ["a", "の", "🫕"])
274
+ def test_large_unicode_characters(c1, c2):
275
+ # c1 and c2 span ascii, 16bit and 32bit range.
276
+ txt = StringIO(f"a,{c1},c,1.0\ne,{c2},2.0,g")
277
+ res = np.loadtxt(txt, dtype=np.dtype('U12'), delimiter=",")
278
+ expected = np.array(
279
+ [f"a,{c1},c,1.0".split(","), f"e,{c2},2.0,g".split(",")],
280
+ dtype=np.dtype('U12')
281
+ )
282
+ assert_equal(res, expected)
283
+
284
+
285
+ def test_unicode_with_converter():
286
+ txt = StringIO("cat,dog\nαβγ,δεζ\nabc,def\n")
287
+ conv = {0: lambda s: s.upper()}
288
+ res = np.loadtxt(
289
+ txt,
290
+ dtype=np.dtype("U12"),
291
+ converters=conv,
292
+ delimiter=",",
293
+ encoding=None
294
+ )
295
+ expected = np.array([['CAT', 'dog'], ['ΑΒΓ', 'δεζ'], ['ABC', 'def']])
296
+ assert_equal(res, expected)
297
+
298
+
299
+ def test_converter_with_structured_dtype():
300
+ txt = StringIO('1.5,2.5,Abc\n3.0,4.0,dEf\n5.5,6.0,ghI\n')
301
+ dt = np.dtype([('m', np.int32), ('r', np.float32), ('code', 'U8')])
302
+ conv = {0: lambda s: int(10 * float(s)), -1: lambda s: s.upper()}
303
+ res = np.loadtxt(txt, dtype=dt, delimiter=",", converters=conv)
304
+ expected = np.array(
305
+ [(15, 2.5, 'ABC'), (30, 4.0, 'DEF'), (55, 6.0, 'GHI')], dtype=dt
306
+ )
307
+ assert_equal(res, expected)
308
+
309
+
310
+ def test_converter_with_unicode_dtype():
311
+ """
312
+ With the 'bytes' encoding, tokens are encoded prior to being
313
+ passed to the converter. This means that the output of the converter may
314
+ be bytes instead of unicode as expected by `read_rows`.
315
+
316
+ This test checks that outputs from the above scenario are properly decoded
317
+ prior to parsing by `read_rows`.
318
+ """
319
+ txt = StringIO('abc,def\nrst,xyz')
320
+ conv = bytes.upper
321
+ res = np.loadtxt(
322
+ txt, dtype=np.dtype("U3"), converters=conv, delimiter=",",
323
+ encoding="bytes")
324
+ expected = np.array([['ABC', 'DEF'], ['RST', 'XYZ']])
325
+ assert_equal(res, expected)
326
+
327
+
328
+ def test_read_huge_row():
329
+ row = "1.5, 2.5," * 50000
330
+ row = row[:-1] + "\n"
331
+ txt = StringIO(row * 2)
332
+ res = np.loadtxt(txt, delimiter=",", dtype=float)
333
+ assert_equal(res, np.tile([1.5, 2.5], (2, 50000)))
334
+
335
+
336
+ @pytest.mark.parametrize("dtype", "edfgFDG")
337
+ def test_huge_float(dtype):
338
+ # Covers a non-optimized path that is rarely taken:
339
+ field = "0" * 1000 + ".123456789"
340
+ dtype = np.dtype(dtype)
341
+ value = np.loadtxt([field], dtype=dtype)[()]
342
+ assert value == dtype.type("0.123456789")
343
+
344
+
345
+ @pytest.mark.parametrize(
346
+ ("given_dtype", "expected_dtype"),
347
+ [
348
+ ("S", np.dtype("S5")),
349
+ ("U", np.dtype("U5")),
350
+ ],
351
+ )
352
+ def test_string_no_length_given(given_dtype, expected_dtype):
353
+ """
354
+ The given dtype is just 'S' or 'U' with no length. In these cases, the
355
+ length of the resulting dtype is determined by the longest string found
356
+ in the file.
357
+ """
358
+ txt = StringIO("AAA,5-1\nBBBBB,0-3\nC,4-9\n")
359
+ res = np.loadtxt(txt, dtype=given_dtype, delimiter=",")
360
+ expected = np.array(
361
+ [['AAA', '5-1'], ['BBBBB', '0-3'], ['C', '4-9']], dtype=expected_dtype
362
+ )
363
+ assert_equal(res, expected)
364
+ assert_equal(res.dtype, expected_dtype)
365
+
366
+
367
+ def test_float_conversion():
368
+ """
369
+ Some tests that the conversion to float64 works as accurately as the
370
+ Python built-in `float` function. In a naive version of the float parser,
371
+ these strings resulted in values that were off by an ULP or two.
372
+ """
373
+ strings = [
374
+ '0.9999999999999999',
375
+ '9876543210.123456',
376
+ '5.43215432154321e+300',
377
+ '0.901',
378
+ '0.333',
379
+ ]
380
+ txt = StringIO('\n'.join(strings))
381
+ res = np.loadtxt(txt)
382
+ expected = np.array([float(s) for s in strings])
383
+ assert_equal(res, expected)
384
+
385
+
386
+ def test_bool():
387
+ # Simple test for bool via integer
388
+ txt = StringIO("1, 0\n10, -1")
389
+ res = np.loadtxt(txt, dtype=bool, delimiter=",")
390
+ assert res.dtype == bool
391
+ assert_array_equal(res, [[True, False], [True, True]])
392
+ # Make sure we use only 1 and 0 on the byte level:
393
+ assert_array_equal(res.view(np.uint8), [[1, 0], [1, 1]])
394
+
395
+
396
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
397
+ reason="PyPy bug in error formatting")
398
+ @pytest.mark.parametrize("dtype", np.typecodes["AllInteger"])
399
+ @pytest.mark.filterwarnings("error:.*integer via a float.*:DeprecationWarning")
400
+ def test_integer_signs(dtype):
401
+ dtype = np.dtype(dtype)
402
+ assert np.loadtxt(["+2"], dtype=dtype) == 2
403
+ if dtype.kind == "u":
404
+ with pytest.raises(ValueError):
405
+ np.loadtxt(["-1\n"], dtype=dtype)
406
+ else:
407
+ assert np.loadtxt(["-2\n"], dtype=dtype) == -2
408
+
409
+ for sign in ["++", "+-", "--", "-+"]:
410
+ with pytest.raises(ValueError):
411
+ np.loadtxt([f"{sign}2\n"], dtype=dtype)
412
+
413
+
414
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
415
+ reason="PyPy bug in error formatting")
416
+ @pytest.mark.parametrize("dtype", np.typecodes["AllInteger"])
417
+ @pytest.mark.filterwarnings("error:.*integer via a float.*:DeprecationWarning")
418
+ def test_implicit_cast_float_to_int_fails(dtype):
419
+ txt = StringIO("1.0, 2.1, 3.7\n4, 5, 6")
420
+ with pytest.raises(ValueError):
421
+ np.loadtxt(txt, dtype=dtype, delimiter=",")
422
+
423
+ @pytest.mark.parametrize("dtype", (np.complex64, np.complex128))
424
+ @pytest.mark.parametrize("with_parens", (False, True))
425
+ def test_complex_parsing(dtype, with_parens):
426
+ s = "(1.0-2.5j),3.75,(7+-5.0j)\n(4),(-19e2j),(0)"
427
+ if not with_parens:
428
+ s = s.replace("(", "").replace(")", "")
429
+
430
+ res = np.loadtxt(StringIO(s), dtype=dtype, delimiter=",")
431
+ expected = np.array(
432
+ [[1.0 - 2.5j, 3.75, 7 - 5j], [4.0, -1900j, 0]], dtype=dtype
433
+ )
434
+ assert_equal(res, expected)
435
+
436
+
437
+ def test_read_from_generator():
438
+ def gen():
439
+ for i in range(4):
440
+ yield f"{i},{2 * i},{i**2}"
441
+
442
+ res = np.loadtxt(gen(), dtype=int, delimiter=",")
443
+ expected = np.array([[0, 0, 0], [1, 2, 1], [2, 4, 4], [3, 6, 9]])
444
+ assert_equal(res, expected)
445
+
446
+
447
+ def test_read_from_generator_multitype():
448
+ def gen():
449
+ for i in range(3):
450
+ yield f"{i} {i / 4}"
451
+
452
+ res = np.loadtxt(gen(), dtype="i, d", delimiter=" ")
453
+ expected = np.array([(0, 0.0), (1, 0.25), (2, 0.5)], dtype="i, d")
454
+ assert_equal(res, expected)
455
+
456
+
457
+ def test_read_from_bad_generator():
458
+ def gen():
459
+ yield from ["1,2", b"3, 5", 12738]
460
+
461
+ with pytest.raises(
462
+ TypeError, match=r"non-string returned while reading data"):
463
+ np.loadtxt(gen(), dtype="i, i", delimiter=",")
464
+
465
+
466
+ @pytest.mark.skipif(not HAS_REFCOUNT, reason="Python lacks refcounts")
467
+ def test_object_cleanup_on_read_error():
468
+ sentinel = object()
469
+ already_read = 0
470
+
471
+ def conv(x):
472
+ nonlocal already_read
473
+ if already_read > 4999:
474
+ raise ValueError("failed half-way through!")
475
+ already_read += 1
476
+ return sentinel
477
+
478
+ txt = StringIO("x\n" * 10000)
479
+
480
+ with pytest.raises(ValueError, match="at row 5000, column 1"):
481
+ np.loadtxt(txt, dtype=object, converters={0: conv})
482
+
483
+ assert sys.getrefcount(sentinel) == 2
484
+
485
+
486
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
487
+ reason="PyPy bug in error formatting")
488
+ def test_character_not_bytes_compatible():
489
+ """Test exception when a character cannot be encoded as 'S'."""
490
+ data = StringIO("–") # == \u2013
491
+ with pytest.raises(ValueError):
492
+ np.loadtxt(data, dtype="S5")
493
+
494
+
495
+ @pytest.mark.parametrize("conv", (0, [float], ""))
496
+ def test_invalid_converter(conv):
497
+ msg = (
498
+ "converters must be a dictionary mapping columns to converter "
499
+ "functions or a single callable."
500
+ )
501
+ with pytest.raises(TypeError, match=msg):
502
+ np.loadtxt(StringIO("1 2\n3 4"), converters=conv)
503
+
504
+
505
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
506
+ reason="PyPy bug in error formatting")
507
+ def test_converters_dict_raises_non_integer_key():
508
+ with pytest.raises(TypeError, match="keys of the converters dict"):
509
+ np.loadtxt(StringIO("1 2\n3 4"), converters={"a": int})
510
+ with pytest.raises(TypeError, match="keys of the converters dict"):
511
+ np.loadtxt(StringIO("1 2\n3 4"), converters={"a": int}, usecols=0)
512
+
513
+
514
+ @pytest.mark.parametrize("bad_col_ind", (3, -3))
515
+ def test_converters_dict_raises_non_col_key(bad_col_ind):
516
+ data = StringIO("1 2\n3 4")
517
+ with pytest.raises(ValueError, match="converter specified for column"):
518
+ np.loadtxt(data, converters={bad_col_ind: int})
519
+
520
+
521
+ def test_converters_dict_raises_val_not_callable():
522
+ with pytest.raises(TypeError,
523
+ match="values of the converters dictionary must be callable"):
524
+ np.loadtxt(StringIO("1 2\n3 4"), converters={0: 1})
525
+
526
+
527
+ @pytest.mark.parametrize("q", ('"', "'", "`"))
528
+ def test_quoted_field(q):
529
+ txt = StringIO(
530
+ f"{q}alpha, x{q}, 2.5\n{q}beta, y{q}, 4.5\n{q}gamma, z{q}, 5.0\n"
531
+ )
532
+ dtype = np.dtype([('f0', 'U8'), ('f1', np.float64)])
533
+ expected = np.array(
534
+ [("alpha, x", 2.5), ("beta, y", 4.5), ("gamma, z", 5.0)], dtype=dtype
535
+ )
536
+
537
+ res = np.loadtxt(txt, dtype=dtype, delimiter=",", quotechar=q)
538
+ assert_array_equal(res, expected)
539
+
540
+
541
+ @pytest.mark.parametrize("q", ('"', "'", "`"))
542
+ def test_quoted_field_with_whitepace_delimiter(q):
543
+ txt = StringIO(
544
+ f"{q}alpha, x{q} 2.5\n{q}beta, y{q} 4.5\n{q}gamma, z{q} 5.0\n"
545
+ )
546
+ dtype = np.dtype([('f0', 'U8'), ('f1', np.float64)])
547
+ expected = np.array(
548
+ [("alpha, x", 2.5), ("beta, y", 4.5), ("gamma, z", 5.0)], dtype=dtype
549
+ )
550
+
551
+ res = np.loadtxt(txt, dtype=dtype, delimiter=None, quotechar=q)
552
+ assert_array_equal(res, expected)
553
+
554
+
555
+ def test_quote_support_default():
556
+ """Support for quoted fields is disabled by default."""
557
+ txt = StringIO('"lat,long", 45, 30\n')
558
+ dtype = np.dtype([('f0', 'U24'), ('f1', np.float64), ('f2', np.float64)])
559
+
560
+ with pytest.raises(ValueError,
561
+ match="the dtype passed requires 3 columns but 4 were"):
562
+ np.loadtxt(txt, dtype=dtype, delimiter=",")
563
+
564
+ # Enable quoting support with non-None value for quotechar param
565
+ txt.seek(0)
566
+ expected = np.array([("lat,long", 45., 30.)], dtype=dtype)
567
+
568
+ res = np.loadtxt(txt, dtype=dtype, delimiter=",", quotechar='"')
569
+ assert_array_equal(res, expected)
570
+
571
+
572
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
573
+ reason="PyPy bug in error formatting")
574
+ def test_quotechar_multichar_error():
575
+ txt = StringIO("1,2\n3,4")
576
+ msg = r".*must be a single unicode character or None"
577
+ with pytest.raises(TypeError, match=msg):
578
+ np.loadtxt(txt, delimiter=",", quotechar="''")
579
+
580
+
581
+ def test_comment_multichar_error_with_quote():
582
+ txt = StringIO("1,2\n3,4")
583
+ msg = (
584
+ "when multiple comments or a multi-character comment is given, "
585
+ "quotes are not supported."
586
+ )
587
+ with pytest.raises(ValueError, match=msg):
588
+ np.loadtxt(txt, delimiter=",", comments="123", quotechar='"')
589
+ with pytest.raises(ValueError, match=msg):
590
+ np.loadtxt(txt, delimiter=",", comments=["#", "%"], quotechar='"')
591
+
592
+ # A single character string in a tuple is unpacked though:
593
+ res = np.loadtxt(txt, delimiter=",", comments=("#",), quotechar="'")
594
+ assert_equal(res, [[1, 2], [3, 4]])
595
+
596
+
597
+ def test_structured_dtype_with_quotes():
598
+ data = StringIO(
599
+
600
+ "1000;2.4;'alpha';-34\n"
601
+ "2000;3.1;'beta';29\n"
602
+ "3500;9.9;'gamma';120\n"
603
+ "4090;8.1;'delta';0\n"
604
+ "5001;4.4;'epsilon';-99\n"
605
+ "6543;7.8;'omega';-1\n"
606
+
607
+ )
608
+ dtype = np.dtype(
609
+ [('f0', np.uint16), ('f1', np.float64), ('f2', 'S7'), ('f3', np.int8)]
610
+ )
611
+ expected = np.array(
612
+ [
613
+ (1000, 2.4, "alpha", -34),
614
+ (2000, 3.1, "beta", 29),
615
+ (3500, 9.9, "gamma", 120),
616
+ (4090, 8.1, "delta", 0),
617
+ (5001, 4.4, "epsilon", -99),
618
+ (6543, 7.8, "omega", -1)
619
+ ],
620
+ dtype=dtype
621
+ )
622
+ res = np.loadtxt(data, dtype=dtype, delimiter=";", quotechar="'")
623
+ assert_array_equal(res, expected)
624
+
625
+
626
+ def test_quoted_field_is_not_empty():
627
+ txt = StringIO('1\n\n"4"\n""')
628
+ expected = np.array(["1", "4", ""], dtype="U1")
629
+ res = np.loadtxt(txt, delimiter=",", dtype="U1", quotechar='"')
630
+ assert_equal(res, expected)
631
+
632
+ def test_quoted_field_is_not_empty_nonstrict():
633
+ # Same as test_quoted_field_is_not_empty but check that we are not strict
634
+ # about missing closing quote (this is the `csv.reader` default also)
635
+ txt = StringIO('1\n\n"4"\n"')
636
+ expected = np.array(["1", "4", ""], dtype="U1")
637
+ res = np.loadtxt(txt, delimiter=",", dtype="U1", quotechar='"')
638
+ assert_equal(res, expected)
639
+
640
+ def test_consecutive_quotechar_escaped():
641
+ txt = StringIO('"Hello, my name is ""Monty""!"')
642
+ expected = np.array('Hello, my name is "Monty"!', dtype="U40")
643
+ res = np.loadtxt(txt, dtype="U40", delimiter=",", quotechar='"')
644
+ assert_equal(res, expected)
645
+
646
+
647
+ @pytest.mark.parametrize("data", ("", "\n\n\n", "# 1 2 3\n# 4 5 6\n"))
648
+ @pytest.mark.parametrize("ndmin", (0, 1, 2))
649
+ @pytest.mark.parametrize("usecols", [None, (1, 2, 3)])
650
+ def test_warn_on_no_data(data, ndmin, usecols):
651
+ """Check that a UserWarning is emitted when no data is read from input."""
652
+ if usecols is not None:
653
+ expected_shape = (0, 3)
654
+ elif ndmin == 2:
655
+ expected_shape = (0, 1) # guess a single column?!
656
+ else:
657
+ expected_shape = (0,)
658
+
659
+ txt = StringIO(data)
660
+ with pytest.warns(UserWarning, match="input contained no data"):
661
+ res = np.loadtxt(txt, ndmin=ndmin, usecols=usecols)
662
+ assert res.shape == expected_shape
663
+
664
+ with NamedTemporaryFile(mode="w") as fh:
665
+ fh.write(data)
666
+ fh.seek(0)
667
+ with pytest.warns(UserWarning, match="input contained no data"):
668
+ res = np.loadtxt(txt, ndmin=ndmin, usecols=usecols)
669
+ assert res.shape == expected_shape
670
+
671
+ @pytest.mark.parametrize("skiprows", (2, 3))
672
+ def test_warn_on_skipped_data(skiprows):
673
+ data = "1 2 3\n4 5 6"
674
+ txt = StringIO(data)
675
+ with pytest.warns(UserWarning, match="input contained no data"):
676
+ np.loadtxt(txt, skiprows=skiprows)
677
+
678
+
679
+ @pytest.mark.parametrize(["dtype", "value"], [
680
+ ("i2", 0x0001), ("u2", 0x0001),
681
+ ("i4", 0x00010203), ("u4", 0x00010203),
682
+ ("i8", 0x0001020304050607), ("u8", 0x0001020304050607),
683
+ # The following values are constructed to lead to unique bytes:
684
+ ("float16", 3.07e-05),
685
+ ("float32", 9.2557e-41), ("complex64", 9.2557e-41 + 2.8622554e-29j),
686
+ ("float64", -1.758571353180402e-24),
687
+ # Here and below, the repr side-steps a small loss of precision in
688
+ # complex `str` in PyPy (which is probably fine, as repr works):
689
+ ("complex128", repr(5.406409232372729e-29 - 1.758571353180402e-24j)),
690
+ # Use integer values that fit into double. Everything else leads to
691
+ # problems due to longdoubles going via double and decimal strings
692
+ # causing rounding errors.
693
+ ("longdouble", 0x01020304050607),
694
+ ("clongdouble", repr(0x01020304050607 + (0x00121314151617 * 1j))),
695
+ ("U2", "\U00010203\U000a0b0c")])
696
+ @pytest.mark.parametrize("swap", [True, False])
697
+ def test_byteswapping_and_unaligned(dtype, value, swap):
698
+ # Try to create "interesting" values within the valid unicode range:
699
+ dtype = np.dtype(dtype)
700
+ data = [f"x,{value}\n"] # repr as PyPy `str` truncates some
701
+ if swap:
702
+ dtype = dtype.newbyteorder()
703
+ full_dt = np.dtype([("a", "S1"), ("b", dtype)], align=False)
704
+ # The above ensures that the interesting "b" field is unaligned:
705
+ assert full_dt.fields["b"][1] == 1
706
+ res = np.loadtxt(data, dtype=full_dt, delimiter=",",
707
+ max_rows=1) # max-rows prevents over-allocation
708
+ assert res["b"] == dtype.type(value)
709
+
710
+
711
+ @pytest.mark.parametrize("dtype",
712
+ np.typecodes["AllInteger"] + "efdFD" + "?")
713
+ def test_unicode_whitespace_stripping(dtype):
714
+ # Test that all numeric types (and bool) strip whitespace correctly
715
+ # \u202F is a narrow no-break space, `\n` is just a whitespace if quoted.
716
+ # Currently, skip float128 as it did not always support this and has no
717
+ # "custom" parsing:
718
+ txt = StringIO(' 3 ,"\u202F2\n"')
719
+ res = np.loadtxt(txt, dtype=dtype, delimiter=",", quotechar='"')
720
+ assert_array_equal(res, np.array([3, 2]).astype(dtype))
721
+
722
+
723
+ @pytest.mark.parametrize("dtype", "FD")
724
+ def test_unicode_whitespace_stripping_complex(dtype):
725
+ # Complex has a few extra cases since it has two components and
726
+ # parentheses
727
+ line = " 1 , 2+3j , ( 4+5j ), ( 6+-7j ) , 8j , ( 9j ) \n"
728
+ data = [line, line.replace(" ", "\u202F")]
729
+ res = np.loadtxt(data, dtype=dtype, delimiter=',')
730
+ assert_array_equal(res, np.array([[1, 2 + 3j, 4 + 5j, 6 - 7j, 8j, 9j]] * 2))
731
+
732
+
733
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
734
+ reason="PyPy bug in error formatting")
735
+ @pytest.mark.parametrize("dtype", "FD")
736
+ @pytest.mark.parametrize("field",
737
+ ["1 +2j", "1+ 2j", "1+2 j", "1+-+3", "(1j", "(1", "(1+2j", "1+2j)"])
738
+ def test_bad_complex(dtype, field):
739
+ with pytest.raises(ValueError):
740
+ np.loadtxt([field + "\n"], dtype=dtype, delimiter=",")
741
+
742
+
743
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
744
+ reason="PyPy bug in error formatting")
745
+ @pytest.mark.parametrize("dtype",
746
+ np.typecodes["AllInteger"] + "efgdFDG" + "?")
747
+ def test_nul_character_error(dtype):
748
+ # Test that a \0 character is correctly recognized as an error even if
749
+ # what comes before is valid (not everything gets parsed internally).
750
+ if dtype.lower() == "g":
751
+ pytest.xfail("longdouble/clongdouble assignment may misbehave.")
752
+ with pytest.raises(ValueError):
753
+ np.loadtxt(["1\000"], dtype=dtype, delimiter=",", quotechar='"')
754
+
755
+
756
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
757
+ reason="PyPy bug in error formatting")
758
+ @pytest.mark.parametrize("dtype",
759
+ np.typecodes["AllInteger"] + "efgdFDG" + "?")
760
+ def test_no_thousands_support(dtype):
761
+ # Mainly to document behaviour, Python supports thousands like 1_1.
762
+ # (e and G may end up using different conversion and support it, this is
763
+ # a bug but happens...)
764
+ if dtype == "e":
765
+ pytest.skip("half assignment currently uses Python float converter")
766
+ if dtype in "eG":
767
+ pytest.xfail("clongdouble assignment is buggy (uses `complex`?).")
768
+
769
+ assert int("1_1") == float("1_1") == complex("1_1") == 11
770
+ with pytest.raises(ValueError):
771
+ np.loadtxt(["1_1\n"], dtype=dtype)
772
+
773
+
774
+ @pytest.mark.parametrize("data", [
775
+ ["1,2\n", "2\n,3\n"],
776
+ ["1,2\n", "2\r,3\n"]])
777
+ def test_bad_newline_in_iterator(data):
778
+ # In NumPy <=1.22 this was accepted, because newlines were completely
779
+ # ignored when the input was an iterable. This could be changed, but right
780
+ # now, we raise an error.
781
+ msg = "Found an unquoted embedded newline within a single line"
782
+ with pytest.raises(ValueError, match=msg):
783
+ np.loadtxt(data, delimiter=",")
784
+
785
+
786
+ @pytest.mark.parametrize("data", [
787
+ ["1,2\n", "2,3\r\n"], # a universal newline
788
+ ["1,2\n", "'2\n',3\n"], # a quoted newline
789
+ ["1,2\n", "'2\r',3\n"],
790
+ ["1,2\n", "'2\r\n',3\n"],
791
+ ])
792
+ def test_good_newline_in_iterator(data):
793
+ # The quoted newlines will be untransformed here, but are just whitespace.
794
+ res = np.loadtxt(data, delimiter=",", quotechar="'")
795
+ assert_array_equal(res, [[1., 2.], [2., 3.]])
796
+
797
+
798
+ @pytest.mark.parametrize("newline", ["\n", "\r", "\r\n"])
799
+ def test_universal_newlines_quoted(newline):
800
+ # Check that universal newline support within the tokenizer is not applied
801
+ # to quoted fields. (note that lines must end in newline or quoted
802
+ # fields will not include a newline at all)
803
+ data = ['1,"2\n"\n', '3,"4\n', '1"\n']
804
+ data = [row.replace("\n", newline) for row in data]
805
+ res = np.loadtxt(data, dtype=object, delimiter=",", quotechar='"')
806
+ assert_array_equal(res, [['1', f'2{newline}'], ['3', f'4{newline}1']])
807
+
808
+
809
+ def test_null_character():
810
+ # Basic tests to check that the NUL character is not special:
811
+ res = np.loadtxt(["1\0002\0003\n", "4\0005\0006"], delimiter="\000")
812
+ assert_array_equal(res, [[1, 2, 3], [4, 5, 6]])
813
+
814
+ # Also not as part of a field (avoid unicode/arrays as unicode strips \0)
815
+ res = np.loadtxt(["1\000,2\000,3\n", "4\000,5\000,6"],
816
+ delimiter=",", dtype=object)
817
+ assert res.tolist() == [["1\000", "2\000", "3"], ["4\000", "5\000", "6"]]
818
+
819
+
820
+ def test_iterator_fails_getting_next_line():
821
+ class BadSequence:
822
+ def __len__(self):
823
+ return 100
824
+
825
+ def __getitem__(self, item):
826
+ if item == 50:
827
+ raise RuntimeError("Bad things happened!")
828
+ return f"{item}, {item + 1}"
829
+
830
+ with pytest.raises(RuntimeError, match="Bad things happened!"):
831
+ np.loadtxt(BadSequence(), dtype=int, delimiter=",")
832
+
833
+
834
+ class TestCReaderUnitTests:
835
+ # These are internal tests for path that should not be possible to hit
836
+ # unless things go very very wrong somewhere.
837
+ def test_not_an_filelike(self):
838
+ with pytest.raises(AttributeError, match=".*read"):
839
+ np._core._multiarray_umath._load_from_filelike(
840
+ object(), dtype=np.dtype("i"), filelike=True)
841
+
842
+ def test_filelike_read_fails(self):
843
+ # Can only be reached if loadtxt opens the file, so it is hard to do
844
+ # via the public interface (although maybe not impossible considering
845
+ # the current "DataClass" backing).
846
+ class BadFileLike:
847
+ counter = 0
848
+
849
+ def read(self, size):
850
+ self.counter += 1
851
+ if self.counter > 20:
852
+ raise RuntimeError("Bad bad bad!")
853
+ return "1,2,3\n"
854
+
855
+ with pytest.raises(RuntimeError, match="Bad bad bad!"):
856
+ np._core._multiarray_umath._load_from_filelike(
857
+ BadFileLike(), dtype=np.dtype("i"), filelike=True)
858
+
859
+ def test_filelike_bad_read(self):
860
+ # Can only be reached if loadtxt opens the file, so it is hard to do
861
+ # via the public interface (although maybe not impossible considering
862
+ # the current "DataClass" backing).
863
+
864
+ class BadFileLike:
865
+ counter = 0
866
+
867
+ def read(self, size):
868
+ return 1234 # not a string!
869
+
870
+ with pytest.raises(TypeError,
871
+ match="non-string returned while reading data"):
872
+ np._core._multiarray_umath._load_from_filelike(
873
+ BadFileLike(), dtype=np.dtype("i"), filelike=True)
874
+
875
+ def test_not_an_iter(self):
876
+ with pytest.raises(TypeError,
877
+ match="error reading from object, expected an iterable"):
878
+ np._core._multiarray_umath._load_from_filelike(
879
+ object(), dtype=np.dtype("i"), filelike=False)
880
+
881
+ def test_bad_type(self):
882
+ with pytest.raises(TypeError, match="internal error: dtype must"):
883
+ np._core._multiarray_umath._load_from_filelike(
884
+ object(), dtype="i", filelike=False)
885
+
886
+ def test_bad_encoding(self):
887
+ with pytest.raises(TypeError, match="encoding must be a unicode"):
888
+ np._core._multiarray_umath._load_from_filelike(
889
+ object(), dtype=np.dtype("i"), filelike=False, encoding=123)
890
+
891
+ @pytest.mark.parametrize("newline", ["\r", "\n", "\r\n"])
892
+ def test_manual_universal_newlines(self, newline):
893
+ # This is currently not available to users, because we should always
894
+ # open files with universal newlines enabled `newlines=None`.
895
+ # (And reading from an iterator uses slightly different code paths.)
896
+ # We have no real support for `newline="\r"` or `newline="\n" as the
897
+ # user cannot specify those options.
898
+ data = StringIO('0\n1\n"2\n"\n3\n4 #\n'.replace("\n", newline),
899
+ newline="")
900
+
901
+ res = np._core._multiarray_umath._load_from_filelike(
902
+ data, dtype=np.dtype("U10"), filelike=True,
903
+ quote='"', comment="#", skiplines=1)
904
+ assert_array_equal(res[:, 0], ["1", f"2{newline}", "3", "4 "])
905
+
906
+
907
+ def test_delimiter_comment_collision_raises():
908
+ with pytest.raises(TypeError, match=".*control characters.*incompatible"):
909
+ np.loadtxt(StringIO("1, 2, 3"), delimiter=",", comments=",")
910
+
911
+
912
+ def test_delimiter_quotechar_collision_raises():
913
+ with pytest.raises(TypeError, match=".*control characters.*incompatible"):
914
+ np.loadtxt(StringIO("1, 2, 3"), delimiter=",", quotechar=",")
915
+
916
+
917
+ def test_comment_quotechar_collision_raises():
918
+ with pytest.raises(TypeError, match=".*control characters.*incompatible"):
919
+ np.loadtxt(StringIO("1 2 3"), comments="#", quotechar="#")
920
+
921
+
922
+ def test_delimiter_and_multiple_comments_collision_raises():
923
+ with pytest.raises(
924
+ TypeError, match="Comment characters.*cannot include the delimiter"
925
+ ):
926
+ np.loadtxt(StringIO("1, 2, 3"), delimiter=",", comments=["#", ","])
927
+
928
+
929
+ @pytest.mark.parametrize(
930
+ "ws",
931
+ (
932
+ " ", # space
933
+ "\t", # tab
934
+ "\u2003", # em
935
+ "\u00A0", # non-break
936
+ "\u3000", # ideographic space
937
+ )
938
+ )
939
+ def test_collision_with_default_delimiter_raises(ws):
940
+ with pytest.raises(TypeError, match=".*control characters.*incompatible"):
941
+ np.loadtxt(StringIO(f"1{ws}2{ws}3\n4{ws}5{ws}6\n"), comments=ws)
942
+ with pytest.raises(TypeError, match=".*control characters.*incompatible"):
943
+ np.loadtxt(StringIO(f"1{ws}2{ws}3\n4{ws}5{ws}6\n"), quotechar=ws)
944
+
945
+
946
+ @pytest.mark.parametrize("nl", ("\n", "\r"))
947
+ def test_control_character_newline_raises(nl):
948
+ txt = StringIO(f"1{nl}2{nl}3{nl}{nl}4{nl}5{nl}6{nl}{nl}")
949
+ msg = "control character.*cannot be a newline"
950
+ with pytest.raises(TypeError, match=msg):
951
+ np.loadtxt(txt, delimiter=nl)
952
+ with pytest.raises(TypeError, match=msg):
953
+ np.loadtxt(txt, comments=nl)
954
+ with pytest.raises(TypeError, match=msg):
955
+ np.loadtxt(txt, quotechar=nl)
956
+
957
+
958
+ @pytest.mark.parametrize(
959
+ ("generic_data", "long_datum", "unitless_dtype", "expected_dtype"),
960
+ [
961
+ ("2012-03", "2013-01-15", "M8", "M8[D]"), # Datetimes
962
+ ("spam-a-lot", "tis_but_a_scratch", "U", "U17"), # str
963
+ ],
964
+ )
965
+ @pytest.mark.parametrize("nrows", (10, 50000, 60000)) # lt, eq, gt chunksize
966
+ def test_parametric_unit_discovery(
967
+ generic_data, long_datum, unitless_dtype, expected_dtype, nrows
968
+ ):
969
+ """Check that the correct unit (e.g. month, day, second) is discovered from
970
+ the data when a user specifies a unitless datetime."""
971
+ # Unit should be "D" (days) due to last entry
972
+ data = [generic_data] * nrows + [long_datum]
973
+ expected = np.array(data, dtype=expected_dtype)
974
+ assert len(data) == nrows + 1
975
+ assert len(data) == len(expected)
976
+
977
+ # file-like path
978
+ txt = StringIO("\n".join(data))
979
+ a = np.loadtxt(txt, dtype=unitless_dtype)
980
+ assert len(a) == len(expected)
981
+ assert a.dtype == expected.dtype
982
+ assert_equal(a, expected)
983
+
984
+ # file-obj path
985
+ fd, fname = mkstemp()
986
+ os.close(fd)
987
+ with open(fname, "w") as fh:
988
+ fh.write("\n".join(data) + "\n")
989
+ # loading the full file...
990
+ a = np.loadtxt(fname, dtype=unitless_dtype)
991
+ assert len(a) == len(expected)
992
+ assert a.dtype == expected.dtype
993
+ assert_equal(a, expected)
994
+ # loading half of the file...
995
+ a = np.loadtxt(fname, dtype=unitless_dtype, max_rows=int(nrows / 2))
996
+ os.remove(fname)
997
+ assert len(a) == int(nrows / 2)
998
+ assert_equal(a, expected[:int(nrows / 2)])
999
+
1000
+
1001
+ def test_str_dtype_unit_discovery_with_converter():
1002
+ data = ["spam-a-lot"] * 60000 + ["XXXtis_but_a_scratch"]
1003
+ expected = np.array(
1004
+ ["spam-a-lot"] * 60000 + ["tis_but_a_scratch"], dtype="U17"
1005
+ )
1006
+ conv = lambda s: s.removeprefix("XXX")
1007
+
1008
+ # file-like path
1009
+ txt = StringIO("\n".join(data))
1010
+ a = np.loadtxt(txt, dtype="U", converters=conv)
1011
+ assert a.dtype == expected.dtype
1012
+ assert_equal(a, expected)
1013
+
1014
+ # file-obj path
1015
+ fd, fname = mkstemp()
1016
+ os.close(fd)
1017
+ with open(fname, "w") as fh:
1018
+ fh.write("\n".join(data))
1019
+ a = np.loadtxt(fname, dtype="U", converters=conv)
1020
+ os.remove(fname)
1021
+ assert a.dtype == expected.dtype
1022
+ assert_equal(a, expected)
1023
+
1024
+
1025
+ @pytest.mark.skipif(IS_PYPY and sys.implementation.version <= (7, 3, 8),
1026
+ reason="PyPy bug in error formatting")
1027
+ def test_control_character_empty():
1028
+ with pytest.raises(TypeError, match="Text reading control character must"):
1029
+ np.loadtxt(StringIO("1 2 3"), delimiter="")
1030
+ with pytest.raises(TypeError, match="Text reading control character must"):
1031
+ np.loadtxt(StringIO("1 2 3"), quotechar="")
1032
+ with pytest.raises(ValueError, match="comments cannot be an empty string"):
1033
+ np.loadtxt(StringIO("1 2 3"), comments="")
1034
+ with pytest.raises(ValueError, match="comments cannot be an empty string"):
1035
+ np.loadtxt(StringIO("1 2 3"), comments=["#", ""])
1036
+
1037
+
1038
+ def test_control_characters_as_bytes():
1039
+ """Byte control characters (comments, delimiter) are supported."""
1040
+ a = np.loadtxt(StringIO("#header\n1,2,3"), comments=b"#", delimiter=b",")
1041
+ assert_equal(a, [1, 2, 3])
1042
+
1043
+
1044
+ @pytest.mark.filterwarnings('ignore::UserWarning')
1045
+ def test_field_growing_cases():
1046
+ # Test empty field appending/growing (each field still takes 1 character)
1047
+ # to see if the final field appending does not create issues.
1048
+ res = np.loadtxt([""], delimiter=",", dtype=bytes)
1049
+ assert len(res) == 0
1050
+
1051
+ for i in range(1, 1024):
1052
+ res = np.loadtxt(["," * i], delimiter=",", dtype=bytes, max_rows=10)
1053
+ assert len(res) == i + 1
1054
+
1055
+ @pytest.mark.parametrize("nmax", (10000, 50000, 55000, 60000))
1056
+ def test_maxrows_exceeding_chunksize(nmax):
1057
+ # tries to read all of the file,
1058
+ # or less, equal, greater than _loadtxt_chunksize
1059
+ file_length = 60000
1060
+
1061
+ # file-like path
1062
+ data = ["a 0.5 1"] * file_length
1063
+ txt = StringIO("\n".join(data))
1064
+ res = np.loadtxt(txt, dtype=str, delimiter=" ", max_rows=nmax)
1065
+ assert len(res) == nmax
1066
+
1067
+ # file-obj path
1068
+ fd, fname = mkstemp()
1069
+ os.close(fd)
1070
+ with open(fname, "w") as fh:
1071
+ fh.write("\n".join(data))
1072
+ res = np.loadtxt(fname, dtype=str, delimiter=" ", max_rows=nmax)
1073
+ os.remove(fname)
1074
+ assert len(res) == nmax
1075
+
1076
+ @pytest.mark.parametrize("nskip", (0, 10000, 12345, 50000, 67891, 100000))
1077
+ def test_skiprow_exceeding_maxrows_exceeding_chunksize(tmpdir, nskip):
1078
+ # tries to read a file in chunks by skipping a variable amount of lines,
1079
+ # less, equal, greater than max_rows
1080
+ file_length = 110000
1081
+ data = "\n".join(f"{i} a 0.5 1" for i in range(1, file_length + 1))
1082
+ expected_length = min(60000, file_length - nskip)
1083
+ expected = np.arange(nskip + 1, nskip + 1 + expected_length).astype(str)
1084
+
1085
+ # file-like path
1086
+ txt = StringIO(data)
1087
+ res = np.loadtxt(txt, dtype='str', delimiter=" ", skiprows=nskip, max_rows=60000)
1088
+ assert len(res) == expected_length
1089
+ # are the right lines read in res?
1090
+ assert_array_equal(expected, res[:, 0])
1091
+
1092
+ # file-obj path
1093
+ tmp_file = tmpdir / "test_data.txt"
1094
+ tmp_file.write(data)
1095
+ fname = str(tmp_file)
1096
+ res = np.loadtxt(fname, dtype='str', delimiter=" ", skiprows=nskip, max_rows=60000)
1097
+ assert len(res) == expected_length
1098
+ # are the right lines read in res?
1099
+ assert_array_equal(expected, res[:, 0])