Your IP : 216.73.216.85
U
i�f� � @ 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/python38/lib64/python3.8/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 5 Q R X W 5 Q R X 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 sl | � t�� t� W 5 Q R X | � t�� td� W 5 Q R X td�}|�� | � t�� t|� W 5 Q R X 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 5 Q R X t td��}| �dt|�� t|��}| �|j� d|d< W 5 Q R X | �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 sL t d�}|�� | �t��}t|� W 5 Q R X | �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 s8 t |�}|�� � }| �|t� | �||� W 5 Q R X d S r )r �rawZassertIsInstancer r )r �objr r r r r r � check_rawc s
zPickleBufferTest.check_rawc
C s: dt d�fD ](}| j|d�� | �||� W 5 Q R X 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 5 Q R X 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 s2 t d�}|�� | �t��}|�� W 5 Q R X 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- ZunittestZtestr r# r ZTestCaser
r �mainr r r r �<module> s