scipy 1.16.2__cp312-cp312-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.cp312-win_arm64.lib +0 -0
  4. scipy/_cyutility.cp312-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.cp312-win_arm64.lib +0 -0
  13. scipy/_lib/_ccallback_c.cp312-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.cp312-win_arm64.lib +0 -0
  18. scipy/_lib/_fpumode.cp312-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.cp312-win_arm64.lib +0 -0
  23. scipy/_lib/_test_ccallback.cp312-win_arm64.pyd +0 -0
  24. scipy/_lib/_test_deprecation_call.cp312-win_arm64.lib +0 -0
  25. scipy/_lib/_test_deprecation_call.cp312-win_arm64.pyd +0 -0
  26. scipy/_lib/_test_deprecation_def.cp312-win_arm64.lib +0 -0
  27. scipy/_lib/_test_deprecation_def.cp312-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.cp312-win_arm64.lib +0 -0
  35. scipy/_lib/_uarray/_uarray.cp312-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.cp312-win_arm64.lib +0 -0
  101. scipy/_lib/messagestream.cp312-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.cp312-win_arm64.lib +0 -0
  148. scipy/cluster/_hierarchy.cp312-win_arm64.pyd +0 -0
  149. scipy/cluster/_optimal_leaf_ordering.cp312-win_arm64.lib +0 -0
  150. scipy/cluster/_optimal_leaf_ordering.cp312-win_arm64.pyd +0 -0
  151. scipy/cluster/_vq.cp312-win_arm64.lib +0 -0
  152. scipy/cluster/_vq.cp312-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.cp312-win_arm64.lib +0 -0
  193. scipy/fft/_pocketfft/pypocketfft.cp312-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.cp312-win_arm64.lib +0 -0
  215. scipy/fftpack/convolve.cp312-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.cp312-win_arm64.lib +0 -0
  233. scipy/integrate/_dop.cp312-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.cp312-win_arm64.lib +0 -0
  248. scipy/integrate/_lsoda.cp312-win_arm64.pyd +0 -0
  249. scipy/integrate/_ode.py +1395 -0
  250. scipy/integrate/_odepack.cp312-win_arm64.lib +0 -0
  251. scipy/integrate/_odepack.cp312-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.cp312-win_arm64.lib +0 -0
  255. scipy/integrate/_quadpack.cp312-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.cp312-win_arm64.lib +0 -0
  265. scipy/integrate/_test_multivariate.cp312-win_arm64.pyd +0 -0
  266. scipy/integrate/_test_odeint_banded.cp312-win_arm64.lib +0 -0
  267. scipy/integrate/_test_odeint_banded.cp312-win_arm64.pyd +0 -0
  268. scipy/integrate/_vode.cp312-win_arm64.lib +0 -0
  269. scipy/integrate/_vode.cp312-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.cp312-win_arm64.lib +0 -0
  290. scipy/interpolate/_dfitpack.cp312-win_arm64.pyd +0 -0
  291. scipy/interpolate/_dierckx.cp312-win_arm64.lib +0 -0
  292. scipy/interpolate/_dierckx.cp312-win_arm64.pyd +0 -0
  293. scipy/interpolate/_fitpack.cp312-win_arm64.lib +0 -0
  294. scipy/interpolate/_fitpack.cp312-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.cp312-win_arm64.lib +0 -0
  300. scipy/interpolate/_interpnd.cp312-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.cp312-win_arm64.lib +0 -0
  307. scipy/interpolate/_ppoly.cp312-win_arm64.pyd +0 -0
  308. scipy/interpolate/_rbf.py +290 -0
  309. scipy/interpolate/_rbfinterp.py +550 -0
  310. scipy/interpolate/_rbfinterp_pythran.cp312-win_arm64.lib +0 -0
  311. scipy/interpolate/_rbfinterp_pythran.cp312-win_arm64.pyd +0 -0
  312. scipy/interpolate/_rgi.py +764 -0
  313. scipy/interpolate/_rgi_cython.cp312-win_arm64.lib +0 -0
  314. scipy/interpolate/_rgi_cython.cp312-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.cp312-win_arm64.lib +0 -0
  343. scipy/io/_fast_matrix_market/_fmm_core.cp312-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.cp312-win_arm64.lib +0 -0
  355. scipy/io/_test_fortran.cp312-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.cp312-win_arm64.lib +0 -0
  386. scipy/io/matlab/_mio5_utils.cp312-win_arm64.pyd +0 -0
  387. scipy/io/matlab/_mio_utils.cp312-win_arm64.lib +0 -0
  388. scipy/io/matlab/_mio_utils.cp312-win_arm64.pyd +0 -0
  389. scipy/io/matlab/_miobase.py +435 -0
  390. scipy/io/matlab/_streams.cp312-win_arm64.lib +0 -0
  391. scipy/io/matlab/_streams.cp312-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.cp312-win_arm64.lib +0 -0
  623. scipy/linalg/_cythonized_array_utils.cp312-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.cp312-win_arm64.lib +0 -0
  630. scipy/linalg/_decomp_interpolative.cp312-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.cp312-win_arm64.lib +0 -0
  634. scipy/linalg/_decomp_lu_cython.cp312-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.cp312-win_arm64.lib +0 -0
  642. scipy/linalg/_decomp_update.cp312-win_arm64.pyd +0 -0
  643. scipy/linalg/_expm_frechet.py +417 -0
  644. scipy/linalg/_fblas.cp312-win_arm64.lib +0 -0
  645. scipy/linalg/_fblas.cp312-win_arm64.pyd +0 -0
  646. scipy/linalg/_flapack.cp312-win_arm64.lib +0 -0
  647. scipy/linalg/_flapack.cp312-win_arm64.pyd +0 -0
  648. scipy/linalg/_lapack_subroutines.h +1521 -0
  649. scipy/linalg/_linalg_pythran.cp312-win_arm64.lib +0 -0
  650. scipy/linalg/_linalg_pythran.cp312-win_arm64.pyd +0 -0
  651. scipy/linalg/_matfuncs.py +1050 -0
  652. scipy/linalg/_matfuncs_expm.cp312-win_arm64.lib +0 -0
  653. scipy/linalg/_matfuncs_expm.cp312-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.cp312-win_arm64.lib +0 -0
  657. scipy/linalg/_matfuncs_schur_sqrtm.cp312-win_arm64.pyd +0 -0
  658. scipy/linalg/_matfuncs_sqrtm.py +107 -0
  659. scipy/linalg/_matfuncs_sqrtm_triu.cp312-win_arm64.lib +0 -0
  660. scipy/linalg/_matfuncs_sqrtm_triu.cp312-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.cp312-win_arm64.lib +0 -0
  665. scipy/linalg/_solve_toeplitz.cp312-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.cp312-win_arm64.lib +0 -0
  672. scipy/linalg/cython_blas.cp312-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.cp312-win_arm64.lib +0 -0
  676. scipy/linalg/cython_lapack.cp312-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.cp312-win_arm64.lib +0 -0
  728. scipy/ndimage/_ctest.cp312-win_arm64.pyd +0 -0
  729. scipy/ndimage/_cytest.cp312-win_arm64.lib +0 -0
  730. scipy/ndimage/_cytest.cp312-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.cp312-win_arm64.lib +0 -0
  738. scipy/ndimage/_nd_image.cp312-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.cp312-win_arm64.lib +0 -0
  742. scipy/ndimage/_ni_label.cp312-win_arm64.pyd +0 -0
  743. scipy/ndimage/_ni_support.py +139 -0
  744. scipy/ndimage/_rank_filter_1d.cp312-win_arm64.lib +0 -0
  745. scipy/ndimage/_rank_filter_1d.cp312-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.cp312-win_arm64.lib +0 -0
  768. scipy/odr/__odrpack.cp312-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.cp312-win_arm64.lib +0 -0
  780. scipy/optimize/_bglu_dense.cp312-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.cp312-win_arm64.lib +0 -0
  790. scipy/optimize/_direct.cp312-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.cp312-win_arm64.lib +0 -0
  795. scipy/optimize/_group_columns.cp312-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.cp312-win_arm64.lib +0 -0
  799. scipy/optimize/_highspy/_core.cp312-win_arm64.pyd +0 -0
  800. scipy/optimize/_highspy/_highs_options.cp312-win_arm64.lib +0 -0
  801. scipy/optimize/_highspy/_highs_options.cp312-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.cp312-win_arm64.lib +0 -0
  805. scipy/optimize/_lbfgsb.cp312-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.cp312-win_arm64.lib +0 -0
  816. scipy/optimize/_lsap.cp312-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.cp312-win_arm64.lib +0 -0
  822. scipy/optimize/_lsq/givens_elimination.cp312-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.cp312-win_arm64.lib +0 -0
  830. scipy/optimize/_minpack.cp312-win_arm64.pyd +0 -0
  831. scipy/optimize/_minpack_py.py +1178 -0
  832. scipy/optimize/_moduleTNC.cp312-win_arm64.lib +0 -0
  833. scipy/optimize/_moduleTNC.cp312-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.cp312-win_arm64.lib +0 -0
  839. scipy/optimize/_pava_pybind.cp312-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.cp312-win_arm64.lib +0 -0
  850. scipy/optimize/_slsqplib.cp312-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.cp312-win_arm64.lib +0 -0
  855. scipy/optimize/_trlib/_trlib.cp312-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.cp312-win_arm64.lib +0 -0
  877. scipy/optimize/_zeros.cp312-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.cp312-win_arm64.lib +0 -0
  882. scipy/optimize/cython_optimize/_zeros.cp312-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.cp312-win_arm64.lib +0 -0
  953. scipy/signal/_max_len_seq_inner.cp312-win_arm64.pyd +0 -0
  954. scipy/signal/_peak_finding.py +1310 -0
  955. scipy/signal/_peak_finding_utils.cp312-win_arm64.lib +0 -0
  956. scipy/signal/_peak_finding_utils.cp312-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.cp312-win_arm64.lib +0 -0
  963. scipy/signal/_sigtools.cp312-win_arm64.pyd +0 -0
  964. scipy/signal/_sosfilt.cp312-win_arm64.lib +0 -0
  965. scipy/signal/_sosfilt.cp312-win_arm64.pyd +0 -0
  966. scipy/signal/_spectral_py.py +2471 -0
  967. scipy/signal/_spline.cp312-win_arm64.lib +0 -0
  968. scipy/signal/_spline.cp312-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.cp312-win_arm64.lib +0 -0
  974. scipy/signal/_upfirdn_apply.cp312-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.cp312-win_arm64.lib +0 -0
  1021. scipy/sparse/_csparsetools.cp312-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.cp312-win_arm64.lib +0 -0
  1032. scipy/sparse/_sparsetools.cp312-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.cp312-win_arm64.lib +0 -0
  1043. scipy/sparse/csgraph/_flow.cp312-win_arm64.pyd +0 -0
  1044. scipy/sparse/csgraph/_laplacian.py +563 -0
  1045. scipy/sparse/csgraph/_matching.cp312-win_arm64.lib +0 -0
  1046. scipy/sparse/csgraph/_matching.cp312-win_arm64.pyd +0 -0
  1047. scipy/sparse/csgraph/_min_spanning_tree.cp312-win_arm64.lib +0 -0
  1048. scipy/sparse/csgraph/_min_spanning_tree.cp312-win_arm64.pyd +0 -0
  1049. scipy/sparse/csgraph/_reordering.cp312-win_arm64.lib +0 -0
  1050. scipy/sparse/csgraph/_reordering.cp312-win_arm64.pyd +0 -0
  1051. scipy/sparse/csgraph/_shortest_path.cp312-win_arm64.lib +0 -0
  1052. scipy/sparse/csgraph/_shortest_path.cp312-win_arm64.pyd +0 -0
  1053. scipy/sparse/csgraph/_tools.cp312-win_arm64.lib +0 -0
  1054. scipy/sparse/csgraph/_tools.cp312-win_arm64.pyd +0 -0
  1055. scipy/sparse/csgraph/_traversal.cp312-win_arm64.lib +0 -0
  1056. scipy/sparse/csgraph/_traversal.cp312-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.cp312-win_arm64.lib +0 -0
  1079. scipy/sparse/linalg/_dsolve/_superlu.cp312-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.cp312-win_arm64.lib +0 -0
  1089. scipy/sparse/linalg/_eigen/arpack/_arpack.cp312-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.cp312-win_arm64.lib +0 -0
  1122. scipy/sparse/linalg/_propack/_cpropack.cp312-win_arm64.pyd +0 -0
  1123. scipy/sparse/linalg/_propack/_dpropack.cp312-win_arm64.lib +0 -0
  1124. scipy/sparse/linalg/_propack/_dpropack.cp312-win_arm64.pyd +0 -0
  1125. scipy/sparse/linalg/_propack/_spropack.cp312-win_arm64.lib +0 -0
  1126. scipy/sparse/linalg/_propack/_spropack.cp312-win_arm64.pyd +0 -0
  1127. scipy/sparse/linalg/_propack/_zpropack.cp312-win_arm64.lib +0 -0
  1128. scipy/sparse/linalg/_propack/_zpropack.cp312-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.cp312-win_arm64.lib +0 -0
  1170. scipy/spatial/_ckdtree.cp312-win_arm64.pyd +0 -0
  1171. scipy/spatial/_distance_pybind.cp312-win_arm64.lib +0 -0
  1172. scipy/spatial/_distance_pybind.cp312-win_arm64.pyd +0 -0
  1173. scipy/spatial/_distance_wrap.cp312-win_arm64.lib +0 -0
  1174. scipy/spatial/_distance_wrap.cp312-win_arm64.pyd +0 -0
  1175. scipy/spatial/_geometric_slerp.py +238 -0
  1176. scipy/spatial/_hausdorff.cp312-win_arm64.lib +0 -0
  1177. scipy/spatial/_hausdorff.cp312-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.cp312-win_arm64.lib +0 -0
  1182. scipy/spatial/_qhull.cp312-win_arm64.pyd +0 -0
  1183. scipy/spatial/_qhull.pyi +213 -0
  1184. scipy/spatial/_spherical_voronoi.py +341 -0
  1185. scipy/spatial/_voronoi.cp312-win_arm64.lib +0 -0
  1186. scipy/spatial/_voronoi.cp312-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.cp312-win_arm64.lib +0 -0
  1236. scipy/spatial/transform/_rigid_transform.cp312-win_arm64.pyd +0 -0
  1237. scipy/spatial/transform/_rotation.cp312-win_arm64.lib +0 -0
  1238. scipy/spatial/transform/_rotation.cp312-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.cp312-win_arm64.lib +0 -0
  1252. scipy/special/_comb.cp312-win_arm64.pyd +0 -0
  1253. scipy/special/_ellip_harm.py +214 -0
  1254. scipy/special/_ellip_harm_2.cp312-win_arm64.lib +0 -0
  1255. scipy/special/_ellip_harm_2.cp312-win_arm64.pyd +0 -0
  1256. scipy/special/_gufuncs.cp312-win_arm64.lib +0 -0
  1257. scipy/special/_gufuncs.cp312-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.cp312-win_arm64.lib +0 -0
  1281. scipy/special/_specfun.cp312-win_arm64.pyd +0 -0
  1282. scipy/special/_special_ufuncs.cp312-win_arm64.lib +0 -0
  1283. scipy/special/_special_ufuncs.cp312-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.cp312-win_arm64.lib +0 -0
  1288. scipy/special/_test_internal.cp312-win_arm64.pyd +0 -0
  1289. scipy/special/_test_internal.pyi +9 -0
  1290. scipy/special/_testutils.py +321 -0
  1291. scipy/special/_ufuncs.cp312-win_arm64.lib +0 -0
  1292. scipy/special/_ufuncs.cp312-win_arm64.pyd +0 -0
  1293. scipy/special/_ufuncs.pyi +522 -0
  1294. scipy/special/_ufuncs.pyx +13173 -0
  1295. scipy/special/_ufuncs_cxx.cp312-win_arm64.lib +0 -0
  1296. scipy/special/_ufuncs_cxx.cp312-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.cp312-win_arm64.lib +0 -0
  1304. scipy/special/cython_special.cp312-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.cp312-win_arm64.lib +0 -0
  1376. scipy/stats/_ansari_swilk_statistics.cp312-win_arm64.pyd +0 -0
  1377. scipy/stats/_axis_nan_policy.py +692 -0
  1378. scipy/stats/_biasedurn.cp312-win_arm64.lib +0 -0
  1379. scipy/stats/_biasedurn.cp312-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.cp312-win_arm64.lib +0 -0
  1404. scipy/stats/_levy_stable/levyst.cp312-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.cp312-win_arm64.lib +0 -0
  1418. scipy/stats/_qmc_cy.cp312-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.cp312-win_arm64.lib +0 -0
  1422. scipy/stats/_qmvnt_cy.cp312-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.cp312-win_arm64.lib +0 -0
  1426. scipy/stats/_rcont/rcont.cp312-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.cp312-win_arm64.lib +0 -0
  1433. scipy/stats/_sobol.cp312-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.cp312-win_arm64.lib +0 -0
  1437. scipy/stats/_stats.cp312-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.cp312-win_arm64.lib +0 -0
  1442. scipy/stats/_stats_pythran.cp312-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.cp312-win_arm64.lib +0 -0
  1447. scipy/stats/_unuran/unuran_wrapper.cp312-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,1701 @@
