Edit File: test_memmap.cpython-37.pyc
B ��Fd� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlmZm Z m Z m Z d dl Z d dl mZmZmZmZmZmZmZmZmZ d dlmZ d dl mZmZmZ d dl m Z m Z m Z m Z m!Z!m"Z"m#Z# G dd � d e �Z$e%d kr�e � dS ) � )�division�absolute_import�print_functionN)�NamedTemporaryFile� TemporaryFile�mktemp�mkdtemp) �memmap�sum�average�product�ndarray�isscalar�add�subtract�multiply)�Path)�arange�allclose�asarray)�TestCase�run_module_suite�assert_� assert_equal�assert_array_equal�dec�suppress_warningsc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e � e dkd�dd� �Z dd� Ze �ejdkd�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� ZdS )+� TestMemmapc C sB t dd�| _t� | _d| _d| _td| jd�| _| j�| j� d S )N�mmap)�prefix)� � �float32� )�dtype) r �tmpfpr �tempdir�shaper$ r �data�resize)�self� r+ ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_memmap.py�setUp s zTestMemmap.setUpc C s | j �� t�| j� d S )N)r% �close�shutil�rmtreer& )r* r+ r+ r, �tearDown s zTestMemmap.tearDownc C st t | j| jd| jd�}| jd d � |d d �< ~t | j| jd| jd�}tt| j|�� t| j|� | �|j j d� d S )Nzw+)r$ �moder' �rF) r r% r$ r' r( r r r � assertEqual�flags� writeable)r* �fpZnewfpr+ r+ r, �test_roundtrip s zTestMemmap.test_roundtripc C s@ t dd| jd�}t|| jd| jd�}| jd d � |d d �< ~d S )N� r )�dirzw+)r$ r2 r' )r r& r r$ r' r( )r* �tmpnamer7 r+ r+ r, �test_open_with_filename. s z"TestMemmap.test_open_with_filenamec C s* t � �}t|| j| jd�}~W d Q R X d S )N)r$ r' )r r r$ r' )r* �fr7 r+ r+ r, �test_unnamed_file5 s z TestMemmap.test_unnamed_filec C sB d}d}t | j| j|| j|d�}| �||j� | �||j� ~d S )N� zw+)r$ r2 r' �offset)r r% r$ r' r4 r@ r2 )r* r@ r2 r7 r+ r+ r, �test_attributes: s zTestMemmap.test_attributesc C sv t dd| jd�}t|| jd| jd�}tj�|�}| jd d � |d d �< | � ||j � |d d� }| � ||j � ~~d S )Nr9 r )r: zw+)r$ r2 r' r? ) r r& r r$ r' �os�path�abspathr( r4 �filename)r* r; r7 rD �br+ r+ r, � test_filenameC s zTestMemmap.test_filenameNzNo pathlib.Pathc C s� t dd| jd�}tt|�| jd| jd�}tj�tj� |��}| j d d � |d d �< | � |t |j �� �� |d d� }| � |t |j �� �� ~~d S )Nr9 r )r: zw+)r$ r2 r' r? )r r& r r r$ r' rB rC �realpathrD r( r4 �strrE �resolve)r* r; r7 rD rF r+ r+ r, � test_pathO s zTestMemmap.test_pathc C s, t | j| jd| jd�}| �|j| jj� d S )Nzw+)r$ r2 r' )r r% r$ r' r4 rE �name)r* r7 r+ r+ r, �test_filename_fileobj\ s z TestMemmap.test_filename_fileobjZgnu0z"This test is known to fail on hurdc C sL t | j| jd| jd�}| jd d � |d d �< t|d | jd � |�� d S )Nzw+)r$ r2 r' r )r r% r$ r' r( r �flush)r* r7 r+ r+ r, � test_flusha s zTestMemmap.test_flushc C sb t | j| jd| jd�}d|d< |dd� }t|d d� ~t|d d� d|d< t|d d� d S )Nzw+)r$ r2 r' � r r? � )r r% r$ r' r )r* Zfp_baseZfp_viewr+ r+ r, �test_deli s zTestMemmap.test_delc C s< t | j| jd| jd�}|d }t|t �r8t|j|jk � d S )Nzw+)r$ r2 r' � )r r% r$ r' � isinstancer �_mmap)r* r7 �tmpr+ r+ r, � test_arithmetic_drops_referencesw s z+TestMemmap.test_arithmetic_drops_referencesc C s@ t | j| jd| jd�}|ddg }t|t �r<t|j|jk � d S )Nzw+)r$ r2 r' )r? � )rX r )r r% r$ r' rT r rU )r* r7 rV r+ r+ r, � test_indexing_drops_references~ s z)TestMemmap.test_indexing_drops_referencesc C s: t | j| jd| jd�}t|d d�d d�f j|jk� d S )Nzw+)r$ r2 r' rX )r r% r$ r' r rU )r* r7 r+ r+ r, � test_slicing_keeps_references� s z(TestMemmap.test_slicing_keeps_referencesc C sZ t | j| j| jd�}|�� }|�� }t|j|k� t|j|k� t|�}t|j|k� d S )N)r$ r' )r r% r$ r' �viewr �baser )r* r7 Znew1Znew2� new_arrayr+ r+ r, � test_view� s zTestMemmap.test_viewc C s8 t | j| j| jd�}| j|d d �< t� �|}|�td� xhtt t gD ]Z}||�}t t |�� t |j | jd j k� t ||dd�j tk� t ||dd�j tk� qBW W d Q R X xRtttgD ]D}t ||| j�j tk� t || j|�j tk� t |||�j tk� q�W |d7 }|j t k�st�t|d|d� |j t k�s4t�d S )N)r$ r' z&np.average currently does not preserve)r r r )�axisr? )�out)r r% r$ r' r( r �filter� FutureWarningr r r r r � __class__r r r r �AssertionError)r* r7 �supZunary_op�resultZ binary_opr+ r+ r, �test_ufunc_return_ndarray� s$ $z$TestMemmap.test_ufunc_return_ndarrayc C sZ t | j| j| jd�}| j|d d �< t|dd �d d�f jt k� t|ddg jtk� d S )N)r$ r' r? ���r )r r% r$ r' r( r rc r )r* r7 r+ r+ r, � test_getitem� s zTestMemmap.test_getitemc C s� G dd� dt �}|| j| j| jd�}| j|d d �< tt|dd�j|k� tt|�j|k� t|dd �d d�f j|k� |ddg j|ks�t�d S )Nc @ s e Zd ZdS )z7TestMemmap.test_memmap_subclass.<locals>.MemmapSubClassN)�__name__� __module__� __qualname__r+ r+ r+ r, �MemmapSubClass� s rm )r$ r' r )r_ r? rh ) r r% r$ r' r( r r rc rd )r* rm r7 r+ r+ r, �test_memmap_subclass� s z TestMemmap.test_memmap_subclassc C s8 dt j }t jd }t| j|d|d�}t|j|k� d S )NrP r? zw+)r' r2 r@ )r �ALLOCATIONGRANULARITYr r% r r@ )r* �sizer@ r7 r+ r+ r, �4test_mmap_offset_greater_than_allocation_granularity� s z?TestMemmap.test_mmap_offset_greater_than_allocation_granularity) rj rk rl r- r1 r8 r< r> rA rG r �skipifr rK rM �knownfailureif�sys�platformrO rR rW rY rZ r^ rg ri rn rq r+ r+ r+ r, r s& r �__main__)&� __future__r r r rt rB r/ �tempfiler r r r r �numpyr r r r r r r r r � numpy.compatr r r r Z numpy.testingr r r r r r r r rj r+ r+ r+ r, �<module> s , $ 5