a dC%@sdZddlZddlZddlZddlZddlZddlmZddlZ ddlm Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlZddlmZmZddlmZdd lmZmZm Z m!Z!d d l"m#Z#ddl$Z$e$j%j&Z&e$j%j'Z'e$j%j(Z(e$j)Z)e$j*Z*e+d Z,e-d ddkZ.ej/0ddZ1e12dkr2dZ1n>e12dkrFdZ1n*ze3e4e1Z1Wne5yndZ1Yn0d+ddZ6d,ddZ7ddZ8ddZ9d-dd Z:d.d!d"Z;d#d$Zd/d)d*Z?dS)0z Testing utilities. N)NamedTemporaryFile)testing) TestCaseassert_ assert_warnsassert_no_warnings assert_equalassert_almost_equalassert_array_equalassert_allcloseassert_array_almost_equalassert_array_almost_equal_nulpassert_array_less)dataio)_fetch) img_as_uint img_as_float img_as_int img_as_ubyte)expected_warningsz%(\s*>>>.*?)(\s*)#\s*skip\s+if\s+(.*)$P Z#SKIMAGE_TEST_STRICT_WARNINGS_GLOBAL0trueTfalseFcCs6|d|}|dur"|d|7}||ks2J|dS)Nz is not lower than : abmsgmessager r `/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/_shared/testing.py assert_less:s r'cCs6|d|}|dur"|d|7}||ks2J|dS)Nz is not greater than rr r!r r r&assert_greaterAs r(c Cs|jd}g}|D]}t|}|dur6||q|\}}}zt||jr^||d}Wn,tyt||j jr||d}Yn0||qd ||_|S)aD Decorator replaces custom skip test markup in doctests Say a function has a docstring:: >>> something, HAVE_AMODULE, HAVE_BMODULE = 0, False, False >>> something # skip if not HAVE_AMODULE 0 >>> something # skip if HAVE_BMODULE 0 This decorator will evaluate the expression after ``skip if``. If this evaluates to True, then the comment is replaced by ``# doctest: +SKIP``. If False, then the comment is just removed. The expression is evaluated in the ``globals`` scope of `func`. For example, if the module global ``HAVE_AMODULE`` is False, and module global ``HAVE_BMODULE`` is False, the returned function will have docstring:: >>> something # doctest: +SKIP >>> something + else # doctest: +SKIP >>> something # doctest: +SKIP  Nz# doctest: +SKIP) __doc__splitSKIP_REmatchappendgroupseval __globals__AttributeError__init__join)funclines new_linesliner-codespaceexprr r r&doctest_skip_parserHs"       r<cCsd|vrd|}t|dd}|j}Wdn1s80Ytj|||dtj||d}zt|WntyYn0|S)z/Save and read an image using a specified plugin.F)suffixdeleteN)plugin)rnamerZimsaveZimreadosremove Exception)imager@r> temp_filefnamenewr r r& roundtripws$ rIpngc Cstt}t|||}t|||dk}t|||}t||tt|}t|||}t||t |}| dvr|d8}t|||} t| |nt|||} t| t|t |} t| ||} t| |dS)zCheck roundtrip behavior for color images. All major input types should be handled as ubytes and read back correctly. ZtifZtiffdN) rrZchelsearIrr astypeboolrrlowerr r@fmtimgZr1Zimg2Zr2Zimg3Zr3Zimg4Zr4Zimg5Zr5r r r& color_checks&          rTc Cstt}t|||}t|||dk}t|||}t||tt|}t|||}|j j dkrxt||nt|t |t |}| dvr|d8}t|||} t| |nt|||} t| t |t |} t| ||} t| | dS)zsCheck the roundtrip behavior for images that support most types. All major input types should be handled. rKfrLrMN)rrZmoonrIrr rNrOrZdtypekindrrrPrQr r r& mono_checks*          rWcCstdtrtjdtdtjddtdtjddtdtjddt dd tjdd t d d tjdd t dtjdd t dtjddt dtjddt ddS)zDefault package level setup routine for skimage tests. Import packages known to raise warnings, and then force warnings to raise errors. Also set the random seed to zero. defaultrerrorz unclosed file)r%categoryz/Conversion of the second argument of issubdtypez.the matrix subclass is not the recommended waynumpy)r%rZmodulezYour installed pillow versionz skimage.ioz.The optional dask dependency is not installed.znumpy.ufunc size changedz# The scipy.sparse array containersz-`np.int0` is a deprecated alias for `np.intp`N) warnings simplefilter_error_on_warningsnprandomseedfilterwarningsResourceWarning FutureWarningPendingDeprecationWarning UserWarningRuntimeWarningDeprecationWarningr r r r& setup_testsV    rjcCstrttddS)zhDefault package level teardown routine for skimage tests. Restore warnings to default behavior rXN)r_r] resetwarningsr^r r r r& teardown_testsrlc Cs:z t|WSttfy4tjd|ddYn0dS)z:Attempt to fetch data, but if unavailable, skip the tests.zUnable to download T)Zallow_module_levelN)rConnectionErrorModuleNotFoundErrorpytestskip)Z data_filenamer r r&fetchs   rqcsdks Jfdd}|S)a`Decorator to run the same function multiple times in parallel. This decorator is useful to ensure that separate threads execute concurrently and correctly while releasing the GIL. Parameters ---------- num_threads : int, optional The number of times the function is run in parallel. warnings_matching: list or None This parameter is passed on to `expected_warnings` so as not to have race conditions with the warnings filters. A single `expected_warnings` context manager is used for all threads. If None, then no warnings are checked. rcstfdd}|S)Ncstrg}tdD]}tj||d}||q|D] }|q>|i||D] }|q^Wdn1s0YdS)Nr)targetargskwargs)rrange threadingThreadr.startr4)rsrtthreadsithread)r5 num_threadswarnings_matchingr r&inner8s   z/run_in_parallel..wrapper..inner) functoolswraps)r5r~r|r})r5r&wrapper7sz run_in_parallel..wrapperr )r|r}rr rr&run_in_parallel"s r)N)N)rJ)rJ)rN)@r*rBrestructrvrtempfilerr[r`rZ numpy.testingrrrrrr r r r r rr]rrZdata._fetchersrutilrrrr _warningsrromarkZskipifZxfailZ parametrizeZraisesZfixturecompiler,calcsizeZarch32environgetr_rPrOint ValueErrorr'r(r<rIrTrWrjrlrqrr r r r&sR  4      / #F