o 8ήc<@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/kre8dSdS)0N) takewhile)njit)CFGraphLoop) compile_extracompile_isolatedFlags CompilerBaseDefaultPassBuilder) PreserveIR)IRLegalization)typesir)TestCaseoverride_config SerialMixinTc@s$eZdZddZddZddZdS)_DummycCs||_||_||dSN)recordername _add_dummy)selfrrrD/tmp/pip-target-vg8gfxp4/lib/python/numba/tests/test_obj_lifetime.py__init__sz_Dummy.__init__cCs&t|tsJt|jd|j|jfS)Nz%s + %s) isinstancerrr)rotherrrr__add__"sz_Dummy.__add__cCst|jd|jS)Nziter(%s))_DummyIteratorrrrrrr__iter__&sz_Dummy.__iter__N)__name__ __module__ __qualname__rrr rrrrrs rc@seZdZdZddZeZdS)rrcCs4|jdkrt|jd7_t|jd|j|jfS)Nz%s#%s)count StopIterationrrrrrrr__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_wrsrrrrr>s  zRefRecorder.__init__cCs t||S)zK Make an object whose deletion will be recorded as *name*. )r)rrrrr make_dummyCs zRefRecorder.make_dummycCst||j}|j|j|<dSr)weakrefref _on_disposalrr0)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.r/appendformat)reventr&rrrmarkOs zRefRecorder.markcCs|j|}|j|dSr)r0popr/r7)rr6rrrrr4Xs zRefRecorder._on_disposalcCsdd|jDS)zC A list of objects which haven't been deleted yet. cSsg|]}|qSrr).0r6rrr asz%RefRecorder.alive..)r0rrrralive\szRefRecorder.alivecCs|jS)z, A list of recorded events. )r/rrrrrecordedcszRefRecorder.recordedN) r!r"r#__doc__rr1r__call__r:r4propertyr>r?rrrrr*7s  r*cCs@|d}|d}|d}||}|d||}|d|S)Nabc--1----2--r:)recrCrDrEdrrrsimple_usecase1ks  rKcCs*|d}|d}|d|}|}d}|S)NrCrDrFrH)rIrCrDxyrrrsimple_usecase2us rNcCsN|d}|d}|d}|}|D] }||}|dq|d||}|S)NrCrDrE--loop bottom-- --loop exit--rH)rIrCrDrErLrMrrrlooping_usecase1~s  rQcCs|d}|d}|d}|D]:}|d||}||}|d|D]}|d||}|dq$|d|rA||}nn |d q|d |d |S) NrCrDcum--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--rH)rIrCrDrRrLzrMrrrlooping_usecase2s(         rWccs"|d}|d}|V|VdSNrCrDrrIrCrDrrrgenerator_usecase1s  rZccs,|d}|d}|D]}|Vq |VdSrXr)rIrCrDrLrrrgenerator_usecase2s  r[c@s eZdZdS)MyErrorN)r!r"r#rrrrr\sr\cCst|rr\)rLrrrdo_raisesr^cCs@|d}|d}|d}|rtd|d}||||dS)NrCrDrJfoorEr^)rIrCrDrJrErrrraising_usecase1s racCs0|d}|d}|r|d}t|||dS)NrCrDrEr`)rIrCrDrErrrraising_usecase2s  rbcCs |d}|d}|rt|dSrXr]rYrrrraising_usecase3s rccCsd}t|D]9}|t|d}t|D]}dS|dkr q|dkr5t|D]}|S|d|d|dq|ddS)zN This test reveal a bug that there is a del on uninitialized variable rFAILEDOK)ranger:str)rInijrrrdel_before_definitions         rnccs. |ddV|d}|r|d q)z^ test to reveal bug of invalid liveness when infinite loop has multiple backedge. TyieldNpbrarH)rIrprrrinf_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 pyobjectforceobj_flags entry_point)rpyfunccrrrrcompileszTestObjLifetime.compileNcCs\t}||}|dur(|| ||Wd|S1s!wY|S|||Sr)r*rz assertRaises)rrxraisesrIcfuncrrrcompile_and_records    z"TestObjLifetime.compile_and_recordcCsPg}|j}t|}|D]}||vr||||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?listr7remove assertEqual)rrIexpectedactualr? remainingrJrrrassertRecordOrders  z!TestObjLifetime.assertRecordOrdercCsJ|t}||j||gd||gd||gddS)N)rCrDrF)rCrErF)rFzb + crG)r~rK assertFalser>rrrIrrr test_simple1s  zTestObjLifetime.test_simple1cCs*|t}||j||gddS)N)rDrFrC)r~rNrr>rrrrr test_simple2 s  zTestObjLifetime.test_simple2cCs:|t}||j||gd||gddS)N)rCrDrPrE) iter(a)#1rO iter(a)#2rO iter(a)#3rOiter(a)rP)r~rQrr>rrrrr test_looping1%s  zTestObjLifetime.test_looping1cCsj|t}||j||ddg||gd||gd||gd||ddgdS)NrCrS)rrTrU)iter(b)z--inner loop exit #1--rz--inner loop exit #2--rz--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~rWrr>rrrrr test_looping21s   zTestObjLifetime.test_looping2cCs ||}t}|| ||}t|||jt|||jWdn1s0wYt}||||}~t ||jWdn1sWwYt}||"||}t|||j~t ||jWddS1swYdSr) rzr*assertRefCountr) assertTruer>rrgccollect)rgenfuncr}rIgenrrrexercise_generatorGs2      "z"TestObjLifetime.exercise_generatorcC|tdSr)rrZrrrrtest_generator1bzTestObjLifetime.test_generator1cCrr)rr[rrrrtest_generator2erzTestObjLifetime.test_generator2cCs |t}||jgddS)N)012)r~rnrr?rrrrtest_del_before_definitionhs z*TestObjLifetime.test_del_before_definitioncCJ|t|jttd}||jWddS1swYdSN)r|)rr^r~rar\rr>rrrr test_raising1l "zTestObjLifetime.test_raising1cCrr)rr^r~rbr\rr>rrrr test_raising2qrzTestObjLifetime.test_raising2cCsJ|t|jttd}||jWddS1swYdSr)rr\r~rcrr>rrrr test_raising3vrzTestObjLifetime.test_raising3cCst|t}t}t||}t|||jgt|||jgt|||jg||jgddS)N)rorprqrorprqro)rzrrr*iterr)rr>r?)rr}rIiteratorrrr test_inf_loop_multiple_back_edge{s  z0TestObjLifetime.test_inf_loop_multiple_back_edger)r!r"r#r@rzr~rrrrrrrrrrrrrrrrrrss"    rsc@seZdZddZddZdS)TestExtendingVariableLifetimescsdd}fdd}|d}gtjfdtjtjtjtjtjtjtj}||||d}gtjfdtjtjfdtj}|||dS) NcSsrGdddt}t|ddd}td|||j|jd}|jd}Wd|S1s2wY|S) Nc@eZdZddZdS)z`TestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompilercS&t|j}|tt||gSrr define_nopython_pipelinestateadd_pass_afterr r finalizerpmrrrdefine_pipelines  zqTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompiler.define_pipelinesNr!r"r#rrrrrIRPreservingCompiler r)pipeline_classcSd}d}||}||}|SN rrCrDrErJrrrr_ zOTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..fooEXTEND_VARIABLE_LIFETIMESr preserved_ir)r rr overloads signaturesmetadata)extend_lifetimesrr_cresfunc_irrrrget_irs    zBTestExtendingVariableLifetimes.test_lifetime_basic..get_ircsJt|jdtt|j}t||jD] \}}||qdS)Nr%) rlenblocksr)rvalueszipbodyassertIsInstance)rexpectblk expect_classgot_stmtrrrchecks zATestExtendingVariableLifetimes.test_lifetime_basic..checkFr$T)rAssignDelReturn)rrrdel_after_use_irrdel_at_block_end_irrrrtest_lifetime_basics  & (z2TestExtendingVariableLifetimes.test_lifetime_basiccCs`dd}|dd}|ddd}|ddd}dd}|||||||||dS) Nc[sNGdddt}tdd|i|dd}||j|jd}|jd}|S) Nc@r)zfTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompilercSrrrrrrrrrzwTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..IRPreservingCompiler.define_pipelinesNrrrrrrrrrcSrrrrrrrr_rzUTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..foorrr)r rrrr)optionsrr_rrrrrrs  zHTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_irT)debug)r_dbg_extend_lifetimesFcSsh|j\}dd|jD}tttj|}t|}tttj||d}|t|}|t|dkS)NcSsg|]}t|tjqSr)rrr)r<stmtrrrr=szoTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_end..r%) rrrrroperatornot_rtruth)firr inst_is_delnot_delsbeginall_delsendrrris_del_grouped_at_the_ends  z[TestExtendingVariableLifetimes.test_dbg_extend_lifetimes..is_del_grouped_at_the_end)rr)rrir_debug ir_debug_extir_debug_no_extrrrrtest_dbg_extend_lifetimess   z8TestExtendingVariableLifetimes.test_dbg_extend_lifetimesN)r!r"r#rrrrrrrs 2r__main__)9r+sysr2rr itertoolsrunittestnumbarnumba.core.controlflowrrnumba.core.compilerrrrr r numba.core.untyped_passesr numba.core.typed_passesr numba.corer rnumba.tests.supportrrrenable_pyobj_flagsenable_pyobjectrvforce_pyobjectno_pyobj_flagsobjectrrr*rKrNrQrWrZr[ RuntimeErrorr\r^rarbrcrnrrrsrr!mainrrrrsR     4    m