U
|
O ±d* ã @ s: d dl Z d dlZd dlZd dlmZmZ G dd dZdS )é N)Úassert_array_equalÚIS_PYPYc @ sø e Zd Zejjedddd Zdd Zdd Z ejjeddd d
|
Z
|
ej d e j e je je je je je je je je je je je jg ¡d d Zdd Zdd Z dd Z ej de d¡dd Z dd Z dd Z!d d Z"d d Z#d d! Z$d"d# Z%d$S )%Ú
|
TestDLPackzPyPy can't get refcounts.)Úreasonc C s< t d¡}| ¡ }t |¡dks$t~t |¡dks8td S ©Né é é )ÚnpÚarangeÚ
|
__dlpack__ÚsysÚ getrefcountÚAssertionError©ÚselfÚxÚy© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_dlpack.pyÚtest_dunder_dlpack_refcount s
|
|
z&TestDLPack.test_dunder_dlpack_refcountc C s< t d¡}|jd d t t¡ |jdd W 5 Q R X d S )Nr )Ústreamé )r
|
r r ÚpytestÚraisesÚ RuntimeError©r r r r r Útest_dunder_dlpack_stream s
|
z$TestDLPack.test_dunder_dlpack_streamc C sT t dt jfdt jfg¡}t jd|d}|d }t t¡ t |¡ W 5 Q R X d S )NÚintÚchar)r ©Údtype) r
|
r! Úint32Úint8Úzerosr r Ú BufferErrorÚ from_dlpack)r Údtr Úzr r r Ú%test_strides_not_multiple_of_itemsize s
|
z0TestDLPack.test_strides_not_multiple_of_itemsizec C s> t d¡}t |¡}t |¡dks&t~t |¡dks:td S r )r
|
r r& r r r r r r r Útest_from_dlpack_refcount s
|
|
|
z$TestDLPack.test_from_dlpack_refcountr! c C s6 t jd|d}t |¡}|j|jks(tt|| d S )Nr r )r
|
r r& r! r r )r r! r r r r r Útest_dtype_passthrough( s
|
z!TestDLPack.test_dtype_passthroughc C s4 t t d¡¡}t t¡ t |¡ W 5 Q R X d S )Nz
|
2021-05-27)r
|
ZasarrayZ
|
datetime64r r r% r& r r r r Útest_invalid_dtype5 s z TestDLPack.test_invalid_dtypec C s@ t d¡ ¡ }t jd|d}t t¡ t |¡ W 5 Q R X d S )Nz=i8r r )r
|
r! Z newbyteorderr r r r% r& )r r' r r r r Útest_invalid_byte_swapping; s z%TestDLPack.test_invalid_byte_swappingc C s¢ t d¡ d¡}|d }t|t |¡ |d d
df }t|t |¡ |dd d
f }t|t |¡ |d }t|t |¡ t |¡ ¡ }t|t |¡ d S )Né )r r r r )r
|
r Zreshaper r& ZdiagonalÚcopy)r r Úy1Úy2Zy3Zy4Zy5r r r Útest_non_contiguousB s z TestDLPack.test_non_contiguousÚndimé! c C s0 d| }t j|t jd}|t |¡jks,td S )N)r r )r
|
r$ Úfloat64r& Úshaper )r r3 r6 r r r r Útest_higher_dimsT s zTestDLPack.test_higher_dimsc C sV t d¡}| ¡ dkstt |¡}| ¡ dks4t|d d d
}| ¡ dksRtd S )Nr )r r r )r
|
r Z__dlpack_device__r r& )r r r r( r r r Útest_dlpack_device[ s
|
|
z TestDLPack.test_dlpack_devicec C s t d¡}| ¡ }td S )Nr )r
|
r r r )r r Ú_r r r Údlpack_deleter_exceptionc s
|
z#TestDLPack.dlpack_deleter_exceptionc C s" t t¡ | ¡ W 5 Q R X d S )N)r r r r: )r r r r Ú test_dlpack_destructor_exceptionh s z+TestDLPack.test_dlpack_destructor_exceptionc C s4 t d¡}d|j_t t¡ | ¡ W 5 Q R X d S )Nr F)r
|
r ÚflagsZ writeabler r r% r r r r r Ú test_readonlyl s
|
zTestDLPack.test_readonlyc C s" t d¡}t |¡}t|| d S )Ng ð?)r
|
Úarrayr& r r r r r Ú
|
test_ndim0r s
|
|
zTestDLPack.test_ndim0c C s8 t jdddt jdt jddd}t |¡}t|| d S )NZf8)é
|
r r )é éP é iè r ÚF)r! r6 ÚstridesÚbufferÚorder)r
|
ZndarrayZonesÚuint8r& r r r r r Útest_size1dims_arraysw s
|
ÿ
|
z TestDLPack.test_size1dims_arraysN)&Ú__name__Ú
|
__module__Ú __qualname__r ÚmarkZskipifr r r r) r* Z parametrizer
|
r# Úint16r" Úint64rH Zuint16Zuint32Zuint64Úfloat16Úfloat32r5 Z complex64Z
|
complex128r+ r, r- r2 Úranger7 r8 r: r; r= r? rI r r r r r sB
|
|
ü
|
|
r )r r Únumpyr
|
Z numpy.testingr r r r r r r Ú<module> s
|
