ó  ‰\c@sdZddlZddlZddlZddlZddlZddlmZddlZ ddlm Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZddlZdd lmZmZmZmZmZm Z ddl!Z!ej"d ƒZ#e!j$j%Z%e!j$j&Z&e!j$j'Z'e!j(Z(e!j)Z)ej*d ƒd d kZ+dd„Z-dd„Z.d„Z/d„Z0dd„Z1dd„Z2d„Z3d„Z4dd„Z5e6dkr e1dƒe2dƒe2ddƒe2ddƒndS(s Testing utilities. iÿÿÿÿN(tNamedTemporaryFile(ttesting( tassert_array_equaltassert_array_almost_equaltassert_array_lesstassert_array_almost_equal_nulpt assert_equaltTestCasetassert_allclosetassert_almost_equaltassert_t assert_warnstassert_no_warningsi(texpected_warningsi(tdatatiot img_as_uintt img_as_floatt img_as_intt img_as_ubytes%(\s*>>>.*?)(\s*)#\s*skip\s+if\s+(.*)$tPii cCsId||f}|dk r-|d|7}n||ksEt|ƒ‚dS(Ns%r is not lower than %rs: (tNonetAssertionError(tatbtmsgtmessage((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt assert_less)s cCsId||f}|dk r-|d|7}n||ksEt|ƒ‚dS(Ns%r is not greater than %rs: (RR(RRRR((s6lib/python2.7/site-packages/skimage/_shared/testing.pytassert_greater0s cCsð|jjdƒ}g}x¿|D]·}tj|ƒ}|dkrS|j|ƒqn|jƒ\}}}y't||jƒrŽ||d}nWn7t k rÈt||j jƒrÉ||d}qÉnX|j|ƒqWdj |ƒ|_|S(sD 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 s s# doctest: +SKIPN( t__doc__tsplittSKIP_REtmatchRtappendtgroupstevalt __globals__tAttributeErrort__init__tjoin(tfunctlinest new_linestlineR tcodetspacetexpr((s6lib/python2.7/site-packages/skimage/_shared/testing.pytdoctest_skip_parser7s"    cCs•d|krd|}ntd|dtƒ}|j}|jƒtj||d|ƒtj|d|ƒ}ytj|ƒWnt k rnX|S(s/Save and read an image using a specified plugint.tsuffixtdeletetplugin( RtFalsetnametcloseRtimsavetimreadtostremovet Exception(timageR3R1t temp_filetfnametnew((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt roundtripfs     tpngc Cs±ttjƒƒ}t|||ƒ}tj||ƒ|dk}t|||ƒ}tj|jtjƒ|ƒt |ƒ}t dgƒt|||ƒ}WdQXtj||ƒt dgƒt |ƒ}WdQX|j ƒd kr+|d8}t dgƒt|||ƒ} WdQXtj| |ƒn>t dgƒ-t|||ƒ} tj| t|ƒƒWdQXt |ƒ} t dgƒt| ||ƒ} WdQXtj| |ƒdS( sƒCheck roundtrip behavior for color images. All major input types should be handled as ubytes and read back correctly. i€sprecision lossNttifttiffids sign lossssign loss|precision loss(RBRC(RRtchelseaR@RRtastypetnptuint8RR RtlowerR( R3tfmttimgtr1timg2tr2timg3tr3timg4tr4timg5tr5((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt color_checkvs0    c CsÆttjƒƒ}t|||ƒ}tj||ƒ|dk}t|||ƒ}tj|jtjƒ|ƒt |ƒ}t dgƒt|||ƒ}WdQX|j j dkrÇtj||ƒntj|t |ƒƒt dgƒt|ƒ}WdQX|jƒd krV|d8}t d gƒt|||ƒ} WdQXtj| |ƒn>t d gƒ-t|||ƒ} tj| t |ƒƒWdQXt |ƒ} t| ||ƒ} tj| | ƒdS( ssCheck the roundtrip behavior for images that support most types. All major input types should be handled. i€sprecision|\A\ZNtfsprecision lossRBRCidssign loss|\A\Zsprecision loss|sign loss(RBRC(RRtmoonR@RRRERFRGRR tdtypetkindRRRH( R3RIRJRKRLRMRNRORPRQRRRS((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt mono_check›s2    cCsvtjdƒddlm}m}m}m}m}ddlm }ddl m }t j jdƒtjdƒdS( s½Default 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. tdefaultiÿÿÿÿ(tsignaltndimagetspecialtoptimizetlinalg(tloadmat(tvieweriterrorN(twarningst simplefiltertscipyR[R\R]R^R_tscipy.ioR`tskimageRaRFtrandomtseed(R[R\R]R^R_R`Ra((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt setup_testÂs  (cCstjdƒdS(shDefault package level teardown routine for skimage tests. Restore warnings to default behavior RZN(RcRd(((s6lib/python2.7/site-packages/skimage/_shared/testing.pyt teardown_testÕscs%ˆdkst‚‡fd†}|S(s>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. ics%tjˆƒ‡‡fd†ƒ}|S(Ncs•g}xCtˆdƒD]1}tjdˆd|d|ƒ}|j|ƒqWx|D]}|jƒqSWˆ||Ž}x|D]}|jƒq}W|S(Nittargettargstkwargs(tranget threadingtThreadR!tstartR'(RmRntthreadstitthreadtresult(R(t num_threads(s6lib/python2.7/site-packages/skimage/_shared/testing.pytinnerís  (t functoolstwraps(R(Rx(Rw(R(s6lib/python2.7/site-packages/skimage/_shared/testing.pytwrapperìs!(R(RwR{((Rws6lib/python2.7/site-packages/skimage/_shared/testing.pyt test_parallelÝs t__main__tpiltbmpRC(7RR9tretstructRpRyttempfileRtnumpyRFRt numpy.testingRRRRRRRR R R R t _warningsR RctRRRRRRtpytesttcompileRtmarktskipiftxfailt parametrizetraisestfixturetcalcsizetarch32RRRR/R@RTRYRjRkR|t__name__(((s6lib/python2.7/site-packages/skimage/_shared/testing.pytsD      L .         /  % '   %