o ¦¸¢i_ ã@sndZddlZddlZddlZddlmZddlmZddl m Z m Z Gdd„de e ƒZ e dkr5e ¡dSdS) z< Tests issues or edge cases for producing invalid NRT refct éN)Únjit)Úrtsys)ÚTestCaseÚEnableNRTStatsMixincs<eZdZ‡fdd„Zdd„Zdd„Zdd„Zd d „Z‡ZS) Ú TestNrtRefCtcst ¡tt|ƒ ¡dS)N)ÚgcÚcollectÚsuperrÚsetUp)Úself©Ú __class__©úX/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/tests/test_nrt_refct.pyr szTestNrtRefCt.setUpcCsTtdd„ƒ}d}t ¡}||ƒt ¡}| |j|j|¡| |j|j|¡dS)z" Test issue #1291 cSst|ƒD]}t d¡}qdS©Nér)ÚrangeÚnpÚzeros)ÚnÚiÚtemprrrÚfoos  z(TestNrtRefCt.test_no_return..fooé N©rrÚget_allocation_statsÚ assertEqualÚallocÚfree)r rrÚ init_statsÚ cur_statsrrrÚtest_no_returns zTestNrtRefCt.test_no_returncCsPtdd„ƒ}t ¡}|dƒt ¡}| |j|jd¡| |j|jd¡dS)z" Test issue #1297 cSs>t |df¡}t|ƒD]}||}q t|ƒD]}||}qdSr)rrr)rÚxrÚyrrrÚg.s     z6TestNrtRefCt.test_escaping_var_init_in_loop..gréNr)r r$rr rrrÚtest_escaping_var_init_in_loop)s  z+TestNrtRefCt.test_escaping_var_init_in_loopcCsRtdd„ƒ}tj d¡}t ¡}||dƒt ¡}| |j|j|j|j¡dS)z" Test issue #1573 cSs0t |¡}t|jdƒD]}q |rt |¡|S)Nr)rÚ zeros_likerÚshape)Úarr1Útest1Útmp_arrrrrrÚ%if_with_allocation_and_initializationEs  z`TestNrtRefCt.test_invalid_computation_of_lifetime..if_with_allocation_and_initialization)ér-FN)rrÚrandomrrrrr)r r,Úarrrr rrrÚ$test_invalid_computation_of_lifetimeAs    ÿz1TestNrtRefCt.test_invalid_computation_of_lifetimecCsNtdd„ƒ}t d¡}t ¡}||ƒt ¡}| |j|j|j|j¡dS)z" Test issue #1734 cSs2d}dD]}||}||dkr|||7}q|S)Nr)rr%r%r)r/ÚresrÚtrrrÚf]s  €z5TestNrtRefCt.test_del_at_beginning_of_loop..f)rrN)rrÚonesrrrrr)r r3r/rr rrrÚtest_del_at_beginning_of_loopYs   ÿz*TestNrtRefCt.test_del_at_beginning_of_loop) Ú__name__Ú __module__Ú __qualname__r r!r&r0r5Ú __classcell__rrr rrs  rÚ__main__)Ú__doc__rÚnumpyrÚunittestÚnumbarÚnumba.core.runtimerÚnumba.tests.supportrrrr6ÚmainrrrrÚs  b ÿ