B \@sdZddlZddlZddlZddlZddlZddlmZddlZ ddlm Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZddlZdd lmZmZmZmZmZm Z ddl!Z!e"d Z#e!j$j%Z%e!j$j&Z&e!j$j'Z'e!j(Z(e!j)Z)e*d d d kZ+d%ddZ,d&ddZ-ddZ.ddZ/d'ddZ0d(ddZ1ddZ2ddZ3d)dd Z4e5d!kr|e0d"e1d"e1d"d#e1d"d$dS)*z Testing utilities. N)NamedTemporaryFile)testing) assert_array_equalassert_array_almost_equalassert_array_lessassert_array_almost_equal_nulp assert_equalTestCaseassert_allcloseassert_almost_equalassert_ assert_warnsassert_no_warnings)expected_warnings)dataio img_as_uint img_as_float img_as_int img_as_ubytez%(\s*>>>.*?)(\s*)#\s*skip\s+if\s+(.*)$P cCs4d||f}|dk r |d|7}||ks0t|dS)Nz%r is not lower than %rz: )AssertionError)abmsgmessager 6lib/python3.7/site-packages/skimage/_shared/testing.py assert_less)s  r"cCs4d||f}|dk r |d|7}||ks0t|dS)Nz%r is not greater than %rz: )r)rrrrr r r!assert_greater0s  r#c Cs|jd}g}x|D]}t|}|dkr8||q|\}}}yt||jr`||d}Wn.tk rt||j jr||d}YnX||qWd ||_|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)funclinesZ new_linesliner(codeZspaceexprr r r!doctest_skip_parser7s"      r5cCspd|krd|}t|dd}|j}|tj|||dtj||d}yt|Wntk rjYnX|S)z/Save and read an image using a specified plugin.F)suffixdelete)plugin) rnamecloserZimsaveZimreadosremove Exception)Zimager9r7Z temp_filefnamenewr r r! roundtripfs rApngc CsPtt}t|||}t|||dk}t|||}t|tj|t |}t dgt|||}WdQRXt||t dgt |}WdQRX| dkr|d8}t dgt|||} WdQRXt| |n2t dg"t|||} t| t|WdQRXt |} t dgt| ||} WdQRXt| |dS)zCheck roundtrip behavior for color images. All major input types should be handled as ubytes and read back correctly. zprecision lossN)tiftiffdz sign losszsign loss|precision loss)rrZchelsearArr astypenpuint8rrrlowerr) r9fmtimgr1img2r2img3r3img4r4img5r5r r r! color_checkvs0            rVc CsZtt}t|||}t|||dk}t|||}t|tj|t |}t dgt|||}WdQRX|j j dkrt||nt|t |t dgt|}WdQRX|dkr|d8}t dgt|||} WdQRXt| |n2t d g"t|||} t| t |WdQRXt |} t| ||} t| | dS) zsCheck the roundtrip behavior for images that support most types. All major input types should be handled. rCzprecision|\A\ZNfzprecision loss)rDrErFzsign loss|\A\Zzprecision loss|sign loss)rrZmoonrArr rGrHrIrrZdtypeZkindrrrJ) r9rKrLrMrNrOrPrQrRrSrTrUr r r! mono_checks2           rXcCsXtdddlm}m}m}m}m}ddlm }ddl m }t j dtddS)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. defaultr)signalndimagespecialoptimizelinalg)loadmat)viewererrorN)warnings simplefilterZscipyrZr[r\r]r^Zscipy.ior_Zskimager`rHZrandomZseed)rZr[r\r]r^r_r`r r r! setup_tests     rdcCstddS)zhDefault package level teardown routine for skimage tests. Restore warnings to default behavior rYN)rbrcr r r r! teardown_testsrecsdks tfdd}|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. rcstfdd}|S)Ncsng}x.tdD]}tj||d}||qWx|D] }|q:W||}x|D] }|qZW|S)Nr)targetargskwargs)range threadingZThreadr)startr/)rgrhZthreadsiZthreadresult)r0 num_threadsr r!inners     z-test_parallel..wrapper..inner) functoolswraps)r0ro)rn)r0r!wrappersztest_parallel..wrapper)r)rnrrr )rnr! test_parallels  rs__main__ZpilZbmprE)N)N)rB)rB)r)6r%r<reZstructrjrpZtempfilerZnumpyrHrZ numpy.testingrrrrrr r r r r r _warningsrrbrrrrrrZpytestcompiler'ZmarkZskipifZxfailZ parametrizeZraisesZfixtureZcalcsizeZarch32r"r#r5rArVrXrdrers__name__r r r r!sD  4     / % ' %