OwlCyberSecurity - MANAGER
Edit File: test_memmap.cpython-38.pyc
U �p�]� � @ s� d dl mZmZmZ d dlZd dlZd dlZd dlZd dlZd dl m Z m Z m Z m Z d dlmZmZmZmZmZmZmZmZmZ d dlmZ d dlmZmZm Z d dl m Z m Z m Z m!Z! G dd � d e"�Z#dS ) � )�division�absolute_import�print_functionN)�NamedTemporaryFile� TemporaryFile�mktemp�mkdtemp) �memmap�sum�average�product�ndarray�isscalar�add�subtract�multiply)�Path)�arange�allclose�asarray)�assert_� assert_equal�assert_array_equal�suppress_warningsc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j j e dkdd�dd� �Zdd� Ze j j ejdkdd�dd� �Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z d.d/� Z dS )0� TestMemmapc C sB t dd�| _t� | _d| _d| _td| jd�| _| j�| j� d S )N�mmap)�prefix)� � Zfloat32� ��dtype) r �tmpfpr �tempdir�shaper! r �dataZresize��self� r( �>/usr/lib/python3/dist-packages/numpy/core/tests/test_memmap.py�setup s zTestMemmap.setupc C s | j �� t�| j� d S )N)r" �close�shutilZrmtreer# r& r( r( r) �teardown s zTestMemmap.teardownc C sr t | j| jd| jd�}| jd d � |d d �< ~t | j| jd| jd�}tt| j|�� t| j|� t|j j d� d S )N�w+�r! �moder$ �rF) r r" r! r$ r% r r r r �flagsZ 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 ��dirr. r/ )r r# r r! r$ r% )r' �tmpnamer3 r( r( r) �test_open_with_filename. s �z"TestMemmap.test_open_with_filenamec C s* t � �}t|| j| jd�}~W 5 Q R X d S �N�r! r$ )r r r! r$ )r' �fr3 r( r( r) �test_unnamed_file5 s z TestMemmap.test_unnamed_filec C s> d}d}t | j| j|| j|d�}t||j� t||j� ~d S )N� r. )r! r0 r$ �offset)r r" r! r$ r r? r0 )r' r? r0 r3 r( r( r) �test_attributes: s � zTestMemmap.test_attributesc C sr t dd| jd�}t|| jd| jd�}tj�|�}| jd d � |d d �< t ||j � |d d� }t ||j � ~~d S �Nr5 r r6 r. r/ r> ) r r# r r! r$ �os�path�abspathr% r �filename�r' r8 r3 rD �br( r( r) � test_filenameC s � zTestMemmap.test_filenameNzNo pathlib.Path)�reasonc C s� t dd| jd�}tt|�| jd| jd�}tt|��� �}| jd d � |d d �< t |t|j �� �� |d d� }t |t|j �� �� ~~d S rA ) r r# r r r! r$ �strZresolver% r rE rF r( r( r) � test_pathO s � zTestMemmap.test_pathc C s* t | j| jd| jd�}t|j| jj� d S )Nr. r/ )r r" r! r$ r rE �name�r' r3 r( r( r) �test_filename_fileobj_ s �z TestMemmap.test_filename_fileobjZgnu0zKnown to fail on hurdc C sL t | j| jd| jd�}| jd d � |d d �< t|d | jd � |�� d S )Nr. r/ r )r r" r! r$ r% r �flushrM r( r( r) � test_flushd 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 )Nr. r/ � r r> � )r r" r! r$ r )r' Zfp_baseZfp_viewr( r( r) �test_delm s � zTestMemmap.test_delc C s< t | j| jd| jd�}|d }t|t �r8t|j|jk � d S )Nr. r/ � �r r" r! r$ � isinstancer �_mmap�r' r3 Ztmpr( r( r) � test_arithmetic_drops_references{ s � z+TestMemmap.test_arithmetic_drops_referencesc C s< t | j| jd| jd�}|d }t|t �r8t|j|jk � d S )Nr. r/ ))r> � )rZ r rU rX 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 )Nr. r/ rZ )r r" r! r$ r rW rM 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 r: )r r" r! r$ Zviewr �baser )r' r3 Znew1Znew2Z new_arrayr( r( r) � test_view� s zTestMemmap.test_viewc C s0 t | j| j| jd�}| j|d d �< t� �x}|�td� tt t fD ]Z}||�}t t |�� t |j | jd j k� t ||dd�j tk� t ||dd�j tk� q@W 5 Q R X tttfD ]D}t ||| j�j tk� t || j|�j tk� t |||�j tk� q�|d7 }|j t k�st�t|d|d� |j t k�s,t�d S )Nr; z&np.average currently does not preserve)r r r �Zaxisr> )�out)r r" r! r$ r% r �filter� FutureWarningr r r r r � __class__r r r r �AssertionError)r' r3 ZsupZunary_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 )Nr; r> ���r )r r" r! r$ r% r rc r rM 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 rl r; r r_ r> rg ) r r" r! r$ r% r r rc rd )r' rl r3 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 )NrQ r> r. )r$ r0 r? )r ZALLOCATIONGRANULARITYr r" r r? )r' �sizer? r3 r( r( r) �4test_mmap_offset_greater_than_allocation_granularity� s z?TestMemmap.test_mmap_offset_greater_than_allocation_granularityc C s* | j �d� t| j dd�}t|jd� d S )Ns aaaaaaaaaaaaaaaaZfloat64r )rZ )r" �writer r r$ )r' Zmmr( r( r) � test_no_shape� s zTestMemmap.test_no_shapec C sJ t jtdd�� t| jddd� W 5 Q R X | j�d� t| jddd� d S )Nz empty file)�match)r r r. )r$ r0 � )�pytestZraises� ValueErrorr r" rp r& r( r( r) �test_empty_array� s zTestMemmap.test_empty_array) ri rj rk r* r- r4 r9 r= r@ rH rt ZmarkZskipifr rK rN �sys�platformrP rS rY r[ r\ r^ rf rh rm ro rq rv r( r( r( r) r s2 � r )$Z __future__r r r rw rB r, r rt Ztempfiler r r r Znumpyr r r r r r r r r Z numpy.compatr r r r Z numpy.testingr r r r �objectr r( r( r( r) �<module> s ,