U id<@sddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZmZmZmZeZde_eZde_ddZd d Zd d Zd dZddZddZddZddZGdddeeZedkre dS)N)compile_isolatedFlagsjit)typeserrorsutils)captured_stdouttagTestCaseEnableNRTStatsMixinTcCs t|dSNprintxrO/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/tests/test_print.py print_valuesrcCst||jdSr )rr)arrirrrprint_array_itemsrcCst|||dSr rabcrrr print_valuessrcCs tdSr rrrrr print_emptysrcCst|dddS)Nhop!g @rrrrr print_string srcCst||f|dSr rrrrr print_vararg#sr cCst|d|f|dS)Nrrrrrrprint_string_vararg&sr!csfdd}tddS)NcsSr rrrrr print_closure*sz)make_print_closure..print_closureTnopythonr)rr"rrrmake_print_closure)s r%c@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS) TestPrintc stfdd}|tjd|tjd|tjd|tjd|tjddtd f|tjd |t d t d d ft tj f}|j}t<|d tj}d}|||dW5QRX<ddddg}||ttj|fW5QRXW5QRXttjdd}t |ftd}|j}t,|t jdt jdtjdW5QRXdS)z8 Test printing a single argument value. c sTt|f}|j}|D]8}t(||tjt|dW5QRXqdS)N )r entry_pointr assertEqualsysstdoutgetvaluestr)typvaluescrcfuncvalpyfuncselfrr check_values7s  z1TestPrint.test_print_values..check_values))r7r8g3/m lg3/m )r7r9l{)TFg?g@xDnan)y??msdg?z 1.10000002384r'r7C)flags dtypez[0 1 2 3 4 5 6 7 8 9] N)rrint32Zint64Zuint64booleanZfloat64floatZ complex64Z NPTimedeltanpZ timedelta64rZfloat32r(r r*r+r, assertTrue startswithendswithZassertNoNRTLeakZassertRefCountListZArrayenable_pyobj_flagsaranger))r5r6r0r1gotexpectrZ arraytyperr3rtest_print_values1s:         ( zTestPrint.test_print_valuesc Cstdg}tjttdd|d}t}tdd|}tt|D]B}t2|||| t j t ||ddW5QRXqBd S) z: Test printing a Numpy character sequence )rZS4r7 rDTr#rr'N)rIrEZ frombuffer bytearrayrangerrlenr r)r*r+r,r-)r5rErr4r1rrrrtest_print_array_itemcs  zTestPrint.test_print_array_itemc CsPt}t|tjfd}|j}t$|ddd|tj dW5QRXdS)Nr>r71 2 3 ) rrrrFr(r r)r*r+r,r5r4r0r1rrrtest_print_multiple_valuesqs  z$TestPrint.test_print_multiple_valuesc CsHt}tddd|}t$|ddd|tjdW5QRXdS)NT)r$Znogilr7rXr>rY)rrr r)r*r+r,r5r4r1rrrtest_print_nogilys  zTestPrint.test_print_nogilc CsBt}t|d}|j}t||tjdW5QRXdS)Nrr')rrr(r r)r*r+r,rZrrrtest_print_emptys  zTestPrint.test_print_emptyc CsHt}t|tjf}|j}t |d|tj dW5QRXdS)Nr7z 1 hop! 3.5 ) rrrrFr(r r)r*r+r,rZrrrtest_print_stringss zTestPrint.test_print_stringsc Cst}tdd|}t$|ddd|tjdW5QRXt}tdd|}t$|ddd|tjdW5QRXdS)NTr#r7)rXr>)y@z1 (2, 3) 4 5j z1 hop! (2, 3) 4 5j )r rr r)r*r+r,r!r\rrrtest_print_varargs  zTestPrint.test_print_varargc sbtddddtddfdd}td}t |||tjdW5QRXdS) NTr#cSs t|dSr rrrrrfoosz*TestPrint.test_inner_fn_print..foocs|ddS)NZhellorrrbrrbarsz*TestPrint.test_inner_fn_print..barr?z[0 1 2 3 4] hello )rrIrOr r)r*r+r,)r5rdrrrcrtest_inner_fn_prints  zTestPrint.test_inner_fn_printc CsHtdddd}|tj }|W5QRXd}||jj|dS)NTr#cSstddddS)NrT)flushrrrrr print_kwargsz8TestPrint.test_print_w_kwarg_raises..print_kwargzKNumba's print() function implementation does not support keyword arguments.)r assertRaisesrZUnsupportedErrorZassertIn exceptionmsg)r5rgZraisesexpectedrrrtest_print_w_kwarg_raisess  z#TestPrint.test_print_w_kwarg_raisesc CsPtdddd}t.||tjddgddW5QRXd S) z9 See: https://github.com/numba/numba/issues/3811 Tr#cSstddgddS)Nr')rjoinrrrrrbsz/TestPrint.test_print_no_truncation..foormrrnr'N)rr r)r*r+r,ro)r5rbrrrtest_print_no_truncations  z"TestPrint.test_print_no_truncationN) __name__ __module__ __qualname__rRrWr[r]r^r_rarerlrprrrrr&/s2 r&__main__)!r*numpyrIZunittestZnumba.core.compilerrrZnumbarZ numba.corerrrZnumba.tests.supportr r r r rNZenable_pyobjectZforce_pyobj_flagsZforce_pyobjectrrrrrr r!r%r&rqmainrrrrs,