o i<@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/kre9dSdS)0N) takewhile)njitjit)CFGraphLoop) compile_extracompile_isolatedFlags CompilerBaseDefaultPassBuilder) PreserveIR)IRLegalization)typesir)TestCaseoverride_config SerialMixinTc@s$eZdZddZddZddZdS)_DummycCs||_||_||dSN)recordername _add_dummy)selfrrrl/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/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 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_dummyCs zRefRecorder.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]:}|d||}||}|d|D]}|d||}|dq$|d|rA||}nn |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_usecase1s  r[ccs,|d}|d}|D]}|Vq |VdSrYr)rJrDrErMrrrgenerator_usecase2s  r\c@s eZdZdS)MyErrorN)r"r#r$rrrrr]sr]cCst|rr])rMrrrdo_raisesr_cCs@|d}|d}|d}|rtd|d}||||dS)NrDrErKfoorFr_)rJrDrErKrFrrrraising_usecase1s rbcCs0|d}|d}|r|d}t|||dS)NrDrErFra)rJrDrErFrrrraising_usecase2s  rccCs |d}|d}|rt|dSrYr^rZrrrraising_usecase3s rdcCsd}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 rZFAILEDOK)ranger;str)rJnijrrrdel_before_definitions         rnccs. |ddV|d}|r|d q)z^ test to reveal bug of invalid liveness when infinite loop has multiple backedge. TyieldNpbrarI)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. cCsttjfddd|}|S)NTF)ZforceobjZlooplift)rrZpyobject)rpyfunccfuncrrrcompileszTestObjLifetime.compileNcCs\t}||}|dur(|| ||Wd|S1s!wY|S|||Sr)r+rv assertRaises)rrtraisesrJrurrrcompile_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@listr8remove assertEqual)rrJexpectedactualr@ remainingrKrrrassertRecordOrder s  z!TestObjLifetime.assertRecordOrdercCsJ|t}||j||gd||gd||gddS)N)rDrErG)rDrFrG)rGzb + crH)ryrL assertFalser?rrrJrrr test_simple1s  zTestObjLifetime.test_simple1cCs*|t}||j||gddS)N)rErGrD)ryrOrr?rrrrr test_simple2"s  zTestObjLifetime.test_simple2cCs:|t}||j||gd||gddS)N)rDrErQrF) iter(a)#1rP iter(a)#2rP iter(a)#3rPiter(a)rQ)ryrRrr?rrrrr test_looping1's  zTestObjLifetime.test_looping1cCsj|t}||j||ddg||gd||gd||gd||ddgdS)NrDrT)rrUrV)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--)ryrXrr?rrrrr test_looping23s   zTestObjLifetime.test_looping2cCs ||}t}|| ||}t|||jt|||jWdn1s0wYt}||||}~t ||jWdn1sWwYt}||"||}t|||j~t ||jWddS1swYdSr) rvr+assertRefCountr* assertTruer?rzrgcZcollect)rZgenfuncrurJgenrrrexercise_generatorIs2      "z"TestObjLifetime.exercise_generatorcC|tdSr)rr[r rrrtest_generator1dzTestObjLifetime.test_generator1cCrr)rr\r rrrtest_generator2grzTestObjLifetime.test_generator2cCs |t}||jgddS)N)012)ryrnr|r@rrrrtest_del_before_definitionjs z*TestObjLifetime.test_del_before_definitioncCJ|t|jttd}||jWddS1swYdSN)rx)rr_ryrbr]rr?rrrr test_raising1n "zTestObjLifetime.test_raising1cCrr)rr_ryrcr]rr?rrrr test_raising2srzTestObjLifetime.test_raising2cCsJ|t|jttd}||jWddS1swYdSr)rr]ryrdrr?rrrr test_raising3xrzTestObjLifetime.test_raising3cCst|t}t}t||}t|||jgt|||jgt|||jg||jgddS)N)rorprqrorprqro)rvrrr+iterr*r|r?r@)rrurJiteratorrrr test_inf_loop_multiple_back_edge}s  z0TestObjLifetime.test_inf_loop_multiple_back_edger)r"r#r$rArvryrrrrrrrrrrrrrrrrrrss"    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 Zdefine_nopython_pipelinestateZadd_pass_afterr r finalizerpmrrrdefine_pipelines  zqTestExtendingVariableLifetimes.test_lifetime_basic..get_ir..IRPreservingCompiler.define_pipelinesNr"r#r$rrrrrIRPreservingCompiler r)pipeline_classcSd}d}||}||}|SN rrDrErFrKrrrr` 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] \}}||qdS)Nr&) r|lenblocksr*rvalueszipbodyZassertIsInstance)rexpectblkZ expect_classZgot_stmtr rrchecks zATestExtendingVariableLifetimes.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}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)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&) rrrrzroperatornot_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$rrrrrrrs 2r__main__):r,sysr3rr 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_flagsobjectrrr+rLrOrRrXr[r\ RuntimeErrorr]r_rbrcrdrnrrrsrr"mainrrrrsR    4    m