a
iC @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z ddl
m Z ddl m
Z
mZmZmZmZ dZe ZdZG dd deZG d d
d
eZG d d d eZG d
d deZdd Zg Z dd Z dd Z dd Z dd Z dd Z!dd Z"d d Z#G d d d Z$G d!d" d"e$Z%G d#d$ d$e%Z&G d%d& d&e%Z'G d'd( d(ej(Z)G d)d* d*eZ*G d+d, d,e*Z+G d-d. d.e*Z,dS )/ N )result)strclass safe_repr_count_diff_all_purpose_count_diff_hashable_common_shorten_reprTz@
Diff is %s characters long. Set self.maxDiff to None to see it.c @ s e Zd ZdS )SkipTestN__name__
__module__ __qualname__ r r %/usr/lib64/python3.9/unittest/case.pyr s r c @ s e Zd ZdS ) _ShouldStopNr
r r r r r ! s r c @ s e Zd ZdS )_UnexpectedSuccessNr
r r r r r & s r c @ s& e Zd ZdddZejdddZdS ) _OutcomeNc C s4 d| _ || _t|d| _d| _g | _d | _g | _d S )NF
addSubTestT)expecting_failurer hasattrresult_supports_subtestssuccessskippedexpectedFailureerrors)selfr r r r __init__- s z_Outcome.__init__Fc
c s | j }d| _ zz
d V W n ty, Y n tyh } z&d| _ | j|t|f W Y d }~npd }~0 tyx Y nX t }| j r|| _
nd| _ | j ||f d }Y n 0 | j r| j r| j |d f W | j o|| _ n| j o|| _ 0 d S )NTF)
r KeyboardInterruptr r appendstrr sysexc_infor r r r )r test_caseisTestZ old_successer! r r r testPartExecutor6 s*
(
z_Outcome.testPartExecutor)N)F)r r r
r
contextlibcontextmanagerr% r r r r r , s
r c C s | S Nr )objr r r _idT s r* c O s t | ||f d S r( )_module_cleanupsr )functionargskwargsr r r addModuleCleanupY s r/ c
C sj g } t rZt \}}}z||i | W q tyV } z| | W Y d }~qd }~0 0 q| rf| d d S Nr )r+ pop Exceptionr )
exceptionsr, r- r. excr r r doModuleCleanups_ s "r5 c s, fdd}t tjr( }d ||S |S )Nc s4 t | ts$t| fdd}|} d| _ | _| S )Nc s t d S r( r r- r. reasonr r skip_wrapperu s z-skip..decorator..skip_wrapperT)
isinstancetype functoolswraps__unittest_skip____unittest_skip_why__) test_itemr: r8 r r decorators s
zskip..decorator )r; types FunctionType)r9 rB rA r r8 r skipo s
rF c C s | r t |S tS r( rF r* Z conditionr9 r r r skipIf s rI c C s | s t |S tS r( rG rH r r r
skipUnless s rJ c C s
d| _ | S )NT) __unittest_expecting_failure__)rA r r r r s r c s4 t | tr t fdd| D S t | to2t| S )Nc 3 s | ]}t | V qd S r( ) _is_subtype).0r$ basetyper r z _is_subtype..)r; tupleallr<
issubclass)expectedrO r rN r rL s
rL c @ s e Zd Zdd Zdd ZdS )_BaseTestCaseContextc C s
|| _ d S r( )r" )r r" r r r r s z _BaseTestCaseContext.__init__c C s | j | j|}| j |d S r( )r" _formatMessagemsgfailureException)r standardMsgrX r r r
_raiseFailure s z"_BaseTestCaseContext._raiseFailureN)r r r
r r[ r r r r rV s rV c @ s e Zd ZdddZdd ZdS )_AssertRaisesBaseContextNc C s@ t | | || _|| _|d ur*t|}|| _d | _d | _d S r( ) rV r rU r" recompileexpected_regexobj_namerX )r rU r" r_ r r r r s
z!_AssertRaisesBaseContext.__init__c C s zt | j| js"td|| jf |sX|dd | _|rNtdtt|f | W d } S |^}}z |j | _
W n t y t || _
Y n0 | ||i | W d n1 s0 Y W d } nd } 0 d S )Nz%s() arg 1 must be %srX z3%r is an invalid keyword argument for this function)
rL rU
_base_type TypeError_base_type_strr1 rX nextiterr r` AttributeErrorr )r namer- r. Z callable_objr r r handle s, .z _AssertRaisesBaseContext.handle)N)r r r
r rh r r r r r\ s
r\ c @ s. e Zd ZeZdZdd Zdd Zee j
Z dS )_AssertRaisesContextz-an exception type or tuple of exception typesc C s | S r( r r r r r __enter__ s z _AssertRaisesContext.__enter__c C s |d u r`z | j j}W n ty0 t| j }Y n0 | jrN| d|| j qj| d| n
t| t || j szdS |
d | _ | j d u rdS | j }|
t|s| d|jt| dS )Nz{} not raised by {}z
{} not raisedFT"{}" does not match "{}")rU r rf r r` r[ format traceback clear_framesrT with_traceback exceptionr_ searchpattern)r exc_type exc_valuetbexc_namer_ r r r __exit__ s,
z _AssertRaisesContext.__exit__N) r r r
BaseExceptionra rc rk rx classmethodrD GenericAlias__class_getitem__r r r r ri s
ri c @ s$ e Zd ZeZdZdd Zdd ZdS )_AssertWarnsContextz(a warning type or tuple of warning typesc C sR t tj D ]}t|dd ri |_qtjdd| _| j | _t
d| j | S )N__warningregistry__T)recordalways) listr modulesvaluesgetattrr~ warningscatch_warningswarnings_managerrk simplefilterrU )r vr r r rk s z _AssertWarnsContext.__enter__c C s | j ||| |d ur d S z | jj}W n tyD t| j}Y n0 d }| jD ]Z}|j}t|| jshqP|d u rt|}| j d ur| j
t|sqP|| _ |j | _ |j
| _
d S |d ur| d| j jt| | jr| d|| j n| d| d S )Nrl z{} not triggered by {}z{} not triggered)r rx rU r rf r r messager; r_ rr warningfilenamelinenor[ rm rs r` )r rt ru rv rw Zfirst_matchingmwr r r rx s>
z _AssertWarnsContext.__exit__N)r r r
Warningra rc rk rx r r r r r} s r} c @ s e Zd Zdd ZdS )_OrderedChainMapc c s8 t }| jD ]&}|D ] }||vr|| |V qq d S r( )setmapsadd)r seenmappingkr r r __iter__, s
z_OrderedChainMap.__iter__N)r r r
r r r r r r + s r c @ s e Zd ZeZdZdZdZdZg Z dddZ
dd Z d
d Z e
d d
Zdd Zdd Ze
dd Ze
dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Ze j e fd(d)Z d*d+ Z d,d- Z!d.d/ Z"d0d1 Z#d2d3 Z$d4d5 Z%d6d7 Z&dd9d:Z'd;d< Z(e
d=d> Z)d?d@ Z*dAdB Z+dCdD Z,ddEdFZ-ddGdHZ.ddIdJZ/dKdL Z0dMdN Z1dOdP Z2ddQdRZ3dSdT Z4ddUdVZ5ddWdXZ6ddYdZZ7dd[d\Z8dd]d^Z9dd_d`Z:dadb Z;ddcddZddidjZ?ddkdlZ@ddmdnZAddodpZBddqdrZCddsdtZDddudvZEddwdxZFddydzZGdd{d|ZHdd}d~ZIdddZJdddZKdddZLdddZMdddZNdd ZOdd ZPdddZQdddZRdd ZSeSe6 ZTZUeSe7 ZVZWeSe8 ZXZYeSe9 ZZZ[eSe/ Z\Z]eSe1Z^eSe.Z_eSeOZ`eSeQZaeSeRZbd8S )TestCaseTi i FrunTestc C s || _ d | _d| _zt| |}W n, tyL |dkrHtd| j|f Y n
0 |j| _g | _d | _ i | _
| t d | t
d | td | td | td | td d S ) NzNo testr z no such test method in %s: %sassertDictEqualassertListEqualassertTupleEqualassertSetEqualassertMultiLineEqual)_testMethodName_outcome_testMethodDocr rf
ValueError __class____doc__ _cleanups_subtest_type_equality_funcsaddTypeEqualityFuncdictr rR r frozensetr )r Z
methodName
testMethodr r r r f s( zTestCase.__init__c C s || j |< d S r( )r )r Ztypeobjr, r r r r s
z TestCase.addTypeEqualityFuncc O s | j |||f d S r( )r r r r, r- r. r r r
addCleanup s zTestCase.addCleanupc O s | j |||f d S r( )_class_cleanupsr clsr, r- r. r r r addClassCleanup s zTestCase.addClassCleanupc C s d S r( r rj r r r setUp s zTestCase.setUpc C s d S r( r rj r r r tearDown s zTestCase.tearDownc C s d S r( r r r r r
setUpClass s zTestCase.setUpClassc C s d S r( r r r r r
tearDownClass s zTestCase.tearDownClassc C s dS )Nr r rj r r r countTestCases s zTestCase.countTestCasesc C s t S r( )r Z
TestResultrj r r r defaultTestResult s zTestCase.defaultTestResultc C s$ | j }|r | dd S d S N
r )r stripsplitr docr r r shortDescription s zTestCase.shortDescriptionc C s dt | j| jf S )Nz%s.%sr r r rj r r r id s z TestCase.idc C s t | t |urtS | j|jkS r( )r< NotImplementedr r otherr r r __eq__ s zTestCase.__eq__c C s t t| | jfS r( )hashr< r rj r r r __hash__ s zTestCase.__hash__c C s d| j t| jf S Nz%s (%s))r r r rj r r r __str__ s zTestCase.__str__c C s dt | j| jf S )Nz<%s testMethod=%s>r rj r r r __repr__ s zTestCase.__repr__c C s< t |dd }|d ur ||| ntdtd || d S )NaddSkipz4TestResult has no addSkip method, skips not reported )r r warnRuntimeWarning
addSuccess)r r r" r9 r r r r _addSkip s zTestCase._addSkipc k s | j d u s| j js d V d S | j}|d u r4t|}n |j|}t| ||| _zr| j j| jdd d V W d n1 s~0 Y | j js| j j }|d ur|j
rt n | j j rt W || _n|| _0 d S )NTr# )
r r r r params new_child_SubTestr% r r Zfailfastr r )r rX r parentZ
params_mapr r r r subTest s$
$zTestCase.subTestc C s` |D ]V\}}t |tr(||j|| q|d urt|d | jrN||| q||| qd S r0 )r; r r r" rT rY
addFailureZaddError)r r r testr! r r r _feedErrorsToResult s
z TestCase._feedErrorsToResultc C sB z
|j }W n( ty2 tdt || Y n 0 || | d S )Nz@TestResult has no addExpectedFailure method, reporting as passes)addExpectedFailurerf r r r r )r r r! r r r r _addExpectedFailure s
z TestCase._addExpectedFailurec C sd z
|j }W nL tyV tdt z
td W n" tyP || t Y n0 Y n
0 || d S )NzCTestResult has no addUnexpectedSuccess method, reporting as failure) addUnexpectedSuccessrf r r r r r r r! )r r r r r r _addUnexpectedSuccess s
z TestCase._addUnexpectedSuccessc C s | d S r( )r rj r r r
_callSetUp" s zTestCase._callSetUpc C s
| d S r( r )r methodr r r _callTestMethod% s zTestCase._callTestMethodc C s | d S r( )r rj r r r
_callTearDown( s zTestCase._callTearDownc O s ||i | d S r( r r r r r _callCleanup+ s zTestCase._callCleanupNc
C s |d u r8| }t|dd }t|dd }|d ur<| nd }|| z,t| | j}t| jddspt|ddrt| jddpt|dd}| || | |W || |d ur| S t| ddpt|dd}t|}zf|| _| | |
W d n1 s
0 Y |j r||_ |j | dd |
| W d n1 sR0 Y d|_ | | | W d n1 s0 Y | |jD ]\}} | ||| q| ||j |j r|r|jr| ||j n
| | n
|| |W |j d |_d | _W || |d ur>| S |j d |_d | _0 W || |d ur| n || |d ur| 0 d S )
N startTestRun stopTestRunr? Fr@ rC rK Tr )r r Z startTestr r r ZstopTestr r r% r r r r r
doCleanupsr r r r r r r clear)
r r r r r skip_whyr outcomer r9 r r r run. s
,
(* (
z TestCase.runc C sl | j p
t }| jrf| j \}}}|| ( | j|g|R i | W d q 1 sZ0 Y q |jS r( )r r r r1 r% r r )r r r, r- r. r r r r q s 8zTestCase.doCleanupsc C sX g | _ | jrT| j \}}}z||i | W q tyP | j t Y q0 qd S r( )ZtearDown_exceptionsr r1 r2 r r r! r r r r doClassCleanups~ s zTestCase.doClassCleanupsc O s | j |i |S r( )r )r r- kwdsr r r __call__ s zTestCase.__call__c C s t | | j}t | jdds&t |ddrHt | jddp>t |dd}t|| | | | | jr| j \}}}| j |g|R i | qbd S )Nr? Fr@ rC )
r r r r r r r r r1 r )r r r r, r- r. r r r debug s
zTestCase.debugc C s t |d S r( r6 )r r9 r r r skipTest s zTestCase.skipTestc C s | |d S r( )rY )r rX r r r fail s z
TestCase.failc C s& |r"| |dt| }| |d S )Nz%s is not falserW r rY r exprrX r r r assertFalse s zTestCase.assertFalsec C s& |s"| |dt| }| |d S )Nz%s is not truer r r r r
assertTrue s zTestCase.assertTruec C sT | j s|p |S |d u r|S zd||f W S tyN dt|t|f Y S 0 d S )Nz%s : %s) longMessageUnicodeDecodeErrorr )r rX rZ r r r rW s
zTestCase._formatMessagec O s* t || }z|d||W d }S d }0 d S )N assertRaisesri rh )r expected_exceptionr- r. contextr r r r s
zTestCase.assertRaisesc O s t || }|d||S )N assertWarnsr} rh )r expected_warningr- r. r r r r r s
zTestCase.assertWarnsc C s ddl m} || ||S )Nr )_AssertLogsContext)_logr )r loggerlevelr r r r
assertLogs s zTestCase.assertLogsc C sF t |t |u r@| jt |}|d ur@t|tr.z; zMismatched values: %s) r r DeprecationWarningitemsr r r r rW ) r ZsubsetZ
dictionaryrX missingZ
mismatchedkeyvaluerZ r r r assertDictContainsSubsetr s4
z!TestCase.assertDictContainsSubsetc
C s t |t | }}zt|}t|}W n tyF t||}Y n0 ||krTd S t||}|rd}dd |D }d|} | || }| ||}| | d S )Nz Element counts were not equal:
c S s g | ] }d | qS )z First has %d, Second has %d: %rr )rM r r r r
rQ z-TestCase.assertCountEqual..r )
r collectionsCounterrb r r r r! rW r )
r r r rX Z first_seqZ
second_seqZ differencesrZ r+ r$ r r r assertCountEqual s
zTestCase.assertCountEqualc C s | |td | |td ||krt|| jks@t|| jkrN| ||| |jdd}|jdd}t|dkr|d|kr|d g}|d g}dt|| }dd t
|| }| ||}| |
|| d S )
Nz First argument is not a stringz Second argument is not a stringT)keependsr z
r r rC )r6 r r _diffThresholdr r r r r r r r! r rW )r r r rX Z
firstlinesZ secondlinesrZ r r r r r s
z TestCase.assertMultiLineEqualc C s2 ||k s.dt |t |f }| | || d S )Nz%s not less than %sr- r abrX rZ r r r
assertLess s zTestCase.assertLessc C s2 ||ks.dt |t |f }| | || d S )Nz %s not less than or equal to %sr- rD r r r assertLessEqual s zTestCase.assertLessEqualc C s2 ||ks.dt |t |f }| | || d S )Nz%s not greater than %sr- rD r r r
assertGreater s zTestCase.assertGreaterc C s2 ||ks.dt |t |f }| | || d S )Nz"%s not greater than or equal to %sr- rD r r r assertGreaterEqual s zTestCase.assertGreaterEqualc C s, |d ur(dt |f }| | || d S )Nz%s is not Noner- r r) rX rZ r r r assertIsNone s zTestCase.assertIsNonec C s" |d u r d}| | || d S )Nzunexpectedly None)r rW rK r r r assertIsNotNone s zTestCase.assertIsNotNonec C s0 t ||s,dt||f }| | || d S )Nz%s is not an instance of %rr; r r rW r r) r rX rZ r r r r6 s
zTestCase.assertIsInstancec C s0 t ||r,dt||f }| | || d S )Nz%s is an instance of %rrN rO r r r assertNotIsInstance s
z TestCase.assertNotIsInstancec O s t || |}|d||S )NassertRaisesRegexr )r r r_ r- r. r r r r rQ s
zTestCase.assertRaisesRegexc O s t || |}|d||S )NassertWarnsRegexr )r r r_ r- r. r r r r rR s zTestCase.assertWarnsRegexc C sJ t |ttfrt|}||sFd|j|f }| ||}| |d S )Nz&Regex didn't match: %r not found in %r) r; r bytesr] r^ rr rs rW rY )r textr_ rX rZ r r r assertRegex s
zTestCase.assertRegexc C s` t |ttfrt|}||}|r\d|| | |j|f }| ||}|
|d S )Nz"Regex matched: %r matches %r in %r) r; r rS r] r^ rr startendrs rW rY )r rT Zunexpected_regexrX matchrZ r r r assertNotRegex# s
zTestCase.assertNotRegexc s fdd}|S )Nc s$ t d jtd | i |S )NzPlease use {0} instead.r )r r rm r r8 r7
original_funcr r deprecated_func3 s
z,TestCase._deprecate..deprecated_funcr )r[ r\ r rZ r
_deprecate2 s zTestCase._deprecate)r )N)N)N)N)NN)N)N)N)NNN)NNN)NN)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)cr r r
AssertionErrorrY r r& rC Z_classSetupFailedr r r r rz r r r r r r r r r r r r r r r& r' _subtest_msg_sentinelr r r r r r r r r r r r r r r r r rW r r r r r r
r r r r% r! r r r r1 r2 r4 r5 r r= rA r rG rH rI rJ rL rM r6 rP rQ rR rU rY r] ZfailUnlessEqualZ assertEqualsZ failIfEqualZassertNotEqualsZfailUnlessAlmostEqualZassertAlmostEqualsZfailIfAlmostEqualZassertNotAlmostEqualsZ
failUnlessZassert_ZfailUnlessRaisesZfailIfZassertRaisesRegexpZassertRegexpMatchesZassertNotRegexpMatchesr r r r r 5 s !
C
!
-
#
c
+
r c sf e Zd Zd fdd Zdd Zdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z Z
S )FunctionTestCaseNc s* t t| || _|| _|| _|| _d S r( )superr` r
_setUpFunc
_tearDownFunc _testFunc _description)r ZtestFuncr r descriptionr r r r Q s
zFunctionTestCase.__init__c C s | j d ur| d S r( )rb rj r r r r X s
zFunctionTestCase.setUpc C s | j d ur| d S r( )rc rj r r r r \ s
zFunctionTestCase.tearDownc C s | d S r( )rd rj r r r r ` s zFunctionTestCase.runTestc C s | j jS r( )rd r rj r r r r c s zFunctionTestCase.idc C s@ t || jstS | j|jko>| j|jko>| j|jko>| j|jkS r( )r; r r rb rc rd re r r r r r f s
zFunctionTestCase.__eq__c C s t t| | j| j| j| jfS r( )r r< rb rc rd re rj r r r r o s zFunctionTestCase.__hash__c C s dt | j| jjf S r )r r rd r rj r r r r s s
zFunctionTestCase.__str__c C s dt | j| jf S )Nz <%s tec=%s>)r r rd rj r r r r w s
zFunctionTestCase.__repr__c C s2 | j d ur| j S | jj}|r.|dd p0d S r )re rd r r r r r r r r { s
z!FunctionTestCase.shortDescription)NNN)r r r
r r r r r r r r r r
__classcell__r r rg r r` H s r` c sD e Zd Z fddZdd Zdd Zdd Zd d
Zd d Z Z S )
r c s( t || _|| _|| _|j| _d S r( )ra r _messager" r rY )r r" r r rg r r r s
z_SubTest.__init__c C s t dd S )Nz subtests cannot be run directly)r rj r r r r s z_SubTest.runTestc C s^ g }| j tur |d| j | jrPddd | j D }|d| d|p\dS )Nz[{}]z, c s s | ]\}}d ||V qdS )z{}={!r}N)rm )rM r r r r r rP s z+_SubTest._subDescription..z({}) z ())ri r_ r rm r r r9 )r partsZ params_descr r r _subDescription s
z_SubTest._subDescriptionc C s d | j | S Nz{} {})rm r" r rl rj r r r r s z _SubTest.idc C s
| j S r( )r" r rj r r r r s z_SubTest.shortDescriptionc C s d | j| S rm )rm r" rl rj r r r r s z_SubTest.__str__)
r r r
r r rl r r r rh r r rg r r s r )-r r= r r r] r r? r& rn rD rC r utilr r r r r Z
__unittestobjectr_ r' r2 r r r r r* r+ r/ r5 rF rI rJ r rL rV r\ ri r} ChainMapr r r` r r r r r sV ( *'5
: