0-Ph$dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZddlmZmZddlmZdd lmZm Z m!Z!m"Z"d d l#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-ej.d Z/ej0ddzdkZ1ddZ2ddZ3dZ4dZ5ddZ6ddZ7dZ8ddZ9dddZ:dS) z 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_warnings)is_wasmz%(\s*>>>.*?)(\s*)#\s*skip\s+if\s+(.*)$P cH|d|}||d|zz }||ks J|dS)Nz is not lower than : abmsgmessages W/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/_shared/testing.py assert_lessr'5s?....G 4#: q555'55555cH|d|}||d|zz }||ks J|dS)Nz is not greater than rr r!s r&assert_greaterr*<s?00100G 4#: q555'55555r(c|jd}g}|D]}t|}|||4|\}}} t ||jr||zdz}n2#t$r%t ||j jr||zdz}YnwxYw||d ||_|S)aCDecorator 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_linesliner0codespaceexprs r&doctest_skip_parserr?Cs 0 L  t $ $EI d## =   T " " " !LLNNeT 9D$*++ 9e|&88 9 9 9D$-344 9e|&88 9 99Y''DL Ks,B  ,B98B9cd|vrd|z}t|d5}|j}dddn #1swxYwYtj|||tj||} t j|n#t$rYnwxYw|S)z/Save and read an image using a specified plugin.F)suffixdeleteN)plugin)rnamerimsaveimreadosremove Exception)imagerDrB temp_filefnamenews r& roundtriprOrs &v 6% 8 8 8IIeU6**** )E& ) ) )C  %      Js/33(A== B  B pngcttj}t|||}t j|||dk}t|||}t j||tt|}t|||}t j||t|}| dvr,|dz}t|||} t j| |n3t|||} t j| t|t|} t| ||} t j| |dS)zCheck roundtrip behavior for color images. All major input types should be handled as ubytes and read back correctly. tiftiffdN) rrchelsearOrr astypeboolrrlowerr rDfmtimgr1img2r2img3r3img4r4img5r5s r& color_checkrgsT t|~~ & &C 3 $ $B C$$$ 9D 4 % %B D"))D//222   D 4 % %B B$$$ c??D yy{{''   tVS ) )D)))) tVS ) )L$6$6777 s  D 4 % %B B$$$$$r(c\ttj}t|||}t j|||dk}t|||}t j||tt|}t|||}|j j dkrt j||n"t j|t|t|}| dvr,|dz}t|||} t j| |n3t|||} t j| t|t|} t| ||} t j| | dS)zsCheck the roundtrip behavior for images that support most types. All major input types should be handled. rRfrSrVN)rrmoonrOrr rXrYrdtypekindrrrZr[s r& mono_checkrms ty{{ # #C 3 $ $B C$$$ 9D 4 % %B D"))D//222   D 4 % %B x}b))))K$4$4555 c??D yy{{''   tVS ) )D)))) tVS ) )K$5$5666 s  D 4 % %B B%%%%%r(c t|S#ttf$rtjd|dYdSwxYw)z:Attempt to fetch data, but if unavailable, skip the tests.zUnable to download T)allow_module_levelN)rConnectionErrorModuleNotFoundErrorpytestskip) data_filenames r&fetchrusbTm$$$ 0 1TTT 9-99dSSSSSSSTs *??c&dksJfd}|S)aDecorator 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. It is currently skipped when running on WASM-based platforms, as the threading module is not supported. 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. rcjtrSddltjfd}|S)NrcVt5g}tdz D]/}||}||0|D]}||i||D]}| ddddS#1swxYwYdS)Nr)targetargskwargs)rrangeThreadr1startr7) rzr{threadsithreadr8 num_threads threadingwarnings_matchings r&innerz/run_in_parallel..wrapper..inners"#455 " "{Q//++A&--TV-TTFNN6****%##FLLNNNNd%f%%%%""FKKMMMM" " " " " " " " " " " " " " " " " " "sA?BB"%B")rr functoolswraps)r8rrrrs` @r&wrapperz run_in_parallel..wrappersb  K    " " " " " " "   " r(r )rrrs`` r&run_in_parallelrs8, ????0 Nr()offsetc d}tjdj}|j|z}|jj}|d|}|D]@}|jd|j}d|d|d|jj d|j } ||ks J| AdS) a6Assert correct stacklevel of captured warnings. When scikit-image raises warnings, the stacklevel should ideally be set so that the origin of the warnings will point to the public function that was called by the user and not necessarily the very place where the warnings were emitted (which may be inside some internal function). This utility function helps with checking that the stacklevel was set correctly on warnings captured by `pytest.warns`. Parameters ---------- warnings : collections.abc.Iterable[warning.WarningMessage] Warnings that were captured by `pytest.warns`. offset : int, optional Offset from the line this function is called to the line were the warning is supposed to originate from. For multiline calls, the first line is relevant. Defaults to -1 which corresponds to the line right above the one where this function is called. Raises ------ AssertionError If a warning in `warnings` does not match the expected line number or file name. Examples -------- >>> def test_something(): ... with pytest.warns(UserWarning, match="some message") as record: ... something_raising_a_warning() ... assert_stacklevel(record) ... >>> def test_another_thing(): ... with pytest.warns(UserWarning, match="some message") as record: ... iam_raising_many_warnings( ... "A long argument that forces the call to wrap." ... ) ... assert_stacklevel(record, offset=-3) Tr:z+Warning with wrong stacklevel: Expected: z Actual: z rN) inspectstackframef_linenof_code co_filenamefilenamelinenocategory__name__r%) warningsr__tracebackhide__r line_numberrexpectedwarningactualr$s r&assert_stacklevelrsP MOOA  $E.6)K|'H**[**H''$77w~77 @# @ @ @ @!* @ @/6o @ @ !!!3!!!!''r()N)rP)rN);r-rHplatformrestructsysrrtempfilernumpynpr numpy.testingrrrrr r r r r rrrrdata._fetchersrutilrrrr _warningsr_dependency_checksrrrmarkskipifxfail parametrizeraisesfixturecompiler/calcsizearch32r'r*r?rOrgrmrurrr r(r&rs  ''''''                          ######FFFFFFFFFFFF((((((''''''    k%  . "*= > >    !R ',,,^   %%%%@ & & & &FTTT0000f+-6'6'6'6'6'6'6'r(