o
    0c                     @   s   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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d Z$d d! Z%d"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                   C   s   t tdd  t tdd  t tdd  t tdd  t tdd  tdgdd	 t td
d  t tdd  t tdd  d S )Nc                   S   
   t dgS )N   r    r   r   c/var/www/html/gps/gps/lib/python3.10/site-packages/numpy/array_api/tests/test_creation_functions.py<lambda>      
 z%test_asarray_errors.<locals>.<lambda>c                   S   r   )Nar   r   r   r   r   r      r   c                   S   s   t dgtjdS N      ?dtype)r   npfloat16r   r   r   r   r          c                   S   s   t dd S Nr   d   r   r   r   r   r   r          c                   S   s   t dd gS r%   r   r   r   r   r   r           r   cpudevicec                   S      t dgddS Nr   gpur*   r   r   r   r   r   r   "   r(   c                   S   s   t dgtdS Nr   r    )r   intr   r   r   r   r   $   r(   c                   S   r,   Nr   ir    r   r   r   r   r   r   %   r(   )r   	TypeError
ValueErrorOverflowErrorr   r   r   r   r   test_asarray_errors   s   r6   c                     s   t dg t  dd} d d< t| d dksJ t d dks#J t dg t  tjjd} d d< t| d dks>J t d dksHJ t dg t  tjjd} d d< t| d dkscJ tt fdd tt fdd d S )Nr   Tcopyr   c                      s   t  ddS )NFr7   r   r   r   r   r   r   7   r'   z#test_asarray_copy.<locals>.<lambda>c                      s   t  tjjdS )Nr7   )r   r"   	_CopyMode	IF_NEEDEDr   r9   r   r   r   9   r$   )r   r   r"   r:   ALWAYSNEVERr   NotImplementedError)br   r9   r   test_asarray_copy(   s$   



r@   c                   C   :   t ddd ttdd  ttdd  ttdd  d S )Nr   r)   r*   c                   S      t dddS r-   r   r   r   r   r   r   >   r'   z$test_arange_errors.<locals>.<lambda>c                   S      t dtdS r/   )r   r0   r   r   r   r   r   ?   r'   c                   S   rB   r1   rC   r   r   r   r   r   @   r'   )r   r   r4   r   r   r   r   test_arange_errors<      rE   c                   C   rA   )Nr   r)   r*   c                   S   rB   NrG   r.   r*   r   r   r   r   r   r   E   r'   z#test_empty_errors.<locals>.<lambda>c                   S   rD   NrG   r    )r   r0   r   r   r   r   r   F   r'   c                   S   rB   NrG   r2   r    rI   r   r   r   r   r   G   r'   )r   r   r4   r   r   r   r   test_empty_errorsC   rF   rL   c                   C   >   t tddd ttdd  ttdd  ttdd  d S )Nr   r)   r*   c                   S      t tdddS r-   r   r   r   r   r   r   r   L   r$   z(test_empty_like_errors.<locals>.<lambda>c                   S      t tdtdS r/   )r   r   r0   r   r   r   r   r   M   r$   c                   S   rN   r1   rO   r   r   r   r   r   N   r$   )r   r   r   r4   r   r   r   r   test_empty_like_errorsJ      rQ   c                   C   rA   )Nr   r)   r*   c                   S   rB   r-   r	   r   r   r   r   r   S   r'   z!test_eye_errors.<locals>.<lambda>c                   S   rD   r/   )r	   r0   r   r   r   r   r   T   r'   c                   S   rB   r1   rS   r   r   r   r   r   U   r'   )r	   r   r4   r   r   r   r   test_eye_errorsQ   rF   rT   c                   C   s<   t dddd ttdd  ttdd  ttdd  d S )	NrG   r   r)   r*   c                   S      t ddddS )NrG   r   r.   r*   r
   r   r   r   r   r   Z   r(   z"test_full_errors.<locals>.<lambda>c                   S   s   t ddtdS )NrG   r   r    )r
   r0   r   r   r   r   r   [   r(   c                   S   rU   )NrG   r   r2   r    rV   r   r   r   r   r   \   r(   )r
   r   r4   r   r   r   r   test_full_errorsX   s   rW   c                   C   s@   t tdddd ttdd  ttdd  ttdd  d S )	Nr   r   r)   r*   c                   S      t tddddS )Nr   r   r.   r*   r   r   r   r   r   r   r   a       z'test_full_like_errors.<locals>.<lambda>c                   S   s   t tddtdS )Nr   r   r    )r   r   r0   r   r   r   r   r   b   rZ   c                   S   rX   )Nr   r   r2   r    rY   r   r   r   r   r   c   rZ   )r   r   r   r4   r   r   r   r   test_full_like_errors_   s   r[   c                   C   s>   t ddddd ttdd  ttdd  ttd	d  d S )
Nr   r   
   r)   r*   c                   S      t dddddS )Nr   r   r\   r.   r*   r   r   r   r   r   r   h   r$   z&test_linspace_errors.<locals>.<lambda>c                   S   s   t dddtdS )Nr   r   r\   r    )r   floatr   r   r   r   r   i   r$   c                   S   r]   )Nr   r   r\   fr    r^   r   r   r   r   r   j   r$   )r   r   r4   r   r   r   r   test_linspace_errorsf   rR   ra   c                   C   rA   )NrG   r)   r*   c                   S   rB   rH   r   r   r   r   r   r   o   r'   z"test_ones_errors.<locals>.<lambda>c                   S   rD   rJ   )r   r0   r   r   r   r   r   p   r'   c                   S   rB   rK   rb   r   r   r   r   r   q   r'   )r   r   r4   r   r   r   r   test_ones_errorsm   rF   rc   c                   C   rM   )Nr   r)   r*   c                   S   rN   r-   r   r   r   r   r   r   r   v   r$   z'test_ones_like_errors.<locals>.<lambda>c                   S   rP   r/   )r   r   r0   r   r   r   r   r   w   r$   c                   S   rN   r1   rd   r   r   r   r   r   x   r$   )r   r   r   r4   r   r   r   r   test_ones_like_errorst   rR   re   c                   C   rA   )NrG   r)   r*   c                   S   rB   rH   r   r   r   r   r   r   }   r'   z#test_zeros_errors.<locals>.<lambda>c                   S   rD   rJ   )r   r0   r   r   r   r   r   ~   r'   c                   S   rB   rK   rf   r   r   r   r   r      r'   )r   r   r4   r   r   r   r   test_zeros_errors{   rF   rg   c                   C   rM   )Nr   r)   r*   c                   S   rN   r-   r   r   r   r   r   r   r      r$   z(test_zeros_like_errors.<locals>.<lambda>c                   S   rP   r/   )r   r   r0   r   r   r   r   r      r$   c                   S   rN   r1   rh   r   r   r   r   r      r$   )r   r   r   r4   r   r   r   r   test_zeros_like_errors   rR   ri   c                   C   sH   t   t tdgtd t tdgtdtdgtd ttdd  d S )Nr   r    c                   S   s   t tdgtdtdgtdS r   )r   r   r   r   r   r   r   r   r      s    z,test_meshgrid_dtype_errors.<locals>.<lambda>)r   r   r   r   r4   r   r   r   r   test_meshgrid_dtype_errors   s   rj   )'numpy.testingr   numpyr"    r   _creation_functionsr   r   r   r   r	   r
   r   r   r   r   r   r   r   _dtypesr   r   _array_objectr   r6   r@   rE   rL   rQ   rT   rW   r[   ra   rc   re   rg   ri   rj   r   r   r   r   <module>   s(    <