scipy 1.16.2__cp313-cp313t-win_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 (1530) hide show
  1. scipy/__config__.py +161 -0
  2. scipy/__init__.py +150 -0
  3. scipy/_cyutility.cp313t-win_arm64.lib +0 -0
  4. scipy/_cyutility.cp313t-win_arm64.pyd +0 -0
  5. scipy/_distributor_init.py +18 -0
  6. scipy/_lib/__init__.py +14 -0
  7. scipy/_lib/_array_api.py +931 -0
  8. scipy/_lib/_array_api_compat_vendor.py +9 -0
  9. scipy/_lib/_array_api_no_0d.py +103 -0
  10. scipy/_lib/_bunch.py +229 -0
  11. scipy/_lib/_ccallback.py +251 -0
  12. scipy/_lib/_ccallback_c.cp313t-win_arm64.lib +0 -0
  13. scipy/_lib/_ccallback_c.cp313t-win_arm64.pyd +0 -0
  14. scipy/_lib/_disjoint_set.py +254 -0
  15. scipy/_lib/_docscrape.py +761 -0
  16. scipy/_lib/_elementwise_iterative_method.py +346 -0
  17. scipy/_lib/_fpumode.cp313t-win_arm64.lib +0 -0
  18. scipy/_lib/_fpumode.cp313t-win_arm64.pyd +0 -0
  19. scipy/_lib/_gcutils.py +105 -0
  20. scipy/_lib/_pep440.py +487 -0
  21. scipy/_lib/_sparse.py +41 -0
  22. scipy/_lib/_test_ccallback.cp313t-win_arm64.lib +0 -0
  23. scipy/_lib/_test_ccallback.cp313t-win_arm64.pyd +0 -0
  24. scipy/_lib/_test_deprecation_call.cp313t-win_arm64.lib +0 -0
  25. scipy/_lib/_test_deprecation_call.cp313t-win_arm64.pyd +0 -0
  26. scipy/_lib/_test_deprecation_def.cp313t-win_arm64.lib +0 -0
  27. scipy/_lib/_test_deprecation_def.cp313t-win_arm64.pyd +0 -0
  28. scipy/_lib/_testutils.py +373 -0
  29. scipy/_lib/_threadsafety.py +58 -0
  30. scipy/_lib/_tmpdirs.py +86 -0
  31. scipy/_lib/_uarray/LICENSE +29 -0
  32. scipy/_lib/_uarray/__init__.py +116 -0
  33. scipy/_lib/_uarray/_backend.py +707 -0
  34. scipy/_lib/_uarray/_uarray.cp313t-win_arm64.lib +0 -0
  35. scipy/_lib/_uarray/_uarray.cp313t-win_arm64.pyd +0 -0
  36. scipy/_lib/_util.py +1283 -0
  37. scipy/_lib/array_api_compat/__init__.py +22 -0
  38. scipy/_lib/array_api_compat/_internal.py +59 -0
  39. scipy/_lib/array_api_compat/common/__init__.py +1 -0
  40. scipy/_lib/array_api_compat/common/_aliases.py +727 -0
  41. scipy/_lib/array_api_compat/common/_fft.py +213 -0
  42. scipy/_lib/array_api_compat/common/_helpers.py +1058 -0
  43. scipy/_lib/array_api_compat/common/_linalg.py +232 -0
  44. scipy/_lib/array_api_compat/common/_typing.py +192 -0
  45. scipy/_lib/array_api_compat/cupy/__init__.py +13 -0
  46. scipy/_lib/array_api_compat/cupy/_aliases.py +156 -0
  47. scipy/_lib/array_api_compat/cupy/_info.py +336 -0
  48. scipy/_lib/array_api_compat/cupy/_typing.py +31 -0
  49. scipy/_lib/array_api_compat/cupy/fft.py +36 -0
  50. scipy/_lib/array_api_compat/cupy/linalg.py +49 -0
  51. scipy/_lib/array_api_compat/dask/__init__.py +0 -0
  52. scipy/_lib/array_api_compat/dask/array/__init__.py +12 -0
  53. scipy/_lib/array_api_compat/dask/array/_aliases.py +376 -0
  54. scipy/_lib/array_api_compat/dask/array/_info.py +416 -0
  55. scipy/_lib/array_api_compat/dask/array/fft.py +21 -0
  56. scipy/_lib/array_api_compat/dask/array/linalg.py +72 -0
  57. scipy/_lib/array_api_compat/numpy/__init__.py +28 -0
  58. scipy/_lib/array_api_compat/numpy/_aliases.py +190 -0
  59. scipy/_lib/array_api_compat/numpy/_info.py +366 -0
  60. scipy/_lib/array_api_compat/numpy/_typing.py +30 -0
  61. scipy/_lib/array_api_compat/numpy/fft.py +35 -0
  62. scipy/_lib/array_api_compat/numpy/linalg.py +143 -0
  63. scipy/_lib/array_api_compat/torch/__init__.py +22 -0
  64. scipy/_lib/array_api_compat/torch/_aliases.py +855 -0
  65. scipy/_lib/array_api_compat/torch/_info.py +369 -0
  66. scipy/_lib/array_api_compat/torch/_typing.py +3 -0
  67. scipy/_lib/array_api_compat/torch/fft.py +85 -0
  68. scipy/_lib/array_api_compat/torch/linalg.py +121 -0
  69. scipy/_lib/array_api_extra/__init__.py +38 -0
  70. scipy/_lib/array_api_extra/_delegation.py +171 -0
  71. scipy/_lib/array_api_extra/_lib/__init__.py +1 -0
  72. scipy/_lib/array_api_extra/_lib/_at.py +463 -0
  73. scipy/_lib/array_api_extra/_lib/_backends.py +46 -0
  74. scipy/_lib/array_api_extra/_lib/_funcs.py +937 -0
  75. scipy/_lib/array_api_extra/_lib/_lazy.py +357 -0
  76. scipy/_lib/array_api_extra/_lib/_testing.py +278 -0
  77. scipy/_lib/array_api_extra/_lib/_utils/__init__.py +1 -0
  78. scipy/_lib/array_api_extra/_lib/_utils/_compat.py +74 -0
  79. scipy/_lib/array_api_extra/_lib/_utils/_compat.pyi +45 -0
  80. scipy/_lib/array_api_extra/_lib/_utils/_helpers.py +559 -0
  81. scipy/_lib/array_api_extra/_lib/_utils/_typing.py +10 -0
  82. scipy/_lib/array_api_extra/_lib/_utils/_typing.pyi +105 -0
  83. scipy/_lib/array_api_extra/testing.py +359 -0
  84. scipy/_lib/cobyqa/__init__.py +20 -0
  85. scipy/_lib/cobyqa/framework.py +1240 -0
  86. scipy/_lib/cobyqa/main.py +1506 -0
  87. scipy/_lib/cobyqa/models.py +1529 -0
  88. scipy/_lib/cobyqa/problem.py +1296 -0
  89. scipy/_lib/cobyqa/settings.py +132 -0
  90. scipy/_lib/cobyqa/subsolvers/__init__.py +14 -0
  91. scipy/_lib/cobyqa/subsolvers/geometry.py +387 -0
  92. scipy/_lib/cobyqa/subsolvers/optim.py +1203 -0
  93. scipy/_lib/cobyqa/utils/__init__.py +18 -0
  94. scipy/_lib/cobyqa/utils/exceptions.py +22 -0
  95. scipy/_lib/cobyqa/utils/math.py +77 -0
  96. scipy/_lib/cobyqa/utils/versions.py +67 -0
  97. scipy/_lib/decorator.py +399 -0
  98. scipy/_lib/deprecation.py +274 -0
  99. scipy/_lib/doccer.py +366 -0
  100. scipy/_lib/messagestream.cp313t-win_arm64.lib +0 -0
  101. scipy/_lib/messagestream.cp313t-win_arm64.pyd +0 -0
  102. scipy/_lib/pyprima/__init__.py +212 -0
  103. scipy/_lib/pyprima/cobyla/__init__.py +0 -0
  104. scipy/_lib/pyprima/cobyla/cobyla.py +559 -0
  105. scipy/_lib/pyprima/cobyla/cobylb.py +714 -0
  106. scipy/_lib/pyprima/cobyla/geometry.py +226 -0
  107. scipy/_lib/pyprima/cobyla/initialize.py +215 -0
  108. scipy/_lib/pyprima/cobyla/trustregion.py +492 -0
  109. scipy/_lib/pyprima/cobyla/update.py +289 -0
  110. scipy/_lib/pyprima/common/__init__.py +0 -0
  111. scipy/_lib/pyprima/common/_bounds.py +34 -0
  112. scipy/_lib/pyprima/common/_linear_constraints.py +46 -0
  113. scipy/_lib/pyprima/common/_nonlinear_constraints.py +54 -0
  114. scipy/_lib/pyprima/common/_project.py +173 -0
  115. scipy/_lib/pyprima/common/checkbreak.py +93 -0
  116. scipy/_lib/pyprima/common/consts.py +47 -0
  117. scipy/_lib/pyprima/common/evaluate.py +99 -0
  118. scipy/_lib/pyprima/common/history.py +38 -0
  119. scipy/_lib/pyprima/common/infos.py +30 -0
  120. scipy/_lib/pyprima/common/linalg.py +435 -0
  121. scipy/_lib/pyprima/common/message.py +290 -0
  122. scipy/_lib/pyprima/common/powalg.py +131 -0
  123. scipy/_lib/pyprima/common/preproc.py +277 -0
  124. scipy/_lib/pyprima/common/present.py +5 -0
  125. scipy/_lib/pyprima/common/ratio.py +54 -0
  126. scipy/_lib/pyprima/common/redrho.py +47 -0
  127. scipy/_lib/pyprima/common/selectx.py +296 -0
  128. scipy/_lib/tests/__init__.py +0 -0
  129. scipy/_lib/tests/test__gcutils.py +110 -0
  130. scipy/_lib/tests/test__pep440.py +67 -0
  131. scipy/_lib/tests/test__testutils.py +32 -0
  132. scipy/_lib/tests/test__threadsafety.py +51 -0
  133. scipy/_lib/tests/test__util.py +641 -0
  134. scipy/_lib/tests/test_array_api.py +322 -0
  135. scipy/_lib/tests/test_bunch.py +169 -0
  136. scipy/_lib/tests/test_ccallback.py +196 -0
  137. scipy/_lib/tests/test_config.py +45 -0
  138. scipy/_lib/tests/test_deprecation.py +10 -0
  139. scipy/_lib/tests/test_doccer.py +143 -0
  140. scipy/_lib/tests/test_import_cycles.py +18 -0
  141. scipy/_lib/tests/test_public_api.py +482 -0
  142. scipy/_lib/tests/test_scipy_version.py +28 -0
  143. scipy/_lib/tests/test_tmpdirs.py +48 -0
  144. scipy/_lib/tests/test_warnings.py +137 -0
  145. scipy/_lib/uarray.py +31 -0
  146. scipy/cluster/__init__.py +31 -0
  147. scipy/cluster/_hierarchy.cp313t-win_arm64.lib +0 -0
  148. scipy/cluster/_hierarchy.cp313t-win_arm64.pyd +0 -0
  149. scipy/cluster/_optimal_leaf_ordering.cp313t-win_arm64.lib +0 -0
  150. scipy/cluster/_optimal_leaf_ordering.cp313t-win_arm64.pyd +0 -0
  151. scipy/cluster/_vq.cp313t-win_arm64.lib +0 -0
  152. scipy/cluster/_vq.cp313t-win_arm64.pyd +0 -0
  153. scipy/cluster/hierarchy.py +4348 -0
  154. scipy/cluster/tests/__init__.py +0 -0
  155. scipy/cluster/tests/hierarchy_test_data.py +145 -0
  156. scipy/cluster/tests/test_disjoint_set.py +202 -0
  157. scipy/cluster/tests/test_hierarchy.py +1238 -0
  158. scipy/cluster/tests/test_vq.py +434 -0
  159. scipy/cluster/vq.py +832 -0
  160. scipy/conftest.py +683 -0
  161. scipy/constants/__init__.py +358 -0
  162. scipy/constants/_codata.py +2266 -0
  163. scipy/constants/_constants.py +369 -0
  164. scipy/constants/codata.py +21 -0
  165. scipy/constants/constants.py +53 -0
  166. scipy/constants/tests/__init__.py +0 -0
  167. scipy/constants/tests/test_codata.py +78 -0
  168. scipy/constants/tests/test_constants.py +83 -0
  169. scipy/datasets/__init__.py +90 -0
  170. scipy/datasets/_download_all.py +71 -0
  171. scipy/datasets/_fetchers.py +225 -0
  172. scipy/datasets/_registry.py +26 -0
  173. scipy/datasets/_utils.py +81 -0
  174. scipy/datasets/tests/__init__.py +0 -0
  175. scipy/datasets/tests/test_data.py +128 -0
  176. scipy/differentiate/__init__.py +27 -0
  177. scipy/differentiate/_differentiate.py +1129 -0
  178. scipy/differentiate/tests/__init__.py +0 -0
  179. scipy/differentiate/tests/test_differentiate.py +694 -0
  180. scipy/fft/__init__.py +114 -0
  181. scipy/fft/_backend.py +196 -0
  182. scipy/fft/_basic.py +1650 -0
  183. scipy/fft/_basic_backend.py +197 -0
  184. scipy/fft/_debug_backends.py +22 -0
  185. scipy/fft/_fftlog.py +223 -0
  186. scipy/fft/_fftlog_backend.py +200 -0
  187. scipy/fft/_helper.py +348 -0
  188. scipy/fft/_pocketfft/LICENSE.md +25 -0
  189. scipy/fft/_pocketfft/__init__.py +9 -0
  190. scipy/fft/_pocketfft/basic.py +251 -0
  191. scipy/fft/_pocketfft/helper.py +249 -0
  192. scipy/fft/_pocketfft/pypocketfft.cp313t-win_arm64.lib +0 -0
  193. scipy/fft/_pocketfft/pypocketfft.cp313t-win_arm64.pyd +0 -0
  194. scipy/fft/_pocketfft/realtransforms.py +109 -0
  195. scipy/fft/_pocketfft/tests/__init__.py +0 -0
  196. scipy/fft/_pocketfft/tests/test_basic.py +1011 -0
  197. scipy/fft/_pocketfft/tests/test_real_transforms.py +505 -0
  198. scipy/fft/_realtransforms.py +706 -0
  199. scipy/fft/_realtransforms_backend.py +63 -0
  200. scipy/fft/tests/__init__.py +0 -0
  201. scipy/fft/tests/mock_backend.py +96 -0
  202. scipy/fft/tests/test_backend.py +98 -0
  203. scipy/fft/tests/test_basic.py +504 -0
  204. scipy/fft/tests/test_fftlog.py +215 -0
  205. scipy/fft/tests/test_helper.py +558 -0
  206. scipy/fft/tests/test_multithreading.py +84 -0
  207. scipy/fft/tests/test_real_transforms.py +247 -0
  208. scipy/fftpack/__init__.py +103 -0
  209. scipy/fftpack/_basic.py +428 -0
  210. scipy/fftpack/_helper.py +115 -0
  211. scipy/fftpack/_pseudo_diffs.py +554 -0
  212. scipy/fftpack/_realtransforms.py +598 -0
  213. scipy/fftpack/basic.py +20 -0
  214. scipy/fftpack/convolve.cp313t-win_arm64.lib +0 -0
  215. scipy/fftpack/convolve.cp313t-win_arm64.pyd +0 -0
  216. scipy/fftpack/helper.py +19 -0
  217. scipy/fftpack/pseudo_diffs.py +22 -0
  218. scipy/fftpack/realtransforms.py +19 -0
  219. scipy/fftpack/tests/__init__.py +0 -0
  220. scipy/fftpack/tests/fftw_double_ref.npz +0 -0
  221. scipy/fftpack/tests/fftw_longdouble_ref.npz +0 -0
  222. scipy/fftpack/tests/fftw_single_ref.npz +0 -0
  223. scipy/fftpack/tests/test.npz +0 -0
  224. scipy/fftpack/tests/test_basic.py +877 -0
  225. scipy/fftpack/tests/test_helper.py +54 -0
  226. scipy/fftpack/tests/test_import.py +33 -0
  227. scipy/fftpack/tests/test_pseudo_diffs.py +388 -0
  228. scipy/fftpack/tests/test_real_transforms.py +836 -0
  229. scipy/integrate/__init__.py +122 -0
  230. scipy/integrate/_bvp.py +1160 -0
  231. scipy/integrate/_cubature.py +729 -0
  232. scipy/integrate/_dop.cp313t-win_arm64.lib +0 -0
  233. scipy/integrate/_dop.cp313t-win_arm64.pyd +0 -0
  234. scipy/integrate/_ivp/__init__.py +8 -0
  235. scipy/integrate/_ivp/base.py +290 -0
  236. scipy/integrate/_ivp/bdf.py +478 -0
  237. scipy/integrate/_ivp/common.py +451 -0
  238. scipy/integrate/_ivp/dop853_coefficients.py +193 -0
  239. scipy/integrate/_ivp/ivp.py +755 -0
  240. scipy/integrate/_ivp/lsoda.py +224 -0
  241. scipy/integrate/_ivp/radau.py +572 -0
  242. scipy/integrate/_ivp/rk.py +601 -0
  243. scipy/integrate/_ivp/tests/__init__.py +0 -0
  244. scipy/integrate/_ivp/tests/test_ivp.py +1287 -0
  245. scipy/integrate/_ivp/tests/test_rk.py +37 -0
  246. scipy/integrate/_lebedev.py +5450 -0
  247. scipy/integrate/_lsoda.cp313t-win_arm64.lib +0 -0
  248. scipy/integrate/_lsoda.cp313t-win_arm64.pyd +0 -0
  249. scipy/integrate/_ode.py +1395 -0
  250. scipy/integrate/_odepack.cp313t-win_arm64.lib +0 -0
  251. scipy/integrate/_odepack.cp313t-win_arm64.pyd +0 -0
  252. scipy/integrate/_odepack_py.py +273 -0
  253. scipy/integrate/_quad_vec.py +674 -0
  254. scipy/integrate/_quadpack.cp313t-win_arm64.lib +0 -0
  255. scipy/integrate/_quadpack.cp313t-win_arm64.pyd +0 -0
  256. scipy/integrate/_quadpack_py.py +1283 -0
  257. scipy/integrate/_quadrature.py +1336 -0
  258. scipy/integrate/_rules/__init__.py +12 -0
  259. scipy/integrate/_rules/_base.py +518 -0
  260. scipy/integrate/_rules/_gauss_kronrod.py +202 -0
  261. scipy/integrate/_rules/_gauss_legendre.py +62 -0
  262. scipy/integrate/_rules/_genz_malik.py +210 -0
  263. scipy/integrate/_tanhsinh.py +1385 -0
  264. scipy/integrate/_test_multivariate.cp313t-win_arm64.lib +0 -0
  265. scipy/integrate/_test_multivariate.cp313t-win_arm64.pyd +0 -0
  266. scipy/integrate/_test_odeint_banded.cp313t-win_arm64.lib +0 -0
  267. scipy/integrate/_test_odeint_banded.cp313t-win_arm64.pyd +0 -0
  268. scipy/integrate/_vode.cp313t-win_arm64.lib +0 -0
  269. scipy/integrate/_vode.cp313t-win_arm64.pyd +0 -0
  270. scipy/integrate/dop.py +15 -0
  271. scipy/integrate/lsoda.py +15 -0
  272. scipy/integrate/odepack.py +17 -0
  273. scipy/integrate/quadpack.py +23 -0
  274. scipy/integrate/tests/__init__.py +0 -0
  275. scipy/integrate/tests/test__quad_vec.py +211 -0
  276. scipy/integrate/tests/test_banded_ode_solvers.py +305 -0
  277. scipy/integrate/tests/test_bvp.py +714 -0
  278. scipy/integrate/tests/test_cubature.py +1375 -0
  279. scipy/integrate/tests/test_integrate.py +840 -0
  280. scipy/integrate/tests/test_odeint_jac.py +74 -0
  281. scipy/integrate/tests/test_quadpack.py +680 -0
  282. scipy/integrate/tests/test_quadrature.py +730 -0
  283. scipy/integrate/tests/test_tanhsinh.py +1171 -0
  284. scipy/integrate/vode.py +15 -0
  285. scipy/interpolate/__init__.py +228 -0
  286. scipy/interpolate/_bary_rational.py +715 -0
  287. scipy/interpolate/_bsplines.py +2469 -0
  288. scipy/interpolate/_cubic.py +973 -0
  289. scipy/interpolate/_dfitpack.cp313t-win_arm64.lib +0 -0
  290. scipy/interpolate/_dfitpack.cp313t-win_arm64.pyd +0 -0
  291. scipy/interpolate/_dierckx.cp313t-win_arm64.lib +0 -0
  292. scipy/interpolate/_dierckx.cp313t-win_arm64.pyd +0 -0
  293. scipy/interpolate/_fitpack.cp313t-win_arm64.lib +0 -0
  294. scipy/interpolate/_fitpack.cp313t-win_arm64.pyd +0 -0
  295. scipy/interpolate/_fitpack2.py +2397 -0
  296. scipy/interpolate/_fitpack_impl.py +811 -0
  297. scipy/interpolate/_fitpack_py.py +898 -0
  298. scipy/interpolate/_fitpack_repro.py +996 -0
  299. scipy/interpolate/_interpnd.cp313t-win_arm64.lib +0 -0
  300. scipy/interpolate/_interpnd.cp313t-win_arm64.pyd +0 -0
  301. scipy/interpolate/_interpolate.py +2266 -0
  302. scipy/interpolate/_ndbspline.py +415 -0
  303. scipy/interpolate/_ndgriddata.py +329 -0
  304. scipy/interpolate/_pade.py +67 -0
  305. scipy/interpolate/_polyint.py +1025 -0
  306. scipy/interpolate/_ppoly.cp313t-win_arm64.lib +0 -0
  307. scipy/interpolate/_ppoly.cp313t-win_arm64.pyd +0 -0
  308. scipy/interpolate/_rbf.py +290 -0
  309. scipy/interpolate/_rbfinterp.py +550 -0
  310. scipy/interpolate/_rbfinterp_pythran.cp313t-win_arm64.lib +0 -0
  311. scipy/interpolate/_rbfinterp_pythran.cp313t-win_arm64.pyd +0 -0
  312. scipy/interpolate/_rgi.py +764 -0
  313. scipy/interpolate/_rgi_cython.cp313t-win_arm64.lib +0 -0
  314. scipy/interpolate/_rgi_cython.cp313t-win_arm64.pyd +0 -0
  315. scipy/interpolate/dfitpack.py +24 -0
  316. scipy/interpolate/fitpack.py +31 -0
  317. scipy/interpolate/fitpack2.py +29 -0
  318. scipy/interpolate/interpnd.py +24 -0
  319. scipy/interpolate/interpolate.py +30 -0
  320. scipy/interpolate/ndgriddata.py +23 -0
  321. scipy/interpolate/polyint.py +24 -0
  322. scipy/interpolate/rbf.py +18 -0
  323. scipy/interpolate/tests/__init__.py +0 -0
  324. scipy/interpolate/tests/data/bug-1310.npz +0 -0
  325. scipy/interpolate/tests/data/estimate_gradients_hang.npy +0 -0
  326. scipy/interpolate/tests/data/gcvspl.npz +0 -0
  327. scipy/interpolate/tests/test_bary_rational.py +368 -0
  328. scipy/interpolate/tests/test_bsplines.py +3754 -0
  329. scipy/interpolate/tests/test_fitpack.py +519 -0
  330. scipy/interpolate/tests/test_fitpack2.py +1431 -0
  331. scipy/interpolate/tests/test_gil.py +64 -0
  332. scipy/interpolate/tests/test_interpnd.py +452 -0
  333. scipy/interpolate/tests/test_interpolate.py +2630 -0
  334. scipy/interpolate/tests/test_ndgriddata.py +308 -0
  335. scipy/interpolate/tests/test_pade.py +107 -0
  336. scipy/interpolate/tests/test_polyint.py +972 -0
  337. scipy/interpolate/tests/test_rbf.py +246 -0
  338. scipy/interpolate/tests/test_rbfinterp.py +534 -0
  339. scipy/interpolate/tests/test_rgi.py +1151 -0
  340. scipy/io/__init__.py +116 -0
  341. scipy/io/_fast_matrix_market/__init__.py +600 -0
  342. scipy/io/_fast_matrix_market/_fmm_core.cp313t-win_arm64.lib +0 -0
  343. scipy/io/_fast_matrix_market/_fmm_core.cp313t-win_arm64.pyd +0 -0
  344. scipy/io/_fortran.py +354 -0
  345. scipy/io/_harwell_boeing/__init__.py +7 -0
  346. scipy/io/_harwell_boeing/_fortran_format_parser.py +316 -0
  347. scipy/io/_harwell_boeing/hb.py +571 -0
  348. scipy/io/_harwell_boeing/tests/__init__.py +0 -0
  349. scipy/io/_harwell_boeing/tests/test_fortran_format.py +74 -0
  350. scipy/io/_harwell_boeing/tests/test_hb.py +70 -0
  351. scipy/io/_idl.py +917 -0
  352. scipy/io/_mmio.py +968 -0
  353. scipy/io/_netcdf.py +1104 -0
  354. scipy/io/_test_fortran.cp313t-win_arm64.lib +0 -0
  355. scipy/io/_test_fortran.cp313t-win_arm64.pyd +0 -0
  356. scipy/io/arff/__init__.py +28 -0
  357. scipy/io/arff/_arffread.py +873 -0
  358. scipy/io/arff/arffread.py +19 -0
  359. scipy/io/arff/tests/__init__.py +0 -0
  360. scipy/io/arff/tests/data/iris.arff +225 -0
  361. scipy/io/arff/tests/data/missing.arff +8 -0
  362. scipy/io/arff/tests/data/nodata.arff +11 -0
  363. scipy/io/arff/tests/data/quoted_nominal.arff +13 -0
  364. scipy/io/arff/tests/data/quoted_nominal_spaces.arff +13 -0
  365. scipy/io/arff/tests/data/test1.arff +10 -0
  366. scipy/io/arff/tests/data/test10.arff +8 -0
  367. scipy/io/arff/tests/data/test11.arff +11 -0
  368. scipy/io/arff/tests/data/test2.arff +15 -0
  369. scipy/io/arff/tests/data/test3.arff +6 -0
  370. scipy/io/arff/tests/data/test4.arff +11 -0
  371. scipy/io/arff/tests/data/test5.arff +26 -0
  372. scipy/io/arff/tests/data/test6.arff +12 -0
  373. scipy/io/arff/tests/data/test7.arff +15 -0
  374. scipy/io/arff/tests/data/test8.arff +12 -0
  375. scipy/io/arff/tests/data/test9.arff +14 -0
  376. scipy/io/arff/tests/test_arffread.py +421 -0
  377. scipy/io/harwell_boeing.py +17 -0
  378. scipy/io/idl.py +17 -0
  379. scipy/io/matlab/__init__.py +66 -0
  380. scipy/io/matlab/_byteordercodes.py +75 -0
  381. scipy/io/matlab/_mio.py +375 -0
  382. scipy/io/matlab/_mio4.py +632 -0
  383. scipy/io/matlab/_mio5.py +901 -0
  384. scipy/io/matlab/_mio5_params.py +281 -0
  385. scipy/io/matlab/_mio5_utils.cp313t-win_arm64.lib +0 -0
  386. scipy/io/matlab/_mio5_utils.cp313t-win_arm64.pyd +0 -0
  387. scipy/io/matlab/_mio_utils.cp313t-win_arm64.lib +0 -0
  388. scipy/io/matlab/_mio_utils.cp313t-win_arm64.pyd +0 -0
  389. scipy/io/matlab/_miobase.py +435 -0
  390. scipy/io/matlab/_streams.cp313t-win_arm64.lib +0 -0
  391. scipy/io/matlab/_streams.cp313t-win_arm64.pyd +0 -0
  392. scipy/io/matlab/byteordercodes.py +17 -0
  393. scipy/io/matlab/mio.py +16 -0
  394. scipy/io/matlab/mio4.py +17 -0
  395. scipy/io/matlab/mio5.py +19 -0
  396. scipy/io/matlab/mio5_params.py +18 -0
  397. scipy/io/matlab/mio5_utils.py +17 -0
  398. scipy/io/matlab/mio_utils.py +17 -0
  399. scipy/io/matlab/miobase.py +16 -0
  400. scipy/io/matlab/streams.py +16 -0
  401. scipy/io/matlab/tests/__init__.py +0 -0
  402. scipy/io/matlab/tests/data/bad_miuint32.mat +0 -0
  403. scipy/io/matlab/tests/data/bad_miutf8_array_name.mat +0 -0
  404. scipy/io/matlab/tests/data/big_endian.mat +0 -0
  405. scipy/io/matlab/tests/data/broken_utf8.mat +0 -0
  406. scipy/io/matlab/tests/data/corrupted_zlib_checksum.mat +0 -0
  407. scipy/io/matlab/tests/data/corrupted_zlib_data.mat +0 -0
  408. scipy/io/matlab/tests/data/debigged_m4.mat +0 -0
  409. scipy/io/matlab/tests/data/japanese_utf8.txt +5 -0
  410. scipy/io/matlab/tests/data/little_endian.mat +0 -0
  411. scipy/io/matlab/tests/data/logical_sparse.mat +0 -0
  412. scipy/io/matlab/tests/data/malformed1.mat +0 -0
  413. scipy/io/matlab/tests/data/miuint32_for_miint32.mat +0 -0
  414. scipy/io/matlab/tests/data/miutf8_array_name.mat +0 -0
  415. scipy/io/matlab/tests/data/nasty_duplicate_fieldnames.mat +0 -0
  416. scipy/io/matlab/tests/data/one_by_zero_char.mat +0 -0
  417. scipy/io/matlab/tests/data/parabola.mat +0 -0
  418. scipy/io/matlab/tests/data/single_empty_string.mat +0 -0
  419. scipy/io/matlab/tests/data/some_functions.mat +0 -0
  420. scipy/io/matlab/tests/data/sqr.mat +0 -0
  421. scipy/io/matlab/tests/data/test3dmatrix_6.1_SOL2.mat +0 -0
  422. scipy/io/matlab/tests/data/test3dmatrix_6.5.1_GLNX86.mat +0 -0
  423. scipy/io/matlab/tests/data/test3dmatrix_7.1_GLNX86.mat +0 -0
  424. scipy/io/matlab/tests/data/test3dmatrix_7.4_GLNX86.mat +0 -0
  425. scipy/io/matlab/tests/data/test_empty_struct.mat +0 -0
  426. scipy/io/matlab/tests/data/test_mat4_le_floats.mat +0 -0
  427. scipy/io/matlab/tests/data/test_skip_variable.mat +0 -0
  428. scipy/io/matlab/tests/data/testbool_8_WIN64.mat +0 -0
  429. scipy/io/matlab/tests/data/testcell_6.1_SOL2.mat +0 -0
  430. scipy/io/matlab/tests/data/testcell_6.5.1_GLNX86.mat +0 -0
  431. scipy/io/matlab/tests/data/testcell_7.1_GLNX86.mat +0 -0
  432. scipy/io/matlab/tests/data/testcell_7.4_GLNX86.mat +0 -0
  433. scipy/io/matlab/tests/data/testcellnest_6.1_SOL2.mat +0 -0
  434. scipy/io/matlab/tests/data/testcellnest_6.5.1_GLNX86.mat +0 -0
  435. scipy/io/matlab/tests/data/testcellnest_7.1_GLNX86.mat +0 -0
  436. scipy/io/matlab/tests/data/testcellnest_7.4_GLNX86.mat +0 -0
  437. scipy/io/matlab/tests/data/testcomplex_4.2c_SOL2.mat +0 -0
  438. scipy/io/matlab/tests/data/testcomplex_6.1_SOL2.mat +0 -0
  439. scipy/io/matlab/tests/data/testcomplex_6.5.1_GLNX86.mat +0 -0
  440. scipy/io/matlab/tests/data/testcomplex_7.1_GLNX86.mat +0 -0
  441. scipy/io/matlab/tests/data/testcomplex_7.4_GLNX86.mat +0 -0
  442. scipy/io/matlab/tests/data/testdouble_4.2c_SOL2.mat +0 -0
  443. scipy/io/matlab/tests/data/testdouble_6.1_SOL2.mat +0 -0
  444. scipy/io/matlab/tests/data/testdouble_6.5.1_GLNX86.mat +0 -0
  445. scipy/io/matlab/tests/data/testdouble_7.1_GLNX86.mat +0 -0
  446. scipy/io/matlab/tests/data/testdouble_7.4_GLNX86.mat +0 -0
  447. scipy/io/matlab/tests/data/testemptycell_5.3_SOL2.mat +0 -0
  448. scipy/io/matlab/tests/data/testemptycell_6.5.1_GLNX86.mat +0 -0
  449. scipy/io/matlab/tests/data/testemptycell_7.1_GLNX86.mat +0 -0
  450. scipy/io/matlab/tests/data/testemptycell_7.4_GLNX86.mat +0 -0
  451. scipy/io/matlab/tests/data/testfunc_7.4_GLNX86.mat +0 -0
  452. scipy/io/matlab/tests/data/testhdf5_7.4_GLNX86.mat +0 -0
  453. scipy/io/matlab/tests/data/testmatrix_4.2c_SOL2.mat +0 -0
  454. scipy/io/matlab/tests/data/testmatrix_6.1_SOL2.mat +0 -0
  455. scipy/io/matlab/tests/data/testmatrix_6.5.1_GLNX86.mat +0 -0
  456. scipy/io/matlab/tests/data/testmatrix_7.1_GLNX86.mat +0 -0
  457. scipy/io/matlab/tests/data/testmatrix_7.4_GLNX86.mat +0 -0
  458. scipy/io/matlab/tests/data/testminus_4.2c_SOL2.mat +0 -0
  459. scipy/io/matlab/tests/data/testminus_6.1_SOL2.mat +0 -0
  460. scipy/io/matlab/tests/data/testminus_6.5.1_GLNX86.mat +0 -0
  461. scipy/io/matlab/tests/data/testminus_7.1_GLNX86.mat +0 -0
  462. scipy/io/matlab/tests/data/testminus_7.4_GLNX86.mat +0 -0
  463. scipy/io/matlab/tests/data/testmulti_4.2c_SOL2.mat +0 -0
  464. scipy/io/matlab/tests/data/testmulti_7.1_GLNX86.mat +0 -0
  465. scipy/io/matlab/tests/data/testmulti_7.4_GLNX86.mat +0 -0
  466. scipy/io/matlab/tests/data/testobject_6.1_SOL2.mat +0 -0
  467. scipy/io/matlab/tests/data/testobject_6.5.1_GLNX86.mat +0 -0
  468. scipy/io/matlab/tests/data/testobject_7.1_GLNX86.mat +0 -0
  469. scipy/io/matlab/tests/data/testobject_7.4_GLNX86.mat +0 -0
  470. scipy/io/matlab/tests/data/testonechar_4.2c_SOL2.mat +0 -0
  471. scipy/io/matlab/tests/data/testonechar_6.1_SOL2.mat +0 -0
  472. scipy/io/matlab/tests/data/testonechar_6.5.1_GLNX86.mat +0 -0
  473. scipy/io/matlab/tests/data/testonechar_7.1_GLNX86.mat +0 -0
  474. scipy/io/matlab/tests/data/testonechar_7.4_GLNX86.mat +0 -0
  475. scipy/io/matlab/tests/data/testscalarcell_7.4_GLNX86.mat +0 -0
  476. scipy/io/matlab/tests/data/testsimplecell.mat +0 -0
  477. scipy/io/matlab/tests/data/testsparse_4.2c_SOL2.mat +0 -0
  478. scipy/io/matlab/tests/data/testsparse_6.1_SOL2.mat +0 -0
  479. scipy/io/matlab/tests/data/testsparse_6.5.1_GLNX86.mat +0 -0
  480. scipy/io/matlab/tests/data/testsparse_7.1_GLNX86.mat +0 -0
  481. scipy/io/matlab/tests/data/testsparse_7.4_GLNX86.mat +0 -0
  482. scipy/io/matlab/tests/data/testsparsecomplex_4.2c_SOL2.mat +0 -0
  483. scipy/io/matlab/tests/data/testsparsecomplex_6.1_SOL2.mat +0 -0
  484. scipy/io/matlab/tests/data/testsparsecomplex_6.5.1_GLNX86.mat +0 -0
  485. scipy/io/matlab/tests/data/testsparsecomplex_7.1_GLNX86.mat +0 -0
  486. scipy/io/matlab/tests/data/testsparsecomplex_7.4_GLNX86.mat +0 -0
  487. scipy/io/matlab/tests/data/testsparsefloat_7.4_GLNX86.mat +0 -0
  488. scipy/io/matlab/tests/data/teststring_4.2c_SOL2.mat +0 -0
  489. scipy/io/matlab/tests/data/teststring_6.1_SOL2.mat +0 -0
  490. scipy/io/matlab/tests/data/teststring_6.5.1_GLNX86.mat +0 -0
  491. scipy/io/matlab/tests/data/teststring_7.1_GLNX86.mat +0 -0
  492. scipy/io/matlab/tests/data/teststring_7.4_GLNX86.mat +0 -0
  493. scipy/io/matlab/tests/data/teststringarray_4.2c_SOL2.mat +0 -0
  494. scipy/io/matlab/tests/data/teststringarray_6.1_SOL2.mat +0 -0
  495. scipy/io/matlab/tests/data/teststringarray_6.5.1_GLNX86.mat +0 -0
  496. scipy/io/matlab/tests/data/teststringarray_7.1_GLNX86.mat +0 -0
  497. scipy/io/matlab/tests/data/teststringarray_7.4_GLNX86.mat +0 -0
  498. scipy/io/matlab/tests/data/teststruct_6.1_SOL2.mat +0 -0
  499. scipy/io/matlab/tests/data/teststruct_6.5.1_GLNX86.mat +0 -0
  500. scipy/io/matlab/tests/data/teststruct_7.1_GLNX86.mat +0 -0
  501. scipy/io/matlab/tests/data/teststruct_7.4_GLNX86.mat +0 -0
  502. scipy/io/matlab/tests/data/teststructarr_6.1_SOL2.mat +0 -0
  503. scipy/io/matlab/tests/data/teststructarr_6.5.1_GLNX86.mat +0 -0
  504. scipy/io/matlab/tests/data/teststructarr_7.1_GLNX86.mat +0 -0
  505. scipy/io/matlab/tests/data/teststructarr_7.4_GLNX86.mat +0 -0
  506. scipy/io/matlab/tests/data/teststructnest_6.1_SOL2.mat +0 -0
  507. scipy/io/matlab/tests/data/teststructnest_6.5.1_GLNX86.mat +0 -0
  508. scipy/io/matlab/tests/data/teststructnest_7.1_GLNX86.mat +0 -0
  509. scipy/io/matlab/tests/data/teststructnest_7.4_GLNX86.mat +0 -0
  510. scipy/io/matlab/tests/data/testunicode_7.1_GLNX86.mat +0 -0
  511. scipy/io/matlab/tests/data/testunicode_7.4_GLNX86.mat +0 -0
  512. scipy/io/matlab/tests/data/testvec_4_GLNX86.mat +0 -0
  513. scipy/io/matlab/tests/test_byteordercodes.py +29 -0
  514. scipy/io/matlab/tests/test_mio.py +1399 -0
  515. scipy/io/matlab/tests/test_mio5_utils.py +179 -0
  516. scipy/io/matlab/tests/test_mio_funcs.py +51 -0
  517. scipy/io/matlab/tests/test_mio_utils.py +45 -0
  518. scipy/io/matlab/tests/test_miobase.py +32 -0
  519. scipy/io/matlab/tests/test_pathological.py +33 -0
  520. scipy/io/matlab/tests/test_streams.py +241 -0
  521. scipy/io/mmio.py +17 -0
  522. scipy/io/netcdf.py +17 -0
  523. scipy/io/tests/__init__.py +0 -0
  524. scipy/io/tests/data/Transparent Busy.ani +0 -0
  525. scipy/io/tests/data/array_float32_1d.sav +0 -0
  526. scipy/io/tests/data/array_float32_2d.sav +0 -0
  527. scipy/io/tests/data/array_float32_3d.sav +0 -0
  528. scipy/io/tests/data/array_float32_4d.sav +0 -0
  529. scipy/io/tests/data/array_float32_5d.sav +0 -0
  530. scipy/io/tests/data/array_float32_6d.sav +0 -0
  531. scipy/io/tests/data/array_float32_7d.sav +0 -0
  532. scipy/io/tests/data/array_float32_8d.sav +0 -0
  533. scipy/io/tests/data/array_float32_pointer_1d.sav +0 -0
  534. scipy/io/tests/data/array_float32_pointer_2d.sav +0 -0
  535. scipy/io/tests/data/array_float32_pointer_3d.sav +0 -0
  536. scipy/io/tests/data/array_float32_pointer_4d.sav +0 -0
  537. scipy/io/tests/data/array_float32_pointer_5d.sav +0 -0
  538. scipy/io/tests/data/array_float32_pointer_6d.sav +0 -0
  539. scipy/io/tests/data/array_float32_pointer_7d.sav +0 -0
  540. scipy/io/tests/data/array_float32_pointer_8d.sav +0 -0
  541. scipy/io/tests/data/example_1.nc +0 -0
  542. scipy/io/tests/data/example_2.nc +0 -0
  543. scipy/io/tests/data/example_3_maskedvals.nc +0 -0
  544. scipy/io/tests/data/fortran-3x3d-2i.dat +0 -0
  545. scipy/io/tests/data/fortran-mixed.dat +0 -0
  546. scipy/io/tests/data/fortran-sf8-11x1x10.dat +0 -0
  547. scipy/io/tests/data/fortran-sf8-15x10x22.dat +0 -0
  548. scipy/io/tests/data/fortran-sf8-1x1x1.dat +0 -0
  549. scipy/io/tests/data/fortran-sf8-1x1x5.dat +0 -0
  550. scipy/io/tests/data/fortran-sf8-1x1x7.dat +0 -0
  551. scipy/io/tests/data/fortran-sf8-1x3x5.dat +0 -0
  552. scipy/io/tests/data/fortran-si4-11x1x10.dat +0 -0
  553. scipy/io/tests/data/fortran-si4-15x10x22.dat +0 -0
  554. scipy/io/tests/data/fortran-si4-1x1x1.dat +0 -0
  555. scipy/io/tests/data/fortran-si4-1x1x5.dat +0 -0
  556. scipy/io/tests/data/fortran-si4-1x1x7.dat +0 -0
  557. scipy/io/tests/data/fortran-si4-1x3x5.dat +0 -0
  558. scipy/io/tests/data/invalid_pointer.sav +0 -0
  559. scipy/io/tests/data/null_pointer.sav +0 -0
  560. scipy/io/tests/data/scalar_byte.sav +0 -0
  561. scipy/io/tests/data/scalar_byte_descr.sav +0 -0
  562. scipy/io/tests/data/scalar_complex32.sav +0 -0
  563. scipy/io/tests/data/scalar_complex64.sav +0 -0
  564. scipy/io/tests/data/scalar_float32.sav +0 -0
  565. scipy/io/tests/data/scalar_float64.sav +0 -0
  566. scipy/io/tests/data/scalar_heap_pointer.sav +0 -0
  567. scipy/io/tests/data/scalar_int16.sav +0 -0
  568. scipy/io/tests/data/scalar_int32.sav +0 -0
  569. scipy/io/tests/data/scalar_int64.sav +0 -0
  570. scipy/io/tests/data/scalar_string.sav +0 -0
  571. scipy/io/tests/data/scalar_uint16.sav +0 -0
  572. scipy/io/tests/data/scalar_uint32.sav +0 -0
  573. scipy/io/tests/data/scalar_uint64.sav +0 -0
  574. scipy/io/tests/data/struct_arrays.sav +0 -0
  575. scipy/io/tests/data/struct_arrays_byte_idl80.sav +0 -0
  576. scipy/io/tests/data/struct_arrays_replicated.sav +0 -0
  577. scipy/io/tests/data/struct_arrays_replicated_3d.sav +0 -0
  578. scipy/io/tests/data/struct_inherit.sav +0 -0
  579. scipy/io/tests/data/struct_pointer_arrays.sav +0 -0
  580. scipy/io/tests/data/struct_pointer_arrays_replicated.sav +0 -0
  581. scipy/io/tests/data/struct_pointer_arrays_replicated_3d.sav +0 -0
  582. scipy/io/tests/data/struct_pointers.sav +0 -0
  583. scipy/io/tests/data/struct_pointers_replicated.sav +0 -0
  584. scipy/io/tests/data/struct_pointers_replicated_3d.sav +0 -0
  585. scipy/io/tests/data/struct_scalars.sav +0 -0
  586. scipy/io/tests/data/struct_scalars_replicated.sav +0 -0
  587. scipy/io/tests/data/struct_scalars_replicated_3d.sav +0 -0
  588. scipy/io/tests/data/test-1234Hz-le-1ch-10S-20bit-extra.wav +0 -0
  589. scipy/io/tests/data/test-44100Hz-2ch-32bit-float-be.wav +0 -0
  590. scipy/io/tests/data/test-44100Hz-2ch-32bit-float-le.wav +0 -0
  591. scipy/io/tests/data/test-44100Hz-be-1ch-4bytes.wav +0 -0
  592. scipy/io/tests/data/test-44100Hz-le-1ch-4bytes-early-eof-no-data.wav +0 -0
  593. scipy/io/tests/data/test-44100Hz-le-1ch-4bytes-early-eof.wav +0 -0
  594. scipy/io/tests/data/test-44100Hz-le-1ch-4bytes-incomplete-chunk.wav +0 -0
  595. scipy/io/tests/data/test-44100Hz-le-1ch-4bytes-rf64.wav +0 -0
  596. scipy/io/tests/data/test-44100Hz-le-1ch-4bytes.wav +0 -0
  597. scipy/io/tests/data/test-48000Hz-2ch-64bit-float-le-wavex.wav +0 -0
  598. scipy/io/tests/data/test-8000Hz-be-3ch-5S-24bit.wav +0 -0
  599. scipy/io/tests/data/test-8000Hz-le-1ch-1byte-ulaw.wav +0 -0
  600. scipy/io/tests/data/test-8000Hz-le-2ch-1byteu.wav +0 -0
  601. scipy/io/tests/data/test-8000Hz-le-3ch-5S-24bit-inconsistent.wav +0 -0
  602. scipy/io/tests/data/test-8000Hz-le-3ch-5S-24bit-rf64.wav +0 -0
  603. scipy/io/tests/data/test-8000Hz-le-3ch-5S-24bit.wav +0 -0
  604. scipy/io/tests/data/test-8000Hz-le-3ch-5S-36bit.wav +0 -0
  605. scipy/io/tests/data/test-8000Hz-le-3ch-5S-45bit.wav +0 -0
  606. scipy/io/tests/data/test-8000Hz-le-3ch-5S-53bit.wav +0 -0
  607. scipy/io/tests/data/test-8000Hz-le-3ch-5S-64bit.wav +0 -0
  608. scipy/io/tests/data/test-8000Hz-le-4ch-9S-12bit.wav +0 -0
  609. scipy/io/tests/data/test-8000Hz-le-5ch-9S-5bit.wav +0 -0
  610. scipy/io/tests/data/various_compressed.sav +0 -0
  611. scipy/io/tests/test_fortran.py +264 -0
  612. scipy/io/tests/test_idl.py +483 -0
  613. scipy/io/tests/test_mmio.py +831 -0
  614. scipy/io/tests/test_netcdf.py +550 -0
  615. scipy/io/tests/test_paths.py +93 -0
  616. scipy/io/tests/test_wavfile.py +501 -0
  617. scipy/io/wavfile.py +938 -0
  618. scipy/linalg/__init__.pxd +1 -0
  619. scipy/linalg/__init__.py +236 -0
  620. scipy/linalg/_basic.py +2146 -0
  621. scipy/linalg/_blas_subroutines.h +164 -0
  622. scipy/linalg/_cythonized_array_utils.cp313t-win_arm64.lib +0 -0
  623. scipy/linalg/_cythonized_array_utils.cp313t-win_arm64.pyd +0 -0
  624. scipy/linalg/_cythonized_array_utils.pxd +40 -0
  625. scipy/linalg/_cythonized_array_utils.pyi +16 -0
  626. scipy/linalg/_decomp.py +1645 -0
  627. scipy/linalg/_decomp_cholesky.py +413 -0
  628. scipy/linalg/_decomp_cossin.py +236 -0
  629. scipy/linalg/_decomp_interpolative.cp313t-win_arm64.lib +0 -0
  630. scipy/linalg/_decomp_interpolative.cp313t-win_arm64.pyd +0 -0
  631. scipy/linalg/_decomp_ldl.py +356 -0
  632. scipy/linalg/_decomp_lu.py +401 -0
  633. scipy/linalg/_decomp_lu_cython.cp313t-win_arm64.lib +0 -0
  634. scipy/linalg/_decomp_lu_cython.cp313t-win_arm64.pyd +0 -0
  635. scipy/linalg/_decomp_lu_cython.pyi +6 -0
  636. scipy/linalg/_decomp_polar.py +113 -0
  637. scipy/linalg/_decomp_qr.py +494 -0
  638. scipy/linalg/_decomp_qz.py +452 -0
  639. scipy/linalg/_decomp_schur.py +336 -0
  640. scipy/linalg/_decomp_svd.py +545 -0
  641. scipy/linalg/_decomp_update.cp313t-win_arm64.lib +0 -0
  642. scipy/linalg/_decomp_update.cp313t-win_arm64.pyd +0 -0
  643. scipy/linalg/_expm_frechet.py +417 -0
  644. scipy/linalg/_fblas.cp313t-win_arm64.lib +0 -0
  645. scipy/linalg/_fblas.cp313t-win_arm64.pyd +0 -0
  646. scipy/linalg/_flapack.cp313t-win_arm64.lib +0 -0
  647. scipy/linalg/_flapack.cp313t-win_arm64.pyd +0 -0
  648. scipy/linalg/_lapack_subroutines.h +1521 -0
  649. scipy/linalg/_linalg_pythran.cp313t-win_arm64.lib +0 -0
  650. scipy/linalg/_linalg_pythran.cp313t-win_arm64.pyd +0 -0
  651. scipy/linalg/_matfuncs.py +1050 -0
  652. scipy/linalg/_matfuncs_expm.cp313t-win_arm64.lib +0 -0
  653. scipy/linalg/_matfuncs_expm.cp313t-win_arm64.pyd +0 -0
  654. scipy/linalg/_matfuncs_expm.pyi +6 -0
  655. scipy/linalg/_matfuncs_inv_ssq.py +886 -0
  656. scipy/linalg/_matfuncs_schur_sqrtm.cp313t-win_arm64.lib +0 -0
  657. scipy/linalg/_matfuncs_schur_sqrtm.cp313t-win_arm64.pyd +0 -0
  658. scipy/linalg/_matfuncs_sqrtm.py +107 -0
  659. scipy/linalg/_matfuncs_sqrtm_triu.cp313t-win_arm64.lib +0 -0
  660. scipy/linalg/_matfuncs_sqrtm_triu.cp313t-win_arm64.pyd +0 -0
  661. scipy/linalg/_misc.py +191 -0
  662. scipy/linalg/_procrustes.py +113 -0
  663. scipy/linalg/_sketches.py +189 -0
  664. scipy/linalg/_solve_toeplitz.cp313t-win_arm64.lib +0 -0
  665. scipy/linalg/_solve_toeplitz.cp313t-win_arm64.pyd +0 -0
  666. scipy/linalg/_solvers.py +862 -0
  667. scipy/linalg/_special_matrices.py +1322 -0
  668. scipy/linalg/_testutils.py +65 -0
  669. scipy/linalg/basic.py +23 -0
  670. scipy/linalg/blas.py +495 -0
  671. scipy/linalg/cython_blas.cp313t-win_arm64.lib +0 -0
  672. scipy/linalg/cython_blas.cp313t-win_arm64.pyd +0 -0
  673. scipy/linalg/cython_blas.pxd +169 -0
  674. scipy/linalg/cython_blas.pyx +1432 -0
  675. scipy/linalg/cython_lapack.cp313t-win_arm64.lib +0 -0
  676. scipy/linalg/cython_lapack.cp313t-win_arm64.pyd +0 -0
  677. scipy/linalg/cython_lapack.pxd +1528 -0
  678. scipy/linalg/cython_lapack.pyx +12045 -0
  679. scipy/linalg/decomp.py +23 -0
  680. scipy/linalg/decomp_cholesky.py +21 -0
  681. scipy/linalg/decomp_lu.py +21 -0
  682. scipy/linalg/decomp_qr.py +20 -0
  683. scipy/linalg/decomp_schur.py +21 -0
  684. scipy/linalg/decomp_svd.py +21 -0
  685. scipy/linalg/interpolative.py +989 -0
  686. scipy/linalg/lapack.py +1081 -0
  687. scipy/linalg/matfuncs.py +23 -0
  688. scipy/linalg/misc.py +21 -0
  689. scipy/linalg/special_matrices.py +22 -0
  690. scipy/linalg/tests/__init__.py +0 -0
  691. scipy/linalg/tests/_cython_examples/extending.pyx +23 -0
  692. scipy/linalg/tests/_cython_examples/meson.build +34 -0
  693. scipy/linalg/tests/data/carex_15_data.npz +0 -0
  694. scipy/linalg/tests/data/carex_18_data.npz +0 -0
  695. scipy/linalg/tests/data/carex_19_data.npz +0 -0
  696. scipy/linalg/tests/data/carex_20_data.npz +0 -0
  697. scipy/linalg/tests/data/carex_6_data.npz +0 -0
  698. scipy/linalg/tests/data/gendare_20170120_data.npz +0 -0
  699. scipy/linalg/tests/test_basic.py +2074 -0
  700. scipy/linalg/tests/test_batch.py +588 -0
  701. scipy/linalg/tests/test_blas.py +1127 -0
  702. scipy/linalg/tests/test_cython_blas.py +118 -0
  703. scipy/linalg/tests/test_cython_lapack.py +22 -0
  704. scipy/linalg/tests/test_cythonized_array_utils.py +130 -0
  705. scipy/linalg/tests/test_decomp.py +3189 -0
  706. scipy/linalg/tests/test_decomp_cholesky.py +268 -0
  707. scipy/linalg/tests/test_decomp_cossin.py +314 -0
  708. scipy/linalg/tests/test_decomp_ldl.py +137 -0
  709. scipy/linalg/tests/test_decomp_lu.py +308 -0
  710. scipy/linalg/tests/test_decomp_polar.py +110 -0
  711. scipy/linalg/tests/test_decomp_update.py +1701 -0
  712. scipy/linalg/tests/test_extending.py +46 -0
  713. scipy/linalg/tests/test_fblas.py +607 -0
  714. scipy/linalg/tests/test_interpolative.py +232 -0
  715. scipy/linalg/tests/test_lapack.py +3620 -0
  716. scipy/linalg/tests/test_matfuncs.py +1125 -0
  717. scipy/linalg/tests/test_matmul_toeplitz.py +136 -0
  718. scipy/linalg/tests/test_procrustes.py +214 -0
  719. scipy/linalg/tests/test_sketches.py +118 -0
  720. scipy/linalg/tests/test_solve_toeplitz.py +150 -0
  721. scipy/linalg/tests/test_solvers.py +844 -0
  722. scipy/linalg/tests/test_special_matrices.py +636 -0
  723. scipy/misc/__init__.py +6 -0
  724. scipy/misc/common.py +6 -0
  725. scipy/misc/doccer.py +6 -0
  726. scipy/ndimage/__init__.py +174 -0
  727. scipy/ndimage/_ctest.cp313t-win_arm64.lib +0 -0
  728. scipy/ndimage/_ctest.cp313t-win_arm64.pyd +0 -0
  729. scipy/ndimage/_cytest.cp313t-win_arm64.lib +0 -0
  730. scipy/ndimage/_cytest.cp313t-win_arm64.pyd +0 -0
  731. scipy/ndimage/_delegators.py +303 -0
  732. scipy/ndimage/_filters.py +2422 -0
  733. scipy/ndimage/_fourier.py +306 -0
  734. scipy/ndimage/_interpolation.py +1033 -0
  735. scipy/ndimage/_measurements.py +1689 -0
  736. scipy/ndimage/_morphology.py +2634 -0
  737. scipy/ndimage/_nd_image.cp313t-win_arm64.lib +0 -0
  738. scipy/ndimage/_nd_image.cp313t-win_arm64.pyd +0 -0
  739. scipy/ndimage/_ndimage_api.py +16 -0
  740. scipy/ndimage/_ni_docstrings.py +214 -0
  741. scipy/ndimage/_ni_label.cp313t-win_arm64.lib +0 -0
  742. scipy/ndimage/_ni_label.cp313t-win_arm64.pyd +0 -0
  743. scipy/ndimage/_ni_support.py +139 -0
  744. scipy/ndimage/_rank_filter_1d.cp313t-win_arm64.lib +0 -0
  745. scipy/ndimage/_rank_filter_1d.cp313t-win_arm64.pyd +0 -0
  746. scipy/ndimage/_support_alternative_backends.py +84 -0
  747. scipy/ndimage/filters.py +27 -0
  748. scipy/ndimage/fourier.py +21 -0
  749. scipy/ndimage/interpolation.py +22 -0
  750. scipy/ndimage/measurements.py +24 -0
  751. scipy/ndimage/morphology.py +27 -0
  752. scipy/ndimage/tests/__init__.py +12 -0
  753. scipy/ndimage/tests/data/label_inputs.txt +21 -0
  754. scipy/ndimage/tests/data/label_results.txt +294 -0
  755. scipy/ndimage/tests/data/label_strels.txt +42 -0
  756. scipy/ndimage/tests/dots.png +0 -0
  757. scipy/ndimage/tests/test_c_api.py +102 -0
  758. scipy/ndimage/tests/test_datatypes.py +67 -0
  759. scipy/ndimage/tests/test_filters.py +3083 -0
  760. scipy/ndimage/tests/test_fourier.py +187 -0
  761. scipy/ndimage/tests/test_interpolation.py +1491 -0
  762. scipy/ndimage/tests/test_measurements.py +1592 -0
  763. scipy/ndimage/tests/test_morphology.py +2950 -0
  764. scipy/ndimage/tests/test_ni_support.py +78 -0
  765. scipy/ndimage/tests/test_splines.py +70 -0
  766. scipy/odr/__init__.py +131 -0
  767. scipy/odr/__odrpack.cp313t-win_arm64.lib +0 -0
  768. scipy/odr/__odrpack.cp313t-win_arm64.pyd +0 -0
  769. scipy/odr/_add_newdocs.py +34 -0
  770. scipy/odr/_models.py +315 -0
  771. scipy/odr/_odrpack.py +1154 -0
  772. scipy/odr/models.py +20 -0
  773. scipy/odr/odrpack.py +21 -0
  774. scipy/odr/tests/__init__.py +0 -0
  775. scipy/odr/tests/test_odr.py +607 -0
  776. scipy/optimize/__init__.pxd +1 -0
  777. scipy/optimize/__init__.py +460 -0
  778. scipy/optimize/_basinhopping.py +741 -0
  779. scipy/optimize/_bglu_dense.cp313t-win_arm64.lib +0 -0
  780. scipy/optimize/_bglu_dense.cp313t-win_arm64.pyd +0 -0
  781. scipy/optimize/_bracket.py +706 -0
  782. scipy/optimize/_chandrupatla.py +551 -0
  783. scipy/optimize/_cobyla_py.py +297 -0
  784. scipy/optimize/_cobyqa_py.py +72 -0
  785. scipy/optimize/_constraints.py +598 -0
  786. scipy/optimize/_dcsrch.py +728 -0
  787. scipy/optimize/_differentiable_functions.py +835 -0
  788. scipy/optimize/_differentialevolution.py +1970 -0
  789. scipy/optimize/_direct.cp313t-win_arm64.lib +0 -0
  790. scipy/optimize/_direct.cp313t-win_arm64.pyd +0 -0
  791. scipy/optimize/_direct_py.py +280 -0
  792. scipy/optimize/_dual_annealing.py +732 -0
  793. scipy/optimize/_elementwise.py +798 -0
  794. scipy/optimize/_group_columns.cp313t-win_arm64.lib +0 -0
  795. scipy/optimize/_group_columns.cp313t-win_arm64.pyd +0 -0
  796. scipy/optimize/_hessian_update_strategy.py +479 -0
  797. scipy/optimize/_highspy/__init__.py +0 -0
  798. scipy/optimize/_highspy/_core.cp313t-win_arm64.lib +0 -0
  799. scipy/optimize/_highspy/_core.cp313t-win_arm64.pyd +0 -0
  800. scipy/optimize/_highspy/_highs_options.cp313t-win_arm64.lib +0 -0
  801. scipy/optimize/_highspy/_highs_options.cp313t-win_arm64.pyd +0 -0
  802. scipy/optimize/_highspy/_highs_wrapper.py +338 -0
  803. scipy/optimize/_isotonic.py +157 -0
  804. scipy/optimize/_lbfgsb.cp313t-win_arm64.lib +0 -0
  805. scipy/optimize/_lbfgsb.cp313t-win_arm64.pyd +0 -0
  806. scipy/optimize/_lbfgsb_py.py +634 -0
  807. scipy/optimize/_linesearch.py +896 -0
  808. scipy/optimize/_linprog.py +733 -0
  809. scipy/optimize/_linprog_doc.py +1434 -0
  810. scipy/optimize/_linprog_highs.py +422 -0
  811. scipy/optimize/_linprog_ip.py +1141 -0
  812. scipy/optimize/_linprog_rs.py +572 -0
  813. scipy/optimize/_linprog_simplex.py +663 -0
  814. scipy/optimize/_linprog_util.py +1521 -0
  815. scipy/optimize/_lsap.cp313t-win_arm64.lib +0 -0
  816. scipy/optimize/_lsap.cp313t-win_arm64.pyd +0 -0
  817. scipy/optimize/_lsq/__init__.py +5 -0
  818. scipy/optimize/_lsq/bvls.py +183 -0
  819. scipy/optimize/_lsq/common.py +731 -0
  820. scipy/optimize/_lsq/dogbox.py +345 -0
  821. scipy/optimize/_lsq/givens_elimination.cp313t-win_arm64.lib +0 -0
  822. scipy/optimize/_lsq/givens_elimination.cp313t-win_arm64.pyd +0 -0
  823. scipy/optimize/_lsq/least_squares.py +1044 -0
  824. scipy/optimize/_lsq/lsq_linear.py +361 -0
  825. scipy/optimize/_lsq/trf.py +587 -0
  826. scipy/optimize/_lsq/trf_linear.py +249 -0
  827. scipy/optimize/_milp.py +394 -0
  828. scipy/optimize/_minimize.py +1199 -0
  829. scipy/optimize/_minpack.cp313t-win_arm64.lib +0 -0
  830. scipy/optimize/_minpack.cp313t-win_arm64.pyd +0 -0
  831. scipy/optimize/_minpack_py.py +1178 -0
  832. scipy/optimize/_moduleTNC.cp313t-win_arm64.lib +0 -0
  833. scipy/optimize/_moduleTNC.cp313t-win_arm64.pyd +0 -0
  834. scipy/optimize/_nnls.py +96 -0
  835. scipy/optimize/_nonlin.py +1634 -0
  836. scipy/optimize/_numdiff.py +963 -0
  837. scipy/optimize/_optimize.py +4169 -0
  838. scipy/optimize/_pava_pybind.cp313t-win_arm64.lib +0 -0
  839. scipy/optimize/_pava_pybind.cp313t-win_arm64.pyd +0 -0
  840. scipy/optimize/_qap.py +760 -0
  841. scipy/optimize/_remove_redundancy.py +522 -0
  842. scipy/optimize/_root.py +732 -0
  843. scipy/optimize/_root_scalar.py +538 -0
  844. scipy/optimize/_shgo.py +1606 -0
  845. scipy/optimize/_shgo_lib/__init__.py +0 -0
  846. scipy/optimize/_shgo_lib/_complex.py +1225 -0
  847. scipy/optimize/_shgo_lib/_vertex.py +460 -0
  848. scipy/optimize/_slsqp_py.py +603 -0
  849. scipy/optimize/_slsqplib.cp313t-win_arm64.lib +0 -0
  850. scipy/optimize/_slsqplib.cp313t-win_arm64.pyd +0 -0
  851. scipy/optimize/_spectral.py +260 -0
  852. scipy/optimize/_tnc.py +438 -0
  853. scipy/optimize/_trlib/__init__.py +12 -0
  854. scipy/optimize/_trlib/_trlib.cp313t-win_arm64.lib +0 -0
  855. scipy/optimize/_trlib/_trlib.cp313t-win_arm64.pyd +0 -0
  856. scipy/optimize/_trustregion.py +318 -0
  857. scipy/optimize/_trustregion_constr/__init__.py +6 -0
  858. scipy/optimize/_trustregion_constr/canonical_constraint.py +390 -0
  859. scipy/optimize/_trustregion_constr/equality_constrained_sqp.py +231 -0
  860. scipy/optimize/_trustregion_constr/minimize_trustregion_constr.py +584 -0
  861. scipy/optimize/_trustregion_constr/projections.py +411 -0
  862. scipy/optimize/_trustregion_constr/qp_subproblem.py +637 -0
  863. scipy/optimize/_trustregion_constr/report.py +49 -0
  864. scipy/optimize/_trustregion_constr/tests/__init__.py +0 -0
  865. scipy/optimize/_trustregion_constr/tests/test_canonical_constraint.py +296 -0
  866. scipy/optimize/_trustregion_constr/tests/test_nested_minimize.py +39 -0
  867. scipy/optimize/_trustregion_constr/tests/test_projections.py +214 -0
  868. scipy/optimize/_trustregion_constr/tests/test_qp_subproblem.py +645 -0
  869. scipy/optimize/_trustregion_constr/tests/test_report.py +34 -0
  870. scipy/optimize/_trustregion_constr/tr_interior_point.py +361 -0
  871. scipy/optimize/_trustregion_dogleg.py +122 -0
  872. scipy/optimize/_trustregion_exact.py +437 -0
  873. scipy/optimize/_trustregion_krylov.py +65 -0
  874. scipy/optimize/_trustregion_ncg.py +126 -0
  875. scipy/optimize/_tstutils.py +972 -0
  876. scipy/optimize/_zeros.cp313t-win_arm64.lib +0 -0
  877. scipy/optimize/_zeros.cp313t-win_arm64.pyd +0 -0
  878. scipy/optimize/_zeros_py.py +1475 -0
  879. scipy/optimize/cobyla.py +19 -0
  880. scipy/optimize/cython_optimize/__init__.py +133 -0
  881. scipy/optimize/cython_optimize/_zeros.cp313t-win_arm64.lib +0 -0
  882. scipy/optimize/cython_optimize/_zeros.cp313t-win_arm64.pyd +0 -0
  883. scipy/optimize/cython_optimize/_zeros.pxd +33 -0
  884. scipy/optimize/cython_optimize/c_zeros.pxd +26 -0
  885. scipy/optimize/cython_optimize.pxd +11 -0
  886. scipy/optimize/elementwise.py +38 -0
  887. scipy/optimize/lbfgsb.py +23 -0
  888. scipy/optimize/linesearch.py +18 -0
  889. scipy/optimize/minpack.py +27 -0
  890. scipy/optimize/minpack2.py +17 -0
  891. scipy/optimize/moduleTNC.py +19 -0
  892. scipy/optimize/nonlin.py +29 -0
  893. scipy/optimize/optimize.py +40 -0
  894. scipy/optimize/slsqp.py +22 -0
  895. scipy/optimize/tests/__init__.py +0 -0
  896. scipy/optimize/tests/_cython_examples/extending.pyx +43 -0
  897. scipy/optimize/tests/_cython_examples/meson.build +32 -0
  898. scipy/optimize/tests/test__basinhopping.py +535 -0
  899. scipy/optimize/tests/test__differential_evolution.py +1703 -0
  900. scipy/optimize/tests/test__dual_annealing.py +416 -0
  901. scipy/optimize/tests/test__linprog_clean_inputs.py +312 -0
  902. scipy/optimize/tests/test__numdiff.py +885 -0
  903. scipy/optimize/tests/test__remove_redundancy.py +228 -0
  904. scipy/optimize/tests/test__root.py +124 -0
  905. scipy/optimize/tests/test__shgo.py +1164 -0
  906. scipy/optimize/tests/test__spectral.py +226 -0
  907. scipy/optimize/tests/test_bracket.py +896 -0
  908. scipy/optimize/tests/test_chandrupatla.py +982 -0
  909. scipy/optimize/tests/test_cobyla.py +195 -0
  910. scipy/optimize/tests/test_cobyqa.py +252 -0
  911. scipy/optimize/tests/test_constraint_conversion.py +286 -0
  912. scipy/optimize/tests/test_constraints.py +255 -0
  913. scipy/optimize/tests/test_cython_optimize.py +92 -0
  914. scipy/optimize/tests/test_differentiable_functions.py +1025 -0
  915. scipy/optimize/tests/test_direct.py +321 -0
  916. scipy/optimize/tests/test_extending.py +28 -0
  917. scipy/optimize/tests/test_hessian_update_strategy.py +300 -0
  918. scipy/optimize/tests/test_isotonic_regression.py +167 -0
  919. scipy/optimize/tests/test_lbfgsb_hessinv.py +65 -0
  920. scipy/optimize/tests/test_lbfgsb_setulb.py +122 -0
  921. scipy/optimize/tests/test_least_squares.py +986 -0
  922. scipy/optimize/tests/test_linear_assignment.py +116 -0
  923. scipy/optimize/tests/test_linesearch.py +328 -0
  924. scipy/optimize/tests/test_linprog.py +2577 -0
  925. scipy/optimize/tests/test_lsq_common.py +297 -0
  926. scipy/optimize/tests/test_lsq_linear.py +287 -0
  927. scipy/optimize/tests/test_milp.py +459 -0
  928. scipy/optimize/tests/test_minimize_constrained.py +845 -0
  929. scipy/optimize/tests/test_minpack.py +1194 -0
  930. scipy/optimize/tests/test_nnls.py +469 -0
  931. scipy/optimize/tests/test_nonlin.py +572 -0
  932. scipy/optimize/tests/test_optimize.py +3344 -0
  933. scipy/optimize/tests/test_quadratic_assignment.py +455 -0
  934. scipy/optimize/tests/test_regression.py +40 -0
  935. scipy/optimize/tests/test_slsqp.py +645 -0
  936. scipy/optimize/tests/test_tnc.py +345 -0
  937. scipy/optimize/tests/test_trustregion.py +110 -0
  938. scipy/optimize/tests/test_trustregion_exact.py +351 -0
  939. scipy/optimize/tests/test_trustregion_krylov.py +170 -0
  940. scipy/optimize/tests/test_zeros.py +998 -0
  941. scipy/optimize/tnc.py +22 -0
  942. scipy/optimize/zeros.py +26 -0
  943. scipy/signal/__init__.py +316 -0
  944. scipy/signal/_arraytools.py +264 -0
  945. scipy/signal/_czt.py +575 -0
  946. scipy/signal/_delegators.py +568 -0
  947. scipy/signal/_filter_design.py +5893 -0
  948. scipy/signal/_fir_filter_design.py +1458 -0
  949. scipy/signal/_lti_conversion.py +534 -0
  950. scipy/signal/_ltisys.py +3546 -0
  951. scipy/signal/_max_len_seq.py +139 -0
  952. scipy/signal/_max_len_seq_inner.cp313t-win_arm64.lib +0 -0
  953. scipy/signal/_max_len_seq_inner.cp313t-win_arm64.pyd +0 -0
  954. scipy/signal/_peak_finding.py +1310 -0
  955. scipy/signal/_peak_finding_utils.cp313t-win_arm64.lib +0 -0
  956. scipy/signal/_peak_finding_utils.cp313t-win_arm64.pyd +0 -0
  957. scipy/signal/_polyutils.py +172 -0
  958. scipy/signal/_savitzky_golay.py +357 -0
  959. scipy/signal/_short_time_fft.py +2228 -0
  960. scipy/signal/_signal_api.py +30 -0
  961. scipy/signal/_signaltools.py +5309 -0
  962. scipy/signal/_sigtools.cp313t-win_arm64.lib +0 -0
  963. scipy/signal/_sigtools.cp313t-win_arm64.pyd +0 -0
  964. scipy/signal/_sosfilt.cp313t-win_arm64.lib +0 -0
  965. scipy/signal/_sosfilt.cp313t-win_arm64.pyd +0 -0
  966. scipy/signal/_spectral_py.py +2471 -0
  967. scipy/signal/_spline.cp313t-win_arm64.lib +0 -0
  968. scipy/signal/_spline.cp313t-win_arm64.pyd +0 -0
  969. scipy/signal/_spline.pyi +34 -0
  970. scipy/signal/_spline_filters.py +848 -0
  971. scipy/signal/_support_alternative_backends.py +73 -0
  972. scipy/signal/_upfirdn.py +219 -0
  973. scipy/signal/_upfirdn_apply.cp313t-win_arm64.lib +0 -0
  974. scipy/signal/_upfirdn_apply.cp313t-win_arm64.pyd +0 -0
  975. scipy/signal/_waveforms.py +687 -0
  976. scipy/signal/_wavelets.py +29 -0
  977. scipy/signal/bsplines.py +21 -0
  978. scipy/signal/filter_design.py +28 -0
  979. scipy/signal/fir_filter_design.py +21 -0
  980. scipy/signal/lti_conversion.py +20 -0
  981. scipy/signal/ltisys.py +25 -0
  982. scipy/signal/signaltools.py +27 -0
  983. scipy/signal/spectral.py +21 -0
  984. scipy/signal/spline.py +18 -0
  985. scipy/signal/tests/__init__.py +0 -0
  986. scipy/signal/tests/_scipy_spectral_test_shim.py +311 -0
  987. scipy/signal/tests/mpsig.py +122 -0
  988. scipy/signal/tests/test_array_tools.py +111 -0
  989. scipy/signal/tests/test_bsplines.py +365 -0
  990. scipy/signal/tests/test_cont2discrete.py +424 -0
  991. scipy/signal/tests/test_czt.py +221 -0
  992. scipy/signal/tests/test_dltisys.py +599 -0
  993. scipy/signal/tests/test_filter_design.py +4744 -0
  994. scipy/signal/tests/test_fir_filter_design.py +851 -0
  995. scipy/signal/tests/test_ltisys.py +1225 -0
  996. scipy/signal/tests/test_max_len_seq.py +71 -0
  997. scipy/signal/tests/test_peak_finding.py +915 -0
  998. scipy/signal/tests/test_result_type.py +51 -0
  999. scipy/signal/tests/test_savitzky_golay.py +363 -0
  1000. scipy/signal/tests/test_short_time_fft.py +1107 -0
  1001. scipy/signal/tests/test_signaltools.py +4735 -0
  1002. scipy/signal/tests/test_spectral.py +2141 -0
  1003. scipy/signal/tests/test_splines.py +427 -0
  1004. scipy/signal/tests/test_upfirdn.py +322 -0
  1005. scipy/signal/tests/test_waveforms.py +400 -0
  1006. scipy/signal/tests/test_wavelets.py +59 -0
  1007. scipy/signal/tests/test_windows.py +987 -0
  1008. scipy/signal/waveforms.py +20 -0
  1009. scipy/signal/wavelets.py +17 -0
  1010. scipy/signal/windows/__init__.py +52 -0
  1011. scipy/signal/windows/_windows.py +2513 -0
  1012. scipy/signal/windows/windows.py +23 -0
  1013. scipy/sparse/__init__.py +350 -0
  1014. scipy/sparse/_base.py +1613 -0
  1015. scipy/sparse/_bsr.py +880 -0
  1016. scipy/sparse/_compressed.py +1328 -0
  1017. scipy/sparse/_construct.py +1454 -0
  1018. scipy/sparse/_coo.py +1581 -0
  1019. scipy/sparse/_csc.py +367 -0
  1020. scipy/sparse/_csparsetools.cp313t-win_arm64.lib +0 -0
  1021. scipy/sparse/_csparsetools.cp313t-win_arm64.pyd +0 -0
  1022. scipy/sparse/_csr.py +558 -0
  1023. scipy/sparse/_data.py +569 -0
  1024. scipy/sparse/_dia.py +677 -0
  1025. scipy/sparse/_dok.py +669 -0
  1026. scipy/sparse/_extract.py +178 -0
  1027. scipy/sparse/_index.py +444 -0
  1028. scipy/sparse/_lil.py +632 -0
  1029. scipy/sparse/_matrix.py +169 -0
  1030. scipy/sparse/_matrix_io.py +167 -0
  1031. scipy/sparse/_sparsetools.cp313t-win_arm64.lib +0 -0
  1032. scipy/sparse/_sparsetools.cp313t-win_arm64.pyd +0 -0
  1033. scipy/sparse/_spfuncs.py +76 -0
  1034. scipy/sparse/_sputils.py +632 -0
  1035. scipy/sparse/base.py +24 -0
  1036. scipy/sparse/bsr.py +22 -0
  1037. scipy/sparse/compressed.py +20 -0
  1038. scipy/sparse/construct.py +38 -0
  1039. scipy/sparse/coo.py +23 -0
  1040. scipy/sparse/csc.py +22 -0
  1041. scipy/sparse/csgraph/__init__.py +210 -0
  1042. scipy/sparse/csgraph/_flow.cp313t-win_arm64.lib +0 -0
  1043. scipy/sparse/csgraph/_flow.cp313t-win_arm64.pyd +0 -0
  1044. scipy/sparse/csgraph/_laplacian.py +563 -0
  1045. scipy/sparse/csgraph/_matching.cp313t-win_arm64.lib +0 -0
  1046. scipy/sparse/csgraph/_matching.cp313t-win_arm64.pyd +0 -0
  1047. scipy/sparse/csgraph/_min_spanning_tree.cp313t-win_arm64.lib +0 -0
  1048. scipy/sparse/csgraph/_min_spanning_tree.cp313t-win_arm64.pyd +0 -0
  1049. scipy/sparse/csgraph/_reordering.cp313t-win_arm64.lib +0 -0
  1050. scipy/sparse/csgraph/_reordering.cp313t-win_arm64.pyd +0 -0
  1051. scipy/sparse/csgraph/_shortest_path.cp313t-win_arm64.lib +0 -0
  1052. scipy/sparse/csgraph/_shortest_path.cp313t-win_arm64.pyd +0 -0
  1053. scipy/sparse/csgraph/_tools.cp313t-win_arm64.lib +0 -0
  1054. scipy/sparse/csgraph/_tools.cp313t-win_arm64.pyd +0 -0
  1055. scipy/sparse/csgraph/_traversal.cp313t-win_arm64.lib +0 -0
  1056. scipy/sparse/csgraph/_traversal.cp313t-win_arm64.pyd +0 -0
  1057. scipy/sparse/csgraph/_validation.py +66 -0
  1058. scipy/sparse/csgraph/tests/__init__.py +0 -0
  1059. scipy/sparse/csgraph/tests/test_connected_components.py +119 -0
  1060. scipy/sparse/csgraph/tests/test_conversions.py +61 -0
  1061. scipy/sparse/csgraph/tests/test_flow.py +209 -0
  1062. scipy/sparse/csgraph/tests/test_graph_laplacian.py +368 -0
  1063. scipy/sparse/csgraph/tests/test_matching.py +307 -0
  1064. scipy/sparse/csgraph/tests/test_pydata_sparse.py +197 -0
  1065. scipy/sparse/csgraph/tests/test_reordering.py +70 -0
  1066. scipy/sparse/csgraph/tests/test_shortest_path.py +540 -0
  1067. scipy/sparse/csgraph/tests/test_spanning_tree.py +66 -0
  1068. scipy/sparse/csgraph/tests/test_traversal.py +148 -0
  1069. scipy/sparse/csr.py +22 -0
  1070. scipy/sparse/data.py +18 -0
  1071. scipy/sparse/dia.py +22 -0
  1072. scipy/sparse/dok.py +22 -0
  1073. scipy/sparse/extract.py +23 -0
  1074. scipy/sparse/lil.py +22 -0
  1075. scipy/sparse/linalg/__init__.py +148 -0
  1076. scipy/sparse/linalg/_dsolve/__init__.py +71 -0
  1077. scipy/sparse/linalg/_dsolve/_add_newdocs.py +147 -0
  1078. scipy/sparse/linalg/_dsolve/_superlu.cp313t-win_arm64.lib +0 -0
  1079. scipy/sparse/linalg/_dsolve/_superlu.cp313t-win_arm64.pyd +0 -0
  1080. scipy/sparse/linalg/_dsolve/linsolve.py +882 -0
  1081. scipy/sparse/linalg/_dsolve/tests/__init__.py +0 -0
  1082. scipy/sparse/linalg/_dsolve/tests/test_linsolve.py +928 -0
  1083. scipy/sparse/linalg/_eigen/__init__.py +22 -0
  1084. scipy/sparse/linalg/_eigen/_svds.py +540 -0
  1085. scipy/sparse/linalg/_eigen/_svds_doc.py +382 -0
  1086. scipy/sparse/linalg/_eigen/arpack/COPYING +45 -0
  1087. scipy/sparse/linalg/_eigen/arpack/__init__.py +20 -0
  1088. scipy/sparse/linalg/_eigen/arpack/_arpack.cp313t-win_arm64.lib +0 -0
  1089. scipy/sparse/linalg/_eigen/arpack/_arpack.cp313t-win_arm64.pyd +0 -0
  1090. scipy/sparse/linalg/_eigen/arpack/arpack.py +1706 -0
  1091. scipy/sparse/linalg/_eigen/arpack/tests/__init__.py +0 -0
  1092. scipy/sparse/linalg/_eigen/arpack/tests/test_arpack.py +717 -0
  1093. scipy/sparse/linalg/_eigen/lobpcg/__init__.py +16 -0
  1094. scipy/sparse/linalg/_eigen/lobpcg/lobpcg.py +1110 -0
  1095. scipy/sparse/linalg/_eigen/lobpcg/tests/__init__.py +0 -0
  1096. scipy/sparse/linalg/_eigen/lobpcg/tests/test_lobpcg.py +725 -0
  1097. scipy/sparse/linalg/_eigen/tests/__init__.py +0 -0
  1098. scipy/sparse/linalg/_eigen/tests/test_svds.py +886 -0
  1099. scipy/sparse/linalg/_expm_multiply.py +816 -0
  1100. scipy/sparse/linalg/_interface.py +920 -0
  1101. scipy/sparse/linalg/_isolve/__init__.py +20 -0
  1102. scipy/sparse/linalg/_isolve/_gcrotmk.py +503 -0
  1103. scipy/sparse/linalg/_isolve/iterative.py +1051 -0
  1104. scipy/sparse/linalg/_isolve/lgmres.py +230 -0
  1105. scipy/sparse/linalg/_isolve/lsmr.py +486 -0
  1106. scipy/sparse/linalg/_isolve/lsqr.py +589 -0
  1107. scipy/sparse/linalg/_isolve/minres.py +372 -0
  1108. scipy/sparse/linalg/_isolve/tests/__init__.py +0 -0
  1109. scipy/sparse/linalg/_isolve/tests/test_gcrotmk.py +183 -0
  1110. scipy/sparse/linalg/_isolve/tests/test_iterative.py +809 -0
  1111. scipy/sparse/linalg/_isolve/tests/test_lgmres.py +225 -0
  1112. scipy/sparse/linalg/_isolve/tests/test_lsmr.py +185 -0
  1113. scipy/sparse/linalg/_isolve/tests/test_lsqr.py +120 -0
  1114. scipy/sparse/linalg/_isolve/tests/test_minres.py +97 -0
  1115. scipy/sparse/linalg/_isolve/tests/test_utils.py +9 -0
  1116. scipy/sparse/linalg/_isolve/tfqmr.py +179 -0
  1117. scipy/sparse/linalg/_isolve/utils.py +121 -0
  1118. scipy/sparse/linalg/_matfuncs.py +940 -0
  1119. scipy/sparse/linalg/_norm.py +195 -0
  1120. scipy/sparse/linalg/_onenormest.py +467 -0
  1121. scipy/sparse/linalg/_propack/_cpropack.cp313t-win_arm64.lib +0 -0
  1122. scipy/sparse/linalg/_propack/_cpropack.cp313t-win_arm64.pyd +0 -0
  1123. scipy/sparse/linalg/_propack/_dpropack.cp313t-win_arm64.lib +0 -0
  1124. scipy/sparse/linalg/_propack/_dpropack.cp313t-win_arm64.pyd +0 -0
  1125. scipy/sparse/linalg/_propack/_spropack.cp313t-win_arm64.lib +0 -0
  1126. scipy/sparse/linalg/_propack/_spropack.cp313t-win_arm64.pyd +0 -0
  1127. scipy/sparse/linalg/_propack/_zpropack.cp313t-win_arm64.lib +0 -0
  1128. scipy/sparse/linalg/_propack/_zpropack.cp313t-win_arm64.pyd +0 -0
  1129. scipy/sparse/linalg/_special_sparse_arrays.py +949 -0
  1130. scipy/sparse/linalg/_svdp.py +309 -0
  1131. scipy/sparse/linalg/dsolve.py +22 -0
  1132. scipy/sparse/linalg/eigen.py +21 -0
  1133. scipy/sparse/linalg/interface.py +20 -0
  1134. scipy/sparse/linalg/isolve.py +22 -0
  1135. scipy/sparse/linalg/matfuncs.py +18 -0
  1136. scipy/sparse/linalg/tests/__init__.py +0 -0
  1137. scipy/sparse/linalg/tests/propack_test_data.npz +0 -0
  1138. scipy/sparse/linalg/tests/test_expm_multiply.py +367 -0
  1139. scipy/sparse/linalg/tests/test_interface.py +561 -0
  1140. scipy/sparse/linalg/tests/test_matfuncs.py +592 -0
  1141. scipy/sparse/linalg/tests/test_norm.py +154 -0
  1142. scipy/sparse/linalg/tests/test_onenormest.py +252 -0
  1143. scipy/sparse/linalg/tests/test_propack.py +165 -0
  1144. scipy/sparse/linalg/tests/test_pydata_sparse.py +272 -0
  1145. scipy/sparse/linalg/tests/test_special_sparse_arrays.py +337 -0
  1146. scipy/sparse/sparsetools.py +17 -0
  1147. scipy/sparse/spfuncs.py +17 -0
  1148. scipy/sparse/sputils.py +17 -0
  1149. scipy/sparse/tests/__init__.py +0 -0
  1150. scipy/sparse/tests/data/csc_py2.npz +0 -0
  1151. scipy/sparse/tests/data/csc_py3.npz +0 -0
  1152. scipy/sparse/tests/test_arithmetic1d.py +341 -0
  1153. scipy/sparse/tests/test_array_api.py +561 -0
  1154. scipy/sparse/tests/test_base.py +5870 -0
  1155. scipy/sparse/tests/test_common1d.py +447 -0
  1156. scipy/sparse/tests/test_construct.py +872 -0
  1157. scipy/sparse/tests/test_coo.py +1119 -0
  1158. scipy/sparse/tests/test_csc.py +98 -0
  1159. scipy/sparse/tests/test_csr.py +214 -0
  1160. scipy/sparse/tests/test_dok.py +209 -0
  1161. scipy/sparse/tests/test_extract.py +51 -0
  1162. scipy/sparse/tests/test_indexing1d.py +603 -0
  1163. scipy/sparse/tests/test_matrix_io.py +109 -0
  1164. scipy/sparse/tests/test_minmax1d.py +128 -0
  1165. scipy/sparse/tests/test_sparsetools.py +344 -0
  1166. scipy/sparse/tests/test_spfuncs.py +97 -0
  1167. scipy/sparse/tests/test_sputils.py +424 -0
  1168. scipy/spatial/__init__.py +129 -0
  1169. scipy/spatial/_ckdtree.cp313t-win_arm64.lib +0 -0
  1170. scipy/spatial/_ckdtree.cp313t-win_arm64.pyd +0 -0
  1171. scipy/spatial/_distance_pybind.cp313t-win_arm64.lib +0 -0
  1172. scipy/spatial/_distance_pybind.cp313t-win_arm64.pyd +0 -0
  1173. scipy/spatial/_distance_wrap.cp313t-win_arm64.lib +0 -0
  1174. scipy/spatial/_distance_wrap.cp313t-win_arm64.pyd +0 -0
  1175. scipy/spatial/_geometric_slerp.py +238 -0
  1176. scipy/spatial/_hausdorff.cp313t-win_arm64.lib +0 -0
  1177. scipy/spatial/_hausdorff.cp313t-win_arm64.pyd +0 -0
  1178. scipy/spatial/_kdtree.py +920 -0
  1179. scipy/spatial/_plotutils.py +274 -0
  1180. scipy/spatial/_procrustes.py +132 -0
  1181. scipy/spatial/_qhull.cp313t-win_arm64.lib +0 -0
  1182. scipy/spatial/_qhull.cp313t-win_arm64.pyd +0 -0
  1183. scipy/spatial/_qhull.pyi +213 -0
  1184. scipy/spatial/_spherical_voronoi.py +341 -0
  1185. scipy/spatial/_voronoi.cp313t-win_arm64.lib +0 -0
  1186. scipy/spatial/_voronoi.cp313t-win_arm64.pyd +0 -0
  1187. scipy/spatial/_voronoi.pyi +4 -0
  1188. scipy/spatial/ckdtree.py +18 -0
  1189. scipy/spatial/distance.py +3147 -0
  1190. scipy/spatial/distance.pyi +210 -0
  1191. scipy/spatial/kdtree.py +25 -0
  1192. scipy/spatial/qhull.py +25 -0
  1193. scipy/spatial/qhull_src/COPYING_QHULL.txt +39 -0
  1194. scipy/spatial/tests/__init__.py +0 -0
  1195. scipy/spatial/tests/data/cdist-X1.txt +10 -0
  1196. scipy/spatial/tests/data/cdist-X2.txt +20 -0
  1197. scipy/spatial/tests/data/degenerate_pointset.npz +0 -0
  1198. scipy/spatial/tests/data/iris.txt +150 -0
  1199. scipy/spatial/tests/data/pdist-boolean-inp.txt +20 -0
  1200. scipy/spatial/tests/data/pdist-chebyshev-ml-iris.txt +1 -0
  1201. scipy/spatial/tests/data/pdist-chebyshev-ml.txt +1 -0
  1202. scipy/spatial/tests/data/pdist-cityblock-ml-iris.txt +1 -0
  1203. scipy/spatial/tests/data/pdist-cityblock-ml.txt +1 -0
  1204. scipy/spatial/tests/data/pdist-correlation-ml-iris.txt +1 -0
  1205. scipy/spatial/tests/data/pdist-correlation-ml.txt +1 -0
  1206. scipy/spatial/tests/data/pdist-cosine-ml-iris.txt +1 -0
  1207. scipy/spatial/tests/data/pdist-cosine-ml.txt +1 -0
  1208. scipy/spatial/tests/data/pdist-double-inp.txt +20 -0
  1209. scipy/spatial/tests/data/pdist-euclidean-ml-iris.txt +1 -0
  1210. scipy/spatial/tests/data/pdist-euclidean-ml.txt +1 -0
  1211. scipy/spatial/tests/data/pdist-hamming-ml.txt +1 -0
  1212. scipy/spatial/tests/data/pdist-jaccard-ml.txt +1 -0
  1213. scipy/spatial/tests/data/pdist-jensenshannon-ml-iris.txt +1 -0
  1214. scipy/spatial/tests/data/pdist-jensenshannon-ml.txt +1 -0
  1215. scipy/spatial/tests/data/pdist-minkowski-3.2-ml-iris.txt +1 -0
  1216. scipy/spatial/tests/data/pdist-minkowski-3.2-ml.txt +1 -0
  1217. scipy/spatial/tests/data/pdist-minkowski-5.8-ml-iris.txt +1 -0
  1218. scipy/spatial/tests/data/pdist-seuclidean-ml-iris.txt +1 -0
  1219. scipy/spatial/tests/data/pdist-seuclidean-ml.txt +1 -0
  1220. scipy/spatial/tests/data/pdist-spearman-ml.txt +1 -0
  1221. scipy/spatial/tests/data/random-bool-data.txt +100 -0
  1222. scipy/spatial/tests/data/random-double-data.txt +100 -0
  1223. scipy/spatial/tests/data/random-int-data.txt +100 -0
  1224. scipy/spatial/tests/data/random-uint-data.txt +100 -0
  1225. scipy/spatial/tests/data/selfdual-4d-polytope.txt +27 -0
  1226. scipy/spatial/tests/test__plotutils.py +91 -0
  1227. scipy/spatial/tests/test__procrustes.py +116 -0
  1228. scipy/spatial/tests/test_distance.py +2389 -0
  1229. scipy/spatial/tests/test_hausdorff.py +199 -0
  1230. scipy/spatial/tests/test_kdtree.py +1536 -0
  1231. scipy/spatial/tests/test_qhull.py +1313 -0
  1232. scipy/spatial/tests/test_slerp.py +417 -0
  1233. scipy/spatial/tests/test_spherical_voronoi.py +358 -0
  1234. scipy/spatial/transform/__init__.py +31 -0
  1235. scipy/spatial/transform/_rigid_transform.cp313t-win_arm64.lib +0 -0
  1236. scipy/spatial/transform/_rigid_transform.cp313t-win_arm64.pyd +0 -0
  1237. scipy/spatial/transform/_rotation.cp313t-win_arm64.lib +0 -0
  1238. scipy/spatial/transform/_rotation.cp313t-win_arm64.pyd +0 -0
  1239. scipy/spatial/transform/_rotation_groups.py +140 -0
  1240. scipy/spatial/transform/_rotation_spline.py +460 -0
  1241. scipy/spatial/transform/rotation.py +21 -0
  1242. scipy/spatial/transform/tests/__init__.py +0 -0
  1243. scipy/spatial/transform/tests/test_rigid_transform.py +1221 -0
  1244. scipy/spatial/transform/tests/test_rotation.py +2569 -0
  1245. scipy/spatial/transform/tests/test_rotation_groups.py +169 -0
  1246. scipy/spatial/transform/tests/test_rotation_spline.py +183 -0
  1247. scipy/special/__init__.pxd +1 -0
  1248. scipy/special/__init__.py +841 -0
  1249. scipy/special/_add_newdocs.py +9961 -0
  1250. scipy/special/_basic.py +3576 -0
  1251. scipy/special/_comb.cp313t-win_arm64.lib +0 -0
  1252. scipy/special/_comb.cp313t-win_arm64.pyd +0 -0
  1253. scipy/special/_ellip_harm.py +214 -0
  1254. scipy/special/_ellip_harm_2.cp313t-win_arm64.lib +0 -0
  1255. scipy/special/_ellip_harm_2.cp313t-win_arm64.pyd +0 -0
  1256. scipy/special/_gufuncs.cp313t-win_arm64.lib +0 -0
  1257. scipy/special/_gufuncs.cp313t-win_arm64.pyd +0 -0
  1258. scipy/special/_input_validation.py +17 -0
  1259. scipy/special/_lambertw.py +149 -0
  1260. scipy/special/_logsumexp.py +426 -0
  1261. scipy/special/_mptestutils.py +453 -0
  1262. scipy/special/_multiufuncs.py +610 -0
  1263. scipy/special/_orthogonal.py +2592 -0
  1264. scipy/special/_orthogonal.pyi +330 -0
  1265. scipy/special/_precompute/__init__.py +0 -0
  1266. scipy/special/_precompute/cosine_cdf.py +17 -0
  1267. scipy/special/_precompute/expn_asy.py +54 -0
  1268. scipy/special/_precompute/gammainc_asy.py +116 -0
  1269. scipy/special/_precompute/gammainc_data.py +124 -0
  1270. scipy/special/_precompute/hyp2f1_data.py +484 -0
  1271. scipy/special/_precompute/lambertw.py +68 -0
  1272. scipy/special/_precompute/loggamma.py +43 -0
  1273. scipy/special/_precompute/struve_convergence.py +131 -0
  1274. scipy/special/_precompute/utils.py +38 -0
  1275. scipy/special/_precompute/wright_bessel.py +342 -0
  1276. scipy/special/_precompute/wright_bessel_data.py +152 -0
  1277. scipy/special/_precompute/wrightomega.py +41 -0
  1278. scipy/special/_precompute/zetac.py +27 -0
  1279. scipy/special/_sf_error.py +15 -0
  1280. scipy/special/_specfun.cp313t-win_arm64.lib +0 -0
  1281. scipy/special/_specfun.cp313t-win_arm64.pyd +0 -0
  1282. scipy/special/_special_ufuncs.cp313t-win_arm64.lib +0 -0
  1283. scipy/special/_special_ufuncs.cp313t-win_arm64.pyd +0 -0
  1284. scipy/special/_spfun_stats.py +106 -0
  1285. scipy/special/_spherical_bessel.py +397 -0
  1286. scipy/special/_support_alternative_backends.py +295 -0
  1287. scipy/special/_test_internal.cp313t-win_arm64.lib +0 -0
  1288. scipy/special/_test_internal.cp313t-win_arm64.pyd +0 -0
  1289. scipy/special/_test_internal.pyi +9 -0
  1290. scipy/special/_testutils.py +321 -0
  1291. scipy/special/_ufuncs.cp313t-win_arm64.lib +0 -0
  1292. scipy/special/_ufuncs.cp313t-win_arm64.pyd +0 -0
  1293. scipy/special/_ufuncs.pyi +522 -0
  1294. scipy/special/_ufuncs.pyx +13173 -0
  1295. scipy/special/_ufuncs_cxx.cp313t-win_arm64.lib +0 -0
  1296. scipy/special/_ufuncs_cxx.cp313t-win_arm64.pyd +0 -0
  1297. scipy/special/_ufuncs_cxx.pxd +142 -0
  1298. scipy/special/_ufuncs_cxx.pyx +427 -0
  1299. scipy/special/_ufuncs_cxx_defs.h +147 -0
  1300. scipy/special/_ufuncs_defs.h +57 -0
  1301. scipy/special/add_newdocs.py +15 -0
  1302. scipy/special/basic.py +87 -0
  1303. scipy/special/cython_special.cp313t-win_arm64.lib +0 -0
  1304. scipy/special/cython_special.cp313t-win_arm64.pyd +0 -0
  1305. scipy/special/cython_special.pxd +259 -0
  1306. scipy/special/cython_special.pyi +3 -0
  1307. scipy/special/orthogonal.py +45 -0
  1308. scipy/special/sf_error.py +20 -0
  1309. scipy/special/specfun.py +24 -0
  1310. scipy/special/spfun_stats.py +17 -0
  1311. scipy/special/tests/__init__.py +0 -0
  1312. scipy/special/tests/_cython_examples/extending.pyx +12 -0
  1313. scipy/special/tests/_cython_examples/meson.build +34 -0
  1314. scipy/special/tests/data/__init__.py +0 -0
  1315. scipy/special/tests/data/boost.npz +0 -0
  1316. scipy/special/tests/data/gsl.npz +0 -0
  1317. scipy/special/tests/data/local.npz +0 -0
  1318. scipy/special/tests/test_basic.py +4815 -0
  1319. scipy/special/tests/test_bdtr.py +112 -0
  1320. scipy/special/tests/test_boost_ufuncs.py +64 -0
  1321. scipy/special/tests/test_boxcox.py +125 -0
  1322. scipy/special/tests/test_cdflib.py +712 -0
  1323. scipy/special/tests/test_cdft_asymptotic.py +49 -0
  1324. scipy/special/tests/test_cephes_intp_cast.py +29 -0
  1325. scipy/special/tests/test_cosine_distr.py +83 -0
  1326. scipy/special/tests/test_cython_special.py +363 -0
  1327. scipy/special/tests/test_data.py +719 -0
  1328. scipy/special/tests/test_dd.py +42 -0
  1329. scipy/special/tests/test_digamma.py +45 -0
  1330. scipy/special/tests/test_ellip_harm.py +278 -0
  1331. scipy/special/tests/test_erfinv.py +89 -0
  1332. scipy/special/tests/test_exponential_integrals.py +118 -0
  1333. scipy/special/tests/test_extending.py +28 -0
  1334. scipy/special/tests/test_faddeeva.py +85 -0
  1335. scipy/special/tests/test_gamma.py +12 -0
  1336. scipy/special/tests/test_gammainc.py +152 -0
  1337. scipy/special/tests/test_hyp2f1.py +2566 -0
  1338. scipy/special/tests/test_hypergeometric.py +234 -0
  1339. scipy/special/tests/test_iv_ratio.py +249 -0
  1340. scipy/special/tests/test_kolmogorov.py +491 -0
  1341. scipy/special/tests/test_lambertw.py +109 -0
  1342. scipy/special/tests/test_legendre.py +1518 -0
  1343. scipy/special/tests/test_log1mexp.py +85 -0
  1344. scipy/special/tests/test_loggamma.py +70 -0
  1345. scipy/special/tests/test_logit.py +162 -0
  1346. scipy/special/tests/test_logsumexp.py +469 -0
  1347. scipy/special/tests/test_mpmath.py +2293 -0
  1348. scipy/special/tests/test_nan_inputs.py +65 -0
  1349. scipy/special/tests/test_ndtr.py +77 -0
  1350. scipy/special/tests/test_ndtri_exp.py +94 -0
  1351. scipy/special/tests/test_orthogonal.py +821 -0
  1352. scipy/special/tests/test_orthogonal_eval.py +275 -0
  1353. scipy/special/tests/test_owens_t.py +53 -0
  1354. scipy/special/tests/test_pcf.py +24 -0
  1355. scipy/special/tests/test_pdtr.py +48 -0
  1356. scipy/special/tests/test_powm1.py +65 -0
  1357. scipy/special/tests/test_precompute_expn_asy.py +24 -0
  1358. scipy/special/tests/test_precompute_gammainc.py +108 -0
  1359. scipy/special/tests/test_precompute_utils.py +36 -0
  1360. scipy/special/tests/test_round.py +18 -0
  1361. scipy/special/tests/test_sf_error.py +146 -0
  1362. scipy/special/tests/test_sici.py +36 -0
  1363. scipy/special/tests/test_specfun.py +48 -0
  1364. scipy/special/tests/test_spence.py +32 -0
  1365. scipy/special/tests/test_spfun_stats.py +61 -0
  1366. scipy/special/tests/test_sph_harm.py +85 -0
  1367. scipy/special/tests/test_spherical_bessel.py +400 -0
  1368. scipy/special/tests/test_support_alternative_backends.py +248 -0
  1369. scipy/special/tests/test_trig.py +72 -0
  1370. scipy/special/tests/test_ufunc_signatures.py +46 -0
  1371. scipy/special/tests/test_wright_bessel.py +205 -0
  1372. scipy/special/tests/test_wrightomega.py +117 -0
  1373. scipy/special/tests/test_zeta.py +301 -0
  1374. scipy/stats/__init__.py +670 -0
  1375. scipy/stats/_ansari_swilk_statistics.cp313t-win_arm64.lib +0 -0
  1376. scipy/stats/_ansari_swilk_statistics.cp313t-win_arm64.pyd +0 -0
  1377. scipy/stats/_axis_nan_policy.py +692 -0
  1378. scipy/stats/_biasedurn.cp313t-win_arm64.lib +0 -0
  1379. scipy/stats/_biasedurn.cp313t-win_arm64.pyd +0 -0
  1380. scipy/stats/_biasedurn.pxd +27 -0
  1381. scipy/stats/_binned_statistic.py +795 -0
  1382. scipy/stats/_binomtest.py +375 -0
  1383. scipy/stats/_bws_test.py +177 -0
  1384. scipy/stats/_censored_data.py +459 -0
  1385. scipy/stats/_common.py +5 -0
  1386. scipy/stats/_constants.py +42 -0
  1387. scipy/stats/_continued_fraction.py +387 -0
  1388. scipy/stats/_continuous_distns.py +12486 -0
  1389. scipy/stats/_correlation.py +210 -0
  1390. scipy/stats/_covariance.py +636 -0
  1391. scipy/stats/_crosstab.py +204 -0
  1392. scipy/stats/_discrete_distns.py +2098 -0
  1393. scipy/stats/_distn_infrastructure.py +4201 -0
  1394. scipy/stats/_distr_params.py +299 -0
  1395. scipy/stats/_distribution_infrastructure.py +5750 -0
  1396. scipy/stats/_entropy.py +428 -0
  1397. scipy/stats/_finite_differences.py +145 -0
  1398. scipy/stats/_fit.py +1351 -0
  1399. scipy/stats/_hypotests.py +2060 -0
  1400. scipy/stats/_kde.py +732 -0
  1401. scipy/stats/_ksstats.py +600 -0
  1402. scipy/stats/_levy_stable/__init__.py +1231 -0
  1403. scipy/stats/_levy_stable/levyst.cp313t-win_arm64.lib +0 -0
  1404. scipy/stats/_levy_stable/levyst.cp313t-win_arm64.pyd +0 -0
  1405. scipy/stats/_mannwhitneyu.py +492 -0
  1406. scipy/stats/_mgc.py +550 -0
  1407. scipy/stats/_morestats.py +4626 -0
  1408. scipy/stats/_mstats_basic.py +3658 -0
  1409. scipy/stats/_mstats_extras.py +521 -0
  1410. scipy/stats/_multicomp.py +449 -0
  1411. scipy/stats/_multivariate.py +7281 -0
  1412. scipy/stats/_new_distributions.py +452 -0
  1413. scipy/stats/_odds_ratio.py +466 -0
  1414. scipy/stats/_page_trend_test.py +486 -0
  1415. scipy/stats/_probability_distribution.py +1964 -0
  1416. scipy/stats/_qmc.py +2956 -0
  1417. scipy/stats/_qmc_cy.cp313t-win_arm64.lib +0 -0
  1418. scipy/stats/_qmc_cy.cp313t-win_arm64.pyd +0 -0
  1419. scipy/stats/_qmc_cy.pyi +54 -0
  1420. scipy/stats/_qmvnt.py +454 -0
  1421. scipy/stats/_qmvnt_cy.cp313t-win_arm64.lib +0 -0
  1422. scipy/stats/_qmvnt_cy.cp313t-win_arm64.pyd +0 -0
  1423. scipy/stats/_quantile.py +335 -0
  1424. scipy/stats/_rcont/__init__.py +4 -0
  1425. scipy/stats/_rcont/rcont.cp313t-win_arm64.lib +0 -0
  1426. scipy/stats/_rcont/rcont.cp313t-win_arm64.pyd +0 -0
  1427. scipy/stats/_relative_risk.py +263 -0
  1428. scipy/stats/_resampling.py +2352 -0
  1429. scipy/stats/_result_classes.py +40 -0
  1430. scipy/stats/_sampling.py +1314 -0
  1431. scipy/stats/_sensitivity_analysis.py +713 -0
  1432. scipy/stats/_sobol.cp313t-win_arm64.lib +0 -0
  1433. scipy/stats/_sobol.cp313t-win_arm64.pyd +0 -0
  1434. scipy/stats/_sobol.pyi +54 -0
  1435. scipy/stats/_sobol_direction_numbers.npz +0 -0
  1436. scipy/stats/_stats.cp313t-win_arm64.lib +0 -0
  1437. scipy/stats/_stats.cp313t-win_arm64.pyd +0 -0
  1438. scipy/stats/_stats.pxd +10 -0
  1439. scipy/stats/_stats_mstats_common.py +322 -0
  1440. scipy/stats/_stats_py.py +11089 -0
  1441. scipy/stats/_stats_pythran.cp313t-win_arm64.lib +0 -0
  1442. scipy/stats/_stats_pythran.cp313t-win_arm64.pyd +0 -0
  1443. scipy/stats/_survival.py +683 -0
  1444. scipy/stats/_tukeylambda_stats.py +199 -0
  1445. scipy/stats/_unuran/__init__.py +0 -0
  1446. scipy/stats/_unuran/unuran_wrapper.cp313t-win_arm64.lib +0 -0
  1447. scipy/stats/_unuran/unuran_wrapper.cp313t-win_arm64.pyd +0 -0
  1448. scipy/stats/_unuran/unuran_wrapper.pyi +179 -0
  1449. scipy/stats/_variation.py +126 -0
  1450. scipy/stats/_warnings_errors.py +38 -0
  1451. scipy/stats/_wilcoxon.py +265 -0
  1452. scipy/stats/biasedurn.py +16 -0
  1453. scipy/stats/contingency.py +521 -0
  1454. scipy/stats/distributions.py +24 -0
  1455. scipy/stats/kde.py +18 -0
  1456. scipy/stats/morestats.py +27 -0
  1457. scipy/stats/mstats.py +140 -0
  1458. scipy/stats/mstats_basic.py +42 -0
  1459. scipy/stats/mstats_extras.py +25 -0
  1460. scipy/stats/mvn.py +17 -0
  1461. scipy/stats/qmc.py +236 -0
  1462. scipy/stats/sampling.py +73 -0
  1463. scipy/stats/stats.py +41 -0
  1464. scipy/stats/tests/__init__.py +0 -0
  1465. scipy/stats/tests/common_tests.py +356 -0
  1466. scipy/stats/tests/data/_mvt.py +171 -0
  1467. scipy/stats/tests/data/fisher_exact_results_from_r.py +607 -0
  1468. scipy/stats/tests/data/jf_skew_t_gamlss_pdf_data.npy +0 -0
  1469. scipy/stats/tests/data/levy_stable/stable-Z1-cdf-sample-data.npy +0 -0
  1470. scipy/stats/tests/data/levy_stable/stable-Z1-pdf-sample-data.npy +0 -0
  1471. scipy/stats/tests/data/levy_stable/stable-loc-scale-sample-data.npy +0 -0
  1472. scipy/stats/tests/data/nist_anova/AtmWtAg.dat +108 -0
  1473. scipy/stats/tests/data/nist_anova/SiRstv.dat +85 -0
  1474. scipy/stats/tests/data/nist_anova/SmLs01.dat +249 -0
  1475. scipy/stats/tests/data/nist_anova/SmLs02.dat +1869 -0
  1476. scipy/stats/tests/data/nist_anova/SmLs03.dat +18069 -0
  1477. scipy/stats/tests/data/nist_anova/SmLs04.dat +249 -0
  1478. scipy/stats/tests/data/nist_anova/SmLs05.dat +1869 -0
  1479. scipy/stats/tests/data/nist_anova/SmLs06.dat +18069 -0
  1480. scipy/stats/tests/data/nist_anova/SmLs07.dat +249 -0
  1481. scipy/stats/tests/data/nist_anova/SmLs08.dat +1869 -0
  1482. scipy/stats/tests/data/nist_anova/SmLs09.dat +18069 -0
  1483. scipy/stats/tests/data/nist_linregress/Norris.dat +97 -0
  1484. scipy/stats/tests/data/rel_breitwigner_pdf_sample_data_ROOT.npy +0 -0
  1485. scipy/stats/tests/data/studentized_range_mpmath_ref.json +1499 -0
  1486. scipy/stats/tests/test_axis_nan_policy.py +1388 -0
  1487. scipy/stats/tests/test_binned_statistic.py +568 -0
  1488. scipy/stats/tests/test_censored_data.py +152 -0
  1489. scipy/stats/tests/test_contingency.py +294 -0
  1490. scipy/stats/tests/test_continued_fraction.py +173 -0
  1491. scipy/stats/tests/test_continuous.py +2198 -0
  1492. scipy/stats/tests/test_continuous_basic.py +1053 -0
  1493. scipy/stats/tests/test_continuous_fit_censored.py +683 -0
  1494. scipy/stats/tests/test_correlation.py +80 -0
  1495. scipy/stats/tests/test_crosstab.py +115 -0
  1496. scipy/stats/tests/test_discrete_basic.py +580 -0
  1497. scipy/stats/tests/test_discrete_distns.py +700 -0
  1498. scipy/stats/tests/test_distributions.py +10413 -0
  1499. scipy/stats/tests/test_entropy.py +322 -0
  1500. scipy/stats/tests/test_fast_gen_inversion.py +435 -0
  1501. scipy/stats/tests/test_fit.py +1090 -0
  1502. scipy/stats/tests/test_hypotests.py +1991 -0
  1503. scipy/stats/tests/test_kdeoth.py +676 -0
  1504. scipy/stats/tests/test_marray.py +289 -0
  1505. scipy/stats/tests/test_mgc.py +217 -0
  1506. scipy/stats/tests/test_morestats.py +3259 -0
  1507. scipy/stats/tests/test_mstats_basic.py +2071 -0
  1508. scipy/stats/tests/test_mstats_extras.py +172 -0
  1509. scipy/stats/tests/test_multicomp.py +405 -0
  1510. scipy/stats/tests/test_multivariate.py +4381 -0
  1511. scipy/stats/tests/test_odds_ratio.py +148 -0
  1512. scipy/stats/tests/test_qmc.py +1492 -0
  1513. scipy/stats/tests/test_quantile.py +199 -0
  1514. scipy/stats/tests/test_rank.py +345 -0
  1515. scipy/stats/tests/test_relative_risk.py +95 -0
  1516. scipy/stats/tests/test_resampling.py +2000 -0
  1517. scipy/stats/tests/test_sampling.py +1450 -0
  1518. scipy/stats/tests/test_sensitivity_analysis.py +310 -0
  1519. scipy/stats/tests/test_stats.py +9707 -0
  1520. scipy/stats/tests/test_survival.py +466 -0
  1521. scipy/stats/tests/test_tukeylambda_stats.py +85 -0
  1522. scipy/stats/tests/test_variation.py +216 -0
  1523. scipy/version.py +12 -0
  1524. scipy-1.16.2.dist-info/DELVEWHEEL +2 -0
  1525. scipy-1.16.2.dist-info/LICENSE.txt +912 -0
  1526. scipy-1.16.2.dist-info/METADATA +1061 -0
  1527. scipy-1.16.2.dist-info/RECORD +1530 -0
  1528. scipy-1.16.2.dist-info/WHEEL +4 -0
  1529. scipy.libs/msvcp140-5f1c5dd31916990d94181e07bc3afb32.dll +0 -0
  1530. scipy.libs/scipy_openblas-f3ac85b1f412f7e86514c923dc4058d1.dll +0 -0
