U
|
P ±dË# ã @ s¾ d dl Zd dlmZ d dlmZmZmZmZ d dl m
|
Z
|
m Z m Z m Z mZmZmZmZmZmZmZmZ d dlmZ d dlmZ G dd deejZG d d
|
d
|
Z G d d d Z G d d dZ 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 ZefddZdd Zedd ZdS )ÚMMatrixc C s t |¡}tj| ||d}|S )N)ÚdataÚmask)ÚnpÚmatrixr Ú__new__)Úclsr r ÚmatÚ_data© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\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 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# Úx1Zx2Zx3Zx4r 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- r0 )Úprotocol)r r r ÚlistÚranger ÚHIGHEST_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|jddjd t|jddjd
|
t|jddd d gg t|jddd gdgg d S )Nr, r- r. r/ )r- r- r0 r )Zaxis)r, r- )r- r, g @r6 g ø?g @)r r r Úzerosr ÚcountÚshapeZmean)r# Úmr r r Útest_count_mean_with_matrixU 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 r0 r2 r5 g F) r r r r Zflatr r ÚallZeyer r; r )r# ÚtestÚcontrolZtestflatrH Ú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æ?TFr7 r0 g à?r r, )
|
r r; r Zbool_r r rP Úanyr ÚT)r# ÚxÚXrN ZmXZmXbigZmXsmallr r r Útest_allany_onmatricesx s4 þ
|
þþ z'TestMaskedMatrix.test_allany_onmatricesc C sj t t ddddg¡ddddgd}| ¡ }t|| tt|tj t|d< | ¡ }t|dddgg d S )Nr, r- r. r/ r r0 )r r )r r r Ú
|
compressedr r rG r )r# rH rS 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. r/ r4 r r0 )r, r4 )r r r Zravelr rM rF )r# rH 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? rH rS r7 r1 r r- )rA Úzipr r< ÚrandomZrandr; r Úfloatr r% r r r rG r )r# Úiteratorr rH rQ r r r Ú test_view¡ s
|
|
|
zTestMaskedMatrix.test_viewN) r' r( r) r> rJ rO rT rY r[ r\ ra 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 )Nr4 r_ r7 r r, r0 )r r< r r ©r# rW Úmxr r r Ú setup_method° s z TestSubclassing.setup_methodc C s | j \}}tt|jtj d S r! )r r rG r r r rc 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 rG r r r rc 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 5 Q R X ttt ||t ttt ||t d S r! ) r r rG r r r r r r r ÚDeprecationWarningÚouterr rc 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 )Nr0 ) r r Z __array__r r rG r r r )r# rW rd Zxmxr r r Ú test_masked_binary_operations2Ï s
|
|
z.TestSubclassing.test_masked_binary_operations2N)r' r( r) re rf rh rk rl r r r r rb s
|
rb 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|jt|j d S )N)Úrr, r- r. ) r r rp r; Zr_r r r: r )r# ÚactualÚexpectedr r r Ú test_matrixÝ s
|
|
z TestConcatenator.test_matrixN)r' r( r) rq ru r r r r rm × s rm ) Únumpyr 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+ rb rm r r r r Ú<module> s 8 *
|
