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