i_ dZddlZddlZddlZddlmZddlmZddl m Z m Z Gdde e Z e dk(rejyy) z< Tests issues or edge cases for producing invalid NRT refct N)njit)rtsys)TestCaseEnableNRTStatsMixinc6eZdZfdZdZdZdZdZxZS) TestNrtRefCtcRtjtt|y)N)gccollectsuperrsetUp)self __class__s o/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_nrt_refct.pyr zTestNrtRefCt.setUps  lD')c$td}d}tj}||tj}|j|j|jz ||j|j |j z |y)z" Test issue #1291 cNt|D]}tjd}yNr)rangenpzeros)nitemps rfooz(TestNrtRefCt.test_no_return..foos 1Xxx{r Nrrget_allocation_stats assertEqualallocfree)rrr init_stats cur_statss rtest_no_returnzTestNrtRefCt.test_no_returns|     //1  A..0  :+;+;;Q? *//91=rc td}tj}|dtj}|j|j|jz d|j|j |j z dy)z" Test issue #1297 ctj|df}t|D]}||} t|D]}||} yr)rrr)rxrys rgz6TestNrtRefCt.test_escaping_var_init_in_loop..g.sK!Q A1XaD1XaDrrNr)rr*r#r$s rtest_escaping_var_init_in_loopz+TestNrtRefCt.test_escaping_var_init_in_loop)sw    //1  "..0  :+;+;;Q? *//91=rc<td}tjjd}tj}||dtj}|j |j |j z |j|jz y)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_initializationEsAmmD)G7==+,- d#Nr)r5FN)rrrandomrrr r!r")rr4arrr#r$s r$test_invalid_computation_of_lifetimez1TestNrtRefCt.test_invalid_computation_of_lifetimeAs    iiv&//1 -c59..0  :+;+;;"*//9 ;rc&td}tjd}tj}||tj}|j |j |j z |j|jz y)z" Test issue #1734 cDd}dD]}||}||dkDs|||z }|S)Nr)rr+r+)r7resrts rfz5TestNrtRefCt.test_del_at_beginning_of_loop..f]s;CFQ4!81Q4KC Jr)rrN)rronesrrr r!r")rr>r7r#r$s rtest_del_at_beginning_of_loopz*TestNrtRefCt.test_del_at_beginning_of_loopYsw    ggfo//1  #..0  :+;+;;"*//9 ;r) __name__ __module__ __qualname__r r%r,r8r@ __classcell__)rs@rrrs* >$>0;0;rr__main__)__doc__r numpyrunittestnumbarnumba.core.runtimernumba.tests.supportrrrrAmainr;rrrMsJ  $=_;&_;D zHMMOr