o
    Je                     @   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   {/home/seandotau/au.memorygallery.club/gallery/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      ?Zdtype)r   npZ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Z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   r3   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 )NFr4   r   r   r   r   r   r   7   r%   z#test_asarray_copy.<locals>.<lambda>c                      s   t  tjjdS )Nr4   )r   r!   	_CopyModeZ	IF_NEEDEDr   r6   r   r   r   9   r"   )r   r   r!   r7   ZALWAYSZNEVERr   NotImplementedError)br   r6   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   r-   r   r   r   r   r   ?   r%   c                   S   r<   r.   r=   r   r   r   r   r   @   r%   )r   r   r1   r   r   r   r   test_arange_errors<      r?   c                   C   r;   )Nr   r'   r(   c                   S   r<   NrA   r+   r(   r   r   r   r   r   r   E   r%   z#test_empty_errors.<locals>.<lambda>c                   S   r>   NrA   r    )r   r-   r   r   r   r   r   F   r%   c                   S   r<   NrA   r/   r    rC   r   r   r   r   r   G   r%   )r   r   r1   r   r   r   r   test_empty_errorsC   r@   rF   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   r-   r   r   r   r   r   M   r"   c                   S   rH   r.   rI   r   r   r   r   r   N   r"   )r   r   r   r1   r   r   r   r   test_empty_like_errorsJ      rK   c                   C   r;   )Nr   r'   r(   c                   S   r<   r*   r	   r   r   r   r   r   S   r%   z!test_eye_errors.<locals>.<lambda>c                   S   r>   r,   )r	   r-   r   r   r   r   r   T   r%   c                   S   r<   r.   rM   r   r   r   r   r   U   r%   )r	   r   r1   r   r   r   r   test_eye_errorsQ   r@   rN   c                   C   s<   t dddd ttdd  ttdd  ttdd  d S )	NrA   r   r'   r(   c                   S      t ddddS )NrA   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 )NrA   r   r    )r
   r-   r   r   r   r   r   [   r&   c                   S   rO   )NrA   r   r/   r    rP   r   r   r   r   r   \   r&   )r
   r   r1   r   r   r   r   test_full_errorsX   s   rQ   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   r-   r   r   r   r   r   b   rT   c                   S   rR   )Nr   r   r/   r    rS   r   r   r   r   r   c   rT   )r   r   r   r1   r   r   r   r   test_full_like_errors_   s   rU   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   rV   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   rV   r    )r   floatr   r   r   r   r   i   r"   c                   S   rW   )Nr   r   rV   fr    rX   r   r   r   r   r   j   r"   )r   r   r1   r   r   r   r   test_linspace_errorsf   rL   r[   c                   C   r;   )NrA   r'   r(   c                   S   r<   rB   r   r   r   r   r   r   o   r%   z"test_ones_errors.<locals>.<lambda>c                   S   r>   rD   )r   r-   r   r   r   r   r   p   r%   c                   S   r<   rE   r\   r   r   r   r   r   q   r%   )r   r   r1   r   r   r   r   test_ones_errorsm   r@   r]   c                   C   rG   )Nr   r'   r(   c                   S   rH   r*   r   r   r   r   r   r   r   v   r"   z'test_ones_like_errors.<locals>.<lambda>c                   S   rJ   r,   )r   r   r-   r   r   r   r   r   w   r"   c                   S   rH   r.   r^   r   r   r   r   r   x   r"   )r   r   r   r1   r   r   r   r   test_ones_like_errorst   rL   r_   c                   C   r;   )NrA   r'   r(   c                   S   r<   rB   r   r   r   r   r   r   }   r%   z#test_zeros_errors.<locals>.<lambda>c                   S   r>   rD   )r   r-   r   r   r   r   r   ~   r%   c                   S   r<   rE   r`   r   r   r   r   r      r%   )r   r   r1   r   r   r   r   test_zeros_errors{   r@   ra   c                   C   rG   )Nr   r'   r(   c                   S   rH   r*   r   r   r   r   r   r   r      r"   z(test_zeros_like_errors.<locals>.<lambda>c                   S   rJ   r,   )r   r   r-   r   r   r   r   r      r"   c                   S   rH   r.   rb   r   r   r   r   r      r"   )r   r   r   r1   r   r   r   r   test_zeros_like_errors   rL   rc   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   r1   r   r   r   r   test_meshgrid_dtype_errors   s   rd   )'Znumpy.testingr   numpyr!    r   Z_creation_functionsr   r   r   r   r	   r
   r   r   r   r   r   r   r   Z_dtypesr   r   Z_array_objectr   r3   r:   r?   rF   rK   rN   rQ   rU   r[   r]   r_   ra   rc   rd   r   r   r   r   <module>   s(    <