scipy 1.16.2__cp313-cp313-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.cp313-win_arm64.lib +0 -0
  4. scipy/_cyutility.cp313-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.cp313-win_arm64.lib +0 -0
  13. scipy/_lib/_ccallback_c.cp313-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.cp313-win_arm64.lib +0 -0
  18. scipy/_lib/_fpumode.cp313-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.cp313-win_arm64.lib +0 -0
  23. scipy/_lib/_test_ccallback.cp313-win_arm64.pyd +0 -0
  24. scipy/_lib/_test_deprecation_call.cp313-win_arm64.lib +0 -0
  25. scipy/_lib/_test_deprecation_call.cp313-win_arm64.pyd +0 -0
  26. scipy/_lib/_test_deprecation_def.cp313-win_arm64.lib +0 -0
  27. scipy/_lib/_test_deprecation_def.cp313-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.cp313-win_arm64.lib +0 -0
  35. scipy/_lib/_uarray/_uarray.cp313-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.cp313-win_arm64.lib +0 -0
  101. scipy/_lib/messagestream.cp313-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.cp313-win_arm64.lib +0 -0
  148. scipy/cluster/_hierarchy.cp313-win_arm64.pyd +0 -0
  149. scipy/cluster/_optimal_leaf_ordering.cp313-win_arm64.lib +0 -0
  150. scipy/cluster/_optimal_leaf_ordering.cp313-win_arm64.pyd +0 -0
  151. scipy/cluster/_vq.cp313-win_arm64.lib +0 -0
  152. scipy/cluster/_vq.cp313-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.cp313-win_arm64.lib +0 -0
  193. scipy/fft/_pocketfft/pypocketfft.cp313-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.cp313-win_arm64.lib +0 -0
  215. scipy/fftpack/convolve.cp313-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.cp313-win_arm64.lib +0 -0
  233. scipy/integrate/_dop.cp313-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.cp313-win_arm64.lib +0 -0
  248. scipy/integrate/_lsoda.cp313-win_arm64.pyd +0 -0
  249. scipy/integrate/_ode.py +1395 -0
  250. scipy/integrate/_odepack.cp313-win_arm64.lib +0 -0
  251. scipy/integrate/_odepack.cp313-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.cp313-win_arm64.lib +0 -0
  255. scipy/integrate/_quadpack.cp313-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.cp313-win_arm64.lib +0 -0
  265. scipy/integrate/_test_multivariate.cp313-win_arm64.pyd +0 -0
  266. scipy/integrate/_test_odeint_banded.cp313-win_arm64.lib +0 -0
  267. scipy/integrate/_test_odeint_banded.cp313-win_arm64.pyd +0 -0
  268. scipy/integrate/_vode.cp313-win_arm64.lib +0 -0
  269. scipy/integrate/_vode.cp313-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.cp313-win_arm64.lib +0 -0
  290. scipy/interpolate/_dfitpack.cp313-win_arm64.pyd +0 -0
  291. scipy/interpolate/_dierckx.cp313-win_arm64.lib +0 -0
  292. scipy/interpolate/_dierckx.cp313-win_arm64.pyd +0 -0
  293. scipy/interpolate/_fitpack.cp313-win_arm64.lib +0 -0
  294. scipy/interpolate/_fitpack.cp313-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.cp313-win_arm64.lib +0 -0
  300. scipy/interpolate/_interpnd.cp313-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.cp313-win_arm64.lib +0 -0
  307. scipy/interpolate/_ppoly.cp313-win_arm64.pyd +0 -0
  308. scipy/interpolate/_rbf.py +290 -0
  309. scipy/interpolate/_rbfinterp.py +550 -0
  310. scipy/interpolate/_rbfinterp_pythran.cp313-win_arm64.lib +0 -0
  311. scipy/interpolate/_rbfinterp_pythran.cp313-win_arm64.pyd +0 -0
  312. scipy/interpolate/_rgi.py +764 -0
  313. scipy/interpolate/_rgi_cython.cp313-win_arm64.lib +0 -0
  314. scipy/interpolate/_rgi_cython.cp313-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.cp313-win_arm64.lib +0 -0
  343. scipy/io/_fast_matrix_market/_fmm_core.cp313-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.cp313-win_arm64.lib +0 -0
  355. scipy/io/_test_fortran.cp313-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.cp313-win_arm64.lib +0 -0
  386. scipy/io/matlab/_mio5_utils.cp313-win_arm64.pyd +0 -0
  387. scipy/io/matlab/_mio_utils.cp313-win_arm64.lib +0 -0
  388. scipy/io/matlab/_mio_utils.cp313-win_arm64.pyd +0 -0
  389. scipy/io/matlab/_miobase.py +435 -0
  390. scipy/io/matlab/_streams.cp313-win_arm64.lib +0 -0
  391. scipy/io/matlab/_streams.cp313-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.cp313-win_arm64.lib +0 -0
  623. scipy/linalg/_cythonized_array_utils.cp313-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.cp313-win_arm64.lib +0 -0
  630. scipy/linalg/_decomp_interpolative.cp313-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.cp313-win_arm64.lib +0 -0
  634. scipy/linalg/_decomp_lu_cython.cp313-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.cp313-win_arm64.lib +0 -0
  642. scipy/linalg/_decomp_update.cp313-win_arm64.pyd +0 -0
  643. scipy/linalg/_expm_frechet.py +417 -0
  644. scipy/linalg/_fblas.cp313-win_arm64.lib +0 -0
  645. scipy/linalg/_fblas.cp313-win_arm64.pyd +0 -0
  646. scipy/linalg/_flapack.cp313-win_arm64.lib +0 -0
  647. scipy/linalg/_flapack.cp313-win_arm64.pyd +0 -0
  648. scipy/linalg/_lapack_subroutines.h +1521 -0
  649. scipy/linalg/_linalg_pythran.cp313-win_arm64.lib +0 -0
  650. scipy/linalg/_linalg_pythran.cp313-win_arm64.pyd +0 -0
  651. scipy/linalg/_matfuncs.py +1050 -0
  652. scipy/linalg/_matfuncs_expm.cp313-win_arm64.lib +0 -0
  653. scipy/linalg/_matfuncs_expm.cp313-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.cp313-win_arm64.lib +0 -0
  657. scipy/linalg/_matfuncs_schur_sqrtm.cp313-win_arm64.pyd +0 -0
  658. scipy/linalg/_matfuncs_sqrtm.py +107 -0
  659. scipy/linalg/_matfuncs_sqrtm_triu.cp313-win_arm64.lib +0 -0
  660. scipy/linalg/_matfuncs_sqrtm_triu.cp313-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.cp313-win_arm64.lib +0 -0
  665. scipy/linalg/_solve_toeplitz.cp313-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.cp313-win_arm64.lib +0 -0
  672. scipy/linalg/cython_blas.cp313-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.cp313-win_arm64.lib +0 -0
  676. scipy/linalg/cython_lapack.cp313-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.cp313-win_arm64.lib +0 -0
  728. scipy/ndimage/_ctest.cp313-win_arm64.pyd +0 -0
  729. scipy/ndimage/_cytest.cp313-win_arm64.lib +0 -0
  730. scipy/ndimage/_cytest.cp313-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.cp313-win_arm64.lib +0 -0
  738. scipy/ndimage/_nd_image.cp313-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.cp313-win_arm64.lib +0 -0
  742. scipy/ndimage/_ni_label.cp313-win_arm64.pyd +0 -0
  743. scipy/ndimage/_ni_support.py +139 -0
  744. scipy/ndimage/_rank_filter_1d.cp313-win_arm64.lib +0 -0
  745. scipy/ndimage/_rank_filter_1d.cp313-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.cp313-win_arm64.lib +0 -0
  768. scipy/odr/__odrpack.cp313-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.cp313-win_arm64.lib +0 -0
  780. scipy/optimize/_bglu_dense.cp313-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.cp313-win_arm64.lib +0 -0
  790. scipy/optimize/_direct.cp313-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.cp313-win_arm64.lib +0 -0
  795. scipy/optimize/_group_columns.cp313-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.cp313-win_arm64.lib +0 -0
  799. scipy/optimize/_highspy/_core.cp313-win_arm64.pyd +0 -0
  800. scipy/optimize/_highspy/_highs_options.cp313-win_arm64.lib +0 -0
  801. scipy/optimize/_highspy/_highs_options.cp313-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.cp313-win_arm64.lib +0 -0
  805. scipy/optimize/_lbfgsb.cp313-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.cp313-win_arm64.lib +0 -0
  816. scipy/optimize/_lsap.cp313-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.cp313-win_arm64.lib +0 -0
  822. scipy/optimize/_lsq/givens_elimination.cp313-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.cp313-win_arm64.lib +0 -0
  830. scipy/optimize/_minpack.cp313-win_arm64.pyd +0 -0
  831. scipy/optimize/_minpack_py.py +1178 -0
  832. scipy/optimize/_moduleTNC.cp313-win_arm64.lib +0 -0
  833. scipy/optimize/_moduleTNC.cp313-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.cp313-win_arm64.lib +0 -0
  839. scipy/optimize/_pava_pybind.cp313-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.cp313-win_arm64.lib +0 -0
  850. scipy/optimize/_slsqplib.cp313-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.cp313-win_arm64.lib +0 -0
  855. scipy/optimize/_trlib/_trlib.cp313-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.cp313-win_arm64.lib +0 -0
  877. scipy/optimize/_zeros.cp313-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.cp313-win_arm64.lib +0 -0
  882. scipy/optimize/cython_optimize/_zeros.cp313-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.cp313-win_arm64.lib +0 -0
  953. scipy/signal/_max_len_seq_inner.cp313-win_arm64.pyd +0 -0
  954. scipy/signal/_peak_finding.py +1310 -0
  955. scipy/signal/_peak_finding_utils.cp313-win_arm64.lib +0 -0
  956. scipy/signal/_peak_finding_utils.cp313-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.cp313-win_arm64.lib +0 -0
  963. scipy/signal/_sigtools.cp313-win_arm64.pyd +0 -0
  964. scipy/signal/_sosfilt.cp313-win_arm64.lib +0 -0
  965. scipy/signal/_sosfilt.cp313-win_arm64.pyd +0 -0
  966. scipy/signal/_spectral_py.py +2471 -0
  967. scipy/signal/_spline.cp313-win_arm64.lib +0 -0
  968. scipy/signal/_spline.cp313-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.cp313-win_arm64.lib +0 -0
  974. scipy/signal/_upfirdn_apply.cp313-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.cp313-win_arm64.lib +0 -0
  1021. scipy/sparse/_csparsetools.cp313-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.cp313-win_arm64.lib +0 -0
  1032. scipy/sparse/_sparsetools.cp313-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.cp313-win_arm64.lib +0 -0
  1043. scipy/sparse/csgraph/_flow.cp313-win_arm64.pyd +0 -0
  1044. scipy/sparse/csgraph/_laplacian.py +563 -0
  1045. scipy/sparse/csgraph/_matching.cp313-win_arm64.lib +0 -0
  1046. scipy/sparse/csgraph/_matching.cp313-win_arm64.pyd +0 -0
  1047. scipy/sparse/csgraph/_min_spanning_tree.cp313-win_arm64.lib +0 -0
  1048. scipy/sparse/csgraph/_min_spanning_tree.cp313-win_arm64.pyd +0 -0
  1049. scipy/sparse/csgraph/_reordering.cp313-win_arm64.lib +0 -0
  1050. scipy/sparse/csgraph/_reordering.cp313-win_arm64.pyd +0 -0
  1051. scipy/sparse/csgraph/_shortest_path.cp313-win_arm64.lib +0 -0
  1052. scipy/sparse/csgraph/_shortest_path.cp313-win_arm64.pyd +0 -0
  1053. scipy/sparse/csgraph/_tools.cp313-win_arm64.lib +0 -0
  1054. scipy/sparse/csgraph/_tools.cp313-win_arm64.pyd +0 -0
  1055. scipy/sparse/csgraph/_traversal.cp313-win_arm64.lib +0 -0
  1056. scipy/sparse/csgraph/_traversal.cp313-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.cp313-win_arm64.lib +0 -0
  1079. scipy/sparse/linalg/_dsolve/_superlu.cp313-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.cp313-win_arm64.lib +0 -0
  1089. scipy/sparse/linalg/_eigen/arpack/_arpack.cp313-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.cp313-win_arm64.lib +0 -0
  1122. scipy/sparse/linalg/_propack/_cpropack.cp313-win_arm64.pyd +0 -0
  1123. scipy/sparse/linalg/_propack/_dpropack.cp313-win_arm64.lib +0 -0
  1124. scipy/sparse/linalg/_propack/_dpropack.cp313-win_arm64.pyd +0 -0
  1125. scipy/sparse/linalg/_propack/_spropack.cp313-win_arm64.lib +0 -0
  1126. scipy/sparse/linalg/_propack/_spropack.cp313-win_arm64.pyd +0 -0
  1127. scipy/sparse/linalg/_propack/_zpropack.cp313-win_arm64.lib +0 -0
  1128. scipy/sparse/linalg/_propack/_zpropack.cp313-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.cp313-win_arm64.lib +0 -0
  1170. scipy/spatial/_ckdtree.cp313-win_arm64.pyd +0 -0
  1171. scipy/spatial/_distance_pybind.cp313-win_arm64.lib +0 -0
  1172. scipy/spatial/_distance_pybind.cp313-win_arm64.pyd +0 -0
  1173. scipy/spatial/_distance_wrap.cp313-win_arm64.lib +0 -0
  1174. scipy/spatial/_distance_wrap.cp313-win_arm64.pyd +0 -0
  1175. scipy/spatial/_geometric_slerp.py +238 -0
  1176. scipy/spatial/_hausdorff.cp313-win_arm64.lib +0 -0
  1177. scipy/spatial/_hausdorff.cp313-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.cp313-win_arm64.lib +0 -0
  1182. scipy/spatial/_qhull.cp313-win_arm64.pyd +0 -0
  1183. scipy/spatial/_qhull.pyi +213 -0
  1184. scipy/spatial/_spherical_voronoi.py +341 -0
  1185. scipy/spatial/_voronoi.cp313-win_arm64.lib +0 -0
  1186. scipy/spatial/_voronoi.cp313-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.cp313-win_arm64.lib +0 -0
  1236. scipy/spatial/transform/_rigid_transform.cp313-win_arm64.pyd +0 -0
  1237. scipy/spatial/transform/_rotation.cp313-win_arm64.lib +0 -0
  1238. scipy/spatial/transform/_rotation.cp313-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.cp313-win_arm64.lib +0 -0
  1252. scipy/special/_comb.cp313-win_arm64.pyd +0 -0
  1253. scipy/special/_ellip_harm.py +214 -0
  1254. scipy/special/_ellip_harm_2.cp313-win_arm64.lib +0 -0
  1255. scipy/special/_ellip_harm_2.cp313-win_arm64.pyd +0 -0
  1256. scipy/special/_gufuncs.cp313-win_arm64.lib +0 -0
  1257. scipy/special/_gufuncs.cp313-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.cp313-win_arm64.lib +0 -0
  1281. scipy/special/_specfun.cp313-win_arm64.pyd +0 -0
  1282. scipy/special/_special_ufuncs.cp313-win_arm64.lib +0 -0
  1283. scipy/special/_special_ufuncs.cp313-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.cp313-win_arm64.lib +0 -0
  1288. scipy/special/_test_internal.cp313-win_arm64.pyd +0 -0
  1289. scipy/special/_test_internal.pyi +9 -0
  1290. scipy/special/_testutils.py +321 -0
  1291. scipy/special/_ufuncs.cp313-win_arm64.lib +0 -0
  1292. scipy/special/_ufuncs.cp313-win_arm64.pyd +0 -0
  1293. scipy/special/_ufuncs.pyi +522 -0
  1294. scipy/special/_ufuncs.pyx +13173 -0
  1295. scipy/special/_ufuncs_cxx.cp313-win_arm64.lib +0 -0
  1296. scipy/special/_ufuncs_cxx.cp313-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.cp313-win_arm64.lib +0 -0
  1304. scipy/special/cython_special.cp313-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.cp313-win_arm64.lib +0 -0
  1376. scipy/stats/_ansari_swilk_statistics.cp313-win_arm64.pyd +0 -0
  1377. scipy/stats/_axis_nan_policy.py +692 -0
  1378. scipy/stats/_biasedurn.cp313-win_arm64.lib +0 -0
  1379. scipy/stats/_biasedurn.cp313-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.cp313-win_arm64.lib +0 -0
  1404. scipy/stats/_levy_stable/levyst.cp313-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.cp313-win_arm64.lib +0 -0
  1418. scipy/stats/_qmc_cy.cp313-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.cp313-win_arm64.lib +0 -0
  1422. scipy/stats/_qmvnt_cy.cp313-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.cp313-win_arm64.lib +0 -0
  1426. scipy/stats/_rcont/rcont.cp313-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.cp313-win_arm64.lib +0 -0
  1433. scipy/stats/_sobol.cp313-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.cp313-win_arm64.lib +0 -0
  1437. scipy/stats/_stats.cp313-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.cp313-win_arm64.lib +0 -0
  1442. scipy/stats/_stats_pythran.cp313-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.cp313-win_arm64.lib +0 -0
  1447. scipy/stats/_unuran/unuran_wrapper.cp313-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,2950 @@
