scipy 1.15.2__cp313-cp313t-macosx_14_0_arm64.whl → 1.16.0rc1__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 (637) hide show
  1. scipy/__config__.py +3 -3
  2. scipy/__init__.py +3 -6
  3. scipy/_cyutility.cpython-313t-darwin.so +0 -0
  4. scipy/_lib/_array_api.py +497 -161
  5. scipy/_lib/_array_api_compat_vendor.py +9 -0
  6. scipy/_lib/_bunch.py +4 -0
  7. scipy/_lib/_ccallback_c.cpython-313t-darwin.so +0 -0
  8. scipy/_lib/_docscrape.py +1 -1
  9. scipy/_lib/_elementwise_iterative_method.py +15 -26
  10. scipy/_lib/_sparse.py +41 -0
  11. scipy/_lib/_test_ccallback.cpython-313t-darwin.so +0 -0
  12. scipy/_lib/_test_deprecation_call.cpython-313t-darwin.so +0 -0
  13. scipy/_lib/_test_deprecation_def.cpython-313t-darwin.so +0 -0
  14. scipy/_lib/_testutils.py +6 -2
  15. scipy/_lib/_util.py +222 -125
  16. scipy/_lib/array_api_compat/__init__.py +4 -4
  17. scipy/_lib/array_api_compat/_internal.py +19 -6
  18. scipy/_lib/array_api_compat/common/__init__.py +1 -1
  19. scipy/_lib/array_api_compat/common/_aliases.py +365 -193
  20. scipy/_lib/array_api_compat/common/_fft.py +94 -64
  21. scipy/_lib/array_api_compat/common/_helpers.py +413 -180
  22. scipy/_lib/array_api_compat/common/_linalg.py +116 -40
  23. scipy/_lib/array_api_compat/common/_typing.py +179 -10
  24. scipy/_lib/array_api_compat/cupy/__init__.py +1 -4
  25. scipy/_lib/array_api_compat/cupy/_aliases.py +61 -41
  26. scipy/_lib/array_api_compat/cupy/_info.py +16 -6
  27. scipy/_lib/array_api_compat/cupy/_typing.py +24 -39
  28. scipy/_lib/array_api_compat/dask/array/__init__.py +6 -3
  29. scipy/_lib/array_api_compat/dask/array/_aliases.py +267 -108
  30. scipy/_lib/array_api_compat/dask/array/_info.py +105 -34
  31. scipy/_lib/array_api_compat/dask/array/fft.py +5 -8
  32. scipy/_lib/array_api_compat/dask/array/linalg.py +21 -22
  33. scipy/_lib/array_api_compat/numpy/__init__.py +13 -15
  34. scipy/_lib/array_api_compat/numpy/_aliases.py +98 -49
  35. scipy/_lib/array_api_compat/numpy/_info.py +36 -16
  36. scipy/_lib/array_api_compat/numpy/_typing.py +27 -43
  37. scipy/_lib/array_api_compat/numpy/fft.py +11 -5
  38. scipy/_lib/array_api_compat/numpy/linalg.py +75 -22
  39. scipy/_lib/array_api_compat/torch/__init__.py +3 -5
  40. scipy/_lib/array_api_compat/torch/_aliases.py +262 -159
  41. scipy/_lib/array_api_compat/torch/_info.py +27 -16
  42. scipy/_lib/array_api_compat/torch/_typing.py +3 -0
  43. scipy/_lib/array_api_compat/torch/fft.py +17 -18
  44. scipy/_lib/array_api_compat/torch/linalg.py +16 -16
  45. scipy/_lib/array_api_extra/__init__.py +26 -3
  46. scipy/_lib/array_api_extra/_delegation.py +171 -0
  47. scipy/_lib/array_api_extra/_lib/__init__.py +1 -0
  48. scipy/_lib/array_api_extra/_lib/_at.py +463 -0
  49. scipy/_lib/array_api_extra/_lib/_backends.py +46 -0
  50. scipy/_lib/array_api_extra/_lib/_funcs.py +937 -0
  51. scipy/_lib/array_api_extra/_lib/_lazy.py +357 -0
  52. scipy/_lib/array_api_extra/_lib/_testing.py +278 -0
  53. scipy/_lib/array_api_extra/_lib/_utils/__init__.py +1 -0
  54. scipy/_lib/array_api_extra/_lib/_utils/_compat.py +74 -0
  55. scipy/_lib/array_api_extra/_lib/_utils/_compat.pyi +45 -0
  56. scipy/_lib/array_api_extra/_lib/_utils/_helpers.py +559 -0
  57. scipy/_lib/array_api_extra/_lib/_utils/_typing.py +10 -0
  58. scipy/_lib/array_api_extra/_lib/_utils/_typing.pyi +105 -0
  59. scipy/_lib/array_api_extra/testing.py +359 -0
  60. scipy/_lib/decorator.py +2 -2
  61. scipy/_lib/doccer.py +1 -7
  62. scipy/_lib/messagestream.cpython-313t-darwin.so +0 -0
  63. scipy/_lib/pyprima/__init__.py +212 -0
  64. scipy/_lib/pyprima/cobyla/__init__.py +0 -0
  65. scipy/_lib/pyprima/cobyla/cobyla.py +559 -0
  66. scipy/_lib/pyprima/cobyla/cobylb.py +714 -0
  67. scipy/_lib/pyprima/cobyla/geometry.py +226 -0
  68. scipy/_lib/pyprima/cobyla/initialize.py +215 -0
  69. scipy/_lib/pyprima/cobyla/trustregion.py +492 -0
  70. scipy/_lib/pyprima/cobyla/update.py +289 -0
  71. scipy/_lib/pyprima/common/__init__.py +0 -0
  72. scipy/_lib/pyprima/common/_bounds.py +34 -0
  73. scipy/_lib/pyprima/common/_linear_constraints.py +46 -0
  74. scipy/_lib/pyprima/common/_nonlinear_constraints.py +54 -0
  75. scipy/_lib/pyprima/common/_project.py +173 -0
  76. scipy/_lib/pyprima/common/checkbreak.py +93 -0
  77. scipy/_lib/pyprima/common/consts.py +47 -0
  78. scipy/_lib/pyprima/common/evaluate.py +99 -0
  79. scipy/_lib/pyprima/common/history.py +38 -0
  80. scipy/_lib/pyprima/common/infos.py +30 -0
  81. scipy/_lib/pyprima/common/linalg.py +435 -0
  82. scipy/_lib/pyprima/common/message.py +290 -0
  83. scipy/_lib/pyprima/common/powalg.py +131 -0
  84. scipy/_lib/pyprima/common/preproc.py +277 -0
  85. scipy/_lib/pyprima/common/present.py +5 -0
  86. scipy/_lib/pyprima/common/ratio.py +54 -0
  87. scipy/_lib/pyprima/common/redrho.py +47 -0
  88. scipy/_lib/pyprima/common/selectx.py +296 -0
  89. scipy/_lib/tests/test__util.py +105 -121
  90. scipy/_lib/tests/test_array_api.py +169 -34
  91. scipy/_lib/tests/test_bunch.py +7 -0
  92. scipy/_lib/tests/test_ccallback.py +2 -10
  93. scipy/_lib/tests/test_public_api.py +13 -0
  94. scipy/cluster/_hierarchy.cpython-313t-darwin.so +0 -0
  95. scipy/cluster/_optimal_leaf_ordering.cpython-313t-darwin.so +0 -0
  96. scipy/cluster/_vq.cpython-313t-darwin.so +0 -0
  97. scipy/cluster/hierarchy.py +393 -223
  98. scipy/cluster/tests/test_hierarchy.py +273 -335
  99. scipy/cluster/tests/test_vq.py +45 -61
  100. scipy/cluster/vq.py +39 -35
  101. scipy/conftest.py +263 -157
  102. scipy/constants/_constants.py +4 -1
  103. scipy/constants/tests/test_codata.py +2 -2
  104. scipy/constants/tests/test_constants.py +11 -18
  105. scipy/datasets/_download_all.py +15 -1
  106. scipy/datasets/_fetchers.py +7 -1
  107. scipy/datasets/_utils.py +1 -1
  108. scipy/differentiate/_differentiate.py +25 -25
  109. scipy/differentiate/tests/test_differentiate.py +24 -25
  110. scipy/fft/_basic.py +20 -0
  111. scipy/fft/_helper.py +3 -34
  112. scipy/fft/_pocketfft/helper.py +29 -1
  113. scipy/fft/_pocketfft/tests/test_basic.py +2 -4
  114. scipy/fft/_pocketfft/tests/test_real_transforms.py +4 -4
  115. scipy/fft/_realtransforms.py +13 -0
  116. scipy/fft/tests/test_basic.py +27 -25
  117. scipy/fft/tests/test_fftlog.py +16 -7
  118. scipy/fft/tests/test_helper.py +18 -34
  119. scipy/fft/tests/test_real_transforms.py +8 -10
  120. scipy/fftpack/convolve.cpython-313t-darwin.so +0 -0
  121. scipy/fftpack/tests/test_basic.py +2 -4
  122. scipy/fftpack/tests/test_real_transforms.py +8 -9
  123. scipy/integrate/_bvp.py +9 -3
  124. scipy/integrate/_cubature.py +3 -2
  125. scipy/integrate/_dop.cpython-313t-darwin.so +0 -0
  126. scipy/integrate/_ivp/common.py +3 -3
  127. scipy/integrate/_ivp/ivp.py +9 -2
  128. scipy/integrate/_ivp/tests/test_ivp.py +19 -0
  129. scipy/integrate/_lsoda.cpython-313t-darwin.so +0 -0
  130. scipy/integrate/_ode.py +9 -2
  131. scipy/integrate/_odepack.cpython-313t-darwin.so +0 -0
  132. scipy/integrate/_quad_vec.py +21 -29
  133. scipy/integrate/_quadpack.cpython-313t-darwin.so +0 -0
  134. scipy/integrate/_quadpack_py.py +11 -7
  135. scipy/integrate/_quadrature.py +3 -3
  136. scipy/integrate/_rules/_base.py +2 -2
  137. scipy/integrate/_tanhsinh.py +57 -54
  138. scipy/integrate/_test_odeint_banded.cpython-313t-darwin.so +0 -0
  139. scipy/integrate/_vode.cpython-313t-darwin.so +0 -0
  140. scipy/integrate/tests/test__quad_vec.py +0 -6
  141. scipy/integrate/tests/test_banded_ode_solvers.py +85 -0
  142. scipy/integrate/tests/test_cubature.py +21 -35
  143. scipy/integrate/tests/test_quadrature.py +6 -8
  144. scipy/integrate/tests/test_tanhsinh.py +61 -43
  145. scipy/interpolate/__init__.py +70 -58
  146. scipy/interpolate/_bary_rational.py +22 -22
  147. scipy/interpolate/_bsplines.py +119 -66
  148. scipy/interpolate/_cubic.py +65 -50
  149. scipy/interpolate/_dfitpack.cpython-313t-darwin.so +0 -0
  150. scipy/interpolate/_dierckx.cpython-313t-darwin.so +0 -0
  151. scipy/interpolate/_fitpack.cpython-313t-darwin.so +0 -0
  152. scipy/interpolate/_fitpack2.py +9 -6
  153. scipy/interpolate/_fitpack_impl.py +32 -26
  154. scipy/interpolate/_fitpack_repro.py +23 -19
  155. scipy/interpolate/_interpnd.cpython-313t-darwin.so +0 -0
  156. scipy/interpolate/_interpolate.py +30 -12
  157. scipy/interpolate/_ndbspline.py +13 -18
  158. scipy/interpolate/_ndgriddata.py +5 -8
  159. scipy/interpolate/_polyint.py +95 -31
  160. scipy/interpolate/_ppoly.cpython-313t-darwin.so +0 -0
  161. scipy/interpolate/_rbf.py +2 -2
  162. scipy/interpolate/_rbfinterp.py +1 -1
  163. scipy/interpolate/_rbfinterp_pythran.cpython-313t-darwin.so +0 -0
  164. scipy/interpolate/_rgi.py +31 -26
  165. scipy/interpolate/_rgi_cython.cpython-313t-darwin.so +0 -0
  166. scipy/interpolate/dfitpack.py +0 -20
  167. scipy/interpolate/interpnd.py +1 -2
  168. scipy/interpolate/tests/test_bary_rational.py +2 -2
  169. scipy/interpolate/tests/test_bsplines.py +97 -1
  170. scipy/interpolate/tests/test_fitpack2.py +39 -1
  171. scipy/interpolate/tests/test_interpnd.py +32 -20
  172. scipy/interpolate/tests/test_interpolate.py +48 -4
  173. scipy/interpolate/tests/test_rgi.py +2 -1
  174. scipy/io/_fast_matrix_market/__init__.py +2 -0
  175. scipy/io/_harwell_boeing/_fortran_format_parser.py +19 -16
  176. scipy/io/_harwell_boeing/hb.py +7 -11
  177. scipy/io/_idl.py +5 -7
  178. scipy/io/_netcdf.py +15 -5
  179. scipy/io/_test_fortran.cpython-313t-darwin.so +0 -0
  180. scipy/io/arff/tests/test_arffread.py +3 -3
  181. scipy/io/matlab/__init__.py +5 -3
  182. scipy/io/matlab/_mio.py +4 -1
  183. scipy/io/matlab/_mio5.py +19 -13
  184. scipy/io/matlab/_mio5_utils.cpython-313t-darwin.so +0 -0
  185. scipy/io/matlab/_mio_utils.cpython-313t-darwin.so +0 -0
  186. scipy/io/matlab/_miobase.py +4 -1
  187. scipy/io/matlab/_streams.cpython-313t-darwin.so +0 -0
  188. scipy/io/matlab/tests/test_mio.py +46 -18
  189. scipy/io/matlab/tests/test_mio_funcs.py +1 -1
  190. scipy/io/tests/test_mmio.py +7 -1
  191. scipy/io/tests/test_wavfile.py +41 -0
  192. scipy/io/wavfile.py +57 -10
  193. scipy/linalg/_basic.py +113 -86
  194. scipy/linalg/_cythonized_array_utils.cpython-313t-darwin.so +0 -0
  195. scipy/linalg/_decomp.py +22 -9
  196. scipy/linalg/_decomp_cholesky.py +28 -13
  197. scipy/linalg/_decomp_cossin.py +45 -30
  198. scipy/linalg/_decomp_interpolative.cpython-313t-darwin.so +0 -0
  199. scipy/linalg/_decomp_ldl.py +4 -1
  200. scipy/linalg/_decomp_lu.py +18 -6
  201. scipy/linalg/_decomp_lu_cython.cpython-313t-darwin.so +0 -0
  202. scipy/linalg/_decomp_polar.py +2 -0
  203. scipy/linalg/_decomp_qr.py +6 -2
  204. scipy/linalg/_decomp_qz.py +3 -0
  205. scipy/linalg/_decomp_schur.py +3 -1
  206. scipy/linalg/_decomp_svd.py +13 -2
  207. scipy/linalg/_decomp_update.cpython-313t-darwin.so +0 -0
  208. scipy/linalg/_expm_frechet.py +4 -0
  209. scipy/linalg/_fblas.cpython-313t-darwin.so +0 -0
  210. scipy/linalg/_flapack.cpython-313t-darwin.so +0 -0
  211. scipy/linalg/_linalg_pythran.cpython-313t-darwin.so +0 -0
  212. scipy/linalg/_matfuncs.py +187 -4
  213. scipy/linalg/_matfuncs_expm.cpython-313t-darwin.so +0 -0
  214. scipy/linalg/_matfuncs_schur_sqrtm.cpython-313t-darwin.so +0 -0
  215. scipy/linalg/_matfuncs_sqrtm.py +1 -99
  216. scipy/linalg/_matfuncs_sqrtm_triu.cpython-313t-darwin.so +0 -0
  217. scipy/linalg/_procrustes.py +2 -0
  218. scipy/linalg/_sketches.py +17 -6
  219. scipy/linalg/_solve_toeplitz.cpython-313t-darwin.so +0 -0
  220. scipy/linalg/_solvers.py +7 -2
  221. scipy/linalg/_special_matrices.py +26 -36
  222. scipy/linalg/cython_blas.cpython-313t-darwin.so +0 -0
  223. scipy/linalg/cython_lapack.cpython-313t-darwin.so +0 -0
  224. scipy/linalg/lapack.py +22 -2
  225. scipy/linalg/tests/_cython_examples/meson.build +7 -0
  226. scipy/linalg/tests/test_basic.py +31 -16
  227. scipy/linalg/tests/test_batch.py +588 -0
  228. scipy/linalg/tests/test_cythonized_array_utils.py +0 -2
  229. scipy/linalg/tests/test_decomp.py +40 -3
  230. scipy/linalg/tests/test_decomp_cossin.py +14 -0
  231. scipy/linalg/tests/test_decomp_ldl.py +1 -1
  232. scipy/linalg/tests/test_interpolative.py +17 -0
  233. scipy/linalg/tests/test_lapack.py +115 -7
  234. scipy/linalg/tests/test_matfuncs.py +157 -102
  235. scipy/linalg/tests/test_procrustes.py +0 -7
  236. scipy/linalg/tests/test_solve_toeplitz.py +1 -1
  237. scipy/linalg/tests/test_special_matrices.py +1 -5
  238. scipy/ndimage/__init__.py +1 -0
  239. scipy/ndimage/_cytest.cpython-313t-darwin.so +0 -0
  240. scipy/ndimage/_delegators.py +8 -2
  241. scipy/ndimage/_filters.py +433 -5
  242. scipy/ndimage/_interpolation.py +36 -6
  243. scipy/ndimage/_measurements.py +4 -2
  244. scipy/ndimage/_morphology.py +5 -0
  245. scipy/ndimage/_nd_image.cpython-313t-darwin.so +0 -0
  246. scipy/ndimage/_ndimage_api.py +2 -1
  247. scipy/ndimage/_ni_docstrings.py +5 -1
  248. scipy/ndimage/_ni_label.cpython-313t-darwin.so +0 -0
  249. scipy/ndimage/_ni_support.py +1 -5
  250. scipy/ndimage/_rank_filter_1d.cpython-313t-darwin.so +0 -0
  251. scipy/ndimage/_support_alternative_backends.py +18 -6
  252. scipy/ndimage/tests/test_filters.py +351 -259
  253. scipy/ndimage/tests/test_fourier.py +7 -9
  254. scipy/ndimage/tests/test_interpolation.py +68 -61
  255. scipy/ndimage/tests/test_measurements.py +18 -35
  256. scipy/ndimage/tests/test_morphology.py +143 -131
  257. scipy/ndimage/tests/test_splines.py +1 -3
  258. scipy/odr/__odrpack.cpython-313t-darwin.so +0 -0
  259. scipy/optimize/_basinhopping.py +13 -7
  260. scipy/optimize/_bglu_dense.cpython-313t-darwin.so +0 -0
  261. scipy/optimize/_bracket.py +46 -26
  262. scipy/optimize/_chandrupatla.py +9 -10
  263. scipy/optimize/_cobyla_py.py +104 -123
  264. scipy/optimize/_constraints.py +14 -10
  265. scipy/optimize/_differentiable_functions.py +371 -230
  266. scipy/optimize/_differentialevolution.py +4 -3
  267. scipy/optimize/_direct.cpython-313t-darwin.so +0 -0
  268. scipy/optimize/_dual_annealing.py +1 -1
  269. scipy/optimize/_elementwise.py +1 -4
  270. scipy/optimize/_group_columns.cpython-313t-darwin.so +0 -0
  271. scipy/optimize/_highspy/_highs_wrapper.py +6 -4
  272. scipy/optimize/_lbfgsb.cpython-313t-darwin.so +0 -0
  273. scipy/optimize/_lbfgsb_py.py +57 -16
  274. scipy/optimize/_linprog_doc.py +2 -2
  275. scipy/optimize/_linprog_highs.py +11 -11
  276. scipy/optimize/_linprog_ip.py +25 -10
  277. scipy/optimize/_linprog_util.py +18 -19
  278. scipy/optimize/_lsap.cpython-313t-darwin.so +0 -0
  279. scipy/optimize/_lsq/common.py +3 -3
  280. scipy/optimize/_lsq/dogbox.py +16 -2
  281. scipy/optimize/_lsq/givens_elimination.cpython-313t-darwin.so +0 -0
  282. scipy/optimize/_lsq/least_squares.py +198 -126
  283. scipy/optimize/_lsq/lsq_linear.py +6 -6
  284. scipy/optimize/_lsq/trf.py +35 -8
  285. scipy/optimize/_milp.py +3 -1
  286. scipy/optimize/_minimize.py +105 -36
  287. scipy/optimize/_minpack.cpython-313t-darwin.so +0 -0
  288. scipy/optimize/_minpack_py.py +21 -14
  289. scipy/optimize/_moduleTNC.cpython-313t-darwin.so +0 -0
  290. scipy/optimize/_nnls.py +20 -21
  291. scipy/optimize/_nonlin.py +34 -3
  292. scipy/optimize/_numdiff.py +288 -110
  293. scipy/optimize/_optimize.py +86 -48
  294. scipy/optimize/_pava_pybind.cpython-313t-darwin.so +0 -0
  295. scipy/optimize/_remove_redundancy.py +5 -5
  296. scipy/optimize/_root_scalar.py +1 -1
  297. scipy/optimize/_shgo.py +6 -0
  298. scipy/optimize/_shgo_lib/_complex.py +1 -1
  299. scipy/optimize/_slsqp_py.py +216 -124
  300. scipy/optimize/_slsqplib.cpython-313t-darwin.so +0 -0
  301. scipy/optimize/_spectral.py +1 -1
  302. scipy/optimize/_tnc.py +8 -1
  303. scipy/optimize/_trlib/_trlib.cpython-313t-darwin.so +0 -0
  304. scipy/optimize/_trustregion.py +20 -6
  305. scipy/optimize/_trustregion_constr/canonical_constraint.py +7 -7
  306. scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +1 -1
  307. scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +11 -3
  308. scipy/optimize/_trustregion_constr/projections.py +12 -8
  309. scipy/optimize/_trustregion_constr/qp_subproblem.py +9 -9
  310. scipy/optimize/_trustregion_constr/tests/test_projections.py +7 -7
  311. scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +77 -77
  312. scipy/optimize/_trustregion_constr/tr_interior_point.py +5 -5
  313. scipy/optimize/_trustregion_exact.py +0 -1
  314. scipy/optimize/_zeros.cpython-313t-darwin.so +0 -0
  315. scipy/optimize/_zeros_py.py +97 -17
  316. scipy/optimize/cython_optimize/_zeros.cpython-313t-darwin.so +0 -0
  317. scipy/optimize/slsqp.py +0 -1
  318. scipy/optimize/tests/test__basinhopping.py +1 -1
  319. scipy/optimize/tests/test__differential_evolution.py +4 -4
  320. scipy/optimize/tests/test__linprog_clean_inputs.py +5 -3
  321. scipy/optimize/tests/test__numdiff.py +66 -22
  322. scipy/optimize/tests/test__remove_redundancy.py +2 -2
  323. scipy/optimize/tests/test__shgo.py +9 -1
  324. scipy/optimize/tests/test_bracket.py +71 -46
  325. scipy/optimize/tests/test_chandrupatla.py +133 -135
  326. scipy/optimize/tests/test_cobyla.py +74 -45
  327. scipy/optimize/tests/test_constraints.py +1 -1
  328. scipy/optimize/tests/test_differentiable_functions.py +226 -6
  329. scipy/optimize/tests/test_lbfgsb_hessinv.py +22 -0
  330. scipy/optimize/tests/test_least_squares.py +125 -13
  331. scipy/optimize/tests/test_linear_assignment.py +3 -3
  332. scipy/optimize/tests/test_linprog.py +3 -3
  333. scipy/optimize/tests/test_lsq_linear.py +5 -5
  334. scipy/optimize/tests/test_minimize_constrained.py +2 -2
  335. scipy/optimize/tests/test_minpack.py +4 -4
  336. scipy/optimize/tests/test_nnls.py +43 -3
  337. scipy/optimize/tests/test_nonlin.py +36 -0
  338. scipy/optimize/tests/test_optimize.py +95 -17
  339. scipy/optimize/tests/test_slsqp.py +36 -4
  340. scipy/optimize/tests/test_zeros.py +34 -1
  341. scipy/signal/__init__.py +12 -23
  342. scipy/signal/_delegators.py +568 -0
  343. scipy/signal/_filter_design.py +459 -241
  344. scipy/signal/_fir_filter_design.py +262 -90
  345. scipy/signal/_lti_conversion.py +3 -2
  346. scipy/signal/_ltisys.py +118 -91
  347. scipy/signal/_max_len_seq_inner.cpython-313t-darwin.so +0 -0
  348. scipy/signal/_peak_finding_utils.cpython-313t-darwin.so +0 -0
  349. scipy/signal/_polyutils.py +172 -0
  350. scipy/signal/_short_time_fft.py +553 -76
  351. scipy/signal/_signal_api.py +30 -0
  352. scipy/signal/_signaltools.py +719 -396
  353. scipy/signal/_sigtools.cpython-313t-darwin.so +0 -0
  354. scipy/signal/_sosfilt.cpython-313t-darwin.so +0 -0
  355. scipy/signal/_spectral_py.py +221 -50
  356. scipy/signal/_spline.cpython-313t-darwin.so +0 -0
  357. scipy/signal/_spline_filters.py +108 -68
  358. scipy/signal/_support_alternative_backends.py +73 -0
  359. scipy/signal/_upfirdn.py +4 -1
  360. scipy/signal/_upfirdn_apply.cpython-313t-darwin.so +0 -0
  361. scipy/signal/_waveforms.py +2 -11
  362. scipy/signal/_wavelets.py +1 -1
  363. scipy/signal/fir_filter_design.py +1 -0
  364. scipy/signal/spline.py +4 -11
  365. scipy/signal/tests/_scipy_spectral_test_shim.py +5 -182
  366. scipy/signal/tests/test_bsplines.py +114 -79
  367. scipy/signal/tests/test_cont2discrete.py +9 -2
  368. scipy/signal/tests/test_filter_design.py +721 -481
  369. scipy/signal/tests/test_fir_filter_design.py +332 -140
  370. scipy/signal/tests/test_savitzky_golay.py +4 -3
  371. scipy/signal/tests/test_short_time_fft.py +231 -5
  372. scipy/signal/tests/test_signaltools.py +2149 -1348
  373. scipy/signal/tests/test_spectral.py +19 -6
  374. scipy/signal/tests/test_splines.py +161 -96
  375. scipy/signal/tests/test_upfirdn.py +84 -50
  376. scipy/signal/tests/test_waveforms.py +20 -0
  377. scipy/signal/tests/test_windows.py +607 -466
  378. scipy/signal/windows/_windows.py +287 -148
  379. scipy/sparse/__init__.py +23 -4
  380. scipy/sparse/_base.py +269 -120
  381. scipy/sparse/_bsr.py +7 -4
  382. scipy/sparse/_compressed.py +59 -234
  383. scipy/sparse/_construct.py +90 -38
  384. scipy/sparse/_coo.py +115 -181
  385. scipy/sparse/_csc.py +4 -4
  386. scipy/sparse/_csparsetools.cpython-313t-darwin.so +0 -0
  387. scipy/sparse/_csr.py +2 -2
  388. scipy/sparse/_data.py +48 -48
  389. scipy/sparse/_dia.py +105 -21
  390. scipy/sparse/_dok.py +0 -23
  391. scipy/sparse/_index.py +4 -4
  392. scipy/sparse/_matrix.py +23 -0
  393. scipy/sparse/_sparsetools.cpython-313t-darwin.so +0 -0
  394. scipy/sparse/_sputils.py +37 -22
  395. scipy/sparse/base.py +0 -9
  396. scipy/sparse/bsr.py +0 -14
  397. scipy/sparse/compressed.py +0 -23
  398. scipy/sparse/construct.py +0 -6
  399. scipy/sparse/coo.py +0 -14
  400. scipy/sparse/csc.py +0 -3
  401. scipy/sparse/csgraph/_flow.cpython-313t-darwin.so +0 -0
  402. scipy/sparse/csgraph/_matching.cpython-313t-darwin.so +0 -0
  403. scipy/sparse/csgraph/_min_spanning_tree.cpython-313t-darwin.so +0 -0
  404. scipy/sparse/csgraph/_reordering.cpython-313t-darwin.so +0 -0
  405. scipy/sparse/csgraph/_shortest_path.cpython-313t-darwin.so +0 -0
  406. scipy/sparse/csgraph/_tools.cpython-313t-darwin.so +0 -0
  407. scipy/sparse/csgraph/_traversal.cpython-313t-darwin.so +0 -0
  408. scipy/sparse/csgraph/tests/test_matching.py +14 -2
  409. scipy/sparse/csgraph/tests/test_pydata_sparse.py +4 -1
  410. scipy/sparse/csgraph/tests/test_shortest_path.py +83 -27
  411. scipy/sparse/csr.py +0 -5
  412. scipy/sparse/data.py +1 -6
  413. scipy/sparse/dia.py +0 -7
  414. scipy/sparse/dok.py +0 -10
  415. scipy/sparse/linalg/_dsolve/_superlu.cpython-313t-darwin.so +0 -0
  416. scipy/sparse/linalg/_dsolve/linsolve.py +9 -0
  417. scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +35 -28
  418. scipy/sparse/linalg/_eigen/arpack/_arpack.cpython-313t-darwin.so +0 -0
  419. scipy/sparse/linalg/_eigen/arpack/arpack.py +28 -20
  420. scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +6 -6
  421. scipy/sparse/linalg/_expm_multiply.py +8 -3
  422. scipy/sparse/linalg/_interface.py +29 -26
  423. scipy/sparse/linalg/_isolve/_gcrotmk.py +6 -5
  424. scipy/sparse/linalg/_isolve/iterative.py +51 -45
  425. scipy/sparse/linalg/_isolve/lgmres.py +6 -6
  426. scipy/sparse/linalg/_isolve/minres.py +5 -5
  427. scipy/sparse/linalg/_isolve/tfqmr.py +7 -7
  428. scipy/sparse/linalg/_isolve/utils.py +2 -8
  429. scipy/sparse/linalg/_matfuncs.py +1 -1
  430. scipy/sparse/linalg/_norm.py +1 -1
  431. scipy/sparse/linalg/_propack/_cpropack.cpython-313t-darwin.so +0 -0
  432. scipy/sparse/linalg/_propack/_dpropack.cpython-313t-darwin.so +0 -0
  433. scipy/sparse/linalg/_propack/_spropack.cpython-313t-darwin.so +0 -0
  434. scipy/sparse/linalg/_propack/_zpropack.cpython-313t-darwin.so +0 -0
  435. scipy/sparse/linalg/_special_sparse_arrays.py +39 -38
  436. scipy/sparse/linalg/tests/test_expm_multiply.py +10 -0
  437. scipy/sparse/linalg/tests/test_interface.py +35 -0
  438. scipy/sparse/linalg/tests/test_pydata_sparse.py +18 -0
  439. scipy/sparse/tests/test_arithmetic1d.py +5 -2
  440. scipy/sparse/tests/test_base.py +217 -40
  441. scipy/sparse/tests/test_common1d.py +17 -12
  442. scipy/sparse/tests/test_construct.py +1 -1
  443. scipy/sparse/tests/test_coo.py +272 -4
  444. scipy/sparse/tests/test_sparsetools.py +5 -0
  445. scipy/sparse/tests/test_sputils.py +36 -7
  446. scipy/spatial/_ckdtree.cpython-313t-darwin.so +0 -0
  447. scipy/spatial/_distance_pybind.cpython-313t-darwin.so +0 -0
  448. scipy/spatial/_distance_wrap.cpython-313t-darwin.so +0 -0
  449. scipy/spatial/_hausdorff.cpython-313t-darwin.so +0 -0
  450. scipy/spatial/_qhull.cpython-313t-darwin.so +0 -0
  451. scipy/spatial/_voronoi.cpython-313t-darwin.so +0 -0
  452. scipy/spatial/distance.py +49 -42
  453. scipy/spatial/tests/test_distance.py +3 -1
  454. scipy/spatial/tests/test_kdtree.py +1 -0
  455. scipy/spatial/tests/test_qhull.py +106 -2
  456. scipy/spatial/transform/__init__.py +5 -3
  457. scipy/spatial/transform/_rigid_transform.cpython-313t-darwin.so +0 -0
  458. scipy/spatial/transform/_rotation.cpython-313t-darwin.so +0 -0
  459. scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
  460. scipy/spatial/transform/tests/test_rotation.py +1342 -790
  461. scipy/spatial/transform/tests/test_rotation_groups.py +3 -3
  462. scipy/spatial/transform/tests/test_rotation_spline.py +29 -8
  463. scipy/special/__init__.py +1 -47
  464. scipy/special/_add_newdocs.py +34 -772
  465. scipy/special/_basic.py +22 -25
  466. scipy/special/_comb.cpython-313t-darwin.so +0 -0
  467. scipy/special/_ellip_harm_2.cpython-313t-darwin.so +0 -0
  468. scipy/special/_gufuncs.cpython-313t-darwin.so +0 -0
  469. scipy/special/_logsumexp.py +83 -69
  470. scipy/special/_orthogonal.pyi +1 -1
  471. scipy/special/_specfun.cpython-313t-darwin.so +0 -0
  472. scipy/special/_special_ufuncs.cpython-313t-darwin.so +0 -0
  473. scipy/special/_spherical_bessel.py +4 -4
  474. scipy/special/_support_alternative_backends.py +212 -119
  475. scipy/special/_test_internal.cpython-313t-darwin.so +0 -0
  476. scipy/special/_testutils.py +4 -4
  477. scipy/special/_ufuncs.cpython-313t-darwin.so +0 -0
  478. scipy/special/_ufuncs.pyi +1 -0
  479. scipy/special/_ufuncs.pyx +215 -1400
  480. scipy/special/_ufuncs_cxx.cpython-313t-darwin.so +0 -0
  481. scipy/special/_ufuncs_cxx.pxd +2 -15
  482. scipy/special/_ufuncs_cxx.pyx +5 -44
  483. scipy/special/_ufuncs_cxx_defs.h +2 -16
  484. scipy/special/_ufuncs_defs.h +0 -8
  485. scipy/special/cython_special.cpython-313t-darwin.so +0 -0
  486. scipy/special/cython_special.pxd +1 -1
  487. scipy/special/tests/_cython_examples/meson.build +10 -1
  488. scipy/special/tests/test_basic.py +153 -20
  489. scipy/special/tests/test_boost_ufuncs.py +3 -0
  490. scipy/special/tests/test_cdflib.py +35 -11
  491. scipy/special/tests/test_gammainc.py +16 -0
  492. scipy/special/tests/test_hyp2f1.py +23 -2
  493. scipy/special/tests/test_log1mexp.py +85 -0
  494. scipy/special/tests/test_logsumexp.py +220 -64
  495. scipy/special/tests/test_mpmath.py +1 -0
  496. scipy/special/tests/test_nan_inputs.py +1 -1
  497. scipy/special/tests/test_orthogonal.py +17 -18
  498. scipy/special/tests/test_sf_error.py +3 -2
  499. scipy/special/tests/test_sph_harm.py +6 -7
  500. scipy/special/tests/test_support_alternative_backends.py +211 -76
  501. scipy/stats/__init__.py +4 -1
  502. scipy/stats/_ansari_swilk_statistics.cpython-313t-darwin.so +0 -0
  503. scipy/stats/_axis_nan_policy.py +4 -3
  504. scipy/stats/_biasedurn.cpython-313t-darwin.so +0 -0
  505. scipy/stats/_continued_fraction.py +387 -0
  506. scipy/stats/_continuous_distns.py +296 -319
  507. scipy/stats/_covariance.py +6 -3
  508. scipy/stats/_discrete_distns.py +39 -32
  509. scipy/stats/_distn_infrastructure.py +39 -12
  510. scipy/stats/_distribution_infrastructure.py +900 -238
  511. scipy/stats/_entropy.py +7 -8
  512. scipy/{_lib → stats}/_finite_differences.py +1 -1
  513. scipy/stats/_hypotests.py +82 -49
  514. scipy/stats/_kde.py +53 -49
  515. scipy/stats/_ksstats.py +1 -1
  516. scipy/stats/_levy_stable/__init__.py +7 -15
  517. scipy/stats/_levy_stable/levyst.cpython-313t-darwin.so +0 -0
  518. scipy/stats/_morestats.py +112 -67
  519. scipy/stats/_mstats_basic.py +13 -17
  520. scipy/stats/_mstats_extras.py +8 -8
  521. scipy/stats/_multivariate.py +89 -113
  522. scipy/stats/_new_distributions.py +97 -20
  523. scipy/stats/_page_trend_test.py +12 -5
  524. scipy/stats/_probability_distribution.py +265 -43
  525. scipy/stats/_qmc.py +14 -9
  526. scipy/stats/_qmc_cy.cpython-313t-darwin.so +0 -0
  527. scipy/stats/_qmvnt.py +16 -95
  528. scipy/stats/_qmvnt_cy.cpython-313t-darwin.so +0 -0
  529. scipy/stats/_quantile.py +335 -0
  530. scipy/stats/_rcont/rcont.cpython-313t-darwin.so +0 -0
  531. scipy/stats/_resampling.py +4 -29
  532. scipy/stats/_sampling.py +1 -1
  533. scipy/stats/_sobol.cpython-313t-darwin.so +0 -0
  534. scipy/stats/_stats.cpython-313t-darwin.so +0 -0
  535. scipy/stats/_stats_mstats_common.py +19 -2
  536. scipy/stats/_stats_py.py +534 -460
  537. scipy/stats/_stats_pythran.cpython-313t-darwin.so +0 -0
  538. scipy/stats/_unuran/unuran_wrapper.cpython-313t-darwin.so +0 -0
  539. scipy/stats/_unuran/unuran_wrapper.pyi +2 -1
  540. scipy/stats/_variation.py +5 -7
  541. scipy/stats/_wilcoxon.py +13 -7
  542. scipy/stats/tests/common_tests.py +6 -4
  543. scipy/stats/tests/test_axis_nan_policy.py +62 -24
  544. scipy/stats/tests/test_continued_fraction.py +173 -0
  545. scipy/stats/tests/test_continuous.py +379 -60
  546. scipy/stats/tests/test_continuous_basic.py +18 -12
  547. scipy/stats/tests/test_discrete_basic.py +14 -8
  548. scipy/stats/tests/test_discrete_distns.py +16 -16
  549. scipy/stats/tests/test_distributions.py +117 -75
  550. scipy/stats/tests/test_entropy.py +40 -48
  551. scipy/stats/tests/test_fit.py +4 -3
  552. scipy/stats/tests/test_hypotests.py +153 -24
  553. scipy/stats/tests/test_kdeoth.py +109 -41
  554. scipy/stats/tests/test_marray.py +289 -0
  555. scipy/stats/tests/test_morestats.py +79 -47
  556. scipy/stats/tests/test_mstats_basic.py +3 -3
  557. scipy/stats/tests/test_multivariate.py +434 -83
  558. scipy/stats/tests/test_qmc.py +13 -10
  559. scipy/stats/tests/test_quantile.py +199 -0
  560. scipy/stats/tests/test_rank.py +119 -112
  561. scipy/stats/tests/test_resampling.py +47 -56
  562. scipy/stats/tests/test_sampling.py +9 -4
  563. scipy/stats/tests/test_stats.py +799 -939
  564. scipy/stats/tests/test_variation.py +8 -6
  565. scipy/version.py +2 -2
  566. {scipy-1.15.2.dist-info → scipy-1.16.0rc1.dist-info}/LICENSE.txt +1 -1
  567. {scipy-1.15.2.dist-info → scipy-1.16.0rc1.dist-info}/METADATA +9 -9
  568. {scipy-1.15.2.dist-info → scipy-1.16.0rc1.dist-info}/RECORD +569 -576
  569. scipy-1.16.0rc1.dist-info/WHEEL +6 -0
  570. scipy/_lib/array_api_extra/_funcs.py +0 -484
  571. scipy/_lib/array_api_extra/_typing.py +0 -8
  572. scipy/interpolate/_bspl.cpython-313t-darwin.so +0 -0
  573. scipy/optimize/_cobyla.cpython-313t-darwin.so +0 -0
  574. scipy/optimize/_cython_nnls.cpython-313t-darwin.so +0 -0
  575. scipy/optimize/_slsqp.cpython-313t-darwin.so +0 -0
  576. scipy/spatial/qhull_src/COPYING.txt +0 -38
  577. scipy/special/libsf_error_state.dylib +0 -0
  578. scipy/special/tests/test_log_softmax.py +0 -109
  579. scipy/special/tests/test_xsf_cuda.py +0 -114
  580. scipy/special/xsf/binom.h +0 -89
  581. scipy/special/xsf/cdflib.h +0 -100
  582. scipy/special/xsf/cephes/airy.h +0 -307
  583. scipy/special/xsf/cephes/besselpoly.h +0 -51
  584. scipy/special/xsf/cephes/beta.h +0 -257
  585. scipy/special/xsf/cephes/cbrt.h +0 -131
  586. scipy/special/xsf/cephes/chbevl.h +0 -85
  587. scipy/special/xsf/cephes/chdtr.h +0 -193
  588. scipy/special/xsf/cephes/const.h +0 -87
  589. scipy/special/xsf/cephes/ellie.h +0 -293
  590. scipy/special/xsf/cephes/ellik.h +0 -251
  591. scipy/special/xsf/cephes/ellpe.h +0 -107
  592. scipy/special/xsf/cephes/ellpk.h +0 -117
  593. scipy/special/xsf/cephes/expn.h +0 -260
  594. scipy/special/xsf/cephes/gamma.h +0 -398
  595. scipy/special/xsf/cephes/hyp2f1.h +0 -596
  596. scipy/special/xsf/cephes/hyperg.h +0 -361
  597. scipy/special/xsf/cephes/i0.h +0 -149
  598. scipy/special/xsf/cephes/i1.h +0 -158
  599. scipy/special/xsf/cephes/igam.h +0 -421
  600. scipy/special/xsf/cephes/igam_asymp_coeff.h +0 -195
  601. scipy/special/xsf/cephes/igami.h +0 -313
  602. scipy/special/xsf/cephes/j0.h +0 -225
  603. scipy/special/xsf/cephes/j1.h +0 -198
  604. scipy/special/xsf/cephes/jv.h +0 -715
  605. scipy/special/xsf/cephes/k0.h +0 -164
  606. scipy/special/xsf/cephes/k1.h +0 -163
  607. scipy/special/xsf/cephes/kn.h +0 -243
  608. scipy/special/xsf/cephes/lanczos.h +0 -112
  609. scipy/special/xsf/cephes/ndtr.h +0 -275
  610. scipy/special/xsf/cephes/poch.h +0 -85
  611. scipy/special/xsf/cephes/polevl.h +0 -167
  612. scipy/special/xsf/cephes/psi.h +0 -194
  613. scipy/special/xsf/cephes/rgamma.h +0 -111
  614. scipy/special/xsf/cephes/scipy_iv.h +0 -811
  615. scipy/special/xsf/cephes/shichi.h +0 -248
  616. scipy/special/xsf/cephes/sici.h +0 -224
  617. scipy/special/xsf/cephes/sindg.h +0 -221
  618. scipy/special/xsf/cephes/tandg.h +0 -139
  619. scipy/special/xsf/cephes/trig.h +0 -58
  620. scipy/special/xsf/cephes/unity.h +0 -186
  621. scipy/special/xsf/cephes/zeta.h +0 -172
  622. scipy/special/xsf/config.h +0 -304
  623. scipy/special/xsf/digamma.h +0 -205
  624. scipy/special/xsf/error.h +0 -57
  625. scipy/special/xsf/evalpoly.h +0 -47
  626. scipy/special/xsf/expint.h +0 -266
  627. scipy/special/xsf/hyp2f1.h +0 -694
  628. scipy/special/xsf/iv_ratio.h +0 -173
  629. scipy/special/xsf/lambertw.h +0 -150
  630. scipy/special/xsf/loggamma.h +0 -163
  631. scipy/special/xsf/sici.h +0 -200
  632. scipy/special/xsf/tools.h +0 -427
  633. scipy/special/xsf/trig.h +0 -164
  634. scipy/special/xsf/wright_bessel.h +0 -843
  635. scipy/special/xsf/zlog1.h +0 -35
  636. scipy/stats/_mvn.cpython-313t-darwin.so +0 -0
  637. 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