OwlCyberSecurity - MANAGER
Edit File: test_masked_matrix.cpython-38.pyc
U �p�]�" � @ s� d dl mZmZmZ d dlZd dlmZmZm Z m Z d dl m Z m Z mZmZmZmZmZmZmZmZmZmZ d dlmZ d dlmZ G dd� deej �Z G d d � d e �Z G d d � d e �Z G d d� de �Z!dS )� )�division�absolute_import�print_functionN)�assert_� assert_equal� assert_raises�assert_array_equal) � masked_array� masked_values�masked�allequal�MaskType�getmask� MaskedArray�nomask�log�add�hypot�divide��mr_)�picklec @ s, e Zd Zefdd�Zdd� Zedd� �ZdS )�MMatrixc C s t �|�}tj| ||d�}|S )N)�data�mask)�np�matrixr �__new__)�clsr r Zmat�_data� r �J/usr/lib/python3/dist-packages/numpy/matrixlib/tests/test_masked_matrix.pyr s zMMatrix.__new__c C s t j�| |� t�| |� d S �N)r r �__array_finalize__r )�self�objr r r! r# s zMMatrix.__array_finalize__c C s | � t�}d|_|S )NF)�viewr Z _sharedmask)r$ Z_viewr r r! �_series s zMMatrix._seriesN)�__name__� __module__� __qualname__r r r# �propertyr' r r r r! r s r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�TestMaskedMatrixc C s t �dddgdddgg�}t|dddgdddggd�}t|dddgdddggd�}t|�}t|� t|� tt|d �t|d �k� t|d |d k� t|d tk� t|d |d � t|ddd �f |ddd �f � t|d d �df |d d �df � t|d d � |d d � � t|dd � |dd � � d |d < d |d < t||� d |ddd �f< d |ddd �f< t||� t|d < t||� t|ddd �f< t||� |dd d �f |dd d �f< t|d < tt t |�t � dddgdddgg��� tdddgdddg�|dd d �f< tt t |�d tdddg��� tt t |d �tdddg��� tdddgdddg�|dd d �f< tt t |d �tdddg��� tt |d tdddg��� t �t � d �d �}t |d�}t||� tt tdddddgtd�|j�� td|j� d S )N� � � � r �r �r- r )r- r- )r r. � �c )r r- � � �?� @�Zdtype)r r r �str�reprr �typer r r r �array�aranger r r Z fill_value)r$ Zx1Zx2Zx3Zx4r r r! �test_matrix_indexing! sP "" (" " �z%TestMaskedMatrix.test_matrix_indexingc C s� t t�ttd���dddddgd d�}tdtjd �D ]B}t�tj||d��}t |j |j � t ||� t t |j tj�� q8d S )N� r- r r. r1 )Zprotocol)r r r �list�ranger ZHIGHEST_PROTOCOL�loads�dumpsr �_maskr � isinstancer )r$ �a�protoZ a_pickledr r r! �test_pickling_subbaseclassM s ( z+TestMaskedMatrix.test_pickling_subbaseclassc C s� t t�ddgddgg�t�d�d�}t|jdd�jd � t|jdd�jd � t|jdd�d d gg� t|jdd�d gdgg� d S )Nr- r. r/ r0 )r. r. r1 r )Zaxis)r- r. )r. r- g @r7 g �?g @)r r r Zzerosr �count�shapeZmean)r$ �mr r r! �test_count_mean_with_matrixV s $z,TestMaskedMatrix.test_count_mean_with_matrixc C sx t t�dddgg�dddgd�}t|jd d� t|jd t� tt�|jdd� |ddd�f k�� t t�dddgg�dddgd�}t dddgdddgd�|_t t�dddgg�dddgd�}t||� t t�dddgg�dddgd�}|j}|dddg |d d �< t||� d|d< t t�t�d��dd�}|j}|d d� }t|j t� ddgg�� t|j t� d d gg�� d S ) Nr- r. r/ r r1 r3 r6 g F) r r r r Zflatr r �allZeyer r< r )r$ �testZcontrolZtestflatrF �bZb01r r r! � test_flat` s$ ( zTestMaskedMatrix.test_flatc C s� t �dddgdddgddd gg�}t �|�}t jd d d gd d d gd d d ggt jd �}t||d �}|dk}|dk }t|�� � t|�� � t|�d�t �d d d g�� t|�d�t �d d d g�j � t|�d�t �d d d g�� t|�d�t �d d d g�j � t|�� � t|�� � t|�d�t �d d d g�� t|�d�t �d d d g�j � t|�d�t �d d d g�� t|�d�t �d d d g�j � d S )Ng�p= ף�?g�p= ף�?g�������?g�Q� ���?g ��Q� �?g)\��(�?gףp= ��?gףp= ��?gffffff�?TFr8 r1 g �?r r- ) r r< r Zbool_r r rM �anyr �T)r$ �x�XrK ZmXZmXbigZmXsmallr r r! �test_allany_onmatricesy s4 � �� z'TestMaskedMatrix.test_allany_onmatricesc C sj t t�ddddg�ddddgd�}|�� }t||� tt|tj�� t|d< |�� }t|dddgg� d S )Nr- r. r/ r0 r r1 )r r )r r r Z compressedr r rE r )r$ rF rO r r r! �test_compressed� s " z TestMaskedMatrix.test_compressedc C sP t t�dddddg�dddddggd�}|�� }t|jd� t|jj|j� d S ) Nr- r. r/ r0 r5 r r1 )r- r5 )r r r Zravelr rJ rD )r$ rF Zaravelr r r! � test_ravel� s ( zTestMaskedMatrix.test_ravelc C s� t tt�d�tj�d���}t�|�}t|dtfdtfgd�}d|j d< |� tdftj �}t ||� t t|tj �� t t|t� � d S )Nr? rF rO r8 r2 r r. )r@ �zipr r= ZrandomZrandr< r �floatr r& r r r rE r )r$ �iteratorr rF rN r r r! � test_view� s zTestMaskedMatrix.test_viewN) r( r) r* r> rH rL rP rU rV rW r[ r r r r! r, s , r, c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestSubclassingc C s2 t jddd�}t|dddddgd�}||f| _d S )Nr5 rY r8 r r- r1 )r r= r r �r$ rS �mxr r r! �setup� s zTestSubclassing.setupc C s | j \}}tt|jtj�� d S r" )r r rE r r r r] r r r! � test_maskedarray_subclassing� s z,TestSubclassing.test_maskedarray_subclassingc C sL | j \}}tjdd��, ttt|�t�� tt|�t�|�� W 5 Q R X d S )N�ignore)r )r r Zerrstater rE r r r r] r r r! � test_masked_unary_operations� s z,TestSubclassing.test_masked_unary_operationsc C s� | j \}}ttt||�t�� ttt||�t�� tt||�|| � ttt||�jtj�� ttt� ||�t�� ttt ||�t�� ttt ||�t�� d S r" ) r r rE r r r r r r Zouterr r] r r r! � test_masked_binary_operations� s z-TestSubclassing.test_masked_binary_operationsc C s` | j \}}t|j �� |jd�}ttt||�t�� ttt||�t�� tt||�t||�� d S )Nr1 ) r r Z __array__r r rE r r r )r$ rS r^ Zxmxr r r! � test_masked_binary_operations2� s z.TestSubclassing.test_masked_binary_operations2N)r( r) r* r_ r` rb rc rd r r r r! r\ � s r\ c @ s e Zd Zdd� Zdd� ZdS )�TestConcatenatorc C s t tjjdd� � d S )Nc S s t d S )Nz 1, 2; 3, 4r r r r r! �<lambda>� � z6TestConcatenator.test_matrix_builder.<locals>.<lambda>)r r �maZMAError)r$ r r r! �test_matrix_builder� s z$TestConcatenator.test_matrix_builderc C sP t d }tj�tjd �}t||� tt|�t|�� tt|j�t|j�� d S )N)�rr- r. r/ ) r r rh r< Zr_r r r; r )r$ ZactualZexpectedr r r! � test_matrix� s z TestConcatenator.test_matrixN)r( r) r* ri rk r r r r! re � s re )"Z __future__r r r Znumpyr Znumpy.ma.testutilsr r r r Z numpy.ma.corer r r r r r r r r r r r Znumpy.ma.extrasr Z numpy.compatr r r �objectr, r\ re r r r r! �<module> s 8 )