U
|
¬ý°d½ ã @ sÖ d dl mZ d dlmZmZ d dlZd dlZd dlmZm Z m
|
Z
|
m Z m Z d dl Z eeddddfddd d d
|
d d d dZdd dddZddd
|
d ddddZddddddZddd d d d Zddd d d!ZdS )"é )Ú annotations)ÚcontextmanagerÚ nullcontextN)Ú GeneratorÚLiteralÚSequenceÚTypeÚcastÚalwaysTz7type[Warning] | bool | tuple[type[Warning], ...] | NonezCLiteral[('error', 'ignore', 'always', 'default', 'module', 'once')]Úboolz
|
str | Nonez4Generator[list[warnings.WarningMessage], None, None])Úexpected_warningÚ filter_levelÚcheck_stacklevelÚraise_on_extra_warningsÚmatchÚreturnc c sj d}t jddP}t |¡ z
|
|V W 5 | rJttt | } t|| ||d |rZt|| d X W 5 Q R X dS )a$
|
Context manager for running code expected to either raise a specific warning,
|
multiple specific warnings, or not raise any warnings. Verifies that the code
|
raises the expected warning(s), and that it does not raise any other unexpected
|
warnings. It is basically a wrapper around ``warnings.catch_warnings``.
|
|
Parameters
|
----------
|
expected_warning : {Warning, False, tuple[Warning, ...], None}, default Warning
|
The type of Exception raised. ``exception.Warning`` is the base
|
class for all warnings. To raise multiple types of exceptions,
|
pass them as a tuple. To check that no warning is returned,
|
specify ``False`` or ``None``.
|
filter_level : str or None, default "always"
|
Specifies whether warnings are ignored, displayed, or turned
|
into errors.
|
Valid values are:
|
|
* "error" - turns matching warnings into exceptions
|
* "ignore" - discard the warning
|
* "always" - always emit a warning
|
* "default" - print the warning the first time it is generated
|
from each location
|
* "module" - print the warning the first time it is generated
|
from each module
|
* "once" - print the warning the first time it is generated
|
|
check_stacklevel : bool, default True
|
If True, displays the line that called the function containing
|
the warning to show were the function is called. Otherwise, the
|
line that implements the function is displayed.
|
raise_on_extra_warnings : bool, default True
|
Whether extra warnings not of the type `expected_warning` should
|
cause the test to fail.
|
match : str, optional
|
Match warning message.
|
|
Examples
|
--------
|
>>> import warnings
|
>>> with assert_produces_warning():
|
... warnings.warn(UserWarning())
|
...
|
>>> with assert_produces_warning(False):
|
... warnings.warn(RuntimeWarning())
|
...
|
Traceback (most recent call last):
|
...
|
AssertionError: Caused unexpected warning(s): ['RuntimeWarning'].
|
>>> with assert_produces_warning(UserWarning):
|
... warnings.warn(RuntimeWarning())
|
Traceback (most recent call last):
|
...
|
AssertionError: Did not see expected warning of class 'UserWarning'.
|
|
..warn:: This is *not* thread-safe.
|
T)Úrecord)Úcaught_warningsr r r )r r N)ÚwarningsÚcatch_warningsÚ simplefilterr r ÚWarningÚ _assert_caught_expected_warningÚ _assert_caught_no_extra_warnings)r r r r r Z__tracebackhide__Úw© r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/_testing/_warnings.pyÚassert_produces_warning s$ C
|
|
üþr z type[Warning])ÚwarningÚ conditionc K s |rt | f|S t S dS )zX
|
Return a context manager that possibly checks a warning based on the condition
|
N)r r )r r Úkwargsr r r Úmaybe_produces_warningl s r! z!Sequence[warnings.WarningMessage]ÚNone)r r r r r c C s¢ d}d}g }| D ]L}t |j|rd}|r0t| |dk rt |t|j¡rPd}q| |j¡ q|svtdt |j
|
|r|stdt |j
|
d| d| dS )zEAssert that there was the expected warning among the caught warnings.FTNz&Did not see expected warning of class zDid not see warning z matching 'z$'. The emitted warning messages are ) Ú
|
issubclassÚcategoryÚ&_assert_raised_with_correct_stacklevelÚreÚsearchÚstrÚmessageÚappendÚAssertionErrorÚreprÚ__name__)r r r r Z saw_warningZmatched_messageZunmatched_messagesÚactual_warningr r r r v s( ÿÿr )r r r c C s~ g }| D ]Z}t ||r|jtkrFdt|jkr0qtdd tjD rFq| |jj |j|j
|
|j f¡ q|rzt dt | dS )zFAssert that no extra warnings apart from the expected ones are caught.zunclosed <ssl.SSLSocketc s s | ]}d |kV qdS )Z
|
matplotlibNr )Ú.0Úmodr r r Ú <genexpr>° s z3_assert_caught_no_extra_warnings.<locals>.<genexpr>z Caused unexpected warning(s): N)Ú_is_unexpected_warningr$ ÚResourceWarningr( r) ÚanyÚsysÚmodulesr* r- ÚfilenameÚlinenor+ r, )r r Zextra_warningsr. r r r r s"
|
|
üÿ r zwarnings.WarningMessage)r. r r c C s, | r |s dS t tt |}tt| j| S )z1Check if the actual warning issued is unexpected.T)r r r r r# r$ )r. r r r r r2 ¿ s r2 )r. r c C sV ddl m}m} || d d }d| j d|j d| j }| j|jksRt|d S )Nr )Ú getframeinfoÚstacké zGWarning not set with correct stacklevel. File where warning is raised: z != z. Warning message: )Úinspectr9 r: r7 r) r+ )r. r9 r: ZcallerÚmsgr r r r% Ê s
|
ÿr% )Ú
|
__future__r Ú
|
contextlibr r r& r5 Útypingr r r r r r r r r! r r r2 r% r r r r Ú<module> s" ù X
|
'"
|
