U id<@sddlZddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z m Z ddlmZmZmZmZmZddlmZddlmZddlmZmZdd 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 Z0d!d"Z1d#d$Z2d%d&Z3d'd(Z4d)d*Z5Gd+d,d,eZ6Gd-d.d.eeZ7e8d/kre9dS)0N) takewhile)njitjit)CFGraphLoop) compile_extracompile_isolatedFlags CompilerBaseDefaultPassBuilder) PreserveIR)IRLegalization)typesir)TestCaseoverride_config SerialMixinTc@s$eZdZddZddZddZdS)_DummycCs||_||_||dSN)recordername _add_dummy)selfrrrV/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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)r rcCs4|jdkrt|jd7_t|jd|j|jfS)Nz%s#%s)count StopIterationrrrr!rrr__next__.s z_DummyIterator.__next__N)r#r$r%r(r*nextrrrrr *sr c@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_disposalrr2)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)r0r1appendformat)reventr(rrrmarkOs zRefRecorder.markcCs|j|}|j|dSr)r2popr1r9)rr8rrrrr6Xs zRefRecorder._on_disposalcCsdd|jDS)zC A list of objects which haven't been deleted yet. cSsg|] }|qSrr).0r8rrr asz%RefRecorder.alive..)r2r!rrralive\szRefRecorder.alivecCs|jS)z, A list of recorded events. )r1r!rrrrecordedcszRefRecorder.recordedN) r#r$r%__doc__rr3r__call__r<r6propertyr@rArrrrr,7s  r,cCs@|d}|d}|d}||}|d||}|d|S)Nabc--1----2--r<)recrErFrGdrrrsimple_usecase1ks  rMcCs*|d}|d}|d|}|}d}|S)NrErFrHrJ)rKrErFxyrrrsimple_usecase2us rPcCsN|d}|d}|d}|}|D]}||}|dq |d||}|S)NrErFrG--loop bottom-- --loop exit--rJ)rKrErFrGrNrOrrrlooping_usecase1~s  rScCs|d}|d}|d}|D]t}|d||}||}|d|D] }|d||}|dqH|d|r||}nq|d q|d |d |S) NrErFcum--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--rJ)rKrErFrTrNzrOrrrlooping_usecase2s(         rYccs |d}|d}|V|VdSNrErFrrKrErFrrrgenerator_usecase1sr\ccs*|d}|d}|D] }|Vq|VdSrZr)rKrErFrNrrrgenerator_usecase2s r]c@s eZdZdS)MyErrorN)r#r$r%rrrrr^sr^cCs t|dSrr^)rNrrrdo_raisesr`cCs@|d}|d}|d}|r4td|d}||||dS)NrErFrLfoorGr`)rKrErFrLrGrrrraising_usecase1srccCs0|d}|d}|r$|d}t|||dS)NrErFrGrb)rKrErFrGrrrraising_usecase2s rdcCs |d}|d}|rt|dSrZr_r[rrrraising_usecase3srecCsd}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)rKnijrrrdel_before_definitions         roccs,|ddV|d}|r|dqdS)z^ test to reveal bug of invalid liveness when infinite loop has multiple backedge. yieldNpbrarJ)rKrqrrrinf_loop_multiple_back_edges   rsc@seZdZdZddZd!ddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd ZdS)"TestObjLifetimezH Test lifetime of Python objects inside jit-compiled functions. cCsttjfddd|}|S)NTF)ZforceobjZlooplift)rrZpyobject)rpyfunccfuncrrrcompileszTestObjLifetime.compileNc CsDt}||}|dk r8||||W5QRXn|||Sr)r,rw assertRaises)rruraisesrKrvrrrcompile_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)rAlistr9remove assertEqual)rrKexpectedactualrA remainingrLrrrassertRecordOrder s  z!TestObjLifetime.assertRecordOrdercCsP|t}||j||dddg||dddg||dddgdS)NrErFrHrGzb + crI)rzrM assertFalser@rrrKrrr test_simple1s   zTestObjLifetime.test_simple1cCs,|t}||j||dddgdS)NrFrHrE)rzrPrr@rrrrr test_simple2"s  zTestObjLifetime.test_simple2c CsJ|t}||j||ddddg||ddddddd dgdS) NrErFrRrG iter(a)#1rQ iter(a)#2 iter(a)#3iter(a))rzrSrr@rrrrr test_looping1's   zTestObjLifetime.test_looping1c Cs||t}||j||ddg||dddg||dddddd g||d d d d ddg||ddgdS)NrErUrrVrWziter(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--)rzrYrr@rrrrr test_looping23s*     zTestObjLifetime.test_looping2c Cs||}t}||6||}t|||jt|||jW5QRXt}||$||}~t ||jW5QRXt}||8||}t|||j~t ||jW5QRXdSr) rwr,assertRefCountr+ assertTruer@r{rgcZcollect)rZgenfuncrvrKgenrrrexercise_generatorIs,      z"TestObjLifetime.exercise_generatorcCs|tdSr)rr\r!rrrtest_generator1dszTestObjLifetime.test_generator1cCs|tdSr)rr]r!rrrtest_generator2gszTestObjLifetime.test_generator2cCs"|t}||jdddgdS)N012)rzror}rArrrrtest_del_before_definitionjs z*TestObjLifetime.test_del_before_definitionc Cs4|t |jttd}||jW5QRXdSN)ry)rr`rzrcr^rr@rrrr test_raising1ns zTestObjLifetime.test_raising1c Cs4|t |jttd}||jW5QRXdSr)rr`rzrdr^rr@rrrr test_raising2ss zTestObjLifetime.test_raising2c Cs4|t |jttd}||jW5QRXdSr)rr^rzrerr@rrrr test_raising3xs zTestObjLifetime.test_raising3c Cs~|t}t}t||}t|||jgt|||jgt|||jg||jdddddddgdS)Nrprqrr)rwrsr,iterr+r}r@rA)rrvrKiteratorrrr test_inf_loop_multiple_back_edge}s  z0TestObjLifetime.test_inf_loop_multiple_back_edge)N)r#r$r%rBrwrzrrrrrrrrrrrrrrrrrrts   rtc@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 finalizerpmrrrdefine_pipeliness  zqTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompiler.define_pipelinesNr#r$r%rrrrrIRPreservingCompilersr)pipeline_classcSsd}d}||}||}|SN rrErFrGrLrrrras zOTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..fooZEXTEND_VARIABLE_LIFETIMESr preserved_ir)r rr overloads signaturesmetadata)Zextend_lifetimesrracresfunc_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)rAssignDelReturn)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}||}||}|Srrrrrrras zUTestExtendingVariableLifetimes.test_dbg_extend_lifetimes..get_ir..foorr)r rrrr)optionsrrarrrrrrs  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>stmtrrrr?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__):r-sysr4rr itertoolsrZunittestZnumbarrZnumba.core.controlflowrrZnumba.core.compilerrrr r r Znumba.core.untyped_passesr Znumba.core.typed_passesr Z numba.corerrZnumba.tests.supportrrrZenable_pyobj_flagsZenable_pyobjectZforceobj_flagsZforce_pyobjectZno_pyobj_flagsobjectrr r,rMrPrSrYr\r] RuntimeErrorr^r`rcrdrerorsrtrr#mainrrrrsN    4    m