a d<@sddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZmZmZmZeZde_eZde_ddZd d Zd d Zd dZddZddZddZddZGdddeeZedkre dS)N)compile_isolatedFlagsjit)typeserrorsutils)captured_stdouttagTestCaseEnableNRTStatsMixinTcCs t|dSNprintxr_/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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||g|RdSr rrrrr print_vararg#sr cCst|d|g|RdS)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}tF|d tj}d}|||dWdn1s0YXgd}|$|ttj|fWdn1s60YWdn1sV0Yttjdd}t |ftd}|j}t6|t jdt jdtjdWdn1s0YdS)z8 Test printing a single argument value. c sht|f}|j}|D]L}t2||tjt|dWdq1sX0YqdS)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'N)r7r7C)flags dtypez[0 1 2 3 4 5 6 7 8 9] )rrint32int64Zuint64booleanfloat64floatZ complex64Z NPTimedeltanpZ timedelta64rfloat32r(r r*r+r, assertTrue startswithendswithZassertNoNRTLeakZassertRefCountListArrayenable_pyobj_flagsaranger))r5r6r0r1gotexpectrZ arraytyperr3rtest_print_values1s:      .  T zTestPrint.test_print_valuesc Cstdg}tjttdd|d}t}tdd|}tt|D]V}t<|||| t j t ||ddWd qB1s0YqBd S) z: Test printing a Numpy character sequence )rZS4r7 rDTr#rr'N)rKrEZ frombuffer bytearrayrangerrlenr r)r*r+r,r-)r5rErr4r1rrrrtest_print_array_itemcs  zTestPrint.test_print_array_itemcCsdt}t|tjfd}|j}t.|ddd|tj dWdn1sV0YdS)Nr>r71 2 3 ) rrrrFr(r r)r*r+r,r5r4r0r1rrrtest_print_multiple_valuesqs  z$TestPrint.test_print_multiple_valuescCs\t}tddd|}t.|ddd|tjdWdn1sN0YdS)NT)r$Znogilr7r\r>r])rrr r)r*r+r,r5r4r1rrrtest_print_nogilys  zTestPrint.test_print_nogilcCsVt}t|d}|j}t(||tjdWdn1sH0YdS)Nrr')rrr(r r)r*r+r,r^rrrtest_print_emptys  zTestPrint.test_print_emptycCs\t}t|tjf}|j}t*|d|tj dWdn1sN0YdS)Nr7z 1 hop! 3.5 ) rrrrFr(r r)r*r+r,r^rrrtest_print_stringss zTestPrint.test_print_stringscCst}tdd|}t.|ddd|tjdWdn1sL0Yt}tdd|}t.|ddd|tjdWdn1s0YdS)NTr#r7)r\r>)y@z1 (2, 3) 4 5j z1 hop! (2, 3) 4 5j )r rr r)r*r+r,r!r`rrrtest_print_varargs 0 zTestPrint.test_print_varargcsvtddddtddfdd}td}t*|||tjdWdn1sh0YdS) NTr#cSs t|dSr rrrrrfoosz*TestPrint.test_inner_fn_print..foocs|ddS)NZhellorrrfrrbarsz*TestPrint.test_inner_fn_print..barr?z[0 1 2 3 4] hello )rrKrSr r)r*r+r,)r5rhrrrgrtest_inner_fn_prints  zTestPrint.test_inner_fn_printcCs\tdddd}|tj}|Wdn1s:0Yd}||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)r5rkZraisesexpectedrrrtest_print_w_kwarg_raisess  $z#TestPrint.test_print_w_kwarg_raisescCsdtdddd}t8||tjddgddWd n1sV0Yd S) z9 See: https://github.com/numba/numba/issues/3811 Tr#cSstddgddS)Nr')rjoinrrrrrfsz/TestPrint.test_print_no_truncation..foorqrrrr'N)rr r)r*r+r,rs)r5rfrrrtest_print_no_truncations  z"TestPrint.test_print_no_truncationN) __name__ __module__ __qualname__rVr[r_rarbrcrerirprtrrrrr&/s2 r&__main__)!r*numpyrKZunittestZnumba.core.compilerrrZnumbarZ numba.corerrrZnumba.tests.supportr r r r rRZenable_pyobjectZforce_pyobj_flagsZforce_pyobjectrrrrrr r!r%r&rumainrrrrs,