g_ dZddlZddlZddlZddlmZddlmZddl m Z m Z Gdde e Z e dkrejdSdS) z< Tests issues or edge cases for producing invalid NRT refct N)njit)rtsys)TestCaseEnableNRTStatsMixinc6eZdZfdZdZdZdZdZxZS) TestNrtRefCtc|tjtt|dS)N)gccollectsuperrsetUp)self __class__s P/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_nrt_refct.pyr zTestNrtRefCt.setUps0  lD!!'')))))ctd}d}tj}||tj}||j|jz |||j|jz |dS)z" Test issue #1291 cRt|D]}tjd}dSNr)rangenpzeros)nitemps rfooz(TestNrtRefCt.test_no_return..foos+1XX # #x{{1r Nrrget_allocation_stats assertEqualallocfree)rrr init_stats cur_statss rtest_no_returnzTestNrtRefCt.test_no_returns      /11  A.00  :+;;Q??? */91=====rctd}tj}|dtj}||j|jz d||j|jz ddS)z" Test issue #1297 ctj|df}t|D] }||} t|D] }||} dSr)rrr)rxrys rgz6TestNrtRefCt.test_escaping_var_init_in_loop..g.s\!Q  A1XX  aD1XX  aD1rrNr)rr*r#r$s rtest_escaping_var_init_in_loopz+TestNrtRefCt.test_escaping_var_init_in_loop)s     /11  ".00  :+;;Q??? */91=====rc(td}tjd}tj}||dtj}||j|jz |j|jz dS)z" Test issue #1573 ctj|}t|jdD]}|rtj||S)Nr)r zeros_likershape)arr1test1tmp_arrrs r%if_with_allocation_and_initializationz`TestNrtRefCt.test_invalid_computation_of_lifetime..if_with_allocation_and_initializationEsOmD))G7=+,,   $ d###Nr)r5FN)rrrandomrrr r!r")rr4arrr#r$s r$test_invalid_computation_of_lifetimez1TestNrtRefCt.test_invalid_computation_of_lifetimeAs     iv&&/11 --c5999.00  :+;;"*/9 ; ; ; ; ;rctd}tjd}tj}||tj}||j|jz |j|jz dS)z" Test issue #1734 cRd}dD]!}||}||dkr |||z }"|S)Nr)rr+r+)r7resrts rfz5TestNrtRefCt.test_del_at_beginning_of_loop..f]sAC  FQ4!881Q4KCJr)rrN)rronesrrr r!r")rr>r7r#r$s rtest_del_at_beginning_of_loopz*TestNrtRefCt.test_del_at_beginning_of_loopYs     gfoo/11  #.00  :+;;"*/9 ; ; ; ; ;r) __name__ __module__ __qualname__r r%r,r8r@ __classcell__)rs@rrrst***** >>>$>>>0;;;0;;;;;;;rr__main__)__doc__r numpyrunittestnumbarnumba.core.runtimernumba.tests.supportrrrrAmainr;rrrMs  $$$$$$========_;_;_;_;_;&_;_;_;D zHMOOOOOr