
    Dg                         d dl mZ d dlZddlmZ ddlmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZ ddlmZmZ ddlmZ d Zd	 Zd
 Zd Zd Zd Zd Zd Z d Z!d Z"d Z#d Z$d Z%d Z&dS )    )assert_raisesN   )all)asarrayarangeempty
empty_likeeyefull	full_likelinspacemeshgridones	ones_likezeros
zeros_like)float32float64Arrayc                     t          t          d            t          t          d            t          t          d            t          t          d            t          t          d            t	          dgd           t          t          d	            t          t          d
            t          t          d            d S )Nc                  "    t          dg          S )N   r        b/var/www/sysmax/venv/lib/python3.11/site-packages/numpy/array_api/tests/test_creation_functions.py<lambda>z%test_asarray_errors.<locals>.<lambda>   s    UA3ZZ r   c                  "    t          dg          S )Nar   r   r   r   r   z%test_asarray_errors.<locals>.<lambda>   s    WcU^^ r   c                  :    t          dgt          j                  S N      ?dtype)r   npfloat16r   r   r   r   z%test_asarray_errors.<locals>.<lambda>   s    gse2:&F&F&F r   c                  &    t          ddz            S Nr   d   r    r   r   r   r   z%test_asarray_errors.<locals>.<lambda>   s    C r   c                  (    t          ddz  g          S r)   r    r   r   r   r   z%test_asarray_errors.<locals>.<lambda>    s    WafX%6%6 r   r   cpudevicec                  &    t          dgd          S Nr   gpur-   r    r   r   r   r   z%test_asarray_errors.<locals>.<lambda>"   s    gqc%&@&@&@ r   c                  0    t          dgt                    S Nr   r$   )r   intr   r   r   r   z%test_asarray_errors.<locals>.<lambda>$   s    gqc&=&=&= r   c                  &    t          dgd          S Nr   ir$   r    r   r   r   r   z%test_asarray_errors.<locals>.<lambda>%   s    gqc&=&=&= r   )r   	TypeError
