File: //usr/local/lib/python3.7/test/__pycache__/test_memoryview.cpython-37.opt-2.pyc
B
��g�E � @ sL d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z G dd� d�Z
G dd� de
�Z G dd� de
�Z G dd � d �Z
G d
d � d �ZG d d
� d
�ZG dd� de je
e �ZG dd� de je
e �ZG dd� de jee �ZG dd� de jee �ZG dd� de jee �ZG dd� de jee �ZG dd� de j�Zed k�rHe �� dS ) � Nc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6S )7�AbstractMemoryTestss abcdefc C s | j S )N)� source_bytes)�self� r �0/usr/local/lib/python3.7/test/test_memoryview.py�_source s zAbstractMemoryTests._sourcec C s t d | j| jg�S )N)�filter�ro_type�rw_type)r r r r �_types s zAbstractMemoryTests._typesc s$ || j �}t�|�}| �|�� | �� d td�� | �� d t� | �� d td�� | �� d td�� | �� d td�� | �t � fdd�� | �t � fd d�� | �t � fd
d�� | �t � fd d�� | �t
� fd d�� | �t
� fd
d�� | �t
� fdd�� d � | �t�|�|� d S )Nr � a� � f���i����c s � d S )N� r r )�mr r �<lambda>' � z=AbstractMemoryTests.check_getitem_with_type.<locals>.<lambda>c s � d S )Ni����r r )r r r r ( r c s
� t j S )N)�sys�maxsizer )r r r r ) r c s � t j S )N)r r r )r r r r * r c s � d S )Nr r )r r r r , r c s � d S )Ng r r )r r r r - r c s � d S )N�ar r )r r r r . r ) r r � getrefcount�_view� assertEqual�ord�assertIsInstance�int� assertRaises�
IndexError� TypeError)r �tp�b� oldrefcountr )r r �check_getitem_with_type s"
z+AbstractMemoryTests.check_getitem_with_typec C s x| j D ]}| �|� qW d S )N)r r# )r r r r r � test_getitem2 s z AbstractMemoryTests.test_getitemc sN xH| j D ]>}|| j�}| �|�� | �t� �� fdd�tt� ��D �� qW d S )Nc s g | ] }� | �qS r r )�.0�i)r r r �
<listcomp>: s z1AbstractMemoryTests.test_iter.<locals>.<listcomp>)r r r r �list�range�len)r r r! r )r r � test_iter6 s
z AbstractMemoryTests.test_iterc s� | j s| �d� | � | j�}t�|�}| �|�� � fdd�}| �t|d� | �t|d� | �t|td�� d � | � t�|�|� d S )Nzno read-only type to testc s | � d< d S )Nr r )�value)r r r �setitemB s z:AbstractMemoryTests.test_setitem_readonly.<locals>.setitemr �A )
r �skipTestr r r r r r �
memoryviewr )r r! r" r- r )r r �test_setitem_readonly< s
z)AbstractMemoryTests.test_setitem_readonlyc s� | j s| �d� | j �| � | j�}t�|�}| �|�� td�� d< | ��|d� �d�� dd�< | ��|d� �d�� dd �< | ��|d
� �d �� dd�< | ��|d
� �d �� d d �< | ��|d � � d
d� � dd �< | ��|d� �d �� d d �< � dd � � d
d�< | ��|d� � �fdd�}| �t |dd� | �t |dd� | �t |tj
d� | �t |tj
d� | �t |dd� | �t |dd� | �t |t ddd�dfd� | �t |dt ddd�fd� | �t |dd� | �t |dd� t ddd�t ddd
�f}| �t
||d� � jdk�r tnt }| �||dd � | �||dd� | �t|t dd�d� | �t|t dd
�d� d � | �t�|�|� d S )Nzno writable type to test� 1r s 1bcdef� 0� s 0bcdefs 12� s 012defr s abcdef� r
s cdedefs ababcfc s �|�� | < d S )Nr )�keyr, )r r r r r- c s z:AbstractMemoryTests.test_setitem_writable.<locals>.setitemr r i����g )r r �cs ab)r
r/ r r r r r �_check_contentsr r r r �slice�NotImplementedError�format�
ValueErrorr )r r! r" r- Zslices�excr )r r r �test_setitem_writableJ sR
z)AbstractMemoryTests.test_setitem_writablec
C sd x^| j D ]T}|| j�}| �|�}| �t�� |d= W d Q R X | �t�� |dd�= W d Q R X qW d S )Nr4 � )r r r r r )r r r! r r r r � test_delitem~ s
z AbstractMemoryTests.test_delitemc s\ xV� j D ]L}� �|� j��}|�� }d�� fdd�dD ��}� �||� � �|t� qW d S )Nr c 3 s | ]}� � t|g��V qd S )N)� getitem_type�bytes)r% r8 )r r r � <genexpr>� s z3AbstractMemoryTests.test_tobytes.<locals>.<genexpr>s abcdef)r r r �tobytes�joinr r rC )r r r r! Zexpectedr )r r � test_tobytes� s z AbstractMemoryTests.test_tobytesc C s< x6| j D ],}| �|| j��}|�� }| �|td�� qW d S )Ns abcdef)r r r �tolistr r( )r r r �lr r r � test_tolist� s z AbstractMemoryTests.test_tolistc s� �x�| j D �]�}| �|| j���xz| j D ]p}| ��|d�k� | ��|d�k� | ��|d�k� | ��|d�k� | ��|d�k� | ��|d�k� q(W | ���k� | ���d d � k� | ��dd� �d d � k� | ��dd� �k� | ��dk� | ��dk� | �d�k� | �d�k� xj�dfD ]^� | �t� �fdd �� | �t� �fd
d �� | �t� �fd d �� | �t� �fd d �� �q6W q
W d S )
Ns abcdefs abcdes abcde1r r r
Zabcdefc s �� k S )Nr r )r8 r r r r � r z2AbstractMemoryTests.test_compare.<locals>.<lambda>c s � �kS )Nr r )r8 r r r r � r c s �� kS )Nr r )r8 r r r r � r c s � �kS )Nr r )r8 r r r r � r )r r r �
assertTrueZ assertFalser r )r r Ztp_compr )r8 r r � test_compare� s, z AbstractMemoryTests.test_comparec C s� | � || j��}| �|j| j� | �|j| j� | �|jd� | �|jd� | �t|�d� | �|j| jf� | �|j d� |S )Nr4 )r r r )
r r r r< �itemsize�ndim�shaper* �strides�
suboffsets)r r r r r r �check_attributes_with_type� s z.AbstractMemoryTests.check_attributes_with_typec C s. | j s| �d� | �| j �}| �|jd� d S )Nzno read-only type to testT)r r/ rR r �readonly)r r r r r �test_attributes_readonly� s
z,AbstractMemoryTests.test_attributes_readonlyc C s. | j s| �d� | �| j �}| �|jd� d S )Nzno writable type to testF)r
r/ rR r rS )r r r r r �test_attributes_writable� s
z,AbstractMemoryTests.test_attributes_writablec C s� x|| j D ]r}|| j�}t�|�}| �|�}t�|�}t|d�}| �|||�d�� | �t�|�|� d }| �t�|�|� qW d S )Nzutf-8) r r r r r �strr9 �encoder )r r r! r" r Zoldviewrefcount�sr r r �test_getbuffer� s
z"AbstractMemoryTests.test_getbufferc C s� x�| j D ]�}t|t�sqG dd� d�}G dd� d|�}G dd� d�}||d��}| �|�}|� }||_||_t�|�}d } }}t� � | �
|� d k|� � ||d��}|� }||_ ||_t�|�}d }}t� � | �
|� d k|� � qW d S )Nc @ s e Zd Zdd� ZdS )z+AbstractMemoryTests.test_gc.<locals>.MyViewc S s t |�| _d S )N)r0 r )r �baser r r �__init__� s z4AbstractMemoryTests.test_gc.<locals>.MyView.__init__N)�__name__�
__module__� __qualname__r[ r r r r �MyView� s r_ c @ s e Zd ZdS )z-AbstractMemoryTests.test_gc.<locals>.MySourceN)r\ r] r^ r r r r �MySource� s r` c @ s e Zd ZdS )z-AbstractMemoryTests.test_gc.<locals>.MyObjectN)r\ r] r^ r r r r �MyObject� s ra s abc) r �
isinstance�typer r �o�weakref�ref�gcZcollectrK �x) r r r_ r` ra r! r rd �wrr r r �test_gc� s.
zAbstractMemoryTests.test_gcc
C s� | � td�}|� t|� W d Q R X |� |�� W d Q R X |� |�� W d Q R X |� |d W d Q R X |� d|d< W d Q R X |� t|� W d Q R X |� |j W d Q R X |� |j W d Q R X |� |j W d Q R X |� |j W d Q R X |� |j
W d Q R X |� |j W d Q R X |� |� W d Q R X W d Q R X | � dt
|�� | � dt|�� | �||� | �|t|| j��� | �||| j�� d S )NZreleasedr � xzreleased memory)ZassertRaisesRegexr= rC rE rH r* r< rM rN rS rO rP ZassertInrV �reprr ZassertNotEqualr0 r )r r r Zcheckr r r �_check_released� sB z#AbstractMemoryTests._check_releasedc
C sr xl| j D ]b}|| j�}| �|�}|�}| �||� W d Q R X | �||� | �|�}|� |�� W d Q R X qW d S )N)r r r �assertIsrm �release)r r r! r �cmr r r �test_contextmanager s
z'AbstractMemoryTests.test_contextmanagerc C sP xJ| j D ]@}|| j�}| �|�}|�� | �||� |�� | �||� qW d S )N)r r r ro rm )r r r! r r r r � test_release# s
z AbstractMemoryTests.test_releasec C sJ | j }|d kr| �d� || j�}| �|�}t�d�}| �t|j|� d S )Nzno read-only type to tests ZZZZ) r r/ r r �io�BytesIOr r �readinto)r r r! r r& r r r �test_writable_readonly- s
z*AbstractMemoryTests.test_writable_readonlyc C s | � t| ji � d S )N)r r r )r r r r �test_getbuf_fail8 s z$AbstractMemoryTests.test_getbuf_failc C s� | j }|d kr| �d� || j�}| �|�}| �t|�td�� |�� | �t|�td�� | �|�}|�� | �tt|� d S )Nzno read-only type to tests abcdef) r r/ r r r �hashro r r= )r r r! r r r r � test_hash; s
z AbstractMemoryTests.test_hashc C s> | j }|d kr| �d� || j�}| �|�}| �tt|� d S )Nzno writable type to test)r
r/ r r r r= rx )r r r! r r r r �test_hash_writableM s
z&AbstractMemoryTests.test_hash_writablec s� xz| j D ]p}|| j�}| �|�}g � |f� fdd� }t�||�}| �|� |� ~tj�� | �|� d � | �� d |� qW d S )Nc s � � |� d S )N)�append)ri r! )�Lr r �callback\ s z2AbstractMemoryTests.test_weakref.<locals>.callbackr ) r r r re rf rn �test�supportZ
gc_collect)r r r! r r} ri r )r| r � test_weakrefV s
z AbstractMemoryTests.test_weakrefc C sn xh| j D ]^}|| j�}| �|�}tt|�� ��}| �tt|��|� | �tt|��t|d d d� �� qW d S )Nr )r r r r( �reversedrH r )r r r! r Zaslistr r r �
test_reversede s
z!AbstractMemoryTests.test_reversedc C s� t � dddddg�}t|�}|�d�}|�d�}|dd� }t|�}~| �|d d� | �|d d� | �|jd� | �|jd� |�d�}| �|d d� | �|d d� | �|jd� | �|jd� d S )N�H� �Br r6 �I)�arrayr0 �castr r< )r r rh r r! r8 �d�_r r r �test_issue22668m s
z#AbstractMemoryTests.test_issue22668N) r\ r] r^ r �propertyr r r# r$ r+ r1 r? rA rG rJ rL rR rT rU rY rj rm rq rr rv rw ry rz r� r� r� r r r r r s6 4
&
r c @ s e Zd ZeZeZeZdZdZ dS )�BaseBytesMemoryTestsr4 r� N)
r\ r] r^ rC r � bytearrayr
rB rM r< r r r r r� � s
r� c @ sT e Zd ZdZdd� Zdd� Ze�d�jZdZe �
d�dd� �Z e �
d �d
d � �Z dS ) �BaseArrayMemoryTestsNc C s t � dt|��S )Nr& )r� r( )r r! r r r r � r z BaseArrayMemoryTests.<lambda>c C s t � dt|���� S )Nr& )r� r( rE )r r! r r r r � r r& z/XXX test should be adapted for non-byte buffersc C s d S )Nr )r r r r rY � s z#BaseArrayMemoryTests.test_getbufferz:XXX NotImplementedError: tolist() only supports byte viewsc C s d S )Nr )r r r r rJ � s z BaseArrayMemoryTests.test_tolist)
r\ r] r^ r r
rB r� rM r< �unittest�skiprY rJ r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�BaseMemoryviewTestsc C s t |�S )N)r0 )r �objr r r r � s zBaseMemoryviewTests._viewc C s | � |||�� d S )N)r )r r r� �contentsr r r r9 � s z#BaseMemoryviewTests._check_contentsN)r\ r] r^ r r9 r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �BaseMemorySliceTestss XabcdefYc C s t |�}|dd� S )Nr4 � )r0 )r r� r r r r r � s zBaseMemorySliceTests._viewc C s | � |dd� ||�� d S )Nr4 r� )r )r r r� r� r r r r9 � s z$BaseMemorySliceTests._check_contentsc C sJ xD| j D ]:}t|| j��}t�|�}|dd� | �t�|�|� qW d S )Nr4 r6 )r r0 r r r r )r r r r" r r r � test_refs� s
z BaseMemorySliceTests.test_refsN)r\ r] r^ r r r9 r� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�BaseMemorySliceSliceTestss XabcdefYc C s t |�}|d d� dd � S )Nr� r4 )r0 )r r� r r r r r � s z BaseMemorySliceSliceTests._viewc C s | � |dd� ||�� d S )Nr4 r� )r )r r r� r� r r r r9 � s z)BaseMemorySliceSliceTests._check_contentsN)r\ r] r^ r r r9 r r r r r� � s r� c @ s e Zd Zdd� ZdS )�BytesMemoryviewTestc C sz xt| j D ]j}|| j�}| �t|�� | �t|d�� | �tt� | �tt||� | jtt|d� | jtt|dd� qW d S )N)�object)�argumentT)r r rK r0 r r )r r Zobr r r �test_constructor� s
z$BytesMemoryviewTest.test_constructorN)r\ r] r^ r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�ArrayMemoryviewTestc C sH t � dtd��}t|�}t � dtddd��}||d d �< | �||� d S )Nr& �
� r )r� r) r0 r )r r r Znew_ar r r �test_array_assign� s
z%ArrayMemoryviewTest.test_array_assignN)r\ r] r^ r� r r r r r� � s r� c @ s e Zd ZdS )�BytesMemorySliceTestN)r\ r] r^ r r r r r� � s r� c @ s e Zd ZdS )�ArrayMemorySliceTestN)r\ r] r^ r r r r r� � s r� c @ s e Zd ZdS )�BytesMemorySliceSliceTestN)r\ r] r^ r r r r r� � s r� c @ s e Zd ZdS )�ArrayMemorySliceSliceTestN)r\ r] r^ r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� OtherTestc
C s� t j�d�}t|�d��}|�� }t|��d�}|d d� |d d�< |dd � |dd �< | �|jd� x�dD ]z}| � |��f |�� }t|��|�}t|��|�d d� |d d�< t|��|�dd � |dd �< | �|jd� W d Q R X qlW d S )N�ctypesg333333�?r� r6 ZBbc)
r~ r �
import_modulerC Zc_doubler0 r� r r, ZsubTest)r r� Zp6r� r r< r r r �test_ctypes_cast� s
zOtherTest.test_ctypes_castc C s6 dd }t |�}|d d d� }| �|�� dd � d S )Nr3 i@
r Z30)r0 r �hex)r rh Zm1Zm2r r r �test_memoryview_hex s z OtherTest.test_memoryview_hexc C s, t d�}| �t�� t�|� W d Q R X d S )Ns abc)r0 r r �copy)r r r r r � test_copy s zOtherTest.test_copyc
C sF t d�}x8ttjd �D ]&}| �t�� t�||� W d Q R X qW d S )Ns abcr4 )r0 r) �pickleZHIGHEST_PROTOCOLr r �dumps)r r �protor r r � test_pickle s zOtherTest.test_pickleN)r\ r] r^ r� r� r� r� r r r r r� � s r� �__main__)r� Z test.supportr~ r rg re r� rs r� r� r r� r� r� r� r� ZTestCaser� r� r� r� r� r� r� r\ �mainr r r r �<module> s> w
'