Your IP : 216.73.216.213
a
XC?h� � @ sj d Z ddlZddlmZ ddlZddlZddlmZ G dd� de�Z G dd� dej
�Z e d krfe�
� dS )
z[Unit tests for the PickleBuffer object.
Pickling tests themselves are in pickletester.py.
� N)� PickleBuffer)�supportc @ s e Zd ZdS )�BN)�__name__�
__module__� __qualname__� r r �;/opt/alt/python39/lib64/python3.9/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 |���}t |���}| �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j|j� | �|j |j � | �|�
� |�
� � W d � n1 s�0 Y W d � n1 s�0 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 s&0 Y | � t�� td� W d � n1 sX0 Y td�}|�� | � t�� t|� W d � n1 s�0 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 sB0 Y t td��}| �dt|�� t|��$}| �|j� d|d< W d � n1 s�0 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 s80 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 sL t |�}|�� �(}| �|t� | �||� W d � n1 s>0 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 � q 1 s>0 Y q d 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 s@ t |�}| �td�� |�� W d � n1 s20 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 sF t d�}|�� | �t��}|�� W d � n1 s80 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__)�__doc__r/ �pickler r- Zunittest�testr r# r ZTestCaser
r �mainr r r r �<module> s