scipy 1.15.2__cp313-cp313t-macosx_14_0_arm64.whl → 1.16.0__cp313-cp313t-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 (642) hide show
  1. scipy/.dylibs/libgcc_s.1.1.dylib +0 -0
  2. scipy/.dylibs/libgfortran.5.dylib +0 -0
  3. scipy/.dylibs/libquadmath.0.dylib +0 -0
  4. scipy/__config__.py +5 -5
  5. scipy/__init__.py +3 -6
  6. scipy/_cyutility.cpython-313t-darwin.so +0 -0
  7. scipy/_lib/_array_api.py +497 -161
  8. scipy/_lib/_array_api_compat_vendor.py +9 -0
  9. scipy/_lib/_bunch.py +4 -0
  10. scipy/_lib/_ccallback_c.cpython-313t-darwin.so +0 -0
  11. scipy/_lib/_docscrape.py +1 -1
  12. scipy/_lib/_elementwise_iterative_method.py +15 -26
  13. scipy/_lib/_sparse.py +41 -0
  14. scipy/_lib/_test_ccallback.cpython-313t-darwin.so +0 -0
  15. scipy/_lib/_test_deprecation_call.cpython-313t-darwin.so +0 -0
  16. scipy/_lib/_test_deprecation_def.cpython-313t-darwin.so +0 -0
  17. scipy/_lib/_testutils.py +6 -2
  18. scipy/_lib/_util.py +222 -125
  19. scipy/_lib/array_api_compat/__init__.py +4 -4
  20. scipy/_lib/array_api_compat/_internal.py +19 -6
  21. scipy/_lib/array_api_compat/common/__init__.py +1 -1
  22. scipy/_lib/array_api_compat/common/_aliases.py +365 -193
  23. scipy/_lib/array_api_compat/common/_fft.py +94 -64
  24. scipy/_lib/array_api_compat/common/_helpers.py +413 -180
  25. scipy/_lib/array_api_compat/common/_linalg.py +116 -40
  26. scipy/_lib/array_api_compat/common/_typing.py +179 -10
  27. scipy/_lib/array_api_compat/cupy/__init__.py +1 -4
  28. scipy/_lib/array_api_compat/cupy/_aliases.py +61 -41
  29. scipy/_lib/array_api_compat/cupy/_info.py +16 -6
  30. scipy/_lib/array_api_compat/cupy/_typing.py +24 -39
  31. scipy/_lib/array_api_compat/dask/array/__init__.py +6 -3
  32. scipy/_lib/array_api_compat/dask/array/_aliases.py +267 -108
  33. scipy/_lib/array_api_compat/dask/array/_info.py +105 -34
  34. scipy/_lib/array_api_compat/dask/array/fft.py +5 -8
  35. scipy/_lib/array_api_compat/dask/array/linalg.py +21 -22
  36. scipy/_lib/array_api_compat/numpy/__init__.py +13 -15
  37. scipy/_lib/array_api_compat/numpy/_aliases.py +98 -49
  38. scipy/_lib/array_api_compat/numpy/_info.py +36 -16
  39. scipy/_lib/array_api_compat/numpy/_typing.py +27 -43
  40. scipy/_lib/array_api_compat/numpy/fft.py +11 -5
  41. scipy/_lib/array_api_compat/numpy/linalg.py +75 -22
  42. scipy/_lib/array_api_compat/torch/__init__.py +3 -5
  43. scipy/_lib/array_api_compat/torch/_aliases.py +262 -159
  44. scipy/_lib/array_api_compat/torch/_info.py +27 -16
  45. scipy/_lib/array_api_compat/torch/_typing.py +3 -0
  46. scipy/_lib/array_api_compat/torch/fft.py +17 -18
  47. scipy/_lib/array_api_compat/torch/linalg.py +16 -16
  48. scipy/_lib/array_api_extra/__init__.py +26 -3
  49. scipy/_lib/array_api_extra/_delegation.py +171 -0
  50. scipy/_lib/array_api_extra/_lib/__init__.py +1 -0
  51. scipy/_lib/array_api_extra/_lib/_at.py +463 -0
  52. scipy/_lib/array_api_extra/_lib/_backends.py +46 -0
  53. scipy/_lib/array_api_extra/_lib/_funcs.py +937 -0
  54. scipy/_lib/array_api_extra/_lib/_lazy.py +357 -0
  55. scipy/_lib/array_api_extra/_lib/_testing.py +278 -0
  56. scipy/_lib/array_api_extra/_lib/_utils/__init__.py +1 -0
  57. scipy/_lib/array_api_extra/_lib/_utils/_compat.py +74 -0
  58. scipy/_lib/array_api_extra/_lib/_utils/_compat.pyi +45 -0
  59. scipy/_lib/array_api_extra/_lib/_utils/_helpers.py +559 -0
  60. scipy/_lib/array_api_extra/_lib/_utils/_typing.py +10 -0
  61. scipy/_lib/array_api_extra/_lib/_utils/_typing.pyi +105 -0
  62. scipy/_lib/array_api_extra/testing.py +359 -0
  63. scipy/_lib/decorator.py +2 -2
  64. scipy/_lib/doccer.py +1 -7
  65. scipy/_lib/messagestream.cpython-313t-darwin.so +0 -0
  66. scipy/_lib/pyprima/__init__.py +212 -0
  67. scipy/_lib/pyprima/cobyla/__init__.py +0 -0
  68. scipy/_lib/pyprima/cobyla/cobyla.py +559 -0
  69. scipy/_lib/pyprima/cobyla/cobylb.py +714 -0
  70. scipy/_lib/pyprima/cobyla/geometry.py +226 -0
  71. scipy/_lib/pyprima/cobyla/initialize.py +215 -0
  72. scipy/_lib/pyprima/cobyla/trustregion.py +492 -0
  73. scipy/_lib/pyprima/cobyla/update.py +289 -0
  74. scipy/_lib/pyprima/common/__init__.py +0 -0
  75. scipy/_lib/pyprima/common/_bounds.py +34 -0
  76. scipy/_lib/pyprima/common/_linear_constraints.py +46 -0
  77. scipy/_lib/pyprima/common/_nonlinear_constraints.py +54 -0
  78. scipy/_lib/pyprima/common/_project.py +173 -0
  79. scipy/_lib/pyprima/common/checkbreak.py +93 -0
  80. scipy/_lib/pyprima/common/consts.py +47 -0
  81. scipy/_lib/pyprima/common/evaluate.py +99 -0
  82. scipy/_lib/pyprima/common/history.py +38 -0
  83. scipy/_lib/pyprima/common/infos.py +30 -0
  84. scipy/_lib/pyprima/common/linalg.py +435 -0
  85. scipy/_lib/pyprima/common/message.py +290 -0
  86. scipy/_lib/pyprima/common/powalg.py +131 -0
  87. scipy/_lib/pyprima/common/preproc.py +277 -0
  88. scipy/_lib/pyprima/common/present.py +5 -0
  89. scipy/_lib/pyprima/common/ratio.py +54 -0
  90. scipy/_lib/pyprima/common/redrho.py +47 -0
  91. scipy/_lib/pyprima/common/selectx.py +296 -0
  92. scipy/_lib/tests/test__util.py +105 -121
  93. scipy/_lib/tests/test_array_api.py +169 -34
  94. scipy/_lib/tests/test_bunch.py +7 -0
  95. scipy/_lib/tests/test_ccallback.py +2 -10
  96. scipy/_lib/tests/test_public_api.py +13 -0
  97. scipy/cluster/_hierarchy.cpython-313t-darwin.so +0 -0
  98. scipy/cluster/_optimal_leaf_ordering.cpython-313t-darwin.so +0 -0
  99. scipy/cluster/_vq.cpython-313t-darwin.so +0 -0
  100. scipy/cluster/hierarchy.py +393 -223
  101. scipy/cluster/tests/test_hierarchy.py +273 -335
  102. scipy/cluster/tests/test_vq.py +45 -61
  103. scipy/cluster/vq.py +39 -35
  104. scipy/conftest.py +282 -151
  105. scipy/constants/_constants.py +4 -1
  106. scipy/constants/tests/test_codata.py +2 -2
  107. scipy/constants/tests/test_constants.py +11 -18
  108. scipy/datasets/_download_all.py +15 -1
  109. scipy/datasets/_fetchers.py +7 -1
  110. scipy/datasets/_utils.py +1 -1
  111. scipy/differentiate/_differentiate.py +25 -25
  112. scipy/differentiate/tests/test_differentiate.py +24 -25
  113. scipy/fft/_basic.py +20 -0
  114. scipy/fft/_helper.py +3 -34
  115. scipy/fft/_pocketfft/helper.py +29 -1
  116. scipy/fft/_pocketfft/tests/test_basic.py +2 -4
  117. scipy/fft/_pocketfft/tests/test_real_transforms.py +4 -4
  118. scipy/fft/_realtransforms.py +13 -0
  119. scipy/fft/tests/test_basic.py +27 -25
  120. scipy/fft/tests/test_fftlog.py +16 -7
  121. scipy/fft/tests/test_helper.py +18 -34
  122. scipy/fft/tests/test_real_transforms.py +8 -10
  123. scipy/fftpack/convolve.cpython-313t-darwin.so +0 -0
  124. scipy/fftpack/tests/test_basic.py +2 -4
  125. scipy/fftpack/tests/test_real_transforms.py +8 -9
  126. scipy/integrate/_bvp.py +9 -3
  127. scipy/integrate/_cubature.py +3 -2
  128. scipy/integrate/_dop.cpython-313t-darwin.so +0 -0
  129. scipy/integrate/_ivp/common.py +3 -3
  130. scipy/integrate/_ivp/ivp.py +9 -2
  131. scipy/integrate/_ivp/tests/test_ivp.py +19 -0
  132. scipy/integrate/_lsoda.cpython-313t-darwin.so +0 -0
  133. scipy/integrate/_ode.py +9 -2
  134. scipy/integrate/_odepack.cpython-313t-darwin.so +0 -0
  135. scipy/integrate/_quad_vec.py +21 -29
  136. scipy/integrate/_quadpack.cpython-313t-darwin.so +0 -0
  137. scipy/integrate/_quadpack_py.py +11 -7
  138. scipy/integrate/_quadrature.py +3 -3
  139. scipy/integrate/_rules/_base.py +2 -2
  140. scipy/integrate/_tanhsinh.py +57 -54
  141. scipy/integrate/_test_odeint_banded.cpython-313t-darwin.so +0 -0
  142. scipy/integrate/_vode.cpython-313t-darwin.so +0 -0
  143. scipy/integrate/tests/test__quad_vec.py +0 -6
  144. scipy/integrate/tests/test_banded_ode_solvers.py +85 -0
  145. scipy/integrate/tests/test_cubature.py +21 -35
  146. scipy/integrate/tests/test_quadrature.py +6 -8
  147. scipy/integrate/tests/test_tanhsinh.py +61 -43
  148. scipy/interpolate/__init__.py +70 -58
  149. scipy/interpolate/_bary_rational.py +22 -22
  150. scipy/interpolate/_bsplines.py +119 -66
  151. scipy/interpolate/_cubic.py +65 -50
  152. scipy/interpolate/_dfitpack.cpython-313t-darwin.so +0 -0
  153. scipy/interpolate/_dierckx.cpython-313t-darwin.so +0 -0
  154. scipy/interpolate/_fitpack.cpython-313t-darwin.so +0 -0
  155. scipy/interpolate/_fitpack2.py +9 -6
  156. scipy/interpolate/_fitpack_impl.py +32 -26
  157. scipy/interpolate/_fitpack_repro.py +23 -19
  158. scipy/interpolate/_interpnd.cpython-313t-darwin.so +0 -0
  159. scipy/interpolate/_interpolate.py +30 -12
  160. scipy/interpolate/_ndbspline.py +13 -18
  161. scipy/interpolate/_ndgriddata.py +5 -8
  162. scipy/interpolate/_polyint.py +95 -31
  163. scipy/interpolate/_ppoly.cpython-313t-darwin.so +0 -0
  164. scipy/interpolate/_rbf.py +2 -2
  165. scipy/interpolate/_rbfinterp.py +1 -1
  166. scipy/interpolate/_rbfinterp_pythran.cpython-313t-darwin.so +0 -0
  167. scipy/interpolate/_rgi.py +31 -26
  168. scipy/interpolate/_rgi_cython.cpython-313t-darwin.so +0 -0
  169. scipy/interpolate/dfitpack.py +0 -20
  170. scipy/interpolate/interpnd.py +1 -2
  171. scipy/interpolate/tests/test_bary_rational.py +2 -2
  172. scipy/interpolate/tests/test_bsplines.py +97 -1
  173. scipy/interpolate/tests/test_fitpack2.py +39 -1
  174. scipy/interpolate/tests/test_interpnd.py +32 -20
  175. scipy/interpolate/tests/test_interpolate.py +48 -4
  176. scipy/interpolate/tests/test_rgi.py +2 -1
  177. scipy/io/_fast_matrix_market/__init__.py +2 -0
  178. scipy/io/_harwell_boeing/_fortran_format_parser.py +19 -16
  179. scipy/io/_harwell_boeing/hb.py +7 -11
  180. scipy/io/_idl.py +5 -7
  181. scipy/io/_netcdf.py +15 -5
  182. scipy/io/_test_fortran.cpython-313t-darwin.so +0 -0
  183. scipy/io/arff/tests/test_arffread.py +3 -3
  184. scipy/io/matlab/__init__.py +5 -3
  185. scipy/io/matlab/_mio.py +4 -1
  186. scipy/io/matlab/_mio5.py +19 -13
  187. scipy/io/matlab/_mio5_utils.cpython-313t-darwin.so +0 -0
  188. scipy/io/matlab/_mio_utils.cpython-313t-darwin.so +0 -0
  189. scipy/io/matlab/_miobase.py +4 -1
  190. scipy/io/matlab/_streams.cpython-313t-darwin.so +0 -0
  191. scipy/io/matlab/tests/test_mio.py +46 -18
  192. scipy/io/matlab/tests/test_mio_funcs.py +1 -1
  193. scipy/io/tests/test_mmio.py +7 -1
  194. scipy/io/tests/test_wavfile.py +41 -0
  195. scipy/io/wavfile.py +57 -10
  196. scipy/linalg/_basic.py +113 -86
  197. scipy/linalg/_cythonized_array_utils.cpython-313t-darwin.so +0 -0
  198. scipy/linalg/_decomp.py +22 -9
  199. scipy/linalg/_decomp_cholesky.py +28 -13
  200. scipy/linalg/_decomp_cossin.py +45 -30
  201. scipy/linalg/_decomp_interpolative.cpython-313t-darwin.so +0 -0
  202. scipy/linalg/_decomp_ldl.py +4 -1
  203. scipy/linalg/_decomp_lu.py +18 -6
  204. scipy/linalg/_decomp_lu_cython.cpython-313t-darwin.so +0 -0
  205. scipy/linalg/_decomp_polar.py +2 -0
  206. scipy/linalg/_decomp_qr.py +6 -2
  207. scipy/linalg/_decomp_qz.py +3 -0
  208. scipy/linalg/_decomp_schur.py +3 -1
  209. scipy/linalg/_decomp_svd.py +13 -2
  210. scipy/linalg/_decomp_update.cpython-313t-darwin.so +0 -0
  211. scipy/linalg/_expm_frechet.py +4 -0
  212. scipy/linalg/_fblas.cpython-313t-darwin.so +0 -0
  213. scipy/linalg/_flapack.cpython-313t-darwin.so +0 -0
  214. scipy/linalg/_linalg_pythran.cpython-313t-darwin.so +0 -0
  215. scipy/linalg/_matfuncs.py +187 -4
  216. scipy/linalg/_matfuncs_expm.cpython-313t-darwin.so +0 -0
  217. scipy/linalg/_matfuncs_schur_sqrtm.cpython-313t-darwin.so +0 -0
  218. scipy/linalg/_matfuncs_sqrtm.py +1 -99
  219. scipy/linalg/_matfuncs_sqrtm_triu.cpython-313t-darwin.so +0 -0
  220. scipy/linalg/_procrustes.py +2 -0
  221. scipy/linalg/_sketches.py +17 -6
  222. scipy/linalg/_solve_toeplitz.cpython-313t-darwin.so +0 -0
  223. scipy/linalg/_solvers.py +7 -2
  224. scipy/linalg/_special_matrices.py +26 -36
  225. scipy/linalg/blas.py +35 -24
  226. scipy/linalg/cython_blas.cpython-313t-darwin.so +0 -0
  227. scipy/linalg/cython_lapack.cpython-313t-darwin.so +0 -0
  228. scipy/linalg/lapack.py +22 -2
  229. scipy/linalg/tests/_cython_examples/meson.build +7 -0
  230. scipy/linalg/tests/test_basic.py +31 -16
  231. scipy/linalg/tests/test_batch.py +588 -0
  232. scipy/linalg/tests/test_cythonized_array_utils.py +0 -2
  233. scipy/linalg/tests/test_decomp.py +40 -3
  234. scipy/linalg/tests/test_decomp_cossin.py +14 -0
  235. scipy/linalg/tests/test_decomp_ldl.py +1 -1
  236. scipy/linalg/tests/test_interpolative.py +17 -0
  237. scipy/linalg/tests/test_lapack.py +115 -7
  238. scipy/linalg/tests/test_matfuncs.py +157 -102
  239. scipy/linalg/tests/test_procrustes.py +0 -7
  240. scipy/linalg/tests/test_solve_toeplitz.py +1 -1
  241. scipy/linalg/tests/test_special_matrices.py +1 -5
  242. scipy/ndimage/__init__.py +1 -0
  243. scipy/ndimage/_cytest.cpython-313t-darwin.so +0 -0
  244. scipy/ndimage/_delegators.py +8 -2
  245. scipy/ndimage/_filters.py +453 -5
  246. scipy/ndimage/_interpolation.py +36 -6
  247. scipy/ndimage/_measurements.py +4 -2
  248. scipy/ndimage/_morphology.py +5 -0
  249. scipy/ndimage/_nd_image.cpython-313t-darwin.so +0 -0
  250. scipy/ndimage/_ndimage_api.py +2 -1
  251. scipy/ndimage/_ni_docstrings.py +5 -1
  252. scipy/ndimage/_ni_label.cpython-313t-darwin.so +0 -0
  253. scipy/ndimage/_ni_support.py +1 -5
  254. scipy/ndimage/_rank_filter_1d.cpython-313t-darwin.so +0 -0
  255. scipy/ndimage/_support_alternative_backends.py +18 -6
  256. scipy/ndimage/tests/test_filters.py +384 -259
  257. scipy/ndimage/tests/test_fourier.py +7 -9
  258. scipy/ndimage/tests/test_interpolation.py +68 -61
  259. scipy/ndimage/tests/test_measurements.py +18 -35
  260. scipy/ndimage/tests/test_morphology.py +143 -131
  261. scipy/ndimage/tests/test_splines.py +1 -3
  262. scipy/odr/__odrpack.cpython-313t-darwin.so +0 -0
  263. scipy/optimize/_basinhopping.py +13 -7
  264. scipy/optimize/_bglu_dense.cpython-313t-darwin.so +0 -0
  265. scipy/optimize/_bracket.py +46 -26
  266. scipy/optimize/_chandrupatla.py +9 -10
  267. scipy/optimize/_cobyla_py.py +104 -123
  268. scipy/optimize/_constraints.py +14 -10
  269. scipy/optimize/_differentiable_functions.py +371 -230
  270. scipy/optimize/_differentialevolution.py +4 -3
  271. scipy/optimize/_direct.cpython-313t-darwin.so +0 -0
  272. scipy/optimize/_dual_annealing.py +1 -1
  273. scipy/optimize/_elementwise.py +1 -4
  274. scipy/optimize/_group_columns.cpython-313t-darwin.so +0 -0
  275. scipy/optimize/_highspy/_highs_wrapper.py +6 -4
  276. scipy/optimize/_lbfgsb.cpython-313t-darwin.so +0 -0
  277. scipy/optimize/_lbfgsb_py.py +80 -24
  278. scipy/optimize/_linprog_doc.py +2 -2
  279. scipy/optimize/_linprog_highs.py +11 -11
  280. scipy/optimize/_linprog_ip.py +25 -10
  281. scipy/optimize/_linprog_util.py +18 -19
  282. scipy/optimize/_lsap.cpython-313t-darwin.so +0 -0
  283. scipy/optimize/_lsq/common.py +3 -3
  284. scipy/optimize/_lsq/dogbox.py +16 -2
  285. scipy/optimize/_lsq/givens_elimination.cpython-313t-darwin.so +0 -0
  286. scipy/optimize/_lsq/least_squares.py +198 -126
  287. scipy/optimize/_lsq/lsq_linear.py +6 -6
  288. scipy/optimize/_lsq/trf.py +35 -8
  289. scipy/optimize/_milp.py +3 -1
  290. scipy/optimize/_minimize.py +105 -36
  291. scipy/optimize/_minpack.cpython-313t-darwin.so +0 -0
  292. scipy/optimize/_minpack_py.py +21 -14
  293. scipy/optimize/_moduleTNC.cpython-313t-darwin.so +0 -0
  294. scipy/optimize/_nnls.py +20 -21
  295. scipy/optimize/_nonlin.py +34 -3
  296. scipy/optimize/_numdiff.py +288 -110
  297. scipy/optimize/_optimize.py +86 -48
  298. scipy/optimize/_pava_pybind.cpython-313t-darwin.so +0 -0
  299. scipy/optimize/_remove_redundancy.py +5 -5
  300. scipy/optimize/_root_scalar.py +1 -1
  301. scipy/optimize/_shgo.py +6 -0
  302. scipy/optimize/_shgo_lib/_complex.py +1 -1
  303. scipy/optimize/_slsqp_py.py +216 -124
  304. scipy/optimize/_slsqplib.cpython-313t-darwin.so +0 -0
  305. scipy/optimize/_spectral.py +1 -1
  306. scipy/optimize/_tnc.py +8 -1
  307. scipy/optimize/_trlib/_trlib.cpython-313t-darwin.so +0 -0
  308. scipy/optimize/_trustregion.py +20 -6
  309. scipy/optimize/_trustregion_constr/canonical_constraint.py +7 -7
  310. scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +1 -1
  311. scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +11 -3
  312. scipy/optimize/_trustregion_constr/projections.py +12 -8
  313. scipy/optimize/_trustregion_constr/qp_subproblem.py +9 -9
  314. scipy/optimize/_trustregion_constr/tests/test_projections.py +7 -7
  315. scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +77 -77
  316. scipy/optimize/_trustregion_constr/tr_interior_point.py +5 -5
  317. scipy/optimize/_trustregion_exact.py +0 -1
  318. scipy/optimize/_zeros.cpython-313t-darwin.so +0 -0
  319. scipy/optimize/_zeros_py.py +97 -17
  320. scipy/optimize/cython_optimize/_zeros.cpython-313t-darwin.so +0 -0
  321. scipy/optimize/slsqp.py +0 -1
  322. scipy/optimize/tests/test__basinhopping.py +1 -1
  323. scipy/optimize/tests/test__differential_evolution.py +4 -4
  324. scipy/optimize/tests/test__linprog_clean_inputs.py +5 -3
  325. scipy/optimize/tests/test__numdiff.py +66 -22
  326. scipy/optimize/tests/test__remove_redundancy.py +2 -2
  327. scipy/optimize/tests/test__shgo.py +9 -1
  328. scipy/optimize/tests/test_bracket.py +71 -46
  329. scipy/optimize/tests/test_chandrupatla.py +133 -135
  330. scipy/optimize/tests/test_cobyla.py +74 -45
  331. scipy/optimize/tests/test_constraints.py +1 -1
  332. scipy/optimize/tests/test_differentiable_functions.py +226 -6
  333. scipy/optimize/tests/test_lbfgsb_hessinv.py +22 -0
  334. scipy/optimize/tests/test_least_squares.py +125 -13
  335. scipy/optimize/tests/test_linear_assignment.py +3 -3
  336. scipy/optimize/tests/test_linprog.py +3 -3
  337. scipy/optimize/tests/test_lsq_linear.py +6 -6
  338. scipy/optimize/tests/test_minimize_constrained.py +2 -2
  339. scipy/optimize/tests/test_minpack.py +4 -4
  340. scipy/optimize/tests/test_nnls.py +43 -3
  341. scipy/optimize/tests/test_nonlin.py +36 -0
  342. scipy/optimize/tests/test_optimize.py +98 -20
  343. scipy/optimize/tests/test_slsqp.py +36 -4
  344. scipy/optimize/tests/test_zeros.py +34 -1
  345. scipy/signal/__init__.py +12 -23
  346. scipy/signal/_delegators.py +568 -0
  347. scipy/signal/_filter_design.py +459 -241
  348. scipy/signal/_fir_filter_design.py +262 -90
  349. scipy/signal/_lti_conversion.py +3 -2
  350. scipy/signal/_ltisys.py +118 -91
  351. scipy/signal/_max_len_seq_inner.cpython-313t-darwin.so +0 -0
  352. scipy/signal/_peak_finding_utils.cpython-313t-darwin.so +0 -0
  353. scipy/signal/_polyutils.py +172 -0
  354. scipy/signal/_short_time_fft.py +553 -76
  355. scipy/signal/_signal_api.py +30 -0
  356. scipy/signal/_signaltools.py +719 -396
  357. scipy/signal/_sigtools.cpython-313t-darwin.so +0 -0
  358. scipy/signal/_sosfilt.cpython-313t-darwin.so +0 -0
  359. scipy/signal/_spectral_py.py +230 -50
  360. scipy/signal/_spline.cpython-313t-darwin.so +0 -0
  361. scipy/signal/_spline_filters.py +108 -68
  362. scipy/signal/_support_alternative_backends.py +73 -0
  363. scipy/signal/_upfirdn.py +4 -1
  364. scipy/signal/_upfirdn_apply.cpython-313t-darwin.so +0 -0
  365. scipy/signal/_waveforms.py +2 -11
  366. scipy/signal/_wavelets.py +1 -1
  367. scipy/signal/fir_filter_design.py +1 -0
  368. scipy/signal/spline.py +4 -11
  369. scipy/signal/tests/_scipy_spectral_test_shim.py +5 -182
  370. scipy/signal/tests/test_bsplines.py +114 -79
  371. scipy/signal/tests/test_cont2discrete.py +9 -2
  372. scipy/signal/tests/test_filter_design.py +721 -481
  373. scipy/signal/tests/test_fir_filter_design.py +332 -140
  374. scipy/signal/tests/test_savitzky_golay.py +4 -3
  375. scipy/signal/tests/test_short_time_fft.py +231 -5
  376. scipy/signal/tests/test_signaltools.py +2150 -1349
  377. scipy/signal/tests/test_spectral.py +50 -6
  378. scipy/signal/tests/test_splines.py +161 -96
  379. scipy/signal/tests/test_upfirdn.py +84 -50
  380. scipy/signal/tests/test_waveforms.py +20 -0
  381. scipy/signal/tests/test_windows.py +607 -466
  382. scipy/signal/windows/_windows.py +287 -148
  383. scipy/sparse/__init__.py +23 -4
  384. scipy/sparse/_base.py +269 -120
  385. scipy/sparse/_bsr.py +7 -4
  386. scipy/sparse/_compressed.py +59 -234
  387. scipy/sparse/_construct.py +90 -38
  388. scipy/sparse/_coo.py +115 -181
  389. scipy/sparse/_csc.py +4 -4
  390. scipy/sparse/_csparsetools.cpython-313t-darwin.so +0 -0
  391. scipy/sparse/_csr.py +2 -2
  392. scipy/sparse/_data.py +48 -48
  393. scipy/sparse/_dia.py +105 -21
  394. scipy/sparse/_dok.py +0 -23
  395. scipy/sparse/_index.py +4 -4
  396. scipy/sparse/_matrix.py +23 -0
  397. scipy/sparse/_sparsetools.cpython-313t-darwin.so +0 -0
  398. scipy/sparse/_sputils.py +37 -22
  399. scipy/sparse/base.py +0 -9
  400. scipy/sparse/bsr.py +0 -14
  401. scipy/sparse/compressed.py +0 -23
  402. scipy/sparse/construct.py +0 -6
  403. scipy/sparse/coo.py +0 -14
  404. scipy/sparse/csc.py +0 -3
  405. scipy/sparse/csgraph/_flow.cpython-313t-darwin.so +0 -0
  406. scipy/sparse/csgraph/_matching.cpython-313t-darwin.so +0 -0
  407. scipy/sparse/csgraph/_min_spanning_tree.cpython-313t-darwin.so +0 -0
  408. scipy/sparse/csgraph/_reordering.cpython-313t-darwin.so +0 -0
  409. scipy/sparse/csgraph/_shortest_path.cpython-313t-darwin.so +0 -0
  410. scipy/sparse/csgraph/_tools.cpython-313t-darwin.so +0 -0
  411. scipy/sparse/csgraph/_traversal.cpython-313t-darwin.so +0 -0
  412. scipy/sparse/csgraph/tests/test_matching.py +14 -2
  413. scipy/sparse/csgraph/tests/test_pydata_sparse.py +4 -1
  414. scipy/sparse/csgraph/tests/test_shortest_path.py +83 -27
  415. scipy/sparse/csr.py +0 -5
  416. scipy/sparse/data.py +1 -6
  417. scipy/sparse/dia.py +0 -7
  418. scipy/sparse/dok.py +0 -10
  419. scipy/sparse/linalg/_dsolve/_superlu.cpython-313t-darwin.so +0 -0
  420. scipy/sparse/linalg/_dsolve/linsolve.py +9 -0
  421. scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +35 -28
  422. scipy/sparse/linalg/_eigen/arpack/_arpack.cpython-313t-darwin.so +0 -0
  423. scipy/sparse/linalg/_eigen/arpack/arpack.py +28 -20
  424. scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +6 -6
  425. scipy/sparse/linalg/_expm_multiply.py +8 -3
  426. scipy/sparse/linalg/_interface.py +29 -26
  427. scipy/sparse/linalg/_isolve/_gcrotmk.py +6 -5
  428. scipy/sparse/linalg/_isolve/iterative.py +51 -45
  429. scipy/sparse/linalg/_isolve/lgmres.py +6 -6
  430. scipy/sparse/linalg/_isolve/minres.py +5 -5
  431. scipy/sparse/linalg/_isolve/tfqmr.py +7 -7
  432. scipy/sparse/linalg/_isolve/utils.py +2 -8
  433. scipy/sparse/linalg/_matfuncs.py +1 -1
  434. scipy/sparse/linalg/_norm.py +1 -1
  435. scipy/sparse/linalg/_propack/_cpropack.cpython-313t-darwin.so +0 -0
  436. scipy/sparse/linalg/_propack/_dpropack.cpython-313t-darwin.so +0 -0
  437. scipy/sparse/linalg/_propack/_spropack.cpython-313t-darwin.so +0 -0
  438. scipy/sparse/linalg/_propack/_zpropack.cpython-313t-darwin.so +0 -0
  439. scipy/sparse/linalg/_special_sparse_arrays.py +39 -38
  440. scipy/sparse/linalg/tests/test_expm_multiply.py +10 -0
  441. scipy/sparse/linalg/tests/test_interface.py +35 -0
  442. scipy/sparse/linalg/tests/test_pydata_sparse.py +18 -0
  443. scipy/sparse/tests/test_arithmetic1d.py +5 -2
  444. scipy/sparse/tests/test_base.py +224 -40
  445. scipy/sparse/tests/test_common1d.py +17 -12
  446. scipy/sparse/tests/test_construct.py +1 -1
  447. scipy/sparse/tests/test_coo.py +272 -4
  448. scipy/sparse/tests/test_sparsetools.py +5 -0
  449. scipy/sparse/tests/test_sputils.py +36 -7
  450. scipy/spatial/_ckdtree.cpython-313t-darwin.so +0 -0
  451. scipy/spatial/_distance_pybind.cpython-313t-darwin.so +0 -0
  452. scipy/spatial/_distance_wrap.cpython-313t-darwin.so +0 -0
  453. scipy/spatial/_hausdorff.cpython-313t-darwin.so +0 -0
  454. scipy/spatial/_qhull.cpython-313t-darwin.so +0 -0
  455. scipy/spatial/_voronoi.cpython-313t-darwin.so +0 -0
  456. scipy/spatial/distance.py +49 -42
  457. scipy/spatial/tests/test_distance.py +15 -1
  458. scipy/spatial/tests/test_kdtree.py +1 -0
  459. scipy/spatial/tests/test_qhull.py +106 -2
  460. scipy/spatial/transform/__init__.py +5 -3
  461. scipy/spatial/transform/_rigid_transform.cpython-313t-darwin.so +0 -0
  462. scipy/spatial/transform/_rotation.cpython-313t-darwin.so +0 -0
  463. scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
  464. scipy/spatial/transform/tests/test_rotation.py +1342 -790
  465. scipy/spatial/transform/tests/test_rotation_groups.py +3 -3
  466. scipy/spatial/transform/tests/test_rotation_spline.py +29 -8
  467. scipy/special/__init__.py +1 -47
  468. scipy/special/_add_newdocs.py +34 -772
  469. scipy/special/_basic.py +22 -25
  470. scipy/special/_comb.cpython-313t-darwin.so +0 -0
  471. scipy/special/_ellip_harm_2.cpython-313t-darwin.so +0 -0
  472. scipy/special/_gufuncs.cpython-313t-darwin.so +0 -0
  473. scipy/special/_logsumexp.py +83 -69
  474. scipy/special/_orthogonal.pyi +1 -1
  475. scipy/special/_specfun.cpython-313t-darwin.so +0 -0
  476. scipy/special/_special_ufuncs.cpython-313t-darwin.so +0 -0
  477. scipy/special/_spherical_bessel.py +4 -4
  478. scipy/special/_support_alternative_backends.py +212 -119
  479. scipy/special/_test_internal.cpython-313t-darwin.so +0 -0
  480. scipy/special/_testutils.py +4 -4
  481. scipy/special/_ufuncs.cpython-313t-darwin.so +0 -0
  482. scipy/special/_ufuncs.pyi +1 -0
  483. scipy/special/_ufuncs.pyx +215 -1400
  484. scipy/special/_ufuncs_cxx.cpython-313t-darwin.so +0 -0
  485. scipy/special/_ufuncs_cxx.pxd +2 -15
  486. scipy/special/_ufuncs_cxx.pyx +5 -44
  487. scipy/special/_ufuncs_cxx_defs.h +2 -16
  488. scipy/special/_ufuncs_defs.h +0 -8
  489. scipy/special/cython_special.cpython-313t-darwin.so +0 -0
  490. scipy/special/cython_special.pxd +1 -1
  491. scipy/special/tests/_cython_examples/meson.build +10 -1
  492. scipy/special/tests/test_basic.py +153 -20
  493. scipy/special/tests/test_boost_ufuncs.py +3 -0
  494. scipy/special/tests/test_cdflib.py +35 -11
  495. scipy/special/tests/test_gammainc.py +16 -0
  496. scipy/special/tests/test_hyp2f1.py +23 -2
  497. scipy/special/tests/test_log1mexp.py +85 -0
  498. scipy/special/tests/test_logsumexp.py +220 -64
  499. scipy/special/tests/test_mpmath.py +1 -0
  500. scipy/special/tests/test_nan_inputs.py +1 -1
  501. scipy/special/tests/test_orthogonal.py +17 -18
  502. scipy/special/tests/test_sf_error.py +3 -2
  503. scipy/special/tests/test_sph_harm.py +6 -7
  504. scipy/special/tests/test_support_alternative_backends.py +211 -76
  505. scipy/stats/__init__.py +4 -1
  506. scipy/stats/_ansari_swilk_statistics.cpython-313t-darwin.so +0 -0
  507. scipy/stats/_axis_nan_policy.py +5 -12
  508. scipy/stats/_biasedurn.cpython-313t-darwin.so +0 -0
  509. scipy/stats/_continued_fraction.py +387 -0
  510. scipy/stats/_continuous_distns.py +296 -319
  511. scipy/stats/_correlation.py +1 -1
  512. scipy/stats/_covariance.py +6 -3
  513. scipy/stats/_discrete_distns.py +39 -32
  514. scipy/stats/_distn_infrastructure.py +39 -12
  515. scipy/stats/_distribution_infrastructure.py +920 -238
  516. scipy/stats/_entropy.py +9 -10
  517. scipy/{_lib → stats}/_finite_differences.py +1 -1
  518. scipy/stats/_hypotests.py +83 -50
  519. scipy/stats/_kde.py +53 -49
  520. scipy/stats/_ksstats.py +1 -1
  521. scipy/stats/_levy_stable/__init__.py +7 -15
  522. scipy/stats/_levy_stable/levyst.cpython-313t-darwin.so +0 -0
  523. scipy/stats/_morestats.py +118 -73
  524. scipy/stats/_mstats_basic.py +13 -17
  525. scipy/stats/_mstats_extras.py +8 -8
  526. scipy/stats/_multivariate.py +89 -113
  527. scipy/stats/_new_distributions.py +97 -20
  528. scipy/stats/_page_trend_test.py +12 -5
  529. scipy/stats/_probability_distribution.py +265 -43
  530. scipy/stats/_qmc.py +14 -9
  531. scipy/stats/_qmc_cy.cpython-313t-darwin.so +0 -0
  532. scipy/stats/_qmvnt.py +16 -95
  533. scipy/stats/_qmvnt_cy.cpython-313t-darwin.so +0 -0
  534. scipy/stats/_quantile.py +335 -0
  535. scipy/stats/_rcont/rcont.cpython-313t-darwin.so +0 -0
  536. scipy/stats/_resampling.py +5 -30
  537. scipy/stats/_sampling.py +1 -1
  538. scipy/stats/_sobol.cpython-313t-darwin.so +0 -0
  539. scipy/stats/_stats.cpython-313t-darwin.so +0 -0
  540. scipy/stats/_stats_mstats_common.py +21 -2
  541. scipy/stats/_stats_py.py +551 -477
  542. scipy/stats/_stats_pythran.cpython-313t-darwin.so +0 -0
  543. scipy/stats/_unuran/unuran_wrapper.cpython-313t-darwin.so +0 -0
  544. scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
  545. scipy/stats/_variation.py +6 -8
  546. scipy/stats/_wilcoxon.py +13 -7
  547. scipy/stats/tests/common_tests.py +6 -4
  548. scipy/stats/tests/test_axis_nan_policy.py +62 -24
  549. scipy/stats/tests/test_continued_fraction.py +173 -0
  550. scipy/stats/tests/test_continuous.py +379 -60
  551. scipy/stats/tests/test_continuous_basic.py +18 -12
  552. scipy/stats/tests/test_discrete_basic.py +14 -8
  553. scipy/stats/tests/test_discrete_distns.py +16 -16
  554. scipy/stats/tests/test_distributions.py +117 -75
  555. scipy/stats/tests/test_entropy.py +40 -48
  556. scipy/stats/tests/test_fit.py +4 -3
  557. scipy/stats/tests/test_hypotests.py +153 -24
  558. scipy/stats/tests/test_kdeoth.py +109 -41
  559. scipy/stats/tests/test_marray.py +289 -0
  560. scipy/stats/tests/test_morestats.py +81 -49
  561. scipy/stats/tests/test_mstats_basic.py +3 -3
  562. scipy/stats/tests/test_multivariate.py +434 -83
  563. scipy/stats/tests/test_qmc.py +13 -10
  564. scipy/stats/tests/test_quantile.py +199 -0
  565. scipy/stats/tests/test_rank.py +119 -112
  566. scipy/stats/tests/test_resampling.py +47 -56
  567. scipy/stats/tests/test_sampling.py +9 -4
  568. scipy/stats/tests/test_stats.py +799 -939
  569. scipy/stats/tests/test_variation.py +8 -6
  570. scipy/version.py +2 -2
  571. {scipy-1.15.2.dist-info → scipy-1.16.0.dist-info}/LICENSE.txt +4 -4
  572. {scipy-1.15.2.dist-info → scipy-1.16.0.dist-info}/METADATA +12 -12
  573. {scipy-1.15.2.dist-info → scipy-1.16.0.dist-info}/RECORD +574 -581
  574. scipy-1.16.0.dist-info/WHEEL +6 -0
  575. scipy/_lib/array_api_extra/_funcs.py +0 -484
  576. scipy/_lib/array_api_extra/_typing.py +0 -8
  577. scipy/interpolate/_bspl.cpython-313t-darwin.so +0 -0
  578. scipy/optimize/_cobyla.cpython-313t-darwin.so +0 -0
  579. scipy/optimize/_cython_nnls.cpython-313t-darwin.so +0 -0
  580. scipy/optimize/_slsqp.cpython-313t-darwin.so +0 -0
  581. scipy/spatial/qhull_src/COPYING.txt +0 -38
  582. scipy/special/libsf_error_state.dylib +0 -0
  583. scipy/special/tests/test_log_softmax.py +0 -109
  584. scipy/special/tests/test_xsf_cuda.py +0 -114
  585. scipy/special/xsf/binom.h +0 -89
  586. scipy/special/xsf/cdflib.h +0 -100
  587. scipy/special/xsf/cephes/airy.h +0 -307
  588. scipy/special/xsf/cephes/besselpoly.h +0 -51
  589. scipy/special/xsf/cephes/beta.h +0 -257
  590. scipy/special/xsf/cephes/cbrt.h +0 -131
  591. scipy/special/xsf/cephes/chbevl.h +0 -85
  592. scipy/special/xsf/cephes/chdtr.h +0 -193
  593. scipy/special/xsf/cephes/const.h +0 -87
  594. scipy/special/xsf/cephes/ellie.h +0 -293
  595. scipy/special/xsf/cephes/ellik.h +0 -251
  596. scipy/special/xsf/cephes/ellpe.h +0 -107
  597. scipy/special/xsf/cephes/ellpk.h +0 -117
  598. scipy/special/xsf/cephes/expn.h +0 -260
  599. scipy/special/xsf/cephes/gamma.h +0 -398
  600. scipy/special/xsf/cephes/hyp2f1.h +0 -596
  601. scipy/special/xsf/cephes/hyperg.h +0 -361
  602. scipy/special/xsf/cephes/i0.h +0 -149
  603. scipy/special/xsf/cephes/i1.h +0 -158
  604. scipy/special/xsf/cephes/igam.h +0 -421
  605. scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
  606. scipy/special/xsf/cephes/igami.h +0 -313
  607. scipy/special/xsf/cephes/j0.h +0 -225
  608. scipy/special/xsf/cephes/j1.h +0 -198
  609. scipy/special/xsf/cephes/jv.h +0 -715
  610. scipy/special/xsf/cephes/k0.h +0 -164
  611. scipy/special/xsf/cephes/k1.h +0 -163
  612. scipy/special/xsf/cephes/kn.h +0 -243
  613. scipy/special/xsf/cephes/lanczos.h +0 -112
  614. scipy/special/xsf/cephes/ndtr.h +0 -275
  615. scipy/special/xsf/cephes/poch.h +0 -85
  616. scipy/special/xsf/cephes/polevl.h +0 -167
  617. scipy/special/xsf/cephes/psi.h +0 -194
  618. scipy/special/xsf/cephes/rgamma.h +0 -111
  619. scipy/special/xsf/cephes/scipy_iv.h +0 -811
  620. scipy/special/xsf/cephes/shichi.h +0 -248
  621. scipy/special/xsf/cephes/sici.h +0 -224
  622. scipy/special/xsf/cephes/sindg.h +0 -221
  623. scipy/special/xsf/cephes/tandg.h +0 -139
  624. scipy/special/xsf/cephes/trig.h +0 -58
  625. scipy/special/xsf/cephes/unity.h +0 -186
  626. scipy/special/xsf/cephes/zeta.h +0 -172
  627. scipy/special/xsf/config.h +0 -304
  628. scipy/special/xsf/digamma.h +0 -205
  629. scipy/special/xsf/error.h +0 -57
  630. scipy/special/xsf/evalpoly.h +0 -47
  631. scipy/special/xsf/expint.h +0 -266
  632. scipy/special/xsf/hyp2f1.h +0 -694
  633. scipy/special/xsf/iv_ratio.h +0 -173
  634. scipy/special/xsf/lambertw.h +0 -150
  635. scipy/special/xsf/loggamma.h +0 -163
  636. scipy/special/xsf/sici.h +0 -200
  637. scipy/special/xsf/tools.h +0 -427
  638. scipy/special/xsf/trig.h +0 -164
  639. scipy/special/xsf/wright_bessel.h +0 -843
  640. scipy/special/xsf/zlog1.h +0 -35
  641. scipy/stats/_mvn.cpython-313t-darwin.so +0 -0
  642. scipy-1.15.2.dist-info/WHEEL +0 -4
