a j@sbdZddlZgdZGdddeZddZdd Zd d Zd d ZdddZ ddZ ddZ dS)z8Tools not exempt from being descended into in tracebacksN)make_decoratorraises set_tracetimed with_setup TimeExpiredistestnottestc@s eZdZdS)rN)__name__ __module__ __qualname__r r ;E:\soccer\ml_env\Lib\site-packages\nose/tools/nontrivial.pyrsrcsfdd}|S)zWraps a test decorator so as to properly replicate metadata of the decorated function, including nose's additional stuff (namely, setup and teardown).csltdrj}nj}j|_j|_j|_t|dsDjj|_z ||_Wnt yf||_Yn0|S)Ncompat_func_namecompat_co_firstlineno) hasattrrr __dict____doc__r __code__co_firstlinenor TypeError)newfuncnamefuncr rdecorates      z make_decorator..decorater )rrr rrr s rcs&dddDfdd}|S)aTest must raise one of expected exceptions to pass. Example: @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 1zraises..cs(jfdd}t|}|S)NcsPz|i|Wn$y$Yn(ty6Yn0df}t|dS)Nz%s() did not raise %s) ExceptionAssertionError)argkwmessage) exceptionsrrvalidr rr6s   z)raises..decorate..newfunc)r rrrr%r&)rrrr3s zraises..decorate)join)r%rr r(rr"srcCs0ddl}ddl}|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)pdbsys __stdout__stdoutZPdbr _getframef_back)r*r+r r rrFsrcsfdd}|S)zTest must finish within specified time limit to pass. Example: @timed(.1) def test_that_fails(): time.sleep(.2) csfdd}t|}|S)Ncs:t}|i|}t}||kr6td|S)NzTime limit (%s) exceeded)timer)r"r#startresultend)rlimitr rrWs   z(timed..decorate..newfunc)rr'r4rrrVs ztimed..decorater )r4rr r5rrPs rcCs||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)NsetupcsdSNr r )_old_sr6r r_sqsz(with_setup..decorate.._steardowncsdSr7r r )_old_tr:r r_t|sz(with_setup..decorate.._t)rr6r:)rr6r:r9r<r )r8r;r6r:rrls  zwith_setup..decorater )r6r:rr r rrcs rcCs d|_|S)z0Decorator to mark a function or method as a testTZ__test__rr r rrsrcCs d|_|S)z6Decorator to mark a function or method as *not* a testFr=rr r rr sr )NN) rr0__all__r!rrrrrrrr r r r rs$  $