3
���h~ � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddl m
Z
m Z m Z m
Z
ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z; G dd� d�Z<G d d
� d
�Z=G d d � d �Z>G d
d� d�Z?G dd� d�Z@G dd� d�ZAG dd� d�ZBG dd� d�ZCG dd� d�ZDG dd� d�ZEG dd � d �ZFG d d � d �ZGG d d � d �ZHG d!d"� d"�ZIdS )#z�Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $
� N)� assert_warns�suppress_warnings)�assert_�assert_array_equal� assert_equal�assert_almost_equal) �array�arange�masked� MaskedArray� masked_array� getmaskarray�shape�nomask�ones�zeros�count)!�
atleast_1d�
atleast_2d�
atleast_3d�mr_�dot�polyfit�cov�corrcoef�median�average�unique�setxor1d� setdiff1d�union1d� intersect1d�in1d�ediff1d�apply_over_axes�apply_along_axis� compress_nd�compress_rowcols� mask_rowcols� clump_masked�clump_unmasked�flatnotmasked_contiguous�notmasked_contiguous�notmasked_edges�
masked_all�masked_all_like�isin�diagflat�stack�vstackc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )� TestGenericc C sJ t d td�}tddgddgtd�}t||� tjddgddgd��}t d
|d�}tddgddg|d�}t||� t d|d�}tddgddggddgddgg|d�}t||� tjddd d gfg�}t d |d�}td d"gd$d&g|d�}t||� t d'|d�}td)d+gd-d/g|d�}t||� t d0|d�}td2ggd4gg|d�}t||� d S )5N� )�dtype� )�maskr6 �a�b�f)�names�formatsr �ba�bb)r5 )r5 )r r )r r )r7 r7 )r7 r7 )r5 r5 )r r )r r )r r )r r )r7 r7 )r7 r7 )r7 r7 )r7 r7 )r9 r; )r> r; )r? r; )r5 �r7 r7 )r7 r@ �r7 r7 )r7 rA �r7 r7 )r7 rB �r7 r7 )r7 rC )r5 �r7 r7 )r7 rD �r7 r7 )r7 rE �r7 r7 )r7 rF �r7 r7 )r7 rG )r7 r7 �r7 r7 )r7 rH �r7 r7 )r7 rI )r. �floatr r �npr6 )�self�test�control�dt� rP �;/tmp/pip-build-5_djhm0z/numpy/numpy/ma/tests/test_extras.py�test_masked_all% s0
zTestGeneric.test_masked_allc C s� t jddtfgdffg�}t jjd|�}tt|d �t jjj� tt|d d �t jjj� tt |d d �d� t|d d j
d� t|d d j j
f � d S )Nr: �cr7 )r7 )r7 )r7 r7 ) rK r6 �object�mar. r �type�corer �lenr � _fill_value)rL �my_dtype�
masked_arrrP rP rQ �"test_masked_all_with_object_nestedC s z.TestGeneric.test_masked_all_with_object_nestedc C sr t jdtdffg�}t jjd|�}tt|d �t jjj� tt |d �d� t|d j
d� t|d j j
f � d S )Nr: r7 )r7 )r7 )r7 r7 ) rK r6 rT rU r. r rV rW r rX r rY )rL rZ r[ rP rP rQ �test_masked_all_with_objectO s z'TestGeneric.test_masked_all_with_objectc C s� t ddgtd�}t|�}t ddgddgtd�}t||� tjddgddgd��}t d
dgddg|d�}t|�}t ddgddg|d�}t||� tjddddgfg�}t ddgd d g|d�}t|�}t||� d S ) Nr7 r5 )r6 )r8 r6 r9 r: r; )r< r= r �
r> r? )r r )r r )r7 r7 )r7 r7 )r^ r^ )r^ r^ )r7 r7 )r7 r7 )r9 r; )r> r; )r? r; �r7 r7 )r7 r_ �r7 r7 )r7 r` �r7 r7 )r7 ra �r7 r7 )r7 rb )r rJ r/ r rK r6 )rL �baserM rN rO rP rP rQ �test_masked_all_likeY s
z TestGeneric.test_masked_all_likec C s� x�t dd�D ]�}x�t d| �D ]�}tj|td�}tj||td�}td| �}|d| @ dk|_d}x"||�D ]}||j| j� 7 }qlW |t kr�t
|j � j� |� q |j |_t
|j � j� |� q W q W d S )Nr7 � r5 )r6 r ) �rangerK r �int�fullr r8 �data�sumr* r �
compressed) rL r; �i�j�k�jar9 �s�slrP rP rQ � check_clumpm s
zTestGeneric.check_clumpc C s^ t tjd��}t|ddddddg< t|�}tdd�tdd �tdd�g}t||� | jt� d S )
Nr^ r r7 r5 � � � � re )r rK r r
r) �slicer rr )rL r9 rM rN rP rP rQ �test_clump_masked} s
z TestGeneric.test_clump_maskedc C sV t tjd��}t|ddddddg< t|�}tdd�td d�g}t||� | jt� d S )
Nr^ r r7 r5 rs rt ru rv re )r rK r r
r* rw r rr )rL r9 rM rN rP rP rQ �test_clump_unmasked� s
z TestGeneric.test_clump_unmaskedc C s� t d�}t|�}t|td|j�g� tjdtd�|_t|td|j�g� t ||dk |dkB |dkB < t|�}t|tdd�tdd�g� t |d d �< t|�}t|g � d S ) Nr^ r )r6 rv rt � rs ru )
r r+ r rw �sizerK r �boolr8 r
)rL r9 rM rP rP rQ � test_flatnotmasked_contiguous� s z)TestGeneric.test_flatnotmasked_contiguousN) �__name__�
__module__� __qualname__rR r\ r] rd rr rx ry r} rP rP rP rQ r4 # s
r4 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � TestAveragec C s< t ddddgddddgd�}tdt|dd �� tdt|ddddgd
�� t|ddddgdd �\}}td|� t|d k� t|d d �< tt|dd �jdg� t ddddgddddgd�}|jd
d
�}t|d d �df< tt|dd �ddg� tt|dd �jd dg� tddgt|dd �� t|ddd�\}}t|ddg� d S )Ng g �?g @g @TF)r8 r )�axis)�weights)r� �returnedg @r5 r7 )r� r� )r r r r r
r8 �reshape)rL Zott�result�wtsrP rP rQ �test_testAverage1� s
z TestAverage.test_testAverage1c C s� ddddddg}ddddddgddddddgg}t dtjd�}tt|dd�d� tt|d|d�d� tt dtjd�dt d� g�}tt|d �tjjtj d��d d
� tt|dd�tj d�d d � tt|dd�t|dd�t|dd�d g� tt|d |d �d� tt|d|d�dddd ddg� tt|dd�t|dd�t|dd�d g� td�}ddddddg}ddddddgddddddgg}t d�}ddddddg} ttt
||�dd�d� ttt
||�dd�d� ttt
||�dd�j dg� ttt
|| �dd�d� tt tt
||�dd��d� t
||�}
tt|
d �d� tt|
dd�ddddddg� tt|
dd�ddg� tt|
d|d�ddddddg� d S )Nr r7 rs )r6 )r� g @)r� r� g @g @g (@)r� g 4@g @g g �?g @g $@Tg �X@g @g @g������
@g������
@)
r rK Zfloat_r r r �add�reducer r r r8 r ) rL �w1�w2�x�y�m1�m2Zm3Zm4Zm5�zrP rP rQ �test_testAverage2� s> &
z TestAverage.test_testAverage2c C sf t d�}t d�d }t||g||ggddd�\}}tt|�t|�� t|j|j� ttd�dddgdd�\}}tt|�t|�� ttd�dd �\}}tt|�t|�� ttd�td�dd
�\}}tt|�t|�� tddgdd ggt�}t|d d gdd gg�}t|dd
�} t| ddg� t|dd
�}
t|
ddg� t|d d
�}
t|
d� t|dd
�}
t|
ddg� d S )Nrs rv r7 T)r� r� r5 r )r� r� r� )r� )r� r� � F)r� g �?g @g �?g @g �?g @)r5 r5 rv )r5 r5 rv )r5 r5 rv )r5 r5 rv g������@)r r r r r r rJ r ) rL r9 r: Zr1r� Zr2r� Za2dZa2dmZa2daZa2dmarP rP rQ �test_testAverage3� s*
z TestAverage.test_testAverage3c C sH t tddg��}t|d� t tddddgddddgd��}t|d� d S ) Nr7 r5 g �?rv r� FT)r8 )r r r )rL r9 rP rP rQ �test_onintegers_with_mask� s
z%TestAverage.test_onintegers_with_maskc
C s t jdddddgdddddggtd�}tddddd gd d d d d gg|d�}t|�}t j|j� �}t|j|j� t|j|j� t|dd�}t|jdd�t|jdd�d }t|j|j� t|j|j� t|dd�}t|jdd�t|jdd�d }t|j|j� t|j|j� t jdddddgdddddgg�} t|| d�}
t j|j� | | d�}t|
j|j� t|
j|j� t|| dd�} t|j| dd�t|j| dd�d }t| j|j� t| j|j� t|| dd�} t|j| dd�t|j| dd�d }t| j|j� t| j|j� d S )!Nr r7 )r6 y @rv y @rz y @re y @y "@y �?r5 y @r� y @y @)r8 )r� g �?g �?g @)r� )r� r� y �? @y @ @y @ @y @ @y �?y @ @y @ @y @ @) rK r r| r r rk r �real�imag)
rL r8 r9 �av�expectedZav0Z expected0Zav1Z expected1r� ZwavZwav0Zwav1rP rP rQ � test_complex� sB zTestAverage.test_complexN)r~ r r� r� r� r� r� r� rP rP rP rQ r� � s
"r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestConcatenatorc C s� t td tddddddg�� td�}dddddg}t||d�}t|dd|f }tt|t�� t |ddddddddddddg � t |jt|dd|f � d S )
Nr7 r5 rv r� rz rs r )r8 )r7 r5 rv r� rz rs ) r r r r r r �
isinstancer r8 )rL r: �m�drS rP rP rQ �test_1d+ s "zTestConcatenator.test_1dc C s0 t jjdd�}t jjdd�}t jt jjdd�d�}t jt jjdd�d�}t||d�}t||d�}td||f }t|jdk� t|d d �d d�f |� t|d d �dd �f |� t|j t j
d||f � t||f }t|jdk� t|d d�d d �f |� t|dd �d d �f |� t|j t j
||f � d S )Nrz r )r8 �1r^ )rz r^ )r^ rz ) rK �random�randZround_r r r r r r8 Zr_)rL Za_1Za_2Zm_1Zm_2Zb_1Zb_2r� rP rP rQ �test_2d6 s zTestConcatenator.test_2dc C sr t tjjdf }t|jddg� t|jd d� t ddgtjjf }t|jdddg� t|jd d� ddg� d S )Nr7 TFr5 )r rK rU r
r r8 ri )rL �actualrP rP rQ �test_masked_constantJ s z%TestConcatenator.test_masked_constantN)r~ r r� r� r� r� rP rP rP rQ r� ( s r� c @ s e Zd Zdd� Zdd� ZdS )�
TestNotMaskedc C s� t tjd�jdd�dddddgdddddgdddddgdddddgdddddggd�}t|d �}t|ddg� t|d�}t|d d
d g� t|d d d
g� t|d�}t|d ddg� t|d ddg� t|jd �}t|ddg� t|jd�}t|d ddg� t|d ddg� t|jd�}t|d ddg� t|d ddg� t|d< t|d�}t|d d d g� t|d d d g� t|d �}t|d d!d"g� t|d d#d$g� d S )%N� rz r r7 )r8 � r5 rv r� )r r r7 r r )r r7 r5 rv r� )rv rv rv r� r� )r r7 r5 rv r� )r r7 r5 rv r� )r r r5 r rv )r r7 r5 rv r� )r� r5 r� r� r� )r r r r r )r r7 r5 rv r� )r� r� r� r� r� )r r7 r5 rv r� ���)r r7 r5 rv r� )r r r r r )r r7 r5 rv r� )r� r� r� r� r� ���)r r r7 r r )r r7 r5 rv r� )r7 r7 r5 r� r� )r r7 r5 rv r� r� )r r7 r5 r� )r r r5 rv )r r7 r5 r� )r� r5 r� r� )r rK r r� r- r ri r
)rL ri rM rP rP rQ �
test_edgesW s:
zTestNotMasked.test_edgesc
C sP t tjd�jdd�ddddddddgddddddddgddddddddggd�}t|d �}t|tddd �tdd d �td
dd �g� t|d�}t|tddd �td dd �gtddd �td dd �gtddd �td dd �gtddd �td dd �gtd dd �gtd dd �gg td dd �gg� t|d�}t|tddd �gg tdd d �td
dd �gg� d S )Nr� rv rt r r7 )r8 r� � � � r5 rs re )r rK r r� r, r rw )rL r9 �tmprP rP rQ �test_contiguousy s0
z TestNotMasked.test_contiguousN)r~ r r� r� r� rP rP rP rQ r� T s "r� c @ s~ e Zd Zdd� Zdd� Zdd� Zejjddd d
g�ejjd d ge j
j d fe j
j d
fg�d
d� ��Z
dd� Zdd� Zdd� ZdS )�TestCompressFunctionsc C s> t jttd?���jddd�}t jd@�jt�}d|dA< t||d�}t|�}t |ddddgd d
d d gd
dddggddddgddddgdddd ggg� t|d�}t |dddddgdd d d d gd d!d
d d gd
d"dddggdd#dddgd$d%d&d'd(gdd)dddgdd*ddd ggg� t|d�}t |dddddgd d!d
d d gd
d"dddggd+d,d-d.d/gd0d1d2d3d4gd5d6d7d8d9ggdd#dddgdd)dddgdd*ddd ggg� t|dB�}t|dC�}t|dE�}t ||� t ||� t ||� t|d�}t |ddddgdd d d gd d
d d gd
dddggd+d-d.d/gd:d;d<d=gd0d2d3d4gd5d7d8d9ggddddgd$d&d'd(gddddgdddd ggg� t|dF�}t|dG�}t|dI�}t ||� t ||� t ||� t|dJ�}t |dddddgd d!d
d d gd
d"dddggdd#dddgdd)dddgdd*ddd ggg� t|dL�}t ||� t|dM�}t |ddddgd d
d d gd
dddggd+d-d.d/gd0d2d3d4gd5d7d8d9ggddddgddddgdddd ggg� t|dO�}t|dQ�}t|dT�}t ||� t ||� t ||� t|dU�}t |ddddgdd d d gd d
d d gd
dddggddddgd$d&d'd(gddddgdddd ggg� t|dW�}t ||� d S )XNrv r� rz Tr7 )r8 r r5 r^ � �
� � � � � �( �* �+ �, �2 �4 �5 �6 �7 �9 �: �; rs re rt ru � r� �) �- �. �/ �0 �1 �3 �8 � � r� r� r� � � � �! �" �# �$ �% �& �' r� � � � r� �< )rv r� rz )r7 r7 r7 )r7 r� r� )r� )r5 r� r� )r� )r r7 r� )r r� )r7 r5 r� )r� r5 r� )r7 r� r� r� )r� r� )r r5 r� )r r� )
rK r �listrf r� r �astyper| r& r )rL r� r� r9 �a2�a3Za4rP rP rQ �test_compress_nd� s�
z&TestCompressFunctions.test_compress_ndc C s� t tjd�jdd�dddgdddgdddggd�}tt|�ddgdd gg� tt|d�dddgd
dd gg� tt|d�dd gddgdd gg� t |jdddgdddgdddggd�}tt|�dd gd
d gg� tt|d�ddd gd
dd gg� tt|d�dd gddgd
d gg� t |jdddgdddgdddggd�}tt|�d gg� tt|d�d
dd gg� tt|d�d gdgd gg� t |jdddgdddgdddggd�}tt|�jd� tt|d�jd� tt|d�jd� d S ) Nru rv r7 r )r8 r� rz re rt rs r5 )r rK r r� r r' �_datar{ )rL r� rP rP rQ �test_compress_rowcols s" "& "& &z+TestCompressFunctions.test_compress_rowcolsc C s� t tjd�jdd�dddgdddgdddggd�}tt|�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� t |jdddgdddgdddggd�}tt|�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� t |jdddgdddgdddggd�}tt|�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� tt|d�jdddgdddgdddgg� t |jdddgdddgdddggd�}tt|�j � t
k� tt|d�j � t
k� tt|d�j � t
k� tt|�jj � � tt|d�jj � � tt|d�jj � � d S )Nru rv r7 r )r8 ) r rK r r� r r( r8 r� r �allr
)rL r� rP rP rQ �test_mask_rowcols s:
&
&
&z'TestCompressFunctions.test_mask_rowcolsr� Nr r7 �func� rowcols_axisc C sf t tjd�jdd�dddgdddgdddggd�}tt��" |||d�}t|t||�� W d Q R X d S )Nru rv r7 r )r8 )r� )r rK r r� r �DeprecationWarningr r( )rL r� r� r� r� �resrP rP rQ �#test_mask_row_cols_axis_deprecation@ s
z9TestCompressFunctions.test_mask_row_cols_axis_deprecationc C s t jdd�}ddddddg}t||d�jdd�}t||d�jdd�}t||dd�}t|jddgddgg� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t j|jd�|jd��� t||d d�}t|t j|jd�|jd��� ddddddg}t||d�jdd�}t||d�jdd�}t||dd�}t|jddgddgg� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t j|jd�|jd��� t|t||�� t||d d�}t|t j|jd�|jd��� ddddddg}t||d�jdd�}t||d�jdd�}t||�}t|jt� t||�}t|jt� t|ddddddgd�jdd�}t|ddddddgd�jdd�}t||dd�}t|jddgddgg� t||d d�}t|t j|jd�|jd��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t j|jd�|jd��� t|ddddddgd�jdd�}t|ddddddgd�jdd�}t||dd�}t|jddgddgg� t||�}t|t j|jd�|jd��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t j|jd�|jd��� t|ddddddgd�jdd�}t|ddddddgd�jdd�}t||dd�}t|jddgddgg� t||d d�}t|t j|jd�|jd��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t j|jd�|jd��� d S )
Nr7 re r )r8 r5 rv T)�strictF) rK r r r� r r r8 �filledr )rL �nr� r9 r: rS rP rP rQ �test_dotL sz $ $
$
$ $z TestCompressFunctions.test_dotc C sn t jd�}t|�}ttt||��tk� ttt||��tk� ttt||��tk� ttt||��tk� d S )Nrv )rK �eyer r rV r r )rL r9 r: rP rP rQ � test_dot_returns_maskedarray� s
z2TestCompressFunctions.test_dot_returns_maskedarrayc C sD t tjd��}t tjd��}t|||d�}t||k� t||� d S )Nrv )�out)rv rv )r rK r� r r r r )rL r9 r� r� rP rP rQ � test_dot_out� s
z"TestCompressFunctions.test_dot_out)r~ r r� r� r� r� �pytest�markZ parametrizerK rU Z mask_rowsZ mask_colsr� r� r� r� rP rP rP rQ r� � s p $
F r� c @ s e Zd Zdd� Zdd� ZdS )�TestApplyAlongAxisc C s@ t d�jddd�}dd� }t|d|�}t|ddgdd gg� d S )
Ng (@r5 rv c S s | d S )Nr7 rP )r: rP rP rQ �myfunc� s z*TestApplyAlongAxis.test_3d.<locals>.myfuncr7 r� re r^ )r r� r% r )rL r9 r� �xarP rP rQ �test_3d� s zTestApplyAlongAxis.test_3dc C sF t d�jddd�}d dd�}t|d|dd�}t|dd gd
d gg� d S )
Nr� r5 rv r c S s | d| S )Nr7 rP )r: �offsetrP rP rQ r� � s z1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfuncr7 )r� rz rt r� )r )r r� r% r )rL r9 r� r� rP rP rQ �test_3d_kwargs� s
z!TestApplyAlongAxis.test_3d_kwargsN)r~ r r� r� r� rP rP rP rQ r� � s
r� c @ s e Zd Zdd� ZdS )�TestApplyOverAxesc C s� t d�jddd�}ttj|ddg�}tjdgdgdggg�}t||� t||d jt �< ttj|ddg�}tjd gd
gdggg�}t||� d S ) Nr� r5 rv r� r r� �\ �| r� r� )
r r� r$ rK rj r r r
r� r| )rL r9 rM ZctrlrP rP rQ �
test_basic� s
z TestApplyOverAxes.test_basicN)r~ r r� r� rP rP rP rQ r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-S ).�
TestMedianc C s4 t jjt jt jgt jt jggdd�}t|t j� d S )Nr7 )r� r� )rK rU r �infr )rL �rrP rP rQ � test_pytype� s $zTestMedian.test_pytypec C s� t jjt jjt jt jgt jt jgg�dd�}t|t j� t jjt jjt jt jgt jt jgg�d d�}t|t j� t jjt jjt jt jgt jt jggdd�dd�}t|jd� t jjt jjt jt jgt jt jggdd�d d�}t|jd� d S )Nr7 )r� T)r8 r� r� )rK rU r r r r r8 )rL r rP rP rQ �test_inf� s zTestMedian.test_infc C sl t jd�}tt jj|�d� ttt jj|��tk � td�}tt jj|�d� ttt jj|��tk � d}tt jj|�d� ttt jj|��tk � t jd �j dd�}tt jj|dd�t j|dd�� tt jj|d d�t j|d d�� tt jj|d d�tk � t jd �j dd�}tt jj|dd�t j|dd�� tt jj|d d�t j|d d�� tt jj|d d�tk � d S )
Nru g @rt g @rz g @r )r� r7 g @�H g R@)
rK r r rU r r rV r rf r� )rL r� rP rP rQ �test_non_masked� s"
zTestMedian.test_non_maskedc C s t tjd�dgd dgd d�}ttjj|�d� ttjj|�jf d� tttjj|��t k � t tjd�j
d d
�dgd dgd d�}ttjj|�d � ttjj|�jf d� tttjj|��t k � tjj|dd
d�}t|ddg� t|jdd� tt|�t k� dS )z5test the examples given in the docstring of ma.medianrt r r� r7 )r8 g �?zshape mismatchr^ r5 rz rs g @T)r� �overwrite_inputg @g @Nr� )r5 ) r rK r r rU r r r rV r r� )rL r� Zma_xrP rP rQ �test_docstring_examples� s "*z"TestMedian.test_docstring_examplesc C s d}�xt d�D ]�}x�dD ]�}td||d�}tjt | |�ddg�}xP|D ]H\}}ytjj|||d� W qJ tk
r� t|||||f ��Y qJX qJW tj|d |gddg�}xR|D ]J\}}ytjj|||d� W n tj k
r� Y q�X t|||||f ��q�W qW qW d S ) Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %srz FTr7 )�ndminr8 )r� r )FT)
rf r � itertools�productrK rU r � Exception�AssertionErrorZ AxisError)rL �msgr r8 r� �argsr� ZoverrP rP rQ �test_axis_argument_errors s"
z$TestMedian.test_axis_argument_errorsc C sD t ddd�}ttjj|�d� t ddd�}ttjj|�tjj� d S )Nr7 F)r8 T)r r rK rU r r
)rL r� rP rP rQ �test_masked_0d s zTestMedian.test_masked_0dc C s� t tjd�dd�}ttjj|�tjj� ttjj|�jf d� tt tjj|��tjj
j k� t tjd�dd�}ttjj|�d� ttjj|�jf d� tt tjj|��t k � t tjd�dddddgd�}ttjj|�d � ttjj|�jf d� tt tjj|��t k � t tjd�dddddgd�}ttjj|�d
� ttjj|�jf d� tt tjj|��t k � t tjd�dddddgd�}ttjj|�d � ttjj|�jf d� tt tjj|��t k � t tjd �dddddgd�}ttjj|�d � ttjj|�jf d� tt tjj|��t k � t tjd
�ddddddgd�}ttjj|�d � ttjj|�jf d� tt tjj|��t k � t tjd�ddddddgd�}ttjj|�d � ttjj|�jf d� tt tjj|��t k � d S )Nrz T)r8 zshape mismatchFg @r r7 g @g g @g @rs g @)
r rK r r rU r r
r r rV rW ZMaskedConstantr )rL r� rP rP rQ �test_masked_1d s@ zTestMedian.test_masked_1dc C sJ t tjjtdddgdddgd��jtjjtdddgdddgd��j� d S )Nr7 r5 rv r )r8 )r rK rU r r r )rL rP rP rQ �test_1d_shape_consistencyF s "z$TestMedian.test_1d_shape_consistencyc C s d \}}t tjd
d|��}t |d d�< |d d �< t tj||ftd��}|d d � |d d �df< tjt|��}x2td|�D ]$}tj j
|� || |d d �|f< qzW t t |d d �df �d� t t |�d� t t |dd�tj
|�� t t |jdd�tj
|�� d S ) N�e r� g �?r^ )r6 r r7 )r� )r r� g �i����)r rK Zlinspacer
�emptyrJ r rX rf r� �shuffler r r �T)rL r� �pr� r� �idxrl rP rP rQ r� J s zTestMedian.test_2dc C s� t tjd�jdd��}t |d d�< |d
d �< tt|�d� tttj j|��t
k � tt|dd�dddg� tttj j|dd��t
k� tt|d d�ddddd
d d dddg
� tttj j|d d��t
k� tt|d d�j d d d ddddd d d g
� d S )Nr� r^ rv g -@r )r� g +@g /@r7 r� r� r� ���) r rK r r� r
r r r rV rU r r8 )rL r� rP rP rQ �
test_2d_waxisZ s & zTestMedian.test_2d_waxisc C s� t jjd�jddd�}t||d dk< tt|d�ddgdd gddgd
d gg� d|_tt|d�d d gd
d gddgg� t jjd�jddd�}t||d dk< tt|d�dd gddgddgg� d S )Nr� rv r� r5 r r� ru rs r� r� �c r^ r� r� r� rz rt r� r� )r� rv r5 )rK rU r r� r
r r r )rL r� rP rP rQ r� f s ("zTestMedian.test_3dc C sL t tjd�jdd��}t |d d�< |dd �< tt|dd�t|dd�� d S )Nr� r^ rv r7 )r� r r� )r rK r r� r
r r )rL r� rP rP rQ �
test_neg_axisq s zTestMedian.test_neg_axisc C s� x�d D ]~}t tj|��}t |d d�< |d
d �< t tjf ��}t||d�}|dkr^t|d� n
t|d� t||k� tt|�t k� qW d S ) Nr� � >@r� � ?@rv )r� g -@g .@)r� r r� r r )
r rK r r
r r r r rV r )rL �vr� r� r rP rP rQ � test_out_1dv s
zTestMedian.test_out_1dc C s �x dD �]}t tj|�jdd��}t |d d�< |dd �< t tjd��}t|d|d�}|dkr�t d gd dd
d d g d gd d
gd dgd d
gd d�}nBt d gd ddddg d gd d
gd dgd d
gd d�}t||� t||k� tt |�t
k� qW d S )Nr� � D@r� � >@r^ r7 rv )r� r� g r� r� r� TFr� )r8 g +@g �1@g �5@g �9@)r� r" r� r# r� r ) r rK r r� r
r r r r rV r )rL r r� r� r �erP rP rQ �test_out� s $ "
zTestMedian.test_outc C sF ddgddgddgg}t jj|d�}ddg}tt jj|dd�|� d S )Ng �?g g @r )r� )rK rU Z masked_equalr r )rL ri r[ r� rP rP rQ �$test_single_non_masked_value_on_axis� s
z/TestMedian.test_single_non_masked_value_on_axisc C s� �x$dt jdtd�fD �] }t jjdt jdgdddgg�}||_t jj|d d�}tt j |�� t
|t j� t jj|j � dd�}tt j |�� t
|t j� t jj|dd�}t t
|�t� t
|dt jdg� t jj|dd�}t t
|�t� t
|t jdg� t jj|d
d�}t t
|�t� t
|t jdg� qW t jjdt jdgdddgg�}t jj|d d �df< t
t jj|d d�t j� t
t jj|dd�dt jdg� t
t jj|dd�t jd g� d S ) NFrs )r6 r7 rv r5 )r� r g �?r� )rK r r| rU r �nanr8 r r �isscalarr Zravelr rV r r
)rL r8 �dmr rP rP rQ �test_nan� s. zTestMedian.test_nanc C s� t jjt jd��}t jjt jd��}t j|d < t jj|d
< tt jj|d|d�|� t jjt jd ��}tt jj|d|d�|� t jjt jf ��}tt jj||d�|� d S ) Nr� rv r5 r7 r )r� )r� )rv r� )r5 r7 )r5 r5 )rv ) rK rU r r r r' r
r r )rL �or� rP rP rQ � test_out_nan� s
zTestMedian.test_out_nanc C s� t jjt jdtd��}t jj|d d d�< t j|d< tt jj|�t j� tt jj|dd�t j� t jjt jdtd�j ddd��}t j|j
�d dk|_ |j � }t j|d < t j|d
< tt jj|�t j� t
t jt jj|��� t jj|dd�}t j|d < t j|d < tt jj|d�|� t jj|dd�}t j|d
< t j|d< tt jj|d�|� t jj|dd�}t j|d< t j|d< tt jj|d�|� d S )Nr� )r6 rv r5 r )r� r� r7 )r7 r5 rv )r7 r7 r5 )r5 rv )r7 r5 )r7 rv )r7 r5 )r r5 )r r5 )rK rU r r rJ r
r' r r r� r{ r8 �copyr r( r )rL r9 Zaorigr: rP rP rQ �test_nan_behavior� s0
z TestMedian.test_nan_behaviorc C s� t jdddgdddggt jd�}t jj||dkd�}tt jj|dd�d� tt jj|dd�jd� tt jj|dd�|d � tt jj|�d� d S ) Nrv � )r6 )r8 r7 )r� Fr )rK r Zuint8rU r r r r8 )rL r9 rP rP rQ �test_ambigous_fill� s z TestMedian.test_ambigous_fillc C s �xt jt j gD �]�}t j|t jgt jt jgg�}t jj|t j|�d�}tt jj|dd�|t jg� tt jj|dd�|t jg� tt jj|�|� t jt jt j|gt jt j|gg�}t jj|t j|�d�}t t jj|dd�|� t t jj|dd�j
d� t t jj|dd�|d � t t jj|�|� t j||g||gg�}tt jj|�|� tt jj|dd�|� tt jj|dd�|� t j|d| dgdt jt jd gd
t jt j|ggt j d �}t jj|t j|�d�}|dk�rtt jj|dd�d d
| dg� tt jj|�d� n2tt jj|dd�dd
| dg� tt jj|�d� tt jj|dd�dd|g� x�t dd�D ]�}x�t dd�D ]�}t jt jg| |g| gd �}t jj|t j|�d�}tt jj|�|� tt jj|dd�|� tt jj|dd�t jg| |g| � �qpW �q`W qW d S ) N)r8 r )r� r7 Fre ru r^ rz r� )r6 g @g @g @g @g $@g "@g @g �?r5 i����i����g $�g "�g �g �g �)
rK r r r' rU r �isnanr r r r8 Zfloat32rf )rL r r9 rl rm rP rP rQ � test_special� sF "
"zTestMedian.test_specialc C st t jjt jg td��}t� �8}|jt� tt jj |�t j
� t |j d j
tk� W d Q R X t jjt jg tdd��}t� �F}|jt� tjddt� tt jj |�t j
� t |j d j
tk� W d Q R X t jjt jg tdd��}tt jj |dd�|� tt jj |d d�|� t jjt jt j
tdd��}tjd
d ��<}tjddt� tt jj |dd�|� t |d j
tk� W d Q R X d S ) N)r6 r rv )r6 r �always� r5 )r� r7 T)�record)rK rU r r rJ r r5 �RuntimeWarningr r r' r �log�category�warnings�filterwarningsr �catch_warnings)rL r9 �wr: rP rP rQ �
test_empty s&
zTestMedian.test_emptyc C sX t jjt jd��}ttt jj|jt���t � t j
|d< ttt jj|jt���t � d S )Ng @r5 ) rK rU r r r rV r r� rT rJ r' )rL r+ rP rP rQ � test_object: s
zTestMedian.test_objectN)r~ r r� r r r r r r r r r� r r� r r! r% r& r* r, r. r0 r2 r= r> rP rP rP rQ r � s, & # + r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestCovc C s t tjjd��| _d S )Nr� )r rK r� r� ri )rL rP rP rQ �setupC s z
TestCov.setupc C sZ | j }ttj|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� d S )NF)�rowvarT)rA �bias)ri r rK r )rL r� rP rP rQ �test_1d_without_missingF s
z TestCov.test_1d_without_missingc C sb | j jdd�}ttj|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� d S )Nrv r� F)rA T)rA rB )ri r� r rK r )rL r� rP rP rQ �test_2d_without_missingN s
z TestCov.test_2d_without_missingc C s@ | j }t|d< ||j� 8 }|j� }ttj|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� yt|dd� W n tk
r� Y nX |dd� }ttj||d d d � �t||d d d
� �� ttj||d d d � dd�t||d d d � dd�� ttj||d d d
� ddd�t||d d d� ddd�� d S )Nr7 F)rA T)rA rB )� allow_maskedr� r� r� r� r� r� r� r� )ri r
�meanrk r rK r �
ValueError)rL r� �nxrP rP rQ �test_1d_with_missingV s$ , z TestCov.test_1d_with_missingc C s6 | j }t|d < |jdd�}tjt|��jt�}tj||j �}||j
d�d d �d f j d�}t t
|�tj
|�|jd d |d � t t
|dd�tj
|dd�|jd | � tj|j |�}||j
d� j d�}t t
|dd �tj
|dd �|jd d |d � t t
|ddd
�tj
|ddd
�|jd | � d S ) Nr7 rv r� r g �?T)rB F)rA )rA rB r� )ri r
r� rK Z logical_notr
r� rg r r rF r� r r r )rL r� Zvalid�fracZxfrP rP rQ �test_2d_with_missingn s " &z TestCov.test_2d_with_missingN)r~ r r� r@ rC rD rI rK rP rP rP rQ r? A s
r? c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� TestCorrcoefc C s( t tjjd��| _t tjjd��| _d S )Nr� )r rK r� r� ri �data2)rL rP rP rQ r@ � s zTestCorrcoef.setupc
C s� | j | j }}tj|�}tj||�}t� ��}tjd� ttt|dd� |j td� t
tj|dd�t|dd�� t
t|dd�|� t
t||d d�|� t
t|dd�|� t
t||dd�|� W d Q R X d S )
Nr3 r7 )�ddofz bias and ddof have no effectr rv r� r� r� ) ri rM rK r r r9 � simplefilterr r� �filterr )rL r� r� r� Z expected2�suprP rP rQ � test_ddof� s
zTestCorrcoef.test_ddofc C s� | j | j }}tj|�}t� �b}tjd� ttt||dd� ttt||dd� ttt|dd� |j td� t
t|dd�|� W d Q R X d S )Nr3 TF)rB z bias and ddof have no effectr7 ) ri rM rK r r r9 rO r r� rP r )rL r� r� r� rQ rP rP rQ � test_bias� s
zTestCorrcoef.test_biasc C sx | j }ttj|�t|�� ttj|dd�t|dd�� t� �2}|jtd� ttj|ddd�t|ddd�� W d Q R X d S )NF)rA z bias and ddof have no effectT)rA rB )ri r rK r r rP r� )rL r� rQ rP rP rQ rC � s z$TestCorrcoef.test_1d_without_missingc C s� | j jdd�}ttj|�t|�� ttj|dd�t|dd�� t� �2}|jtd� ttj|ddd�t|ddd�� W d Q R X d S )Nrv r� F)rA z bias and ddof have no effectT)rA rB )ri r� r rK r r rP r� )rL r� rQ rP rP rQ rD � s z$TestCorrcoef.test_2d_without_missingc C s� | j }t|d < ||j� 8 }|j� }ttj|�t|�� ttj|dd�t|dd�� t� �2}|jt d� ttj|ddd�t|ddd�� W d Q R X yt|dd� W n t
k
r� Y nX |dd � }ttj||d d d
� �t||d d d� �� ttj||d d d� dd�t||d d d� dd�� t� �r}|jt d� ttj||d d d� �t||d d d� dd�� ttj||d d d� �t||d d d� d d
�� W d Q R X d S )Nr7 F)rA z bias and ddof have no effectT)rA rB )rE )rB r5 )rN r� r� r� r� r� r� r� r� r� r� ) ri r
rF rk r rK r r rP r� rG )rL r� rH rQ rP rP rQ rI � s2 , z!TestCorrcoef.test_1d_with_missingc
C s | j }t|d< |jdd�}t|�}tj|�}t|d d �d d
�f |d d �d d �f � t� ��}|jtd� tt|d
d�d d�d d�f |d d�d d�f � tt|dd�d d�d d�f |d d�d d�f � tt|dd�d d�d d�f |d d�d d�f � W d Q R X d S )Nr7 rv r� z bias and ddof have no effectr5 )rN )rB r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ) ri r
r� r rK r r rP r� )rL r� rM rN rQ rP rP rQ rK � s
* z!TestCorrcoef.test_2d_with_missingN)
r~ r r� r@ rR rS rC rD rI rK rP rP rP rQ rL � s
rL c @ s e Zd Zdd� Zdd� ZdS )�TestPolynomialc C s� t jjd�}t jjd�jd
d�}tt||d�t j||d�� |jt�}t|d< |jt�}t |d < |d< t||d d �df ddd�\}}}}}t j|dd � |dd �df j � ddd�\}} }
} } x4t
|||||f|| |
| | f�D ]\}
}t|
|� q�W t||d d �df ddd�\}}}}}t j|dd� |dd�df ddd�\}} }
} } x6t
|||||f|| |
| | f�D ]\}
}t|
|� �qnW t||ddd�\}}}}}t j|dd� |dd�d d �f ddd�\}} }
} } x6t
|||||f|| |
| | f�D ]\}
}t|
|� �q�W t jjd�d }|j � }|dd� }|dd� }|dd� }t||dd|d �\}}}}}t j||dd|d �\}} }
} } t ||� x6t
|||||f|| |
| | f�D ]\}
}t|
|� �q�W d S )Nr^ r� r7 r5 rv r T)rh )rh r< r� )r r r� r� )r� r� r� r� r� r� r� r� r� r� r� )
rK r� r� r� r r �viewr r
rk �zipr- r )rL r� r� �C�R�K�S�DrS r rn rp r� r9 �a_r< Zwo�xsZys�wsrP rP rQ � test_polyfit� s<
&"(&0(4(
(zTestPolynomial.test_polyfitc C s� t jjd�}t jjd�jd d�}t j|d< t j|d < |jt�}|jt�}t|d< t|d< t||ddd�\}}}}}t j|dd� |dd�d d �f ddd�\}} }
} } x4t |||||f|| |
| | f�D ]\}
}t
|
|� q�W d S )Nr^ r� r7 r5 r rv T)rh r� r� r� )r� r� r� r� )r� r� r� r� ) rK r� r� r� r' rU r r
r rV r )rL r� r� rW rX rY rZ r[ rS r rn rp r� r9 r\ rP rP rQ � test_polyfit_with_masked_NaNs s
4(z,TestPolynomial.test_polyfit_with_masked_NaNsN)r~ r r� r_ r` rP rP rP rQ rT � s 'rT c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�TestArraySetOpsc C s� ddddddg}t |ddd�}tt|d t�� t|d tdddgdddgd�� t|d dddg� t|d ddddddg� d S ) Nr7 r5 rv T)� return_index�return_inverser )r8 rz )r r r� r r r )rL ri rM rP rP rQ �test_unique_onlist+ s "z"TestArraySetOps.test_unique_onlistc C s t ddddddgddddddgd�}t|ddd�}t|d t dddd
gddddgd�� t|d ddddg� t|d ddddddg� d|_t ddddddgddddddgdd �}t|ddd�}t|d t dddd gddddgd�� t|d ddddg� t|d ddddddg� d S ) Nr7 r5 rv r )r8 T)rb rc rz )ri r8 �
fill_valuer� r� )r r r re )rL ri rM rP rP rQ �test_unique_onmaskedarray4 s $&&z)TestArraySetOps.test_unique_onmaskedarrayc C s� t dddgdd�}t|ddd�}t|d t dgdgd�� t|d dg� t|d dddg� t}t|ddd�}t|d t t�� t|d dg� t|d dg� d S )Nr7 T)r8 )rb rc r r5 )r r r r
)rL ri rM rP rP rQ �test_unique_allmaskedD s z%TestArraySetOps.test_unique_allmaskedc C sr t tjd�dddddgd�}tddddgddddgd�}t|�}t||� t|jd�|jd�� t|j|j� d S )Nrz r7 r )r8 r� )r rK r r r# r r� r8 )rL r� rN rM rP rP rQ � test_ediff1dS s
z TestArraySetOps.test_ediff1dc C s� t tjd�dddddgd�}t|td�}tdddddgdddddgd�}t||� t|jd�|jd�� t|j|j� t|dddgd�}tdddddddgdddddddgd�}t||� t|jd�|jd�� t|j|j� d S ) Nrz r7 r )r8 )�to_beginr� r5 rv ) r rK r r# r
r r r� r8 )rL r� rM rN rP rP rQ �test_ediff1d_tobegin\ s
(
z$TestArraySetOps.test_ediff1d_tobeginc C s� t tjd�dddddgd�}t|td�}tdddddgdddddgd�}t||� t|jd�|jd�� t|j|j� t|dddgd�}tdddddddgdddddddgd�}t||� t|jd�|jd�� t|j|j� d S ) Nrz r7 r )r8 )�to_endr� r5 rv ) r rK r r# r
r r r� r8 )rL r� rM rN rP rP rQ �test_ediff1d_toendk s
(
z"TestArraySetOps.test_ediff1d_toendc
C s� t tjd�dddddgd�}t|ttd�}tddddddgddddddgd�}t||� t|jd�|jd�� t|j|j� t|dddgtd�}tddddddddgddddddddgd�}t||� t|jd�|jd�� t|j|j� d S ) Nrz r7 r )r8 )rk ri r� r5 rv ) r rK r r# r
r r r� r8 )rL r� rM rN rP rP rQ �test_ediff1d_tobegin_toendz s $
z*TestArraySetOps.test_ediff1d_tobegin_toendc C s� t jd�}t|�}tddddgddddgd�}t||� tt|t�� t|jd�|jd�� t|j |j � t|t
t
d�}tddddddgddddddgd�}tt|t�� t|jd�|jd�� t|j |j � d S )Nrz r7 r )r8 )rk ri ) rK r r# r r r r� r r� r8 r
)rL r� rM rN rP rP rQ �test_ediff1d_ndarray� s
$z$TestArraySetOps.test_ediff1d_ndarrayc C sh t ddddgddddgd�}t ddddgddddgd�}t||�}t dddgdddgd�}t||� d S )Nr7 rv r )r8 r� )r r! r )rL r� r� rM rN rP rP rQ �test_intersect1d� s
z TestArraySetOps.test_intersect1dc C s� t ddddd gdddddgd�}t dddddd gddddddgd�}t||�}t|t dddg�� t ddddd
gdddddgd�}dddddg}t||�}t|t ddddgddddgd�� t dddg�}t d ddg�}t||�}tt|t�� t|dddddd g� t dd
ddgddddgd�}t d ddd
gddddgd�}t||�}tt|t�� t|dddddd g� tg tg g �� d S )Nr7 r5 rz re r )r8 rv r� rs rt r� r� r� r� )r r r r r� r r )rL r9 r: rM rP rP rQ �
test_setxor1d� s& $
"
z TestArraySetOps.test_setxor1dc C s� t jd�jdddg�}t jdddg�}d|d< t||d�}tddd d
ddd d d
g dddddddddg d�}tdtd�}d|d< d|d< d|d< t||�}tt|t �� t
||� t j|||j �|j @ }t
||� d S )Nr� r5 rv r� r7 r )r8 r^ r� r� r� r� r� )ri r8 )r6 T)r7 r5 r )r5 rv r� )r r r )r r r7 )r r5 rv ) rK r r� r r r| r0 r r� r r r8 )rL r9 r8 r: ZecrS r� rP rP rQ � test_isin� s
zTestArraySetOps.test_isinc C s� t ddddd gdddddgd�}t dddddd gddddddgd�}t||�}t|d d d d
d g� t ddddd
gdddddgd�}t dddgdddgd�}t||�}t|d d d
d d g� tg tg g �� d S )Nr7 r5 rz re r )r8 rv r� TFr� r� r� r� )r r" r r )rL r9 r: rM rP rP rQ � test_in1d� s $
zTestArraySetOps.test_in1dc C s� t ddddd gdddddgd�}t dddddd gddddddgd�}ttjt||��t||d d
�� t ddddd
gdddddgd�}t dddgdddgd�}ttjt||��t||d d
�� tg tg g d d
�� d S )Nr7 r5 rz re r )r8 rv r� T)�invertr� r� r� r� )r r rK rs r" r )rL r9 r: rP rP rQ �test_in1d_invert� s $ z TestArraySetOps.test_in1d_invertc C s t dddddd gddddddgd�}t dddddd
gddddddgd�}t||�}t ddddddd gdddddddgd�}t||� t dddgdddggdddgdddggd�}t dddddgdddddgd�}t ddddddgddddddgd�}t||�}t||� tg tg g �� d S ) Nr7 r5 rz re r )r8 rv r� r� r� r� )r r r r ) rL r9 r: rM rN r� r� Zezr� rP rP rQ � test_union1d� s $$
(
, $
z TestArraySetOps.test_union1dc
C s� t ddddddddgddddddddgd�}t ddd d dddg�}t||�}t|t ddd
gdddgd�� td
�}td �}tt||�t d d g�� t g tjg d�}tt|g �jtj� d S )Nrs rz r� re r7 r5 r )r8 rv r^ rt ru r� )r r r r rK Zuint32r6 )rL r9 r: rM rP rP rQ �test_setdiff1d� s ,
z TestArraySetOps.test_setdiff1dc C s< t jdddg�}t jdddg�}tt||�t jdg�� d S )Nr9 r: rS rp )rK r r r )rL r9 r: rP rP rQ �test_setdiff1d_char_array
s z)TestArraySetOps.test_setdiff1d_char_arrayN)r~ r r� rd rf rg rh rj rl rm rn ro rp rq rr rt ru rv rw rP rP rP rQ ra ) s
ra c @ s e Zd Zdd� Zdd� ZdS )�
TestShapeBasec C sr t dddgdddgd�}t|�}t|jd� t|jj|jj� t|jd� t|jj|jj� t|jj|jj� d S )Nr r7 r5 )r8 rv )r7 rv )rv )r r r r r8 ri )rL r9 r: rP rP rQ �test_atleast_2d s z TestShapeBase.test_atleast_2dc C s� t d�}t|jd� t|jj|j� t|jj|j� t dd�}x8|D ]0}t|jd� t|jj|j� t|jj|j� qDW td�}t|jd� t|jj|j� t|jj|j� tdd�}x8|D ]0}t|jd� t|jj|j� t|jj|j� q�W td�}t|jd� t|jj|j� t|jj|j� tdd�}x:|D ]2}t|jd � t|jj|j� t|jj|j� �q4W td�}t|jd
� t|jj|jj� d S ) Ng �?r7 g @)r7 )r7 )r7 r7 )r7 r7 )r7 r7 r7 )r7 r7 r7 )r7 r7 )r r r r8 ri r r r1 )rL r: r9 rP rP rQ �test_shape_scalar s<
z TestShapeBase.test_shape_scalarN)r~ r r� ry rz rP rP rP rQ rx s
rx c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestStackc C s� t dddgdddgd�}t dddgdddgd�}t||gdd�}t|jd
� t|j|d j� t|j|d j� t||g�}t|j|j� t|j|j� t||gdd�}t|jd � t|j|d d �df j� t|j|d d �df j� d S ) Nr r7 r5 )r8 ru rt re )r� rv )r5 rv )rv r5 )r r2 r r r r8 r3 ri )rL r9 r: rS r� rP rP rQ �
test_stack_1dK s zTestStack.test_stack_1dc C s� t dddgdd�}t dddgd d�}t||gdd
�}t|jd � t|j|d j� t|j|d j� t||g�}t|j|j� t|j|j� t||gdd
�}t|jd
� t|j|d d �df j� t|j|d d �df j� d S )Nr r7 r5 T)r8 ru rt re F)r� rv )r5 rv )rv r5 )r r2 r r r r8 r3 ri )rL r9 r: rS r� rP rP rQ �test_stack_masks] s zTestStack.test_stack_masksc
C s� d }t jjdd|�}t jjdd|�}t jjdd|�jt�}t jjdd|�jt�}t||d�}t||d�}t||gdd�}d | } t|j| � t |j
|d j
� t |j
|d j
� t||gd
d�}|d } t|j| � t |j
|d j
� t |j
|d j
� d}t jjdd|�}t jjdd|�}t jjdd|�jt�}t jjdd|�jt�}t||d�}t||d�}t||gdd�}d| } t|j| � t |j
|d j
� t |j
|d j
� t||gdd�}|d } t|j| � t |j
|d j
� t |j
|d j
� d S )Nrv r5 r r^ )r8 )r� r7 .r� rz )rv r5 )r5 r� )r5 ).r ).r7 )rv r5 r� rz )r5 r� )r5 ).r ).r7 ) rK r� �randintr� r| r r2 r r r r8 )
rL ZshpZd1Zd2r� r� Za1r� rS Zc_shprP rP rQ �
test_stack_ndo sD zTestStack.test_stack_ndN)r~ r r� r| r} r rP rP rP rQ r{ I s r{ )J�__doc__r9 r
r� ZnumpyrK Z
numpy.testingr r Znumpy.ma.testutilsr r r r Z
numpy.ma.corer r r
r r r
r r r r r Znumpy.ma.extrasr r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r� r� r� r� r� r� r r? rL rT ra rx r{ rP rP rP rQ �<module> s: 4�
,F
}Ck: i8 |