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