o 8Î®cã@shdZddlZddlZddlZddlmZddlmZddl m Z Gdd„de ƒZedkr2e ¡dSdS) z< Tests issues or edge cases for producing invalid NRT refct éN)Únjit)Úrtsys)ÚTestCasec@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestNrtRefCtcCst ¡dS)N)ÚgcÚcollect)Úself©r úA/tmp/pip-target-vg8gfxp4/lib/python/numba/tests/test_nrt_refct.pyÚsetUpszTestNrtRefCt.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Útempr r r Úfoosz(TestNrtRefCt.test_no_return..fooé N©rrÚget_allocation_statsÚassertEqualÚallocÚfree)rrrÚ init_statsÚ cur_statsr r r Útest_no_returns zTestNrtRefCt.test_no_returncCsPtdd„ƒ}t ¡}|dƒt ¡}| |j|jd¡| |j|jd¡dS)z" Test issue #1297 cSs>t |df¡}t|ƒD]}||}qt|ƒD]}||}qdSr)rrr)rÚxrÚyr r r Úg-s z6TestNrtRefCt.test_escaping_var_init_in_loop..gréNr)rr rrr r r Ú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_arrrr r r Ú%if_with_allocation_and_initializationDs z`TestNrtRefCt.test_invalid_computation_of_lifetime..if_with_allocation_and_initialization)ér)FN)rrÚrandomrrrrr)rr(Úarrrrr r r Ú$test_invalid_computation_of_lifetime@s ÿ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Útr r r Úf\s€z5TestNrtRefCt.test_del_at_beginning_of_loop..f)r r N)rrÚonesrrrrr)rr/r+rrr r r Útest_del_at_beginning_of_loopXs ÿz*TestNrtRefCt.test_del_at_beginning_of_loopN)Ú__name__Ú __module__Ú__qualname__rrr"r,r1r r r r rsrÚ__main__)Ú__doc__rÚnumpyrÚunittestÚnumbarÚnumba.core.runtimerÚnumba.tests.supportrrr2Úmainr r r r Úsaÿ