B u9aÔã@sØdZddlZddlZddlZddlZddlmZddlmZddl m Z mZmZddl mZddlZddlmZmZGdd „d ejƒZd d„Zedd „ƒZdd„Zdd„ZiZded<ded<ded<edkrÔeƒdS)z"Test suite for the profile module.éN)Úunified_diff)ÚStringIO)ÚTESTFNÚrun_unittestÚunlink)Úcontextmanager)ÚtestfuncÚtimerc@s`eZdZejZeZdddgZdZdd„Z dd„Z ed d „ƒZdd„Z d d„Zdd„Zdd„ZdS)ÚProfileTestÚprint_statsÚ print_callersÚ print_calleesz:0(max)cCsttƒdS)N)rr)Úself©rú"/usr/lib/python3.7/test_profile.pyÚtearDownszProfileTest.tearDowncCstS)N)Ú_ProfileOutput)rrrrÚget_expected_outputszProfileTest.get_expected_outputcsºg}| td¡}tƒ}| dtƒtƒ¡| tƒ|¡x||jD]r}tƒ}tj ||d}| ¡ d¡t||ƒƒ| ¡ ¡}tj dd¡d‰‡fdd„|Dƒ}| d |¡¡q@W|S) Ngü©ñÒMbP?z testfunc())ÚstreamZstdnameÚ.écsg|]}ˆ|kr| ¡‘qSr)Úrstrip)Ú.0Úline)Úmod_namerrú .sz,ProfileTest.do_profiling..Ú )Ú profilerclassr ÚrunctxÚglobalsÚlocalsÚappendÚmethodnamesrÚpstatsÚStatsZ strip_dirsZ sort_statsÚgetattrÚgetvalueÚ splitlinesrÚ __module__ÚrsplitÚjoin)ÚclsÚresultsÚprofZstart_timerZ methodnameÚsÚstatsÚoutputr)rrÚdo_profilingszProfileTest.do_profilingcCs¬| ¡}| ¡}| |dd¡g}xnt|jƒD]`\}}||}||d}||kr0| d|›d|jj›d¡| t | d¡| d¡dd ¡q0W|r¨| d |¡¡dS) Nrièrz Stats.z output for z does not fit expectation:rÚ)Zlineterm) r1rZassertEqualÚ enumerater"r!rÚ__name__ÚextendrÚsplitÚfailr*)rr,Zexpectedr7ÚiÚmethodÚaÚbrrrÚ test_cprofile2szProfileTest.test_cprofilec Cs|ddddddg}xf|D]^}tƒ}| td¡}| |tƒtƒ¡tj||d}| ¡| ¡}| |j|d ||¡¡qWdS) Nzmax([0])zmax([0], key=int)zmax([0], **dict(key=int))zmax(*([0],))zmax(*([0],), key=int)zmax(*([0],), **dict(key=int))gü©ñÒMbP?)rz&Profiling {0!r} didn't report max: {1}) rrr rrr r#r$rr&ZassertInÚexpected_max_outputÚformat)rZstmtsZstmtr.r-r/ZresrrrÚtest_calling_conventionsCs z$ProfileTest.test_calling_conventionsc CsDtƒ|j d¡WdQRX|jjdtd| tj t¡¡dS)Nzint('1'))Úfilename)ÚsilentÚprofilermoduleÚrunrÚ assertTrueÚosÚpathÚexists)rrrrÚtest_runXszProfileTest.test_runc CsTtƒ|j dtƒtƒ¡WdQRX|jjdtƒtƒtd| tj t¡¡dS)Nz testfunc())r@) rArBrrr rrDrErFrG)rrrrÚtest_runctx^s zProfileTest.test_runctxN)r4r(Ú__qualname__ÚprofileZProfilerrBr"r=rrÚclassmethodr1r<r?rHrIrrrrr s r c CsÆ| d¡}td|ƒ| ¡}g}t|dƒ*}x"|D]}| |¡| d¡r4Pq4WWdQRXt|dƒV}| |¡| d¡x0t|j ƒD]"\}}| d|||df¡qˆW| d ¡WdQRXdS) NÚcozRegenerating %s...Úrz#--cutÚwz_ProfileOutput = {} z _ProfileOutput[%r] = """\ %s""" rz' if __name__ == "__main__": main() ) rÚprintr1Úopenr!Ú startswithÚ writelinesÚwriter3r")r@r+r,ZnewfileÚfrr8r9rrrÚregenerate_expected_outputfs rVccs&tj}ztƒt_dVWd|t_XdS)N)ÚsysÚstdoutr)rXrrrrAzs rAcCsttƒdS)N)rr rrrrÚ test_mainƒsrYcCs dtjkrtƒn tttƒdS)Nz-r)rWÚargvrYrVÚ__file__r rrrrÚmain†s r\aƒ 28 27.972 0.999 27.972 0.999 profilee.py:110(__getattr__) 1 269.996 269.996 999.769 999.769 profilee.py:25(testfunc) 23/3 149.937 6.519 169.917 56.639 profilee.py:35(factorial) 20 19.980 0.999 19.980 0.999 profilee.py:48(mul) 2 39.986 19.993 599.830 299.915 profilee.py:55(helper) 4 115.984 28.996 119.964 29.991 profilee.py:73(helper1) 2 -0.006 -0.003 139.946 69.973 profilee.py:84(helper2_indirect) 8 311.976 38.997 399.912 49.989 profilee.py:88(helper2) 8 63.976 7.997 79.960 9.995 profilee.py:98(subhelper)raä:0(append) <- profilee.py:73(helper1)(4) 119.964 :0(exc_info) <- profilee.py:73(helper1)(4) 119.964 :0(hasattr) <- profilee.py:73(helper1)(4) 119.964 profilee.py:88(helper2)(8) 399.912 profilee.py:110(__getattr__) <- :0(hasattr)(12) 11.964 profilee.py:98(subhelper)(16) 79.960 profilee.py:25(testfunc) <- :1()(1) 999.767 profilee.py:35(factorial) <- profilee.py:25(testfunc)(1) 999.769 profilee.py:35(factorial)(20) 169.917 profilee.py:84(helper2_indirect)(2) 139.946 profilee.py:48(mul) <- profilee.py:35(factorial)(20) 169.917 profilee.py:55(helper) <- profilee.py:25(testfunc)(2) 999.769 profilee.py:73(helper1) <- profilee.py:55(helper)(4) 599.830 profilee.py:84(helper2_indirect) <- profilee.py:55(helper)(2) 599.830 profilee.py:88(helper2) <- profilee.py:55(helper)(6) 599.830 profilee.py:84(helper2_indirect)(2) 139.946 profilee.py:98(subhelper) <- profilee.py:88(helper2)(8) 399.912ra’:0(hasattr) -> profilee.py:110(__getattr__)(12) 27.972 :1() -> profilee.py:25(testfunc)(1) 999.769 profilee.py:110(__getattr__) -> profilee.py:25(testfunc) -> profilee.py:35(factorial)(1) 169.917 profilee.py:55(helper)(2) 599.830 profilee.py:35(factorial) -> profilee.py:35(factorial)(20) 169.917 profilee.py:48(mul)(20) 19.980 profilee.py:48(mul) -> profilee.py:55(helper) -> profilee.py:73(helper1)(4) 119.964 profilee.py:84(helper2_indirect)(2) 139.946 profilee.py:88(helper2)(6) 399.912 profilee.py:73(helper1) -> :0(append)(4) -0.004 profilee.py:84(helper2_indirect) -> profilee.py:35(factorial)(2) 169.917 profilee.py:88(helper2)(2) 399.912 profilee.py:88(helper2) -> :0(hasattr)(8) 11.964 profilee.py:98(subhelper)(8) 79.960 profilee.py:98(subhelper) -> profilee.py:110(__getattr__)(16) 27.972r Ú__main__)Ú__doc__rWr#ZunittestrEZdifflibrÚiorZtest.supportrrrÚ contextlibrrKZ test.profileerr ZTestCaser rVrArYr\rr4rrrrÚs*V