
    J/Ph                         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
 d dlmZ  G d de          Zedk    r ej                     dS dS )	    N)jitnjit)types)usecases)TestCasec                      e Zd Zej        d             Zej        d             Zej        d             Zej        d             Zej        d             Z	ej        d             Z
ej        d             Zej        d             Zej        d	             Zej        d
             Zej        d             Zej        d             Zej        d             ZdS )TestUsecasesc                     t           j        } t          t          j        t          j        f          |          }d}d}t          j        ||          D ]"}|                      ||  || d|           #d S )N)r      	   
      zargs )r   andorr   r   int32	itertoolsproductassertEqual)selfpyfunccfuncxsysargss         Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/tests/test_usecases.py
test_andorzTestUsecases.test_andor   s    0ek5;/0088 ! %b"-- 	O 	ODVVT]EE4LLtt:MNNNN	O 	O    c                     t           j        } t          t          j        t          j        f          |          }d}d}t          j        ||          D ]}|                      ||  || |            d S Nr   r   r   d      )r   sum1dr   r   r   r   r   r   r   r   r   ssesr   s         r   
test_sum1dzTestUsecases.test_sum1d       0ek5;/0088%b"-- 	@ 	@DVVT]EE4L$????	@ 	@r   c                     t           j        } t          t          j        t          j        fd          |          }d}d}t          j        ||          D ]}|                      ||  || |            d S )NTforceobjr    )r   r#   r   r   r   r   r   r   r$   s         r   test_sum1d_pyobjzTestUsecases.test_sum1d_pyobj&   s    >U[%+.>>>vFF%b"-- 	@ 	@DVVT]EE4L$????	@ 	@r   c                     t           j        } t          t          j        t          j        f          |          }d}d}t          j        ||          D ]}|                      ||  || |            d S r   )r   sum2dr   r   r   r   r   r   r$   s         r   
test_sum2dzTestUsecases.test_sum2d1   r(   r   c                     t           j        } t          t          j        t          j        f          |          }d}d}t          j        ||          D ]}|                      ||  || |            d S r   )r   while_countr   r   r   r   r   r   r$   s         r   test_while_countzTestUsecases.test_while_count<   s    %0ek5;/0088%b"-- 	@ 	@DVVT]EE4L$????	@ 	@r   c                 P   t           j        }t          j        t          j        dd          } t          ||f          |          }d}|D ]Z}t          j        |d          }t          j        |          }||f} ||  | 	                    ||t          |                     [d S )Nr   A)r   r   r   r!   r   dtypemsg)r   copy_arraysr   Arrayr   r   nparange
empty_likeassertPreciseEqualstr)	r   r   	arraytyper   ndandabr   s	            r   test_copy_arrayszTestUsecases.test_copy_arraysG   s    %KQ44	,i+,,V44 	9 	9B	"G,,,Aa  Aa4DE4LL##Aqc$ii#8888	9 	9r   c                    t           j        }t          j        t          j        dd          } t          ||f          |          }d}|D ]v}|\  }}t          j        ||z  d                              ||          }t          j	        |          }	||	f}
 ||
  | 
                    ||	t          |
                     wd S )N   r4   ))r   r   )r   r   )rG      )      r   r5   r7   )r   copy_arrays2dr   r:   r   r   r;   r<   reshaper=   r>   r?   )r   r   r@   r   rA   rB   d1d2rC   rD   r   s              r   test_copy_arrays2dzTestUsecases.test_copy_arrays2dW   s    'KQ44	,i+,,V44- 	9 	9BFB	"r'11199"bAAAa  Aa4DE4LL##Aqc$ii#8888	9 	9r   c                     t           j        } t          t          j        t          j        fd          |          }d}d}t          j        ||          D ]&\  }}||f}|                      ||  || |           'd S )NTr*   )r   r   r   )r   string_concatr   r   r   r   r   r   )r   r   r   r   r   xyr   s           r   test_string_concatzTestUsecases.test_string_concath   s    '>U[%+.>>>vFF%b"-- 	@ 	@DAqa4DVVT]EE4L$????	@ 	@r   c                 d   t           j        } t          t          j        fd          |          }d}|                      ||           ||                     d}|                      ||           ||                     d}|                      ||           ||                     d S )NTr*   1234561 )r   
string_lenr   r   pyobjectr   r   r   r   test_strs       r   test_string_lenzTestUsecases.test_string_lent   s    $5U^%555f==))55??;;;))55??;;;))55??;;;;;r   c                 z   t           j        } t          t          j        fdz  d          |          }d}|                      ||dd           ||dd                     |                      ||dd           ||dd                     |                      ||dd           ||dd                     d S )	N   Tr*   rV   r   r   rH   rG   )r   string_slicingr   r   rZ   r   r[   s       r   test_string_slicingz TestUsecases.test_string_slicing   s    (9U^%)D999&AA!Q//xA1F1FGGG!Q//xA1F1FGGG!Q//xA1F1FGGGGGr   c                 R   t           j        } t          t          j        fd          |          }|                      |d           |d                      t          t          j        fd          |          }|                      |d           |d                     d S )NTr*   r   g?)r   string_conversionr   r   r   r   float32)r   r   r   s      r   test_string_conversionz#TestUsecases.test_string_conversion   s    +2U[NT2226::EE!HH---4U]$t444V<<eeCjj11111r   c                 T   dd l }t          j        } t          t          j        t          j        t          j        fd          |          }d}d}|j        }|                      ||||           ||||                     d}d}|j        }|                      ||||           ||||                     d}d}|j        }|                      ||||           ||||                     d}d}|j        }|                      ||||           ||||                     d S )Nr   Tr*   123456)	operatorr   string_comparisonr   r   rZ   eqr   ne)r   ri   r   r   	test_str1	test_str2ops          r   test_string_comparisonsz$TestUsecases.test_string_comparisons   s   +#U^U^U^D!# # ##)+ + 		[	9b99E)Y++	- 	- 	- 		[	9b99E)Y++	- 	- 	- 		[	9b99E)Y++	- 	- 	- 		[	9b99E)Y++	- 	- 	- 	- 	-r   c                     t           j        } t          t          j        f          |          }d}|D ]"}|f}|                      ||  || |           #d S )N)g      r   g      ?)r   blackscholes_cndr   r   rd   r   )r   r   r   dsdr   s         r   test_blackscholes_cndz"TestUsecases.test_blackscholes_cnd   sv    *&em%&&v.. 	@ 	@A4DVVT]EE4L$????	@ 	@r   N)__name__
__module____qualname__r   run_test_in_subprocessr   r'   r,   r/   r2   rE   rO   rT   r]   ra   re   rp   ru    r   r   r	   r	      s        $	O 	O %$	O $@ @ %$@ $@ @ %$@ $@ @ %$@ $@ @ %$@ $9 9 %$9 $9 9 %$9  $	@ 	@ %$	@ $	< 	< %$	< $H H %$H $2 2 %$2 $- - %$-< $@ @ %$@ @ @r   r	   __main__)r   unittestnumpyr;   numbar   r   
numba.corer   numba.testsr   numba.tests.supportr   r	   rv   mainrz   r   r   <module>r      s                                       ( ( ( ( ( (q@ q@ q@ q@ q@8 q@ q@ q@h zHMOOOOO r   