U
|
O ±d- ã @ sì d dl mZ d dlZddlmZ ddlmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZ ddlmZmZ ddlmZ dd Zd
|
d Zd d Zdd Z dd Z dd Z dd Z dd Z dd Z!dd Z"d d Z#d d Z$d d! Z%d"d# Z&dS )$é )Ú 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 tdd t tdd t tdd t tdd t tdd tdgdd t td
|
d t td d t td d d S ) Nc S s
|
t dgS )Né r © r r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/tests/test_creation_functions.pyÚ<lambda> ó z%test_asarray_errors.<locals>.<lambda>c S s
|
t dgS )NÚa©r r r r r r r c S s t dgtjdS ©Nç ð?©Zdtype)r ÚnpÚfloat16r r r r r r c S s t dd S ©Nr éd r r r r r r r c S s t dd gS r" r r r r r r r r Úcpu©Zdevicec S s t dgddS ©Nr Úgpur% r r r r r r " r c S s t dgtdS ©Nr r )r Úintr r r r r $ r c S s t dgddS ©Nr Úir r r r r r r % r )r Ú TypeErrorÚ
|
ValueErrorÚ OverflowErrorr r r r r Útest_asarray_errors s r/ c sî t dg t dd} d d< t| d dks2tt d dksFtt dg t tjjd} d d< t| d dks|tt d dkstt dg t tjjd} d d< t| d dksÆttt fdd tt fdd d S )Nr T©Úcopyr c s t ddS )NFr0 r r ©r r r r 7 r z#test_asarray_copy.<locals>.<lambda>c s t tjjdS )Nr0 )r r Ú _CopyModeZ IF_NEEDEDr r2 r r r 9 r ) r r ÚAssertionErrorr r3 ZALWAYSZNEVERr ÚNotImplementedError)Úbr r2 r Útest_asarray_copy( s$
|
|
|
|
ÿr7 c C s: t ddd ttdd ttdd ttdd d S )Nr r$ r% c S s t dddS r& ©r r r r r r > r z$test_arange_errors.<locals>.<lambda>c S s t dtdS r( )r r) r r r r r ? r c S s t dddS r* r8 r r r r r @ r )r r r- r r r r Útest_arange_errors< s r9 c C s: t ddd ttdd ttdd ttdd d S )N©r r$ r% c S s t dddS ©Nr: r' r% ©r r r r r r E r z#test_empty_errors.<locals>.<lambda>c S s t dtdS ©Nr: r )r r) r r r r r F r c S s t dddS ©Nr: r+ r r< r r r r r G r )r r r- r r r r Útest_empty_errorsC s r? c C s> t tddd ttdd ttdd ttdd d S )Nr r$ r% c S s t tdddS r& ©r r r r r r r L r z(test_empty_like_errors.<locals>.<lambda>c S s t tdtdS r( )r r r) r r r r r M r c S s t tdddS r* r@ r r r r r N r )r r r r- r r r r Útest_empty_like_errorsJ s rA c C s: t ddd ttdd ttdd ttdd d S )Nr r$ r% c S s t dddS r& ©r r r r r r S r z!test_eye_errors.<locals>.<lambda>c S s t dtdS r( )r r) r r r r r T r c S s t dddS r* rB r r r r r U r )r r r- r r r r Útest_eye_errorsQ s rC c C s< t dddd ttdd ttdd ttdd d S ) Nr: r r$ r% c S s t ddddS )Nr: r r' r% ©r
|
r r r r r Z r z"test_full_errors.<locals>.<lambda>c S s t ddtdS )Nr: r r )r
|
r) r r r r r [ r c S s t ddddS )Nr: r r+ r rD r r r r r \ r )r
|
r r- r r r r Útest_full_errorsX s rE c C s@ t tdddd ttdd ttdd ttdd d S ) Nr r r$ r% c S s t tddddS )Nr r r' r% ©r r r r r r r a r z'test_full_like_errors.<locals>.<lambda>c S s t tddtdS )Nr r r )r r r) r r r r r b r c S s t tddddS )Nr r r+ r rF r r r r r c r )r r r r- r r r r Útest_full_like_errors_ s rG c C s> t ddddd ttdd ttdd ttd d d S )
|
Nr r é
|
r$ r% c S s t dddddS )Nr r rH r' r% ©r r r r r r h r z&test_linspace_errors.<locals>.<lambda>c S s t dddtdS )Nr r rH r )r Úfloatr r r r r i r c S s t dddddS )Nr r rH Úfr rI r r r r r j r )r r r- r r r r Útest_linspace_errorsf s rL c C s: t ddd ttdd ttdd ttdd d S )Nr: r$ r% c S s t dddS r; ©r r r r r r o r z"test_ones_errors.<locals>.<lambda>c S s t dtdS r= )r r) r r r r r p r c S s t dddS r> rM r r r r r q r )r r r- r r r r Útest_ones_errorsm s rN c C s> t tddd ttdd ttdd ttdd d S )Nr r$ r% c S s t tdddS r& ©r r r r r r r v r z'test_ones_like_errors.<locals>.<lambda>c S s t tdtdS r( )r r r) r r r r r w r c S s t tdddS r* rO r r r r r x r )r r r r- r r r r Útest_ones_like_errorst s rP c C s: t ddd ttdd ttdd ttdd d S )Nr: r$ r% c S s t dddS r; ©r r r r r r } r z#test_zeros_errors.<locals>.<lambda>c S s t dtdS r= )r r) r r r r r ~ r c S s t dddS r> rQ r r r r r r )r r r- r r r r Útest_zeros_errors{ s rR c C s> t tddd ttdd ttdd ttdd d S )Nr r$ r% c S s t tdddS r& ©r r r r r r r r z(test_zeros_like_errors.<locals>.<lambda>c S s t tdtdS r( )r r r) r r r r r
r c S s t tdddS r* rS r r r r r r )r r r r- r r r r Útest_zeros_like_errors s rT c C sH t t tdgtd t tdgtdtdgtd ttdd d S )Nr r c S s t tdgtdtdgtdS r )r r r r r r r r r r z,test_meshgrid_dtype_errors.<locals>.<lambda>)r r r r r- r r r r Útest_meshgrid_dtype_errors s rU )'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 r/ r7 r9 r? rA rC rE rG rL rN rP rR rT rU r r r r Ú<module> s& <
|
