Your IP : 216.73.216.52
3
� f�E � @ sP d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
G dd� d�Z G dd� de �Z G dd� de �Z
G d d
� d
�ZG d d � d �ZG d
d� d�ZG dd� dejee �ZG dd� dejee
�ZG dd� dejee �ZG dd� dejee
�ZG dd� dejee �ZG dd� dejee
�ZG dd � d ej�Zed k�rLej� dS ) z�Unit tests for the memoryview
Some tests are in test_bytes. Many tests that require _testbuffer.ndarray
are in test_buffer.
� 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 �9/opt/alt/python36/lib64/python3.6/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 �}tj|�}| j|�� | j� d td�� | j� d t� | j� d td�� | j� d td�� | j� d td�� | jt � fdd�� | jt � fd d�� | jt � fd
d�� | jt � fd d�� | jt
� fd d�� | jt
� fd
d�� | jt
� fdd�� d � | jtj|�|� d S )Nr � a� � f� � c s � d S )Nr r r )�mr r �<lambda>' s z=AbstractMemoryTests.check_getitem_with_type.<locals>.<lambda>c s � d S )N� i����r r )r r r r ( s c s
� t j S )N)�sys�maxsizer )r r r r ) s c s � t j S )N)r r r )r r r r * s c s � d S )Nr r )r r r r , s c s � d S )Ng r r )r r r r - s c s � d S )N�ar r )r r r r . s ���i����) 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 ]}| j|� qW d S )N)r r$ )r r! r r r � test_getitem2 s z AbstractMemoryTests.test_getitemc sN xH| j D ]>}|| j�}| j|�� | jt� �� 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| jd� | j | j�}tj|�}| j|�� � fdd�}| jt|d� | jt|d� | jt|td�� d � | j tj|�|� 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| jd� | j �| j | j�}tj|�}| j|�� td�� d< | j�|d� �d�� dd�< | j�|d� �d�� dd �< | j�|d
� �d �� dd�< | j�|d
� �d �� d d �< | j�|d � � d
d� � dd �< | j�|d� �d �� d d �< � dd � � d
d�< | j�|d� � �fdd�}| jt |dd� | jt |dd� | jt |tj
d� | jt |tj
d� | jt |dd� | jt |dd� | jt |t ddd�dfd� | jt |dt ddd�fd� | jt |d d� | jt |dd� t ddd�t ddd
�f}| jt
||d� � jdk�r tnt }| j||dd � | j||dd� | jt|t dd�d� | jt|t dd
�d� d � | jtj|�|� d S ) Nzno writable type to test� 1r s 1bcdef� 0r s 0bcdefs 12� s 012def� 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 r g r �cs abi����)r )r )r
r0 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�}| j|�}| jt�� |d= W d Q R X | jt�� |dd�= W d Q R X qW d S )Nr � )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|� j��}|j� }dj� fdd�dD ��}� j||� � j|t� qW d S )Nr6 c 3 s | ]}� j t|g��V qd S )N)� getitem_type�bytes)r& r9 )r r r � <genexpr>� s z3AbstractMemoryTests.test_tobytes.<locals>.<genexpr>s abcdef)r r r �tobytes�joinr r rD )r r! r r" Zexpectedr )r r � test_tobytes� s z AbstractMemoryTests.test_tobytesc C s< x6| j D ],}| j|| j��}|j� }| 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|| j���xz| j D ]p}| j�|d�k� | j�|d�k� | j�|d�k� | j�|d�k� | j�|d�k� | j�|d�k� q(W | j��k� | j��d d � k� | j�dd� �d d � k� | j�dd� �k� | j�dk� | j�dk� | jd�k� | jd�k� xj�dfD ]^� | jt� �fdd �� | jt� �fd
d �� | jt� �fd d �� | jt� �fd d �� �q6W q
W d S )
Ns abcdefs abcdes abcde1r r r
Zabcdefc s �� k S )Nr r )r9 r r r r � s z2AbstractMemoryTests.test_compare.<locals>.<lambda>c s � �kS )Nr r )r9 r r r r � s c s �� kS )Nr r )r9 r r r r � s c s � �kS )Nr r )r9 r r r r � s )r r r �
assertTrueZ assertFalser r )r r! Ztp_compr )r9 r r � test_compare� s, z AbstractMemoryTests.test_comparec C s� | j || j��}| j|j| j� | j|j| j� | j|jd� | j|jd� | jt|�d� | j|j| jf� | j|j f � |S )Nr 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| jd� | j| j �}| j|jd� d S )Nzno read-only type to testT)r r0 rS r �readonly)r r r r r �test_attributes_readonly� s
z,AbstractMemoryTests.test_attributes_readonlyc C s. | j s| jd� | j| j �}| j|jd� d S )Nzno writable type to testF)r
r0 rS r rT )r r r r r �test_attributes_writable� s
z,AbstractMemoryTests.test_attributes_writablec C s� x|| j D ]r}|| j�}tj|�}| j|�}tj|�}t|d�}| j|||jd�� | jtj|�|� d }| jtj|�|� qW d S )Nzutf-8) r r r r r �strr: �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��}| j|�}|� }||_||_tj|�}d } }}tj � | j
|� d k|� � ||d��}|� }||_ ||_tj|�}d }}tj � | j
|� d k|� � qW d S )Nc @ s e Zd Zdd� ZdS )z+AbstractMemoryTests.test_gc.<locals>.MyViewc S s t |�| _d S )N)r1 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 ra c @ s e Zd ZdS )z-AbstractMemoryTests.test_gc.<locals>.MyObjectN)r] r^ r_ r r r r �MyObject� s rb s abc) r �
isinstance�typer r �o�weakref�ref�gcZcollectrL �x) r r! r` ra rb r" r re �wrr r r �test_gc� s.
zAbstractMemoryTests.test_gcc g C s� | j td�}|� t|� W d Q R X |� |j� W d Q R X |� |j� 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 | j dt
|�� | j dt|�� | j||� | j|t|| j��� | j||| j�� d S )NZreleasedr � xzreleased memory)ZassertRaisesRegexr> rD rF rI r+ r= rN rO rT rP rQ ZassertInrW �reprr ZassertNotEqualr1 r )r r r! Zcheckr r r �_check_released� sB z#AbstractMemoryTests._check_releasedc C sr xl| j D ]b}|| j�}| j|�}|�}| j||� W d Q R X | j||� | j|�}|� |j� W d Q R X qW d S )N)r r r �assertIsrn �release)r r! r" r �cmr r r �test_contextmanager s
z'AbstractMemoryTests.test_contextmanagerc C sP xJ| j D ]@}|| j�}| j|�}|j� | j||� |j� | j||� qW d S )N)r r r rp rn )r r! r" r r r r � test_release# s
z AbstractMemoryTests.test_releasec C sJ | j }|d kr| jd� || j�}| j|�}tjd�}| jt|j|� d S )Nzno read-only type to tests ZZZZ) r r0 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 | j 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| jd� || j�}| j|�}| jt|�td�� |j� | jt|�td�� | j|�}|j� | jtt|� d S )Nzno read-only type to tests abcdef) r r0 r r r �hashrp r r> )r r! r" r r r r � test_hash; s
z AbstractMemoryTests.test_hashc C s> | j }|d kr| jd� || j�}| j|�}| jtt|� d S )Nzno writable type to test)r
r0 r r r r> ry )r r! r" r r r r �test_hash_writableM s
z&AbstractMemoryTests.test_hash_writablec s� xz| j D ]p}|| j�}| j|�}g � |f� fdd� }tj||�}| j|� |� ~tjj� | j|� d � | j� d |� qW d S )Nc s � j |� d S )N)�append)rj r" )�Lr r �callback\ s z2AbstractMemoryTests.test_weakref.<locals>.callbackr ) r r r rf rg ro �test�supportZ
gc_collect)r r! r" r r~ rj r )r} r � test_weakrefV s
z AbstractMemoryTests.test_weakrefc C sn xh| j D ]^}|| j�}| j|�}tt|j� ��}| jtt|��|� | jtt|��t|d d d� �� qW d S )Nr r )r r r r) �reversedrI r )r r! r" r Zaslistr r r �
test_reversede s
z!AbstractMemoryTests.test_reversedc C s� t j dddddg�}t|�}|jd�}|jd�}|dd� }t|�}~| j|d d� | j|d d� | j|jd� | j|jd� |jd�}| j|d d� | j|d d� | j|jd� | j|jd� d S )N�H� �Br r7 �I)�arrayr1 �castr r= )r r ri r r" r9 �d�_r r r �test_issue22668m s
z#AbstractMemoryTests.test_issue22668N) r] r^ r_ r �propertyr r r$ r% r, r2 r@ rB rH rK rM rS rU rV rZ rk rn rr rs rw rx rz r{ r� r� r� r r r r r s6 4
&
r c @ s e Zd ZeZeZeZdZdZ dS )�BaseBytesMemoryTestsr r� N)
r] r^ r_ rD r � bytearrayr
rC rN r= r r r r r� � s
r� c @ sT e Zd ZdZdd� Zdd� Zejd�jZdZe j
d�dd� �Z e j
d �d
d � �Z dS ) �BaseArrayMemoryTestsNc C s t j dt|��S )Nr' )r� r) )r r" r r r r � s z BaseArrayMemoryTests.<lambda>c C s t j dt|��j� S )Nr' )r� r) rF )r r" r r r r � s r' z/XXX test should be adapted for non-byte buffersc C s d S )Nr )r r r r rZ � s z#BaseArrayMemoryTests.test_getbufferz:XXX NotImplementedError: tolist() only supports byte viewsc C s d S )Nr )r r r r rK � s z BaseArrayMemoryTests.test_tolist)
r] r^ r_ r r
rC r� rN r= �unittest�skiprZ rK r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�BaseMemoryviewTestsc C s t |�S )N)r1 )r �objr r r r � s zBaseMemoryviewTests._viewc C s | j |||�� d S )N)r )r r! r� �contentsr r r r: � s z#BaseMemoryviewTests._check_contentsN)r] r^ r_ r r: r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �BaseMemorySliceTestss XabcdefYc C s t |�}|dd� S )Nr r )r1 )r r� r r r r r � s zBaseMemorySliceTests._viewc C s | j |dd� ||�� d S )Nr r )r )r r! r� r� r r r r: � s z$BaseMemorySliceTests._check_contentsc C sJ xD| j D ]:}t|| j��}tj|�}|dd� | jtj|�|� qW d S )Nr r7 )r r1 r r r r )r r! r r# r r r � test_refs� s
z BaseMemorySliceTests.test_refsN)r] r^ r_ r r r: 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 r )r1 )r r� r r r r r � s z BaseMemorySliceSliceTests._viewc C s | j |dd� ||�� d S )Nr r )r )r r! r� r� r r r r: � s z)BaseMemorySliceSliceTests._check_contentsN)r] r^ r_ r r r: r r r r r� � s r� c @ s e Zd Zdd� ZdS )�BytesMemoryviewTestc C sz xt| j D ]j}|| j�}| jt|�� | jt|d�� | jtt� | jtt||� | jtt|d� | jtt|dd� qW d S )N)�object)ZargumentT)r r rL r1 r r )r r! �obr 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 j dtd��}t|�}t j dtddd��}||d d �< | j||� d S )Nr' �
� r r r )r� r* r1 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 jjd�}t|jd��}|j� }t|�jd�}|d d� |d d�< |dd � |dd �< | j|jd� x�dD ]z}| j |��f |j� }t|�j|�}t|�j|�d d� |d d�< t|�j|�dd � |dd �< | j|jd� W d Q R X qlW d S )N�ctypesg333333�?r� r7 ZBbc)
r r� �
import_modulerD Zc_doubler1 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� }| j|j� dd � d S )Nr4 i@
r Z30r )r1 r �hex)r ri �m1�m2r r r �test_memoryview_hex s z OtherTest.test_memoryview_hexc C s, t d�}| jt�� tj|� W d Q R X d S )Ns abc)r1 r r �copy)r r r r r � test_copy s zOtherTest.test_copyc C sF t d�}x8ttjd �D ]&}| jt�� tj||� W d Q R X qW d S )Ns abcr )r1 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__)�__doc__r� Z test.supportr r rh rf r� rt r� r� r r� r� r� r� r� ZTestCaser� r� r� r� r� r� r� r] �mainr r r r �<module> s@ w
'