a ¤Â=bJã@shddlmZddlZddlZddlZddlZddlZgd¢Zejej ddZ edd„ƒZ edd „ƒZ dS) é)ÚcontextmanagerN)Ú all_warningsÚexpected_warningsÚwarné)Ú stacklevelc cs®ddl}| ¡}|r2| |¡D]}i|djd<q~ttj ¡ƒD]*\}}z|j  ¡WqBt yjYqB0qBt j dd }t   d¡|VWdƒn1s 0YdS)a& Context for use in testing to ensure that all warnings are raised. Examples -------- >>> import warnings >>> def foo(): ... warnings.warn(RuntimeWarning("bar"), stacklevel=2) We raise the warning once, while the warning filter is set to "once". Hereafter, the warning is invisible, even with custom filters: >>> with warnings.catch_warnings(): ... warnings.simplefilter('once') ... foo() # doctest: +SKIP We can now run ``foo()`` without a warning being raised: >>> from numpy.testing import assert_warns >>> foo() # doctest: +SKIP To catch the warning, we call in the help of ``all_warnings``: >>> with all_warnings(): ... assert_warns(RuntimeWarning, foo) rNÚ__warningregistry__T)ÚrecordÚalways)ÚinspectZ currentframeZgetouterframesÚf_localsÚlistÚsysÚmodulesÚitemsrÚclearÚAttributeErrorÚwarningsÚcatch_warningsÚ simplefilter)r ÚframeÚfÚmod_nameÚmodÚw©rú8lib/python3.9/site-packages/skimage/_shared/_warnings.pyrs   rccs>t|tƒrtdƒ‚|dur$dVdStj dd¡}| ¡dkrDd}n| ¡dkrVd}n tt|ƒƒ}t ƒÀ}|Vd|vr„|  d¡qpd d „|Dƒ}|D]X}d}|D]0}t   |t|j ƒ¡dur¢d}||vr¢|  |¡q¢|r–|s–td t|j ƒƒ‚q–|rt|ƒd krd d |¡}t|ƒ‚Wdƒn1s00YdS)a‹Context for use in testing to catch known warnings matching regexes Parameters ---------- matching : None or a list of strings or compiled regexes Regexes for the desired warning to catch If matching is None, this behaves as a no-op. Examples -------- >>> import numpy as np >>> rng = np.random.default_rng() >>> image = rng.integers(0, 2**16, size=(100, 100), dtype=np.uint16) >>> # rank filters are slow when bit-depth exceeds 10 bits >>> from skimage import filters >>> with expected_warnings(['Bad rank filter performance']): ... median_filtered = filters.rank.median(image) Notes ----- Uses `all_warnings` to ensure all warnings are raised. Upon exiting, it checks the recorded warnings for the desired matching pattern(s). Raises a ValueError if any match was not found or an unexpected warning was raised. Allows for three types of behaviors: `and`, `or`, and `optional` matches. This is done to accommodate different build environments or loop conditions that may produce different warnings. The behaviors can be combined. If you pass multiple patterns, you get an orderless `and`, where all of the warnings must be raised. If you use the `|` operator in a pattern, you can catch one of several warnings. Finally, you can use `|\A\Z` in a pattern to signify it as optional. zA``matching`` should be a list of strings and not a string itself.NZSKIMAGE_TEST_STRICT_WARNINGSÚ1ÚtrueTZfalseFcSsg|]}d| d¡vr|‘qS)z\A\Zú|)Úsplit)Ú.0ÚmrrrÚ †óz%expected_warnings..zUnexpected warning: %srzNo warning raised matching: %sÚ )Ú isinstanceÚstrÚ ValueErrorÚosÚenvironÚgetÚlowerÚboolÚintrÚremoveÚreÚsearchÚmessageÚlenÚjoin)ZmatchingZstrict_warningsrZ remainingrÚfoundÚmatchÚmsgrrrrIs8%      r) Ú contextlibrrrr0Ú functoolsr)Ú__all__ÚpartialrrrrrrrÚs  8