U ,‰dã@sddZddlZddlZddlZddlmZddlmZddl m Z Gdd„de ƒZedkr`e ¡dS) 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)ÚgcZcollect)Úself©rú>/tmp/pip-unpacked-wheel-eu7e0c37/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Útemprrr Úfoosz(TestNrtRefCt.test_no_return..fooé N©rrÚget_allocation_statsÚassertEqualÚallocÚfree)rrrÚ 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]}||}qt|ƒD]}||}q,dSr)rrr )rÚxrÚyrrr Úg-s z6TestNrtRefCt.test_escaping_var_init_in_loop..gréNr)rrrrrrr Ú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|r,t |¡|S)Nr)rZ zeros_liker Úshape)Zarr1Ztest1Ztmp_arrrrrr Ú%if_with_allocation_and_initializationDs z`TestNrtRefCt.test_invalid_computation_of_lifetime..if_with_allocation_and_initialization)ér$FN)rrÚrandomrrrrr)rr#Úarrrrrrr Ú$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Útrrr Úf\sz5TestNrtRefCt.test_del_at_beginning_of_loop..f)rrN)rrZonesrrrrr)rr*r&rrrrr Útest_del_at_beginning_of_loopXs ÿz*TestNrtRefCt.test_del_at_beginning_of_loopN)Ú__name__Ú __module__Ú__qualname__r rr!r'r+rrrr rs rÚ__main__)Ú__doc__rZnumpyrZunittestZnumbarZnumba.core.runtimerZnumba.tests.supportrrr,Úmainrrrr Úsa