@@ -1,114 +0,0 @@
1
- import os
2
- import pytest
3
- import scipy.special as sc
4
- import shutil
5
- import tempfile
6
-
7
- from uuid import uuid4
8
-
9
- from scipy.special._testutils import check_version
10
- from scipy.special._testutils import MissingModule
11
-
12
- try:
13
- import cupy # type: ignore
14
- except (ImportError, AttributeError):
15
- cupy = MissingModule('cupy')
16
-
17
-
18
- def get_test_cases():
19
- cases_source = [
20
- (sc.beta, "cephes/beta.h", "out0 = xsf::cephes::beta(in0, in1)"),
21
- (sc.binom, "binom.h", "out0 = xsf::binom(in0, in1)"),
22
- (sc.digamma, "digamma.h", "xsf::digamma(in0)"),
23
- (sc.expn, "cephes/expn.h", "out0 = xsf::cephes::expn(in0, in1)"),
24
- (sc.hyp2f1, "hyp2f1.h", "out0 = xsf::hyp2f1(in0, in1, in2, in3)"),
25
- (sc._ufuncs._lambertw, "lambertw.h", "out0 = xsf::lambertw(in0, in1, in2)"),
26
- (sc.ellipkinc, "cephes/ellik.h", "out0 = xsf::cephes::ellik(in0, in1)"),
27
- (sc.ellipeinc, "cephes/ellie.h", "out0 = xsf::cephes::ellie(in0, in1)"),
28
- (sc.gdtrib, "cdflib.h", "out0 = xsf::gdtrib(in0, in1, in2)"),
29
- (sc.sici, "sici.h", "xsf::sici(in0, &out0, &out1)"),
30
- (sc.shichi, "sici.h", "xsf::shichi(in0, &out0, &out1)"),
31
- ]
32
-
33
- cases = []
34
- for ufunc, header, routine in cases_source:
35
- preamble = f"#include <xsf/{header}>"
36
- for signature in ufunc.types:
37
- cases.append((signature, preamble, routine))
38
- return cases
39
-
40
-
41
- dtype_map = {
42
- "f": "float32",
43
- "d": "float64",
44
- "F": "complex64",
45
- "D": "complex128",
46
- "i": "int32",
47
- "l": "int64",
48
- }
49
-
50
-
51
- def get_params(signature):
52
- in_, out = signature.split("->")
53
- in_params = []
54
- out_params = []
55
- for i, typecode in enumerate(in_):
56
- in_params.append(f"{dtype_map[typecode]} in{i}")
57
- for i, typecode in enumerate(out):
58
- out_params.append(f"{dtype_map[typecode]} out{i}")
59
- in_params = ", ".join(in_params)
60
- out_params = ", ".join(out_params)
61
- return in_params, out_params
62
-
63
-
64
- def get_sample_input(signature, xp):
65
- dtype_map = {
66
- "f": xp.float32,
67
- "d": xp.float64,
68
- "F": xp.complex64,
69
- "D": xp.complex128,
70
- "i": xp.int32,
71
- "l": xp.int64,
72
- }
73
-
74
- in_, _ = signature.split("->")
75
- args = []
76
- for typecode in in_:
77
- args.append(xp.zeros(2, dtype=dtype_map[typecode]))
78
- return args
79
-
80
-
81
- @pytest.fixture(scope="module", autouse=True)
82
- def manage_cupy_cache():
83
- # Temporarily change cupy kernel cache location so kernel cache will not be polluted
84
- # by these tests. Remove temporary cache in teardown.
85
- temp_cache_dir = tempfile.mkdtemp()
86
- original_cache_dir = os.environ.get('CUPY_CACHE_DIR', None)
87
- os.environ['CUPY_CACHE_DIR'] = temp_cache_dir
88
-
89
- yield
90
-
91
- if original_cache_dir is not None:
92
- os.environ['CUPY_CACHE_DIR'] = original_cache_dir
93
- else:
94
- del os.environ['CUPY_CACHE_DIR']
95
- shutil.rmtree(temp_cache_dir)
96
-
97
-
98
- @check_version(cupy, "13.0.0")
99
- @pytest.mark.parametrize("signature,preamble,routine", get_test_cases())
100
- @pytest.mark.xslow
101
- def test_compiles_in_cupy(signature, preamble, routine, manage_cupy_cache):
102
- name = f"x{uuid4().hex}"
103
- in_params, out_params = get_params(signature)
104
-
105
- func = cupy.ElementwiseKernel(
106
- in_params,
107
- out_params,
108
- routine,
109
- name,
110
- preamble=preamble,
111
- options=(f"--include-path={sc._get_include()}", "-std=c++17")
112
- )
113
-
114
- _ = func(*get_sample_input(signature, cupy))
scipy/special/xsf/binom.h DELETED
@@ -1,89 +0,0 @@
1
- /* Translated from Cython into C++ by SciPy developers in 2024.
2
- *
3
- * Original authors: Pauli Virtanen, Eric Moore
4
- */
5
-
6
- // Binomial coefficient
7
-
8
- #pragma once
9
-
10
- #include "config.h"
11
-
12
- #include "cephes/beta.h"
13
- #include "cephes/gamma.h"
14
-
15
- namespace xsf {
16
-
17
- XSF_HOST_DEVICE inline double binom(double n, double k) {
18
- double kx, nx, num, den, dk, sgn;
19
-
20
- if (n < 0) {
21
- nx = std::floor(n);
22
- if (n == nx) {
23
- // Undefined
24
- return std::numeric_limits<double>::quiet_NaN();
25
- }
26
- }
27
-
28
- kx = std::floor(k);
29
- if (k == kx && (std::abs(n) > 1E-8 || n == 0)) {
30
- /* Integer case: use multiplication formula for less rounding
31
- * error for cases where the result is an integer.
32
- *
33
- * This cannot be used for small nonzero n due to loss of
34
- * precision. */
35
- nx = std::floor(n);
36
- if (nx == n && kx > nx / 2 && nx > 0) {
37
- // Reduce kx by symmetry
38
- kx = nx - kx;
39
- }
40
-
41
- if (kx >= 0 && kx < 20) {
42
- num = 1.0;
43
- den = 1.0;
44
- for (int i = 1; i < 1 + static_cast<int>(kx); i++) {
45
- num *= i + n - kx;
46
- den *= i;
47
- if (std::abs(num) > 1E50) {
48
- num /= den;
49
- den = 1.0;
50
- }
51
- }
52
- return num / den;
53
- }
54
- }
55
-
56
- // general case
57
- if (n >= 1E10 * k and k > 0) {
58
- // avoid under/overflows intermediate results
59
- return std::exp(-cephes::lbeta(1 + n - k, 1 + k) - std::log(n + 1));
60
- }
61
- if (k > 1E8 * std::abs(n)) {
62
- // avoid loss of precision
63
- num = cephes::Gamma(1 + n) / std::abs(k) + cephes::Gamma(1 + n) * n / (2 * k * k); // + ...
64
- num /= M_PI * std::pow(std::abs(k), n);
65
- if (k > 0) {
66
- kx = std::floor(k);
67
- if (static_cast<int>(kx) == kx) {
68
- dk = k - kx;
69
- sgn = (static_cast<int>(kx) % 2 == 0) ? 1 : -1;
70
- } else {
71
- dk = k;
72
- sgn = 1;
73
- }
74
- return num * std::sin((dk - n) * M_PI) * sgn;
75
- }
76
- kx = std::floor(k);
77
- if (static_cast<int>(kx) == kx) {
78
- return 0;
79
- }
80
- return num * std::sin(k * M_PI);
81
- }
82
- return 1 / (n + 1) / cephes::beta(1 + n - k, 1 + k);
83
- }
84
-
85
- XSF_HOST_DEVICE inline float binom(float n, float k) {
86
- return binom(static_cast<double>(n), static_cast<double>(k));
87
- }
88
-
89
- } // namespace xsf
@@ -1,100 +0,0 @@
1
-
2
- #pragma once
3
-
4
- #include "cephes/igam.h"
5
- #include "config.h"
6
- #include "error.h"
7
- #include "tools.h"
8
-
9
- namespace xsf {
10
-
11
- XSF_HOST_DEVICE inline double gdtrib(double a, double p, double x) {
12
- if (std::isnan(p) || std::isnan(a) || std::isnan(x)) {
13
- return std::numeric_limits<double>::quiet_NaN();
14
- }
15
- if (!((0 <= p) && (p <= 1))) {
16
- set_error("gdtrib", SF_ERROR_DOMAIN, "Input parameter p is out of range");
17
- return std::numeric_limits<double>::quiet_NaN();
18
- }
19
- if (!(a > 0) || std::isinf(a)) {
20
- set_error("gdtrib", SF_ERROR_DOMAIN, "Input parameter a is out of range");
21
- return std::numeric_limits<double>::quiet_NaN();
22
- }
23
- if (!(x >= 0) || std::isinf(x)) {
24
- set_error("gdtrib", SF_ERROR_DOMAIN, "Input parameter x is out of range");
25
- return std::numeric_limits<double>::quiet_NaN();
26
- }
27
- if (x == 0.0) {
28
- if (p == 0.0) {
29
- set_error("gdtrib", SF_ERROR_DOMAIN, "Indeterminate result for (x, p) == (0, 0).");
30
- return std::numeric_limits<double>::quiet_NaN();
31
- }
32
- /* gdtrib(a, p, x) tends to 0 as x -> 0 when p > 0 */
33
- return 0.0;
34
- }
35
- if (p == 0.0) {
36
- /* gdtrib(a, p, x) tends to infinity as p -> 0 from the right when x > 0. */
37
- set_error("gdtrib", SF_ERROR_SINGULAR, NULL);
38
- return std::numeric_limits<double>::infinity();
39
- }
40
- if (p == 1.0) {
41
- /* gdtrib(a, p, x) tends to 0 as p -> 1.0 from the left when x > 0. */
42
- return 0.0;
43
- }
44
- double q = 1.0 - p;
45
- auto func = [a, p, q, x](double b) {
46
- if (p <= q) {
47
- return cephes::igam(b, a * x) - p;
48
- }
49
- return q - cephes::igamc(b, a * x);
50
- };
51
- double lower_bound = std::numeric_limits<double>::min();
52
- double upper_bound = std::numeric_limits<double>::max();
53
- /* To explain the magic constants used below:
54
- * 1.0 is the initial guess for the root. -0.875 is the initial step size
55
- * for the leading bracket endpoint if the bracket search will proceed to the
56
- * left, likewise 7.0 is the initial step size when the bracket search will
57
- * proceed to the right. 0.125 is the scale factor for a left moving bracket
58
- * search and 8.0 the scale factor for a right moving bracket search. These
59
- * constants are chosen so that:
60
- *
61
- * 1. The scale factor and bracket endpoints remain powers of 2, allowing for
62
- * exact arithmetic, preventing roundoff error from causing numerical catastrophe
63
- * which could lead to unexpected results.
64
- * 2. The bracket sizes remain constant in a relative sense. Each candidate bracket
65
- * will contain roughly the same number of floating point values. This means that
66
- * the number of necessary function evaluations in the worst case scenario for
67
- * Chandrupatla's algorithm will remain constant.
68
- *
69
- * false specifies that the function is not decreasing. 342 is equal to
70
- * max(ceil(log_8(DBL_MAX)), ceil(log_(1/8)(DBL_MIN))). An upper bound for the
71
- * number of iterations needed in this bracket search to check all normalized
72
- * floating point values.
73
- */
74
- auto [xl, xr, f_xl, f_xr, bracket_status] = detail::bracket_root_for_cdf_inversion(
75
- func, 1.0, lower_bound, upper_bound, -0.875, 7.0, 0.125, 8, false, 342
76
- );
77
- if (bracket_status == 1) {
78
- set_error("gdtrib", SF_ERROR_UNDERFLOW, NULL);
79
- return 0.0;
80
- }
81
- if (bracket_status == 2) {
82
- set_error("gdtrib", SF_ERROR_OVERFLOW, NULL);
83
- return std::numeric_limits<double>::infinity();
84
- }
85
- if (bracket_status >= 3) {
86
- set_error("gdtrib", SF_ERROR_OTHER, "Computational Error");
87
- return std::numeric_limits<double>::quiet_NaN();
88
- }
89
- auto [result, root_status] = detail::find_root_chandrupatla(
90
- func, xl, xr, f_xl, f_xr, std::numeric_limits<double>::epsilon(), 1e-100, 100
91
- );
92
- if (root_status) {
93
- /* The root finding return should only fail if there's a bug in our code. */
94
- set_error("gdtrib", SF_ERROR_OTHER, "Computational Error, (%.17g, %.17g, %.17g)", a, p, x);
95
- return std::numeric_limits<double>::quiet_NaN();
96
- }
97
- return result;
98
- }
99
-
100
- } // namespace xsf
@@ -1,307 +0,0 @@
1
- /* Translated into C++ by SciPy developers in 2024.
2
- * Original header with Copyright information appears below.
3
- */
4
-
5
- /* airy.c
6
- *
7
- * Airy function
8
- *
9
- *
10
- *
11
- * SYNOPSIS:
12
- *
13
- * double x, ai, aip, bi, bip;
14
- * int airy();
15
- *
16
- * airy( x, _&ai, _&aip, _&bi, _&bip );
17
- *
18
- *
19
- *
20
- * DESCRIPTION:
21
- *
22
- * Solution of the differential equation
23
- *
24
- * y"(x) = xy.
25
- *
26
- * The function returns the two independent solutions Ai, Bi
27
- * and their first derivatives Ai'(x), Bi'(x).
28
- *
29
- * Evaluation is by power series summation for small x,
30
- * by rational minimax approximations for large x.
31
- *
32
- *
33
- *
34
- * ACCURACY:
35
- * Error criterion is absolute when function <= 1, relative
36
- * when function > 1, except * denotes relative error criterion.
37
- * For large negative x, the absolute error increases as x^1.5.
38
- * For large positive x, the relative error increases as x^1.5.
39
- *
40
- * Arithmetic domain function # trials peak rms
41
- * IEEE -10, 0 Ai 10000 1.6e-15 2.7e-16
42
- * IEEE 0, 10 Ai 10000 2.3e-14* 1.8e-15*
43
- * IEEE -10, 0 Ai' 10000 4.6e-15 7.6e-16
44
- * IEEE 0, 10 Ai' 10000 1.8e-14* 1.5e-15*
45
- * IEEE -10, 10 Bi 30000 4.2e-15 5.3e-16
46
- * IEEE -10, 10 Bi' 30000 4.9e-15 7.3e-16
47
- *
48
- */
49
- /* airy.c */
50
-
51
- /*
52
- * Cephes Math Library Release 2.8: June, 2000
53
- * Copyright 1984, 1987, 1989, 2000 by Stephen L. Moshier
54
- */
55
- #pragma once
56
-
57
- #include "../config.h"
58
- #include "const.h"
59
- #include "polevl.h"
60
-
61
- namespace xsf {
62
- namespace cephes {
63
-
64
- namespace detail {
65
-
66
- constexpr double airy_c1 = 0.35502805388781723926;
67
- constexpr double airy_c2 = 0.258819403792806798405;
68
- constexpr double MAXAIRY = 103.892;
69
-
70
- constexpr double airy_AN[8] = {
71
- 3.46538101525629032477E-1, 1.20075952739645805542E1, 7.62796053615234516538E1, 1.68089224934630576269E2,
72
- 1.59756391350164413639E2, 7.05360906840444183113E1, 1.40264691163389668864E1, 9.99999999999999995305E-1,
73
- };
74
-
75
- constexpr double airy_AD[8] = {
76
- 5.67594532638770212846E-1, 1.47562562584847203173E1, 8.45138970141474626562E1, 1.77318088145400459522E2,
77
- 1.64234692871529701831E2, 7.14778400825575695274E1, 1.40959135607834029598E1, 1.00000000000000000470E0,
78
- };
79
-
80
- constexpr double airy_APN[8] = {
81
- 6.13759184814035759225E-1, 1.47454670787755323881E1, 8.20584123476060982430E1, 1.71184781360976385540E2,
82
- 1.59317847137141783523E2, 6.99778599330103016170E1, 1.39470856980481566958E1, 1.00000000000000000550E0,
83
- };
84
-
85
- constexpr double airy_APD[8] = {
86
- 3.34203677749736953049E-1, 1.11810297306158156705E1, 7.11727352147859965283E1, 1.58778084372838313640E2,
87
- 1.53206427475809220834E2, 6.86752304592780337944E1, 1.38498634758259442477E1, 9.99999999999999994502E-1,
88
- };
89
-
90
- constexpr double airy_BN16[5] = {
91
- -2.53240795869364152689E-1, 5.75285167332467384228E-1, -3.29907036873225371650E-1,
92
- 6.44404068948199951727E-2, -3.82519546641336734394E-3,
93
- };
94
-
95
- constexpr double airy_BD16[5] = {
96
- /* 1.00000000000000000000E0, */
97
- -7.15685095054035237902E0, 1.06039580715664694291E1, -5.23246636471251500874E0,
98
- 9.57395864378383833152E-1, -5.50828147163549611107E-2,
99
- };
100
-
101
- constexpr double airy_BPPN[5] = {
102
- 4.65461162774651610328E-1, -1.08992173800493920734E0, 6.38800117371827987759E-1,
103
- -1.26844349553102907034E-1, 7.62487844342109852105E-3,
104
- };
105
-
106
- constexpr double airy_BPPD[5] = {
107
- /* 1.00000000000000000000E0, */
108
- -8.70622787633159124240E0, 1.38993162704553213172E1, -7.14116144616431159572E0,
109
- 1.34008595960680518666E0, -7.84273211323341930448E-2,
110
- };
111
-
112
- constexpr double airy_AFN[9] = {
113
- -1.31696323418331795333E-1, -6.26456544431912369773E-1, -6.93158036036933542233E-1,
114
- -2.79779981545119124951E-1, -4.91900132609500318020E-2, -4.06265923594885404393E-3,
115
- -1.59276496239262096340E-4, -2.77649108155232920844E-6, -1.67787698489114633780E-8,
116
- };
117
-
118
- constexpr double airy_AFD[9] = {
119
- /* 1.00000000000000000000E0, */
120
- 1.33560420706553243746E1, 3.26825032795224613948E1, 2.67367040941499554804E1,
121
- 9.18707402907259625840E0, 1.47529146771666414581E0, 1.15687173795188044134E-1,
122
- 4.40291641615211203805E-3, 7.54720348287414296618E-5, 4.51850092970580378464E-7,
123
- };
124
-
125
- constexpr double airy_AGN[11] = {
126
- 1.97339932091685679179E-2, 3.91103029615688277255E-1, 1.06579897599595591108E0, 9.39169229816650230044E-1,
127
- 3.51465656105547619242E-1, 6.33888919628925490927E-2, 5.85804113048388458567E-3, 2.82851600836737019778E-4,
128
- 6.98793669997260967291E-6, 8.11789239554389293311E-8, 3.41551784765923618484E-10,
129
- };
130
-
131
- constexpr double airy_AGD[10] = {
132
- /* 1.00000000000000000000E0, */
133
- 9.30892908077441974853E0, 1.98352928718312140417E1, 1.55646628932864612953E1, 5.47686069422975497931E0,
134
- 9.54293611618961883998E-1, 8.64580826352392193095E-2, 4.12656523824222607191E-3, 1.01259085116509135510E-4,
135
- 1.17166733214413521882E-6, 4.91834570062930015649E-9,
136
- };
137
-
138
- constexpr double airy_APFN[9] = {
139
- 1.85365624022535566142E-1, 8.86712188052584095637E-1, 9.87391981747398547272E-1,
140
- 4.01241082318003734092E-1, 7.10304926289631174579E-2, 5.90618657995661810071E-3,
141
- 2.33051409401776799569E-4, 4.08718778289035454598E-6, 2.48379932900442457853E-8,
142
- };
143
-
144
- constexpr double airy_APFD[9] = {
145
- /* 1.00000000000000000000E0, */
146
- 1.47345854687502542552E1, 3.75423933435489594466E1, 3.14657751203046424330E1,
147
- 1.09969125207298778536E1, 1.78885054766999417817E0, 1.41733275753662636873E-1,
148
- 5.44066067017226003627E-3, 9.39421290654511171663E-5, 5.65978713036027009243E-7,
149
- };
150
-
151
- constexpr double airy_APGN[11] = {
152
- -3.55615429033082288335E-2, -6.37311518129435504426E-1, -1.70856738884312371053E0,
153
- -1.50221872117316635393E0, -5.63606665822102676611E-1, -1.02101031120216891789E-1,
154
- -9.48396695961445269093E-3, -4.60325307486780994357E-4, -1.14300836484517375919E-5,
155
- -1.33415518685547420648E-7, -5.63803833958893494476E-10,
156
- };
157
-
158
- constexpr double airy_APGD[11] = {
159
- /* 1.00000000000000000000E0, */
160
- 9.85865801696130355144E0, 2.16401867356585941885E1, 1.73130776389749389525E1, 6.17872175280828766327E0,
161
- 1.08848694396321495475E0, 9.95005543440888479402E-2, 4.78468199683886610842E-3, 1.18159633322838625562E-4,
162
- 1.37480673554219441465E-6, 5.79912514929147598821E-9,
163
- };
164
-
165
- } // namespace detail
166
-
167
- XSF_HOST_DEVICE inline int airy(double x, double *ai, double *aip, double *bi, double *bip) {
168
- double z, zz, t, f, g, uf, ug, k, zeta, theta;
169
- int domflg;
170
-
171
- domflg = 0;
172
- if (x > detail::MAXAIRY) {
173
- *ai = 0;
174
- *aip = 0;
175
- *bi = std::numeric_limits<double>::infinity();
176
- *bip = std::numeric_limits<double>::infinity();
177
- return (-1);
178
- }
179
-
180
- if (x < -2.09) {
181
- domflg = 15;
182
- t = std::sqrt(-x);
183
- zeta = -2.0 * x * t / 3.0;
184
- t = std::sqrt(t);
185
- k = detail::SQRT1OPI / t;
186
- z = 1.0 / zeta;
187
- zz = z * z;
188
- uf = 1.0 + zz * polevl(zz, detail::airy_AFN, 8) / p1evl(zz, detail::airy_AFD, 9);
189
- ug = z * polevl(zz, detail::airy_AGN, 10) / p1evl(zz, detail::airy_AGD, 10);
190
- theta = zeta + 0.25 * M_PI;
191
- f = std::sin(theta);
192
- g = std::cos(theta);
193
- *ai = k * (f * uf - g * ug);
194
- *bi = k * (g * uf + f * ug);
195
- uf = 1.0 + zz * polevl(zz, detail::airy_APFN, 8) / p1evl(zz, detail::airy_APFD, 9);
196
- ug = z * polevl(zz, detail::airy_APGN, 10) / p1evl(zz, detail::airy_APGD, 10);
197
- k = detail::SQRT1OPI * t;
198
- *aip = -k * (g * uf + f * ug);
199
- *bip = k * (f * uf - g * ug);
200
- return (0);
201
- }
202
-
203
- if (x >= 2.09) { /* cbrt(9) */
204
- domflg = 5;
205
- t = std::sqrt(x);
206
- zeta = 2.0 * x * t / 3.0;
207
- g = std::exp(zeta);
208
- t = std::sqrt(t);
209
- k = 2.0 * t * g;
210
- z = 1.0 / zeta;
211
- f = polevl(z, detail::airy_AN, 7) / polevl(z, detail::airy_AD, 7);
212
- *ai = detail::SQRT1OPI * f / k;
213
- k = -0.5 * detail::SQRT1OPI * t / g;
214
- f = polevl(z, detail::airy_APN, 7) / polevl(z, detail::airy_APD, 7);
215
- *aip = f * k;
216
-
217
- if (x > 8.3203353) { /* zeta > 16 */
218
- f = z * polevl(z, detail::airy_BN16, 4) / p1evl(z, detail::airy_BD16, 5);
219
- k = detail::SQRT1OPI * g;
220
- *bi = k * (1.0 + f) / t;
221
- f = z * polevl(z, detail::airy_BPPN, 4) / p1evl(z, detail::airy_BPPD, 5);
222
- *bip = k * t * (1.0 + f);
223
- return (0);
224
- }
225
- }
226
-
227
- f = 1.0;
228
- g = x;
229
- t = 1.0;
230
- uf = 1.0;
231
- ug = x;
232
- k = 1.0;
233
- z = x * x * x;
234
- while (t > detail::MACHEP) {
235
- uf *= z;
236
- k += 1.0;
237
- uf /= k;
238
- ug *= z;
239
- k += 1.0;
240
- ug /= k;
241
- uf /= k;
242
- f += uf;
243
- k += 1.0;
244
- ug /= k;
245
- g += ug;
246
- t = std::abs(uf / f);
247
- }
248
- uf = detail::airy_c1 * f;
249
- ug = detail::airy_c2 * g;
250
- if ((domflg & 1) == 0) {
251
- *ai = uf - ug;
252
- }
253
- if ((domflg & 2) == 0) {
254
- *bi = detail::SQRT3 * (uf + ug);
255
- }
256
-
257
- /* the deriviative of ai */
258
- k = 4.0;
259
- uf = x * x / 2.0;
260
- ug = z / 3.0;
261
- f = uf;
262
- g = 1.0 + ug;
263
- uf /= 3.0;
264
- t = 1.0;
265
-
266
- while (t > detail::MACHEP) {
267
- uf *= z;
268
- ug /= k;
269
- k += 1.0;
270
- ug *= z;
271
- uf /= k;
272
- f += uf;
273
- k += 1.0;
274
- ug /= k;
275
- uf /= k;
276
- g += ug;
277
- k += 1.0;
278
- t = std::abs(ug / g);
279
- }
280
-
281
- uf = detail::airy_c1 * f;
282
- ug = detail::airy_c2 * g;
283
- if ((domflg & 4) == 0) {
284
- *aip = uf - ug;
285
- }
286
- if ((domflg & 8) == 0) {
287
- *bip = detail::SQRT3 * (uf + ug);
288
- };
289
- return (0);
290
- }
291
-
292
- inline int airy(float xf, float *aif, float *aipf, float *bif, float *bipf) {
293
- double ai;
294
- double aip;
295
- double bi;
296
- double bip;
297
- int res = cephes::airy(xf, &ai, &aip, &bi, &bip);
298
-
299
- *aif = ai;
300
- *aipf = aip;
301
- *bif = bi;
302
- *bipf = bip;
303
- return res;
304
- }
305
-
306
- } // namespace cephes
307
- } // namespace xsf
@@ -1,51 +0,0 @@
1
- /* Translated into C++ by SciPy developers in 2024.
2
- *
3
- * This was not part of the original cephes library.
4
- */
5
- #pragma once
6
-
7
- #include "../config.h"
8
- #include "gamma.h"
9
-
10
- namespace xsf {
11
- namespace cephes {
12
- namespace detail {
13
-
14
- constexpr double besselpoly_EPS = 1.0e-17;
15
- }
16
-
17
- XSF_HOST_DEVICE inline double besselpoly(double a, double lambda, double nu) {
18
-
19
- int m, factor = 0;
20
- double Sm, relerr, Sol;
21
- double sum = 0.0;
22
-
23
- /* Special handling for a = 0.0 */
24
- if (a == 0.0) {
25
- if (nu == 0.0) {
26
- return 1.0 / (lambda + 1);
27
- } else {
28
- return 0.0;
29
- }
30
- }
31
- /* Special handling for negative and integer nu */
32
- if ((nu < 0) && (std::floor(nu) == nu)) {
33
- nu = -nu;
34
- factor = static_cast<int>(nu) % 2;
35
- }
36
- Sm = std::exp(nu * std::log(a)) / (Gamma(nu + 1) * (lambda + nu + 1));
37
- m = 0;
38
- do {
39
- sum += Sm;
40
- Sol = Sm;
41
- Sm *= -a * a * (lambda + nu + 1 + 2 * m) / ((nu + m + 1) * (m + 1) * (lambda + nu + 1 + 2 * m + 2));
42
- m++;
43
- relerr = std::abs((Sm - Sol) / Sm);
44
- } while (relerr > detail::besselpoly_EPS && m < 1000);
45
- if (!factor)
46
- return sum;
47
- else
48
- return -sum;
49
- }
50
- } // namespace cephes
51
- } // namespace xsf