o 8ήcÚã@spddlZddlmZddlmZddlmZddlZdd„Z dd„Z Gd d „d eej ƒZ e d kr6e ¡dSdS) éN)Úcompile_isolated)Útypeof)ÚMemoryLeakMixincCs d||<|S)Né{©©ÚaÚirrúD/tmp/pip-target-vg8gfxp4/lib/python/numba/tests/test_array_return.pyÚ array_return sr cCs$t|jƒD] }||d7<q|S)Né)ÚrangeÚsizerrrr Úarray_return_start_with_loopsrc@seZdZdd„Zdd„ZdS)ÚTestArrayReturncCsJt d¡}d}t|ƒt|ƒ}}tt||fƒ}|j}| ||||ƒ¡dS)Né é)ÚnpÚarangerrr Ú entry_pointÚassertIs)Úselfrr ÚatÚitÚcresÚcfuncrrr Útest_array_returns z!TestArrayReturn.test_array_returncCs8t d¡}t|ƒ}tt|gƒ}|j}| |||ƒ¡dS)zN A bug breaks array return if the function starts with a loop rN)rrrrrrr)rrrrrrrr Ú!test_array_return_start_with_loops  z1TestArrayReturn.test_array_return_start_with_loopN)Ú__name__Ú __module__Ú __qualname__rrrrrr rs rÚ__main__)ÚnumpyrÚnumba.core.compilerrÚnumbarÚnumba.tests.supportrÚunittestr rÚTestCaserrÚmainrrrr Ús    ÿ