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