a 5êdgâ"ã@sðdZddlZddlZddlZddlZddlmZddlmZddl m Z mZmZm Z mZddlmZddlZddlmZmZddlmZmZGd d „d ejƒZdd„Zed d„ƒZdd„Zdd„ZiZded<ded<ded<edkrìeƒdS)z"Test suite for the profile module.éN)Úunified_diff)ÚStringIO)ÚTESTFNÚrun_unittestÚunlinkÚtemp_dirÚ change_cwd)Úcontextmanager)ÚtestfuncÚtimer)Úassert_python_failureÚassert_python_okc@sneZdZejZeZgd¢ZdZdd„Z dd„Z edd„ƒZd d „Z dd„Zd d„Zdd„Zdd„Zdd„ZdS)ÚProfileTest)Úprint_statsÚ print_callersÚ print_calleesz:0(max)cCsttƒdS©N)rr©Úself©rú-/usr/local/lib/python3.9/test/test_profile.pyÚtearDownszProfileTest.tearDowncCstSr)Ú_ProfileOutputrrrrÚget_expected_outputszProfileTest.get_expected_outputcs¶g}| td¡}tƒ}| dtƒtƒ¡| tƒ|¡|jD]r}tƒ}tj ||d}| ¡ d¡t||ƒƒ| ¡ ¡}tj dd¡d‰‡fdd„|Dƒ}| d |¡¡q>|S) Nçü©ñÒMbP?ú testfunc()©ÚstreamZstdnameÚ.écsg|]}ˆ|vr| ¡‘qSr)Úrstrip)Ú.0Úline©Zmod_namerrÚ /óz,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Úoutputrr#rÚdo_profilings zProfileTest.do_profilingcCs¨| ¡}| ¡}| |dd¡g}t|jƒD]`\}}||}||d}||kr.| d|›d|jj›d¡| t | d¡| d¡dd ¡q.|r¤| d |¡¡dS) Nrièrz Stats.z output for z does not fit expectation:r&Ú)Zlineterm) r;rZassertEqualÚ enumerater,r+r'Ú__name__ÚextendrÚsplitÚfailr4)rr6ZexpectedrAÚiÚmethodÚaÚbrrrÚ test_cprofile3s ÿ ÿ zProfileTest.test_cprofilec Cspgd¢}|D]^}tƒ}| td¡}| |tƒtƒ¡tj||d}| ¡| ¡}| |j|d ||¡¡qdS)N)zmax([0])zmax([0], key=int)zmax([0], **dict(key=int))zmax(*([0],))zmax(*([0],), key=int)zmax(*([0],), **dict(key=int))rrz&Profiling {0!r} didn't report max: {1}) rr'rr(r)r*r-r.rr0ZassertInÚexpected_max_outputÚformat)rZstmtsZstmtr8r7r9ÚresrrrÚtest_calling_conventionsDs ÿz$ProfileTest.test_calling_conventionscCsXtƒ|j d¡Wdƒn1s(0Y|jjdtd| tj t¡¡dS)Nzint('1')©Úfilename)ÚsilentÚprofilermoduleÚrunrÚ assertTrueÚosÚpathÚexistsrrrrÚtest_runYs*zProfileTest.test_runcCshtƒ$|j dtƒtƒ¡Wdƒn1s00Y|jjdtƒtƒtd| tj t¡¡dS)NrrK) rMrNr(r)r*rrPrQrRrSrrrrÚtest_runctx_s2ÿzProfileTest.test_runctxcCs<td|jjdƒtd|jjddƒtd|jjddddƒdS)Nú-mZrandom_module_xyzZtimeitz-nÚ1)rrNr>r rrrrÚtest_run_profile_as_modulefs ÿ ÿz&ProfileTest.test_run_profile_as_modulec Cs¶tƒœ}t|ƒtt d¡tddƒ}| d¡Wdƒn1sF0Ytd|jjdddƒ| tj d¡¡Wdƒn1sŠ0YWdƒn1s¨0YdS)NÚdestzdemo.pyÚwzimport os; os.chdir("dest")rVz-oz out.pstats)rrrQÚmkdirÚopenÚwriter rNr>rPrRrS)rZtmpdirÚfrrrÚ(test_output_file_when_changing_directoryrs (ýz4ProfileTest.test_output_file_when_changing_directoryN)r>r2Ú__qualname__ÚprofileZProfiler'rNr,rGrrÚclassmethodr;rFrJrTrUrXr_rrrrrs rc Csè| d¡}td|ƒ| ¡}g}t|dƒ2}|D]}| |¡| d¡r2qPq2Wdƒn1sd0Yt|dƒ\}| |¡| d¡t|j ƒD]"\}}| d|||df¡q˜| d ¡Wdƒn1sÚ0YdS) NÚcozRegenerating %s...Úrz#--cutrZz_ProfileOutput = {} z _ProfileOutput[%r] = """\ %s""" rz' if __name__ == "__main__": main() ) r Úprintr;r\r+Ú startswithÚ writelinesr]r=r,)rLr5r6Únewfiler^r"rBrCrrrÚregenerate_expected_outputs" $ ÿ riccs,tj}ztƒt_dVW|t_n|t_0dSr)ÚsysÚstdoutr)rkrrrrM•s rMcCsttƒdSr)rrrrrrÚ test_mainžsrlcCs dtjvrtƒn tttƒdS)Nz-r)rjÚargvrlriÚ__file__rrrrrÚmain¡s roaƒ 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__rjr-ZunittestrQZdifflibrÚiorZtest.supportrrrrrÚ contextlibr raZ test.profileer rZtest.support.script_helperrr ZTestCaserrirMrlrorr>rrrrÚs.p