ValueErrorOverflowErrorr   r   r   r   test_asarray_errorsr;      s    )//000)33444*FFGGG-!8!8999 )66777QC*@@AAA*==>>>*==>>>>>r   c                     t          dg          t          d          } dd<   t          | d         dk              sJ t          d         dk              sJ t          dg          t          t          j        j                  } dd<   t          | d         dk              sJ t          d         dk              sJ t          dg          t          t          j        j                  } dd<   t          | d         dk              sJ t          t          fd           t          t          fd           d S )Nr   Tcopyr   c                  &    t           d          S )NFr=   r    r   s   r   r   z#test_asarray_copy.<locals>.<lambda>7   s    wqu/E/E/E r   c                  D    t           t          j        j                  S )Nr=   )r   r&   	_CopyMode	IF_NEEDEDr@   s   r   r   z#test_asarray_copy.<locals>.<lambda>9   s    '!",*@AAA r   )r   r   r&   rB   ALWAYSNEVERr   NotImplementedError)br   s    @r   test_asarray_copyrH   (   sR   AAAaDqtqy>>qtqy>>A+,,,AAaDqtqy>>qtqy>>A*+++AAaDqtqy>>%'E'E'E'EFFF%AAAAC C C C Cr   c                      t          dd           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  $    t          dd          S r0   r   r   r   r   r   z$test_arange_errors.<locals>.<lambda>>   s    fQu&=&=&= r   c                  .    t          dt                    S r3   )r   r4   r   r   r   r   z$test_arange_errors.<locals>.<lambda>?   s    fQc&:&:&: r   c                  $    t          dd          S r6   rK   r   r   r   r   z$test_arange_errors.<locals>.<lambda>@   s    fQc&:&:&: r   )r   r   r9   r   r   r   test_arange_errorsrN   <   sW    
1U*==>>>*::;;;*::;;;;;r   c                      t          dd           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  $    t          dd          S NrP   r1   r-   r   r   r   r   r   z#test_empty_errors.<locals>.<lambda>E       eD&?&?&? r   c                  .    t          dt                    S NrP   r$   )r   r4   r   r   r   r   z#test_empty_errors.<locals>.<lambda>F       eD&<&<&< r   c                  $    t          dd          S NrP   r7   r$   rS   r   r   r   r   z#test_empty_errors.<locals>.<lambda>G       eD&<&<&< r   )r   r   r9   r   r   r   test_empty_errorsr[   C   W    	$u*??@@@*<<===*<<=====r   c                      t          t          d          d           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  >    t          t          d          d          S r0   r	   r   r   r   r   r   z(test_empty_like_errors.<locals>.<lambda>L       jE&J&J&J r   c                  H    t          t          d          t                    S r3   )r	   r   r4   r   r   r   r   z(test_empty_like_errors.<locals>.<lambda>M       j3&G&G&G r   c                  >    t          t          d          d          S r6   r_   r   r   r   r   z(test_empty_like_errors.<locals>.<lambda>N       j3&G&G&G r   )r	   r   r   r9   r   r   r   test_empty_like_errorsre   J   ]    wqzz%((((*JJKKK*GGHHH*GGHHHHHr   c                      t          dd           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  $    t          dd          S r0   r
   r   r   r   r   z!test_eye_errors.<locals>.<lambda>S   s    c!E&:&:&: r   c                  .    t          dt                    S r3   )r
   r4   r   r   r   r   z!test_eye_errors.<locals>.<lambda>T   s    c!3&7&7&7 r   c                  $    t          dd          S r6   ri   r   r   r   r   z!test_eye_errors.<locals>.<lambda>U   s    c!3&7&7&7 r   )r
   r   r9   r   r   r   test_eye_errorsrl   Q   sW    %*::;;;*77888*7788888r   c                      t          ddd           t          t          d            t          t          d            t          t          d            d S )NrP   r   r,   r-   c                  &    t          ddd          S )NrP   r   r1   r-   r   r   r   r   r   z"test_full_errors.<locals>.<lambda>Z   s    d45&A&A&A r   c                  0    t          ddt                    S )NrP   r   r$   )r   r4   r   r   r   r   z"test_full_errors.<locals>.<lambda>[   s    d4#&>&>&> r   c                  &    t          ddd          S )NrP   r   r7   r$   ro   r   r   r   r   z"test_full_errors.<locals>.<lambda>\   s    d4#&>&>&> r   )r   r   r9   r   r   r   test_full_errorsrr   X   sY    q*AABBB*>>???*>>?????r   c                      t          t          d          dd           t          t          d            t          t          d            t          t          d            d S )Nr   r   r,   r-   c                  @    t          t          d          dd          S )Nr   r   r1   r-   r   r   r   r   r   r   z'test_full_like_errors.<locals>.<lambda>a   s    i

Ae&L&L&L r   c                  J    t          t          d          dt                    S )Nr   r   r$   )r   r   r4   r   r   r   r   z'test_full_like_errors.<locals>.<lambda>b   s    i

AS&I&I&I r   c                  @    t          t          d          dd          S )Nr   r   r7   r$   ru   r   r   r   r   z'test_full_like_errors.<locals>.<lambda>c   s    i

