U ,d<@sddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZddlmZddlmZddlmZmZdd lmZmZmZeZd e_eZ d e _!eZ"Gd d d e#Z$Gd dde$Z%Gddde#Z&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,Gddde-Z.dd Z/d!d"Z0d#d$Z1d%d&Z2d'd(Z3d)d*Z4Gd+d,d,eZ5Gd-d.d.eeZ6e7d/kre8dS)0N) takewhile)njit)CFGraphLoop) compile_extracompile_isolatedFlags CompilerBaseDefaultPassBuilder) PreserveIR)IRLegalization)typesir)TestCaseoverride_config SerialMixinTc@s$eZdZddZddZddZdS)_DummycCs||_||_||dSN)recordername _add_dummy)selfrrrA/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_obj_lifetime.py__init__sz_Dummy.__init__cCs&t|tstt|jd|j|jfS)Nz%s + %s) isinstancerAssertionErrorrr)rotherrrr__add__"sz_Dummy.__add__cCst|jd|jS)Nziter(%s))_DummyIteratorrrrrrr__iter__&sz_Dummy.__iter__N)__name__ __module__ __qualname__rrr!rrrrrsrc@seZdZdZddZeZdS)rrcCs4|jdkrt|jd7_t|jd|j|jfS)Nz%s#%s)count StopIterationrrrr rrr__next__.s z_DummyIterator.__next__N)r"r#r$r'r)nextrrrrr*src@sTeZdZdZddZddZddZeZdd Zd d Z e d d Z e ddZ dS) RefRecorderz An object which records events when instances created through it are deleted. Custom events can also be recorded to aid in diagnosis. cCstt|_g|_i|_dSr) collections defaultdictint_counts_events_wrsr rrrr>s zRefRecorder.__init__cCs t||S)zK Make an object whose deletion will be recorded as *name*. )r)rrrrr make_dummyCszRefRecorder.make_dummycCst||j}|j|j|<dSr)weakrefref _on_disposalrr1)rdummywrrrrrIszRefRecorder._add_dummycCs0|j|d}||j|<|j|j|ddS)zr Manually append *event* to the recorded events. *event* can be formatted using format(). r&)r'N)r/r0appendformat)reventr'rrrmarkOs zRefRecorder.markcCs|j|}|j|dSr)r1popr0r8)rr7rrrrr5Xs zRefRecorder._on_disposalcCsdd|jDS)zC A list of objects which haven't been deleted yet. cSsg|] }|qSrr).0r7rrr asz%RefRecorder.alive..)r1r rrralive\szRefRecorder.alivecCs|jS)z, A list of recorded events. )r0r rrrrecordedcszRefRecorder.recordedN) r"r#r$__doc__rr2r__call__r;r5propertyr?r@rrrrr+7s  r+cCs@|d}|d}|d}||}|d||}|d|S)Nabc--1----2--r;)recrDrErFdrrrsimple_usecase1ks  rLcCs*|d}|d}|d|}|}d}|S)NrDrErGrI)rJrDrExyrrrsimple_usecase2us rOcCsN|d}|d}|d}|}|D]}||}|dq |d||}|S)NrDrErF--loop bottom-- --loop exit--rI)rJrDrErFrMrNrrrlooping_usecase1~s  rRcCs|d}|d}|d}|D]t}|d||}||}|d|D] }|d||}|dqH|d|r||}nq|d q|d |d |S) NrDrEcum--outer loop top--z--inner loop entry #{count}--z--inner loop top #{count}--z--inner loop bottom #{count}--z--inner loop exit #{count}--z--outer loop bottom #{count}----outer loop else----outer loop exit--rI)rJrDrErSrMzrNrrrlooping_usecase2s(         rXccs |d}|d}|V|VdSNrDrErrJrDrErrrgenerator_usecase1sr[ccs*|d}|d}|D] }|Vq|VdSrYr)rJrDrErMrrrgenerator_usecase2s r\c@s eZdZdS)MyErrorN)r"r#r$rrrrr]sr]cCs t|dSrr])rMrrrdo_raisesr_cCs@|d}|d}|d}|r4td|d}||||dS)NrDrErKfoorFr_)rJrDrErKrFrrrraising_usecase1srbcCs0|d}|d}|r$|d}t|||dS)NrDrErFra)rJrDrErFrrrraising_usecase2s rccCs |d}|d}|rt|dSrYr^rZrrrraising_usecase3srdcCsd}t|D]t}|t|d}t|D] }dS|dkrBq n*|dkrlt|D]}|S|d|d|dq |ddS)zN This test reveal a bug that there is a del on uninitialized variable rZFAILEDOK)ranger;str)rJnijrrrdel_before_definitions         rnccs,|ddV|d}|r|dqdS)z^ test to reveal bug of invalid liveness when infinite loop has multiple backedge. yieldNpbrarI)rJrprrrinf_loop_multiple_back_edges   rrc@seZdZdZddZd!ddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd ZdS)"TestObjLifetimezH Test lifetime of Python objects inside jit-compiled functions. cCst|tjftd}|jS)N)flags)rr Zpyobjectforceobj_flagsZ entry_point)rpyfuncZcrrrrcompileszTestObjLifetime.compileNc CsDt}||}|dk r8||||W5QRXn|||Sr)r+rw assertRaises)rrvraisesrJcfuncrrrcompile_and_records  z"TestObjLifetime.compile_and_recordcCsPg}|j}t|}|D] }||kr||||q|||d|fdS)zk Check that the *expected* markers occur in that order in *rec*'s recorded events. z'the full list of recorded events is: %rN)r@listr8remove assertEqual)rrJexpectedactualr@ remainingrKrrrassertRecordOrders  z!TestObjLifetime.assertRecordOrdercCsP|t}||j||dddg||dddg||dddgdS)NrDrErGrFzb + crH)r{rL assertFalser?rrrJrrr test_simple1s   zTestObjLifetime.test_simple1cCs,|t}||j||dddgdS)NrErGrD)r{rOrr?rrrrr test_simple2 s  zTestObjLifetime.test_simple2c CsJ|t}||j||ddddg||ddddddd dgdS) NrDrErQrF iter(a)#1rP iter(a)#2 iter(a)#3iter(a))r{rRrr?rrrrr test_looping1%s   zTestObjLifetime.test_looping1c Cs||t}||j||ddg||dddg||dddddd g||d d d d ddg||ddgdS)NrDrTrrUrVziter(b)z--inner loop exit #1--z--inner loop exit #2--z--inner loop exit #3--rz--inner loop entry #1--rz--inner loop entry #2--rz--inner loop entry #3--ziter(a)#1 + iter(a)#1z--outer loop bottom #1--)r{rXrr?rrrrr test_looping21s*     zTestObjLifetime.test_looping2c Cs||}t}||6||}t|||jt|||jW5QRXt}||$||}~t ||jW5QRXt}||8||}t|||j~t ||jW5QRXdSr) rwr+assertRefCountr* assertTruer?r|rgcZcollect)rZgenfuncrzrJgenrrrexercise_generatorGs,      z"TestObjLifetime.exercise_generatorcCs|tdSr)rr[r rrrtest_generator1bszTestObjLifetime.test_generator1cCs|tdSr)rr\r rrrtest_generator2eszTestObjLifetime.test_generator2cCs"|t}||jdddgdS)N012)r{rnr~r@rrrrtest_del_before_definitionhs z*TestObjLifetime.test_del_before_definitionc Cs4|t |jttd}||jW5QRXdSN)ry)rr_r{rbr]rr?rrrr test_raising1ls zTestObjLifetime.test_raising1c Cs4|t |jttd}||jW5QRXdSr)rr_r{rcr]rr?rrrr test_raising2qs zTestObjLifetime.test_raising2c Cs4|t |jttd}||jW5QRXdSr)rr]r{rdrr?rrrr test_raising3vs zTestObjLifetime.test_raising3c Cs~|t}t}t||}t|||jgt|||jgt|||jg||jdddddddgdS)Nrorprq)rwrrr+iterr*r~r?r@)rrzrJiteratorrrr test_inf_loop_multiple_back_edge{s  z0TestObjLifetime.test_inf_loop_multiple_back_edge)N)r"r#r$rArwr{rrrrrrrrrrrrrrrrrrss   rsc@seZdZddZddZdS)TestExtendingVariableLifetimescsdd}fdd}|d}tjfdtjtjtjtjtjtjtjf}||||d}tjfdtjftjfdtjf}|||dS) Nc Ss\Gdddt}t|ddd}td|&||j|jd}|jd}W5QRX|S) Nc@seZdZddZdS)z`TestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompilercSs&t|j}|tt||gSrr Zdefine_nopython_pipelinestateZadd_pass_afterr r finalizerZpmrrrdefine_pipeliness  zqTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompiler.define_pipelinesNr"r#r$rrrrrIRPreservingCompilersr)pipeline_classcSsd}d}||}||}|SN rrDrErFrKrrrr`s zOTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..fooZEXTEND_VARIABLE_LIFETIMESr preserved_ir)r rr overloads signaturesmetadata)Zextend_lifetimesrr`cresfunc_irrrrget_irs  zBTestExtendingVariableLifetimes.test_lifetime_basic..get_ircsJt|jdtt|j}t||jD]\}}||q0dS)Nr&) r~lenblocksr*rvalueszipbodyZassertIsInstance)rexpectblkZ expect_classZgot_stmtr rrcheckszATestExtendingVariableLifetimes.test_lifetime_basic..checkFr%T)rZAssignDelZReturn)rrrZdel_after_use_irrZdel_at_block_end_irrr rtest_lifetime_basics  $z2TestExtendingVariableLifetimes.test_lifetime_basiccCs`dd}|dd}|ddd}|ddd}dd}|||||||||dS) Nc[sNGdddt}tfd|i|dd}||j|jd}|jd}|S)Nc@seZdZddZdS)zfTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompilercSs&t|j}|tt||gSrrrrrrrs  zwTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompiler.define_pipelinesNrrrrrrsrrcSsd}d}||}||}|Srrrrrrr`s zUTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..foorr)r rrrr)optionsrr`rrrrrrs  zHTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_irT)debug)rZ_dbg_extend_lifetimesFcSsh|j\}dd|jD}tttj|}t|}tttj||d}|t|}|t|dkS)NcSsg|]}t|tjqSr)rrr)r=Zstmtrrrr>szoTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_end..r&) rrrr|roperatornot_rtruth)ZfirrZ inst_is_delZnot_delsbeginZall_delsendrrris_del_grouped_at_the_ends  z[TestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_end)rr)rrZir_debugZ ir_debug_extZir_debug_no_extrrrrtest_dbg_extend_lifetimess   z8TestExtendingVariableLifetimes.test_dbg_extend_lifetimesN)r"r#r$rrrrrrrs2r__main__)9r,sysr3rr itertoolsrZunittestZnumbarZnumba.core.controlflowrrZnumba.core.compilerrrrr r Znumba.core.untyped_passesr Znumba.core.typed_passesr Z numba.corer rZnumba.tests.supportrrrZenable_pyobj_flagsZenable_pyobjectruZforce_pyobjectZno_pyobj_flagsobjectrrr+rLrOrRrXr[r\ RuntimeErrorr]r_rbrcrdrnrrrsrr"mainrrrrsN     4    m