o gJ@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 S/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/nose/tools/nontrivial.pyr srcfdd}|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). csltdr j}nj}j|_j|_j|_t|ds"jj|_z||_W|St y5||_Y|Sw)Ncompat_func_namecompat_co_firstlineno) hasattrrr __dict____doc__r __code__co_firstlinenor TypeError)newfuncnamefuncr rdecorates    z make_decorator..decorater )rrr rrrs rcs&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)NcsBz |i|Wn yYdSdf}t|)Nz%s() did not raise %s)AssertionError)argkwmessage) exceptionsrrvalidr rr:s  z)raises..decorate..newfunc)r rrrr$r%)rrrr8s zraises..decorate)join)r$rr r'rr'srcCs6ddl}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__Pdbr _getframef_back)r)r*r+r r rrIs rcr)zTest must finish within specified time limit to pass. Example use:: @timed(.1) def test_that_fails(): time.sleep(.2) csfdd}t|}|S)Ncs:t}|i|}t}||krtd|S)NzTime limit (%s) exceeded)timer)r!r"startresultend)rlimitr rr_s   z(timed..decorate..newfunc)rr&r4rrr^s ztimed..decorater )r4rr r5rrUs  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. csfrt|dr|jfdd}||_n|_r1t|dr.|jfdd}||_|S|_|S)NsetupcsdSNr r )_old_sr6r r_sy z(with_setup..decorate.._steardowncsdSr7r r )_old_tr;r r_tr:z(with_setup..decorate.._t)rr6r;)rr6r;r9r=r )r8r<r6r;rrus  zwith_setup..decorater )r6r;rr r rrks rcC d|_|S)z5Decorator to mark a function or method as a test T__test__rr r rrrcCr>)z;Decorator to mark a function or method as *not* a test Fr?rr r rr rAr )NN) rr0__all__r rrrrrrrr r r r rs" !