Your IP : 216.73.216.196
o
0�i� � @ sl d dl Z d dlmZ d dlZd dlZd dlmZ G dd� de�ZG dd� dej �Z
e dkr4e� � dS dS ) � N)� PickleBuffer)�
import_helperc @ s e Zd ZdS )�BN)�__name__�
__module__� __qualname__� r r �=/opt/alt/python310/lib64/python3.10/test/test_picklebuffer.pyr s r c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�PickleBufferTestc C s� t |��o}t |��R}| �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j |j � | �|�
� |�
� � W d � n1 s^w Y W d � d S W d � d S 1 svw Y d S �N) �
memoryview� assertEqual�nbytes�readonly�itemsize�shape�strides� c_contiguous� f_contiguous�format�tobytes)�self�pb�equiv�mZexpectedr r r �check_memoryview s
��"�z!PickleBufferTest.check_memoryviewc C s� | � t�� t� W d � n1 sw Y | � t�� td� W d � n1 s,w Y td�}|�� | � t��
t|� W d � d S 1 sNw Y d S )NZfoo� foo)� assertRaises� TypeErrorr r �release�
ValueError)r r r r r �test_constructor_failure! s �
�
"�z)PickleBufferTest.test_constructor_failurec C s� t d�}| �dt|�� t|��}| �|j� W d � n1 s!w Y t td��}| �dt|�� t|��}| �|j� d|d< W d � n1 sMw Y | �dt|�� d S )Nr �0 r s 0oo)r r
�bytesr �
assertTruer � bytearray� assertFalse)r r r r r r � test_basics, s
�
�z PickleBufferTest.test_basicsc C s` t d�}|�� | �t�� }t|� W d � n1 s w Y | �dt|j�� |�� d S )Nr z3operation forbidden on released PickleBuffer object)r r r r r ZassertIn�strZ exception�r r Zraisesr r r � test_release9 s
�� z PickleBufferTest.test_releasec C s< t d�}t|�}||_t�|�}~~t�� | �|� � d S �Nr )r r �cycle�weakref�ref�gcZcollectZ assertIsNone)r �br Zwpbr r r �
test_cycleC s
zPickleBufferTest.test_cyclec C s� t �d�j}|ttd��ddd�}| �|j� | �|j� t |�}| �
||� |d d d� }| �|j� | �|j� t |�}| �
||� |ttd��dddd �}| �|j� | �|j� t |�}| �
||� d S )
N� _testbuffer� �� � �<i�r r � )r6 r5 )r5 r3 �r r r ) r �
import_module�ndarray�list�ranger$ r r& r r r )r r< �arrr r r r �test_ndarray_2dL s z PickleBufferTest.test_ndarray_2dc C sN t |�}|�� �}| �|t� | �||� W d � d S 1 s w Y d S r )r �rawZassertIsInstancer r )r �objr r r r r r � check_rawc s
"�zPickleBufferTest.check_rawc C sN dt d�fD ] }| j|d�� | �||� W d � n1 s w Y qd S )Nr )rB )r% ZsubTestrC )r rB r r r �test_rawi s ���zPickleBufferTest.test_rawc C s� t �d�j}|ttd��ddd�}d}| �||� |ttd��ddd�}d }| �||� |ttd��dd
dd �}d }| �||� |d d
dd�}d}| �||� d S )Nr2 r6 )r6 z<hr8 s � )r9 r6 s )r9 r5 r: i� r r7 s � )r r; r< r= r> rC )r r< r? r r r r �test_raw_ndarrayn s � z!PickleBufferTest.test_raw_ndarrayc C sB t |�}| �td��
|�� W d � d S 1 sw Y d S )Nznon-contiguous)r ZassertRaisesRegex� BufferErrorrA )r rB r r r r �check_raw_non_contiguous� s
"�z)PickleBufferTest.check_raw_non_contiguousc C sd t �d�j}|ttd��ddd�d d d� }| �|� |ttd��ddd�d d d� }| �|� d S ) Nr2 rE )rE r7 r8 r9 r3 r4 )r r; r< r= r> rH )r r< r? r r r �test_raw_non_contiguous� s
z(PickleBufferTest.test_raw_non_contiguousc C sH t d�}|�� | �t��
}|�� W d � d S 1 s w Y d S r+ )r r r r rA r) r r r �test_raw_released� s
"�z"PickleBufferTest.test_raw_releasedN)r r r r r! r' r* r1 r@ rC rD rF rH rI rJ r r r r r
s
r
�__main__)
r/ �pickler r- ZunittestZ test.supportr r# r ZTestCaser
r �mainr r r r �<module> s �