U
|
®ý°dÍ! ã @ sh d Z ddlZddlZddlmZmZ ddlmZ ej e
|
e e e eeegddd Zej e
|
e fe e fe e
|
feefee feefe efe efgdd d d
|
d Zd d Zej d¡dd Zej dddddddddddd g ¡d d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Z d+d, Z d-d. Z d/d0 Z d1d2 Z ej d3d4dg¡G d5d6 d6Z!d7d8 Z"d9d: Z#dS );zG"
|
Test module for testing ``pandas._testing.assert_produces_warning``.
|
é N)Ú DtypeWarningÚPerformanceWarning)Úparamsc C s | j S )zx
|
Return unique warning.
|
|
Useful for testing behavior of tm.assert_produces_warning with various categories.
|
©Úparam©Úrequest© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_produces_warning.pyÚcategory s r c C s
|
t | jS ©N)ÚtypeÚ__name__)Úxr r r
|
Ú<lambda>/ ó r )r Úidsc C s | j S )z
|
Return pair or different warnings.
|
|
Useful for testing how several different warnings are handled
|
in tm.assert_produces_warning.
|
r r r r r
|
Úpair_different_warnings$ s r c C s t dt¡ t dt¡ d S )NÚf1Úf2)ÚwarningsÚwarnÚ FutureWarningÚRuntimeWarningr r r r
|
Úf; s r zignore:f1:FutureWarningc
|
C s^ d} t jt| d" t t¡ t W 5 Q R X W 5 Q R X tjtdd t W 5 Q R X d S )Nú Caused unexpected warning\(s\)©ÚmatchF©Zraise_on_extra_warnings)ÚpytestÚraisesÚAssertionErrorÚtmÚassert_produces_warningr r ©Úmsgr r r
|
Ú*test_assert_produces_warning_honors_filter@ s r& zmessage, match)Ú N)r' r' )úWarning messagez.*)r( ZWar)r( z
|
[Ww]arning)r( Zage)r( zage$)úMessage 12-234 with numbersz \d{2}-\d{3})r) z^Mes.*\d{2}-\d{3})r) z\d{2}-\d{3}\s\S+)z Message, which we do not matchNc C s* t j| |d t || ¡ W 5 Q R X d S )Nr ©r" r# r r )r Úmessager r r r
|
Ú%test_catch_warning_category_and_matchL s r, c
|
C s\ t } d}d}tjt|d8 tj| |d t d| ¡ t d| ¡ W 5 Q R X W 5 Q R X d S )NzDid not see this warningzÇDid not see warning 'RuntimeWarning' matching 'Did not see this warning'. The emitted warning messages are \[RuntimeWarning\('This is not a match.'\), RuntimeWarning\('Another unmatched warning.'\)\]r úThis is not a match.úAnother unmatched warning.)r r r r! r" r# r r ©r r Ú unmatchedr r r
|
Ú"test_fail_to_match_runtime_warninga s ÿ r1 c
|
C s\ t } d}d}tjt|d8 tj| |d t d| ¡ t d| ¡ W 5 Q R X W 5 Q R X d S )NÚWarningz³Did not see warning 'FutureWarning' matching 'Warning'. The emitted warning messages are \[FutureWarning\('This is not a match.'\), FutureWarning\('Another unmatched warning.'\)\]r r- r. )r r r r! r" r# r r r/ r r r
|
Ú!test_fail_to_match_future_warningp s ÿ r3 c
|
C s\ t } d}d}tjt|d8 tj| |d t d| ¡ t d| ¡ W 5 Q R X W 5 Q R X d S )Nz\d+z·Did not see warning 'ResourceWarning' matching '\\d\+'. The emitted warning messages are \[ResourceWarning\('This is not a match.'\), ResourceWarning\('Another unmatched warning.'\)\]r r- r. )ÚResourceWarningr r r! r" r# r r r/ r r r
|
Ú#test_fail_to_match_resource_warning s ÿ r5 c
|
C sL | \}}d}t jt|d( t |¡ t d|¡ W 5 Q R X W 5 Q R X d S )Nz%Did not see expected warning of classr zwarning message©r r r! r" r# r r )r Úexpected_categoryZactual_categoryr r r r
|
Ú!test_fail_to_catch_actual_warning s
|
r8 c C s> | \}}t j|dd t d|¡ t d|¡ W 5 Q R X d S )NFr úExpected warningzUnexpected warning OKr* )r r7 Úextra_categoryr r r
|
Útest_ignore_extra_warning s r; c
|
C sX | \}}d}t jt|d4 t |¡ t d|¡ t d|¡ W 5 Q R X W 5 Q R X d S )Nr r r9 zUnexpected warning NOT OKr6 )r r7 r: r r r r
|
Útest_raise_on_extra_warning s r< c C sF t } tj| dd* t d| ¡ t d| ¡ t d| ¡ W 5 Q R X d S )Nú ^Match thisr ú
|
Match thisúDo not match thatúDo not match that either)Ú UserWarningr" r# r r ©r r r r
|
Ú1test_same_category_different_messages_first_match¦ s
|
rC c C sF t } tj| dd* t d| ¡ t d| ¡ t d| ¡ W 5 Q R X d S )Nr= r r? r@ r> )ÚDeprecationWarningr" r# r r rB r r r
|
Ú0test_same_category_different_messages_last_match® s
|
rE c C s> t tf} tj| dd t dt ¡ t dt¡ W 5 Q R X d S )Nr= r r> zMatch this too)r rA r" r# r r rB r r r
|
Ú test_match_multiple_warnings¶ s rF c
|
C sX | \}}t jtdd8 tj|dd t d|¡ t d|¡ W 5 Q R X W 5 Q R X d S )Nz Did not see warning.*matchingr r= zDo not match itr> r6 )r Ztarget_categoryZother_categoryr r r
|
Ú&test_right_category_wrong_match_raises¾ s
|
rG Ú false_or_noneFc @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestFalseOrNoneExpectedWarningc
|
C s> d}t jt|d" t |¡ t W 5 Q R X W 5 Q R X d S )Nr r )r r r! r" r# r )ÚselfrH r% r r r
|
Útest_raise_on_warningÈ s z4TestFalseOrNoneExpectedWarning.test_raise_on_warningc C s t |¡ W 5 Q R X d S r )r" r# ©rJ rH r r r
|
Ú test_no_raise_without_warningÎ s z<TestFalseOrNoneExpectedWarning.test_no_raise_without_warningc C s$ t j|dd t W 5 Q R X d S )NFr )r" r# r rL r r r
|
Ú'test_no_raise_with_false_raise_on_extraÒ s zFTestFalseOrNoneExpectedWarning.test_no_raise_with_false_raise_on_extraN)r Ú
|
__module__Ú __qualname__rK rM rN r r r r
|
rI Æ s rI c
|
C sÀ d} t jt| d t t¡
|
tW 5 Q R X W 5 Q R X t jt| d, t t¡ t dt ¡ t
|
W 5 Q R X W 5 Q R X d} t jt| d, t d ¡ t dt ¡ t W 5 Q R X W 5 Q R X d S )Nz3Did not see expected warning of class 'UserWarning'r r zCaused unexpected warning) r r r! r" r# rA Ú
|
ValueErrorr r r Ú
|
IndexErrorÚ SystemErrorr$ r r r
|
Ú test_raises_during_exception× s rT c
|
C s t jtdd$ t d ¡ tdW 5 Q R X W 5 Q R X t jtdd4 tjtdd t dt¡ tdW 5 Q R X W 5 Q R X d S )NÚErrorr r ) r r Ú SyntaxErrorr" r# rQ r r r r r r r
|
Ú test_passes_during_exceptioné s rW )$Ú__doc__r r Z pandas.errorsr r Zpandas._testingZ_testingr" Zfixturer r4 rA r rD r r r ÚmarkÚfilterwarningsr& Z parametrizer, r1 r3 r5 r8 r; r< rC rE rF rG rI rT rW r r r r
|
Ú<module> sz ùÿ
|
ø
|
õ
|
|
|
|
õþ
|
|