AS&I&I&I r   )r   r   r   r9   r   r   r   test_full_like_errorsrx   _   s_    gajj!E*****LLMMM*IIJJJ*IIJJJJJr   c                      t          dddd           t          t          d            t          t          d            t          t          d            d S )	Nr   r   
   r,   r-   c                  (    t          dddd          S )Nr   r   rz   r1   r-   r   r   r   r   r   z&test_linspace_errors.<locals>.<lambda>h   s    hq!R&F&F&F r   c                  2    t          dddt                    S )Nr   r   rz   r$   )r   floatr   r   r   r   z&test_linspace_errors.<locals>.<lambda>i   s    hq!Ru&E&E&E r   c                  (    t          dddd          S )Nr   r   rz   fr$   r|   r   r   r   r   z&test_linspace_errors.<locals>.<lambda>j   s    hq!Rs&C&C&C r   )r   r   r9   r   r   r   test_linspace_errorsr   f   s[    Q2e$$$$*FFGGG*EEFFF*CCDDDDDr   c                      t          dd           t          t          d            t          t          d            t          t          d            d S )NrP   r,   r-   c                  $    t          dd          S rR   r   r   r   r   r   z"test_ones_errors.<locals>.<lambda>o   s    d4&>&>&> r   c                  .    t          dt                    S rV   )r   r4   r   r   r   r   z"test_ones_errors.<locals>.<lambda>p   s    d4s&;&;&; r   c                  $    t          dd          S rY   r   r   r   r   r   z"test_ones_errors.<locals>.<lambda>q   s    d4s&;&;&; r   )r   r   r9   r   r   r   test_ones_errorsr   m   sW    e*>>???*;;<<<*;;<<<<<r   c                      t          t          d          d           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  >    t          t          d          d          S r0   r   r   r   r   r   r   z'test_ones_like_errors.<locals>.<lambda>v   s    i

5&I&I&I r   c                  H    t          t          d          t                    S r3   )r   r   r4   r   r   r   r   z'test_ones_like_errors.<locals>.<lambda>w   s    i

#&F&F&F r   c                  >    t          t          d          d          S r6   r   r   r   r   r   z'test_ones_like_errors.<locals>.<lambda>x   s    i

#&F&F&F r   )r   r   r   r9   r   r   r   test_ones_like_errorsr   t   s]    gajj''''*IIJJJ*FFGGG*FFGGGGGr   c                      t          dd           t          t          d            t          t          d            t          t          d            d S )NrP   r,   r-   c                  $    t          dd          S rR   r   r   r   r   r   z#test_zeros_errors.<locals>.<lambda>}   rT   r   c                  .    t          dt                    S rV   )r   r4   r   r   r   r   z#test_zeros_errors.<locals>.<lambda>~   rW   r   c                  $    t          dd          S rY   r   r   r   r   r   z#test_zeros_errors.<locals>.<lambda>   rZ   r   )r   r   r9   r   r   r   test_zeros_errorsr   {   r\   r   c                      t          t          d          d           t          t          d            t          t          d            t          t          d            d S )Nr   r,   r-   c                  >    t          t          d          d          S r0   r   r   r   r   r   r   z(test_zeros_like_errors.<locals>.<lambda>   r`   r   c                  H    t          t          d          t                    S r3   )r   r   r4   r   r   r   r   z(test_zeros_like_errors.<locals>.<lambda>   rb   r   c                  >    t          t          d          d          S r6   r   r   r   r   r   z(test_zeros_like_errors.<locals>.<lambda>   rd   r   )r   r   r   r9   r   r   r   test_zeros_like_errorsr      rf   r   c                  
   t                       t          t          dgt                               t          t          dgt                    t          dgt                               t          t          d            d S )Nr#   r$   c                  v    t          t          dgt                    t          dgt                              S r"   )r   r   r   r   r   r   r   r   z,test_meshgrid_dtype_errors.<locals>.<lambda>   s1    hwt7/K/K/KWVXUYahMiMiMi&j&j r   )r   r   r   r   r9   r   r   r   test_meshgrid_dtype_errorsr      sp    JJJWbT)))***WbT)))7B4w+G+G+GHHH*jjkkkkkr   )'numpy.testingr   numpyr&    r   _creation_functionsr   r   r   r	   r
   r   r   r   r   r   r   r   r   _dtypesr   r   _array_objectr   r;   rH   rN   r[   re   rl   rr   rx   r   r   r   r   r   r   r   r   r   <module>r      s   ' ' ' ' ' '                                        ' & & & & & & & ! ! ! ! ! !? ? ? C C C(< < <> > >I I I9 9 9@ @ @K K KE E E= = =H H H> > >I I Il l l l lr   