1
+ import itertools
2
+
3
+ import numpy as np
4
+ from numpy.testing import assert_, assert_allclose, assert_equal
5
+ from pytest import raises as assert_raises
6
+ from scipy import linalg
7
+ import scipy.linalg._decomp_update as _decomp_update
8
+ from scipy.linalg._decomp_update import qr_delete, qr_update, qr_insert
9
+
10
+ def assert_unitary(a, rtol=None, atol=None, assert_sqr=True):
11
+ if rtol is None:
12
+ rtol = 10.0 ** -(np.finfo(a.dtype).precision-2)
13
+ if atol is None:
14
+ atol = 10*np.finfo(a.dtype).eps
15
+
16
+ if assert_sqr:
17
+ assert_(a.shape[0] == a.shape[1], 'unitary matrices must be square')
18
+ aTa = np.dot(a.T.conj(), a)
19
+ assert_allclose(aTa, np.eye(a.shape[1]), rtol=rtol, atol=atol)
20
+
21
+ def assert_upper_tri(a, rtol=None, atol=None):
22
+ if rtol is None:
23
+ rtol = 10.0 ** -(np.finfo(a.dtype).precision-2)
24
+ if atol is None:
25
+ atol = 2*np.finfo(a.dtype).eps
26
+ mask = np.tri(a.shape[0], a.shape[1], -1, np.bool_)
27
+ assert_allclose(a[mask], 0.0, rtol=rtol, atol=atol)
28
+
29
+ def check_qr(q, r, a, rtol, atol, assert_sqr=True):
30
+ assert_unitary(q, rtol, atol, assert_sqr)
31
+ assert_upper_tri(r, rtol, atol)
32
+ assert_allclose(q.dot(r), a, rtol=rtol, atol=atol)
33
+
34
+ def make_strided(arrs):
35
+ strides = [(3, 7), (2, 2), (3, 4), (4, 2), (5, 4), (2, 3), (2, 1), (4, 5)]
36
+ kmax = len(strides)
37
+ k = 0
38
+ ret = []
39
+ for a in arrs:
40
+ if a.ndim == 1:
41
+ s = strides[k % kmax]
42
+ k += 1
43
+ base = np.zeros(s[0]*a.shape[0]+s[1], a.dtype)
44
+ view = base[s[1]::s[0]]
45
+ view[...] = a
46
+ elif a.ndim == 2:
47
+ s = strides[k % kmax]
48
+ t = strides[(k+1) % kmax]
49
+ k += 2
50
+ base = np.zeros((s[0]*a.shape[0]+s[1], t[0]*a.shape[1]+t[1]),
51
+ a.dtype)
52
+ view = base[s[1]::s[0], t[1]::t[0]]
53
+ view[...] = a
54
+ else:
55
+ raise ValueError('make_strided only works for ndim = 1 or'
56
+ ' 2 arrays')
57
+ ret.append(view)
58
+ return ret
59
+
60
+ def negate_strides(arrs):
61
+ ret = []
62
+ for a in arrs:
63
+ b = np.zeros_like(a)
64
+ if b.ndim == 2:
65
+ b = b[::-1, ::-1]
66
+ elif b.ndim == 1:
67
+ b = b[::-1]
68
+ else:
69
+ raise ValueError('negate_strides only works for ndim = 1 or'
70
+ ' 2 arrays')
71
+ b[...] = a
72
+ ret.append(b)
73
+ return ret
74
+
75
+ def nonitemsize_strides(arrs):
76
+ out = []
77
+ for a in arrs:
78
+ a_dtype = a.dtype
79
+ b = np.zeros(a.shape, [('a', a_dtype), ('junk', 'S1')])
80
+ c = b.getfield(a_dtype)
81
+ c[...] = a
82
+ out.append(c)
83
+ return out
84
+
85
+
86
+ def make_nonnative(arrs):
87
+ return [a.astype(a.dtype.newbyteorder()) for a in arrs]
88
+
89
+
90
+ class BaseQRdeltas:
91
+ def setup_method(self):
92
+ self.rtol = 10.0 ** -(np.finfo(self.dtype).precision-2)
93
+ self.atol = 10 * np.finfo(self.dtype).eps
94
+
95
+ def generate(self, type, mode='full'):
96
+ np.random.seed(29382)
97
+ shape = {'sqr': (8, 8), 'tall': (12, 7), 'fat': (7, 12),
98
+ 'Mx1': (8, 1), '1xN': (1, 8), '1x1': (1, 1)}[type]
99
+ a = np.random.random(shape)
100
+ if np.iscomplexobj(self.dtype.type(1)):
101
+ b = np.random.random(shape)
102
+ a = a + 1j * b
103
+ a = a.astype(self.dtype)
104
+ q, r = linalg.qr(a, mode=mode)
105
+ return a, q, r
106
+
107
+ class BaseQRdelete(BaseQRdeltas):
108
+ def test_sqr_1_row(self):
109
+ a, q, r = self.generate('sqr')
110
+ for row in range(r.shape[0]):
111
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
112
+ a1 = np.delete(a, row, 0)
113
+ check_qr(q1, r1, a1, self.rtol, self.atol)
114
+
115
+ def test_sqr_p_row(self):
116
+ a, q, r = self.generate('sqr')
117
+ for ndel in range(2, 6):
118
+ for row in range(a.shape[0]-ndel):
119
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
120
+ a1 = np.delete(a, slice(row, row+ndel), 0)
121
+ check_qr(q1, r1, a1, self.rtol, self.atol)
122
+
123
+ def test_sqr_1_col(self):
124
+ a, q, r = self.generate('sqr')
125
+ for col in range(r.shape[1]):
126
+ q1, r1 = qr_delete(q, r, col, which='col', overwrite_qr=False)
127
+ a1 = np.delete(a, col, 1)
128
+ check_qr(q1, r1, a1, self.rtol, self.atol)
129
+
130
+ def test_sqr_p_col(self):
131
+ a, q, r = self.generate('sqr')
132
+ for ndel in range(2, 6):
133
+ for col in range(r.shape[1]-ndel):
134
+ q1, r1 = qr_delete(q, r, col, ndel, which='col',
135
+ overwrite_qr=False)
136
+ a1 = np.delete(a, slice(col, col+ndel), 1)
137
+ check_qr(q1, r1, a1, self.rtol, self.atol)
138
+
139
+ def test_tall_1_row(self):
140
+ a, q, r = self.generate('tall')
141
+ for row in range(r.shape[0]):
142
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
143
+ a1 = np.delete(a, row, 0)
144
+ check_qr(q1, r1, a1, self.rtol, self.atol)
145
+
146
+ def test_tall_p_row(self):
147
+ a, q, r = self.generate('tall')
148
+ for ndel in range(2, 6):
149
+ for row in range(a.shape[0]-ndel):
150
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
151
+ a1 = np.delete(a, slice(row, row+ndel), 0)
152
+ check_qr(q1, r1, a1, self.rtol, self.atol)
153
+
154
+ def test_tall_1_col(self):
155
+ a, q, r = self.generate('tall')
156
+ for col in range(r.shape[1]):
157
+ q1, r1 = qr_delete(q, r, col, which='col', overwrite_qr=False)
158
+ a1 = np.delete(a, col, 1)
159
+ check_qr(q1, r1, a1, self.rtol, self.atol)
160
+
161
+ def test_tall_p_col(self):
162
+ a, q, r = self.generate('tall')
163
+ for ndel in range(2, 6):
164
+ for col in range(r.shape[1]-ndel):
165
+ q1, r1 = qr_delete(q, r, col, ndel, which='col',
166
+ overwrite_qr=False)
167
+ a1 = np.delete(a, slice(col, col+ndel), 1)
168
+ check_qr(q1, r1, a1, self.rtol, self.atol)
169
+
170
+ def test_fat_1_row(self):
171
+ a, q, r = self.generate('fat')
172
+ for row in range(r.shape[0]):
173
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
174
+ a1 = np.delete(a, row, 0)
175
+ check_qr(q1, r1, a1, self.rtol, self.atol)
176
+
177
+ def test_fat_p_row(self):
178
+ a, q, r = self.generate('fat')
179
+ for ndel in range(2, 6):
180
+ for row in range(a.shape[0]-ndel):
181
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
182
+ a1 = np.delete(a, slice(row, row+ndel), 0)
183
+ check_qr(q1, r1, a1, self.rtol, self.atol)
184
+
185
+ def test_fat_1_col(self):
186
+ a, q, r = self.generate('fat')
187
+ for col in range(r.shape[1]):
188
+ q1, r1 = qr_delete(q, r, col, which='col', overwrite_qr=False)
189
+ a1 = np.delete(a, col, 1)
190
+ check_qr(q1, r1, a1, self.rtol, self.atol)
191
+
192
+ def test_fat_p_col(self):
193
+ a, q, r = self.generate('fat')
194
+ for ndel in range(2, 6):
195
+ for col in range(r.shape[1]-ndel):
196
+ q1, r1 = qr_delete(q, r, col, ndel, which='col',
197
+ overwrite_qr=False)
198
+ a1 = np.delete(a, slice(col, col+ndel), 1)
199
+ check_qr(q1, r1, a1, self.rtol, self.atol)
200
+
201
+ def test_economic_1_row(self):
202
+ # this test always starts and ends with an economic decomp.
203
+ a, q, r = self.generate('tall', 'economic')
204
+ for row in range(r.shape[0]):
205
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
206
+ a1 = np.delete(a, row, 0)
207
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
208
+
209
+ # for economic row deletes
210
+ # eco - prow = eco
211
+ # eco - prow = sqr
212
+ # eco - prow = fat
213
+ def base_economic_p_row_xxx(self, ndel):
214
+ a, q, r = self.generate('tall', 'economic')
215
+ for row in range(a.shape[0]-ndel):
216
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
217
+ a1 = np.delete(a, slice(row, row+ndel), 0)
218
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
219
+
220
+ def test_economic_p_row_economic(self):
221
+ # (12, 7) - (3, 7) = (9,7) --> stays economic
222
+ self.base_economic_p_row_xxx(3)
223
+
224
+ def test_economic_p_row_sqr(self):
225
+ # (12, 7) - (5, 7) = (7, 7) --> becomes square
226
+ self.base_economic_p_row_xxx(5)
227
+
228
+ def test_economic_p_row_fat(self):
229
+ # (12, 7) - (7,7) = (5, 7) --> becomes fat
230
+ self.base_economic_p_row_xxx(7)
231
+
232
+ def test_economic_1_col(self):
233
+ a, q, r = self.generate('tall', 'economic')
234
+ for col in range(r.shape[1]):
235
+ q1, r1 = qr_delete(q, r, col, which='col', overwrite_qr=False)
236
+ a1 = np.delete(a, col, 1)
237
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
238
+
239
+ def test_economic_p_col(self):
240
+ a, q, r = self.generate('tall', 'economic')
241
+ for ndel in range(2, 6):
242
+ for col in range(r.shape[1]-ndel):
243
+ q1, r1 = qr_delete(q, r, col, ndel, which='col',
244
+ overwrite_qr=False)
245
+ a1 = np.delete(a, slice(col, col+ndel), 1)
246
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
247
+
248
+ def test_Mx1_1_row(self):
249
+ a, q, r = self.generate('Mx1')
250
+ for row in range(r.shape[0]):
251
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
252
+ a1 = np.delete(a, row, 0)
253
+ check_qr(q1, r1, a1, self.rtol, self.atol)
254
+
255
+ def test_Mx1_p_row(self):
256
+ a, q, r = self.generate('Mx1')
257
+ for ndel in range(2, 6):
258
+ for row in range(a.shape[0]-ndel):
259
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
260
+ a1 = np.delete(a, slice(row, row+ndel), 0)
261
+ check_qr(q1, r1, a1, self.rtol, self.atol)
262
+
263
+ def test_1xN_1_col(self):
264
+ a, q, r = self.generate('1xN')
265
+ for col in range(r.shape[1]):
266
+ q1, r1 = qr_delete(q, r, col, which='col', overwrite_qr=False)
267
+ a1 = np.delete(a, col, 1)
268
+ check_qr(q1, r1, a1, self.rtol, self.atol)
269
+
270
+ def test_1xN_p_col(self):
271
+ a, q, r = self.generate('1xN')
272
+ for ndel in range(2, 6):
273
+ for col in range(r.shape[1]-ndel):
274
+ q1, r1 = qr_delete(q, r, col, ndel, which='col',
275
+ overwrite_qr=False)
276
+ a1 = np.delete(a, slice(col, col+ndel), 1)
277
+ check_qr(q1, r1, a1, self.rtol, self.atol)
278
+
279
+ def test_Mx1_economic_1_row(self):
280
+ a, q, r = self.generate('Mx1', 'economic')
281
+ for row in range(r.shape[0]):
282
+ q1, r1 = qr_delete(q, r, row, overwrite_qr=False)
283
+ a1 = np.delete(a, row, 0)
284
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
285
+
286
+ def test_Mx1_economic_p_row(self):
287
+ a, q, r = self.generate('Mx1', 'economic')
288
+ for ndel in range(2, 6):
289
+ for row in range(a.shape[0]-ndel):
290
+ q1, r1 = qr_delete(q, r, row, ndel, overwrite_qr=False)
291
+ a1 = np.delete(a, slice(row, row+ndel), 0)
292
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
293
+
294
+ def test_delete_last_1_row(self):
295
+ # full and eco are the same for 1xN
296
+ a, q, r = self.generate('1xN')
297
+ q1, r1 = qr_delete(q, r, 0, 1, 'row')
298
+ assert_equal(q1, np.ndarray(shape=(0, 0), dtype=q.dtype))
299
+ assert_equal(r1, np.ndarray(shape=(0, r.shape[1]), dtype=r.dtype))
300
+
301
+ def test_delete_last_p_row(self):
302
+ a, q, r = self.generate('tall', 'full')
303
+ q1, r1 = qr_delete(q, r, 0, a.shape[0], 'row')
304
+ assert_equal(q1, np.ndarray(shape=(0, 0), dtype=q.dtype))
305
+ assert_equal(r1, np.ndarray(shape=(0, r.shape[1]), dtype=r.dtype))
306
+
307
+ a, q, r = self.generate('tall', 'economic')
308
+ q1, r1 = qr_delete(q, r, 0, a.shape[0], 'row')
309
+ assert_equal(q1, np.ndarray(shape=(0, 0), dtype=q.dtype))
310
+ assert_equal(r1, np.ndarray(shape=(0, r.shape[1]), dtype=r.dtype))
311
+
312
+ def test_delete_last_1_col(self):
313
+ a, q, r = self.generate('Mx1', 'economic')
314
+ q1, r1 = qr_delete(q, r, 0, 1, 'col')
315
+ assert_equal(q1, np.ndarray(shape=(q.shape[0], 0), dtype=q.dtype))
316
+ assert_equal(r1, np.ndarray(shape=(0, 0), dtype=r.dtype))
317
+
318
+ a, q, r = self.generate('Mx1', 'full')
319
+ q1, r1 = qr_delete(q, r, 0, 1, 'col')
320
+ assert_unitary(q1)
321
+ assert_(q1.dtype == q.dtype)
322
+ assert_(q1.shape == q.shape)
323
+ assert_equal(r1, np.ndarray(shape=(r.shape[0], 0), dtype=r.dtype))
324
+
325
+ def test_delete_last_p_col(self):
326
+ a, q, r = self.generate('tall', 'full')
327
+ q1, r1 = qr_delete(q, r, 0, a.shape[1], 'col')
328
+ assert_unitary(q1)
329
+ assert_(q1.dtype == q.dtype)
330
+ assert_(q1.shape == q.shape)
331
+ assert_equal(r1, np.ndarray(shape=(r.shape[0], 0), dtype=r.dtype))
332
+
333
+ a, q, r = self.generate('tall', 'economic')
334
+ q1, r1 = qr_delete(q, r, 0, a.shape[1], 'col')
335
+ assert_equal(q1, np.ndarray(shape=(q.shape[0], 0), dtype=q.dtype))
336
+ assert_equal(r1, np.ndarray(shape=(0, 0), dtype=r.dtype))
337
+
338
+ def test_delete_1x1_row_col(self):
339
+ a, q, r = self.generate('1x1')
340
+ q1, r1 = qr_delete(q, r, 0, 1, 'row')
341
+ assert_equal(q1, np.ndarray(shape=(0, 0), dtype=q.dtype))
342
+ assert_equal(r1, np.ndarray(shape=(0, r.shape[1]), dtype=r.dtype))
343
+
344
+ a, q, r = self.generate('1x1')
345
+ q1, r1 = qr_delete(q, r, 0, 1, 'col')
346
+ assert_unitary(q1)
347
+ assert_(q1.dtype == q.dtype)
348
+ assert_(q1.shape == q.shape)
349
+ assert_equal(r1, np.ndarray(shape=(r.shape[0], 0), dtype=r.dtype))
350
+
351
+ # all full qr, row deletes and single column deletes should be able to
352
+ # handle any non negative strides. (only row and column vector
353
+ # operations are used.) p column delete require fortran ordered
354
+ # Q and R and will make a copy as necessary. Economic qr row deletes
355
+ # require a contiguous q.
356
+
357
+ def base_non_simple_strides(self, adjust_strides, ks, p, which,
358
+ overwriteable):
359
+ if which == 'row':
360
+ qind = (slice(p,None), slice(p,None))
361
+ rind = (slice(p,None), slice(None))
362
+ else:
363
+ qind = (slice(None), slice(None))
364
+ rind = (slice(None), slice(None,-p))
365
+
366
+ for type, k in itertools.product(['sqr', 'tall', 'fat'], ks):
367
+ a, q0, r0, = self.generate(type)
368
+ qs, rs = adjust_strides((q0, r0))
369
+ if p == 1:
370
+ a1 = np.delete(a, k, 0 if which == 'row' else 1)
371
+ else:
372
+ s = slice(k,k+p)
373
+ if k < 0:
374
+ s = slice(k, k + p +
375
+ (a.shape[0] if which == 'row' else a.shape[1]))
376
+ a1 = np.delete(a, s, 0 if which == 'row' else 1)
377
+
378
+ # for each variable, q, r we try with it strided and
379
+ # overwrite=False. Then we try with overwrite=True, and make
380
+ # sure that q and r are still overwritten.
381
+
382
+ q = q0.copy('F')
383
+ r = r0.copy('F')
384
+ q1, r1 = qr_delete(qs, r, k, p, which, False)
385
+ check_qr(q1, r1, a1, self.rtol, self.atol)
386
+ q1o, r1o = qr_delete(qs, r, k, p, which, True)
387
+ check_qr(q1o, r1o, a1, self.rtol, self.atol)
388
+ if overwriteable:
389
+ assert_allclose(q1o, qs[qind], rtol=self.rtol, atol=self.atol)
390
+ assert_allclose(r1o, r[rind], rtol=self.rtol, atol=self.atol)
391
+
392
+ q = q0.copy('F')
393
+ r = r0.copy('F')
394
+ q2, r2 = qr_delete(q, rs, k, p, which, False)
395
+ check_qr(q2, r2, a1, self.rtol, self.atol)
396
+ q2o, r2o = qr_delete(q, rs, k, p, which, True)
397
+ check_qr(q2o, r2o, a1, self.rtol, self.atol)
398
+ if overwriteable:
399
+ assert_allclose(q2o, q[qind], rtol=self.rtol, atol=self.atol)
400
+ assert_allclose(r2o, rs[rind], rtol=self.rtol, atol=self.atol)
401
+
402
+ q = q0.copy('F')
403
+ r = r0.copy('F')
404
+ # since some of these were consumed above
405
+ qs, rs = adjust_strides((q, r))
406
+ q3, r3 = qr_delete(qs, rs, k, p, which, False)
407
+ check_qr(q3, r3, a1, self.rtol, self.atol)
408
+ q3o, r3o = qr_delete(qs, rs, k, p, which, True)
409
+ check_qr(q3o, r3o, a1, self.rtol, self.atol)
410
+ if overwriteable:
411
+ assert_allclose(q2o, qs[qind], rtol=self.rtol, atol=self.atol)
412
+ assert_allclose(r3o, rs[rind], rtol=self.rtol, atol=self.atol)
413
+
414
+ def test_non_unit_strides_1_row(self):
415
+ self.base_non_simple_strides(make_strided, [0], 1, 'row', True)
416
+
417
+ def test_non_unit_strides_p_row(self):
418
+ self.base_non_simple_strides(make_strided, [0], 3, 'row', True)
419
+
420
+ def test_non_unit_strides_1_col(self):
421
+ self.base_non_simple_strides(make_strided, [0], 1, 'col', True)
422
+
423
+ def test_non_unit_strides_p_col(self):
424
+ self.base_non_simple_strides(make_strided, [0], 3, 'col', False)
425
+
426
+ def test_neg_strides_1_row(self):
427
+ self.base_non_simple_strides(negate_strides, [0], 1, 'row', False)
428
+
429
+ def test_neg_strides_p_row(self):
430
+ self.base_non_simple_strides(negate_strides, [0], 3, 'row', False)
431
+
432
+ def test_neg_strides_1_col(self):
433
+ self.base_non_simple_strides(negate_strides, [0], 1, 'col', False)
434
+
435
+ def test_neg_strides_p_col(self):
436
+ self.base_non_simple_strides(negate_strides, [0], 3, 'col', False)
437
+
438
+ def test_non_itemize_strides_1_row(self):
439
+ self.base_non_simple_strides(nonitemsize_strides, [0], 1, 'row', False)
440
+
441
+ def test_non_itemize_strides_p_row(self):
442
+ self.base_non_simple_strides(nonitemsize_strides, [0], 3, 'row', False)
443
+
444
+ def test_non_itemize_strides_1_col(self):
445
+ self.base_non_simple_strides(nonitemsize_strides, [0], 1, 'col', False)
446
+
447
+ def test_non_itemize_strides_p_col(self):
448
+ self.base_non_simple_strides(nonitemsize_strides, [0], 3, 'col', False)
449
+
450
+ def test_non_native_byte_order_1_row(self):
451
+ self.base_non_simple_strides(make_nonnative, [0], 1, 'row', False)
452
+
453
+ def test_non_native_byte_order_p_row(self):
454
+ self.base_non_simple_strides(make_nonnative, [0], 3, 'row', False)
455
+
456
+ def test_non_native_byte_order_1_col(self):
457
+ self.base_non_simple_strides(make_nonnative, [0], 1, 'col', False)
458
+
459
+ def test_non_native_byte_order_p_col(self):
460
+ self.base_non_simple_strides(make_nonnative, [0], 3, 'col', False)
461
+
462
+ def test_neg_k(self):
463
+ a, q, r = self.generate('sqr')
464
+ for k, p, w in itertools.product([-3, -7], [1, 3], ['row', 'col']):
465
+ q1, r1 = qr_delete(q, r, k, p, w, overwrite_qr=False)
466
+ if w == 'row':
467
+ a1 = np.delete(a, slice(k+a.shape[0], k+p+a.shape[0]), 0)
468
+ else:
469
+ a1 = np.delete(a, slice(k+a.shape[0], k+p+a.shape[1]), 1)
470
+ check_qr(q1, r1, a1, self.rtol, self.atol)
471
+
472
+ def base_overwrite_qr(self, which, p, test_C, test_F, mode='full'):
473
+ assert_sqr = True if mode == 'full' else False
474
+ if which == 'row':
475
+ qind = (slice(p,None), slice(p,None))
476
+ rind = (slice(p,None), slice(None))
477
+ else:
478
+ qind = (slice(None), slice(None))
479
+ rind = (slice(None), slice(None,-p))
480
+ a, q0, r0 = self.generate('sqr', mode)
481
+ if p == 1:
482
+ a1 = np.delete(a, 3, 0 if which == 'row' else 1)
483
+ else:
484
+ a1 = np.delete(a, slice(3, 3+p), 0 if which == 'row' else 1)
485
+
486
+ # don't overwrite
487
+ q = q0.copy('F')
488
+ r = r0.copy('F')
489
+ q1, r1 = qr_delete(q, r, 3, p, which, False)
490
+ check_qr(q1, r1, a1, self.rtol, self.atol, assert_sqr)
491
+ check_qr(q, r, a, self.rtol, self.atol, assert_sqr)
492
+
493
+ if test_F:
494
+ q = q0.copy('F')
495
+ r = r0.copy('F')
496
+ q2, r2 = qr_delete(q, r, 3, p, which, True)
497
+ check_qr(q2, r2, a1, self.rtol, self.atol, assert_sqr)
498
+ # verify the overwriting
499
+ assert_allclose(q2, q[qind], rtol=self.rtol, atol=self.atol)
500
+ assert_allclose(r2, r[rind], rtol=self.rtol, atol=self.atol)
501
+
502
+ if test_C:
503
+ q = q0.copy('C')
504
+ r = r0.copy('C')
505
+ q3, r3 = qr_delete(q, r, 3, p, which, True)
506
+ check_qr(q3, r3, a1, self.rtol, self.atol, assert_sqr)
507
+ assert_allclose(q3, q[qind], rtol=self.rtol, atol=self.atol)
508
+ assert_allclose(r3, r[rind], rtol=self.rtol, atol=self.atol)
509
+
510
+ def test_overwrite_qr_1_row(self):
511
+ # any positively strided q and r.
512
+ self.base_overwrite_qr('row', 1, True, True)
513
+
514
+ def test_overwrite_economic_qr_1_row(self):
515
+ # Any contiguous q and positively strided r.
516
+ self.base_overwrite_qr('row', 1, True, True, 'economic')
517
+
518
+ def test_overwrite_qr_1_col(self):
519
+ # any positively strided q and r.
520
+ # full and eco share code paths
521
+ self.base_overwrite_qr('col', 1, True, True)
522
+
523
+ def test_overwrite_qr_p_row(self):
524
+ # any positively strided q and r.
525
+ self.base_overwrite_qr('row', 3, True, True)
526
+
527
+ def test_overwrite_economic_qr_p_row(self):
528
+ # any contiguous q and positively strided r
529
+ self.base_overwrite_qr('row', 3, True, True, 'economic')
530
+
531
+ def test_overwrite_qr_p_col(self):
532
+ # only F ordered q and r can be overwritten for cols
533
+ # full and eco share code paths
534
+ self.base_overwrite_qr('col', 3, False, True)
535
+
536
+ def test_bad_which(self):
537
+ a, q, r = self.generate('sqr')
538
+ assert_raises(ValueError, qr_delete, q, r, 0, which='foo')
539
+
540
+ def test_bad_k(self):
541
+ a, q, r = self.generate('tall')
542
+ assert_raises(ValueError, qr_delete, q, r, q.shape[0], 1)
543
+ assert_raises(ValueError, qr_delete, q, r, -q.shape[0]-1, 1)
544
+ assert_raises(ValueError, qr_delete, q, r, r.shape[0], 1, 'col')
545
+ assert_raises(ValueError, qr_delete, q, r, -r.shape[0]-1, 1, 'col')
546
+
547
+ def test_bad_p(self):
548
+ a, q, r = self.generate('tall')
549
+ # p must be positive
550
+ assert_raises(ValueError, qr_delete, q, r, 0, -1)
551
+ assert_raises(ValueError, qr_delete, q, r, 0, -1, 'col')
552
+
553
+ # and nonzero
554
+ assert_raises(ValueError, qr_delete, q, r, 0, 0)
555
+ assert_raises(ValueError, qr_delete, q, r, 0, 0, 'col')
556
+
557
+ # must have at least k+p rows or cols, depending.
558
+ assert_raises(ValueError, qr_delete, q, r, 3, q.shape[0]-2)
559
+ assert_raises(ValueError, qr_delete, q, r, 3, r.shape[1]-2, 'col')
560
+
561
+ def test_empty_q(self):
562
+ a, q, r = self.generate('tall')
563
+ # same code path for 'row' and 'col'
564
+ assert_raises(ValueError, qr_delete, np.array([]), r, 0, 1)
565
+
566
+ def test_empty_r(self):
567
+ a, q, r = self.generate('tall')
568
+ # same code path for 'row' and 'col'
569
+ assert_raises(ValueError, qr_delete, q, np.array([]), 0, 1)
570
+
571
+ def test_mismatched_q_and_r(self):
572
+ a, q, r = self.generate('tall')
573
+ r = r[1:]
574
+ assert_raises(ValueError, qr_delete, q, r, 0, 1)
575
+
576
+ def test_unsupported_dtypes(self):
577
+ dts = ['int8', 'int16', 'int32', 'int64',
578
+ 'uint8', 'uint16', 'uint32', 'uint64',
579
+ 'float16', 'longdouble', 'clongdouble',
580
+ 'bool']
581
+ a, q0, r0 = self.generate('tall')
582
+ for dtype in dts:
583
+ q = q0.real.astype(dtype)
584
+ with np.errstate(invalid="ignore"):
585
+ r = r0.real.astype(dtype)
586
+ assert_raises(ValueError, qr_delete, q, r0, 0, 1, 'row')
587
+ assert_raises(ValueError, qr_delete, q, r0, 0, 2, 'row')
588
+ assert_raises(ValueError, qr_delete, q, r0, 0, 1, 'col')
589
+ assert_raises(ValueError, qr_delete, q, r0, 0, 2, 'col')
590
+
591
+ assert_raises(ValueError, qr_delete, q0, r, 0, 1, 'row')
592
+ assert_raises(ValueError, qr_delete, q0, r, 0, 2, 'row')
593
+ assert_raises(ValueError, qr_delete, q0, r, 0, 1, 'col')
594
+ assert_raises(ValueError, qr_delete, q0, r, 0, 2, 'col')
595
+
596
+ def test_check_finite(self):
597
+ a0, q0, r0 = self.generate('tall')
598
+
599
+ q = q0.copy('F')
600
+ q[1,1] = np.nan
601
+ assert_raises(ValueError, qr_delete, q, r0, 0, 1, 'row')
602
+ assert_raises(ValueError, qr_delete, q, r0, 0, 3, 'row')
603
+ assert_raises(ValueError, qr_delete, q, r0, 0, 1, 'col')
604
+ assert_raises(ValueError, qr_delete, q, r0, 0, 3, 'col')
605
+
606
+ r = r0.copy('F')
607
+ r[1,1] = np.nan
608
+ assert_raises(ValueError, qr_delete, q0, r, 0, 1, 'row')
609
+ assert_raises(ValueError, qr_delete, q0, r, 0, 3, 'row')
610
+ assert_raises(ValueError, qr_delete, q0, r, 0, 1, 'col')
611
+ assert_raises(ValueError, qr_delete, q0, r, 0, 3, 'col')
612
+
613
+ def test_qr_scalar(self):
614
+ a, q, r = self.generate('1x1')
615
+ assert_raises(ValueError, qr_delete, q[0, 0], r, 0, 1, 'row')
616
+ assert_raises(ValueError, qr_delete, q, r[0, 0], 0, 1, 'row')
617
+ assert_raises(ValueError, qr_delete, q[0, 0], r, 0, 1, 'col')
618
+ assert_raises(ValueError, qr_delete, q, r[0, 0], 0, 1, 'col')
619
+
620
+ class TestQRdelete_f(BaseQRdelete):
621
+ dtype = np.dtype('f')
622
+
623
+ class TestQRdelete_F(BaseQRdelete):
624
+ dtype = np.dtype('F')
625
+
626
+ class TestQRdelete_d(BaseQRdelete):
627
+ dtype = np.dtype('d')
628
+
629
+ class TestQRdelete_D(BaseQRdelete):
630
+ dtype = np.dtype('D')
631
+
632
+ class BaseQRinsert(BaseQRdeltas):
633
+ def generate(self, type, mode='full', which='row', p=1):
634
+ a, q, r = super().generate(type, mode)
635
+
636
+ assert_(p > 0)
637
+ rng = np.random.RandomState(1234)
638
+
639
+ # super call set the seed...
640
+ if which == 'row':
641
+ if p == 1:
642
+ u = rng.random(a.shape[1])
643
+ else:
644
+ u = rng.random((p, a.shape[1]))
645
+ elif which == 'col':
646
+ if p == 1:
647
+ u = rng.random(a.shape[0])
648
+ else:
649
+ u = rng.random((a.shape[0], p))
650
+ else:
651
+ ValueError('which should be either "row" or "col"')
652
+
653
+ if np.iscomplexobj(self.dtype.type(1)):
654
+ b = rng.random(u.shape)
655
+ u = u + 1j * b
656
+
657
+ u = u.astype(self.dtype)
658
+ return a, q, r, u
659
+
660
+ def test_sqr_1_row(self):
661
+ a, q, r, u = self.generate('sqr', which='row')
662
+ for row in range(r.shape[0] + 1):
663
+ q1, r1 = qr_insert(q, r, u, row)
664
+ a1 = np.insert(a, row, u, 0)
665
+ check_qr(q1, r1, a1, self.rtol, self.atol)
666
+
667
+ def test_sqr_p_row(self):
668
+ # sqr + rows --> fat always
669
+ a, q, r, u = self.generate('sqr', which='row', p=3)
670
+ for row in range(r.shape[0] + 1):
671
+ q1, r1 = qr_insert(q, r, u, row)
672
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
673
+ check_qr(q1, r1, a1, self.rtol, self.atol)
674
+
675
+ def test_sqr_1_col(self):
676
+ a, q, r, u = self.generate('sqr', which='col')
677
+ for col in range(r.shape[1] + 1):
678
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
679
+ a1 = np.insert(a, col, u, 1)
680
+ check_qr(q1, r1, a1, self.rtol, self.atol)
681
+
682
+ def test_sqr_p_col(self):
683
+ # sqr + cols --> fat always
684
+ a, q, r, u = self.generate('sqr', which='col', p=3)
685
+ for col in range(r.shape[1] + 1):
686
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
687
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
688
+ check_qr(q1, r1, a1, self.rtol, self.atol)
689
+
690
+ def test_tall_1_row(self):
691
+ a, q, r, u = self.generate('tall', which='row')
692
+ for row in range(r.shape[0] + 1):
693
+ q1, r1 = qr_insert(q, r, u, row)
694
+ a1 = np.insert(a, row, u, 0)
695
+ check_qr(q1, r1, a1, self.rtol, self.atol)
696
+
697
+ def test_tall_p_row(self):
698
+ # tall + rows --> tall always
699
+ a, q, r, u = self.generate('tall', which='row', p=3)
700
+ for row in range(r.shape[0] + 1):
701
+ q1, r1 = qr_insert(q, r, u, row)
702
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
703
+ check_qr(q1, r1, a1, self.rtol, self.atol)
704
+
705
+ def test_tall_1_col(self):
706
+ a, q, r, u = self.generate('tall', which='col')
707
+ for col in range(r.shape[1] + 1):
708
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
709
+ a1 = np.insert(a, col, u, 1)
710
+ check_qr(q1, r1, a1, self.rtol, self.atol)
711
+
712
+ # for column adds to tall matrices there are three cases to test
713
+ # tall + pcol --> tall
714
+ # tall + pcol --> sqr
715
+ # tall + pcol --> fat
716
+ def base_tall_p_col_xxx(self, p):
717
+ a, q, r, u = self.generate('tall', which='col', p=p)
718
+ for col in range(r.shape[1] + 1):
719
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
720
+ a1 = np.insert(a, np.full(p, col, np.intp), u, 1)
721
+ check_qr(q1, r1, a1, self.rtol, self.atol)
722
+
723
+ def test_tall_p_col_tall(self):
724
+ # 12x7 + 12x3 = 12x10 --> stays tall
725
+ self.base_tall_p_col_xxx(3)
726
+
727
+ def test_tall_p_col_sqr(self):
728
+ # 12x7 + 12x5 = 12x12 --> becomes sqr
729
+ self.base_tall_p_col_xxx(5)
730
+
731
+ def test_tall_p_col_fat(self):
732
+ # 12x7 + 12x7 = 12x14 --> becomes fat
733
+ self.base_tall_p_col_xxx(7)
734
+
735
+ def test_fat_1_row(self):
736
+ a, q, r, u = self.generate('fat', which='row')
737
+ for row in range(r.shape[0] + 1):
738
+ q1, r1 = qr_insert(q, r, u, row)
739
+ a1 = np.insert(a, row, u, 0)
740
+ check_qr(q1, r1, a1, self.rtol, self.atol)
741
+
742
+ # for row adds to fat matrices there are three cases to test
743
+ # fat + prow --> fat
744
+ # fat + prow --> sqr
745
+ # fat + prow --> tall
746
+ def base_fat_p_row_xxx(self, p):
747
+ a, q, r, u = self.generate('fat', which='row', p=p)
748
+ for row in range(r.shape[0] + 1):
749
+ q1, r1 = qr_insert(q, r, u, row)
750
+ a1 = np.insert(a, np.full(p, row, np.intp), u, 0)
751
+ check_qr(q1, r1, a1, self.rtol, self.atol)
752
+
753
+ def test_fat_p_row_fat(self):
754
+ # 7x12 + 3x12 = 10x12 --> stays fat
755
+ self.base_fat_p_row_xxx(3)
756
+
757
+ def test_fat_p_row_sqr(self):
758
+ # 7x12 + 5x12 = 12x12 --> becomes sqr
759
+ self.base_fat_p_row_xxx(5)
760
+
761
+ def test_fat_p_row_tall(self):
762
+ # 7x12 + 7x12 = 14x12 --> becomes tall
763
+ self.base_fat_p_row_xxx(7)
764
+
765
+ def test_fat_1_col(self):
766
+ a, q, r, u = self.generate('fat', which='col')
767
+ for col in range(r.shape[1] + 1):
768
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
769
+ a1 = np.insert(a, col, u, 1)
770
+ check_qr(q1, r1, a1, self.rtol, self.atol)
771
+
772
+ def test_fat_p_col(self):
773
+ # fat + cols --> fat always
774
+ a, q, r, u = self.generate('fat', which='col', p=3)
775
+ for col in range(r.shape[1] + 1):
776
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
777
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
778
+ check_qr(q1, r1, a1, self.rtol, self.atol)
779
+
780
+ def test_economic_1_row(self):
781
+ a, q, r, u = self.generate('tall', 'economic', 'row')
782
+ for row in range(r.shape[0] + 1):
783
+ q1, r1 = qr_insert(q, r, u, row, overwrite_qru=False)
784
+ a1 = np.insert(a, row, u, 0)
785
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
786
+
787
+ def test_economic_p_row(self):
788
+ # tall + rows --> tall always
789
+ a, q, r, u = self.generate('tall', 'economic', 'row', 3)
790
+ for row in range(r.shape[0] + 1):
791
+ q1, r1 = qr_insert(q, r, u, row, overwrite_qru=False)
792
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
793
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
794
+
795
+ def test_economic_1_col(self):
796
+ a, q, r, u = self.generate('tall', 'economic', which='col')
797
+ for col in range(r.shape[1] + 1):
798
+ q1, r1 = qr_insert(q, r, u.copy(), col, 'col', overwrite_qru=False)
799
+ a1 = np.insert(a, col, u, 1)
800
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
801
+
802
+ def test_economic_1_col_bad_update(self):
803
+ # When the column to be added lies in the span of Q, the update is
804
+ # not meaningful. This is detected, and a LinAlgError is issued.
805
+ q = np.eye(5, 3, dtype=self.dtype)
806
+ r = np.eye(3, dtype=self.dtype)
807
+ u = np.array([1, 0, 0, 0, 0], self.dtype)
808
+ assert_raises(linalg.LinAlgError, qr_insert, q, r, u, 0, 'col')
809
+
810
+ # for column adds to economic matrices there are three cases to test
811
+ # eco + pcol --> eco
812
+ # eco + pcol --> sqr
813
+ # eco + pcol --> fat
814
+ def base_economic_p_col_xxx(self, p):
815
+ a, q, r, u = self.generate('tall', 'economic', which='col', p=p)
816
+ for col in range(r.shape[1] + 1):
817
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
818
+ a1 = np.insert(a, np.full(p, col, np.intp), u, 1)
819
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
820
+
821
+ def test_economic_p_col_eco(self):
822
+ # 12x7 + 12x3 = 12x10 --> stays eco
823
+ self.base_economic_p_col_xxx(3)
824
+
825
+ def test_economic_p_col_sqr(self):
826
+ # 12x7 + 12x5 = 12x12 --> becomes sqr
827
+ self.base_economic_p_col_xxx(5)
828
+
829
+ def test_economic_p_col_fat(self):
830
+ # 12x7 + 12x7 = 12x14 --> becomes fat
831
+ self.base_economic_p_col_xxx(7)
832
+
833
+ def test_Mx1_1_row(self):
834
+ a, q, r, u = self.generate('Mx1', which='row')
835
+ for row in range(r.shape[0] + 1):
836
+ q1, r1 = qr_insert(q, r, u, row)
837
+ a1 = np.insert(a, row, u, 0)
838
+ check_qr(q1, r1, a1, self.rtol, self.atol)
839
+
840
+ def test_Mx1_p_row(self):
841
+ a, q, r, u = self.generate('Mx1', which='row', p=3)
842
+ for row in range(r.shape[0] + 1):
843
+ q1, r1 = qr_insert(q, r, u, row)
844
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
845
+ check_qr(q1, r1, a1, self.rtol, self.atol)
846
+
847
+ def test_Mx1_1_col(self):
848
+ a, q, r, u = self.generate('Mx1', which='col')
849
+ for col in range(r.shape[1] + 1):
850
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
851
+ a1 = np.insert(a, col, u, 1)
852
+ check_qr(q1, r1, a1, self.rtol, self.atol)
853
+
854
+ def test_Mx1_p_col(self):
855
+ a, q, r, u = self.generate('Mx1', which='col', p=3)
856
+ for col in range(r.shape[1] + 1):
857
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
858
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
859
+ check_qr(q1, r1, a1, self.rtol, self.atol)
860
+
861
+ def test_Mx1_economic_1_row(self):
862
+ a, q, r, u = self.generate('Mx1', 'economic', 'row')
863
+ for row in range(r.shape[0] + 1):
864
+ q1, r1 = qr_insert(q, r, u, row)
865
+ a1 = np.insert(a, row, u, 0)
866
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
867
+
868
+ def test_Mx1_economic_p_row(self):
869
+ a, q, r, u = self.generate('Mx1', 'economic', 'row', 3)
870
+ for row in range(r.shape[0] + 1):
871
+ q1, r1 = qr_insert(q, r, u, row)
872
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
873
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
874
+
875
+ def test_Mx1_economic_1_col(self):
876
+ a, q, r, u = self.generate('Mx1', 'economic', 'col')
877
+ for col in range(r.shape[1] + 1):
878
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
879
+ a1 = np.insert(a, col, u, 1)
880
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
881
+
882
+ def test_Mx1_economic_p_col(self):
883
+ a, q, r, u = self.generate('Mx1', 'economic', 'col', 3)
884
+ for col in range(r.shape[1] + 1):
885
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
886
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
887
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
888
+
889
+ def test_1xN_1_row(self):
890
+ a, q, r, u = self.generate('1xN', which='row')
891
+ for row in range(r.shape[0] + 1):
892
+ q1, r1 = qr_insert(q, r, u, row)
893
+ a1 = np.insert(a, row, u, 0)
894
+ check_qr(q1, r1, a1, self.rtol, self.atol)
895
+
896
+ def test_1xN_p_row(self):
897
+ a, q, r, u = self.generate('1xN', which='row', p=3)
898
+ for row in range(r.shape[0] + 1):
899
+ q1, r1 = qr_insert(q, r, u, row)
900
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
901
+ check_qr(q1, r1, a1, self.rtol, self.atol)
902
+
903
+ def test_1xN_1_col(self):
904
+ a, q, r, u = self.generate('1xN', which='col')
905
+ for col in range(r.shape[1] + 1):
906
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
907
+ a1 = np.insert(a, col, u, 1)
908
+ check_qr(q1, r1, a1, self.rtol, self.atol)
909
+
910
+ def test_1xN_p_col(self):
911
+ a, q, r, u = self.generate('1xN', which='col', p=3)
912
+ for col in range(r.shape[1] + 1):
913
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
914
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
915
+ check_qr(q1, r1, a1, self.rtol, self.atol)
916
+
917
+ def test_1x1_1_row(self):
918
+ a, q, r, u = self.generate('1x1', which='row')
919
+ for row in range(r.shape[0] + 1):
920
+ q1, r1 = qr_insert(q, r, u, row)
921
+ a1 = np.insert(a, row, u, 0)
922
+ check_qr(q1, r1, a1, self.rtol, self.atol)
923
+
924
+ def test_1x1_p_row(self):
925
+ a, q, r, u = self.generate('1x1', which='row', p=3)
926
+ for row in range(r.shape[0] + 1):
927
+ q1, r1 = qr_insert(q, r, u, row)
928
+ a1 = np.insert(a, np.full(3, row, np.intp), u, 0)
929
+ check_qr(q1, r1, a1, self.rtol, self.atol)
930
+
931
+ def test_1x1_1_col(self):
932
+ a, q, r, u = self.generate('1x1', which='col')
933
+ for col in range(r.shape[1] + 1):
934
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
935
+ a1 = np.insert(a, col, u, 1)
936
+ check_qr(q1, r1, a1, self.rtol, self.atol)
937
+
938
+ def test_1x1_p_col(self):
939
+ a, q, r, u = self.generate('1x1', which='col', p=3)
940
+ for col in range(r.shape[1] + 1):
941
+ q1, r1 = qr_insert(q, r, u, col, 'col', overwrite_qru=False)
942
+ a1 = np.insert(a, np.full(3, col, np.intp), u, 1)
943
+ check_qr(q1, r1, a1, self.rtol, self.atol)
944
+
945
+ def test_1x1_1_scalar(self):
946
+ a, q, r, u = self.generate('1x1', which='row')
947
+ assert_raises(ValueError, qr_insert, q[0, 0], r, u, 0, 'row')
948
+ assert_raises(ValueError, qr_insert, q, r[0, 0], u, 0, 'row')
949
+ assert_raises(ValueError, qr_insert, q, r, u[0], 0, 'row')
950
+
951
+ assert_raises(ValueError, qr_insert, q[0, 0], r, u, 0, 'col')
952
+ assert_raises(ValueError, qr_insert, q, r[0, 0], u, 0, 'col')
953
+ assert_raises(ValueError, qr_insert, q, r, u[0], 0, 'col')
954
+
955
+ def base_non_simple_strides(self, adjust_strides, k, p, which):
956
+ for type in ['sqr', 'tall', 'fat']:
957
+ a, q0, r0, u0 = self.generate(type, which=which, p=p)
958
+ qs, rs, us = adjust_strides((q0, r0, u0))
959
+ if p == 1:
960
+ ai = np.insert(a, k, u0, 0 if which == 'row' else 1)
961
+ else:
962
+ ai = np.insert(a, np.full(p, k, np.intp),
963
+ u0 if which == 'row' else u0,
964
+ 0 if which == 'row' else 1)
965
+
966
+ # for each variable, q, r, u we try with it strided and
967
+ # overwrite=False. Then we try with overwrite=True. Nothing
968
+ # is checked to see if it can be overwritten, since only
969
+ # F ordered Q can be overwritten when adding columns.
970
+
971
+ q = q0.copy('F')
972
+ r = r0.copy('F')
973
+ u = u0.copy('F')
974
+ q1, r1 = qr_insert(qs, r, u, k, which, overwrite_qru=False)
975
+ check_qr(q1, r1, ai, self.rtol, self.atol)
976
+ q1o, r1o = qr_insert(qs, r, u, k, which, overwrite_qru=True)
977
+ check_qr(q1o, r1o, ai, self.rtol, self.atol)
978
+
979
+ q = q0.copy('F')
980
+ r = r0.copy('F')
981
+ u = u0.copy('F')
982
+ q2, r2 = qr_insert(q, rs, u, k, which, overwrite_qru=False)
983
+ check_qr(q2, r2, ai, self.rtol, self.atol)
984
+ q2o, r2o = qr_insert(q, rs, u, k, which, overwrite_qru=True)
985
+ check_qr(q2o, r2o, ai, self.rtol, self.atol)
986
+
987
+ q = q0.copy('F')
988
+ r = r0.copy('F')
989
+ u = u0.copy('F')
990
+ q3, r3 = qr_insert(q, r, us, k, which, overwrite_qru=False)
991
+ check_qr(q3, r3, ai, self.rtol, self.atol)
992
+ q3o, r3o = qr_insert(q, r, us, k, which, overwrite_qru=True)
993
+ check_qr(q3o, r3o, ai, self.rtol, self.atol)
994
+
995
+ q = q0.copy('F')
996
+ r = r0.copy('F')
997
+ u = u0.copy('F')
998
+ # since some of these were consumed above
999
+ qs, rs, us = adjust_strides((q, r, u))
1000
+ q5, r5 = qr_insert(qs, rs, us, k, which, overwrite_qru=False)
1001
+ check_qr(q5, r5, ai, self.rtol, self.atol)
1002
+ q5o, r5o = qr_insert(qs, rs, us, k, which, overwrite_qru=True)
1003
+ check_qr(q5o, r5o, ai, self.rtol, self.atol)
1004
+
1005
+ def test_non_unit_strides_1_row(self):
1006
+ self.base_non_simple_strides(make_strided, 0, 1, 'row')
1007
+
1008
+ def test_non_unit_strides_p_row(self):
1009
+ self.base_non_simple_strides(make_strided, 0, 3, 'row')
1010
+
1011
+ def test_non_unit_strides_1_col(self):
1012
+ self.base_non_simple_strides(make_strided, 0, 1, 'col')
1013
+
1014
+ def test_non_unit_strides_p_col(self):
1015
+ self.base_non_simple_strides(make_strided, 0, 3, 'col')
1016
+
1017
+ def test_neg_strides_1_row(self):
1018
+ self.base_non_simple_strides(negate_strides, 0, 1, 'row')
1019
+
1020
+ def test_neg_strides_p_row(self):
1021
+ self.base_non_simple_strides(negate_strides, 0, 3, 'row')
1022
+
1023
+ def test_neg_strides_1_col(self):
1024
+ self.base_non_simple_strides(negate_strides, 0, 1, 'col')
1025
+
1026
+ def test_neg_strides_p_col(self):
1027
+ self.base_non_simple_strides(negate_strides, 0, 3, 'col')
1028
+
1029
+ def test_non_itemsize_strides_1_row(self):
1030
+ self.base_non_simple_strides(nonitemsize_strides, 0, 1, 'row')
1031
+
1032
+ def test_non_itemsize_strides_p_row(self):
1033
+ self.base_non_simple_strides(nonitemsize_strides, 0, 3, 'row')
1034
+
1035
+ def test_non_itemsize_strides_1_col(self):
1036
+ self.base_non_simple_strides(nonitemsize_strides, 0, 1, 'col')
1037
+
1038
+ def test_non_itemsize_strides_p_col(self):
1039
+ self.base_non_simple_strides(nonitemsize_strides, 0, 3, 'col')
1040
+
1041
+ def test_non_native_byte_order_1_row(self):
1042
+ self.base_non_simple_strides(make_nonnative, 0, 1, 'row')
1043
+
1044
+ def test_non_native_byte_order_p_row(self):
1045
+ self.base_non_simple_strides(make_nonnative, 0, 3, 'row')
1046
+
1047
+ def test_non_native_byte_order_1_col(self):
1048
+ self.base_non_simple_strides(make_nonnative, 0, 1, 'col')
1049
+
1050
+ def test_non_native_byte_order_p_col(self):
1051
+ self.base_non_simple_strides(make_nonnative, 0, 3, 'col')
1052
+
1053
+ def test_overwrite_qu_rank_1(self):
1054
+ # when inserting rows, the size of both Q and R change, so only
1055
+ # column inserts can overwrite q. Only complex column inserts
1056
+ # with C ordered Q overwrite u. Any contiguous Q is overwritten
1057
+ # when inserting 1 column
1058
+ a, q0, r, u, = self.generate('sqr', which='col', p=1)
1059
+ q = q0.copy('C')
1060
+ u0 = u.copy()
1061
+ # don't overwrite
1062
+ q1, r1 = qr_insert(q, r, u, 0, 'col', overwrite_qru=False)
1063
+ a1 = np.insert(a, 0, u0, 1)
1064
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1065
+ check_qr(q, r, a, self.rtol, self.atol)
1066
+
1067
+ # try overwriting
1068
+ q2, r2 = qr_insert(q, r, u, 0, 'col', overwrite_qru=True)
1069
+ check_qr(q2, r2, a1, self.rtol, self.atol)
1070
+ # verify the overwriting
1071
+ assert_allclose(q2, q, rtol=self.rtol, atol=self.atol)
1072
+ assert_allclose(u, u0.conj(), self.rtol, self.atol)
1073
+
1074
+ # now try with a fortran ordered Q
1075
+ qF = q0.copy('F')
1076
+ u1 = u0.copy()
1077
+ q3, r3 = qr_insert(qF, r, u1, 0, 'col', overwrite_qru=False)
1078
+ check_qr(q3, r3, a1, self.rtol, self.atol)
1079
+ check_qr(qF, r, a, self.rtol, self.atol)
1080
+
1081
+ # try overwriting
1082
+ q4, r4 = qr_insert(qF, r, u1, 0, 'col', overwrite_qru=True)
1083
+ check_qr(q4, r4, a1, self.rtol, self.atol)
1084
+ assert_allclose(q4, qF, rtol=self.rtol, atol=self.atol)
1085
+
1086
+ def test_overwrite_qu_rank_p(self):
1087
+ # when inserting rows, the size of both Q and R change, so only
1088
+ # column inserts can potentially overwrite Q. In practice, only
1089
+ # F ordered Q are overwritten with a rank p update.
1090
+ a, q0, r, u, = self.generate('sqr', which='col', p=3)
1091
+ q = q0.copy('F')
1092
+ a1 = np.insert(a, np.zeros(3, np.intp), u, 1)
1093
+
1094
+ # don't overwrite
1095
+ q1, r1 = qr_insert(q, r, u, 0, 'col', overwrite_qru=False)
1096
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1097
+ check_qr(q, r, a, self.rtol, self.atol)
1098
+
1099
+ # try overwriting
1100
+ q2, r2 = qr_insert(q, r, u, 0, 'col', overwrite_qru=True)
1101
+ check_qr(q2, r2, a1, self.rtol, self.atol)
1102
+ assert_allclose(q2, q, rtol=self.rtol, atol=self.atol)
1103
+
1104
+ def test_empty_inputs(self):
1105
+ a, q, r, u = self.generate('sqr', which='row')
1106
+ assert_raises(ValueError, qr_insert, np.array([]), r, u, 0, 'row')
1107
+ assert_raises(ValueError, qr_insert, q, np.array([]), u, 0, 'row')
1108
+ assert_raises(ValueError, qr_insert, q, r, np.array([]), 0, 'row')
1109
+ assert_raises(ValueError, qr_insert, np.array([]), r, u, 0, 'col')
1110
+ assert_raises(ValueError, qr_insert, q, np.array([]), u, 0, 'col')
1111
+ assert_raises(ValueError, qr_insert, q, r, np.array([]), 0, 'col')
1112
+
1113
+ def test_mismatched_shapes(self):
1114
+ a, q, r, u = self.generate('tall', which='row')
1115
+ assert_raises(ValueError, qr_insert, q, r[1:], u, 0, 'row')
1116
+ assert_raises(ValueError, qr_insert, q[:-2], r, u, 0, 'row')
1117
+ assert_raises(ValueError, qr_insert, q, r, u[1:], 0, 'row')
1118
+ assert_raises(ValueError, qr_insert, q, r[1:], u, 0, 'col')
1119
+ assert_raises(ValueError, qr_insert, q[:-2], r, u, 0, 'col')
1120
+ assert_raises(ValueError, qr_insert, q, r, u[1:], 0, 'col')
1121
+
1122
+ def test_unsupported_dtypes(self):
1123
+ dts = ['int8', 'int16', 'int32', 'int64',
1124
+ 'uint8', 'uint16', 'uint32', 'uint64',
1125
+ 'float16', 'longdouble', 'clongdouble',
1126
+ 'bool']
1127
+ a, q0, r0, u0 = self.generate('sqr', which='row')
1128
+ for dtype in dts:
1129
+ q = q0.real.astype(dtype)
1130
+ with np.errstate(invalid="ignore"):
1131
+ r = r0.real.astype(dtype)
1132
+ u = u0.real.astype(dtype)
1133
+ assert_raises(ValueError, qr_insert, q, r0, u0, 0, 'row')
1134
+ assert_raises(ValueError, qr_insert, q, r0, u0, 0, 'col')
1135
+ assert_raises(ValueError, qr_insert, q0, r, u0, 0, 'row')
1136
+ assert_raises(ValueError, qr_insert, q0, r, u0, 0, 'col')
1137
+ assert_raises(ValueError, qr_insert, q0, r0, u, 0, 'row')
1138
+ assert_raises(ValueError, qr_insert, q0, r0, u, 0, 'col')
1139
+
1140
+ def test_check_finite(self):
1141
+ a0, q0, r0, u0 = self.generate('sqr', which='row', p=3)
1142
+
1143
+ q = q0.copy('F')
1144
+ q[1,1] = np.nan
1145
+ assert_raises(ValueError, qr_insert, q, r0, u0[:,0], 0, 'row')
1146
+ assert_raises(ValueError, qr_insert, q, r0, u0, 0, 'row')
1147
+ assert_raises(ValueError, qr_insert, q, r0, u0[:,0], 0, 'col')
1148
+ assert_raises(ValueError, qr_insert, q, r0, u0, 0, 'col')
1149
+
1150
+ r = r0.copy('F')
1151
+ r[1,1] = np.nan
1152
+ assert_raises(ValueError, qr_insert, q0, r, u0[:,0], 0, 'row')
1153
+ assert_raises(ValueError, qr_insert, q0, r, u0, 0, 'row')
1154
+ assert_raises(ValueError, qr_insert, q0, r, u0[:,0], 0, 'col')
1155
+ assert_raises(ValueError, qr_insert, q0, r, u0, 0, 'col')
1156
+
1157
+ u = u0.copy('F')
1158
+ u[0,0] = np.nan
1159
+ assert_raises(ValueError, qr_insert, q0, r0, u[:,0], 0, 'row')
1160
+ assert_raises(ValueError, qr_insert, q0, r0, u, 0, 'row')
1161
+ assert_raises(ValueError, qr_insert, q0, r0, u[:,0], 0, 'col')
1162
+ assert_raises(ValueError, qr_insert, q0, r0, u, 0, 'col')
1163
+
1164
+ class TestQRinsert_f(BaseQRinsert):
1165
+ dtype = np.dtype('f')
1166
+
1167
+ class TestQRinsert_F(BaseQRinsert):
1168
+ dtype = np.dtype('F')
1169
+
1170
+ class TestQRinsert_d(BaseQRinsert):
1171
+ dtype = np.dtype('d')
1172
+
1173
+ class TestQRinsert_D(BaseQRinsert):
1174
+ dtype = np.dtype('D')
1175
+
1176
+ class BaseQRupdate(BaseQRdeltas):
1177
+ def generate(self, type, mode='full', p=1):
1178
+ a, q, r = super().generate(type, mode)
1179
+
1180
+ # super call set the seed...
1181
+ if p == 1:
1182
+ u = np.random.random(q.shape[0])
1183
+ v = np.random.random(r.shape[1])
1184
+ else:
1185
+ u = np.random.random((q.shape[0], p))
1186
+ v = np.random.random((r.shape[1], p))
1187
+
1188
+ if np.iscomplexobj(self.dtype.type(1)):
1189
+ b = np.random.random(u.shape)
1190
+ u = u + 1j * b
1191
+
1192
+ c = np.random.random(v.shape)
1193
+ v = v + 1j * c
1194
+
1195
+ u = u.astype(self.dtype)
1196
+ v = v.astype(self.dtype)
1197
+ return a, q, r, u, v
1198
+
1199
+ def test_sqr_rank_1(self):
1200
+ a, q, r, u, v = self.generate('sqr')
1201
+ q1, r1 = qr_update(q, r, u, v, False)
1202
+ a1 = a + np.outer(u, v.conj())
1203
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1204
+
1205
+ def test_sqr_rank_p(self):
1206
+ # test ndim = 2, rank 1 updates here too
1207
+ for p in [1, 2, 3, 5]:
1208
+ a, q, r, u, v = self.generate('sqr', p=p)
1209
+ if p == 1:
1210
+ u = u.reshape(u.size, 1)
1211
+ v = v.reshape(v.size, 1)
1212
+ q1, r1 = qr_update(q, r, u, v, False)
1213
+ a1 = a + np.dot(u, v.T.conj())
1214
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1215
+
1216
+ def test_tall_rank_1(self):
1217
+ a, q, r, u, v = self.generate('tall')
1218
+ q1, r1 = qr_update(q, r, u, v, False)
1219
+ a1 = a + np.outer(u, v.conj())
1220
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1221
+
1222
+ def test_tall_rank_p(self):
1223
+ for p in [1, 2, 3, 5]:
1224
+ a, q, r, u, v = self.generate('tall', p=p)
1225
+ if p == 1:
1226
+ u = u.reshape(u.size, 1)
1227
+ v = v.reshape(v.size, 1)
1228
+ q1, r1 = qr_update(q, r, u, v, False)
1229
+ a1 = a + np.dot(u, v.T.conj())
1230
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1231
+
1232
+ def test_fat_rank_1(self):
1233
+ a, q, r, u, v = self.generate('fat')
1234
+ q1, r1 = qr_update(q, r, u, v, False)
1235
+ a1 = a + np.outer(u, v.conj())
1236
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1237
+
1238
+ def test_fat_rank_p(self):
1239
+ for p in [1, 2, 3, 5]:
1240
+ a, q, r, u, v = self.generate('fat', p=p)
1241
+ if p == 1:
1242
+ u = u.reshape(u.size, 1)
1243
+ v = v.reshape(v.size, 1)
1244
+ q1, r1 = qr_update(q, r, u, v, False)
1245
+ a1 = a + np.dot(u, v.T.conj())
1246
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1247
+
1248
+ def test_economic_rank_1(self):
1249
+ a, q, r, u, v = self.generate('tall', 'economic')
1250
+ q1, r1 = qr_update(q, r, u, v, False)
1251
+ a1 = a + np.outer(u, v.conj())
1252
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1253
+
1254
+ def test_economic_rank_p(self):
1255
+ for p in [1, 2, 3, 5]:
1256
+ a, q, r, u, v = self.generate('tall', 'economic', p)
1257
+ if p == 1:
1258
+ u = u.reshape(u.size, 1)
1259
+ v = v.reshape(v.size, 1)
1260
+ q1, r1 = qr_update(q, r, u, v, False)
1261
+ a1 = a + np.dot(u, v.T.conj())
1262
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1263
+
1264
+ def test_Mx1_rank_1(self):
1265
+ a, q, r, u, v = self.generate('Mx1')
1266
+ q1, r1 = qr_update(q, r, u, v, False)
1267
+ a1 = a + np.outer(u, v.conj())
1268
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1269
+
1270
+ def test_Mx1_rank_p(self):
1271
+ # when M or N == 1, only a rank 1 update is allowed. This isn't
1272
+ # fundamental limitation, but the code does not support it.
1273
+ a, q, r, u, v = self.generate('Mx1', p=1)
1274
+ u = u.reshape(u.size, 1)
1275
+ v = v.reshape(v.size, 1)
1276
+ q1, r1 = qr_update(q, r, u, v, False)
1277
+ a1 = a + np.dot(u, v.T.conj())
1278
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1279
+
1280
+ def test_Mx1_economic_rank_1(self):
1281
+ a, q, r, u, v = self.generate('Mx1', 'economic')
1282
+ q1, r1 = qr_update(q, r, u, v, False)
1283
+ a1 = a + np.outer(u, v.conj())
1284
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1285
+
1286
+ def test_Mx1_economic_rank_p(self):
1287
+ # when M or N == 1, only a rank 1 update is allowed. This isn't
1288
+ # fundamental limitation, but the code does not support it.
1289
+ a, q, r, u, v = self.generate('Mx1', 'economic', p=1)
1290
+ u = u.reshape(u.size, 1)
1291
+ v = v.reshape(v.size, 1)
1292
+ q1, r1 = qr_update(q, r, u, v, False)
1293
+ a1 = a + np.dot(u, v.T.conj())
1294
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1295
+
1296
+ def test_1xN_rank_1(self):
1297
+ a, q, r, u, v = self.generate('1xN')
1298
+ q1, r1 = qr_update(q, r, u, v, False)
1299
+ a1 = a + np.outer(u, v.conj())
1300
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1301
+
1302
+ def test_1xN_rank_p(self):
1303
+ # when M or N == 1, only a rank 1 update is allowed. This isn't
1304
+ # fundamental limitation, but the code does not support it.
1305
+ a, q, r, u, v = self.generate('1xN', p=1)
1306
+ u = u.reshape(u.size, 1)
1307
+ v = v.reshape(v.size, 1)
1308
+ q1, r1 = qr_update(q, r, u, v, False)
1309
+ a1 = a + np.dot(u, v.T.conj())
1310
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1311
+
1312
+ def test_1x1_rank_1(self):
1313
+ a, q, r, u, v = self.generate('1x1')
1314
+ q1, r1 = qr_update(q, r, u, v, False)
1315
+ a1 = a + np.outer(u, v.conj())
1316
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1317
+
1318
+ def test_1x1_rank_p(self):
1319
+ # when M or N == 1, only a rank 1 update is allowed. This isn't
1320
+ # fundamental limitation, but the code does not support it.
1321
+ a, q, r, u, v = self.generate('1x1', p=1)
1322
+ u = u.reshape(u.size, 1)
1323
+ v = v.reshape(v.size, 1)
1324
+ q1, r1 = qr_update(q, r, u, v, False)
1325
+ a1 = a + np.dot(u, v.T.conj())
1326
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1327
+
1328
+ def test_1x1_rank_1_scalar(self):
1329
+ a, q, r, u, v = self.generate('1x1')
1330
+ assert_raises(ValueError, qr_update, q[0, 0], r, u, v)
1331
+ assert_raises(ValueError, qr_update, q, r[0, 0], u, v)
1332
+ assert_raises(ValueError, qr_update, q, r, u[0], v)
1333
+ assert_raises(ValueError, qr_update, q, r, u, v[0])
1334
+
1335
+ def base_non_simple_strides(self, adjust_strides, mode, p, overwriteable):
1336
+ assert_sqr = False if mode == 'economic' else True
1337
+ for type in ['sqr', 'tall', 'fat']:
1338
+ a, q0, r0, u0, v0 = self.generate(type, mode, p)
1339
+ qs, rs, us, vs = adjust_strides((q0, r0, u0, v0))
1340
+ if p == 1:
1341
+ aup = a + np.outer(u0, v0.conj())
1342
+ else:
1343
+ aup = a + np.dot(u0, v0.T.conj())
1344
+
1345
+ # for each variable, q, r, u, v we try with it strided and
1346
+ # overwrite=False. Then we try with overwrite=True, and make
1347
+ # sure that if p == 1, r and v are still overwritten.
1348
+ # a strided q and u must always be copied.
1349
+
1350
+ q = q0.copy('F')
1351
+ r = r0.copy('F')
1352
+ u = u0.copy('F')
1353
+ v = v0.copy('C')
1354
+ q1, r1 = qr_update(qs, r, u, v, False)
1355
+ check_qr(q1, r1, aup, self.rtol, self.atol, assert_sqr)
1356
+ q1o, r1o = qr_update(qs, r, u, v, True)
1357
+ check_qr(q1o, r1o, aup, self.rtol, self.atol, assert_sqr)
1358
+ if overwriteable:
1359
+ assert_allclose(r1o, r, rtol=self.rtol, atol=self.atol)
1360
+ assert_allclose(v, v0.conj(), rtol=self.rtol, atol=self.atol)
1361
+
1362
+ q = q0.copy('F')
1363
+ r = r0.copy('F')
1364
+ u = u0.copy('F')
1365
+ v = v0.copy('C')
1366
+ q2, r2 = qr_update(q, rs, u, v, False)
1367
+ check_qr(q2, r2, aup, self.rtol, self.atol, assert_sqr)
1368
+ q2o, r2o = qr_update(q, rs, u, v, True)
1369
+ check_qr(q2o, r2o, aup, self.rtol, self.atol, assert_sqr)
1370
+ if overwriteable:
1371
+ assert_allclose(r2o, rs, rtol=self.rtol, atol=self.atol)
1372
+ assert_allclose(v, v0.conj(), rtol=self.rtol, atol=self.atol)
1373
+
1374
+ q = q0.copy('F')
1375
+ r = r0.copy('F')
1376
+ u = u0.copy('F')
1377
+ v = v0.copy('C')
1378
+ q3, r3 = qr_update(q, r, us, v, False)
1379
+ check_qr(q3, r3, aup, self.rtol, self.atol, assert_sqr)
1380
+ q3o, r3o = qr_update(q, r, us, v, True)
1381
+ check_qr(q3o, r3o, aup, self.rtol, self.atol, assert_sqr)
1382
+ if overwriteable:
1383
+ assert_allclose(r3o, r, rtol=self.rtol, atol=self.atol)
1384
+ assert_allclose(v, v0.conj(), rtol=self.rtol, atol=self.atol)
1385
+
1386
+ q = q0.copy('F')
1387
+ r = r0.copy('F')
1388
+ u = u0.copy('F')
1389
+ v = v0.copy('C')
1390
+ q4, r4 = qr_update(q, r, u, vs, False)
1391
+ check_qr(q4, r4, aup, self.rtol, self.atol, assert_sqr)
1392
+ q4o, r4o = qr_update(q, r, u, vs, True)
1393
+ check_qr(q4o, r4o, aup, self.rtol, self.atol, assert_sqr)
1394
+ if overwriteable:
1395
+ assert_allclose(r4o, r, rtol=self.rtol, atol=self.atol)
1396
+ assert_allclose(vs, v0.conj(), rtol=self.rtol, atol=self.atol)
1397
+
1398
+ q = q0.copy('F')
1399
+ r = r0.copy('F')
1400
+ u = u0.copy('F')
1401
+ v = v0.copy('C')
1402
+ # since some of these were consumed above
1403
+ qs, rs, us, vs = adjust_strides((q, r, u, v))
1404
+ q5, r5 = qr_update(qs, rs, us, vs, False)
1405
+ check_qr(q5, r5, aup, self.rtol, self.atol, assert_sqr)
1406
+ q5o, r5o = qr_update(qs, rs, us, vs, True)
1407
+ check_qr(q5o, r5o, aup, self.rtol, self.atol, assert_sqr)
1408
+ if overwriteable:
1409
+ assert_allclose(r5o, rs, rtol=self.rtol, atol=self.atol)
1410
+ assert_allclose(vs, v0.conj(), rtol=self.rtol, atol=self.atol)
1411
+
1412
+ def test_non_unit_strides_rank_1(self):
1413
+ self.base_non_simple_strides(make_strided, 'full', 1, True)
1414
+
1415
+ def test_non_unit_strides_economic_rank_1(self):
1416
+ self.base_non_simple_strides(make_strided, 'economic', 1, True)
1417
+
1418
+ def test_non_unit_strides_rank_p(self):
1419
+ self.base_non_simple_strides(make_strided, 'full', 3, False)
1420
+
1421
+ def test_non_unit_strides_economic_rank_p(self):
1422
+ self.base_non_simple_strides(make_strided, 'economic', 3, False)
1423
+
1424
+ def test_neg_strides_rank_1(self):
1425
+ self.base_non_simple_strides(negate_strides, 'full', 1, False)
1426
+
1427
+ def test_neg_strides_economic_rank_1(self):
1428
+ self.base_non_simple_strides(negate_strides, 'economic', 1, False)
1429
+
1430
+ def test_neg_strides_rank_p(self):
1431
+ self.base_non_simple_strides(negate_strides, 'full', 3, False)
1432
+
1433
+ def test_neg_strides_economic_rank_p(self):
1434
+ self.base_non_simple_strides(negate_strides, 'economic', 3, False)
1435
+
1436
+ def test_non_itemsize_strides_rank_1(self):
1437
+ self.base_non_simple_strides(nonitemsize_strides, 'full', 1, False)
1438
+
1439
+ def test_non_itemsize_strides_economic_rank_1(self):
1440
+ self.base_non_simple_strides(nonitemsize_strides, 'economic', 1, False)
1441
+
1442
+ def test_non_itemsize_strides_rank_p(self):
1443
+ self.base_non_simple_strides(nonitemsize_strides, 'full', 3, False)
1444
+
1445
+ def test_non_itemsize_strides_economic_rank_p(self):
1446
+ self.base_non_simple_strides(nonitemsize_strides, 'economic', 3, False)
1447
+
1448
+ def test_non_native_byte_order_rank_1(self):
1449
+ self.base_non_simple_strides(make_nonnative, 'full', 1, False)
1450
+
1451
+ def test_non_native_byte_order_economic_rank_1(self):
1452
+ self.base_non_simple_strides(make_nonnative, 'economic', 1, False)
1453
+
1454
+ def test_non_native_byte_order_rank_p(self):
1455
+ self.base_non_simple_strides(make_nonnative, 'full', 3, False)
1456
+
1457
+ def test_non_native_byte_order_economic_rank_p(self):
1458
+ self.base_non_simple_strides(make_nonnative, 'economic', 3, False)
1459
+
1460
+ def test_overwrite_qruv_rank_1(self):
1461
+ # Any positive strided q, r, u, and v can be overwritten for a rank 1
1462
+ # update, only checking C and F contiguous.
1463
+ a, q0, r0, u0, v0 = self.generate('sqr')
1464
+ a1 = a + np.outer(u0, v0.conj())
1465
+ q = q0.copy('F')
1466
+ r = r0.copy('F')
1467
+ u = u0.copy('F')
1468
+ v = v0.copy('F')
1469
+
1470
+ # don't overwrite
1471
+ q1, r1 = qr_update(q, r, u, v, False)
1472
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1473
+ check_qr(q, r, a, self.rtol, self.atol)
1474
+
1475
+ q2, r2 = qr_update(q, r, u, v, True)
1476
+ check_qr(q2, r2, a1, self.rtol, self.atol)
1477
+ # verify the overwriting, no good way to check u and v.
1478
+ assert_allclose(q2, q, rtol=self.rtol, atol=self.atol)
1479
+ assert_allclose(r2, r, rtol=self.rtol, atol=self.atol)
1480
+
1481
+ q = q0.copy('C')
1482
+ r = r0.copy('C')
1483
+ u = u0.copy('C')
1484
+ v = v0.copy('C')
1485
+ q3, r3 = qr_update(q, r, u, v, True)
1486
+ check_qr(q3, r3, a1, self.rtol, self.atol)
1487
+ assert_allclose(q3, q, rtol=self.rtol, atol=self.atol)
1488
+ assert_allclose(r3, r, rtol=self.rtol, atol=self.atol)
1489
+
1490
+ def test_overwrite_qruv_rank_1_economic(self):
1491
+ # updating economic decompositions can overwrite any contiguous r,
1492
+ # and positively strided r and u. V is only ever read.
1493
+ # only checking C and F contiguous.
1494
+ a, q0, r0, u0, v0 = self.generate('tall', 'economic')
1495
+ a1 = a + np.outer(u0, v0.conj())
1496
+ q = q0.copy('F')
1497
+ r = r0.copy('F')
1498
+ u = u0.copy('F')
1499
+ v = v0.copy('F')
1500
+
1501
+ # don't overwrite
1502
+ q1, r1 = qr_update(q, r, u, v, False)
1503
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1504
+ check_qr(q, r, a, self.rtol, self.atol, False)
1505
+
1506
+ q2, r2 = qr_update(q, r, u, v, True)
1507
+ check_qr(q2, r2, a1, self.rtol, self.atol, False)
1508
+ # verify the overwriting, no good way to check u and v.
1509
+ assert_allclose(q2, q, rtol=self.rtol, atol=self.atol)
1510
+ assert_allclose(r2, r, rtol=self.rtol, atol=self.atol)
1511
+
1512
+ q = q0.copy('C')
1513
+ r = r0.copy('C')
1514
+ u = u0.copy('C')
1515
+ v = v0.copy('C')
1516
+ q3, r3 = qr_update(q, r, u, v, True)
1517
+ check_qr(q3, r3, a1, self.rtol, self.atol, False)
1518
+ assert_allclose(q3, q, rtol=self.rtol, atol=self.atol)
1519
+ assert_allclose(r3, r, rtol=self.rtol, atol=self.atol)
1520
+
1521
+ def test_overwrite_qruv_rank_p(self):
1522
+ # for rank p updates, q r must be F contiguous, v must be C (v.T --> F)
1523
+ # and u can be C or F, but is only overwritten if Q is C and complex
1524
+ a, q0, r0, u0, v0 = self.generate('sqr', p=3)
1525
+ a1 = a + np.dot(u0, v0.T.conj())
1526
+ q = q0.copy('F')
1527
+ r = r0.copy('F')
1528
+ u = u0.copy('F')
1529
+ v = v0.copy('C')
1530
+
1531
+ # don't overwrite
1532
+ q1, r1 = qr_update(q, r, u, v, False)
1533
+ check_qr(q1, r1, a1, self.rtol, self.atol)
1534
+ check_qr(q, r, a, self.rtol, self.atol)
1535
+
1536
+ q2, r2 = qr_update(q, r, u, v, True)
1537
+ check_qr(q2, r2, a1, self.rtol, self.atol)
1538
+ # verify the overwriting, no good way to check u and v.
1539
+ assert_allclose(q2, q, rtol=self.rtol, atol=self.atol)
1540
+ assert_allclose(r2, r, rtol=self.rtol, atol=self.atol)
1541
+
1542
+ def test_empty_inputs(self):
1543
+ a, q, r, u, v = self.generate('tall')
1544
+ assert_raises(ValueError, qr_update, np.array([]), r, u, v)
1545
+ assert_raises(ValueError, qr_update, q, np.array([]), u, v)
1546
+ assert_raises(ValueError, qr_update, q, r, np.array([]), v)
1547
+ assert_raises(ValueError, qr_update, q, r, u, np.array([]))
1548
+
1549
+ def test_mismatched_shapes(self):
1550
+ a, q, r, u, v = self.generate('tall')
1551
+ assert_raises(ValueError, qr_update, q, r[1:], u, v)
1552
+ assert_raises(ValueError, qr_update, q[:-2], r, u, v)
1553
+ assert_raises(ValueError, qr_update, q, r, u[1:], v)
1554
+ assert_raises(ValueError, qr_update, q, r, u, v[1:])
1555
+
1556
+ def test_unsupported_dtypes(self):
1557
+ dts = ['int8', 'int16', 'int32', 'int64',
1558
+ 'uint8', 'uint16', 'uint32', 'uint64',
1559
+ 'float16', 'longdouble', 'clongdouble',
1560
+ 'bool']
1561
+ a, q0, r0, u0, v0 = self.generate('tall')
1562
+ for dtype in dts:
1563
+ q = q0.real.astype(dtype)
1564
+ with np.errstate(invalid="ignore"):
1565
+ r = r0.real.astype(dtype)
1566
+ u = u0.real.astype(dtype)
1567
+ v = v0.real.astype(dtype)
1568
+ assert_raises(ValueError, qr_update, q, r0, u0, v0)
1569
+ assert_raises(ValueError, qr_update, q0, r, u0, v0)
1570
+ assert_raises(ValueError, qr_update, q0, r0, u, v0)
1571
+ assert_raises(ValueError, qr_update, q0, r0, u0, v)
1572
+
1573
+ def test_integer_input(self):
1574
+ q = np.arange(16).reshape(4, 4)
1575
+ r = q.copy() # doesn't matter
1576
+ u = q[:, 0].copy()
1577
+ v = r[0, :].copy()
1578
+ assert_raises(ValueError, qr_update, q, r, u, v)
1579
+
1580
+ def test_check_finite(self):
1581
+ a0, q0, r0, u0, v0 = self.generate('tall', p=3)
1582
+
1583
+ q = q0.copy('F')
1584
+ q[1,1] = np.nan
1585
+ assert_raises(ValueError, qr_update, q, r0, u0[:,0], v0[:,0])
1586
+ assert_raises(ValueError, qr_update, q, r0, u0, v0)
1587
+
1588
+ r = r0.copy('F')
1589
+ r[1,1] = np.nan
1590
+ assert_raises(ValueError, qr_update, q0, r, u0[:,0], v0[:,0])
1591
+ assert_raises(ValueError, qr_update, q0, r, u0, v0)
1592
+
1593
+ u = u0.copy('F')
1594
+ u[0,0] = np.nan
1595
+ assert_raises(ValueError, qr_update, q0, r0, u[:,0], v0[:,0])
1596
+ assert_raises(ValueError, qr_update, q0, r0, u, v0)
1597
+
1598
+ v = v0.copy('F')
1599
+ v[0,0] = np.nan
1600
+ assert_raises(ValueError, qr_update, q0, r0, u[:,0], v[:,0])
1601
+ assert_raises(ValueError, qr_update, q0, r0, u, v)
1602
+
1603
+ def test_economic_check_finite(self):
1604
+ a0, q0, r0, u0, v0 = self.generate('tall', mode='economic', p=3)
1605
+
1606
+ q = q0.copy('F')
1607
+ q[1,1] = np.nan
1608
+ assert_raises(ValueError, qr_update, q, r0, u0[:,0], v0[:,0])
1609
+ assert_raises(ValueError, qr_update, q, r0, u0, v0)
1610
+
1611
+ r = r0.copy('F')
1612
+ r[1,1] = np.nan
1613
+ assert_raises(ValueError, qr_update, q0, r, u0[:,0], v0[:,0])
1614
+ assert_raises(ValueError, qr_update, q0, r, u0, v0)
1615
+
1616
+ u = u0.copy('F')
1617
+ u[0,0] = np.nan
1618
+ assert_raises(ValueError, qr_update, q0, r0, u[:,0], v0[:,0])
1619
+ assert_raises(ValueError, qr_update, q0, r0, u, v0)
1620
+
1621
+ v = v0.copy('F')
1622
+ v[0,0] = np.nan
1623
+ assert_raises(ValueError, qr_update, q0, r0, u[:,0], v[:,0])
1624
+ assert_raises(ValueError, qr_update, q0, r0, u, v)
1625
+
1626
+ def test_u_exactly_in_span_q(self):
1627
+ q = np.array([[0, 0], [0, 0], [1, 0], [0, 1]], self.dtype)
1628
+ r = np.array([[1, 0], [0, 1]], self.dtype)
1629
+ u = np.array([0, 0, 0, -1], self.dtype)
1630
+ v = np.array([1, 2], self.dtype)
1631
+ q1, r1 = qr_update(q, r, u, v)
1632
+ a1 = np.dot(q, r) + np.outer(u, v.conj())
1633
+ check_qr(q1, r1, a1, self.rtol, self.atol, False)
1634
+
1635
+ class TestQRupdate_f(BaseQRupdate):
1636
+ dtype = np.dtype('f')
1637
+
1638
+ class TestQRupdate_F(BaseQRupdate):
1639
+ dtype = np.dtype('F')
1640
+
1641
+ class TestQRupdate_d(BaseQRupdate):
1642
+ dtype = np.dtype('d')
1643
+
1644
+ class TestQRupdate_D(BaseQRupdate):
1645
+ dtype = np.dtype('D')
1646
+
1647
+ def test_form_qTu():
1648
+ # We want to ensure that all of the code paths through this function are
1649
+ # tested. Most of them should be hit with the rest of test suite, but
1650
+ # explicit tests make clear precisely what is being tested.
1651
+ #
1652
+ # This function expects that Q is either C or F contiguous and square.
1653
+ # Economic mode decompositions (Q is (M, N), M != N) do not go through this
1654
+ # function. U may have any positive strides.
1655
+ #
1656
+ # Some of these test are duplicates, since contiguous 1d arrays are both C
1657
+ # and F.
1658
+
1659
+ q_order = ['F', 'C']
1660
+ q_shape = [(8, 8), ]
1661
+ u_order = ['F', 'C', 'A'] # here A means is not F not C
1662
+ u_shape = [1, 3]
1663
+ dtype = ['f', 'd', 'F', 'D']
1664
+
1665
+ for qo, qs, uo, us, d in \
1666
+ itertools.product(q_order, q_shape, u_order, u_shape, dtype):
1667
+ if us == 1:
1668
+ check_form_qTu(qo, qs, uo, us, 1, d)
1669
+ check_form_qTu(qo, qs, uo, us, 2, d)
1670
+ else:
1671
+ check_form_qTu(qo, qs, uo, us, 2, d)
1672
+
1673
+ def check_form_qTu(q_order, q_shape, u_order, u_shape, u_ndim, dtype):
1674
+ np.random.seed(47)
1675
+ if u_shape == 1 and u_ndim == 1:
1676
+ u_shape = (q_shape[0],)
1677
+ else:
1678
+ u_shape = (q_shape[0], u_shape)
1679
+ dtype = np.dtype(dtype)
1680
+
1681
+ if dtype.char in 'fd':
1682
+ q = np.random.random(q_shape)
1683
+ u = np.random.random(u_shape)
1684
+ elif dtype.char in 'FD':
1685
+ q = np.random.random(q_shape) + 1j*np.random.random(q_shape)
1686
+ u = np.random.random(u_shape) + 1j*np.random.random(u_shape)
1687
+ else:
1688
+ ValueError("form_qTu doesn't support this dtype")
1689
+
1690
+ q = np.require(q, dtype, q_order)
1691
+ if u_order != 'A':
1692
+ u = np.require(u, dtype, u_order)
1693
+ else:
1694
+ u, = make_strided((u.astype(dtype),))
1695
+
1696
+ rtol = 10.0 ** -(np.finfo(dtype).precision-2)
1697
+ atol = 2*np.finfo(dtype).eps
1698
+
1699
+ expected = np.dot(q.T.conj(), u)
1700
+ res = _decomp_update._form_qTu(q, u)
1701
+ assert_allclose(res, expected, rtol=rtol, atol=atol)