?????????? ????????? - ??????????????? - /usr/lib64/python3.9/site-packages/numpy/core/tests/__pycache__/test_dlpack.cpython-39.pyc
???????
a z[yc� � @ s: d dl Z d dlZd dlZd dlmZmZ G dd� d�ZdS )� N)�assert_array_equal�IS_PYPYc @ s� e Zd Zejjedd�dd� �Zdd� Zdd� Z ejjedd�d 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$J �~t�|�dks8J �d S �N� � � )�np�arange� __dlpack__�sys� getrefcount��self�x�y� r �B/usr/lib64/python3.9/site-packages/numpy/core/tests/test_dlpack.py�test_dunder_dlpack_refcount s z&TestDLPack.test_dunder_dlpack_refcountc C sP t �d�}|jd d� t�t�� |jdd� W d � n1 sB0 Y 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 sh t �dt jfdt jfg�}t jd|d�}|d }t�t�� t �|� W d � n1 sZ0 Y d S )N�int�char)r ��dtype) r r �int32�int8�zerosr r r � 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&J �~t�|�dks:J �d S 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(J �t||� d S )Nr r )r r r$ r r )r r r r r r r �test_dtype_passthrough( s z!TestDLPack.test_dtype_passthroughc C sH t �t �d��}t�t�� t �|� W d � n1 s:0 Y d S )Nz 2021-05-27)r ZasarrayZ datetime64r r � TypeErrorr$ r r r r �test_invalid_dtype5 s z TestDLPack.test_invalid_dtypec C sT t �d��� }t jd|d�}t�t�� t �|� W d � n1 sF0 Y 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 Zy1Zy2Zy3Zy4Zy5r r r �test_non_contiguousB s z TestDLPack.test_non_contiguous�ndim�! c C s0 d| }t j|t jd�}|t �|�jks,J �d S )N)r r )r r# �float64r$ �shape)r r0 r3 r r r r �test_higher_dimsT s zTestDLPack.test_higher_dimsc C sV t �d�}|�� dksJ �t �|�}|�� dks4J �|d d d� }|�� dksRJ �d S )Nr )r r r )r r Z__dlpack_device__r$ )r r r r& r r r �test_dlpack_device[ s z TestDLPack.test_dlpack_devicec C s t �d�}|�� }t�d S )Nr )r r r r )r r �_r r r �dlpack_deleter_exceptionc s z#TestDLPack.dlpack_deleter_exceptionc C s6 t �t�� | �� W d � n1 s(0 Y d S )N)r r r r7 )r r r r � test_dlpack_destructor_exceptionh s z+TestDLPack.test_dlpack_destructor_exceptionc C sH t �d�}d|j_t�t�� |�� W d � n1 s:0 Y 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 Zarrayr$ r r r r r � test_ndim0r s zTestDLPack.test_ndim0c C s8 t jdddt jdt jd�dd�}t �|�}t||� d S )NZf8)� r r )� �P � i� r �F)r r3 �strides�buffer�order)r ZndarrayZones�uint8r$ r r r r r �test_size1dims_arraysw s � z TestDLPack.test_size1dims_arraysN)&�__name__� __module__� __qualname__r ZmarkZskipifr r r r' r( Z parametrizer r" Zint16r! Zint64rD Zuint16Zuint32Zuint64Zfloat16Zfloat32r2 Z complex64Z complex128r) r+ r, r/ �ranger4 r5 r7 r8 r: r; rE r r r r r s0 � r )r r Znumpyr Z numpy.testingr r r r r r r �<module> s
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????