a lc^@sdddgZddlZddlZddlZddlZddlZddlmZddlm Z eddZ eddZ dS) all_warningsexpected_warningsN)contextmanager)mockc cst}|r*t|D]}i|djd<q~ttjD]*\}}z|j Wq:t ybYq:0q:t j ddT}t jtjddi t d|VWdn1s0YWdn1s0YdS) a Context for use in testing to ensure that all warnings are raised. Examples -------- >>> import warnings >>> def foo(): ... warnings.warn(RuntimeWarning("bar")) 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() We can now run ``foo()`` without a warning being raised: >>> from numpy.testing import assert_warns # doctest: +SKIP >>> foo() # doctest: +SKIP To catch the warning, we call in the help of ``all_warnings``: >>> with all_warnings(): # doctest: +SKIP ... assert_warns(RuntimeWarning, foo) r__warningregistry__T)recordZTRAITLETS_ALL_DEPRECATIONS1alwaysN)inspect currentframegetouterframesf_localslistsysmodulesitemsrclearAttributeErrorwarningscatch_warningsrpatchdictosenviron simplefilter)framef_modwr U/var/www/html/django/DPS/env/lib/python3.9/site-packages/traitlets/tests/_warnings.pyrs"   ccst}|Vdd|D}|D]T}d}|D]0}t|t|jdur,d}||vr,||q,|s tdt|jq t|dkrdd |}t|Wdn1s0YdS) aContext for use in testing to catch known warnings matching regexes Parameters ---------- matching : list of strings or compiled regexes Regexes for the desired warning to catch Examples -------- >>> from skimage import data, img_as_ubyte, img_as_float # doctest: +SKIP >>> with expected_warnings(["precision loss"]): # doctest: +SKIP ... d = img_as_ubyte(img_as_float(data.coins())) # doctest: +SKIP 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 accomodate different build enviroments 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. cSsg|]}d|dvr|qS)z\A\Z|)split).0mr r r! fz%expected_warnings..FNTzUnexpected warning: %srzNo warning raised matching: %s ) rresearchstrmessageremove ValueErrorlenjoin)Zmatchingr remainingwarnfoundmatchmsgr r r!rDs  ) __all__r rr)rr contextlibrZunittestrrrr r r r!s   4