B ِ<[J@sndZddlZddddddd d gZGd ddeZd dZd dZddZddZdddZ dd Z dd Z dS)z8Tools not exempt from being descended into in tracebacksNmake_decoratorraises set_tracetimed with_setup TimeExpiredistestnottestc@s eZdZdS)rN)__name__ __module__ __qualname__r r 4lib/python3.7/site-packages/nose/tools/nontrivial.pyr scsfdd}|S)z Wraps a test decorator so as to properly replicate metadata of the decorated function, including nose's additional stuff (namely, setup and teardown). csntdrj}nj}j|_j|_j|_t|dsDjj|_y ||_Wnt k rh||_YnX|S)Ncompat_func_namecompat_co_firstlineno) hasattrrr __dict____doc__r __code__co_firstlinenor TypeError)newfuncname)funcr rdecorates     z make_decorator..decorater )rrr )rrrs cs&dddDfdd}|S)aTest must raise one of expected exceptions to pass. Example use:: @raises(TypeError, ValueError) def test_raises_type_error(): raise TypeError("This test passes") @raises(Exception) def test_that_fails_by_passing(): pass If you want to test many assertions about exceptions in a single test, you may want to use `assert_raises` instead. z or cSsg|] }|jqSr )r ).0er r r 7szraises..cs(jfdd}t|}|S)NcsHy||Wn k r"Yn"YnXdf}t|dS)Nz%s() did not raise %s)AssertionError)argkwmessage) exceptionsrrvalidr rr:s z)raises..decorate..newfunc)r r)rr)r"r#)rrrr8s zraises..decorate)join)r"rr )r"r#rr'scCs6ddl}ddl}|j}|j|_||jdS)zCall pdb.set_trace in the calling frame, first restoring sys.stdout to the real output stream. Note that sys.stdout is NOT reset to whatever it was before the call once pdb is done! rN)pdbsysstdout __stdout__ZPdbr _getframef_back)r%r&r'r r rrIs csfdd}|S)zTest must finish within specified time limit to pass. Example use:: @timed(.1) def test_that_fails(): time.sleep(.2) csfdd}t|}|S)Ncs6t}||}t}||kr2td|S)NzTime limit (%s) exceeded)timer)rr startresultend)rlimitr rr_s    z(timed..decorate..newfunc)r)rr)r/)rrr^s ztimed..decorater )r/rr )r/rrUs cCs||fdd}|S)aDecorator to add setup and/or teardown methods to a test function:: @with_setup(setup, teardown) def test_something(): " ... " Note that `with_setup` is useful *only* for test functions, not for test methods or inside of TestCase subclasses. csdr0t|dr*|jfdd}||_n|_r`t|drZ|jfdd}||_n|_|S)NsetupcsdS)Nr r )_old_sr0r r_sysz(with_setup..decorate.._steardowncsdS)Nr r )_old_tr3r r_tsz(with_setup..decorate.._t)rr0r3)rr0r3r2r5r )r1r4r0r3rrus  zwith_setup..decorater )r0r3rr r rrks cCs d|_|S)z5Decorator to mark a function or method as a test T)__test__)rr r rrscCs d|_|S)z;Decorator to mark a function or method as *not* a test F)r6)rr r rr s)NN) rr+__all__rrrrrrrrr r r r rs  "  !