*}Zq H d dl Z d dlZd dlZd dlmZmZ G d d ZdS ) N)assert_array_equalIS_PYPYc . e Zd Zej ed d Zd Zd Z ej ed d Z
ej de j
e j e j e j e j e j e j e j e j e j e j e j e j e j g d Zd Z d
Z d Z ej d e d
d Z d Z!d Z"d Z#d Z$d Z%d Z&dS )
TestDLPackzPyPy can't get refcounts.)reasonc t j d }| }t j | dk sJ ~t j | dk sJ d S N )nparange
__dlpack__sys getrefcountselfxys p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/tests/test_dlpack.pytest_dunder_dlpack_refcountz&TestDLPack.test_dunder_dlpack_refcount s[ IaLL
LLNNq!!Q&&&&
q!!Q&&&&&& c t j d }| d t j t
5 | d d d d d S # 1 swxY w Y d S )Nr
)stream )r
r r pytestraises RuntimeErrorr r s r test_dunder_dlpack_streamz$TestDLPack.test_dunder_dlpack_stream s IaLL D !!!
]<
(
( # #
LLL " " " # # # # # # # # # # # # # # # # # #s A((A,/A,c * t j dt j fdt j fg }t j d| }|d }t j t 5 t j | d d d d S # 1 swxY w Y d S )Nintchar)r
dtype) r
r% int32int8zerosr r BufferError from_dlpack)r dtr zs r %test_strides_not_multiple_of_itemsizez0TestDLPack.test_strides_not_multiple_of_itemsize s
Xrx(627*;<
=
= HT $ $ $
eH
];
'
' N1 s &BB B c t j d }t j | }t j | dk sJ ~t j | dk sJ d S r )r
r r* r r r s r test_from_dlpack_refcountz$TestDLPack.test_from_dlpack_refcount s] IaLL N1 q!!Q&&&&
q!!Q&&&&&&r r% c t j d | }t j | }|j |j k sJ t || d S Nr
)r
r astyper* r% r )r r% r r s r test_dtype_passthroughz!TestDLPack.test_dtype_passthrough( sY
IaLL & & N1 w!'!!!!1a r c t j t j d }t j t
5 t j | d d d d S # 1 swxY w Y d S )Nz
2021-05-27)r
asarray
datetime64r r r) r* r s r test_invalid_dtypez TestDLPack.test_invalid_dtype6 s Jr}\22 3 3
];
'
' N1 s A""A&)A&c
t j d }t j d| }t j t 5 t j | d d d d S # 1 swxY w Y d S )Nz=i8r
r$ )r
r% newbyteorderr r r r) r* )r r+ r s r test_invalid_byte_swappingz%TestDLPack.test_invalid_byte_swapping< s
Xe__
)
)
+
+ Iar " " "
];
'
' N1 s A88A<?A<c D t j d d }|d }t |t j | |d d df }t |t j | |dd d f }t |t j | |d }t |t j | t j | }t |t j | d S )N )r
r
r r )r
r reshaper r* diagonalcopy)r r y1y2y3y4y5s r test_non_contiguousz TestDLPack.test_non_contiguousC s IbMM ! !& ) )
qT2r~b 1 1222
qqq!tW2r~b 1 1222
q!!!tW2r~b 1 1222
qT2r~b 1 1222
[^^
"
"2r~b 1 122222r ndim! c d|z }t j |t j }|t j | j k sJ d S )N)r r$ )r
r( float64r* shape)r rF rJ r s r test_higher_dimszTestDLPack.test_higher_dimsU sE t HU"* - - -q))///////r c t j d }| dk sJ t j | }| dk sJ |d d d }| dk sJ d S )Nr
)r r r )r
r __dlpack_device__r* )r r r r, s r test_dlpack_devicez TestDLPack.test_dlpack_device\ s IaLL""$$.... N1 ""$$....
cccF""$$......r c ` t j d }| }t r1 )r
r r r )r r _s r dlpack_deleter_exceptionz#TestDLPack.dlpack_deleter_exceptiond s" IaLL
LLNNr c t j t 5 | d d d d S # 1 swxY w Y d S )N)r r r rQ )r s r test_dlpack_destructor_exceptionz+TestDLPack.test_dlpack_destructor_exceptioni s
]<
(
( , , ) ) + + + , , , , , , , , , , , , , , , , , ,s <A A c t j d }d|j _ t j t 5 | d d d d S # 1 swxY w Y d S )Nr
F)r
r flags writeabler r r) r r s r
test_readonlyzTestDLPack.test_readonlym s IaLL !
];
'
'
LLNNN s A A #A c v t j d }t j | }t || d S )Ng ?)r
arrayr* r r s r
test_ndim0zTestDLPack.test_ndim0s s4 HSMM N1 1a r c t j dddt j dt j d }t j | }t || d S )Nf8)
r
r ) P i r$ F)r% rJ stridesbufferorder)r
ndarrayonesuint8r* r r s r test_size1dims_arraysz TestDLPack.test_size1dims_arraysx s\ JTZ gd"( ; ; ;3
H
H
H N1 1a r N)'__name__
__module__ __qualname__r markskipifr r r r- r/ parametrizer
bool_r' int16r& int64rg uint16uint32uint64float16float32rI complex64
complex128r3 r7 r: rE rangerK rN rQ rS rW rZ rh r r r r s [(CDD' ' ED'# # # [(CDD' ' ED' [ W
28RX
")RY
BJ
bm ' ! ! ! 3 3 3$ [ VUU2YY//0 0 0/0 / / /
, , , ! ! !
! ! ! ! !r r )r r numpyr
numpy.testingr r r rz r r r} sw
5 5 5 5 5 5 5 5t! t! t! t! t! t! t! t! t! t!r