- d dl mZ d dlZddlmZ ddlmZmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ ddlmZmZ ddlmZ d Zd Zd
Zd Z d Z d
Z d Z d Z d Z!d Z"d Z#d Z$d Z%d Z&dS ) )
assert_raisesN )all)
asarrayarangeempty
empty_likeeyefull full_likelinspacemeshgridones ones_likezeros
zeros_like)float32float64Arrayc t t d t t d t t d t t d t t d t dgd t t d t t d
t t d d S ) Nc " t dg S )N r /builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/tests/test_creation_functions.pyz%test_asarray_errors.. s UA3ZZ r c " t dg S )Nar r r r r z%test_asarray_errors.. s WcU^^ r c : t dgt j S N ?dtype)r npfloat16r r r r z%test_asarray_errors.. s gse2:&F&F&F r c & t ddz S Nr d r r r r r z%test_asarray_errors.. s C r c ( t ddz g S r) r r r r r z%test_asarray_errors.. s WafX%6%6 r r cpudevicec & t dgd S Nr gpur- r r r r r z%test_asarray_errors.." s gqc%&@&@&@ r c 0 t dgt S Nr r$ )r intr r r r z%test_asarray_errors..$ s gqc&=&=&= r c & t dgd S Nr ir$ r r r r r z%test_asarray_errors..% s gqc&=&=&= r )r TypeError
ValueError
OverflowErrorr r r r test_asarray_errorsr; s ) / /000) 3 3444* F FGGG-!8!8999 ) 6 6777 QC * @ @AAA* = =>>>* = =>>>>>r c t dg t d } dd< t | d dk sJ t d dk sJ t dg t t j j } dd< t | d dk sJ t d dk sJ t dg t t j j } dd< t | d dk sJ t
t fd t
t fd d S )Nr Tcopyr c & t d S )NFr= r r s r r z#test_asarray_copy..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..9 s '!",*@AAA r )r r r& rB ALWAYSNEVERr NotImplementedError)br s @r test_asarray_copyrH ( sM A A AaD qtqy>>> qtqy>>> A +,,,A AaD qtqy>>> qtqy>>> A *+++A AaD qtqy>>>%'E'E'E'EFFF%AAAAC C C C Cr c t dd t t d t t d t t d d S )Nr r, r- c $ t dd S r0 r r r r r z$test_arange_errors..> s fQu&=&=&= r c . t dt S r3 )r r4 r r r r z$test_arange_errors..? s fQc&:&:&: r c $ t dd S r6 rK r r r r z$test_arange_errors..@ s fQc&:&:&: r )r r r9 r r r test_arange_errorsrN < sW
1U* = =>>>* : :;;;* : :;;;;;r c t dd t t d t t d t t d d S )Nr r, r- c $ t dd S NrP r1 r- r r r r r z#test_empty_errors..E eD&?&?&? r c . t dt S NrP r$ )r r4 r r r r z#test_empty_errors..F eD&<&<&< r c $ t dd S NrP r7 r$ rS r r r r z#test_empty_errors..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..L jE&J&J&J r c H t t d t S r3 )r r r4 r r r r z(test_empty_like_errors..M j3&G&G&G r c > t t d d S r6 r_ r r r r z(test_empty_like_errors..N j3&G&G&G r )r r r r9 r r r test_empty_like_errorsre J ] wqzz%((((* J JKKK* G GHHH* G GHHHHHr c t dd t t d t t d t t d d S )Nr r, r- c $ t dd S r0 r
r r r r z!test_eye_errors..S s c!E&:&:&: r c . t dt S r3 )r
r4 r r r r z!test_eye_errors..T s c!3&7&7&7 r c $ t dd S r6 ri r r r r z!test_eye_errors..U s c!3&7&7&7 r )r
r r9 r r r test_eye_errorsrl Q sW %* : :;;;* 7 7888* 7 788888r c t ddd t t d t t d t t d d S )NrP r r, r- c & t ddd S )NrP r r1 r- r r r r r z"test_full_errors..Z s d45&A&A&A r c 0 t ddt S )NrP r r$ )r r4 r r r r z"test_full_errors..[ s d4#&>&>&> r c & t ddd S )NrP r r7 r$ ro r r r r z"test_full_errors..\ s d4#&>&>&> r )r r r9 r r r test_full_errorsrr X sY q * A ABBB* > >???* > >?????r c t t d dd t t d t t d t t d d S )Nr r r, r- c @ t t d dd S )Nr r r1 r- r r r r r r z'test_full_like_errors..a s i
Ae&L&L&L r c J t t d dt S )Nr r r$ )r r r4 r r r r z'test_full_like_errors..b s i
AS&I&I&I r c @ t t d dd S )Nr r r7 r$ ru r r r r z'test_full_like_errors..c s i
AS&I&I&I r )r r r r9 r r r test_full_like_errorsrx _ s_
gajj!E***** L LMMM* I IJJJ* I IJJJJJr c t dddd t t d t t d t t d d S ) Nr r
r, r- c ( t dddd S )Nr r rz r1 r- r
r r r r z&test_linspace_errors..h s hq!R&F&F&F r c 2 t dddt S )Nr r rz r$ )r
floatr r r r z&test_linspace_errors..i s hq!Ru&E&E&E r c ( t dddd S )Nr r rz fr$ r| r r r r z&test_linspace_errors..j s hq!Rs&C&C&C r )r
r r9 r r r test_linspace_errorsr f s[ Q2e$$$$* F FGGG* E EFFF* C CDDDDDr c t dd t t d t t d t t d d S )NrP r, r- c $ t dd S rR r r r r r z"test_ones_errors..o s d4&>&>&> r c . t dt S rV )r r4 r r r r z"test_ones_errors..p s d4s&;&;&; r c $ t dd S rY r r r r r z"test_ones_errors..q s d4s&;&;&; r )r r r9 r r r test_ones_errorsr m sW e * > >???* ; ;<<<* ; ;<<<< t t d d S r0 r r r r r r z'test_ones_like_errors..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..w s i
#&F&F&F r c > t t d d S r6 r r r r r z'test_ones_like_errors..x s i
#&F&F&F r )r r r r9 r r r test_ones_like_errorsr t s]
gajj''''* I IJJJ* F FGGG* F FGGGGGr c t dd t t d t t d t t d d S )NrP r, r- c $ t dd S rR r r r r r z#test_zeros_errors..} rT r c . t dt S rV )r r4 r r r r z#test_zeros_errors..~ rW r c $ t dd S rY r r r r r z#test_zeros_errors.. 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.. r` r c H t t d t S r3 )r r r4 r r r r z(test_zeros_like_errors.. rb r c > t t d d S r6 r r r r r z(test_zeros_like_errors.. rd r )r r r r9 r r r test_zeros_like_errorsr rf r c
t t t dgt t t dgt t dgt t t d d S )Nr# r$ c v t t dgt t dgt S r" )r r r r r r r r z,test_meshgrid_dtype_errors.. s1 hwt7/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+GHHH* j jkkkkkr )'
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 r s ' ' ' ' ' ' ' & & & & & & & ! ! ! ! ! !
?
?
? C C C(< < <> > >I I I9 9 9@ @ @K K KE E E= = =H H H> > >I I I l l l l lr