@@ -0,0 +1,1221 @@
1
+ import pytest
2
+
3
+ import numpy as np
4
+ from scipy.spatial.transform import Rotation, RigidTransform
5
+ from scipy.spatial.transform._rigid_transform import normalize_dual_quaternion
6
+ from scipy._lib._array_api import (
7
+ is_lazy_array,
8
+ xp_vector_norm,
9
+ is_numpy,
10
+ xp_assert_close,
11
+ )
12
+ import scipy._lib.array_api_extra as xpx
13
+
14
+
15
+ pytestmark = pytest.mark.skip_xp_backends(np_only=True)
16
+
17
+
18
+ def rotation_to_xp(r: Rotation, xp):
19
+ return Rotation.from_quat(xp.asarray(r.as_quat()))
20
+
21
+
22
+ def rigid_transform_to_xp(r: RigidTransform, xp):
23
+ return RigidTransform.from_matrix(xp.asarray(r.as_matrix()))
24
+
25
+
26
+ def test_repr(xp):
27
+ actual = repr(RigidTransform.from_matrix(xp.eye(4)))
28
+ expected = """\
29
+ RigidTransform.from_matrix(array([[1., 0., 0., 0.],
30
+ [0., 1., 0., 0.],
31
+ [0., 0., 1., 0.],
32
+ [0., 0., 0., 1.]]))"""
33
+ assert actual == expected
34
+
35
+ tf = RigidTransform.from_matrix(xp.asarray(RigidTransform.identity(2).as_matrix()))
36
+ actual = repr(tf)
37
+ expected = """\
38
+ RigidTransform.from_matrix(array([[[1., 0., 0., 0.],
39
+ [0., 1., 0., 0.],
40
+ [0., 0., 1., 0.],
41
+ [0., 0., 0., 1.]],
42
+
43
+ [[1., 0., 0., 0.],
44
+ [0., 1., 0., 0.],
45
+ [0., 0., 1., 0.],
46
+ [0., 0., 0., 1.]]]))"""
47
+ assert actual == expected
48
+
49
+
50
+ def test_from_rotation(xp):
51
+ atol = 1e-12
52
+
53
+ # Test single rotation
54
+ r = Rotation.from_matrix(xp.eye(3))
55
+ tf = RigidTransform.from_rotation(r)
56
+ xp_assert_close(tf.as_matrix(), xp.eye(4), atol=atol)
57
+ assert tf.single
58
+
59
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
60
+ tf = RigidTransform.from_rotation(r)
61
+ xp_assert_close(tf.as_matrix()[:3, :3], r.as_matrix(), atol=atol)
62
+ xp_assert_close(tf.as_matrix()[:3, 3], xp.asarray([0.0, 0, 0]), atol=atol)
63
+ xp_assert_close(tf.as_matrix()[3, :], xp.asarray([0.0, 0, 0, 1]), atol=atol)
64
+ assert tf.single
65
+
66
+ # Test multiple rotations
67
+ r = Rotation.from_euler('zyx', xp.asarray([[90, 0, 0], [0, 90, 0]]), degrees=True)
68
+ tf = RigidTransform.from_rotation(r)
69
+ xp_assert_close(tf.as_matrix()[:, :3, :3], r.as_matrix(), atol=atol)
70
+ xp_assert_close(tf.as_matrix()[:, :3, 3], xp.asarray([[0.0, 0, 0], [0, 0, 0]]),
71
+ atol=atol)
72
+ xp_assert_close(tf.as_matrix()[:, 3, :], xp.asarray([[0.0, 0, 0, 1], [0, 0, 0, 1]]),
73
+ atol=atol)
74
+ assert not tf.single
75
+
76
+
77
+ def test_from_translation(xp):
78
+ # Test single translation
79
+ t = xp.asarray([1, 2, 3])
80
+ tf = RigidTransform.from_translation(t)
81
+ expected = xp.eye(4)
82
+ expected = xpx.at(expected)[..., :3, 3].set(t)
83
+ xp_assert_close(tf.as_matrix(), expected)
84
+ assert tf.single
85
+
86
+ # Test multiple translations
87
+ t = xp.asarray([[1, 2, 3], [4, 5, 6]])
88
+ tf = RigidTransform.from_translation(t)
89
+ for i in range(t.shape[0]):
90
+ expected = xp.eye(4)
91
+ expected = xpx.at(expected)[..., :3, 3].set(t[i, ...])
92
+ xp_assert_close(tf.as_matrix()[i, ...], expected)
93
+ assert not tf.single
94
+
95
+
96
+ def test_from_translation_array_like():
97
+ # Test single translation
98
+ t = [1, 2, 3]
99
+ tf = RigidTransform.from_translation(t)
100
+ tf_expected = RigidTransform.from_translation(np.array(t))
101
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix())
102
+ assert tf.single
103
+
104
+ # Test multiple translations
105
+ t = [[1, 2, 3], [4, 5, 6]]
106
+ tf = RigidTransform.from_translation(t)
107
+ tf_expected = RigidTransform.from_translation(np.array(t))
108
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix())
109
+ assert not tf.single
110
+
111
+
112
+ def test_from_matrix(xp):
113
+ atol = 1e-12
114
+
115
+ # Test single transform matrix
116
+ matrix = xp.eye(4)
117
+ matrix = xpx.at(matrix)[..., :3, 3].set(xp.asarray([1, 2, 3]))
118
+ tf = RigidTransform.from_matrix(matrix)
119
+ xp_assert_close(tf.as_matrix(), matrix, atol=atol)
120
+ assert tf.single
121
+
122
+ # Test multiple transform matrices
123
+ matrices = xp.repeat(xp.eye(4)[None, ...], 2, axis=0)
124
+ matrices = xpx.at(matrices)[0, :3, 3].set(xp.asarray([1, 2, 3]))
125
+ matrices = xpx.at(matrices)[1, :3, 3].set(xp.asarray([4, 5, 6]))
126
+ tf = RigidTransform.from_matrix(matrices)
127
+ xp_assert_close(tf.as_matrix(), matrices, atol=atol)
128
+ assert not tf.single
129
+
130
+ # Test non-1 determinant
131
+ matrix = xp.eye(4)
132
+ matrix = xpx.at(matrix)[..., :3, :3].set(xp.eye(3) * 2.0)
133
+ tf = RigidTransform.from_matrix(matrix)
134
+ xp_assert_close(tf.as_matrix(), xp.eye(4), atol=atol)
135
+
136
+ # Test non-orthogonal rotation matrix
137
+ matrix = xp.asarray([[1, 1, 0, 0],
138
+ [0, 1, 0, 0],
139
+ [0, 0, 1, 0],
140
+ [0, 0, 0, 1]])
141
+ tf = RigidTransform.from_matrix(matrix)
142
+ expected = xp.asarray([[0.894427, 0.447214, 0, 0],
143
+ [-0.447214, 0.894427, 0, 0],
144
+ [0, 0, 1, 0],
145
+ [0, 0, 0, 1]])
146
+ xp_assert_close(tf.as_matrix(), expected, atol=1e-6)
147
+
148
+ # Test invalid matrix
149
+ invalid = xp.eye(4)
150
+ invalid = xpx.at(invalid)[..., 3, 3].set(2) # Invalid last row
151
+ if is_lazy_array(invalid):
152
+ tf = RigidTransform.from_matrix(invalid)
153
+ assert xp.all(xp.isnan(tf.as_matrix()))
154
+ else:
155
+ with pytest.raises(ValueError):
156
+ RigidTransform.from_matrix(invalid)
157
+
158
+
159
+ def test_from_matrix_array_like():
160
+ # Test single transform matrix
161
+ matrix = [[1, 0, 0, 0],
162
+ [0, 1, 0, 0],
163
+ [0, 0, 1, 0],
164
+ [0, 0, 0, 1]]
165
+ expected = np.eye(4)
166
+ tf = RigidTransform.from_matrix(matrix)
167
+ xp_assert_close(tf.as_matrix(), expected)
168
+ assert tf.single
169
+
170
+ # Test multiple transform matrices
171
+ matrices = [matrix, matrix]
172
+ tf = RigidTransform.from_matrix(matrices)
173
+ for i in range(len(matrices)):
174
+ xp_assert_close(tf.as_matrix()[i, ...], expected)
175
+ assert not tf.single
176
+
177
+
178
+ def test_from_components(xp):
179
+ atol = 1e-12
180
+
181
+ # Test single rotation and translation
182
+ t = xp.asarray([1, 2, 3])
183
+ r = Rotation.from_euler("zyx", xp.asarray([90, 0, 0]), degrees=True)
184
+ tf = RigidTransform.from_components(t, r)
185
+
186
+ expected = xp.zeros((4, 4))
187
+ expected = xpx.at(expected)[..., :3, :3].set(r.as_matrix())
188
+ expected = xpx.at(expected)[..., :3, 3].set(t)
189
+ expected = xpx.at(expected)[..., 3, 3].set(1)
190
+ xp_assert_close(tf.as_matrix(), expected, atol=atol)
191
+ assert tf.single
192
+
193
+ # Test single rotation and multiple translations
194
+ t = xp.asarray([[1, 2, 3], [4, 5, 6]])
195
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
196
+ tf = RigidTransform.from_components(t, r)
197
+ assert not tf.single
198
+
199
+ for i in range(t.shape[0]):
200
+ expected = xp.zeros((4, 4))
201
+ expected = xpx.at(expected)[..., :3, :3].set(r.as_matrix())
202
+ expected = xpx.at(expected)[..., :3, 3].set(t[i, ...])
203
+ expected = xpx.at(expected)[..., 3, 3].set(1)
204
+ xp_assert_close(tf.as_matrix()[i, ...], expected, atol=atol)
205
+
206
+ # Test multiple rotations and translations
207
+ t = xp.asarray([[1, 2, 3], [4, 5, 6]])
208
+ r = Rotation.from_euler('zyx', xp.asarray([[90, 0, 0], [0, 90, 0]]), degrees=True)
209
+ tf = RigidTransform.from_components(t, r)
210
+ assert not tf.single
211
+
212
+ for i in range(t.shape[0]):
213
+ expected = xp.zeros((4, 4))
214
+ expected = xpx.at(expected)[..., :3, :3].set(r.as_matrix()[i, ...])
215
+ expected = xpx.at(expected)[..., :3, 3].set(t[i, ...])
216
+ expected = xpx.at(expected)[..., 3, 3].set(1)
217
+ xp_assert_close(tf.as_matrix()[i, ...], expected, atol=atol)
218
+
219
+
220
+ def test_from_components_array_like():
221
+ rng = np.random.default_rng(123)
222
+ # Test single rotation and translation
223
+ t = [1, 2, 3]
224
+ r = Rotation.random(rng=rng)
225
+ tf = RigidTransform.from_components(t, r)
226
+ tf_expected = RigidTransform.from_components(np.array(t), r)
227
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
228
+ assert tf.single
229
+
230
+ # Test multiple rotations and translations
231
+ t = [[1, 2, 3], [4, 5, 6]]
232
+ r = Rotation.random(len(t), rng=rng)
233
+ tf = RigidTransform.from_components(t, r)
234
+ tf_expected = RigidTransform.from_components(np.array(t), r)
235
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
236
+ assert not tf.single
237
+
238
+
239
+
240
+ def test_as_components(xp):
241
+ atol = 1e-12
242
+ n = 10
243
+ rng = np.random.default_rng(123)
244
+ t = xp.asarray(rng.normal(size=(n, 3)))
245
+ r = rotation_to_xp(Rotation.random(n, rng=rng), xp=xp)
246
+ tf = RigidTransform.from_components(t, r)
247
+ new_t, new_r = tf.as_components()
248
+ assert all(new_r.approx_equal(r, atol=atol))
249
+ xp_assert_close(new_t, t, atol=atol)
250
+
251
+
252
+ def test_from_exp_coords(xp):
253
+ # example from 3.3 of
254
+ # https://hades.mech.northwestern.edu/images/2/25/MR-v2.pdf
255
+ angle1 = np.deg2rad(30.0)
256
+ mat = xp.asarray([
257
+ [np.cos(angle1), -np.sin(angle1), 0.0, 1.0],
258
+ [np.sin(angle1), np.cos(angle1), 0.0, 2.0],
259
+ [0.0, 0.0, 1.0, 0.0],
260
+ [0.0, 0.0, 0.0, 1.0]
261
+ ])
262
+ tf1 = RigidTransform.from_matrix(mat)
263
+ angle2 = np.deg2rad(60.0)
264
+ mat = xp.asarray([
265
+ [np.cos(angle2), -np.sin(angle2), 0.0, 2.0],
266
+ [np.sin(angle2), np.cos(angle2), 0.0, 1.0],
267
+ [0.0, 0.0, 1.0, 0.0],
268
+ [0.0, 0.0, 0.0, 1.0]
269
+ ])
270
+ tf2 = RigidTransform.from_matrix(mat)
271
+ expected = tf2 * tf1.inv()
272
+ actual = RigidTransform.from_exp_coords(
273
+ np.deg2rad(30.0) * xp.asarray([0.0, 0.0, 1.0, 3.37, -3.37, 0.0]))
274
+ xp_assert_close(actual.as_matrix(), expected.as_matrix(), atol=1e-2)
275
+
276
+ # test cases generated by comparison to pytransform3d
277
+ exp_coords = xp.asarray([
278
+ [-2.01041204, -0.52983629, 0.65773501,
279
+ 0.10386614, 0.05855009, 0.54959179],
280
+ [-0.22537438, -0.24132627, -2.4747121,
281
+ -0.09158594, 1.88075832, -0.03197204]
282
+ ])
283
+ expected_matrix = xp.asarray([
284
+ [[0.76406621, 0.10504613, -0.63652819, -0.10209961],
285
+ [0.59956454, -0.47987325, 0.64050295, 0.40158789],
286
+ [-0.2381705, -0.87102639, -0.42963687, 0.19637636],
287
+ [0., 0., 0., 1.]],
288
+ [[-0.78446989, 0.61157488, 0.10287448, 1.33330055],
289
+ [-0.58017785, -0.78232107, 0.22664378, 0.52660831],
290
+ [0.21909052, 0.11810973, 0.96852952, -0.02968529],
291
+ [0., 0., 0., 1.]]
292
+ ])
293
+ xp_assert_close(
294
+ RigidTransform.from_exp_coords(exp_coords).as_matrix(),
295
+ expected_matrix, atol=1e-8)
296
+
297
+ # identity
298
+ xp_assert_close(
299
+ RigidTransform.from_exp_coords(xp.zeros(6)).as_matrix(),
300
+ xp.eye(4), atol=1e-12)
301
+
302
+ # only translation
303
+ expected_matrix = xp.asarray([
304
+ [[1.0, 0.0, 0.0, 3.0],
305
+ [0.0, 1.0, 0.0, -5.4],
306
+ [0.0, 0.0, 1.0, 100.2],
307
+ [0.0, 0.0, 0.0, 1.0]],
308
+ [[1.0, 0.0, 0.0, -3.0],
309
+ [0.0, 1.0, 0.0, 13.3],
310
+ [0.0, 0.0, 1.0, 1.3],
311
+ [0.0, 0.0, 0.0, 1.0]]
312
+ ])
313
+ actual = RigidTransform.from_exp_coords(xp.asarray([
314
+ [0.0, 0.0, 0.0, 3.0, -5.4, 100.2],
315
+ [0.0, 0.0, 0.0, -3.0, 13.3, 1.3],
316
+ ]))
317
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
318
+
319
+ # only rotation
320
+ rot = Rotation.from_euler(
321
+ 'zyx',
322
+ xp.asarray([[34, -12, 0.5],
323
+ [-102, -55, 30]]),
324
+ degrees=True)
325
+ rotvec = rot.as_rotvec()
326
+ expected_matrix = xp.repeat(xp.eye(4)[None, ...], 2, axis=0)
327
+ expected_matrix = xpx.at(expected_matrix)[..., :3, :3].set(rot.as_matrix())
328
+ actual = RigidTransform.from_exp_coords(
329
+ xp.concat((rotvec, xp.zeros((2, 3))), axis=-1))
330
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
331
+
332
+
333
+ def test_from_exp_coords_array_like():
334
+ rng = np.random.default_rng(123)
335
+ # Test single transform
336
+ t = np.array([1, 2, 3])
337
+ r = Rotation.random(rng=rng)
338
+ tf_expected = RigidTransform.from_components(t, r)
339
+ exp_coords = tf_expected.as_exp_coords().tolist()
340
+ assert isinstance(exp_coords, list)
341
+ tf = RigidTransform.from_exp_coords(exp_coords)
342
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
343
+
344
+ # Test multiple transforms
345
+ t = [[1, 2, 3], [4, 5, 6]]
346
+ r = Rotation.random(len(t), rng=rng)
347
+ tf_expected = RigidTransform.from_components(t, r)
348
+ exp_coords = tf_expected.as_exp_coords().tolist()
349
+ assert isinstance(exp_coords, list)
350
+ tf = RigidTransform.from_exp_coords(exp_coords)
351
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
352
+
353
+
354
+ def test_as_exp_coords(xp):
355
+ # identity
356
+ expected = xp.zeros(6)
357
+ actual = RigidTransform.from_exp_coords(expected).as_exp_coords()
358
+ xp_assert_close(actual, expected, atol=1e-12)
359
+
360
+ rng = np.random.default_rng(10)
361
+
362
+ # pure rotation
363
+ rot_vec = xp.asarray(rng.normal(scale=0.1, size=(1000, 3)))
364
+ tf = RigidTransform.from_rotation(Rotation.from_rotvec(rot_vec))
365
+ exp_coords = tf.as_exp_coords()
366
+ xp_assert_close(exp_coords[:, :3], rot_vec, rtol=1e-13)
367
+ expected = xp.zeros_like(rot_vec)
368
+ xp_assert_close(exp_coords[:, 3:], expected, atol=1e-16)
369
+
370
+ # pure translation
371
+ translation = xp.asarray(rng.normal(scale=100.0, size=(1000, 3)))
372
+ tf = RigidTransform.from_translation(translation)
373
+ exp_coords = tf.as_exp_coords()
374
+ xp_assert_close(exp_coords[:, :3], expected, atol=1e-16)
375
+ xp_assert_close(exp_coords[:, 3:], translation, rtol=1e-15)
376
+
377
+
378
+ def test_from_dual_quat(xp):
379
+ # identity
380
+ xp_assert_close(
381
+ RigidTransform.from_dual_quat(
382
+ xp.asarray([0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0])).as_matrix(),
383
+ xp.eye(4), atol=1e-12)
384
+ xp_assert_close(
385
+ RigidTransform.from_dual_quat(
386
+ xp.asarray([1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]),
387
+ scalar_first=True).as_matrix(),
388
+ xp.eye(4), atol=1e-12)
389
+
390
+ # only translation
391
+ actual = RigidTransform.from_dual_quat(
392
+ xp.asarray([0, 0, 0, 1, 0.25, 0.15, -0.7, 0]))
393
+ expected_matrix = xp.asarray([
394
+ [1, 0, 0, 0.5],
395
+ [0, 1, 0, 0.3],
396
+ [0, 0, 1, -1.4],
397
+ [0, 0, 0, 1]
398
+ ])
399
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
400
+ actual = RigidTransform.from_dual_quat(
401
+ xp.asarray([1, 0, 0, 0, 0, 0.25, 0.15, -0.7]), scalar_first=True)
402
+ expected_matrix = xp.asarray([
403
+ [1, 0, 0, 0.5],
404
+ [0, 1, 0, 0.3],
405
+ [0, 0, 1, -1.4],
406
+ [0, 0, 0, 1]
407
+ ])
408
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
409
+
410
+ # only rotation
411
+ actual_rot = Rotation.from_euler("xyz", xp.asarray([65, -13, 90]), degrees=True)
412
+ actual = RigidTransform.from_dual_quat(
413
+ xp.concat((actual_rot.as_quat(), xp.zeros(4)), axis=-1))
414
+ expected_matrix = xp.eye(4)
415
+ expected_matrix = xpx.at(expected_matrix)[..., :3, :3].set(actual_rot.as_matrix())
416
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
417
+
418
+ actual = RigidTransform.from_dual_quat(
419
+ xp.concat((actual_rot.as_quat(scalar_first=True), xp.zeros(4)), axis=-1),
420
+ scalar_first=True)
421
+ expected_matrix = xp.eye(4)
422
+ expected_matrix = xpx.at(expected_matrix)[..., :3, :3].set(actual_rot.as_matrix())
423
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12)
424
+
425
+ # rotation and translation
426
+ # rtol is set to 1e-7 because xp_assert_close deviates from
427
+ # np.testing.assert_allclose in that it does not automatically default to 1e-7 for
428
+ # floating point inputs.
429
+ # See https://numpy.org/doc/2.2/reference/generated/numpy.testing.assert_allclose.html
430
+ actual = RigidTransform.from_dual_quat(
431
+ xp.asarray(
432
+ [[0.0617101, -0.06483886, 0.31432811, 0.94508498,
433
+ 0.04985168, -0.26119618, 0.1691491, -0.07743254],
434
+ [0.19507259, 0.49404931, -0.06091285, 0.8450749,
435
+ 0.65049656, -0.30782513, 0.16566752, 0.04174544]]))
436
+ expected_matrix = xp.asarray(
437
+ [[[0.79398752, -0.60213598, -0.08376202, 0.24605262],
438
+ [0.58613113, 0.79477941, -0.15740392, -0.4932833],
439
+ [0.16135089, 0.07588122, 0.98397557, 0.34262676],
440
+ [0., 0., 0., 1.]],
441
+ [[0.50440981, 0.2957028, 0.81125249, 1.20934468],
442
+ [0.08979911, 0.91647262, -0.3898898, -0.70540077],
443
+ [-0.8587822, 0.26951399, 0.43572393, -0.47776265],
444
+ [0., 0., 0., 1.]]])
445
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12, rtol=1e-7)
446
+
447
+ actual = RigidTransform.from_dual_quat(
448
+ xp.asarray(
449
+ [[0.94508498, 0.0617101, -0.06483886, 0.31432811,
450
+ -0.07743254, 0.04985168, -0.26119618, 0.1691491],
451
+ [0.8450749, 0.19507259, 0.49404931, -0.06091285,
452
+ 0.04174544, 0.65049656, -0.30782513, 0.16566752]]),
453
+ scalar_first=True)
454
+ xp_assert_close(actual.as_matrix(), expected_matrix, atol=1e-12, rtol=1e-7)
455
+
456
+ # unnormalized dual quaternions
457
+
458
+ # invalid real quaternion with norm 0
459
+ actual = RigidTransform.from_dual_quat(xp.zeros(8))
460
+ xp_assert_close(actual.as_matrix(), xp.eye(4), atol=1e-12)
461
+
462
+ # real quaternion with norm != 1
463
+ unnormalized_dual_quat = xp.asarray(
464
+ [-0.2547655, 1.23506123, 0.20230088, 0.24247194, # norm 1.3
465
+ 0.38559628, 0.08184063, 0.1755943, -0.1582222] # orthogonal
466
+ )
467
+ xp_assert_close(xp_vector_norm(unnormalized_dual_quat[:4]), xp.asarray(1.3)[()],
468
+ atol=1e-12)
469
+ xp_assert_close(xp.vecdot(unnormalized_dual_quat[:4],
470
+ unnormalized_dual_quat[4:])[()],
471
+ xp.asarray(0.0)[()], atol=1e-8)
472
+
473
+ dual_quat = RigidTransform.from_dual_quat(
474
+ unnormalized_dual_quat).as_dual_quat()
475
+ xp_assert_close(xp_vector_norm(dual_quat[:4]), xp.asarray(1.0)[()], atol=1e-12)
476
+ xp_assert_close(xp.vecdot(dual_quat[:4], dual_quat[4:])[()], xp.asarray(0.0)[()],
477
+ atol=1e-12)
478
+
479
+ # real and dual quaternion are not orthogonal
480
+ unnormalized_dual_quat = xp.asarray(
481
+ [0.20824458, 0.75098079, 0.54542913, -0.30849493, # unit norm
482
+ -0.16051025, 0.10742978, 0.21277201, 0.20596935] # not orthogonal
483
+ )
484
+ xp_assert_close(xp_vector_norm(unnormalized_dual_quat[:4]), xp.asarray(1.0)[()],
485
+ atol=1e-12)
486
+ assert xp.vecdot(unnormalized_dual_quat[:4], unnormalized_dual_quat[4:]) != 0.0
487
+ dual_quat = RigidTransform.from_dual_quat(
488
+ unnormalized_dual_quat).as_dual_quat()
489
+ xp_assert_close(xp_vector_norm(dual_quat[:4]), xp.asarray(1.0)[()], atol=1e-12)
490
+ xp_assert_close(xp.vecdot(dual_quat[:4], dual_quat[4:])[()], xp.asarray(0.0)[()],
491
+ atol=1e-12)
492
+
493
+ # invalid real quaternion with norm 0, non-orthogonal dual quaternion
494
+ unnormalized_dual_quat = xp.asarray(
495
+ [0.0, 0.0, 0.0, 0.0, -0.16051025, 0.10742978, 0.21277201, 0.20596935])
496
+ assert xp.vecdot(xp.asarray([0.0, 0, 0, 1]), unnormalized_dual_quat[4:]) != 0.0
497
+ dual_quat = RigidTransform.from_dual_quat(
498
+ unnormalized_dual_quat).as_dual_quat()
499
+ xp_assert_close(dual_quat[:4], xp.asarray([0.0, 0, 0, 1]), atol=1e-12)
500
+ xp_assert_close(xp.vecdot(dual_quat[:4], dual_quat[4:])[()], xp.asarray(0.0)[()],
501
+ atol=1e-12)
502
+
503
+ # compensation for precision loss in real quaternion
504
+ rng = np.random.default_rng(1000)
505
+ t = xp.asarray(rng.normal(size=(3,)))
506
+ r = rotation_to_xp(Rotation.random(10, rng=rng), xp=xp)
507
+ random_dual_quats = RigidTransform.from_components(t, r).as_dual_quat()
508
+
509
+ # ensure that random quaternions are not normalized
510
+ random_dual_quats = xpx.at(random_dual_quats)[:, :4].add(0.01)
511
+ assert not xp.any(xpx.isclose(xp_vector_norm(random_dual_quats[:, :4], axis=1),
512
+ 1.0, atol=0.0001))
513
+ dual_quat_norm = RigidTransform.from_dual_quat(
514
+ random_dual_quats).as_dual_quat()
515
+ expected = xp.ones(dual_quat_norm.shape[0])
516
+ xp_assert_close(xp_vector_norm(dual_quat_norm[:, :4], axis=1), expected, atol=1e-12)
517
+
518
+ # compensation for precision loss in dual quaternion, results in violation
519
+ # of orthogonality constraint
520
+ t = xp.asarray(rng.normal(size=(10, 3)))
521
+ r = rotation_to_xp(Rotation.random(10, rng=rng), xp=xp)
522
+ random_dual_quats = RigidTransform.from_components(t, r).as_dual_quat()
523
+
524
+ # ensure that random quaternions are not normalized
525
+ random_dual_quats = xpx.at(random_dual_quats)[:, 4:].add(0.01)
526
+ q_norm = xp.vecdot(random_dual_quats[:, :4], random_dual_quats[:, 4:])
527
+ assert not xp.any(xpx.isclose(q_norm, 0.0, atol=0.0001))
528
+ dual_quat_norm = RigidTransform.from_dual_quat(
529
+ random_dual_quats).as_dual_quat()
530
+ expected = xp.zeros(dual_quat_norm.shape[0])
531
+ xp_assert_close(xp.vecdot(dual_quat_norm[:, :4], dual_quat_norm[:, 4:]), expected,
532
+ atol=1e-12)
533
+ xp_assert_close(random_dual_quats[:, :4], dual_quat_norm[:, :4], atol=1e-12)
534
+
535
+
536
+ def test_from_dual_quat_array_like():
537
+ rng = np.random.default_rng(123)
538
+ # Test single transform
539
+ t = np.array([1, 2, 3])
540
+ r = Rotation.random(rng=rng)
541
+ tf_expected = RigidTransform.from_components(t, r)
542
+ dual_quat = tf_expected.as_dual_quat().tolist()
543
+ assert isinstance(dual_quat, list)
544
+ tf = RigidTransform.from_dual_quat(dual_quat)
545
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
546
+
547
+ # Test multiple transforms
548
+ t = [[1, 2, 3], [4, 5, 6]]
549
+ r = Rotation.random(len(t), rng=rng)
550
+ tf_expected = RigidTransform.from_components(t, r)
551
+ dual_quat = tf_expected.as_dual_quat().tolist()
552
+ assert isinstance(dual_quat, list)
553
+ tf = RigidTransform.from_dual_quat(dual_quat)
554
+ xp_assert_close(tf.as_matrix(), tf_expected.as_matrix(), atol=1e-12)
555
+
556
+
557
+ def test_as_dual_quat(xp):
558
+ # identity
559
+ expected = xp.asarray([0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0])
560
+ actual = xp.asarray(RigidTransform.identity().as_dual_quat())
561
+ xp_assert_close(actual, expected, atol=1e-12)
562
+
563
+ expected = xp.asarray([1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0])
564
+ actual = xp.asarray(RigidTransform.identity().as_dual_quat(scalar_first=True))
565
+ xp_assert_close(actual, expected, atol=1e-12)
566
+
567
+ rng = np.random.default_rng(10)
568
+
569
+ # only rotation
570
+ for _ in range(10):
571
+ real_part = xp.asarray(Rotation.random(rng=rng).as_quat())
572
+ dual_part = xp.zeros(4)
573
+ expected = xp.concat((real_part, dual_part), axis=-1)
574
+ actual = RigidTransform.from_dual_quat(expected).as_dual_quat()
575
+ # because of double cover:
576
+ if xp.sign(expected[0]) != xp.sign(actual[0]):
577
+ actual = -actual
578
+ xp_assert_close(actual, expected, atol=1e-12)
579
+
580
+ # only translation
581
+ for _ in range(10):
582
+ tf = 0.5 * rng.normal(size=3)
583
+ expected = xp.asarray([0.0, 0, 0, 1, *tf.tolist(), 0])
584
+ actual = RigidTransform.from_dual_quat(expected).as_dual_quat()
585
+ # because of double cover:
586
+ if xp.sign(expected[0]) != xp.sign(actual[0]):
587
+ actual = -actual
588
+ xp_assert_close(actual, expected, atol=1e-12)
589
+
590
+ # rotation and translation
591
+ for _ in range(10):
592
+ t = xp.asarray(rng.normal(size=3))
593
+ r = rotation_to_xp(Rotation.random(rng=rng), xp=xp)
594
+ expected = RigidTransform.from_components(t, r).as_dual_quat()
595
+ actual = RigidTransform.from_dual_quat(expected).as_dual_quat()
596
+ # because of double cover:
597
+ if xp.sign(expected[0]) != xp.sign(actual[0]):
598
+ actual = -actual
599
+ xp_assert_close(actual, expected, atol=1e-12)
600
+
601
+
602
+ def test_from_as_internal_consistency(xp):
603
+ atol = 1e-12
604
+ n = 1000
605
+ rng = np.random.default_rng(10)
606
+ t = xp.asarray(rng.normal(size=(n, 3)))
607
+ r = rotation_to_xp(Rotation.random(n, rng=rng), xp=xp)
608
+ tf0 = RigidTransform.from_components(t, r)
609
+
610
+ tf1 = RigidTransform.from_components(*tf0.as_components())
611
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
612
+
613
+ tf1 = RigidTransform.from_components(tf0.translation, tf0.rotation)
614
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
615
+
616
+ tf1 = RigidTransform.from_exp_coords(tf0.as_exp_coords())
617
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
618
+
619
+ tf1 = RigidTransform.from_matrix(tf0.as_matrix())
620
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
621
+
622
+ tf1 = RigidTransform.from_dual_quat(tf0.as_dual_quat())
623
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
624
+
625
+ # exp_coords small rotation
626
+ tf0 = RigidTransform.from_components(
627
+ xp.asarray(rng.normal(scale=1000.0, size=(1000, 3))),
628
+ Rotation.from_rotvec(xp.asarray(rng.normal(scale=1e-10, size=(1000, 3)))))
629
+ tf1 = RigidTransform.from_exp_coords(tf0.as_exp_coords())
630
+ xp_assert_close(tf0.as_matrix(), tf1.as_matrix(), atol=atol)
631
+
632
+
633
+ def test_identity():
634
+ # We do not use xp here because identity always returns numpy arrays
635
+ atol = 1e-12
636
+
637
+ # Test single identity
638
+ tf = RigidTransform.identity()
639
+ xp_assert_close(tf.as_matrix(), np.eye(4), atol=atol)
640
+
641
+ # Test multiple identities
642
+ tf = RigidTransform.identity(5)
643
+ xp_assert_close(tf.as_matrix(), np.array([np.eye(4)] * 5), atol=atol)
644
+
645
+
646
+ def test_apply(xp):
647
+ atol = 1e-12
648
+
649
+ ## Single transform
650
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
651
+ t = xp.asarray([2, 3, 4])
652
+ tf = RigidTransform.from_components(t, r)
653
+
654
+ # Single vector, single transform
655
+ vec = xp.asarray([1, 0, 0])
656
+ expected = t + r.apply(vec)
657
+ res = tf.apply(vec)
658
+ xp_assert_close(res, expected, atol=atol)
659
+
660
+ # Multiple vectors, single transform
661
+ vecs = xp.asarray([[1, 0, 0], [0, 1, 0]])
662
+ expected = t + r.apply(vecs)
663
+ xp_assert_close(tf.apply(vecs), expected, atol=atol)
664
+
665
+ ## Multiple transforms
666
+ r = Rotation.from_euler('z', xp.asarray([90, 0]), degrees=True)
667
+ t = xp.asarray([[2, 3, 4], [5, 6, 7]])
668
+ tf = RigidTransform.from_components(t, r)
669
+
670
+ # Single vector, multiple transforms
671
+ vec = xp.asarray([1, 0, 0])
672
+ expected = t + r.apply(vec)
673
+ xp_assert_close(tf.apply(vec), expected, atol=atol)
674
+
675
+ # Multiple vectors, multiple transforms
676
+ vecs = xp.asarray([[1, 0, 0], [0, 1, 0]])
677
+ expected = t + r.apply(vecs)
678
+ xp_assert_close(tf.apply(vecs), expected, atol=atol)
679
+
680
+
681
+ def test_apply_array_like():
682
+ rng = np.random.default_rng(123)
683
+ # Single vector
684
+ t = np.array([1, 2, 3])
685
+ r = Rotation.random(rng=rng)
686
+ tf = RigidTransform.from_components(t, r)
687
+ vec = [1, 0, 0]
688
+ expected = t + r.apply(vec)
689
+ xp_assert_close(tf.apply(vec), expected, atol=1e-12)
690
+
691
+ # Multiple vectors
692
+ t = np.array([[1, 2, 3], [4, 5, 6]])
693
+ r = Rotation.random(len(t), rng=rng)
694
+ tf = RigidTransform.from_components(t, r)
695
+ vec = [[1, 0, 0], [0, 1, 0]]
696
+ expected = t + r.apply(vec)
697
+ xp_assert_close(tf.apply(vec), expected, atol=1e-12)
698
+
699
+
700
+ def test_inverse_apply(xp):
701
+ atol = 1e-12
702
+
703
+ # Test applying inverse transform
704
+ t = xp.asarray([1, 2, 3])
705
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
706
+ tf = RigidTransform.from_components(t, r)
707
+
708
+ # Test single vector
709
+ vec = xp.asarray([1, 0, 0])
710
+ expected = tf.inv().apply(vec)
711
+ xp_assert_close(tf.apply(vec, inverse=True), expected, atol=atol)
712
+
713
+ # Test multiple vectors
714
+ vecs = xp.asarray([[1, 0, 0], [0, 1, 0]])
715
+ expected = tf.inv().apply(vecs)
716
+ xp_assert_close(tf.apply(vecs, inverse=True), expected, atol=atol)
717
+
718
+
719
+ def test_rotation_alone(xp):
720
+ atol = 1e-12
721
+
722
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
723
+ tf = RigidTransform.from_rotation(r)
724
+ vec = xp.asarray([1, 0, 0])
725
+ expected = r.apply(vec)
726
+ xp_assert_close(tf.apply(vec), expected, atol=atol)
727
+
728
+
729
+ def test_translation_alone(xp):
730
+ atol = 1e-12
731
+ t = xp.asarray([1.0, 2, 3])
732
+ tf = RigidTransform.from_translation(t)
733
+ vec = xp.asarray([5.0, 6, 7])
734
+ expected = t + vec
735
+ xp_assert_close(tf.apply(vec), expected, atol=atol)
736
+
737
+
738
+ def test_composition(xp):
739
+ atol = 1e-12
740
+
741
+ # Test composing single transforms
742
+ t1 = xp.asarray([1.0, 0, 0])
743
+ r1 = Rotation.from_euler('z', xp.asarray(90), degrees=True)
744
+ tf1 = RigidTransform.from_components(t1, r1)
745
+
746
+ t2 = xp.asarray([0, 1, 0])
747
+ r2 = Rotation.from_euler('x', xp.asarray(90), degrees=True)
748
+ tf2 = RigidTransform.from_components(t2, r2)
749
+
750
+ composed = tf2 * tf1
751
+ vec = xp.asarray([1, 0, 0])
752
+ expected = tf2.apply(tf1.apply(vec))
753
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
754
+ assert composed.single
755
+
756
+ expected = t2 + r2.apply(t1 + r1.apply(vec))
757
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
758
+
759
+ # Multiple transforms with single transform
760
+ t2 = np.array([[1, 2, 3], [4, 5, 6]])
761
+ tf2 = RigidTransform.from_components(t2, r2)
762
+
763
+ composed = tf2 * tf1
764
+ expected = tf2.apply(tf1.apply(vec))
765
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
766
+ assert not composed.single
767
+
768
+ expected = t2 + r2.apply(t1 + r1.apply(vec))
769
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
770
+
771
+ # Multiple transforms with multiple transforms
772
+ t1 = xp.asarray([[1, 0, 0], [0, -1, 1]])
773
+ tf1 = RigidTransform.from_components(t1, r1)
774
+
775
+ composed = tf2 * tf1
776
+ expected = tf2.apply(tf1.apply(vec))
777
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
778
+ assert not composed.single
779
+
780
+ expected = t2 + r2.apply(t1 + r1.apply(vec))
781
+ xp_assert_close(composed.apply(vec), expected, atol=atol)
782
+
783
+
784
+ def test_pow(xp):
785
+ atol = 1e-12
786
+ num = 10
787
+ rng = np.random.default_rng(100)
788
+ t = xp.asarray(rng.normal(size=(num, 3)))
789
+ r = rotation_to_xp(Rotation.random(num, rng=rng), xp=xp)
790
+ p = RigidTransform.from_components(t, r)
791
+ p_inv = p.inv()
792
+
793
+ # Test the short-cuts and other integers
794
+ for n in [-5, -2, -1, 0, 1, 2, 5]:
795
+ q = p**n
796
+ r = rigid_transform_to_xp(RigidTransform.identity(num), xp=xp)
797
+ for _ in range(abs(n)):
798
+ if n > 0:
799
+ r = r * p
800
+ else:
801
+ r = r * p_inv
802
+ xp_assert_close(q.as_matrix(), r.as_matrix(), atol=atol)
803
+
804
+ # Test shape preservation
805
+ r = RigidTransform.from_rotation(Rotation.from_quat(xp.asarray([0, 0, 0, 1])))
806
+ assert (r**n).as_matrix().shape == (4, 4)
807
+ r = RigidTransform.from_rotation(Rotation.from_quat(xp.asarray([[0, 0, 0, 1]])))
808
+ assert (r**n).as_matrix().shape == (1, 4, 4)
809
+
810
+ # Test fractional powers
811
+ q = p**0.5
812
+ xp_assert_close((q * q).as_matrix(), p.as_matrix(), atol=atol)
813
+ q = p**-0.5
814
+ xp_assert_close((q * q).as_matrix(), p.inv().as_matrix(), atol=atol)
815
+ q = p** 1.5
816
+ xp_assert_close((q * q).as_matrix(), (p**3).as_matrix(), atol=atol)
817
+ q = p** -1.5
818
+ xp_assert_close((q * q).as_matrix(), (p**-3).as_matrix(), atol=atol)
819
+
820
+ # pow function
821
+ tf = pow(RigidTransform.from_matrix(xp.eye(4)), 2)
822
+ xp_assert_close(tf.as_matrix(), xp.eye(4), atol=atol)
823
+
824
+
825
+ def test_pow_equivalence_with_rotation(xp):
826
+ atol = 1e-12
827
+ num = 10
828
+ rng = np.random.default_rng(100)
829
+ r = rotation_to_xp(Rotation.random(num, rng=rng), xp=xp)
830
+ p = RigidTransform.from_rotation(r)
831
+ for n in [-5, -2, -1.5, -1, -0.5, 0.0, 0.5, 1, 1.5, 2, 5]:
832
+ xp_assert_close((p**n).rotation.as_matrix(), (r**n).as_matrix(), atol=atol)
833
+
834
+
835
+ def test_inverse(xp):
836
+ atol = 1e-12
837
+
838
+ # Test inverse transform
839
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
840
+ t = xp.asarray([1, 2, 3])
841
+ tf = RigidTransform.from_components(t, r)
842
+
843
+ # Test that tf * tf.inv() equals identity
844
+ tf_inv = tf.inv()
845
+ composed = tf * tf_inv
846
+ xp_assert_close(composed.as_matrix(), xp.eye(4), atol=atol)
847
+
848
+ n = 10
849
+ rng = np.random.default_rng(1000)
850
+ t = xp.asarray(rng.normal(size=(n, 3)))
851
+ r = rotation_to_xp(Rotation.random(n, rng=rng), xp=xp)
852
+ tf = RigidTransform.from_components(t, r)
853
+ tf_inv = tf.inv()
854
+ composed = tf * tf_inv
855
+ expected = xp.repeat(xp.eye(4)[None, ...], n, axis=0)
856
+ xp_assert_close(composed.as_matrix(), expected, atol=atol)
857
+
858
+ # Test multiple transforms
859
+ r = Rotation.from_euler('zyx', xp.asarray([[90, 0, 0], [0, 90, 0]]), degrees=True)
860
+ t = xp.asarray([[1, 2, 3], [4, 5, 6]])
861
+ tf = RigidTransform.from_components(t, r)
862
+ tf_inv = tf.inv()
863
+ composed = tf * tf_inv
864
+ expected = xp.repeat(xp.eye(4)[None, ...], 2, axis=0)
865
+ xp_assert_close(composed.as_matrix(), expected, atol=atol)
866
+
867
+
868
+ def test_properties(xp):
869
+ atol = 1e-12
870
+
871
+ # Test rotation and translation properties for single transform
872
+ r = Rotation.from_euler('z', xp.asarray(90), degrees=True)
873
+ t = xp.asarray([1.0, 2, 3])
874
+ tf = RigidTransform.from_components(t, r)
875
+
876
+ xp_assert_close(tf.rotation.as_matrix(), r.as_matrix(), atol=atol)
877
+ assert tf.rotation.approx_equal(r)
878
+ xp_assert_close(tf.translation, t, atol=atol)
879
+
880
+ # Test rotation and translation properties for multiple transforms
881
+ r = Rotation.from_euler('zyx', xp.asarray([[90, 0, 0], [0, 90, 0]]), degrees=True)
882
+ t = xp.asarray([[1.0, 2, 3], [4, 5, 6]])
883
+ tf = RigidTransform.from_components(t, r)
884
+
885
+ xp_assert_close(tf.rotation.as_matrix(), r.as_matrix(), atol=atol)
886
+ assert all(tf.rotation.approx_equal(r))
887
+ xp_assert_close(tf.translation, t, atol=atol)
888
+
889
+
890
+ def test_indexing(xp):
891
+ atol = 1e-12
892
+
893
+ # Test indexing for multiple transforms
894
+ r = Rotation.from_euler('zyx', xp.asarray([[90, 0, 0], [0, 90, 0]]), degrees=True)
895
+ t = xp.asarray([[1.0, 2, 3], [4, 5, 6]])
896
+ tf = RigidTransform.from_components(t, r)
897
+
898
+ # Test single index
899
+ xp_assert_close(tf[0].as_matrix()[:3, :3], r[0].as_matrix(), atol=atol)
900
+ xp_assert_close(tf[0].as_matrix()[:3, 3], t[0, ...], atol=atol)
901
+
902
+ # Test slice
903
+ tf_slice = tf[0:2]
904
+ xp_assert_close(tf_slice.as_matrix()[:, :3, :3], r[0:2].as_matrix(), atol=atol)
905
+ xp_assert_close(tf_slice.as_matrix()[:, :3, 3], t[0:2, ...], atol=atol)
906
+
907
+ # Test boolean indexing
908
+ tf_masked = tf[xp.asarray([True, True])]
909
+ xp_assert_close(tf_masked.as_matrix()[:, :3, :3], r.as_matrix(), atol=atol)
910
+ xp_assert_close(tf_masked.as_matrix()[:, :3, 3], t, atol=atol)
911
+
912
+ tf_masked = tf[xp.asarray([False, True])]
913
+ xp_assert_close(tf_masked.as_matrix()[:, :3, :3],
914
+ r[xp.asarray([False, True])].as_matrix(), atol=atol)
915
+ xp_assert_close(tf_masked.as_matrix()[:, :3, 3], t[xp.asarray([False, True])],
916
+ atol=atol)
917
+
918
+ tf_masked = tf[xp.asarray([False, False])]
919
+ assert len(tf_masked) == 0
920
+
921
+
922
+ def test_indexing_array_like():
923
+ atol = 1e-12
924
+
925
+ r = Rotation.from_euler('zyx', np.array([[90, 0, 0], [0, 90, 0]]), degrees=True)
926
+ t = np.array([[1.0, 2, 3], [4, 5, 6]])
927
+ tf = RigidTransform.from_components(t, r)
928
+
929
+ tf_masked = tf[[False, True]]
930
+ xp_assert_close(tf_masked.as_matrix()[:, :3, :3], r[[False, True]].as_matrix(),
931
+ atol=atol)
932
+ xp_assert_close(tf_masked.as_matrix()[:, :3, 3], t[[False, True]], atol=atol)
933
+ tf_masked = tf[[False, False]]
934
+ assert len(tf_masked) == 0
935
+
936
+
937
+ def test_concatenate(xp):
938
+ atol = 1e-12
939
+
940
+ # Test concatenation of transforms
941
+ t1 = xp.asarray([1, 0, 0])
942
+ r1 = Rotation.from_euler('z', xp.asarray(90), degrees=True)
943
+ tf1 = RigidTransform.from_components(t1, r1)
944
+
945
+ t2 = xp.asarray([0, 1, 0])
946
+ r2 = Rotation.from_euler('x', xp.asarray(90), degrees=True)
947
+ tf2 = RigidTransform.from_components(t2, r2)
948
+
949
+ # Concatenate single transforms
950
+ concatenated1 = RigidTransform.concatenate([tf1, tf2])
951
+ xp_assert_close(concatenated1[0].as_matrix(), tf1.as_matrix(), atol=atol)
952
+ xp_assert_close(concatenated1[1].as_matrix(), tf2.as_matrix(), atol=atol)
953
+
954
+ # Concatenate multiple transforms
955
+ concatenated2 = RigidTransform.concatenate([tf1, concatenated1])
956
+ xp_assert_close(concatenated2[0].as_matrix(), tf1.as_matrix(), atol=atol)
957
+ xp_assert_close(concatenated2[1].as_matrix(), tf1.as_matrix(), atol=atol)
958
+ xp_assert_close(concatenated2[2].as_matrix(), tf2.as_matrix(), atol=atol)
959
+
960
+
961
+ def test_input_validation(xp):
962
+ # Test invalid matrix shapes
963
+ inputs = [xp.eye(3), xp.zeros((4, 3)), [], xp.zeros((1, 1, 4, 4))]
964
+ for input in inputs:
965
+ with pytest.raises(ValueError, match="Expected `matrix` to have shape"):
966
+ RigidTransform.from_matrix(input)
967
+
968
+ # Test invalid last row
969
+ matrix = xp.eye(4)
970
+ matrix = xpx.at(matrix)[3, :].set(xp.asarray([1, 0, 0, 1]))
971
+ if is_lazy_array(matrix):
972
+ matrix = RigidTransform.from_matrix(matrix).as_matrix()
973
+ assert xp.all(xp.isnan(matrix))
974
+ else:
975
+ with pytest.raises(ValueError, match="last row of transformation matrix 0"):
976
+ RigidTransform.from_matrix(matrix)
977
+
978
+ # Test invalid last row for multiple transforms
979
+ matrix = xp.zeros((2, 4, 4))
980
+ matrix = xpx.at(matrix)[...].set(xp.eye(4))
981
+ matrix = xpx.at(matrix)[1, 3, :].set(xp.asarray([1, 0, 0, 1]))
982
+ if is_lazy_array(matrix):
983
+ matrix = RigidTransform.from_matrix(matrix).as_matrix()
984
+ assert not xp.any(xp.isnan(matrix[0, ...]))
985
+ assert xp.all(xp.isnan(matrix[1, ...]))
986
+ else:
987
+ with pytest.raises(ValueError, match="last row of transformation matrix 1"):
988
+ RigidTransform.from_matrix(matrix)
989
+
990
+ # Test left handed rotation matrix
991
+ matrix = xp.eye(4)
992
+ matrix = xpx.at(matrix)[0, 0].set(-1)
993
+ if is_lazy_array(matrix):
994
+ matrix = RigidTransform.from_matrix(matrix).as_matrix()
995
+ assert xp.all(xp.isnan(matrix[..., :3, :3]))
996
+ else:
997
+ with pytest.raises(ValueError, match="Non-positive determinant"):
998
+ RigidTransform(matrix, normalize=True)
999
+
1000
+ # Test non-Rotation input
1001
+ with pytest.raises(ValueError,
1002
+ match="Expected `rotation` to be a `Rotation` instance"):
1003
+ RigidTransform.from_rotation(xp.eye(3))
1004
+
1005
+
1006
+ def test_translation_validation(xp):
1007
+ # Test invalid translation shapes
1008
+ with pytest.raises(ValueError, match="Expected `translation` to have shape"):
1009
+ RigidTransform.from_translation(xp.asarray([1, 2]))
1010
+
1011
+ with pytest.raises(ValueError, match="Expected `translation` to have shape"):
1012
+ RigidTransform.from_translation(xp.zeros((2, 2)))
1013
+
1014
+ with pytest.raises(ValueError, match="Expected `translation` to have shape"):
1015
+ RigidTransform.from_translation(xp.zeros((1, 1, 3)))
1016
+
1017
+
1018
+ def test_vector_validation(xp):
1019
+ tf = rigid_transform_to_xp(RigidTransform.identity(2), xp=xp)
1020
+
1021
+ # Test invalid vector shapes
1022
+ with pytest.raises(ValueError, match="Expected vector to have shape"):
1023
+ tf.apply(xp.asarray([1, 2]))
1024
+
1025
+ with pytest.raises(ValueError, match="Expected vector to have shape"):
1026
+ tf.apply(xp.zeros((2, 2)))
1027
+
1028
+ with pytest.raises(ValueError, match="Expected vector to have shape"):
1029
+ tf.apply(xp.zeros((1, 1, 3)))
1030
+
1031
+
1032
+ def test_indexing_validation(xp):
1033
+ tf = RigidTransform.from_matrix(xp.eye(4))
1034
+
1035
+ # Test indexing on single transform
1036
+ with pytest.raises(TypeError, match="Single transform is not subscriptable"):
1037
+ tf[0]
1038
+
1039
+ with pytest.raises(TypeError, match="Single transform is not subscriptable"):
1040
+ tf[0:1]
1041
+
1042
+ # Test length on single transform
1043
+ with pytest.raises(TypeError, match="Single transform has no len"):
1044
+ len(tf)
1045
+
1046
+
1047
+ def test_composition_validation(xp):
1048
+ tf2 = RigidTransform.from_translation(xp.asarray([[1, 2, 3], [4, 5, 6]]))
1049
+ tf3 = RigidTransform.from_translation(xp.asarray([[1, 2, 3], [4, 5, 6], [7, 8, 9]]))
1050
+
1051
+ # Test incompatible shapes
1052
+ with pytest.raises(ValueError, match="Expected equal number of transforms"):
1053
+ tf2 * tf3
1054
+
1055
+
1056
+ def test_concatenate_validation(xp):
1057
+ tf = RigidTransform.from_matrix(xp.eye(4))
1058
+
1059
+ # Test invalid inputs
1060
+ with pytest.raises(TypeError,
1061
+ match="input must contain RigidTransform objects"):
1062
+ RigidTransform.concatenate([tf, xp.eye(4)])
1063
+
1064
+
1065
+ def test_setitem_validation(xp):
1066
+ tf = RigidTransform.from_translation(xp.asarray([[1, 2, 3], [4, 5, 6]]))
1067
+ single = RigidTransform.from_matrix(xp.eye(4))
1068
+
1069
+ # Test setting item on single transform
1070
+ with pytest.raises(TypeError, match="Single transform is not subscriptable"):
1071
+ single[0] = tf
1072
+
1073
+ # Test invalid value type
1074
+ with pytest.raises(TypeError, match="value must be a RigidTransform"):
1075
+ tf[0] = xp.eye(4)
1076
+
1077
+
1078
+ @pytest.mark.skip_xp_backends("jax.numpy",
1079
+ reason="JAX does not support memory sharing")
1080
+ def test_copy_flag(xp):
1081
+ # Test that copy=True creates new memory
1082
+ matrix = xp.eye(4)
1083
+ tf = RigidTransform(matrix, normalize=False, copy=True)
1084
+ matrix[0, 0] = 2
1085
+ assert tf.as_matrix()[0, 0] == 1
1086
+
1087
+ # Test that copy=False shares memory
1088
+ matrix = xp.eye(4)
1089
+ tf = RigidTransform(matrix, normalize=False, copy=False)
1090
+ matrix[0, 0] = 2
1091
+ assert tf.as_matrix()[0, 0] == 2
1092
+
1093
+
1094
+ def test_normalize_dual_quaternion(xp):
1095
+ dual_quat = normalize_dual_quaternion(xp.zeros((1, 8)))
1096
+ xp_assert_close(xp_vector_norm(dual_quat[0, :4], axis=-1), xp.asarray(1.0)[()],
1097
+ atol=1e-12)
1098
+ xp_assert_close(xp.vecdot(dual_quat[0, :4], dual_quat[0, 4:])[()],
1099
+ xp.asarray(0.0)[()], atol=1e-12)
1100
+
1101
+ rng = np.random.default_rng(103213650)
1102
+ dual_quat = xp.asarray(rng.normal(size=(1000, 8)))
1103
+ dual_quat = normalize_dual_quaternion(dual_quat)
1104
+ expected = xp.ones(dual_quat.shape[0])
1105
+ xp_assert_close(xp_vector_norm(dual_quat[:, :4], axis=-1), expected, atol=1e-12)
1106
+ expected = xp.zeros(dual_quat.shape[0])
1107
+ xp_assert_close(xp.vecdot(dual_quat[:, :4], dual_quat[:, 4:]), expected, atol=1e-12)
1108
+
1109
+
1110
+ def test_empty_transform_construction(xp):
1111
+ tf = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1112
+ assert len(tf) == 0
1113
+ assert not tf.single
1114
+
1115
+ tf = RigidTransform.from_rotation(Rotation.from_quat(xp.zeros((0, 4))))
1116
+ assert len(tf) == 0
1117
+ assert not tf.single
1118
+
1119
+ tf = RigidTransform.from_translation(xp.empty((0, 3)))
1120
+ assert len(tf) == 0
1121
+ assert not tf.single
1122
+
1123
+ empty_rot = Rotation.from_quat(xp.zeros((0, 4)))
1124
+ tf = RigidTransform.from_components(xp.empty((0, 3)), empty_rot)
1125
+ assert len(tf) == 0
1126
+ assert not tf.single
1127
+
1128
+ tf = RigidTransform.from_exp_coords(xp.empty((0, 6)))
1129
+ assert len(tf) == 0
1130
+ assert not tf.single
1131
+
1132
+ tf = RigidTransform.from_dual_quat(xp.empty((0, 8)))
1133
+ assert len(tf) == 0
1134
+ assert not tf.single
1135
+
1136
+ tf = RigidTransform.identity(0)
1137
+ assert len(tf) == 0
1138
+ assert not tf.single
1139
+
1140
+
1141
+ def test_empty_transform_representation(xp):
1142
+ tf = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1143
+
1144
+ assert len(tf.rotation) == 0
1145
+ assert tf.translation.shape == (0, 3)
1146
+
1147
+ t, r = tf.as_components()
1148
+ assert t.shape == (0, 3)
1149
+ assert len(r) == 0
1150
+
1151
+ assert tf.as_matrix().shape == (0, 4, 4)
1152
+ assert tf.as_exp_coords().shape == (0, 6)
1153
+ assert tf.as_dual_quat().shape == (0, 8)
1154
+
1155
+
1156
+ def test_empty_transform_application(xp):
1157
+ tf = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1158
+
1159
+ assert tf.apply(xp.zeros((3,))).shape == (0, 3)
1160
+ assert tf.apply(xp.empty((0, 3))).shape == (0, 3)
1161
+
1162
+ with pytest.raises(ValueError, match="operands could not be broadcast together"):
1163
+ tf.apply(xp.zeros((2, 3)))
1164
+
1165
+
1166
+ def test_empty_transform_composition(xp):
1167
+ tf_empty = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1168
+ tf_single = RigidTransform.from_matrix(xp.eye(4))
1169
+ tf_many = rigid_transform_to_xp(RigidTransform.identity(3), xp=xp)
1170
+
1171
+ assert len(tf_empty * tf_empty) == 0
1172
+ assert len(tf_empty * tf_single) == 0
1173
+ assert len(tf_single * tf_empty) == 0
1174
+
1175
+ with pytest.raises(ValueError, match="Expected equal number of transforms"):
1176
+ tf_many * tf_empty
1177
+
1178
+ with pytest.raises(ValueError, match="Expected equal number of transforms"):
1179
+ tf_empty * tf_many
1180
+
1181
+
1182
+ def test_empty_transform_concatenation(xp):
1183
+ tf_empty = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1184
+ tf_single = RigidTransform.from_matrix(xp.eye(4))
1185
+ tf_many = rigid_transform_to_xp(RigidTransform.identity(2), xp=xp)
1186
+
1187
+ assert len(RigidTransform.concatenate([tf_empty, tf_empty])) == 0
1188
+ assert len(RigidTransform.concatenate([tf_empty, tf_single])) == 1
1189
+ assert len(RigidTransform.concatenate([tf_single, tf_empty])) == 1
1190
+ assert len(RigidTransform.concatenate([tf_empty, tf_many])) == 2
1191
+ assert len(RigidTransform.concatenate([tf_many, tf_empty])) == 2
1192
+ assert len(RigidTransform.concatenate([tf_many, tf_empty, tf_single])) == 3
1193
+
1194
+
1195
+ def test_empty_transform_inv_and_pow(xp):
1196
+ tf = RigidTransform.from_matrix(xp.empty((0, 4, 4)))
1197
+ assert len(tf.inv()) == 0
1198
+ assert len(tf ** 0) == 0
1199
+ assert len(tf ** 1) == 0
1200
+ assert len(tf ** -1) == 0
1201
+ assert len(tf ** 0.5) == 0
1202
+
1203
+
1204
+ def test_empty_transform_indexing(xp):
1205
+ tf_many = rigid_transform_to_xp(RigidTransform.identity(3), xp=xp)
1206
+ tf_zero = tf_many[xp.asarray([], dtype=xp.int64)]
1207
+ assert len(tf_zero) == 0
1208
+
1209
+ assert len(tf_zero[xp.asarray([], dtype=xp.int64)]) == 0
1210
+ # Array API does not specify out-of-bounds indexing. Only check for numpy.
1211
+ if is_numpy(xp):
1212
+ assert len(tf_zero[:5]) == 0 # Slices can go out of bounds.
1213
+
1214
+ with pytest.raises(IndexError):
1215
+ tf_zero[0]
1216
+
1217
+ with pytest.raises(IndexError):
1218
+ tf_zero[xp.asarray([0, 2])]
1219
+
1220
+ with pytest.raises(IndexError):
1221
+ tf_zero[xp.asarray([False, True])]