File: //usr/local/lib/python3.10/test/__pycache__/test_picklebuffer.cpython-310.opt-1.pyc
o
�i� � @ sn d Z ddlZddlmZ ddlZddlZddlmZ G dd� de�Z G dd� dej
�Z e d kr5e�
� dS dS )
z[Unit tests for the PickleBuffer object.
Pickling tests themselves are in pickletester.py.
� N)� PickleBuffer)�
import_helperc @ s e Zd ZdS )�BN)�__name__�
__module__� __qualname__� r r �3/usr/local/lib/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�str� 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 � )r7 r6 )r6 r4 �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 )rC )r% ZsubTestrD )r rC 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 )Nr3 r7 )r7 z<hr9 s � )r: r7 s )r: r6 r; i� r r8 s � )r r< r= r> r? rD )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� BufferErrorrB )r rC 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 ) Nr3 rF )rF r8 r9 r: r4 r5 )r r< r= r> r? rI )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 rB r* r r r �test_raw_released� s
"�z"PickleBufferTest.test_raw_releasedN)r r r r r! r' r+ r2 rA rD rE rG rI rJ rK r r r r r
s
r
�__main__)�__doc__r0 �pickler r. ZunittestZ test.supportr r# r ZTestCaser
r �mainr r r r �<module> s �