g<&ddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZmZmZmZeZde_eZde_dZdZd Zd Zd Zd Zd ZdZGddeeZedkrej dSdS)N)compile_isolatedFlagsjit)typeserrorsutils)captured_stdouttagTestCaseEnableNRTStatsMixinTc$t|dSNprintxs L/granary/edev/quantum/lib/python3.11/site-packages/numba/tests/test_print.py print_valuers !HHHHHc:t||jdSr)rr)arris rprint_array_itemrs #a&(OOOOOrc(t|||dSrrabcs r print_valuesr s !QNNNNNrc"tdSrrrr print_emptyr#s GGGGGrc(t|dddS)Nhop!g @rrs r print_stringr& s !VSrc"t||g|RdSrrrs r print_varargr(#s !QOOOOOOOrc$t|d|g|RdS)Nr%rrs rprint_string_varargr*&s" !VQrc@fd}tdS)NcSrr"rsr print_closurez)make_print_closure..print_closure*srTnopythonr)rr-s` rmake_print_closurer0)s7 3   a  rcJeZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) TestPrintctfd}|tjd|tjd|tjd|tjd|tjddtdf|tjd |tj d tj d d fttj f}|j}t5|d t j}d }|||ddddn #1swxYwY5gd}|5|tjtj|fdddn #1swxYwYdddn #1swxYwYtjtjdd}t|ft4}|j}t5|tjdtjt jdddddS#1swxYwYdS)z8 Test printing a single argument value. c(t|f}|j}|D]u}t5||tjt|dzdddn #1swxYwYvdS)N )r entry_pointr assertEqualsysstdoutgetvaluestr)typvaluescrcfuncvalpyfuncselfs r check_valuesz1TestPrint.test_print_values..check_values7s!&3&11BNE M M$&&MME#JJJ$$SZ%8%8%:%:CHHtOLLLMMMMMMMMMMMMMMM M MsABB B ))rDrEg3/m lg3/m )rDrFl{)TFg?g@xDnan)y??msdg?z 1.10000002384r5N)rDrDC)flags dtypez[0 1 2 3 4 5 6 7 8 9] )rrint32int64uint64booleanfloat64float complex64 NPTimedeltanp timedelta64rfloat32r6r r8r9r: assertTrue startswithendswithassertNoNRTLeakassertRefCountListArrayenable_pyobj_flagsaranger7) rBrCr>r?gotexpectr arraytyperAs ` @rtest_print_valueszTestPrint.test_print_values1s M M M M M M  U[),,, U[#L M M M U\$D E E E U]M222 U]S+uU||$DEEE U_g... U&t,,r~c4/H/H.JKKK fu}&6 7 7    0 0 E#JJJ*%%''C$F OOCNN622 3 3 3 OOCLL.. / / /  0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 ! ! # # < < A$$Q'' < < UZ 44qd;;; < < < < < < < < < < < < < < < < < < < < < < < < < < < < < < K Q44 fyl:L M M M    8 8 E")Bbh/// 0 0 0   SZ00226 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s\A C> c*t}t|tjfdz}|j}t 5|ddd|tj dddddS#1swxYwYdS)NrKrD1 2 3 ) r rrrSr6r r7r8r9r:rBrAr>r?s rtest_print_multiple_valuesz$TestPrint.test_print_multiple_valuesqs fu{nq&8 9 9    ? ? E!QNNN   SZ0022I > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sABB B ct}tdd|}t5|ddd|tjdddddS#1swxYwYdS)NT)r/nogilrDrtrKru)r rr r7r8r9r:rBrAr?s rtest_print_nogilzTestPrint.test_print_nogilys.T...v66    ? ? E!QNNN   SZ0022I > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sAA==BBct}t|d}|j}t5||t jdddddS#1swxYwYdS)Nr"r5)r#rr6r r7r8r9r:rvs rtest_print_emptyzTestPrint.test_print_emptys fb ) )    : : EGGG   SZ0022D 9 9 9 : : : : : : : : : : : : : : : : : :s=A77A;>A;c t}t|tjf}|j}t 5|d|tj dddddS#1swxYwYdS)NrDz 1 hop! 3.5 ) r&rrrSr6r r7r8r9r:rvs rtest_print_stringszTestPrint.test_print_stringss fu{n 5 5    D D E!HHH   SZ0022N C C C D D D D D D D D D D D D D D D D D Ds>BB Bct}td|}t5|ddd|tjddddn #1swxYwYt}td|}t5|ddd|tjdddddS#1swxYwYdS)NTr.rD)rtrK)y@z1 (2, 3) 4 5j z1 hop! (2, 3) 4 5j )r(rr r7r8r9r:r*rzs rtest_print_varargzTestPrint.test_print_varargs"T"""6**    G G E!VW % % %   SZ00224E F F F G G G G G G G G G G G G G G G%"T"""6**    L L E!VW % % %   SZ00224J K K K L L L L L L L L L L L L L L L L L Ls%AA;;A?A?4ADDDc`tddtdfd}tjd}t5|||t jdddddS#1swxYwYdS)NTr.c$t|dSrrrs rfooz*TestPrint.test_inner_fn_print..foos !HHHHHrc4|ddS)Nhellor")rrs rbarz*TestPrint.test_inner_fn_print..bars! CFFF CLLLLLrrLz[0 1 2 3 4] hello )rr[rfr r7r8r9r:)rBrrrs @rtest_inner_fn_printzTestPrint.test_inner_fn_prints d         d           IaLL    L L CFFF   SZ00224J K K K L L L L L L L L L L L L L L L L L Ls>B##B'*B'ctdd}|tj5}|dddn #1swxYwYd}||jj|dS)NTr.c(tdddS)NrT)flushrr"rr print_kwargz8TestPrint.test_print_w_kwarg_raises..print_kwargs #T " " " " " "rzKNumba's print() function implementation does not support keyword arguments.)r assertRaisesrUnsupportedErrorassertIn exceptionmsg)rBrraisesexpecteds rtest_print_w_kwarg_raisesz#TestPrint.test_print_w_kwarg_raisess d    # #   #  v6 7 7 6 KMMM               ) f&*H55555s AAAc2tdd}t5||tjddgdzdzddddS#1swxYwYdS) z9 See: https://github.com/numba/numba/issues/3811 Tr.cRtddgdzdS)Nr')rjoinr"rrrz/TestPrint.test_print_no_truncation..foos( "''3%%-(( ) ) ) ) )rrrrr5N)rr r7r8r9r:r)rBrs rtest_print_no_truncationz"TestPrint.test_print_no_truncations d    * *   *    S S CEEE   SZ0022BGGSEEM4J4JT4Q R R R S S S S S S S S S S S S S S S S S SsAB  BBN) __name__ __module__ __qualname__rjrrrwr{r}rrrrrr"rrr2r2/s080808d Q Q Q??????:::DDD L L LLLL" 6 6 6SSSSSrr2__main__)!r8numpyr[unittestnumba.core.compilerrrnumbar numba.corerrr numba.tests.supportr r r r reenable_pyobjectforce_pyobj_flagsforce_pyobjectrrr r#r&r(r*r0r2rmainr"rrrs 77777777++++++++++666666666666UWW%)"EGG#'       !!! TSTSTSTSTS#XTSTSTSl zHMOOOOOr