o hk@shddlmZddlZddlZddlZddlZddlZgdZejej ddZ eddZ edd Z dS) )contextmanagerN) all_warningsexpected_warningswarn) stacklevelc csddl}|}|r||D] }i|djd<q~ttjD]\}}z|j Wq"t y6Yq"wt j dd}t d|VWddS1sQwYdS)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)recordalways)inspect currentframegetouterframesf_localslistsysmodulesitemsrclearAttributeErrorwarningscatch_warnings simplefilter)r framefmod_namemodwrT/var/www/html/scripts/venv/lib/python3.10/site-packages/skimage/_shared/_warnings.pyrs"   "rc cs^t|tr td|durdVdStjdd}|dkr#d}n|dkr,d}ntt|}t p}|Vd|vrF| dd|vs=d d |D}|D]-}d}|D]}t |t|j durmd}||vrm| |qU|r||s|td t|j qO|rt|d krd }d|||}t|WddSWddS1swYdS)aContext 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.NSKIMAGE_TEST_STRICT_WARNINGS1trueTfalseFcSsg|] }d|dvr|qS)z\A\Z|)split).0mrrr sz%expected_warnings..zUnexpected warning: r zNo warning raised matching:) isinstancestr ValueErrorosenvirongetlowerboolintrremoveresearchmessagelenjoin) matchingstrict_warningsr remainingrfoundmatchnewlinemsgrrrrIsJ %     "r) contextlibrrrr3 functoolsr,__all__partialrrrrrrrs  8