a rc@seZdZddZdS) DirectorycCs|j}|j}|do|dkS)Nbench_z.py)Z purebasenameext startswith)selfpathbrr r r filefilterszDirectory.filefilterN)__name__ __module__ __qualname__rr r r rrsrc@seZdZddZdS)ModulecCs|dp|dS)Nrtimeit_)r)rnamer r rfuncnamefilter"szModule.funcnamefilterN)rrrr"r r r rr src@s eZdZdejefddZdS)TimerpasscCs`||_t|d}t|d}tj||d}||_t|tjd}i}t||||d|_dS)N)stmtsetupexecinner) timertimeitZreindenttemplatesrccompileZdummy_src_namer)r*)rr'r(r+globalsr.coder r r r__init__)s   zTimer.__init__N)rrrr,Z default_timerr0r2r r r rr#'sr#cs$eZdZfddZddZZS)Functioncs"tj|i|d|_d|_dSN)superr2 benchtime benchtitle)rargskw __class__r rr2:szFunction.__init__c Gst|}d|dd}|jdur2|j|_n|d|_t||jd}|j drd}d}t ddD]`}| |}|dkr|d|9}t t |}q|d kr|d9}qp|d|9}t t |}qqpt|||||_n | d|_dS) N r r)r0r r g?g{Gz?)rjoin splitlinesZfunc_docr7stripr#Z func_globalsr!rranger,int_ceilminrepeatr6) rtargetr8r.r+rEnumberrtr r rexecute?s,         zFunction.execute)rrrr2rI __classcell__r r r:rr38s r3cs0eZdZfddZfddZddZZS) BenchSessioncst|dSr4)r5headerrZcolitemsr:r rrLmszBenchSession.headercs$t||jd|dS)Nr<)r5footeroutwriteprint_bench_resultsrMr:r rrNps  zBenchSession.footerc s|jd|jd|jd|jdg}|jD]~\}}t|tr:|j}|dur`d}nD|dkrtttt |d d}nd}dt |t |t |f}| |j||jgq:dgtt }dgtt }|D]~} | d }|\} } t| } z| d \} }Wnty*| d } }Yn0tt| || || <tt||| || <q|D]} | d }|\} } t| } z| d \} }Wnty| d } }Yn0| || } ||| }|rd | |f} nd | |f} d }ttt D]6| kr|| 7}n|d ||d 7}qd || f| d <q^tdD]:tfdd|D}|D]} | || <qvqX|D]} |jdt| qdS)Nz============================== z *** BENCHMARKING RESULTS *** r<z---grz%.*g %srr . z%s %sc3s|]}t|VqdSr4)len)rrrr r rz3BenchSession.print_bench_results..z%s | %s | %s )rOrPZ_memo isinstancerr6rDrB_floorr precisionscalingunitsappendr!r7rVsplitunitn ValueErrormaxrjustljustr@r>rAtuple)rresultsitemoutcomebestZtstrorderZwmZwernuunmetxtwr rWrrQvsl            z BenchSession.print_bench_results)rrrrLrNrQrJr r r:rrKks  rKcCsBddlm}t|_t|_t|_tjj}dd|_tjj |dS)Nr)defaultconftestcSstSr4)rKr r r rrzmain..) Z py.__.testrrrrr3pytestconfigZ_getsessionclassZcmdlinemain)r8rrrvr r rrws  rw)N)__doc__rtZpy.__.test.itemrZpy.__.test.terminal.terminalrmathrrCrrZrr,inspectrr]r\ enumerater`r[ruZcollectrrr#__rgr3rKrwr r r rs     3f