o 1&i@s|ddlZddlZddlZddlmZmZmZddlm Z ddl m Z m Z ddlmZGddde Zedkrsz!TestEvent.setUp..)rlenev _registered _TestEvent__registered_listenersselfr r r setUpszTestEvent.setUpcCs|ttj|jdSr ) assertEqualrrrrrr r r tearDownszTestEvent.tearDowncCsbtdd}td }|dWdn1swY||tj|t|jddS)NcS||Sr r xr r r fooz.TestEvent.test_recording_listener..foo numba:compile)rrinstall_recorderassertIsInstanceZRecordingListenerassertGreaterEqualrbuffer)rrrecr r r test_recording_listeners   z!TestEvent.test_recording_listenercCLtdd}|d||jd}|dd}||t||ddS)NcSrr r rr r r r%rz/TestEvent.test_compiler_lock_event..foorrtimers compiler_lockr get_metadata signaturesr float assertGreaterrrmdZ lock_durationr r r test_compiler_lock_event$   z"TestEvent.test_compiler_lock_eventcCr%)NcSrr r rr r r r0rz+TestEvent.test_llvm_lock_event..foorrr& llvm_lockr(r-r r r test_llvm_lock_event/r0zTestEvent.test_llvm_lock_eventcCstdd}td }|dWdn1swY|t|jd|jD]7\}}|j}||dt||dt||dt||d t ||d t||d tq+dS) NcSrr r rr r r r;rz*TestEvent.test_run_pass_event..fooznumba:run_passrrnamequalnamemoduleflagsargs return_type) rrrr,rr"datar strr)rrZrecorder_rr9r r r test_run_pass_event:s   zTestEvent.test_run_pass_eventcsr|Gfdddtj}tdd|}td| }dWdn1s,wY|||dS)Nc(eZdZfddZfddZdS)z3TestEvent.test_install_listener..MyListenercs`|jtjj|jd|jd|jd}||jd|j dSNr dispatcherr7) rstatusr EventStatusSTARTkindassertIsr9 assertNotIn overloadsrrr?rutr r on_startQs   z.MyListener.on_startcN|jtjj|jd|jd}||jd|j dSr> rr@rrAZENDrCr9rDZassertInrFrGrHr r on_endZs   z:TestEvent.test_install_listener..MyListener.on_endN__name__ __module__ __qualname__rJrMr rHr r MyListenerP rRcS|Sr r rr r r rbz,TestEvent.test_install_listener..foorr)rListenerrZinstall_listenerrD)rrRlisteneryieldedr rHr test_install_listenerMs  zTestEvent.test_install_listenercsR|Gfdddtj}tdd|}td|dtd|dS)Ncr=)z2TestEvent.test_global_register..MyListenercrKr>) rr@rrArBrCr9rDrErFrGrHr r rJq   z;TestEvent.test_global_register..MyListener.on_startcrKr>rLrGrHr r rMzrZz9TestEvent.test_global_register..MyListener.on_endNrNr rHr r rRprSrRcSrTr r rr r r rrUz+TestEvent.test_global_register..foorr)rrVrregister unregister)rrRrWr rHr test_global_registerms  zTestEvent.test_global_registercCstdddd}td }|Wdn1swY|t|jd|j|jd}|j\}|j|jd}| |j dd t | |j dd t dS) NT)ZforceobjcSs$td}tdD]}||7}q |S)Nr )objectrange)cir r r rs   z-TestEvent.test_lifted_dispatcher..foorrr&r'r1) rrrr!rr"rFr*Zliftedr metadatar+)rrr#ZcresZldispZ lifted_cresr r r test_lifted_dispatchers"    z TestEvent.test_lifted_dispatchercsttjtfddtfdd}tdd |dWdn1s)wYdd }||d }|d }||d }|d }||d |d ||d |d |||d |||d ||d |d ||d |d dS)NcsHd}tD]}|d|ivr|d7}q|t|d7}qt||fS)Nr1r)rr)rnpsqrtsin)raccrb)ar r bars    z-TestEvent.test_timing_properties..barcst||fSr )rgZzerosr)rlr r rsz-TestEvent.test_timing_properties..fooZLLVM_PASS_TIMINGSTrcSs||jd}||S)Nr)r)r*)fnpropr.r r r get_timerssz4TestEvent.test_timing_properties..get_timersr&Zllvm_pass_timingsr1r')tuplestringascii_lowercaserrZ assertLessZget_total_time)rrroZ foo_timersZ bar_timersZfoo_llvm_timerZbar_llvm_timerr )rkrlr test_timing_propertiess>              z TestEvent.test_timing_propertiesN) rOrPrQrrr$r/r2r<rYr]rersr r r r r s     r __main__)ZunittestrqnumpyrgZnumbarrrZ numba.corerrZnumba.tests.supportrrZnumba.core.utilsrr rOmainr r r r s  N