Current File : //lib64/python3.9/site-packages/numpy/matrixlib/tests/__pycache__/test_masked_matrix.cpython-39.pyc
a
z[yc�" � @ s� d dl Zd dlmZ d dlmZmZmZmZ d dl m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZ d dlmZ d dlmZ G dd� deej�ZG d d
� d
�Z G d d � d �Z G d
d� d�Z dS )� N)� assert_warns)�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 �N/usr/lib64/python3.9/site-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 �g d�g d�g�}t|g d�g d�gd�}t|g d�g d�gd�}t|�}t|� t|� tt|d �t|d �u � t|d |d k� t|d tu � t|d |d � t|d d
d �f |d d
d �f � t|d d �d f |d d �d f � t|d d � |d d � � t|d
d � |d
d � � d |d< d |d< t||� d
|d d
d �f< d
|d d
d �f< t||� t|d< t||� t|d d
d �f< t||� |d d d �f |d d d �f< t|d< tt t
|�t � g d�g d�g��� tg d�g d��|d
d d �f< tt t
|�d
tg d���� tt t
|d
�tg d���� tg d�g d��|d
d d �f< tt t
|d
�tg d���� tt |d
tg d���� t �t � d�d �}t
|d�}t||� tt tg d�td�|j�� td|j� d S )N�� � � )� r. r- �r, r r )r r, r �r �r, r )r, r, )r r- r r, r- � �c )r r, )r, r, r � � �?� @)r r r 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 sz t t�ttd���g d�d d�}tdtjd �D ]B}t�tj||d��}t |j
|j
� t ||� t t |j
tj�� q2d S )N�
)r, r r, r r r- r1 r, )Zprotocol)r r r �list�ranger ZHIGHEST_PROTOCOL�loads�dumpsr �_maskr �
isinstancer )r" �a�protoZ a_pickledr r r �test_pickling_subbaseclassL 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. r/ )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_matrixU s
$z,TestMaskedMatrix.test_count_mean_with_matrixc C sb t t�g d�g�g d�d�}t|jd d� t|jd t� tt�|jdd� |ddd�f k�� t t�g d�g�g d�d�}t g d�g d�d�|_t t�g d�g�g d�d�}t||� t t�g d�g�g d�d�}|j}|g d � |d d �< t||� d
|d< t t�t�d��dd�}|j}|d d� }t|j t�
d d gg�� t|j t�
d
d
gg�� d S )Nr+ )r r r, r1 r, r- r )r. r- r, r0 )r- r, r 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 st t �g d�g d�g d�g�}t �|�}t jg d�g d�g d�gt jd�}t||d�}|d k}|d k }t|�� � t|�� � t|�d
�t �g d ��� t|�d �t �g d ��j � t|�d
�t �g d ��� t|�d �t �g d
��j � t|�� � t|�� � t|�d
�t �g d��� t|�d �t �g d��j � t|�d
�t �g d��� t|�d �t �g d��j � d S )N)g�p=
ף�?g�p=
ף�?g�������?)g�Q� ���?g ��Q� �?g)\��(�?)gףp=
��?gףp=
��?gffffff�?)TFF)FFF)TTFr8 r1 g �?r )FFTr, )TTT)
r r< r Zbool_r r rM �anyr �T)r" �x�XrK ZmXZmXbigZmXsmallr r r �test_allany_onmatricesx s4
�
�� z'TestMaskedMatrix.test_allany_onmatricesc C s` t t�g d��g d�d�}|�� }t||� tt|tj�� t|d< |�� }t|g d�g� d S )N)r, r- r. r/ )r r r r r1 )r r )r- r. r/ )r r r �
compressedr r rE r )r" rF rO r r r �test_compressed� s
z TestMaskedMatrix.test_compressedc C sD t t�g d��g d�gd�}|�� }t|jd� t|jj|j� d S )N)r, r- r. r/ r5 �r r, r r 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 rW rY r] r r r r r* s ,
r* c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestSubclassingc C s, t jddd�}t|g d�d�}||f| _d S )Nr5 r[ r8 rX r1 )r r= r r �r" rS �mxr r r � setup_method� s z TestSubclassing.setup_methodc 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 s` | j \}}tjdd��6 ttt|�t�� tt|�t�|�� W d � n1 sR0 Y 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�� t t
��& ttt� ||�t�� W d � n1 s�0 Y ttt ||�t�� ttt ||�t�� d S r )
r r rE r r r r r r r �DeprecationWarningZouterr r_ r r r � test_masked_binary_operations� s
4z-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( ra rb rd rf rg 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 rk r< Zr_r r r; r )r" �actualZexpectedr r r � test_matrix� s
z TestConcatenator.test_matrixN)r& r' r( rl ro r r r r rh � s rh ) Znumpyr Z
numpy.testingr 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 r* r^ rh r r r r �<module> s 8 *
Mini Shell