1
+ import numpy as np
2
+ from scipy._lib._array_api import (
3
+ is_cupy, is_numpy,
4
+ xp_assert_close, xp_assert_equal, assert_array_almost_equal
5
+ )
6
+ import pytest
7
+ from pytest import raises as assert_raises
8
+
9
+ from scipy import ndimage
10
+
11
+ from . import types
12
+
13
+ skip_xp_backends = pytest.mark.skip_xp_backends
14
+ xfail_xp_backends = pytest.mark.xfail_xp_backends
15
+ pytestmark = [skip_xp_backends(cpu_only=True, exceptions=['cupy', 'jax.numpy'])]
16
+
17
+
18
+ class TestNdimageMorphology:
19
+
20
+ @xfail_xp_backends('cupy', reason='CuPy does not have distance_transform_bf.')
21
+ @pytest.mark.parametrize('dtype', types)
22
+ def test_distance_transform_bf01(self, dtype, xp):
23
+ dtype = getattr(xp, dtype)
24
+
25
+ # brute force (bf) distance transform
26
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
27
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
28
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
29
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
30
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
31
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
32
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
33
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
34
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
35
+ out, ft = ndimage.distance_transform_bf(data, 'euclidean',
36
+ return_indices=True)
37
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0, 0],
38
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
39
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
40
+ [0, 0, 1, 2, 4, 2, 1, 0, 0],
41
+ [0, 0, 1, 4, 8, 4, 1, 0, 0],
42
+ [0, 0, 1, 2, 4, 2, 1, 0, 0],
43
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
44
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
45
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]]
46
+ expected = xp.asarray(expected)
47
+ assert_array_almost_equal(out * out, expected)
48
+
49
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
50
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
51
+ [2, 2, 2, 2, 1, 2, 2, 2, 2],
52
+ [3, 3, 3, 2, 1, 2, 3, 3, 3],
53
+ [4, 4, 4, 4, 6, 4, 4, 4, 4],
54
+ [5, 5, 6, 6, 7, 6, 6, 5, 5],
55
+ [6, 6, 6, 7, 7, 7, 6, 6, 6],
56
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
57
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
58
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
59
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
60
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
61
+ [0, 1, 1, 2, 4, 6, 7, 7, 8],
62
+ [0, 1, 1, 1, 6, 7, 7, 7, 8],
63
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
64
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
65
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
66
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
67
+ expected = xp.asarray(expected)
68
+ assert_array_almost_equal(ft, expected)
69
+
70
+ @xfail_xp_backends('cupy', reason='CuPy does not have distance_transform_bf.')
71
+ @pytest.mark.parametrize('dtype', types)
72
+ def test_distance_transform_bf02(self, dtype, xp):
73
+ dtype = getattr(xp, dtype)
74
+
75
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
76
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
77
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
78
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
79
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
80
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
81
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
82
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
83
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
84
+ out, ft = ndimage.distance_transform_bf(data, 'cityblock',
85
+ return_indices=True)
86
+
87
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0, 0],
88
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
89
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
90
+ [0, 0, 1, 2, 2, 2, 1, 0, 0],
91
+ [0, 0, 1, 2, 3, 2, 1, 0, 0],
92
+ [0, 0, 1, 2, 2, 2, 1, 0, 0],
93
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
94
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
95
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]]
96
+ expected = xp.asarray(expected)
97
+ assert_array_almost_equal(out, expected)
98
+
99
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
100
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
101
+ [2, 2, 2, 2, 1, 2, 2, 2, 2],
102
+ [3, 3, 3, 3, 1, 3, 3, 3, 3],
103
+ [4, 4, 4, 4, 7, 4, 4, 4, 4],
104
+ [5, 5, 6, 7, 7, 7, 6, 5, 5],
105
+ [6, 6, 6, 7, 7, 7, 6, 6, 6],
106
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
107
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
108
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
109
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
110
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
111
+ [0, 1, 1, 1, 4, 7, 7, 7, 8],
112
+ [0, 1, 1, 1, 4, 7, 7, 7, 8],
113
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
114
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
115
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
116
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
117
+ expected = xp.asarray(expected)
118
+ assert_array_almost_equal(ft, expected)
119
+
120
+ @xfail_xp_backends('cupy', reason='CuPy does not have distance_transform_bf.')
121
+ @pytest.mark.parametrize('dtype', types)
122
+ def test_distance_transform_bf03(self, dtype, xp):
123
+ dtype = getattr(xp, dtype)
124
+
125
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
126
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
127
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
128
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
129
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
130
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
131
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
132
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
133
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
134
+ out, ft = ndimage.distance_transform_bf(data, 'chessboard',
135
+ return_indices=True)
136
+
137
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0, 0],
138
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
139
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
140
+ [0, 0, 1, 1, 2, 1, 1, 0, 0],
141
+ [0, 0, 1, 2, 2, 2, 1, 0, 0],
142
+ [0, 0, 1, 1, 2, 1, 1, 0, 0],
143
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
144
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
145
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]]
146
+ expected = xp.asarray(expected)
147
+ assert_array_almost_equal(out, expected)
148
+
149
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
150
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
151
+ [2, 2, 2, 2, 1, 2, 2, 2, 2],
152
+ [3, 3, 4, 2, 2, 2, 4, 3, 3],
153
+ [4, 4, 5, 6, 6, 6, 5, 4, 4],
154
+ [5, 5, 6, 6, 7, 6, 6, 5, 5],
155
+ [6, 6, 6, 7, 7, 7, 6, 6, 6],
156
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
157
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
158
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
159
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
160
+ [0, 1, 2, 2, 5, 6, 6, 7, 8],
161
+ [0, 1, 1, 2, 6, 6, 7, 7, 8],
162
+ [0, 1, 1, 2, 6, 7, 7, 7, 8],
163
+ [0, 1, 2, 2, 6, 6, 7, 7, 8],
164
+ [0, 1, 2, 4, 5, 6, 6, 7, 8],
165
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
166
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
167
+ expected = xp.asarray(expected)
168
+ assert_array_almost_equal(ft, expected)
169
+
170
+ @skip_xp_backends(
171
+ np_only=True, reason='inplace distances= arrays are numpy-specific'
172
+ )
173
+ @pytest.mark.parametrize('dtype', types)
174
+ def test_distance_transform_bf04(self, dtype, xp):
175
+ dtype = getattr(xp, dtype)
176
+
177
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
178
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
179
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
180
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
181
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
182
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
183
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
184
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
185
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
186
+ tdt, tft = ndimage.distance_transform_bf(data, return_indices=1)
187
+ dts = []
188
+ fts = []
189
+ dt = xp.zeros(data.shape, dtype=xp.float64)
190
+ ndimage.distance_transform_bf(data, distances=dt)
191
+ dts.append(dt)
192
+ ft = ndimage.distance_transform_bf(
193
+ data, return_distances=False, return_indices=1)
194
+ fts.append(ft)
195
+ ft = np.indices(data.shape, dtype=xp.int32)
196
+ ndimage.distance_transform_bf(
197
+ data, return_distances=False, return_indices=True, indices=ft)
198
+ fts.append(ft)
199
+ dt, ft = ndimage.distance_transform_bf(
200
+ data, return_indices=1)
201
+ dts.append(dt)
202
+ fts.append(ft)
203
+ dt = xp.zeros(data.shape, dtype=xp.float64)
204
+ ft = ndimage.distance_transform_bf(
205
+ data, distances=dt, return_indices=True)
206
+ dts.append(dt)
207
+ fts.append(ft)
208
+ ft = np.indices(data.shape, dtype=xp.int32)
209
+ dt = ndimage.distance_transform_bf(
210
+ data, return_indices=True, indices=ft)
211
+ dts.append(dt)
212
+ fts.append(ft)
213
+ dt = xp.zeros(data.shape, dtype=xp.float64)
214
+ ft = np.indices(data.shape, dtype=xp.int32)
215
+ ndimage.distance_transform_bf(
216
+ data, distances=dt, return_indices=True, indices=ft)
217
+ dts.append(dt)
218
+ fts.append(ft)
219
+ for dt in dts:
220
+ assert_array_almost_equal(tdt, dt)
221
+ for ft in fts:
222
+ assert_array_almost_equal(tft, ft)
223
+
224
+ @xfail_xp_backends('cupy', reason='CuPy does not have distance_transform_bf')
225
+ @pytest.mark.parametrize('dtype', types)
226
+ def test_distance_transform_bf05(self, dtype, xp):
227
+ dtype = getattr(xp, dtype)
228
+
229
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
230
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
231
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
232
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
233
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
234
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
235
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
236
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
237
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
238
+ out, ft = ndimage.distance_transform_bf(
239
+ data, 'euclidean', return_indices=True, sampling=[2, 2])
240
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0, 0],
241
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
242
+ [0, 0, 0, 4, 4, 4, 0, 0, 0],
243
+ [0, 0, 4, 8, 16, 8, 4, 0, 0],
244
+ [0, 0, 4, 16, 32, 16, 4, 0, 0],
245
+ [0, 0, 4, 8, 16, 8, 4, 0, 0],
246
+ [0, 0, 0, 4, 4, 4, 0, 0, 0],
247
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
248
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]]
249
+ expected = xp.asarray(expected)
250
+ assert_array_almost_equal(out * out, expected)
251
+
252
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
253
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
254
+ [2, 2, 2, 2, 1, 2, 2, 2, 2],
255
+ [3, 3, 3, 2, 1, 2, 3, 3, 3],
256
+ [4, 4, 4, 4, 6, 4, 4, 4, 4],
257
+ [5, 5, 6, 6, 7, 6, 6, 5, 5],
258
+ [6, 6, 6, 7, 7, 7, 6, 6, 6],
259
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
260
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
261
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
262
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
263
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
264
+ [0, 1, 1, 2, 4, 6, 7, 7, 8],
265
+ [0, 1, 1, 1, 6, 7, 7, 7, 8],
266
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
267
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
268
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
269
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
270
+ expected = xp.asarray(expected)
271
+ assert_array_almost_equal(ft, expected)
272
+
273
+ @xfail_xp_backends('cupy', reason='CuPy does not have distance_transform_bf')
274
+ @pytest.mark.parametrize('dtype', types)
275
+ def test_distance_transform_bf06(self, dtype, xp):
276
+ dtype = getattr(xp, dtype)
277
+
278
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
279
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
280
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
281
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
282
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
283
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
284
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
285
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
286
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
287
+ out, ft = ndimage.distance_transform_bf(
288
+ data, 'euclidean', return_indices=True, sampling=[2, 1])
289
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0, 0],
290
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
291
+ [0, 0, 0, 1, 4, 1, 0, 0, 0],
292
+ [0, 0, 1, 4, 8, 4, 1, 0, 0],
293
+ [0, 0, 1, 4, 9, 4, 1, 0, 0],
294
+ [0, 0, 1, 4, 8, 4, 1, 0, 0],
295
+ [0, 0, 0, 1, 4, 1, 0, 0, 0],
296
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
297
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]]
298
+ expected = xp.asarray(expected)
299
+ assert_array_almost_equal(out * out, expected)
300
+
301
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
302
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
303
+ [2, 2, 2, 2, 2, 2, 2, 2, 2],
304
+ [3, 3, 3, 3, 2, 3, 3, 3, 3],
305
+ [4, 4, 4, 4, 4, 4, 4, 4, 4],
306
+ [5, 5, 5, 5, 6, 5, 5, 5, 5],
307
+ [6, 6, 6, 6, 7, 6, 6, 6, 6],
308
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
309
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
310
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
311
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
312
+ [0, 1, 2, 2, 6, 6, 6, 7, 8],
313
+ [0, 1, 1, 1, 6, 7, 7, 7, 8],
314
+ [0, 1, 1, 1, 7, 7, 7, 7, 8],
315
+ [0, 1, 1, 1, 6, 7, 7, 7, 8],
316
+ [0, 1, 2, 2, 4, 6, 6, 7, 8],
317
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
318
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
319
+ expected = xp.asarray(expected)
320
+ assert_array_almost_equal(ft, expected)
321
+
322
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_bf")
323
+ def test_distance_transform_bf07(self, xp):
324
+ # test input validation per discussion on PR #13302
325
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
326
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
327
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
328
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
329
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
330
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
331
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
332
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
333
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]])
334
+ with assert_raises(RuntimeError):
335
+ ndimage.distance_transform_bf(
336
+ data, return_distances=False, return_indices=False
337
+ )
338
+
339
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_cdt")
340
+ @pytest.mark.parametrize('dtype', types)
341
+ def test_distance_transform_cdt01(self, dtype, xp):
342
+ dtype = getattr(xp, dtype)
343
+
344
+ # chamfer type distance (cdt) transform
345
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
346
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
347
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
348
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
349
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
350
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
351
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
352
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
353
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
354
+ out, ft = ndimage.distance_transform_cdt(
355
+ data, 'cityblock', return_indices=True)
356
+ bf = ndimage.distance_transform_bf(data, 'cityblock')
357
+ assert_array_almost_equal(bf, out)
358
+
359
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
360
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
361
+ [2, 2, 2, 1, 1, 1, 2, 2, 2],
362
+ [3, 3, 2, 1, 1, 1, 2, 3, 3],
363
+ [4, 4, 4, 4, 1, 4, 4, 4, 4],
364
+ [5, 5, 5, 5, 7, 7, 6, 5, 5],
365
+ [6, 6, 6, 6, 7, 7, 6, 6, 6],
366
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
367
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
368
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
369
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
370
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
371
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
372
+ [0, 1, 1, 1, 4, 7, 7, 7, 8],
373
+ [0, 1, 1, 1, 4, 5, 6, 7, 8],
374
+ [0, 1, 2, 2, 4, 5, 6, 7, 8],
375
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
376
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
377
+ expected = xp.asarray(expected)
378
+ assert_array_almost_equal(ft, expected)
379
+
380
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_cdt")
381
+ @pytest.mark.parametrize('dtype', types)
382
+ def test_distance_transform_cdt02(self, dtype, xp):
383
+ dtype = getattr(xp, dtype)
384
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
385
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
386
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
387
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
388
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
389
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
390
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
391
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
392
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
393
+ out, ft = ndimage.distance_transform_cdt(data, 'chessboard',
394
+ return_indices=True)
395
+ bf = ndimage.distance_transform_bf(data, 'chessboard')
396
+ assert_array_almost_equal(bf, out)
397
+
398
+ expected = [[[0, 0, 0, 0, 0, 0, 0, 0, 0],
399
+ [1, 1, 1, 1, 1, 1, 1, 1, 1],
400
+ [2, 2, 2, 1, 1, 1, 2, 2, 2],
401
+ [3, 3, 2, 2, 1, 2, 2, 3, 3],
402
+ [4, 4, 3, 2, 2, 2, 3, 4, 4],
403
+ [5, 5, 4, 6, 7, 6, 4, 5, 5],
404
+ [6, 6, 6, 6, 7, 7, 6, 6, 6],
405
+ [7, 7, 7, 7, 7, 7, 7, 7, 7],
406
+ [8, 8, 8, 8, 8, 8, 8, 8, 8]],
407
+ [[0, 1, 2, 3, 4, 5, 6, 7, 8],
408
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
409
+ [0, 1, 2, 2, 3, 4, 6, 7, 8],
410
+ [0, 1, 1, 2, 2, 6, 6, 7, 8],
411
+ [0, 1, 1, 1, 2, 6, 7, 7, 8],
412
+ [0, 1, 1, 2, 6, 6, 7, 7, 8],
413
+ [0, 1, 2, 2, 5, 6, 6, 7, 8],
414
+ [0, 1, 2, 3, 4, 5, 6, 7, 8],
415
+ [0, 1, 2, 3, 4, 5, 6, 7, 8]]]
416
+ expected = xp.asarray(expected)
417
+ assert_array_almost_equal(ft, expected)
418
+
419
+ @skip_xp_backends(
420
+ np_only=True, reason='inplace indices= arrays are numpy-specific'
421
+ )
422
+ @pytest.mark.parametrize('dtype', types)
423
+ def test_distance_transform_cdt03(self, dtype, xp):
424
+ dtype = getattr(xp, dtype)
425
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
426
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
427
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
428
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
429
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
430
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
431
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
432
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
433
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
434
+ tdt, tft = ndimage.distance_transform_cdt(data, return_indices=True)
435
+ dts = []
436
+ fts = []
437
+ dt = xp.zeros(data.shape, dtype=xp.int32)
438
+ ndimage.distance_transform_cdt(data, distances=dt)
439
+ dts.append(dt)
440
+ ft = ndimage.distance_transform_cdt(
441
+ data, return_distances=False, return_indices=True)
442
+ fts.append(ft)
443
+ ft = xp.asarray(np.indices(data.shape, dtype=np.int32))
444
+ ndimage.distance_transform_cdt(
445
+ data, return_distances=False, return_indices=True, indices=ft)
446
+ fts.append(ft)
447
+ dt, ft = ndimage.distance_transform_cdt(
448
+ data, return_indices=True)
449
+ dts.append(dt)
450
+ fts.append(ft)
451
+ dt = xp.zeros(data.shape, dtype=xp.int32)
452
+ ft = ndimage.distance_transform_cdt(
453
+ data, distances=dt, return_indices=True)
454
+ dts.append(dt)
455
+ fts.append(ft)
456
+ ft = xp.asarray(np.indices(data.shape, dtype=np.int32))
457
+ dt = ndimage.distance_transform_cdt(
458
+ data, return_indices=True, indices=ft)
459
+ dts.append(dt)
460
+ fts.append(ft)
461
+ dt = xp.zeros(data.shape, dtype=xp.int32)
462
+ ft = xp.asarray(np.indices(data.shape, dtype=np.int32))
463
+ ndimage.distance_transform_cdt(data, distances=dt,
464
+ return_indices=True, indices=ft)
465
+ dts.append(dt)
466
+ fts.append(ft)
467
+ for dt in dts:
468
+ assert_array_almost_equal(tdt, dt)
469
+ for ft in fts:
470
+ assert_array_almost_equal(tft, ft)
471
+
472
+ @skip_xp_backends(
473
+ np_only=True, reason='XXX: does not raise unless indices is a numpy array'
474
+ )
475
+ def test_distance_transform_cdt04(self, xp):
476
+ # test input validation per discussion on PR #13302
477
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
478
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
479
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
480
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
481
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
482
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
483
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
484
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
485
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]])
486
+ indices_out = xp.zeros((data.ndim,) + data.shape, dtype=xp.int32)
487
+ with assert_raises(RuntimeError):
488
+ ndimage.distance_transform_bf(
489
+ data,
490
+ return_distances=True,
491
+ return_indices=False,
492
+ indices=indices_out
493
+ )
494
+
495
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_cdt")
496
+ @xfail_xp_backends("torch", reason="int overflow")
497
+ @pytest.mark.parametrize('dtype', types)
498
+ def test_distance_transform_cdt05(self, dtype, xp):
499
+ dtype = getattr(xp, dtype)
500
+
501
+ # test custom metric type per discussion on issue #17381
502
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
503
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
504
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
505
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
506
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
507
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
508
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
509
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
510
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
511
+ metric_arg = xp.ones((3, 3))
512
+ actual = ndimage.distance_transform_cdt(data, metric=metric_arg)
513
+ assert xp.sum(actual) == -21
514
+
515
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_bf")
516
+ @pytest.mark.parametrize('dtype', types)
517
+ def test_distance_transform_edt01(self, dtype, xp):
518
+ dtype = getattr(xp, dtype)
519
+
520
+ # euclidean distance transform (edt)
521
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
522
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
523
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
524
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
525
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
526
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
527
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
528
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
529
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
530
+ out, ft = ndimage.distance_transform_edt(data, return_indices=True)
531
+ bf = ndimage.distance_transform_bf(data, 'euclidean')
532
+ assert_array_almost_equal(bf, out)
533
+
534
+ # np-specific check
535
+ np_ft = np.asarray(ft)
536
+ dt = np_ft - np.indices(np_ft.shape[1:], dtype=np_ft.dtype)
537
+ dt = dt.astype(np.float64)
538
+ np.multiply(dt, dt, dt)
539
+ dt = np.add.reduce(dt, axis=0)
540
+ np.sqrt(dt, dt)
541
+
542
+ dt = xp.asarray(dt)
543
+ assert_array_almost_equal(bf, dt)
544
+
545
+ @skip_xp_backends(
546
+ np_only=True, reason='inplace distances= are numpy-specific'
547
+ )
548
+ @pytest.mark.parametrize('dtype', types)
549
+ def test_distance_transform_edt02(self, dtype, xp):
550
+ dtype = getattr(xp, dtype)
551
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
552
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
553
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
554
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
555
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
556
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
557
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
558
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
559
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
560
+ tdt, tft = ndimage.distance_transform_edt(data, return_indices=True)
561
+ dts = []
562
+ fts = []
563
+
564
+ dt = xp.zeros(data.shape, dtype=xp.float64)
565
+ ndimage.distance_transform_edt(data, distances=dt)
566
+ dts.append(dt)
567
+
568
+ ft = ndimage.distance_transform_edt(
569
+ data, return_distances=0, return_indices=True)
570
+ fts.append(ft)
571
+
572
+ ft = np.indices(data.shape, dtype=xp.int32)
573
+ ft = xp.asarray(ft)
574
+ ndimage.distance_transform_edt(
575
+ data, return_distances=False, return_indices=True, indices=ft)
576
+ fts.append(ft)
577
+
578
+ dt, ft = ndimage.distance_transform_edt(
579
+ data, return_indices=True)
580
+ dts.append(dt)
581
+ fts.append(ft)
582
+
583
+ dt = xp.zeros(data.shape, dtype=xp.float64)
584
+ ft = ndimage.distance_transform_edt(
585
+ data, distances=dt, return_indices=True)
586
+ dts.append(dt)
587
+ fts.append(ft)
588
+
589
+ ft = np.indices(data.shape, dtype=xp.int32)
590
+ ft = xp.asarray(ft)
591
+ dt = ndimage.distance_transform_edt(
592
+ data, return_indices=True, indices=ft)
593
+ dts.append(dt)
594
+ fts.append(ft)
595
+
596
+ dt = xp.zeros(data.shape, dtype=xp.float64)
597
+ ft = np.indices(data.shape, dtype=xp.int32)
598
+ ft = xp.asarray(ft)
599
+ ndimage.distance_transform_edt(
600
+ data, distances=dt, return_indices=True, indices=ft)
601
+ dts.append(dt)
602
+ fts.append(ft)
603
+
604
+ for dt in dts:
605
+ assert_array_almost_equal(tdt, dt)
606
+ for ft in fts:
607
+ assert_array_almost_equal(tft, ft)
608
+
609
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_bf")
610
+ @pytest.mark.parametrize('dtype', types)
611
+ def test_distance_transform_edt03(self, dtype, xp):
612
+ dtype = getattr(xp, dtype)
613
+
614
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
615
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
616
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
617
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
618
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
619
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
620
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
621
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
622
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
623
+ ref = ndimage.distance_transform_bf(data, 'euclidean', sampling=[2, 2])
624
+ out = ndimage.distance_transform_edt(data, sampling=[2, 2])
625
+ assert_array_almost_equal(out, ref)
626
+
627
+ @xfail_xp_backends("cupy", reason="CuPy does not have distance_transform_bf")
628
+ @pytest.mark.parametrize('dtype', types)
629
+ def test_distance_transform_edt4(self, dtype, xp):
630
+ dtype = getattr(xp, dtype)
631
+
632
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
633
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
634
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
635
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
636
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
637
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
638
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
639
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
640
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
641
+ ref = ndimage.distance_transform_bf(data, 'euclidean', sampling=[2, 1])
642
+ out = ndimage.distance_transform_edt(data, sampling=[2, 1])
643
+ assert_array_almost_equal(out, ref)
644
+
645
+ @xfail_xp_backends(
646
+ "cupy", reason="Only 2D and 3D distance transforms are supported in CuPy"
647
+ )
648
+ def test_distance_transform_edt5(self, xp):
649
+ # Ticket #954 regression test
650
+ out = ndimage.distance_transform_edt(xp.asarray(False))
651
+ assert_array_almost_equal(out, xp.asarray([0.]))
652
+
653
+ @xfail_xp_backends(
654
+ np_only=True, reason='XXX: does not raise unless indices is a numpy array'
655
+ )
656
+ def test_distance_transform_edt6(self, xp):
657
+ # test input validation per discussion on PR #13302
658
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0, 0],
659
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
660
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
661
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
662
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
663
+ [0, 0, 1, 1, 1, 1, 1, 0, 0],
664
+ [0, 0, 0, 1, 1, 1, 0, 0, 0],
665
+ [0, 0, 0, 0, 0, 0, 0, 0, 0],
666
+ [0, 0, 0, 0, 0, 0, 0, 0, 0]])
667
+ distances_out = xp.zeros(data.shape, dtype=xp.float64)
668
+ with assert_raises(RuntimeError):
669
+ ndimage.distance_transform_bf(
670
+ data,
671
+ return_indices=True,
672
+ return_distances=False,
673
+ distances=distances_out
674
+ )
675
+
676
+ @skip_xp_backends(np_only=True,
677
+ reason="generate_binary_structure always generates numpy objects")
678
+ def test_generate_structure01(self, xp):
679
+ struct = ndimage.generate_binary_structure(0, 1)
680
+ assert struct == 1
681
+
682
+ @skip_xp_backends(np_only=True,
683
+ reason="generate_binary_structure always generates numpy objects")
684
+ def test_generate_structure02(self, xp):
685
+ struct = ndimage.generate_binary_structure(1, 1)
686
+ assert_array_almost_equal(struct, [1, 1, 1])
687
+
688
+ @skip_xp_backends(np_only=True,
689
+ reason="generate_binary_structure always generates numpy objects")
690
+ def test_generate_structure03(self, xp):
691
+ struct = ndimage.generate_binary_structure(2, 1)
692
+ assert_array_almost_equal(struct, [[0, 1, 0],
693
+ [1, 1, 1],
694
+ [0, 1, 0]])
695
+
696
+ @skip_xp_backends(np_only=True,
697
+ reason="generate_binary_structure always generates numpy objects")
698
+ def test_generate_structure04(self, xp):
699
+ struct = ndimage.generate_binary_structure(2, 2)
700
+ assert_array_almost_equal(struct, [[1, 1, 1],
701
+ [1, 1, 1],
702
+ [1, 1, 1]])
703
+
704
+ def test_iterate_structure01(self, xp):
705
+ struct = [[0, 1, 0],
706
+ [1, 1, 1],
707
+ [0, 1, 0]]
708
+ struct = xp.asarray(struct)
709
+ out = ndimage.iterate_structure(struct, 2)
710
+ expected = np.asarray([[0, 0, 1, 0, 0],
711
+ [0, 1, 1, 1, 0],
712
+ [1, 1, 1, 1, 1],
713
+ [0, 1, 1, 1, 0],
714
+ [0, 0, 1, 0, 0]], dtype=bool)
715
+ expected = xp.asarray(expected)
716
+ assert_array_almost_equal(out, expected)
717
+
718
+ def test_iterate_structure02(self, xp):
719
+ struct = [[0, 1],
720
+ [1, 1],
721
+ [0, 1]]
722
+ struct = xp.asarray(struct)
723
+ out = ndimage.iterate_structure(struct, 2)
724
+ expected = np.asarray([[0, 0, 1],
725
+ [0, 1, 1],
726
+ [1, 1, 1],
727
+ [0, 1, 1],
728
+ [0, 0, 1]], dtype=bool)
729
+ expected = xp.asarray(expected)
730
+
731
+ assert_array_almost_equal(out, expected)
732
+
733
+ def test_iterate_structure03(self, xp):
734
+ struct = [[0, 1, 0],
735
+ [1, 1, 1],
736
+ [0, 1, 0]]
737
+ struct = xp.asarray(struct)
738
+ out = ndimage.iterate_structure(struct, 2, 1)
739
+ expected = [[0, 0, 1, 0, 0],
740
+ [0, 1, 1, 1, 0],
741
+ [1, 1, 1, 1, 1],
742
+ [0, 1, 1, 1, 0],
743
+ [0, 0, 1, 0, 0]]
744
+ expected = np.asarray(expected, dtype=bool)
745
+ expected = xp.asarray(expected)
746
+ assert_array_almost_equal(out[0], expected)
747
+ assert out[1] == [2, 2]
748
+
749
+ @pytest.mark.parametrize('dtype', types)
750
+ def test_binary_erosion01(self, dtype, xp):
751
+ dtype = getattr(xp, dtype)
752
+ data = xp.ones([], dtype=dtype)
753
+ out = ndimage.binary_erosion(data)
754
+ assert out == xp.asarray(1, dtype=out.dtype)
755
+
756
+ @pytest.mark.parametrize('dtype', types)
757
+ def test_binary_erosion02(self, dtype, xp):
758
+ dtype = getattr(xp, dtype)
759
+ data = xp.ones([], dtype=dtype)
760
+ out = ndimage.binary_erosion(data, border_value=1)
761
+ assert out == xp.asarray(1, dtype=out.dtype)
762
+
763
+ @pytest.mark.parametrize('dtype', types)
764
+ def test_binary_erosion03(self, dtype, xp):
765
+ dtype = getattr(xp, dtype)
766
+ data = xp.ones([1], dtype=dtype)
767
+ out = ndimage.binary_erosion(data)
768
+ assert_array_almost_equal(out, xp.asarray([0]))
769
+
770
+ @pytest.mark.parametrize('dtype', types)
771
+ def test_binary_erosion04(self, dtype, xp):
772
+ dtype = getattr(xp, dtype)
773
+ data = xp.ones([1], dtype=dtype)
774
+ out = ndimage.binary_erosion(data, border_value=1)
775
+ assert_array_almost_equal(out, xp.asarray([1]))
776
+
777
+ @pytest.mark.parametrize('dtype', types)
778
+ def test_binary_erosion05(self, dtype, xp):
779
+ dtype = getattr(xp, dtype)
780
+ data = xp.ones([3], dtype=dtype)
781
+ out = ndimage.binary_erosion(data)
782
+ assert_array_almost_equal(out, xp.asarray([0, 1, 0]))
783
+
784
+ @pytest.mark.parametrize('dtype', types)
785
+ @xfail_xp_backends("cupy", reason="https://github.com/cupy/cupy/issues/8912")
786
+ def test_binary_erosion05_broadcasted(self, dtype, xp):
787
+ dtype = getattr(xp, dtype)
788
+ data = xp.ones((1, ), dtype=dtype)
789
+ data = xp.broadcast_to(data, (3, ))
790
+ out = ndimage.binary_erosion(data)
791
+ assert_array_almost_equal(out, xp.asarray([0, 1, 0]))
792
+
793
+ @pytest.mark.parametrize('dtype', types)
794
+ def test_binary_erosion06(self, dtype, xp):
795
+ dtype = getattr(xp, dtype)
796
+ data = xp.ones([3], dtype=dtype)
797
+ out = ndimage.binary_erosion(data, border_value=1)
798
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1]))
799
+
800
+ @pytest.mark.parametrize('dtype', types)
801
+ def test_binary_erosion07(self, dtype, xp):
802
+ dtype = getattr(xp, dtype)
803
+ data = xp.ones([5], dtype=dtype)
804
+ out = ndimage.binary_erosion(data)
805
+ assert_array_almost_equal(out, xp.asarray([0, 1, 1, 1, 0]))
806
+
807
+ @pytest.mark.parametrize('dtype', types)
808
+ def test_binary_erosion08(self, dtype, xp):
809
+ dtype = getattr(xp, dtype)
810
+ data = xp.ones([5], dtype=dtype)
811
+ out = ndimage.binary_erosion(data, border_value=1)
812
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1, 1, 1]))
813
+
814
+ @pytest.mark.parametrize('dtype', types)
815
+ def test_binary_erosion09(self, dtype, xp):
816
+ data = np.ones([5], dtype=dtype)
817
+ data[2] = 0
818
+ data = xp.asarray(data)
819
+ out = ndimage.binary_erosion(data)
820
+ assert_array_almost_equal(out, xp.asarray([0, 0, 0, 0, 0]))
821
+
822
+ @pytest.mark.parametrize('dtype', types)
823
+ def test_binary_erosion10(self, dtype, xp):
824
+ data = np.ones([5], dtype=dtype)
825
+ data[2] = 0
826
+ data = xp.asarray(data)
827
+ out = ndimage.binary_erosion(data, border_value=1)
828
+ assert_array_almost_equal(out, xp.asarray([1, 0, 0, 0, 1]))
829
+
830
+ @pytest.mark.parametrize('dtype', types)
831
+ def test_binary_erosion11(self, dtype, xp):
832
+ data = np.ones([5], dtype=dtype)
833
+ data[2] = 0
834
+ data = xp.asarray(data)
835
+ struct = xp.asarray([1, 0, 1])
836
+ out = ndimage.binary_erosion(data, struct, border_value=1)
837
+ assert_array_almost_equal(out, xp.asarray([1, 0, 1, 0, 1]))
838
+
839
+ @pytest.mark.parametrize('dtype', types)
840
+ def test_binary_erosion12(self, dtype, xp):
841
+ data = np.ones([5], dtype=dtype)
842
+ data[2] = 0
843
+ data = xp.asarray(data)
844
+ struct = xp.asarray([1, 0, 1])
845
+ out = ndimage.binary_erosion(data, struct, border_value=1, origin=-1)
846
+ assert_array_almost_equal(out, xp.asarray([0, 1, 0, 1, 1]))
847
+
848
+ @pytest.mark.parametrize('dtype', types)
849
+ def test_binary_erosion13(self, dtype, xp):
850
+ data = np.ones([5], dtype=dtype)
851
+ data[2] = 0
852
+ data = xp.asarray(data)
853
+ struct = xp.asarray([1, 0, 1])
854
+ out = ndimage.binary_erosion(data, struct, border_value=1, origin=1)
855
+ assert_array_almost_equal(out, xp.asarray([1, 1, 0, 1, 0]))
856
+
857
+ @pytest.mark.parametrize('dtype', types)
858
+ def test_binary_erosion14(self, dtype, xp):
859
+ data = np.ones([5], dtype=dtype)
860
+ data[2] = 0
861
+ data = xp.asarray(data)
862
+ struct = xp.asarray([1, 1])
863
+ out = ndimage.binary_erosion(data, struct, border_value=1)
864
+ assert_array_almost_equal(out, xp.asarray([1, 1, 0, 0, 1]))
865
+
866
+ @pytest.mark.parametrize('dtype', types)
867
+ def test_binary_erosion15(self, dtype, xp):
868
+ data = np.ones([5], dtype=dtype)
869
+ data[2] = 0
870
+ data = xp.asarray(data)
871
+ struct = xp.asarray([1, 1])
872
+ out = ndimage.binary_erosion(data, struct, border_value=1, origin=-1)
873
+ assert_array_almost_equal(out, xp.asarray([1, 0, 0, 1, 1]))
874
+
875
+ @pytest.mark.parametrize('dtype', types)
876
+ def test_binary_erosion16(self, dtype, xp):
877
+ dtype = getattr(xp, dtype)
878
+ data = xp.ones([1, 1], dtype=dtype)
879
+ out = ndimage.binary_erosion(data, border_value=1)
880
+ assert_array_almost_equal(out, xp.asarray([[1]]))
881
+
882
+ @pytest.mark.parametrize('dtype', types)
883
+ def test_binary_erosion17(self, dtype, xp):
884
+ dtype = getattr(xp, dtype)
885
+ data = xp.ones([1, 1], dtype=dtype)
886
+ out = ndimage.binary_erosion(data)
887
+ assert_array_almost_equal(out, xp.asarray([[0]]))
888
+
889
+ @pytest.mark.parametrize('dtype', types)
890
+ def test_binary_erosion18(self, dtype, xp):
891
+ dtype = getattr(xp, dtype)
892
+ data = xp.ones([1, 3], dtype=dtype)
893
+ out = ndimage.binary_erosion(data)
894
+ assert_array_almost_equal(out, xp.asarray([[0, 0, 0]]))
895
+
896
+ @pytest.mark.parametrize('dtype', types)
897
+ def test_binary_erosion19(self, dtype, xp):
898
+ dtype = getattr(xp, dtype)
899
+ data = xp.ones([1, 3], dtype=dtype)
900
+ out = ndimage.binary_erosion(data, border_value=1)
901
+ assert_array_almost_equal(out, xp.asarray([[1, 1, 1]]))
902
+
903
+ @pytest.mark.parametrize('dtype', types)
904
+ def test_binary_erosion20(self, dtype, xp):
905
+ dtype = getattr(xp, dtype)
906
+ data = xp.ones([3, 3], dtype=dtype)
907
+ out = ndimage.binary_erosion(data)
908
+ assert_array_almost_equal(out, xp.asarray([[0, 0, 0],
909
+ [0, 1, 0],
910
+ [0, 0, 0]]))
911
+
912
+ @pytest.mark.parametrize('dtype', types)
913
+ def test_binary_erosion21(self, dtype, xp):
914
+ dtype = getattr(xp, dtype)
915
+ data = xp.ones([3, 3], dtype=dtype)
916
+ out = ndimage.binary_erosion(data, border_value=1)
917
+ assert_array_almost_equal(out, xp.asarray([[1, 1, 1],
918
+ [1, 1, 1],
919
+ [1, 1, 1]]))
920
+
921
+ @pytest.mark.parametrize('dtype', types)
922
+ def test_binary_erosion22(self, dtype, xp):
923
+ dtype = getattr(xp, dtype)
924
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
925
+ [0, 0, 0, 0, 0, 0, 0, 0],
926
+ [0, 0, 0, 0, 0, 0, 0, 0],
927
+ [0, 0, 0, 0, 0, 1, 0, 0],
928
+ [0, 0, 0, 1, 1, 0, 0, 0],
929
+ [0, 0, 1, 0, 0, 1, 0, 0],
930
+ [0, 0, 0, 0, 0, 0, 0, 0],
931
+ [0, 0, 0, 0, 0, 0, 0, 0]]
932
+ expected = xp.asarray(expected)
933
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
934
+ [0, 1, 0, 0, 0, 0, 0, 0],
935
+ [0, 0, 0, 0, 0, 1, 1, 1],
936
+ [0, 0, 1, 1, 1, 1, 1, 1],
937
+ [0, 0, 1, 1, 1, 1, 0, 0],
938
+ [0, 1, 1, 1, 1, 1, 1, 0],
939
+ [0, 1, 1, 0, 0, 1, 1, 0],
940
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
941
+ out = ndimage.binary_erosion(data, border_value=1)
942
+ assert_array_almost_equal(out, expected)
943
+
944
+ @pytest.mark.parametrize('dtype', types)
945
+ def test_binary_erosion23(self, dtype, xp):
946
+ dtype = getattr(xp, dtype)
947
+ struct = ndimage.generate_binary_structure(2, 2)
948
+ struct = xp.asarray(struct)
949
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
950
+ [0, 0, 0, 0, 0, 0, 0, 0],
951
+ [0, 0, 0, 0, 0, 0, 0, 0],
952
+ [0, 0, 0, 0, 0, 0, 0, 0],
953
+ [0, 0, 0, 1, 1, 0, 0, 0],
954
+ [0, 0, 0, 0, 0, 0, 0, 0],
955
+ [0, 0, 0, 0, 0, 0, 0, 0],
956
+ [0, 0, 0, 0, 0, 0, 0, 0]]
957
+ expected = xp.asarray(expected)
958
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
959
+ [0, 1, 0, 0, 0, 0, 0, 0],
960
+ [0, 0, 0, 0, 0, 1, 1, 1],
961
+ [0, 0, 1, 1, 1, 1, 1, 1],
962
+ [0, 0, 1, 1, 1, 1, 0, 0],
963
+ [0, 1, 1, 1, 1, 1, 1, 0],
964
+ [0, 1, 1, 0, 0, 1, 1, 0],
965
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
966
+ out = ndimage.binary_erosion(data, struct, border_value=1)
967
+ assert_array_almost_equal(out, expected)
968
+
969
+ @pytest.mark.parametrize('dtype', types)
970
+ def test_binary_erosion24(self, dtype, xp):
971
+ dtype = getattr(xp, dtype)
972
+ struct = xp.asarray([[0, 1],
973
+ [1, 1]])
974
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
975
+ [0, 0, 0, 0, 0, 0, 0, 0],
976
+ [0, 0, 0, 0, 0, 0, 0, 0],
977
+ [0, 0, 0, 0, 0, 1, 1, 1],
978
+ [0, 0, 0, 1, 1, 1, 0, 0],
979
+ [0, 0, 1, 1, 1, 1, 0, 0],
980
+ [0, 0, 1, 0, 0, 0, 1, 0],
981
+ [0, 0, 0, 0, 0, 0, 0, 0]]
982
+ expected = xp.asarray(expected)
983
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
984
+ [0, 1, 0, 0, 0, 0, 0, 0],
985
+ [0, 0, 0, 0, 0, 1, 1, 1],
986
+ [0, 0, 1, 1, 1, 1, 1, 1],
987
+ [0, 0, 1, 1, 1, 1, 0, 0],
988
+ [0, 1, 1, 1, 1, 1, 1, 0],
989
+ [0, 1, 1, 0, 0, 1, 1, 0],
990
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
991
+ out = ndimage.binary_erosion(data, struct, border_value=1)
992
+ assert_array_almost_equal(out, expected)
993
+
994
+ @pytest.mark.parametrize('dtype', types)
995
+ def test_binary_erosion25(self, dtype, xp):
996
+ dtype = getattr(xp, dtype)
997
+ struct = [[0, 1, 0],
998
+ [1, 0, 1],
999
+ [0, 1, 0]]
1000
+ struct = xp.asarray(struct)
1001
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
1002
+ [0, 0, 0, 0, 0, 0, 0, 0],
1003
+ [0, 0, 0, 0, 0, 0, 0, 0],
1004
+ [0, 0, 0, 0, 0, 1, 0, 0],
1005
+ [0, 0, 0, 1, 0, 0, 0, 0],
1006
+ [0, 0, 1, 0, 0, 1, 0, 0],
1007
+ [0, 0, 0, 0, 0, 0, 0, 0],
1008
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1009
+ expected = xp.asarray(expected)
1010
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1011
+ [0, 1, 0, 0, 0, 0, 0, 0],
1012
+ [0, 0, 0, 0, 0, 1, 1, 1],
1013
+ [0, 0, 1, 1, 1, 0, 1, 1],
1014
+ [0, 0, 1, 0, 1, 1, 0, 0],
1015
+ [0, 1, 0, 1, 1, 1, 1, 0],
1016
+ [0, 1, 1, 0, 0, 1, 1, 0],
1017
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1018
+ out = ndimage.binary_erosion(data, struct, border_value=1)
1019
+ assert_array_almost_equal(out, expected)
1020
+
1021
+ @pytest.mark.parametrize('dtype', types)
1022
+ def test_binary_erosion26(self, dtype, xp):
1023
+ dtype = getattr(xp, dtype)
1024
+ struct = [[0, 1, 0],
1025
+ [1, 0, 1],
1026
+ [0, 1, 0]]
1027
+ struct = xp.asarray(struct)
1028
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
1029
+ [0, 0, 0, 0, 0, 0, 0, 1],
1030
+ [0, 0, 0, 0, 1, 0, 0, 1],
1031
+ [0, 0, 1, 0, 0, 0, 0, 0],
1032
+ [0, 1, 0, 0, 1, 0, 0, 0],
1033
+ [0, 0, 0, 0, 0, 0, 0, 0],
1034
+ [0, 0, 0, 0, 0, 0, 0, 0],
1035
+ [0, 0, 0, 0, 0, 0, 0, 1]]
1036
+ expected = xp.asarray(expected)
1037
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1038
+ [0, 1, 0, 0, 0, 0, 0, 0],
1039
+ [0, 0, 0, 0, 0, 1, 1, 1],
1040
+ [0, 0, 1, 1, 1, 0, 1, 1],
1041
+ [0, 0, 1, 0, 1, 1, 0, 0],
1042
+ [0, 1, 0, 1, 1, 1, 1, 0],
1043
+ [0, 1, 1, 0, 0, 1, 1, 0],
1044
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1045
+ out = ndimage.binary_erosion(data, struct, border_value=1,
1046
+ origin=(-1, -1))
1047
+ assert_array_almost_equal(out, expected)
1048
+
1049
+ @xfail_xp_backends(
1050
+ "cupy", reason="CuPy: NotImplementedError: only brute_force iteration"
1051
+ )
1052
+ def test_binary_erosion27(self, xp):
1053
+ struct = [[0, 1, 0],
1054
+ [1, 1, 1],
1055
+ [0, 1, 0]]
1056
+ struct = xp.asarray(struct)
1057
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1058
+ [0, 0, 0, 0, 0, 0, 0],
1059
+ [0, 0, 0, 0, 0, 0, 0],
1060
+ [0, 0, 0, 1, 0, 0, 0],
1061
+ [0, 0, 0, 0, 0, 0, 0],
1062
+ [0, 0, 0, 0, 0, 0, 0],
1063
+ [0, 0, 0, 0, 0, 0, 0]]
1064
+ expected = xp.asarray(expected)
1065
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0],
1066
+ [0, 0, 0, 1, 0, 0, 0],
1067
+ [0, 0, 1, 1, 1, 0, 0],
1068
+ [0, 1, 1, 1, 1, 1, 0],
1069
+ [0, 0, 1, 1, 1, 0, 0],
1070
+ [0, 0, 0, 1, 0, 0, 0],
1071
+ [0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1072
+ data = xp.asarray(data)
1073
+ out = ndimage.binary_erosion(data, struct, border_value=1,
1074
+ iterations=2)
1075
+ assert_array_almost_equal(out, expected)
1076
+
1077
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1078
+ reason='inplace out= arguments are numpy-specific')
1079
+ @xfail_xp_backends("cupy",
1080
+ reason="NotImplementedError: only brute_force iteration")
1081
+ def test_binary_erosion28(self, xp):
1082
+ struct = [[0, 1, 0],
1083
+ [1, 1, 1],
1084
+ [0, 1, 0]]
1085
+ struct = xp.asarray(struct)
1086
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1087
+ [0, 0, 0, 0, 0, 0, 0],
1088
+ [0, 0, 0, 0, 0, 0, 0],
1089
+ [0, 0, 0, 1, 0, 0, 0],
1090
+ [0, 0, 0, 0, 0, 0, 0],
1091
+ [0, 0, 0, 0, 0, 0, 0],
1092
+ [0, 0, 0, 0, 0, 0, 0]]
1093
+ expected = np.asarray(expected, dtype=bool)
1094
+ expected = xp.asarray(expected)
1095
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0],
1096
+ [0, 0, 0, 1, 0, 0, 0],
1097
+ [0, 0, 1, 1, 1, 0, 0],
1098
+ [0, 1, 1, 1, 1, 1, 0],
1099
+ [0, 0, 1, 1, 1, 0, 0],
1100
+ [0, 0, 0, 1, 0, 0, 0],
1101
+ [0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1102
+ data = xp.asarray(data)
1103
+ out = np.zeros(data.shape, dtype=bool)
1104
+ out = xp.asarray(out)
1105
+ ndimage.binary_erosion(data, struct, border_value=1,
1106
+ iterations=2, output=out)
1107
+ assert_array_almost_equal(out, expected)
1108
+
1109
+ @xfail_xp_backends(
1110
+ "cupy", reason="CuPy: NotImplementedError: only brute_force iteration"
1111
+ )
1112
+ def test_binary_erosion29(self, xp):
1113
+ struct = [[0, 1, 0],
1114
+ [1, 1, 1],
1115
+ [0, 1, 0]]
1116
+ struct = xp.asarray(struct)
1117
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1118
+ [0, 0, 0, 0, 0, 0, 0],
1119
+ [0, 0, 0, 0, 0, 0, 0],
1120
+ [0, 0, 0, 1, 0, 0, 0],
1121
+ [0, 0, 0, 0, 0, 0, 0],
1122
+ [0, 0, 0, 0, 0, 0, 0],
1123
+ [0, 0, 0, 0, 0, 0, 0]]
1124
+ expected = xp.asarray(expected)
1125
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1126
+ [0, 0, 1, 1, 1, 0, 0],
1127
+ [0, 1, 1, 1, 1, 1, 0],
1128
+ [1, 1, 1, 1, 1, 1, 1],
1129
+ [0, 1, 1, 1, 1, 1, 0],
1130
+ [0, 0, 1, 1, 1, 0, 0],
1131
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1132
+ data = xp.asarray(data)
1133
+ out = ndimage.binary_erosion(data, struct,
1134
+ border_value=1, iterations=3)
1135
+ assert_array_almost_equal(out, expected)
1136
+
1137
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1138
+ reason='inplace out= arguments are numpy-specific')
1139
+ @xfail_xp_backends("cupy",
1140
+ reason="NotImplementedError: only brute_force iteration")
1141
+ def test_binary_erosion30(self, xp):
1142
+ struct = [[0, 1, 0],
1143
+ [1, 1, 1],
1144
+ [0, 1, 0]]
1145
+ struct = xp.asarray(struct)
1146
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1147
+ [0, 0, 0, 0, 0, 0, 0],
1148
+ [0, 0, 0, 0, 0, 0, 0],
1149
+ [0, 0, 0, 1, 0, 0, 0],
1150
+ [0, 0, 0, 0, 0, 0, 0],
1151
+ [0, 0, 0, 0, 0, 0, 0],
1152
+ [0, 0, 0, 0, 0, 0, 0]]
1153
+ expected = np.asarray(expected, dtype=bool)
1154
+ expected = xp.asarray(expected)
1155
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1156
+ [0, 0, 1, 1, 1, 0, 0],
1157
+ [0, 1, 1, 1, 1, 1, 0],
1158
+ [1, 1, 1, 1, 1, 1, 1],
1159
+ [0, 1, 1, 1, 1, 1, 0],
1160
+ [0, 0, 1, 1, 1, 0, 0],
1161
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1162
+ data = xp.asarray(data)
1163
+ out = np.zeros(data.shape, dtype=bool)
1164
+ out = xp.asarray(out)
1165
+ ndimage.binary_erosion(data, struct, border_value=1,
1166
+ iterations=3, output=out)
1167
+ assert_array_almost_equal(out, expected)
1168
+
1169
+ # test with output memory overlap
1170
+ ndimage.binary_erosion(data, struct, border_value=1,
1171
+ iterations=3, output=data)
1172
+ assert_array_almost_equal(data, expected)
1173
+
1174
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1175
+ reason='inplace out= arguments are numpy-specific')
1176
+ def test_binary_erosion31(self, xp):
1177
+ struct = [[0, 1, 0],
1178
+ [1, 1, 1],
1179
+ [0, 1, 0]]
1180
+ struct = xp.asarray(struct)
1181
+ expected = [[0, 0, 1, 0, 0, 0, 0],
1182
+ [0, 1, 1, 1, 0, 0, 0],
1183
+ [1, 1, 1, 1, 1, 0, 1],
1184
+ [0, 1, 1, 1, 0, 0, 0],
1185
+ [0, 0, 1, 0, 0, 0, 0],
1186
+ [0, 0, 0, 0, 0, 0, 0],
1187
+ [0, 0, 1, 0, 0, 0, 1]]
1188
+ expected = np.asarray(expected, dtype=bool)
1189
+ expected = xp.asarray(expected)
1190
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1191
+ [0, 0, 1, 1, 1, 0, 0],
1192
+ [0, 1, 1, 1, 1, 1, 0],
1193
+ [1, 1, 1, 1, 1, 1, 1],
1194
+ [0, 1, 1, 1, 1, 1, 0],
1195
+ [0, 0, 1, 1, 1, 0, 0],
1196
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1197
+ data = xp.asarray(data)
1198
+ out = np.zeros(data.shape, dtype=bool)
1199
+ out = xp.asarray(out)
1200
+ ndimage.binary_erosion(data, struct, border_value=1,
1201
+ iterations=1, output=out, origin=(-1, -1))
1202
+ assert_array_almost_equal(out, expected)
1203
+
1204
+ @xfail_xp_backends("cupy",
1205
+ reason="NotImplementedError: only brute_force iteration")
1206
+ def test_binary_erosion32(self, xp):
1207
+ struct = [[0, 1, 0],
1208
+ [1, 1, 1],
1209
+ [0, 1, 0]]
1210
+ struct = xp.asarray(struct)
1211
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1212
+ [0, 0, 0, 0, 0, 0, 0],
1213
+ [0, 0, 0, 0, 0, 0, 0],
1214
+ [0, 0, 0, 1, 0, 0, 0],
1215
+ [0, 0, 0, 0, 0, 0, 0],
1216
+ [0, 0, 0, 0, 0, 0, 0],
1217
+ [0, 0, 0, 0, 0, 0, 0]]
1218
+ expected = xp.asarray(expected)
1219
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0],
1220
+ [0, 0, 0, 1, 0, 0, 0],
1221
+ [0, 0, 1, 1, 1, 0, 0],
1222
+ [0, 1, 1, 1, 1, 1, 0],
1223
+ [0, 0, 1, 1, 1, 0, 0],
1224
+ [0, 0, 0, 1, 0, 0, 0],
1225
+ [0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1226
+ data = xp.asarray(data)
1227
+ out = ndimage.binary_erosion(data, struct,
1228
+ border_value=1, iterations=2)
1229
+ assert_array_almost_equal(out, expected)
1230
+
1231
+ @xfail_xp_backends("cupy",
1232
+ reason="NotImplementedError: only brute_force iteration")
1233
+ def test_binary_erosion33(self, xp):
1234
+ struct = [[0, 1, 0],
1235
+ [1, 1, 1],
1236
+ [0, 1, 0]]
1237
+ struct = xp.asarray(struct)
1238
+ expected = [[0, 0, 0, 0, 0, 1, 1],
1239
+ [0, 0, 0, 0, 0, 0, 1],
1240
+ [0, 0, 0, 0, 0, 0, 0],
1241
+ [0, 0, 0, 0, 0, 0, 0],
1242
+ [0, 0, 0, 0, 0, 0, 0],
1243
+ [0, 0, 0, 0, 0, 0, 0],
1244
+ [0, 0, 0, 0, 0, 0, 0]]
1245
+ expected = xp.asarray(expected)
1246
+ mask = [[1, 1, 1, 1, 1, 0, 0],
1247
+ [1, 1, 1, 1, 1, 1, 0],
1248
+ [1, 1, 1, 1, 1, 1, 1],
1249
+ [1, 1, 1, 1, 1, 1, 1],
1250
+ [1, 1, 1, 1, 1, 1, 1],
1251
+ [1, 1, 1, 1, 1, 1, 1],
1252
+ [1, 1, 1, 1, 1, 1, 1]]
1253
+ mask = xp.asarray(mask)
1254
+ data = np.asarray([[0, 0, 0, 0, 0, 1, 1],
1255
+ [0, 0, 0, 1, 0, 0, 1],
1256
+ [0, 0, 1, 1, 1, 0, 0],
1257
+ [0, 0, 1, 1, 1, 0, 0],
1258
+ [0, 0, 1, 1, 1, 0, 0],
1259
+ [0, 0, 0, 1, 0, 0, 0],
1260
+ [0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1261
+ data = xp.asarray(data)
1262
+ out = ndimage.binary_erosion(data, struct,
1263
+ border_value=1, mask=mask, iterations=-1)
1264
+ assert_array_almost_equal(out, expected)
1265
+
1266
+ def test_binary_erosion34(self, xp):
1267
+ struct = [[0, 1, 0],
1268
+ [1, 1, 1],
1269
+ [0, 1, 0]]
1270
+ struct = xp.asarray(struct)
1271
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1272
+ [0, 0, 0, 1, 0, 0, 0],
1273
+ [0, 0, 0, 1, 0, 0, 0],
1274
+ [0, 1, 1, 1, 1, 1, 0],
1275
+ [0, 0, 0, 1, 0, 0, 0],
1276
+ [0, 0, 0, 1, 0, 0, 0],
1277
+ [0, 0, 0, 0, 0, 0, 0]]
1278
+ expected = xp.asarray(expected)
1279
+ mask = [[0, 0, 0, 0, 0, 0, 0],
1280
+ [0, 0, 0, 0, 0, 0, 0],
1281
+ [0, 0, 1, 1, 1, 0, 0],
1282
+ [0, 0, 1, 0, 1, 0, 0],
1283
+ [0, 0, 1, 1, 1, 0, 0],
1284
+ [0, 0, 0, 0, 0, 0, 0],
1285
+ [0, 0, 0, 0, 0, 0, 0]]
1286
+ mask = xp.asarray(mask)
1287
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0],
1288
+ [0, 0, 0, 1, 0, 0, 0],
1289
+ [0, 0, 1, 1, 1, 0, 0],
1290
+ [0, 1, 1, 1, 1, 1, 0],
1291
+ [0, 0, 1, 1, 1, 0, 0],
1292
+ [0, 0, 0, 1, 0, 0, 0],
1293
+ [0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1294
+ data = xp.asarray(data)
1295
+ out = ndimage.binary_erosion(data, struct,
1296
+ border_value=1, mask=mask)
1297
+ assert_array_almost_equal(out, expected)
1298
+
1299
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1300
+ reason='inplace out= arguments are numpy-specific')
1301
+ def test_binary_erosion35(self, xp):
1302
+ struct = [[0, 1, 0],
1303
+ [1, 1, 1],
1304
+ [0, 1, 0]]
1305
+ struct = xp.asarray(struct)
1306
+ mask = [[0, 0, 0, 0, 0, 0, 0],
1307
+ [0, 0, 0, 0, 0, 0, 0],
1308
+ [0, 0, 1, 1, 1, 0, 0],
1309
+ [0, 0, 1, 0, 1, 0, 0],
1310
+ [0, 0, 1, 1, 1, 0, 0],
1311
+ [0, 0, 0, 0, 0, 0, 0],
1312
+ [0, 0, 0, 0, 0, 0, 0]]
1313
+ mask = np.asarray(mask, dtype=bool)
1314
+ mask = xp.asarray(mask)
1315
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1316
+ [0, 0, 1, 1, 1, 0, 0],
1317
+ [0, 1, 1, 1, 1, 1, 0],
1318
+ [1, 1, 1, 1, 1, 1, 1],
1319
+ [0, 1, 1, 1, 1, 1, 0],
1320
+ [0, 0, 1, 1, 1, 0, 0],
1321
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1322
+ data = xp.asarray(data)
1323
+ tmp = [[0, 0, 1, 0, 0, 0, 0],
1324
+ [0, 1, 1, 1, 0, 0, 0],
1325
+ [1, 1, 1, 1, 1, 0, 1],
1326
+ [0, 1, 1, 1, 0, 0, 0],
1327
+ [0, 0, 1, 0, 0, 0, 0],
1328
+ [0, 0, 0, 0, 0, 0, 0],
1329
+ [0, 0, 1, 0, 0, 0, 1]]
1330
+ tmp = np.asarray(tmp, dtype=bool)
1331
+ tmp = xp.asarray(tmp)
1332
+ expected = xp.logical_and(tmp, mask)
1333
+ tmp = xp.logical_and(data, xp.logical_not(mask))
1334
+ expected = xp.logical_or(expected, tmp)
1335
+ out = np.zeros(data.shape, dtype=bool)
1336
+ out = xp.asarray(out)
1337
+ ndimage.binary_erosion(data, struct, border_value=1,
1338
+ iterations=1, output=out,
1339
+ origin=(-1, -1), mask=mask)
1340
+ assert_array_almost_equal(out, expected)
1341
+
1342
+ @xfail_xp_backends("cupy",
1343
+ reason="NotImplementedError: only brute_force iteration")
1344
+ def test_binary_erosion36(self, xp):
1345
+ struct = [[0, 1, 0],
1346
+ [1, 0, 1],
1347
+ [0, 1, 0]]
1348
+ struct = xp.asarray(struct)
1349
+ mask = [[0, 0, 0, 0, 0, 0, 0, 0],
1350
+ [0, 0, 0, 0, 0, 0, 0, 0],
1351
+ [0, 0, 1, 1, 1, 0, 0, 0],
1352
+ [0, 0, 1, 0, 1, 0, 0, 0],
1353
+ [0, 0, 1, 1, 1, 0, 0, 0],
1354
+ [0, 0, 1, 1, 1, 0, 0, 0],
1355
+ [0, 0, 1, 1, 1, 0, 0, 0],
1356
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1357
+ mask = np.asarray(mask, dtype=bool)
1358
+ mask = xp.asarray(mask)
1359
+ tmp = [[0, 0, 0, 0, 0, 0, 0, 0],
1360
+ [0, 0, 0, 0, 0, 0, 0, 1],
1361
+ [0, 0, 0, 0, 1, 0, 0, 1],
1362
+ [0, 0, 1, 0, 0, 0, 0, 0],
1363
+ [0, 1, 0, 0, 1, 0, 0, 0],
1364
+ [0, 0, 0, 0, 0, 0, 0, 0],
1365
+ [0, 0, 0, 0, 0, 0, 0, 0],
1366
+ [0, 0, 0, 0, 0, 0, 0, 1]]
1367
+ tmp = np.asarray(tmp, dtype=bool)
1368
+ tmp = xp.asarray(tmp)
1369
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1370
+ [0, 1, 0, 0, 0, 0, 0, 0],
1371
+ [0, 0, 0, 0, 0, 1, 1, 1],
1372
+ [0, 0, 1, 1, 1, 0, 1, 1],
1373
+ [0, 0, 1, 0, 1, 1, 0, 0],
1374
+ [0, 1, 0, 1, 1, 1, 1, 0],
1375
+ [0, 1, 1, 0, 0, 1, 1, 0],
1376
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1377
+ data = xp.asarray(data)
1378
+ expected = xp.logical_and(tmp, mask)
1379
+ tmp = xp.logical_and(data, xp.logical_not(mask))
1380
+ expected = xp.logical_or(expected, tmp)
1381
+ out = ndimage.binary_erosion(data, struct, mask=mask,
1382
+ border_value=1, origin=(-1, -1))
1383
+ assert_array_almost_equal(out, expected)
1384
+
1385
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1386
+ reason='inplace out= arguments are numpy-specific')
1387
+ @xfail_xp_backends("cupy",
1388
+ reason="NotImplementedError: only brute_force iteration")
1389
+ def test_binary_erosion37(self, xp):
1390
+ a = np.asarray([[1, 0, 1],
1391
+ [0, 1, 0],
1392
+ [1, 0, 1]], dtype=bool)
1393
+ a = xp.asarray(a)
1394
+ b = xp.zeros_like(a)
1395
+ out = ndimage.binary_erosion(a, structure=a, output=b, iterations=0,
1396
+ border_value=True, brute_force=True)
1397
+ assert out is b
1398
+ xp_assert_equal(
1399
+ ndimage.binary_erosion(a, structure=a, iterations=0,
1400
+ border_value=True),
1401
+ b)
1402
+
1403
+ def test_binary_erosion38(self, xp):
1404
+ data = np.asarray([[1, 0, 1],
1405
+ [0, 1, 0],
1406
+ [1, 0, 1]], dtype=bool)
1407
+ data = xp.asarray(data)
1408
+ iterations = 2.0
1409
+ with assert_raises(TypeError):
1410
+ _ = ndimage.binary_erosion(data, iterations=iterations)
1411
+
1412
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1413
+ reason='inplace out= arguments are numpy-specific')
1414
+ @xfail_xp_backends("cupy",
1415
+ reason="NotImplementedError: only brute_force iteration")
1416
+ def test_binary_erosion39(self, xp):
1417
+ iterations = np.int32(3)
1418
+ struct = [[0, 1, 0],
1419
+ [1, 1, 1],
1420
+ [0, 1, 0]]
1421
+ struct = xp.asarray(struct)
1422
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1423
+ [0, 0, 0, 0, 0, 0, 0],
1424
+ [0, 0, 0, 0, 0, 0, 0],
1425
+ [0, 0, 0, 1, 0, 0, 0],
1426
+ [0, 0, 0, 0, 0, 0, 0],
1427
+ [0, 0, 0, 0, 0, 0, 0],
1428
+ [0, 0, 0, 0, 0, 0, 0]]
1429
+ expected = xp.asarray(expected, dtype=bool)
1430
+ expected = xp.asarray(expected)
1431
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1432
+ [0, 0, 1, 1, 1, 0, 0],
1433
+ [0, 1, 1, 1, 1, 1, 0],
1434
+ [1, 1, 1, 1, 1, 1, 1],
1435
+ [0, 1, 1, 1, 1, 1, 0],
1436
+ [0, 0, 1, 1, 1, 0, 0],
1437
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1438
+ data = xp.asarray(data)
1439
+ out = np.zeros(data.shape, dtype=bool)
1440
+ out = xp.asarray(out)
1441
+ ndimage.binary_erosion(data, struct, border_value=1,
1442
+ iterations=iterations, output=out)
1443
+ assert_array_almost_equal(out, expected)
1444
+
1445
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1446
+ reason='inplace out= arguments are numpy-specific')
1447
+ @xfail_xp_backends("cupy",
1448
+ reason="NotImplementedError: only brute_force iteration")
1449
+ def test_binary_erosion40(self, xp):
1450
+ iterations = np.int64(3)
1451
+ struct = [[0, 1, 0],
1452
+ [1, 1, 1],
1453
+ [0, 1, 0]]
1454
+ struct = xp.asarray(struct)
1455
+ expected = [[0, 0, 0, 0, 0, 0, 0],
1456
+ [0, 0, 0, 0, 0, 0, 0],
1457
+ [0, 0, 0, 0, 0, 0, 0],
1458
+ [0, 0, 0, 1, 0, 0, 0],
1459
+ [0, 0, 0, 0, 0, 0, 0],
1460
+ [0, 0, 0, 0, 0, 0, 0],
1461
+ [0, 0, 0, 0, 0, 0, 0]]
1462
+ expected = np.asarray(expected, dtype=bool)
1463
+ expected = xp.asarray(expected)
1464
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
1465
+ [0, 0, 1, 1, 1, 0, 0],
1466
+ [0, 1, 1, 1, 1, 1, 0],
1467
+ [1, 1, 1, 1, 1, 1, 1],
1468
+ [0, 1, 1, 1, 1, 1, 0],
1469
+ [0, 0, 1, 1, 1, 0, 0],
1470
+ [0, 0, 0, 1, 0, 0, 0]], dtype=bool)
1471
+ data = xp.asarray(data)
1472
+ out = np.zeros(data.shape, dtype=bool)
1473
+ out = xp.asarray(out)
1474
+ ndimage.binary_erosion(data, struct, border_value=1,
1475
+ iterations=iterations, output=out)
1476
+ assert_array_almost_equal(out, expected)
1477
+
1478
+ @pytest.mark.parametrize('dtype', types)
1479
+ def test_binary_dilation01(self, dtype, xp):
1480
+ dtype = getattr(xp, dtype)
1481
+ data = xp.ones([], dtype=dtype)
1482
+ out = ndimage.binary_dilation(data)
1483
+ assert out == xp.asarray(1, dtype=out.dtype)
1484
+
1485
+ @pytest.mark.parametrize('dtype', types)
1486
+ def test_binary_dilation02(self, dtype, xp):
1487
+ dtype = getattr(xp, dtype)
1488
+ data = xp.zeros([], dtype=dtype)
1489
+ out = ndimage.binary_dilation(data)
1490
+ assert out == xp.asarray(False)
1491
+
1492
+ @pytest.mark.parametrize('dtype', types)
1493
+ def test_binary_dilation03(self, dtype, xp):
1494
+ dtype = getattr(xp, dtype)
1495
+ data = xp.ones([1], dtype=dtype)
1496
+ out = ndimage.binary_dilation(data)
1497
+ assert_array_almost_equal(out, xp.asarray([1], dtype=out.dtype))
1498
+
1499
+ @pytest.mark.parametrize('dtype', types)
1500
+ def test_binary_dilation04(self, dtype, xp):
1501
+ dtype = getattr(xp, dtype)
1502
+ data = xp.zeros([1], dtype=dtype)
1503
+ out = ndimage.binary_dilation(data)
1504
+ assert_array_almost_equal(out, xp.asarray([0]))
1505
+
1506
+ @pytest.mark.parametrize('dtype', types)
1507
+ def test_binary_dilation05(self, dtype, xp):
1508
+ dtype = getattr(xp, dtype)
1509
+ data = xp.ones([3], dtype=dtype)
1510
+ out = ndimage.binary_dilation(data)
1511
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1]))
1512
+
1513
+ @pytest.mark.parametrize('dtype', types)
1514
+ def test_binary_dilation05_broadcasted(self, dtype, xp):
1515
+ dtype = getattr(xp, dtype)
1516
+ data = xp.ones((1, ), dtype=dtype)
1517
+ data = xp.broadcast_to(data, (3,))
1518
+ out = ndimage.binary_dilation(data)
1519
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1]))
1520
+
1521
+ @pytest.mark.parametrize('dtype', types)
1522
+ def test_binary_dilation06(self, dtype, xp):
1523
+ dtype = getattr(xp, dtype)
1524
+ data = xp.zeros([3], dtype=dtype)
1525
+ out = ndimage.binary_dilation(data)
1526
+ assert_array_almost_equal(out, xp.asarray([0, 0, 0]))
1527
+
1528
+ @pytest.mark.parametrize('dtype', types)
1529
+ def test_binary_dilation07(self, dtype, xp):
1530
+ data = np.zeros([3], dtype=dtype)
1531
+ data[1] = 1
1532
+ data = xp.asarray(data)
1533
+ out = ndimage.binary_dilation(data)
1534
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1]))
1535
+
1536
+ @pytest.mark.parametrize('dtype', types)
1537
+ def test_binary_dilation08(self, dtype, xp):
1538
+ data = np.zeros([5], dtype=dtype)
1539
+ data[1] = 1
1540
+ data[3] = 1
1541
+ data = xp.asarray(data)
1542
+ out = ndimage.binary_dilation(data)
1543
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1, 1, 1]))
1544
+
1545
+ @pytest.mark.parametrize('dtype', types)
1546
+ def test_binary_dilation09(self, dtype, xp):
1547
+ data = np.zeros([5], dtype=dtype)
1548
+ data[1] = 1
1549
+ data = xp.asarray(data)
1550
+ out = ndimage.binary_dilation(data)
1551
+ assert_array_almost_equal(out, xp.asarray([1, 1, 1, 0, 0]))
1552
+
1553
+ @pytest.mark.parametrize('dtype', types)
1554
+ def test_binary_dilation10(self, dtype, xp):
1555
+ data = np.zeros([5], dtype=dtype)
1556
+ data[1] = 1
1557
+ data = xp.asarray(data)
1558
+ out = ndimage.binary_dilation(data, origin=-1)
1559
+ assert_array_almost_equal(out, xp.asarray([0, 1, 1, 1, 0]))
1560
+
1561
+ @pytest.mark.parametrize('dtype', types)
1562
+ def test_binary_dilation11(self, dtype, xp):
1563
+ data = np.zeros([5], dtype=dtype)
1564
+ data[1] = 1
1565
+ data = xp.asarray(data)
1566
+ out = ndimage.binary_dilation(data, origin=1)
1567
+ assert_array_almost_equal(out, xp.asarray([1, 1, 0, 0, 0]))
1568
+
1569
+ @pytest.mark.parametrize('dtype', types)
1570
+ def test_binary_dilation12(self, dtype, xp):
1571
+ data = np.zeros([5], dtype=dtype)
1572
+ data[1] = 1
1573
+ data = xp.asarray(data)
1574
+ struct = xp.asarray([1, 0, 1])
1575
+ out = ndimage.binary_dilation(data, struct)
1576
+ assert_array_almost_equal(out, xp.asarray([1, 0, 1, 0, 0]))
1577
+
1578
+ @pytest.mark.parametrize('dtype', types)
1579
+ def test_binary_dilation13(self, dtype, xp):
1580
+ data = np.zeros([5], dtype=dtype)
1581
+ data[1] = 1
1582
+ data = xp.asarray(data)
1583
+ struct = xp.asarray([1, 0, 1])
1584
+ out = ndimage.binary_dilation(data, struct, border_value=1)
1585
+ assert_array_almost_equal(out, xp.asarray([1, 0, 1, 0, 1]))
1586
+
1587
+ @pytest.mark.parametrize('dtype', types)
1588
+ def test_binary_dilation14(self, dtype, xp):
1589
+ data = np.zeros([5], dtype=dtype)
1590
+ data[1] = 1
1591
+ data = xp.asarray(data)
1592
+ struct = xp.asarray([1, 0, 1])
1593
+ out = ndimage.binary_dilation(data, struct, origin=-1)
1594
+ assert_array_almost_equal(out, xp.asarray([0, 1, 0, 1, 0]))
1595
+
1596
+ @pytest.mark.parametrize('dtype', types)
1597
+ def test_binary_dilation15(self, dtype, xp):
1598
+ data = np.zeros([5], dtype=dtype)
1599
+ data[1] = 1
1600
+ data = xp.asarray(data)
1601
+ struct = xp.asarray([1, 0, 1])
1602
+ out = ndimage.binary_dilation(data, struct,
1603
+ origin=-1, border_value=1)
1604
+ assert_array_almost_equal(out, xp.asarray([1, 1, 0, 1, 0]))
1605
+
1606
+ @pytest.mark.parametrize('dtype', types)
1607
+ def test_binary_dilation16(self, dtype, xp):
1608
+ dtype = getattr(xp, dtype)
1609
+ data = xp.ones([1, 1], dtype=dtype)
1610
+ out = ndimage.binary_dilation(data)
1611
+ assert_array_almost_equal(out, xp.asarray([[1]]))
1612
+
1613
+ @pytest.mark.parametrize('dtype', types)
1614
+ def test_binary_dilation17(self, dtype, xp):
1615
+ dtype = getattr(xp, dtype)
1616
+ data = xp.zeros([1, 1], dtype=dtype)
1617
+ out = ndimage.binary_dilation(data)
1618
+ assert_array_almost_equal(out, xp.asarray([[0]]))
1619
+
1620
+ @pytest.mark.parametrize('dtype', types)
1621
+ def test_binary_dilation18(self, dtype, xp):
1622
+ dtype = getattr(xp, dtype)
1623
+ data = xp.ones([1, 3], dtype=dtype)
1624
+ out = ndimage.binary_dilation(data)
1625
+ assert_array_almost_equal(out, xp.asarray([[1, 1, 1]]))
1626
+
1627
+ @pytest.mark.parametrize('dtype', types)
1628
+ def test_binary_dilation19(self, dtype, xp):
1629
+ dtype = getattr(xp, dtype)
1630
+ data = xp.ones([3, 3], dtype=dtype)
1631
+ out = ndimage.binary_dilation(data)
1632
+ assert_array_almost_equal(out, xp.asarray([[1, 1, 1],
1633
+ [1, 1, 1],
1634
+ [1, 1, 1]]))
1635
+
1636
+ @pytest.mark.parametrize('dtype', types)
1637
+ def test_binary_dilation20(self, dtype, xp):
1638
+ data = np.zeros([3, 3], dtype=dtype)
1639
+ data[1, 1] = 1
1640
+ data = xp.asarray(data)
1641
+ out = ndimage.binary_dilation(data)
1642
+ assert_array_almost_equal(out, xp.asarray([[0, 1, 0],
1643
+ [1, 1, 1],
1644
+ [0, 1, 0]]))
1645
+
1646
+ @pytest.mark.parametrize('dtype', types)
1647
+ def test_binary_dilation21(self, dtype, xp):
1648
+ struct = ndimage.generate_binary_structure(2, 2)
1649
+ struct = xp.asarray(struct)
1650
+ data = np.zeros([3, 3], dtype=dtype)
1651
+ data[1, 1] = 1
1652
+ data = xp.asarray(data)
1653
+ out = ndimage.binary_dilation(data, struct)
1654
+ assert_array_almost_equal(out, xp.asarray([[1, 1, 1],
1655
+ [1, 1, 1],
1656
+ [1, 1, 1]]))
1657
+
1658
+ @pytest.mark.parametrize('dtype', types)
1659
+ def test_binary_dilation22(self, dtype, xp):
1660
+ dtype = getattr(xp, dtype)
1661
+ expected = [[0, 1, 0, 0, 0, 0, 0, 0],
1662
+ [1, 1, 1, 0, 0, 0, 0, 0],
1663
+ [0, 1, 0, 0, 0, 1, 0, 0],
1664
+ [0, 0, 0, 1, 1, 1, 1, 0],
1665
+ [0, 0, 1, 1, 1, 1, 0, 0],
1666
+ [0, 1, 1, 1, 1, 1, 1, 0],
1667
+ [0, 0, 1, 0, 0, 1, 0, 0],
1668
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1669
+ expected = xp.asarray(expected)
1670
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1671
+ [0, 1, 0, 0, 0, 0, 0, 0],
1672
+ [0, 0, 0, 0, 0, 0, 0, 0],
1673
+ [0, 0, 0, 0, 0, 1, 0, 0],
1674
+ [0, 0, 0, 1, 1, 0, 0, 0],
1675
+ [0, 0, 1, 0, 0, 1, 0, 0],
1676
+ [0, 0, 0, 0, 0, 0, 0, 0],
1677
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1678
+ out = ndimage.binary_dilation(data)
1679
+ assert_array_almost_equal(out, expected)
1680
+
1681
+ @pytest.mark.parametrize('dtype', types)
1682
+ def test_binary_dilation23(self, dtype, xp):
1683
+ dtype = getattr(xp, dtype)
1684
+ expected = [[1, 1, 1, 1, 1, 1, 1, 1],
1685
+ [1, 1, 1, 0, 0, 0, 0, 1],
1686
+ [1, 1, 0, 0, 0, 1, 0, 1],
1687
+ [1, 0, 0, 1, 1, 1, 1, 1],
1688
+ [1, 0, 1, 1, 1, 1, 0, 1],
1689
+ [1, 1, 1, 1, 1, 1, 1, 1],
1690
+ [1, 0, 1, 0, 0, 1, 0, 1],
1691
+ [1, 1, 1, 1, 1, 1, 1, 1]]
1692
+ expected = xp.asarray(expected)
1693
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1694
+ [0, 1, 0, 0, 0, 0, 0, 0],
1695
+ [0, 0, 0, 0, 0, 0, 0, 0],
1696
+ [0, 0, 0, 0, 0, 1, 0, 0],
1697
+ [0, 0, 0, 1, 1, 0, 0, 0],
1698
+ [0, 0, 1, 0, 0, 1, 0, 0],
1699
+ [0, 0, 0, 0, 0, 0, 0, 0],
1700
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1701
+ out = ndimage.binary_dilation(data, border_value=1)
1702
+ assert_array_almost_equal(out, expected)
1703
+
1704
+ @pytest.mark.parametrize('dtype', types)
1705
+ def test_binary_dilation24(self, dtype, xp):
1706
+ dtype = getattr(xp, dtype)
1707
+ expected = [[1, 1, 0, 0, 0, 0, 0, 0],
1708
+ [1, 0, 0, 0, 1, 0, 0, 0],
1709
+ [0, 0, 1, 1, 1, 1, 0, 0],
1710
+ [0, 1, 1, 1, 1, 0, 0, 0],
1711
+ [1, 1, 1, 1, 1, 1, 0, 0],
1712
+ [0, 1, 0, 0, 1, 0, 0, 0],
1713
+ [0, 0, 0, 0, 0, 0, 0, 0],
1714
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1715
+ expected = xp.asarray(expected)
1716
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1717
+ [0, 1, 0, 0, 0, 0, 0, 0],
1718
+ [0, 0, 0, 0, 0, 0, 0, 0],
1719
+ [0, 0, 0, 0, 0, 1, 0, 0],
1720
+ [0, 0, 0, 1, 1, 0, 0, 0],
1721
+ [0, 0, 1, 0, 0, 1, 0, 0],
1722
+ [0, 0, 0, 0, 0, 0, 0, 0],
1723
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1724
+ out = ndimage.binary_dilation(data, origin=(1, 1))
1725
+ assert_array_almost_equal(out, expected)
1726
+
1727
+ @pytest.mark.parametrize('dtype', types)
1728
+ def test_binary_dilation25(self, dtype, xp):
1729
+ dtype = getattr(xp, dtype)
1730
+ expected = [[1, 1, 0, 0, 0, 0, 1, 1],
1731
+ [1, 0, 0, 0, 1, 0, 1, 1],
1732
+ [0, 0, 1, 1, 1, 1, 1, 1],
1733
+ [0, 1, 1, 1, 1, 0, 1, 1],
1734
+ [1, 1, 1, 1, 1, 1, 1, 1],
1735
+ [0, 1, 0, 0, 1, 0, 1, 1],
1736
+ [1, 1, 1, 1, 1, 1, 1, 1],
1737
+ [1, 1, 1, 1, 1, 1, 1, 1]]
1738
+ expected = xp.asarray(expected)
1739
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1740
+ [0, 1, 0, 0, 0, 0, 0, 0],
1741
+ [0, 0, 0, 0, 0, 0, 0, 0],
1742
+ [0, 0, 0, 0, 0, 1, 0, 0],
1743
+ [0, 0, 0, 1, 1, 0, 0, 0],
1744
+ [0, 0, 1, 0, 0, 1, 0, 0],
1745
+ [0, 0, 0, 0, 0, 0, 0, 0],
1746
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1747
+ out = ndimage.binary_dilation(data, origin=(1, 1), border_value=1)
1748
+ assert_array_almost_equal(out, expected)
1749
+
1750
+ @pytest.mark.parametrize('dtype', types)
1751
+ def test_binary_dilation26(self, dtype, xp):
1752
+ dtype = getattr(xp, dtype)
1753
+ struct = ndimage.generate_binary_structure(2, 2)
1754
+ expected = [[1, 1, 1, 0, 0, 0, 0, 0],
1755
+ [1, 1, 1, 0, 0, 0, 0, 0],
1756
+ [1, 1, 1, 0, 1, 1, 1, 0],
1757
+ [0, 0, 1, 1, 1, 1, 1, 0],
1758
+ [0, 1, 1, 1, 1, 1, 1, 0],
1759
+ [0, 1, 1, 1, 1, 1, 1, 0],
1760
+ [0, 1, 1, 1, 1, 1, 1, 0],
1761
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1762
+ struct = xp.asarray(struct)
1763
+ expected = xp.asarray(expected)
1764
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1765
+ [0, 1, 0, 0, 0, 0, 0, 0],
1766
+ [0, 0, 0, 0, 0, 0, 0, 0],
1767
+ [0, 0, 0, 0, 0, 1, 0, 0],
1768
+ [0, 0, 0, 1, 1, 0, 0, 0],
1769
+ [0, 0, 1, 0, 0, 1, 0, 0],
1770
+ [0, 0, 0, 0, 0, 0, 0, 0],
1771
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1772
+ out = ndimage.binary_dilation(data, struct)
1773
+ assert_array_almost_equal(out, expected)
1774
+
1775
+ @pytest.mark.parametrize('dtype', types)
1776
+ def test_binary_dilation27(self, dtype, xp):
1777
+ dtype = getattr(xp, dtype)
1778
+ struct = [[0, 1],
1779
+ [1, 1]]
1780
+ expected = [[0, 1, 0, 0, 0, 0, 0, 0],
1781
+ [1, 1, 0, 0, 0, 0, 0, 0],
1782
+ [0, 0, 0, 0, 0, 1, 0, 0],
1783
+ [0, 0, 0, 1, 1, 1, 0, 0],
1784
+ [0, 0, 1, 1, 1, 1, 0, 0],
1785
+ [0, 1, 1, 0, 1, 1, 0, 0],
1786
+ [0, 0, 0, 0, 0, 0, 0, 0],
1787
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1788
+ struct = xp.asarray(struct)
1789
+ expected = xp.asarray(expected)
1790
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1791
+ [0, 1, 0, 0, 0, 0, 0, 0],
1792
+ [0, 0, 0, 0, 0, 0, 0, 0],
1793
+ [0, 0, 0, 0, 0, 1, 0, 0],
1794
+ [0, 0, 0, 1, 1, 0, 0, 0],
1795
+ [0, 0, 1, 0, 0, 1, 0, 0],
1796
+ [0, 0, 0, 0, 0, 0, 0, 0],
1797
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
1798
+ out = ndimage.binary_dilation(data, struct)
1799
+ assert_array_almost_equal(out, expected)
1800
+
1801
+ @pytest.mark.parametrize('dtype', types)
1802
+ def test_binary_dilation28(self, dtype, xp):
1803
+ dtype = getattr(xp, dtype)
1804
+ expected = [[1, 1, 1, 1],
1805
+ [1, 0, 0, 1],
1806
+ [1, 0, 0, 1],
1807
+ [1, 1, 1, 1]]
1808
+ expected = xp.asarray(expected)
1809
+ data = xp.asarray([[0, 0, 0, 0],
1810
+ [0, 0, 0, 0],
1811
+ [0, 0, 0, 0],
1812
+ [0, 0, 0, 0]], dtype=dtype)
1813
+ out = ndimage.binary_dilation(data, border_value=1)
1814
+ assert_array_almost_equal(out, expected)
1815
+
1816
+ @xfail_xp_backends("cupy",
1817
+ reason="NotImplementedError: only brute_force iteration")
1818
+ def test_binary_dilation29(self, xp):
1819
+ struct = [[0, 1],
1820
+ [1, 1]]
1821
+ expected = [[0, 0, 0, 0, 0],
1822
+ [0, 0, 0, 1, 0],
1823
+ [0, 0, 1, 1, 0],
1824
+ [0, 1, 1, 1, 0],
1825
+ [0, 0, 0, 0, 0]]
1826
+ struct = xp.asarray(struct)
1827
+ expected = xp.asarray(expected)
1828
+ data = np.asarray([[0, 0, 0, 0, 0],
1829
+ [0, 0, 0, 0, 0],
1830
+ [0, 0, 0, 0, 0],
1831
+ [0, 0, 0, 1, 0],
1832
+ [0, 0, 0, 0, 0]], dtype=bool)
1833
+ data = xp.asarray(data)
1834
+ out = ndimage.binary_dilation(data, struct, iterations=2)
1835
+ assert_array_almost_equal(out, expected)
1836
+
1837
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1838
+ reason='output= arrays are numpy-specific')
1839
+ @xfail_xp_backends("cupy",
1840
+ reason="NotImplementedError: only brute_force iteration")
1841
+ def test_binary_dilation30(self, xp):
1842
+ struct = [[0, 1],
1843
+ [1, 1]]
1844
+ expected = [[0, 0, 0, 0, 0],
1845
+ [0, 0, 0, 1, 0],
1846
+ [0, 0, 1, 1, 0],
1847
+ [0, 1, 1, 1, 0],
1848
+ [0, 0, 0, 0, 0]]
1849
+ struct = xp.asarray(struct)
1850
+ expected = xp.asarray(expected)
1851
+ data = xp.asarray([[0, 0, 0, 0, 0],
1852
+ [0, 0, 0, 0, 0],
1853
+ [0, 0, 0, 0, 0],
1854
+ [0, 0, 0, 1, 0],
1855
+ [0, 0, 0, 0, 0]], dtype=bool)
1856
+ data = xp.asarray(data)
1857
+ out = np.zeros(data.shape, dtype=bool)
1858
+ out = xp.asarray(out)
1859
+ ndimage.binary_dilation(data, struct, iterations=2, output=out)
1860
+ assert_array_almost_equal(out, expected)
1861
+
1862
+ @xfail_xp_backends("cupy",
1863
+ reason="NotImplementedError: only brute_force iteration")
1864
+ def test_binary_dilation31(self, xp):
1865
+ struct = [[0, 1],
1866
+ [1, 1]]
1867
+ expected = [[0, 0, 0, 1, 0],
1868
+ [0, 0, 1, 1, 0],
1869
+ [0, 1, 1, 1, 0],
1870
+ [1, 1, 1, 1, 0],
1871
+ [0, 0, 0, 0, 0]]
1872
+ struct = xp.asarray(struct)
1873
+ expected = xp.asarray(expected)
1874
+ data = np.asarray([[0, 0, 0, 0, 0],
1875
+ [0, 0, 0, 0, 0],
1876
+ [0, 0, 0, 0, 0],
1877
+ [0, 0, 0, 1, 0],
1878
+ [0, 0, 0, 0, 0]], dtype=bool)
1879
+ data = xp.asarray(data)
1880
+ out = ndimage.binary_dilation(data, struct, iterations=3)
1881
+ assert_array_almost_equal(out, expected)
1882
+
1883
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1884
+ reason='output= arrays are numpy-specific')
1885
+ @xfail_xp_backends("cupy",
1886
+ reason="NotImplementedError: only brute_force iteration")
1887
+ def test_binary_dilation32(self, xp):
1888
+ struct = [[0, 1],
1889
+ [1, 1]]
1890
+ expected = [[0, 0, 0, 1, 0],
1891
+ [0, 0, 1, 1, 0],
1892
+ [0, 1, 1, 1, 0],
1893
+ [1, 1, 1, 1, 0],
1894
+ [0, 0, 0, 0, 0]]
1895
+ struct = xp.asarray(struct)
1896
+ expected = xp.asarray(expected)
1897
+ data = np.asarray([[0, 0, 0, 0, 0],
1898
+ [0, 0, 0, 0, 0],
1899
+ [0, 0, 0, 0, 0],
1900
+ [0, 0, 0, 1, 0],
1901
+ [0, 0, 0, 0, 0]], dtype=bool)
1902
+ data = xp.asarray(data)
1903
+ out = np.zeros(data.shape, dtype=bool)
1904
+ out = xp.asarray(out)
1905
+ ndimage.binary_dilation(data, struct, iterations=3, output=out)
1906
+ assert_array_almost_equal(out, expected)
1907
+
1908
+ @xfail_xp_backends("cupy",
1909
+ reason="NotImplementedError: only brute_force iteration")
1910
+ def test_binary_dilation33(self, xp):
1911
+ struct = [[0, 1, 0],
1912
+ [1, 1, 1],
1913
+ [0, 1, 0]]
1914
+ struct = xp.asarray(struct)
1915
+ expected = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
1916
+ [0, 0, 0, 0, 0, 0, 0, 0],
1917
+ [0, 0, 0, 0, 0, 0, 0, 0],
1918
+ [0, 0, 0, 0, 1, 1, 0, 0],
1919
+ [0, 0, 1, 1, 1, 0, 0, 0],
1920
+ [0, 1, 1, 0, 1, 1, 0, 0],
1921
+ [0, 0, 0, 0, 0, 0, 0, 0],
1922
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1923
+ expected = xp.asarray(expected)
1924
+ mask = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
1925
+ [0, 0, 0, 0, 0, 0, 0, 0],
1926
+ [0, 0, 0, 0, 0, 0, 1, 0],
1927
+ [0, 0, 0, 0, 1, 1, 0, 0],
1928
+ [0, 0, 1, 1, 1, 0, 0, 0],
1929
+ [0, 1, 1, 0, 1, 1, 0, 0],
1930
+ [0, 0, 0, 0, 0, 0, 0, 0],
1931
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1932
+ mask = xp.asarray(mask)
1933
+ data = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
1934
+ [0, 0, 0, 0, 0, 0, 0, 0],
1935
+ [0, 0, 0, 0, 0, 0, 0, 0],
1936
+ [0, 0, 0, 0, 0, 0, 0, 0],
1937
+ [0, 0, 0, 0, 0, 0, 0, 0],
1938
+ [0, 1, 0, 0, 0, 0, 0, 0],
1939
+ [0, 0, 0, 0, 0, 0, 0, 0],
1940
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1941
+ data = xp.asarray(data)
1942
+
1943
+ out = ndimage.binary_dilation(data, struct, iterations=-1,
1944
+ mask=mask, border_value=0)
1945
+ assert_array_almost_equal(out, expected)
1946
+
1947
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
1948
+ reason='inplace output= arrays are numpy-specific')
1949
+ @xfail_xp_backends("cupy",
1950
+ reason="NotImplementedError: only brute_force iteration")
1951
+ def test_binary_dilation34(self, xp):
1952
+ struct = [[0, 1, 0],
1953
+ [1, 1, 1],
1954
+ [0, 1, 0]]
1955
+ struct = xp.asarray(struct)
1956
+ expected = [[0, 1, 0, 0, 0, 0, 0, 0],
1957
+ [0, 1, 1, 0, 0, 0, 0, 0],
1958
+ [0, 0, 1, 0, 0, 0, 0, 0],
1959
+ [0, 0, 0, 0, 0, 0, 0, 0],
1960
+ [0, 0, 0, 0, 0, 0, 0, 0],
1961
+ [0, 0, 0, 0, 0, 0, 0, 0],
1962
+ [0, 0, 0, 0, 0, 0, 0, 0],
1963
+ [0, 0, 0, 0, 0, 0, 0, 0]]
1964
+ mask = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
1965
+ [0, 1, 1, 0, 0, 0, 0, 0],
1966
+ [0, 0, 1, 0, 0, 0, 0, 0],
1967
+ [0, 0, 0, 0, 0, 1, 0, 0],
1968
+ [0, 0, 0, 1, 1, 0, 0, 0],
1969
+ [0, 0, 1, 0, 0, 1, 0, 0],
1970
+ [0, 0, 0, 0, 0, 0, 0, 0],
1971
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
1972
+ mask = xp.asarray(mask)
1973
+ data = np.zeros(mask.shape, dtype=bool)
1974
+ data = xp.asarray(data)
1975
+ out = ndimage.binary_dilation(data, struct, iterations=-1,
1976
+ mask=mask, border_value=1)
1977
+ assert_array_almost_equal(out, expected)
1978
+
1979
+ @pytest.mark.parametrize('dtype', types)
1980
+ def test_binary_dilation35(self, dtype, xp):
1981
+ dtype = getattr(xp, dtype)
1982
+ tmp = [[1, 1, 0, 0, 0, 0, 1, 1],
1983
+ [1, 0, 0, 0, 1, 0, 1, 1],
1984
+ [0, 0, 1, 1, 1, 1, 1, 1],
1985
+ [0, 1, 1, 1, 1, 0, 1, 1],
1986
+ [1, 1, 1, 1, 1, 1, 1, 1],
1987
+ [0, 1, 0, 0, 1, 0, 1, 1],
1988
+ [1, 1, 1, 1, 1, 1, 1, 1],
1989
+ [1, 1, 1, 1, 1, 1, 1, 1]]
1990
+
1991
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
1992
+ [0, 1, 0, 0, 0, 0, 0, 0],
1993
+ [0, 0, 0, 0, 0, 0, 0, 0],
1994
+ [0, 0, 0, 0, 0, 1, 0, 0],
1995
+ [0, 0, 0, 1, 1, 0, 0, 0],
1996
+ [0, 0, 1, 0, 0, 1, 0, 0],
1997
+ [0, 0, 0, 0, 0, 0, 0, 0],
1998
+ [0, 0, 0, 0, 0, 0, 0, 0]])
1999
+ mask = [[0, 0, 0, 0, 0, 0, 0, 0],
2000
+ [0, 0, 0, 0, 0, 0, 0, 0],
2001
+ [0, 0, 0, 0, 0, 0, 0, 0],
2002
+ [0, 0, 1, 1, 1, 1, 0, 0],
2003
+ [0, 0, 1, 1, 1, 1, 0, 0],
2004
+ [0, 0, 1, 1, 1, 1, 0, 0],
2005
+ [0, 0, 0, 0, 0, 0, 0, 0],
2006
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2007
+ mask = np.asarray(mask, dtype=bool)
2008
+
2009
+ expected = np.logical_and(tmp, mask)
2010
+ tmp = np.logical_and(data, np.logical_not(mask))
2011
+ expected = np.logical_or(expected, tmp)
2012
+
2013
+ mask = xp.asarray(mask)
2014
+ expected = xp.asarray(expected)
2015
+
2016
+ data = xp.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2017
+ [0, 1, 0, 0, 0, 0, 0, 0],
2018
+ [0, 0, 0, 0, 0, 0, 0, 0],
2019
+ [0, 0, 0, 0, 0, 1, 0, 0],
2020
+ [0, 0, 0, 1, 1, 0, 0, 0],
2021
+ [0, 0, 1, 0, 0, 1, 0, 0],
2022
+ [0, 0, 0, 0, 0, 0, 0, 0],
2023
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2024
+ out = ndimage.binary_dilation(data, mask=mask,
2025
+ origin=(1, 1), border_value=1)
2026
+ assert_array_almost_equal(out, expected)
2027
+
2028
+ def test_binary_dilation36(self, xp):
2029
+ # gh-21009
2030
+ data = np.zeros([], dtype=bool)
2031
+ data = xp.asarray(data)
2032
+ out = ndimage.binary_dilation(data, iterations=-1)
2033
+ assert out == xp.asarray(False)
2034
+
2035
+ def test_binary_propagation01(self, xp):
2036
+ struct = [[0, 1, 0],
2037
+ [1, 1, 1],
2038
+ [0, 1, 0]]
2039
+ struct = xp.asarray(struct)
2040
+ expected = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2041
+ [0, 0, 0, 0, 0, 0, 0, 0],
2042
+ [0, 0, 0, 0, 0, 0, 0, 0],
2043
+ [0, 0, 0, 0, 1, 1, 0, 0],
2044
+ [0, 0, 1, 1, 1, 0, 0, 0],
2045
+ [0, 1, 1, 0, 1, 1, 0, 0],
2046
+ [0, 0, 0, 0, 0, 0, 0, 0],
2047
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2048
+ expected = xp.asarray(expected)
2049
+ mask = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2050
+ [0, 0, 0, 0, 0, 0, 0, 0],
2051
+ [0, 0, 0, 0, 0, 0, 1, 0],
2052
+ [0, 0, 0, 0, 1, 1, 0, 0],
2053
+ [0, 0, 1, 1, 1, 0, 0, 0],
2054
+ [0, 1, 1, 0, 1, 1, 0, 0],
2055
+ [0, 0, 0, 0, 0, 0, 0, 0],
2056
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2057
+ mask = xp.asarray(mask)
2058
+ data = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2059
+ [0, 0, 0, 0, 0, 0, 0, 0],
2060
+ [0, 0, 0, 0, 0, 0, 0, 0],
2061
+ [0, 0, 0, 0, 0, 0, 0, 0],
2062
+ [0, 0, 0, 0, 0, 0, 0, 0],
2063
+ [0, 1, 0, 0, 0, 0, 0, 0],
2064
+ [0, 0, 0, 0, 0, 0, 0, 0],
2065
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2066
+ data = xp.asarray(data)
2067
+ out = ndimage.binary_propagation(data, struct,
2068
+ mask=mask, border_value=0)
2069
+ assert_array_almost_equal(out, expected)
2070
+
2071
+ def test_binary_propagation02(self, xp):
2072
+ struct = [[0, 1, 0],
2073
+ [1, 1, 1],
2074
+ [0, 1, 0]]
2075
+ expected = [[0, 1, 0, 0, 0, 0, 0, 0],
2076
+ [0, 1, 1, 0, 0, 0, 0, 0],
2077
+ [0, 0, 1, 0, 0, 0, 0, 0],
2078
+ [0, 0, 0, 0, 0, 0, 0, 0],
2079
+ [0, 0, 0, 0, 0, 0, 0, 0],
2080
+ [0, 0, 0, 0, 0, 0, 0, 0],
2081
+ [0, 0, 0, 0, 0, 0, 0, 0],
2082
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2083
+ expected = xp.asarray(expected)
2084
+ struct = xp.asarray(struct)
2085
+ mask = np.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2086
+ [0, 1, 1, 0, 0, 0, 0, 0],
2087
+ [0, 0, 1, 0, 0, 0, 0, 0],
2088
+ [0, 0, 0, 0, 0, 1, 0, 0],
2089
+ [0, 0, 0, 1, 1, 0, 0, 0],
2090
+ [0, 0, 1, 0, 0, 1, 0, 0],
2091
+ [0, 0, 0, 0, 0, 0, 0, 0],
2092
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2093
+ mask = xp.asarray(mask)
2094
+ data = np.zeros(mask.shape, dtype=bool)
2095
+ data = xp.asarray(data)
2096
+ out = ndimage.binary_propagation(data, struct,
2097
+ mask=mask, border_value=1)
2098
+ assert_array_almost_equal(out, expected)
2099
+
2100
+ def test_binary_propagation03(self, xp):
2101
+ # gh-21009
2102
+ data = xp.asarray(np.zeros([], dtype=bool))
2103
+ expected = xp.asarray(np.zeros([], dtype=bool))
2104
+ out = ndimage.binary_propagation(data)
2105
+ assert out == expected
2106
+
2107
+ @pytest.mark.parametrize('dtype', types)
2108
+ def test_binary_opening01(self, dtype, xp):
2109
+ dtype = getattr(xp, dtype)
2110
+ expected = [[0, 1, 0, 0, 0, 0, 0, 0],
2111
+ [1, 1, 1, 0, 0, 0, 0, 0],
2112
+ [0, 1, 0, 0, 0, 1, 0, 0],
2113
+ [0, 0, 0, 0, 1, 1, 1, 0],
2114
+ [0, 0, 1, 0, 0, 1, 0, 0],
2115
+ [0, 1, 1, 1, 1, 1, 1, 0],
2116
+ [0, 0, 1, 0, 0, 1, 0, 0],
2117
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2118
+ expected = xp.asarray(expected)
2119
+ data = xp.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2120
+ [1, 1, 1, 0, 0, 0, 0, 0],
2121
+ [0, 1, 0, 0, 0, 1, 0, 0],
2122
+ [0, 0, 0, 1, 1, 1, 1, 0],
2123
+ [0, 0, 1, 1, 0, 1, 0, 0],
2124
+ [0, 1, 1, 1, 1, 1, 1, 0],
2125
+ [0, 0, 1, 0, 0, 1, 0, 0],
2126
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2127
+ out = ndimage.binary_opening(data)
2128
+ assert_array_almost_equal(out, expected)
2129
+
2130
+ @pytest.mark.parametrize('dtype', types)
2131
+ def test_binary_opening02(self, dtype, xp):
2132
+ dtype = getattr(xp, dtype)
2133
+ struct = ndimage.generate_binary_structure(2, 2)
2134
+ expected = [[1, 1, 1, 0, 0, 0, 0, 0],
2135
+ [1, 1, 1, 0, 0, 0, 0, 0],
2136
+ [1, 1, 1, 0, 0, 0, 0, 0],
2137
+ [0, 0, 0, 0, 0, 0, 0, 0],
2138
+ [0, 1, 1, 1, 0, 0, 0, 0],
2139
+ [0, 1, 1, 1, 0, 0, 0, 0],
2140
+ [0, 1, 1, 1, 0, 0, 0, 0],
2141
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2142
+ expected = xp.asarray(expected)
2143
+ struct = xp.asarray(struct)
2144
+ data = xp.asarray([[1, 1, 1, 0, 0, 0, 0, 0],
2145
+ [1, 1, 1, 0, 0, 0, 0, 0],
2146
+ [1, 1, 1, 1, 1, 1, 1, 0],
2147
+ [0, 0, 1, 1, 1, 1, 1, 0],
2148
+ [0, 1, 1, 1, 0, 1, 1, 0],
2149
+ [0, 1, 1, 1, 1, 1, 1, 0],
2150
+ [0, 1, 1, 1, 1, 1, 1, 0],
2151
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2152
+ out = ndimage.binary_opening(data, struct)
2153
+ assert_array_almost_equal(out, expected)
2154
+
2155
+ @pytest.mark.parametrize('dtype', types)
2156
+ def test_binary_closing01(self, dtype, xp):
2157
+ dtype = getattr(xp, dtype)
2158
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
2159
+ [0, 1, 1, 0, 0, 0, 0, 0],
2160
+ [0, 1, 1, 1, 0, 1, 0, 0],
2161
+ [0, 0, 1, 1, 1, 1, 1, 0],
2162
+ [0, 0, 1, 1, 1, 1, 0, 0],
2163
+ [0, 1, 1, 1, 1, 1, 1, 0],
2164
+ [0, 0, 1, 0, 0, 1, 0, 0],
2165
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2166
+ expected = xp.asarray(expected)
2167
+ data = xp.asarray([[0, 1, 0, 0, 0, 0, 0, 0],
2168
+ [1, 1, 1, 0, 0, 0, 0, 0],
2169
+ [0, 1, 0, 0, 0, 1, 0, 0],
2170
+ [0, 0, 0, 1, 1, 1, 1, 0],
2171
+ [0, 0, 1, 1, 0, 1, 0, 0],
2172
+ [0, 1, 1, 1, 1, 1, 1, 0],
2173
+ [0, 0, 1, 0, 0, 1, 0, 0],
2174
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2175
+ out = ndimage.binary_closing(data)
2176
+ assert_array_almost_equal(out, expected)
2177
+
2178
+ @pytest.mark.parametrize('dtype', types)
2179
+ def test_binary_closing02(self, dtype, xp):
2180
+ dtype = getattr(xp, dtype)
2181
+ struct = ndimage.generate_binary_structure(2, 2)
2182
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
2183
+ [0, 1, 1, 0, 0, 0, 0, 0],
2184
+ [0, 1, 1, 1, 1, 1, 1, 0],
2185
+ [0, 1, 1, 1, 1, 1, 1, 0],
2186
+ [0, 1, 1, 1, 1, 1, 1, 0],
2187
+ [0, 1, 1, 1, 1, 1, 1, 0],
2188
+ [0, 1, 1, 1, 1, 1, 1, 0],
2189
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2190
+ expected = xp.asarray(expected)
2191
+ struct = xp.asarray(struct)
2192
+ data = xp.asarray([[1, 1, 1, 0, 0, 0, 0, 0],
2193
+ [1, 1, 1, 0, 0, 0, 0, 0],
2194
+ [1, 1, 1, 1, 1, 1, 1, 0],
2195
+ [0, 0, 1, 1, 1, 1, 1, 0],
2196
+ [0, 1, 1, 1, 0, 1, 1, 0],
2197
+ [0, 1, 1, 1, 1, 1, 1, 0],
2198
+ [0, 1, 1, 1, 1, 1, 1, 0],
2199
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2200
+ out = ndimage.binary_closing(data, struct)
2201
+ assert_array_almost_equal(out, expected)
2202
+
2203
+ def test_binary_fill_holes01(self, xp):
2204
+ expected = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2205
+ [0, 0, 1, 1, 1, 1, 0, 0],
2206
+ [0, 0, 1, 1, 1, 1, 0, 0],
2207
+ [0, 0, 1, 1, 1, 1, 0, 0],
2208
+ [0, 0, 1, 1, 1, 1, 0, 0],
2209
+ [0, 0, 1, 1, 1, 1, 0, 0],
2210
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2211
+ expected = xp.asarray(expected)
2212
+
2213
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2214
+ [0, 0, 1, 1, 1, 1, 0, 0],
2215
+ [0, 0, 1, 0, 0, 1, 0, 0],
2216
+ [0, 0, 1, 0, 0, 1, 0, 0],
2217
+ [0, 0, 1, 0, 0, 1, 0, 0],
2218
+ [0, 0, 1, 1, 1, 1, 0, 0],
2219
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2220
+ data = xp.asarray(data)
2221
+
2222
+ out = ndimage.binary_fill_holes(data)
2223
+ assert_array_almost_equal(out, expected)
2224
+
2225
+ def test_binary_fill_holes02(self, xp):
2226
+ expected = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2227
+ [0, 0, 0, 1, 1, 0, 0, 0],
2228
+ [0, 0, 1, 1, 1, 1, 0, 0],
2229
+ [0, 0, 1, 1, 1, 1, 0, 0],
2230
+ [0, 0, 1, 1, 1, 1, 0, 0],
2231
+ [0, 0, 0, 1, 1, 0, 0, 0],
2232
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2233
+ expected = xp.asarray(expected)
2234
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2235
+ [0, 0, 0, 1, 1, 0, 0, 0],
2236
+ [0, 0, 1, 0, 0, 1, 0, 0],
2237
+ [0, 0, 1, 0, 0, 1, 0, 0],
2238
+ [0, 0, 1, 0, 0, 1, 0, 0],
2239
+ [0, 0, 0, 1, 1, 0, 0, 0],
2240
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2241
+ data = xp.asarray(data)
2242
+ out = ndimage.binary_fill_holes(data)
2243
+ assert_array_almost_equal(out, expected)
2244
+
2245
+ def test_binary_fill_holes03(self, xp):
2246
+ expected = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2247
+ [0, 0, 1, 0, 0, 0, 0, 0],
2248
+ [0, 1, 1, 1, 0, 1, 1, 1],
2249
+ [0, 1, 1, 1, 0, 1, 1, 1],
2250
+ [0, 1, 1, 1, 0, 1, 1, 1],
2251
+ [0, 0, 1, 0, 0, 1, 1, 1],
2252
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2253
+ expected = xp.asarray(expected)
2254
+ data = np.asarray([[0, 0, 0, 0, 0, 0, 0, 0],
2255
+ [0, 0, 1, 0, 0, 0, 0, 0],
2256
+ [0, 1, 0, 1, 0, 1, 1, 1],
2257
+ [0, 1, 0, 1, 0, 1, 0, 1],
2258
+ [0, 1, 0, 1, 0, 1, 0, 1],
2259
+ [0, 0, 1, 0, 0, 1, 1, 1],
2260
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=bool)
2261
+ data = xp.asarray(data)
2262
+ out = ndimage.binary_fill_holes(data)
2263
+ assert_array_almost_equal(out, expected)
2264
+
2265
+ @skip_xp_backends(cpu_only=True)
2266
+ @skip_xp_backends(
2267
+ "cupy", reason="these filters do not yet have axes support in CuPy")
2268
+ @skip_xp_backends(
2269
+ "jax.numpy", reason="these filters are not implemented in JAX.numpy")
2270
+ @pytest.mark.parametrize('border_value',[0, 1])
2271
+ @pytest.mark.parametrize('origin', [(0, 0), (-1, 0)])
2272
+ @pytest.mark.parametrize('expand_axis', [0, 1, 2])
2273
+ @pytest.mark.parametrize('func_name', ["binary_erosion",
2274
+ "binary_dilation",
2275
+ "binary_opening",
2276
+ "binary_closing",
2277
+ "binary_hit_or_miss",
2278
+ "binary_propagation",
2279
+ "binary_fill_holes"])
2280
+ def test_binary_axes(self, xp, func_name, expand_axis, origin, border_value):
2281
+ struct = np.asarray([[0, 1, 0],
2282
+ [1, 1, 1],
2283
+ [0, 1, 0]], bool)
2284
+ struct = xp.asarray(struct)
2285
+
2286
+ data = np.asarray([[0, 0, 0, 1, 0, 0, 0],
2287
+ [0, 0, 0, 1, 0, 0, 0],
2288
+ [0, 0, 1, 1, 0, 1, 0],
2289
+ [0, 1, 0, 1, 1, 0, 1],
2290
+ [0, 1, 1, 1, 1, 1, 0],
2291
+ [0, 0, 1, 1, 0, 0, 0],
2292
+ [0, 0, 0, 1, 0, 0, 0]], bool)
2293
+ data = xp.asarray(data)
2294
+ if func_name == "binary_hit_or_miss":
2295
+ kwargs = dict(origin1=origin, origin2=origin)
2296
+ else:
2297
+ kwargs = dict(origin=origin)
2298
+ border_supported = func_name not in ["binary_hit_or_miss",
2299
+ "binary_fill_holes"]
2300
+ if border_supported:
2301
+ kwargs['border_value'] = border_value
2302
+ elif border_value != 0:
2303
+ pytest.skip('border_value !=0 unsupported by this function')
2304
+ func = getattr(ndimage, func_name)
2305
+ expected = func(data, struct, **kwargs)
2306
+
2307
+ # replicate data and expected result along a new axis
2308
+ n_reps = 5
2309
+ expected = xp.stack([expected] * n_reps, axis=expand_axis)
2310
+ data = xp.stack([data] * n_reps, axis=expand_axis)
2311
+
2312
+ # filter all axes except expand_axis
2313
+ axes = [0, 1, 2]
2314
+ axes.remove(expand_axis)
2315
+ if is_numpy(xp) or is_cupy(xp):
2316
+ out = xp.asarray(np.zeros(data.shape, bool))
2317
+ func(data, struct, output=out, axes=axes, **kwargs)
2318
+ else:
2319
+ # inplace output= is unsupported by JAX
2320
+ out = func(data, struct, axes=axes, **kwargs)
2321
+ xp_assert_close(out, expected)
2322
+
2323
+ def test_grey_erosion01(self, xp):
2324
+ array = xp.asarray([[3, 2, 5, 1, 4],
2325
+ [7, 6, 9, 3, 5],
2326
+ [5, 8, 3, 7, 1]])
2327
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2328
+ output = ndimage.grey_erosion(array, footprint=footprint)
2329
+ assert_array_almost_equal(output,
2330
+ xp.asarray([[2, 2, 1, 1, 1],
2331
+ [2, 3, 1, 3, 1],
2332
+ [5, 5, 3, 3, 1]]))
2333
+
2334
+ @skip_xp_backends("jax.numpy", reason="output=array requires buffer view")
2335
+ @skip_xp_backends("dask.array", reason="output=array requires buffer view")
2336
+ @xfail_xp_backends("cupy", reason="https://github.com/cupy/cupy/issues/8398")
2337
+ def test_grey_erosion01_overlap(self, xp):
2338
+
2339
+ array = xp.asarray([[3, 2, 5, 1, 4],
2340
+ [7, 6, 9, 3, 5],
2341
+ [5, 8, 3, 7, 1]])
2342
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2343
+ ndimage.grey_erosion(array, footprint=footprint, output=array)
2344
+ assert_array_almost_equal(array,
2345
+ xp.asarray([[2, 2, 1, 1, 1],
2346
+ [2, 3, 1, 3, 1],
2347
+ [5, 5, 3, 3, 1]])
2348
+ )
2349
+
2350
+ def test_grey_erosion02(self, xp):
2351
+ array = xp.asarray([[3, 2, 5, 1, 4],
2352
+ [7, 6, 9, 3, 5],
2353
+ [5, 8, 3, 7, 1]])
2354
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2355
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2356
+ output = ndimage.grey_erosion(array, footprint=footprint,
2357
+ structure=structure)
2358
+ assert_array_almost_equal(output,
2359
+ xp.asarray([[2, 2, 1, 1, 1],
2360
+ [2, 3, 1, 3, 1],
2361
+ [5, 5, 3, 3, 1]])
2362
+ )
2363
+
2364
+ def test_grey_erosion03(self, xp):
2365
+ array = xp.asarray([[3, 2, 5, 1, 4],
2366
+ [7, 6, 9, 3, 5],
2367
+ [5, 8, 3, 7, 1]])
2368
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2369
+ structure = xp.asarray([[1, 1, 1], [1, 1, 1]])
2370
+ output = ndimage.grey_erosion(array, footprint=footprint,
2371
+ structure=structure)
2372
+ assert_array_almost_equal(output,
2373
+ xp.asarray([[1, 1, 0, 0, 0],
2374
+ [1, 2, 0, 2, 0],
2375
+ [4, 4, 2, 2, 0]])
2376
+ )
2377
+
2378
+ def test_grey_dilation01(self, xp):
2379
+ array = xp.asarray([[3, 2, 5, 1, 4],
2380
+ [7, 6, 9, 3, 5],
2381
+ [5, 8, 3, 7, 1]])
2382
+ footprint = xp.asarray([[0, 1, 1], [1, 0, 1]])
2383
+ output = ndimage.grey_dilation(array, footprint=footprint)
2384
+ assert_array_almost_equal(output,
2385
+ xp.asarray([[7, 7, 9, 9, 5],
2386
+ [7, 9, 8, 9, 7],
2387
+ [8, 8, 8, 7, 7]]),
2388
+ )
2389
+
2390
+ def test_grey_dilation02(self, xp):
2391
+ array = xp.asarray([[3, 2, 5, 1, 4],
2392
+ [7, 6, 9, 3, 5],
2393
+ [5, 8, 3, 7, 1]])
2394
+ footprint = xp.asarray([[0, 1, 1], [1, 0, 1]])
2395
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2396
+ output = ndimage.grey_dilation(array, footprint=footprint,
2397
+ structure=structure)
2398
+ assert_array_almost_equal(output,
2399
+ xp.asarray([[7, 7, 9, 9, 5],
2400
+ [7, 9, 8, 9, 7],
2401
+ [8, 8, 8, 7, 7]]),
2402
+ )
2403
+
2404
+ def test_grey_dilation03(self, xp):
2405
+ array = xp.asarray([[3, 2, 5, 1, 4],
2406
+ [7, 6, 9, 3, 5],
2407
+ [5, 8, 3, 7, 1]])
2408
+ footprint = xp.asarray([[0, 1, 1], [1, 0, 1]])
2409
+ structure = xp.asarray([[1, 1, 1], [1, 1, 1]])
2410
+ output = ndimage.grey_dilation(array, footprint=footprint,
2411
+ structure=structure)
2412
+ assert_array_almost_equal(output,
2413
+ xp.asarray([[8, 8, 10, 10, 6],
2414
+ [8, 10, 9, 10, 8],
2415
+ [9, 9, 9, 8, 8]]),
2416
+ )
2417
+
2418
+ def test_grey_opening01(self, xp):
2419
+ array = xp.asarray([[3, 2, 5, 1, 4],
2420
+ [7, 6, 9, 3, 5],
2421
+ [5, 8, 3, 7, 1]])
2422
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2423
+ tmp = ndimage.grey_erosion(array, footprint=footprint)
2424
+ expected = ndimage.grey_dilation(tmp, footprint=footprint)
2425
+ output = ndimage.grey_opening(array, footprint=footprint)
2426
+ assert_array_almost_equal(output, expected)
2427
+
2428
+ def test_grey_opening02(self, xp):
2429
+ array = xp.asarray([[3, 2, 5, 1, 4],
2430
+ [7, 6, 9, 3, 5],
2431
+ [5, 8, 3, 7, 1]])
2432
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2433
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2434
+ tmp = ndimage.grey_erosion(array, footprint=footprint,
2435
+ structure=structure)
2436
+ expected = ndimage.grey_dilation(tmp, footprint=footprint,
2437
+ structure=structure)
2438
+ output = ndimage.grey_opening(array, footprint=footprint,
2439
+ structure=structure)
2440
+ assert_array_almost_equal(output, expected)
2441
+
2442
+ def test_grey_closing01(self, xp):
2443
+ array = xp.asarray([[3, 2, 5, 1, 4],
2444
+ [7, 6, 9, 3, 5],
2445
+ [5, 8, 3, 7, 1]])
2446
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2447
+ tmp = ndimage.grey_dilation(array, footprint=footprint)
2448
+ expected = ndimage.grey_erosion(tmp, footprint=footprint)
2449
+ output = ndimage.grey_closing(array, footprint=footprint)
2450
+ assert_array_almost_equal(output, expected)
2451
+
2452
+ def test_grey_closing02(self, xp):
2453
+ array = xp.asarray([[3, 2, 5, 1, 4],
2454
+ [7, 6, 9, 3, 5],
2455
+ [5, 8, 3, 7, 1]])
2456
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2457
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2458
+ tmp = ndimage.grey_dilation(array, footprint=footprint,
2459
+ structure=structure)
2460
+ expected = ndimage.grey_erosion(tmp, footprint=footprint,
2461
+ structure=structure)
2462
+ output = ndimage.grey_closing(array, footprint=footprint,
2463
+ structure=structure)
2464
+ assert_array_almost_equal(output, expected)
2465
+
2466
+ @skip_xp_backends(np_only=True, reason='output= arrays are numpy-specific')
2467
+ def test_morphological_gradient01(self, xp):
2468
+ array = xp.asarray([[3, 2, 5, 1, 4],
2469
+ [7, 6, 9, 3, 5],
2470
+ [5, 8, 3, 7, 1]])
2471
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2472
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2473
+ tmp1 = ndimage.grey_dilation(array, footprint=footprint,
2474
+ structure=structure)
2475
+ tmp2 = ndimage.grey_erosion(array, footprint=footprint,
2476
+ structure=structure)
2477
+ expected = tmp1 - tmp2
2478
+ output = xp.zeros(array.shape, dtype=array.dtype)
2479
+ ndimage.morphological_gradient(array, footprint=footprint,
2480
+ structure=structure, output=output)
2481
+ assert_array_almost_equal(output, expected)
2482
+
2483
+ def test_morphological_gradient02(self, xp):
2484
+ array = xp.asarray([[3, 2, 5, 1, 4],
2485
+ [7, 6, 9, 3, 5],
2486
+ [5, 8, 3, 7, 1]])
2487
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2488
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2489
+ tmp1 = ndimage.grey_dilation(array, footprint=footprint,
2490
+ structure=structure)
2491
+ tmp2 = ndimage.grey_erosion(array, footprint=footprint,
2492
+ structure=structure)
2493
+ expected = tmp1 - tmp2
2494
+ output = ndimage.morphological_gradient(array, footprint=footprint,
2495
+ structure=structure)
2496
+ assert_array_almost_equal(output, expected)
2497
+
2498
+ @skip_xp_backends(np_only=True, reason='output= arrays are numpy-specific')
2499
+ def test_morphological_laplace01(self, xp):
2500
+ array = xp.asarray([[3, 2, 5, 1, 4],
2501
+ [7, 6, 9, 3, 5],
2502
+ [5, 8, 3, 7, 1]])
2503
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2504
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2505
+ tmp1 = ndimage.grey_dilation(array, footprint=footprint,
2506
+ structure=structure)
2507
+ tmp2 = ndimage.grey_erosion(array, footprint=footprint,
2508
+ structure=structure)
2509
+ expected = tmp1 + tmp2 - 2 * array
2510
+ output = xp.zeros(array.shape, dtype=array.dtype)
2511
+ ndimage.morphological_laplace(array, footprint=footprint,
2512
+ structure=structure, output=output)
2513
+ assert_array_almost_equal(output, expected)
2514
+
2515
+ def test_morphological_laplace02(self, xp):
2516
+ array = xp.asarray([[3, 2, 5, 1, 4],
2517
+ [7, 6, 9, 3, 5],
2518
+ [5, 8, 3, 7, 1]])
2519
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2520
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2521
+ tmp1 = ndimage.grey_dilation(array, footprint=footprint,
2522
+ structure=structure)
2523
+ tmp2 = ndimage.grey_erosion(array, footprint=footprint,
2524
+ structure=structure)
2525
+ expected = tmp1 + tmp2 - 2 * array
2526
+ output = ndimage.morphological_laplace(array, footprint=footprint,
2527
+ structure=structure)
2528
+ assert_array_almost_equal(output, expected)
2529
+
2530
+ @skip_xp_backends("jax.numpy", reason="output=array requires buffer view")
2531
+ @skip_xp_backends("dask.array", reason="output=array requires buffer view")
2532
+ def test_white_tophat01(self, xp):
2533
+ array = xp.asarray([[3, 2, 5, 1, 4],
2534
+ [7, 6, 9, 3, 5],
2535
+ [5, 8, 3, 7, 1]])
2536
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2537
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2538
+ tmp = ndimage.grey_opening(array, footprint=footprint,
2539
+ structure=structure)
2540
+ expected = array - tmp
2541
+ output = xp.zeros(array.shape, dtype=array.dtype)
2542
+ ndimage.white_tophat(array, footprint=footprint,
2543
+ structure=structure, output=output)
2544
+ assert_array_almost_equal(output, expected)
2545
+
2546
+ def test_white_tophat02(self, xp):
2547
+ array = xp.asarray([[3, 2, 5, 1, 4],
2548
+ [7, 6, 9, 3, 5],
2549
+ [5, 8, 3, 7, 1]])
2550
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2551
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2552
+ tmp = ndimage.grey_opening(array, footprint=footprint,
2553
+ structure=structure)
2554
+ expected = array - tmp
2555
+ output = ndimage.white_tophat(array, footprint=footprint,
2556
+ structure=structure)
2557
+ assert_array_almost_equal(output, expected)
2558
+
2559
+ @xfail_xp_backends('cupy', reason="cupy#8399")
2560
+ def test_white_tophat03(self, xp):
2561
+
2562
+ array = np.asarray([[1, 0, 0, 0, 0, 0, 0],
2563
+ [0, 1, 1, 1, 1, 1, 0],
2564
+ [0, 1, 1, 1, 1, 1, 0],
2565
+ [0, 1, 1, 1, 1, 1, 0],
2566
+ [0, 1, 1, 1, 0, 1, 0],
2567
+ [0, 1, 1, 1, 1, 1, 0],
2568
+ [0, 0, 0, 0, 0, 0, 1]], dtype=bool)
2569
+ array = xp.asarray(array)
2570
+ structure = np.ones((3, 3), dtype=bool)
2571
+ structure = xp.asarray(structure)
2572
+ expected = np.asarray([[0, 1, 1, 0, 0, 0, 0],
2573
+ [1, 0, 0, 1, 1, 1, 0],
2574
+ [1, 0, 0, 1, 1, 1, 0],
2575
+ [0, 1, 1, 0, 0, 0, 1],
2576
+ [0, 1, 1, 0, 1, 0, 1],
2577
+ [0, 1, 1, 0, 0, 0, 1],
2578
+ [0, 0, 0, 1, 1, 1, 1]], dtype=bool)
2579
+ expected = xp.asarray(expected)
2580
+
2581
+ output = ndimage.white_tophat(array, structure=structure)
2582
+ xp_assert_equal(output, expected)
2583
+
2584
+ @skip_xp_backends("jax.numpy", reason="output=array requires buffer view")
2585
+ @skip_xp_backends("dask.array", reason="output=array requires buffer view")
2586
+ def test_white_tophat04(self, xp):
2587
+ array = np.eye(5, dtype=bool)
2588
+ structure = np.ones((3, 3), dtype=bool)
2589
+
2590
+ array = xp.asarray(array)
2591
+ structure = xp.asarray(structure)
2592
+
2593
+ # Check that type mismatch is properly handled
2594
+ output = xp.empty_like(array, dtype=xp.float64)
2595
+ ndimage.white_tophat(array, structure=structure, output=output)
2596
+
2597
+ @skip_xp_backends("jax.numpy", reason="output=array requires buffer view")
2598
+ @skip_xp_backends("dask.array", reason="output=array requires buffer view")
2599
+ def test_black_tophat01(self, xp):
2600
+ array = xp.asarray([[3, 2, 5, 1, 4],
2601
+ [7, 6, 9, 3, 5],
2602
+ [5, 8, 3, 7, 1]])
2603
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2604
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2605
+ tmp = ndimage.grey_closing(array, footprint=footprint,
2606
+ structure=structure)
2607
+ expected = tmp - array
2608
+ output = xp.zeros(array.shape, dtype=array.dtype)
2609
+ ndimage.black_tophat(array, footprint=footprint,
2610
+ structure=structure, output=output)
2611
+ assert_array_almost_equal(output, expected)
2612
+
2613
+ def test_black_tophat02(self, xp):
2614
+ array = xp.asarray([[3, 2, 5, 1, 4],
2615
+ [7, 6, 9, 3, 5],
2616
+ [5, 8, 3, 7, 1]])
2617
+ footprint = xp.asarray([[1, 0, 1], [1, 1, 0]])
2618
+ structure = xp.asarray([[0, 0, 0], [0, 0, 0]])
2619
+ tmp = ndimage.grey_closing(array, footprint=footprint,
2620
+ structure=structure)
2621
+ expected = tmp - array
2622
+ output = ndimage.black_tophat(array, footprint=footprint,
2623
+ structure=structure)
2624
+ assert_array_almost_equal(output, expected)
2625
+
2626
+ @xfail_xp_backends('cupy', reason="cupy/cupy#8399")
2627
+ def test_black_tophat03(self, xp):
2628
+
2629
+ array = np.asarray([[1, 0, 0, 0, 0, 0, 0],
2630
+ [0, 1, 1, 1, 1, 1, 0],
2631
+ [0, 1, 1, 1, 1, 1, 0],
2632
+ [0, 1, 1, 1, 1, 1, 0],
2633
+ [0, 1, 1, 1, 0, 1, 0],
2634
+ [0, 1, 1, 1, 1, 1, 0],
2635
+ [0, 0, 0, 0, 0, 0, 1]], dtype=bool)
2636
+ array = xp.asarray(array)
2637
+ structure = np.ones((3, 3), dtype=bool)
2638
+ structure = xp.asarray(structure)
2639
+ expected = np.asarray([[0, 1, 1, 1, 1, 1, 1],
2640
+ [1, 0, 0, 0, 0, 0, 1],
2641
+ [1, 0, 0, 0, 0, 0, 1],
2642
+ [1, 0, 0, 0, 0, 0, 1],
2643
+ [1, 0, 0, 0, 1, 0, 1],
2644
+ [1, 0, 0, 0, 0, 0, 1],
2645
+ [1, 1, 1, 1, 1, 1, 0]], dtype=bool)
2646
+ expected = xp.asarray(expected)
2647
+
2648
+ output = ndimage.black_tophat(array, structure=structure)
2649
+ xp_assert_equal(output, expected)
2650
+
2651
+ @skip_xp_backends("jax.numpy", reason="output=array requires buffer view")
2652
+ @skip_xp_backends("dask.array", reason="output=array requires buffer view")
2653
+ def test_black_tophat04(self, xp):
2654
+ array = xp.asarray(np.eye(5, dtype=bool))
2655
+ structure = xp.asarray(np.ones((3, 3), dtype=bool))
2656
+
2657
+ # Check that type mismatch is properly handled
2658
+ output = xp.empty_like(array, dtype=xp.float64)
2659
+ ndimage.black_tophat(array, structure=structure, output=output)
2660
+
2661
+ @skip_xp_backends(cpu_only=True)
2662
+ @skip_xp_backends(
2663
+ "cupy", reason="these filters do not yet have axes support in CuPy")
2664
+ @skip_xp_backends(
2665
+ "jax.numpy", reason="these filters are not implemented in JAX.numpy")
2666
+ @pytest.mark.parametrize('origin', [(0, 0), (-1, 0)])
2667
+ @pytest.mark.parametrize('expand_axis', [0, 1, 2])
2668
+ @pytest.mark.parametrize('mode', ['reflect', 'constant', 'nearest',
2669
+ 'mirror', 'wrap'])
2670
+ @pytest.mark.parametrize('footprint_mode', ['size', 'footprint',
2671
+ 'structure'])
2672
+ @pytest.mark.parametrize('func_name', ["grey_erosion",
2673
+ "grey_dilation",
2674
+ "grey_opening",
2675
+ "grey_closing",
2676
+ "morphological_laplace",
2677
+ "morphological_gradient",
2678
+ "white_tophat",
2679
+ "black_tophat"])
2680
+ def test_grey_axes(self, xp, func_name, expand_axis, origin, footprint_mode,
2681
+ mode):
2682
+
2683
+ data = xp.asarray([[0, 0, 0, 1, 0, 0, 0],
2684
+ [0, 0, 0, 4, 0, 0, 0],
2685
+ [0, 0, 2, 1, 0, 2, 0],
2686
+ [0, 3, 0, 6, 5, 0, 1],
2687
+ [0, 4, 5, 3, 3, 4, 0],
2688
+ [0, 0, 9, 3, 0, 0, 0],
2689
+ [0, 0, 0, 2, 0, 0, 0]])
2690
+ kwargs = dict(origin=origin, mode=mode)
2691
+ if footprint_mode == 'size':
2692
+ kwargs['size'] = (2, 3)
2693
+ else:
2694
+ kwargs['footprint'] = xp.asarray([[1, 0, 1], [1, 1, 0]])
2695
+ if footprint_mode == 'structure':
2696
+ kwargs['structure'] = xp.ones_like(kwargs['footprint'])
2697
+ func = getattr(ndimage, func_name)
2698
+ expected = func(data, **kwargs)
2699
+
2700
+ # replicate data and expected result along a new axis
2701
+ n_reps = 5
2702
+ expected = xp.stack([expected] * n_reps, axis=expand_axis)
2703
+ data = xp.stack([data] * n_reps, axis=expand_axis)
2704
+
2705
+ # filter all axes except expand_axis
2706
+ axes = [0, 1, 2]
2707
+ axes.remove(expand_axis)
2708
+
2709
+ if is_numpy(xp) or is_cupy(xp):
2710
+ out = xp.zeros(expected.shape, dtype=expected.dtype)
2711
+ func(data, output=out, axes=axes, **kwargs)
2712
+ else:
2713
+ # inplace output= is unsupported by JAX
2714
+ out = func(data, axes=axes, **kwargs)
2715
+ xp_assert_close(out, expected)
2716
+
2717
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
2718
+ reason="inplace output= is numpy-specific")
2719
+ @pytest.mark.parametrize('dtype', types)
2720
+ def test_hit_or_miss01(self, dtype, xp):
2721
+ dtype = getattr(xp, dtype)
2722
+ struct = [[0, 1, 0],
2723
+ [1, 1, 1],
2724
+ [0, 1, 0]]
2725
+ struct = xp.asarray(struct)
2726
+ expected = [[0, 0, 0, 0, 0],
2727
+ [0, 1, 0, 0, 0],
2728
+ [0, 0, 0, 0, 0],
2729
+ [0, 0, 0, 0, 0],
2730
+ [0, 0, 0, 0, 0],
2731
+ [0, 0, 0, 0, 0],
2732
+ [0, 0, 0, 0, 0],
2733
+ [0, 0, 0, 0, 0]]
2734
+ expected = xp.asarray(expected)
2735
+ data = xp.asarray([[0, 1, 0, 0, 0],
2736
+ [1, 1, 1, 0, 0],
2737
+ [0, 1, 0, 1, 1],
2738
+ [0, 0, 1, 1, 1],
2739
+ [0, 1, 1, 1, 0],
2740
+ [0, 1, 1, 1, 1],
2741
+ [0, 1, 1, 1, 1],
2742
+ [0, 0, 0, 0, 0]], dtype=dtype)
2743
+ out = xp.asarray(np.zeros(data.shape, dtype=bool))
2744
+ ndimage.binary_hit_or_miss(data, struct, output=out)
2745
+ assert_array_almost_equal(out, expected)
2746
+
2747
+ @pytest.mark.parametrize('dtype', types)
2748
+ def test_hit_or_miss02(self, dtype, xp):
2749
+ dtype = getattr(xp, dtype)
2750
+ struct = [[0, 1, 0],
2751
+ [1, 1, 1],
2752
+ [0, 1, 0]]
2753
+ expected = [[0, 0, 0, 0, 0, 0, 0, 0],
2754
+ [0, 1, 0, 0, 0, 0, 0, 0],
2755
+ [0, 0, 0, 0, 0, 0, 0, 0],
2756
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2757
+ struct = xp.asarray(struct)
2758
+ expected = xp.asarray(expected)
2759
+ data = xp.asarray([[0, 1, 0, 0, 1, 1, 1, 0],
2760
+ [1, 1, 1, 0, 0, 1, 0, 0],
2761
+ [0, 1, 0, 1, 1, 1, 1, 0],
2762
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2763
+ out = ndimage.binary_hit_or_miss(data, struct)
2764
+ assert_array_almost_equal(out, expected)
2765
+
2766
+ @pytest.mark.parametrize('dtype', types)
2767
+ def test_hit_or_miss03(self, dtype, xp):
2768
+ dtype = getattr(xp, dtype)
2769
+ struct1 = [[0, 0, 0],
2770
+ [1, 1, 1],
2771
+ [0, 0, 0]]
2772
+ struct2 = [[1, 1, 1],
2773
+ [0, 0, 0],
2774
+ [1, 1, 1]]
2775
+ expected = [[0, 0, 0, 0, 0, 1, 0, 0],
2776
+ [0, 0, 0, 0, 0, 0, 0, 0],
2777
+ [0, 0, 0, 0, 0, 0, 0, 0],
2778
+ [0, 0, 0, 0, 0, 0, 0, 0],
2779
+ [0, 0, 0, 0, 0, 0, 0, 0],
2780
+ [0, 0, 0, 0, 0, 0, 0, 0],
2781
+ [0, 0, 1, 0, 0, 0, 0, 0],
2782
+ [0, 0, 0, 0, 0, 0, 0, 0]]
2783
+ struct1 = xp.asarray(struct1)
2784
+ struct2 = xp.asarray(struct2)
2785
+ expected = xp.asarray(expected)
2786
+ data = xp.asarray([[0, 1, 0, 0, 1, 1, 1, 0],
2787
+ [1, 1, 1, 0, 0, 0, 0, 0],
2788
+ [0, 1, 0, 1, 1, 1, 1, 0],
2789
+ [0, 0, 1, 1, 1, 1, 1, 0],
2790
+ [0, 1, 1, 1, 0, 1, 1, 0],
2791
+ [0, 0, 0, 0, 1, 1, 1, 0],
2792
+ [0, 1, 1, 1, 1, 1, 1, 0],
2793
+ [0, 0, 0, 0, 0, 0, 0, 0]], dtype=dtype)
2794
+ out = ndimage.binary_hit_or_miss(data, struct1, struct2)
2795
+ assert_array_almost_equal(out, expected)
2796
+
2797
+
2798
+ class TestDilateFix:
2799
+
2800
+ # pytest's setup_method seems to clash with the autouse `xp` fixture
2801
+ # so call _setup manually from all methods
2802
+ def _setup(self, xp):
2803
+ # dilation related setup
2804
+ self.array = xp.asarray([[0, 0, 0, 0, 0],
2805
+ [0, 0, 0, 0, 0],
2806
+ [0, 0, 0, 1, 0],
2807
+ [0, 0, 1, 1, 0],
2808
+ [0, 0, 0, 0, 0]], dtype=xp.uint8)
2809
+
2810
+ self.sq3x3 = xp.ones((3, 3))
2811
+ dilated3x3 = ndimage.binary_dilation(self.array, structure=self.sq3x3)
2812
+
2813
+ if is_numpy(xp):
2814
+ self.dilated3x3 = dilated3x3.view(xp.uint8)
2815
+ else:
2816
+ self.dilated3x3 = xp.astype(dilated3x3, xp.uint8)
2817
+
2818
+ def test_dilation_square_structure(self, xp):
2819
+ self._setup(xp)
2820
+ result = ndimage.grey_dilation(self.array, structure=self.sq3x3)
2821
+ # +1 accounts for difference between grey and binary dilation
2822
+ assert_array_almost_equal(result, self.dilated3x3 + 1)
2823
+
2824
+ def test_dilation_scalar_size(self, xp):
2825
+ self._setup(xp)
2826
+ result = ndimage.grey_dilation(self.array, size=3)
2827
+ assert_array_almost_equal(result, self.dilated3x3)
2828
+
2829
+
2830
+ class TestBinaryOpeningClosing:
2831
+
2832
+ def _setup(self, xp):
2833
+ a = np.zeros((5, 5), dtype=bool)
2834
+ a[1:4, 1:4] = True
2835
+ a[4, 4] = True
2836
+ self.array = xp.asarray(a)
2837
+ self.sq3x3 = xp.ones((3, 3))
2838
+ self.opened_old = ndimage.binary_opening(self.array, self.sq3x3,
2839
+ 1, None, 0)
2840
+ self.closed_old = ndimage.binary_closing(self.array, self.sq3x3,
2841
+ 1, None, 0)
2842
+
2843
+ def test_opening_new_arguments(self, xp):
2844
+ self._setup(xp)
2845
+ opened_new = ndimage.binary_opening(self.array, self.sq3x3, 1, None,
2846
+ 0, None, 0, False)
2847
+ xp_assert_equal(opened_new, self.opened_old)
2848
+
2849
+ def test_closing_new_arguments(self, xp):
2850
+ self._setup(xp)
2851
+ closed_new = ndimage.binary_closing(self.array, self.sq3x3, 1, None,
2852
+ 0, None, 0, False)
2853
+ xp_assert_equal(closed_new, self.closed_old)
2854
+
2855
+
2856
+ def test_binary_erosion_noninteger_iterations(xp):
2857
+ # regression test for gh-9905, gh-9909: ValueError for
2858
+ # non integer iterations
2859
+ data = xp.ones([1])
2860
+ assert_raises(TypeError, ndimage.binary_erosion, data, iterations=0.5)
2861
+ assert_raises(TypeError, ndimage.binary_erosion, data, iterations=1.5)
2862
+
2863
+
2864
+ def test_binary_dilation_noninteger_iterations(xp):
2865
+ # regression test for gh-9905, gh-9909: ValueError for
2866
+ # non integer iterations
2867
+ data = xp.ones([1])
2868
+ assert_raises(TypeError, ndimage.binary_dilation, data, iterations=0.5)
2869
+ assert_raises(TypeError, ndimage.binary_dilation, data, iterations=1.5)
2870
+
2871
+
2872
+ def test_binary_opening_noninteger_iterations(xp):
2873
+ # regression test for gh-9905, gh-9909: ValueError for
2874
+ # non integer iterations
2875
+ data = xp.ones([1])
2876
+ assert_raises(TypeError, ndimage.binary_opening, data, iterations=0.5)
2877
+ assert_raises(TypeError, ndimage.binary_opening, data, iterations=1.5)
2878
+
2879
+
2880
+ def test_binary_closing_noninteger_iterations(xp):
2881
+ # regression test for gh-9905, gh-9909: ValueError for
2882
+ # non integer iterations
2883
+ data = xp.ones([1])
2884
+ assert_raises(TypeError, ndimage.binary_closing, data, iterations=0.5)
2885
+ assert_raises(TypeError, ndimage.binary_closing, data, iterations=1.5)
2886
+
2887
+
2888
+ @xfail_xp_backends(
2889
+ "cupy", reason="CuPy: NotImplementedError: only brute_force iteration"
2890
+ )
2891
+ def test_binary_closing_noninteger_brute_force_passes_when_true(xp):
2892
+ # regression test for gh-9905, gh-9909: ValueError for non integer iterations
2893
+ data = xp.ones([1])
2894
+ xp_assert_equal(ndimage.binary_erosion(data, iterations=2, brute_force=1.5),
2895
+ ndimage.binary_erosion(data, iterations=2, brute_force=bool(1.5))
2896
+ )
2897
+ xp_assert_equal(ndimage.binary_erosion(data, iterations=2, brute_force=0.0),
2898
+ ndimage.binary_erosion(data, iterations=2, brute_force=bool(0.0))
2899
+ )
2900
+
2901
+
2902
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
2903
+ reason="inplace output= is numpy-specific")
2904
+ @xfail_xp_backends("cupy", reason="NotImplementedError: only brute_force iteration")
2905
+ @pytest.mark.parametrize(
2906
+ 'function',
2907
+ ['binary_erosion', 'binary_dilation', 'binary_opening', 'binary_closing'],
2908
+ )
2909
+ @pytest.mark.parametrize('iterations', [1, 5])
2910
+ @pytest.mark.parametrize('brute_force', [False, True])
2911
+ def test_binary_input_as_output(function, iterations, brute_force, xp):
2912
+ rstate = np.random.RandomState(123)
2913
+ data = rstate.randint(low=0, high=2, size=100).astype(bool)
2914
+ data = xp.asarray(data)
2915
+ ndi_func = getattr(ndimage, function)
2916
+
2917
+ # input data is not modified
2918
+ data_orig = data.copy()
2919
+ expected = ndi_func(data, brute_force=brute_force, iterations=iterations)
2920
+ xp_assert_equal(data, data_orig)
2921
+
2922
+ # data should now contain the expected result
2923
+ ndi_func(data, brute_force=brute_force, iterations=iterations, output=data)
2924
+ xp_assert_equal(data, expected)
2925
+
2926
+
2927
+ @skip_xp_backends(np_only=True, exceptions=["cupy"],
2928
+ reason="inplace output= is numpy-specific")
2929
+ def test_binary_hit_or_miss_input_as_output(xp):
2930
+ rstate = np.random.RandomState(123)
2931
+ data = rstate.randint(low=0, high=2, size=100).astype(bool)
2932
+ data = xp.asarray(data)
2933
+
2934
+ # input data is not modified
2935
+ data_orig = data.copy()
2936
+ expected = ndimage.binary_hit_or_miss(data)
2937
+ xp_assert_equal(data, data_orig)
2938
+
2939
+ # data should now contain the expected result
2940
+ ndimage.binary_hit_or_miss(data, output=data)
2941
+ xp_assert_equal(data, expected)
2942
+
2943
+
2944
+ @xfail_xp_backends("cupy",
2945
+ reason="CuPy does not have distance_transform_cdt")
2946
+ def test_distance_transform_cdt_invalid_metric(xp):
2947
+ msg = 'invalid metric provided'
2948
+ with pytest.raises(ValueError, match=msg):
2949
+ ndimage.distance_transform_cdt(xp.ones((5, 5)),
2950
+ metric="garbage")