o
    i<                     @   s   d dl Z d dlZd dlZd dlmZmZ d dlmZ d dl	m
Z
mZmZ d dlmZmZmZmZ e Zde_e Zde_dd Zd	d
 Zdd Zdd Zdd Zdd Zdd Zdd ZG dd deeZedkroe   dS dS )    N)compile_isolatedFlagsjit)typeserrorsutils)captured_stdouttagTestCaseEnableNRTStatsMixinTc                 C      t |  d S Nprintx r   e/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/tests/test_print.pyprint_value   s   r   c                 C   s   t | | j d S r   )r   r   )arrir   r   r   print_array_item   s   r   c                 C   s   t | || d S r   r   abcr   r   r   print_values      r   c                   C   s
   t   d S r   r   r   r   r   r   print_empty   s   
r   c                 C   s   t | dd d S )Nhop!g      @r   r   r   r   r   print_string    r   r!   c                 C   s   t | |g|R   d S r   r   r   r   r   r   print_vararg#   s   r"   c                 C   s   t | d|g|R   d S )Nr    r   r   r   r   r   print_string_vararg&   s   r#   c                    s    fdd}t dd S )Nc                      s    S r   r   r   r   r   r   print_closure*   s   z)make_print_closure.<locals>.print_closureTnopythonr   )r   r$   r   r   r   make_print_closure)   s   r'   c                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )	TestPrintc              	      s  t   fdd}|tjd |tjd |tjd |tjd |tjddtd	f |tjd
 |t	dt
ddf t tjf}|j}t # |d tj }d}|| |d W d   n1 srw   Y   + g d}| |ttj|f W d   n1 sw   Y  W d   n1 sw   Y  ttjdd}t |ftd}|j}t  |t
jdt
jd tj d W d   dS 1 sw   Y  dS )z8
        Test printing a single argument value.
        c              	      sh   t  | f}|j}|D ]&}t  || tj t|d  W d    n1 s,w   Y  qd S )N
)r   entry_pointr	   assertEqualsysstdoutgetvaluestr)typvaluescrcfuncvalpyfuncselfr   r   check_values7   s   z1TestPrint.test_print_values.<locals>.check_values)   )r9   r:      g3/mlg3/m)r9      r;   l   {        )TFg      ?g@xDnan)y      ?      ?msd   g?z1.10000002384r)   N)r9            r9   C)flags
   dtypez[0 1 2 3 4 5 6 7 8 9]
)r   r   int32Zint64Zuint64booleanZfloat64floatZ	complex64ZNPTimedeltanpZtimedelta64r   Zfloat32r*   r	   r,   r-   r.   
assertTrue
startswithendswithZassertNoNRTLeakZassertRefCountListZArrayenable_pyobj_flagsaranger+   )r7   r8   r2   r3   gotexpectr   Z	arraytyper   r5   r   test_print_values1   sD   	

	"zTestPrint.test_print_valuesc              	   C   s   t dg}t jttdd|d}t}tdd|}tt|D ]+}t  ||| | 	t
j t|| d d  W d	   n1 sGw   Y  q!d	S )
z:
        Test printing a Numpy character sequence
        )r   ZS4r9   	   rF   Tr%   r   r)   N)rK   rG   Z
frombuffer	bytearrayranger   r   lenr	   r+   r,   r-   r.   r/   )r7   rG   r   r6   r3   r   r   r   r   test_print_array_itemc   s   
$zTestPrint.test_print_array_itemc                 C   sf   t }t|tjfd }|j}t  |ddd | tj	 d W d    d S 1 s,w   Y  d S )Nr@   r9      1 2 3
)
r   r   r   rH   r*   r	   r+   r,   r-   r.   r7   r6   r2   r3   r   r   r   test_print_multiple_valuesq   s   "z$TestPrint.test_print_multiple_valuesc                 C   s^   t }tddd|}t  |ddd | tj d W d    d S 1 s(w   Y  d S )NT)r&   Znogilr9   rZ   r@   r[   )r   r   r	   r+   r,   r-   r.   r7   r6   r3   r   r   r   test_print_nogily   s   "zTestPrint.test_print_nogilc                 C   sX   t }t|d}|j}t  |  | tj d W d    d S 1 s%w   Y  d S )Nr   r)   )r   r   r*   r	   r+   r,   r-   r.   r\   r   r   r   test_print_empty   s   
"zTestPrint.test_print_emptyc                 C   s^   t }t|tjf}|j}t  |d | tj	 d W d    d S 1 s(w   Y  d S )Nr9   z1 hop! 3.5
)
r!   r   r   rH   r*   r	   r+   r,   r-   r.   r\   r   r   r   test_print_strings   s   "zTestPrint.test_print_stringsc                 C   s   t }tdd|}t  |ddd | tj d W d    n1 s&w   Y  t}tdd|}t  |ddd | tj d W d    d S 1 sRw   Y  d S )NTr%   r9   )rZ   r@   )   y              @z1 (2, 3) 4 5j
z1 hop! (2, 3) 4 5j
)r"   r   r	   r+   r,   r-   r.   r#   r^   r   r   r   test_print_vararg   s   "zTestPrint.test_print_varargc                    sx   t dddd  t dd fdd}td}t  || | tj d W d    d S 1 s5w   Y  d S )	NTr%   c                 S   r   r   r   r   r   r   r   foo   s   z*TestPrint.test_inner_fn_print.<locals>.fooc                    s    |   d d S )NZhellor   r   rd   r   r   bar   s   z*TestPrint.test_inner_fn_print.<locals>.barrA   z[0 1 2 3 4]
hello
)r   rK   rQ   r	   r+   r,   r-   r.   )r7   rf   r   r   re   r   test_inner_fn_print   s   

"zTestPrint.test_inner_fn_printc                 C   s\   t dddd }| tj}|  W d    n1 sw   Y  d}| |jj| d S )NTr%   c                   S   s   t ddd d S )Nr   T)flushr   r   r   r   r   print_kwarg   s   z8TestPrint.test_print_w_kwarg_raises.<locals>.print_kwargzKNumba's print() function implementation does not support keyword arguments.)r   assertRaisesr   ZUnsupportedErrorZassertIn	exceptionmsg)r7   ri   Zraisesexpectedr   r   r   test_print_w_kwarg_raises   s   
z#TestPrint.test_print_w_kwarg_raisesc                 C   sf   t dddd }t  |  | tj ddgd d  W d	   d	S 1 s,w   Y  d	S )
z9 See: https://github.com/numba/numba/issues/3811
        Tr%   c                   S   s   t ddgd  d S )N r   '  )r   joinr   r   r   r   rd      s   z/TestPrint.test_print_no_truncation.<locals>.fooro   r   rp   r)   N)r   r	   r+   r,   r-   r.   rq   )r7   rd   r   r   r   test_print_no_truncation   s   
$"z"TestPrint.test_print_no_truncationN)__name__
__module____qualname__rT   rY   r]   r_   r`   ra   rc   rg   rn   rr   r   r   r   r   r(   /   s    2r(   __main__)!r,   numpyrK   ZunittestZnumba.core.compilerr   r   Znumbar   Z
numba.corer   r   r   Znumba.tests.supportr	   r
   r   r   rP   Zenable_pyobjectZforce_pyobj_flagsZforce_pyobjectr   r   r   r   r!   r"   r#   r'   r(   rs   mainr   r   r   r   <module>   s0     