Edit File: test_extras.cpython-37.pyc
B ��Fd�� � @ s d Z ddlmZmZmZ ddlZddlZddlZddl m Z m Z m Z m Z mZ ddlmZmZmZmZ ddlmZmZmZmZmZmZmZm Z m Z m Z m Z ddl 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;m<Z<m=Z=m>Z>m?Z? ddl m@ mAZB G dd � d e �ZCG d d � d e �ZDG d d � d e �ZEG dd� de �ZFG dd� de �ZGG dd� de �ZHG dd� de �ZIG dd� de �ZJG dd� de �ZKG dd� de �ZLG d d � d e �ZMG d d � d e �ZNG d d!� d!e �ZOePd"k�re � dS )#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 $ � )�division�absolute_import�print_functionN)�TestCase�run_module_suite� assert_warns�suppress_warnings� assert_raises)�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�diagflatc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestGenericc C sJ t dtd�}tddgddgtd�}t||� t�ddgddgd��}t d|d�}td d gd d g|d�}t||� t d |d�}td d gd d ggd d gd d gg|d�}t||� t�d dd dgfg�}t d|d�}tddgddg|d�}t||� t d|d�}tddgddg|d�}t||� t d |d�}tdggdgg|d�}t||� d S )N)� )�dtype� )�maskr: �a�b�f)�names�formats)r r )r; r; )r9 r9 )r= r? )�bar? )�bbr? )r; )r; r; )r4 �floatr r �npr: )�self�test�control�dt� rJ ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/tests/test_extras.py�test_masked_all( s0 zTestGeneric.test_masked_allc C s� t ddgtd�}t|�}t ddgddgtd�}t||� t�ddgddgd��}t d d gd d g|d�}t|�}t d d gd d g|d�}t||� t�d dd dgfg�}t ddgddg|d�}t|�}t||� d S )Nr; r9 )r: )r< r: r= r> r? )r@ rA )r r )r; r; )� rM )r= r? )rB r? )rC r? )r; )r; r; )r rD r5 r rE r: )rF �baserG rH rI rJ rJ rK �test_masked_all_likeF 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| �� 7 }qlW |t kr�t |� � �� |� q |j |_t |� � �� |� q W q W d S )Nr; � r9 )r: r ) �rangerE r �int�fullr r< �data�sumr0 r � compressed) rF r? �i�j�k�jar= �s�slrJ rJ rK � check_clumpZ s zTestGeneric.check_clumpc C s^ t t�d��}t|ddddddg< t|�}tdd�tdd �tdd�g}t||� | �t� d S ) NrM r r; r9 � � � � rP )r rE r r r/ �slicer r] )rF r= rG rH rJ rJ rK �test_clump_maskedj s z TestGeneric.test_clump_maskedc C sV t t�d��}t|ddddddg< t|�}tdd�td d�g}t||� | �t� d S ) NrM r r; r9 r^ r_ r` ra rP )r rE r r r0 rb r r] )rF r= rG rH rJ rJ rK �test_clump_unmaskedu s z TestGeneric.test_clump_unmaskedc C s� t d�}t|�}t|td|j�� t||dk |dkB |dkB < t|�}t|tdd�tdd�g� t|d d �< t|�}t|d � d S )NrM r ra r_ � r^ r` )r r1 r rb �sizer )rF r= rG rJ rJ rK � test_flatnotmasked_contiguous s z)TestGeneric.test_flatnotmasked_contiguousN) �__name__� __module__� __qualname__rL rO r] rc rd rg rJ rJ rJ rK r8 & s r8 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestAveragec C s> t ddddgddddgd�}tdt|dd �� tdt|ddddgd �� t|ddddgd d �\}}td|� | �|d k� t|d d �< tt|dd �jdg� t ddddgddddgd�}|�dd�}t|d d �d f< tt|dd �ddg� tt|d d �jd dg� tddgt|dd �� t|dd d�\}}t|ddg� d S )Ng g �?g @g @TF)r< r )�axis)�weightsr; )rm �returnedg @r9 )rl rn )r r r" � assertTruer r< �reshape)rF �ott�result�wtsrJ rJ rK �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 �tj�t� d��d d � tt|dd�t� d�d d � tt|dd�t|dd�t|dd�d g� tt|d |d �d � tt|d|d�d ddd 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�d dddddg� tt| dd�ddg� tt| d|d�d dddddg� d S )Nr r; r^ )r: )rl g @)rl rm g @g @g (@)rm g������ @g g �?g @g $@Tg �X@g @g @) r rE �float_r r" r �add�reducer r r r< r ) rF �w1�w2�x�y�m1�m2�m3�m4Zm5�zrJ rJ rK �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|�� tdd gdd ggt�}t|d d gd d gg�}t|dd�} t| ddg� t|dd�} t| ddg� t|d d�} t| d� t|dd�} t| ddg� d S )Nr^ ra r; )rl rn )r9 r9 ra r )rl rm rn )rn )rm rn r9 � FT)rl g �?g @g �?g������@g �?g @)r r" r r r r rD r ) rF r= r> �r1rx �r2ry �a2d�a2dmZa2daZa2dmarJ rJ rK �test_testAverage3� s* z TestAverage.test_testAverage3c C sH t tddg��}t|d� t tddddgddddgd��}t|d� d S ) Nr; r9 g �?ra r� FT)r< )r" r r )rF r= rJ rJ rK �test_onintegers_with_mask� s z%TestAverage.test_onintegers_with_maskc C s t jdddddgdddddggtd�}tdddddgdd d d d gg|d �}t|�}t �|�� �}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 �dddddgdddddgg�} t|| d�} t 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 r; )r: y �? @y @ @y @ @y @ @y "@y �?y @ @y @ @y @ @)r< )rl g �?g �?g @)rm )rm rl ) rE r �boolr r" rV r �real�imag) rF r< r= �av�expectedZav0� expected0Zav1� expected1rs ZwavZwav0Zwav1rJ rJ rK � test_complex� sB zTestAverage.test_complexN)rh ri rj rt r� r� r� r� rJ rJ rJ rK rk � s "rk c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestConcatenatorc C s� t td tddddddg�� td�}dddddg}t||d �}t|dd|f }| �t|t�� t |ddddddddddddg � t |jt|dd|f � d S ) N)r; r9 ra r� re r^ r; r9 ra r� re r^ r )r< ) r r r r r ro � isinstancer r< )rF r> �m�d�crJ rJ rK �test_1d s "zTestConcatenator.test_1dc C s4 t j�dd�}t j�dd�}t �t j�dd�d�}t �t j�dd�d�}t||d�}t||d�}td||f }| �|jdk� t|d d �d d�f |� t|d d �dd �f |� t|j t j d||f � t||f }| �|jdk� t|d d�d d �f |� t|dd �d d �f |� t|j t j ||f � d S )Nre r )r< �1)re rM )rM re ) rE �random�rand�round_r r ro r r r< �r_)rF �a_1Za_2Zm_1Zm_2Zb_1Zb_2r� rJ rJ rK �test_2d s zTestConcatenator.test_2dc C s t tjjdd� � d S )Nc S s t d S )Nz 1, 2; 3, 4)r rJ rJ rJ rK �<lambda>5 � z6TestConcatenator.test_matrix_builder.<locals>.<lambda>)r rE �ma�MAError)rF rJ rJ rK �test_matrix_builder4 s z$TestConcatenator.test_matrix_builderc C sP t d }tj�tjd �}t||� tt|�t|�� tt|j�t|j�� d S )N)�rr; r9 ra ) r rE r� r r� r r �typerT )rF �actualr� rJ rJ rK � test_matrix7 s z TestConcatenator.test_matrixN)rh ri rj r� r� r� r� rJ rJ rJ rK r� s r� c @ s e Zd Zdd� Zdd� ZdS )� TestNotMaskedc C s� t t�d��dd�dddddgdddddgdddddgdddddgdddddggd�}t|d �}t|ddg� t|d�}t|d ddg� t|d d dg� t|d�}t|d dd g� t|d dd g� t|jd �}t|ddg� t|jd�}t|d d dg� t|d d dg� t|jd�}t|d dd g� t|d dd g� t|d< t|d�}t|d ddg� t|d ddg� t|d�}t|d ddg� t|d ddg� d S )N� re r r; )r< � )r r r; r r )r r; r9 ra r� )ra ra ra r� r� )r r r9 r ra )r� r9 r� r� r� )r r r r r )r� r� r� r� r� ������)r; r; r9 r� r� )r r; r9 r� )r r r9 ra )r� r9 r� r� )r rE r rp r3 r rT r )rF rT rG rJ rJ rK � test_edgesD s: zTestNotMasked.test_edgesc C sd t t�d��dd�ddddddddgddddddddgddddddddggd�}t|d �}t|d tddd �� t|d td d d �� t|d tdd d �� t|d�}| �t|d �dk� | �|d d k� t|d |d � | �t|d �dk� t|d�}t|d d tdd d �� | �|d d k� t|d d tddd �� t|d d tddd �� d S )Nr� ra r_ r r; )r< r� � r� � � ���r� r9 rP r^ ) r rE r rp r2 r rb ro �len)rF r= �tmprJ rJ rK �test_contiguousf s$ z TestNotMasked.test_contiguousN)rh ri rj r� r� rJ rJ rJ rK r� A s "r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestCompressFunctionsc C s> t �ttd����ddd�}t �d��t�}d|d< t||d�}t|�}t |d d ddgd d d dgddddggddddgddddgdd d d ggg� t|d �}t |d d d ddgdd d!d"d#gd d$d d dgdd%dddggdd&dddgd'd(d)d*d+gdd,dddgdd-d d d ggg� t|d �}t |d d d ddgd d$d d dgdd%dddggd.d/d0d1d2gd3d4d5d6d7gd8d9d:d;d<ggdd&dddgdd,dddgdd-d d d ggg� t|d=�}t|d>�}t|d?�}t ||� t ||� t ||� t|d �}t |d d ddgdd!d"d#gd d d dgddddggd.d0d1d2gd@dAdBdCgd3d5d6d7gd8d:d;d<ggddddgd'd)d*d+gddddgdd d d ggg� t|dD�}t|dE�}t|dF�}t ||� t ||� t ||� t|dG�}t |d d d ddgd d$d d dgdd%dddggdd&dddgdd,dddgdd-d d d ggg� t|dH�}t ||� t|dI�}t |d d ddgd d d dgddddggd.d0d1d2gd3d5d6d7gd8d:d;d<ggddddgddddgdd d d ggg� t|dJ�}t|dK�}t|dL�}t ||� t ||� t ||� t|dM�}t |d d ddgdd!d"d#gd d d dgddddggddddgd'd)d*d+gddddgdd d d ggg� t|dN�}t ||� d S )ON�< ra r� re )ra r� re T)r; r; r; )r< r r9 rM � � � � � � � �( �* �+ �, �2 �4 �5 �6 �7 �9 �: �; r; r^ rP r_ r` � r� �) �- �. �/ �0 �1 �3 �8 � � r� r� r� � � � �! �" �# �$ �% �& �' )r; r� )r� r� � � � )r9 r� )r� )r r; )r r� )r; r9 )r� r9 )r; r� )r� r� )r r9 )r r� ) rE r �listrQ rp r �astyper� r, r )rF rz r� r= �a2�a3�a4rJ rJ rK �test_compress_nd� s� z&TestCompressFunctions.test_compress_ndc C s� t t�d��dd�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 ) Nr` ra r; r )r< r� re rP r_ r^ r9 )r rE r rp r r- �_datarf )rF rz rJ rJ rK �test_compress_rowcols� s" "& "& &z+TestCompressFunctions.test_compress_rowcolsc C s� t t�d��dd�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�}| �t|�� � t k� | �t|d�� � t k� | �t|d�� � t k� | �t|�j� � � | �t|d�j� � � | �t|d�j� � � d S )Nr` ra r; r )r< ) r rE r rp r r. r< r� ro �allr )rF rz rJ rJ rK �test_mask_rowcols s: & & &z'TestCompressFunctions.test_mask_rowcolsc C s t �dd�}ddddddg}t||d��dd�}t||d��dd�}t||dd�}t|jddgddgg� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t �|�d�|�d��� t||d d�}t|t �|�d�|�d��� ddddddg}t||d��dd�}t||d��dd�}t||dd�}t|jddgddgg� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t �|�d�|�d��� t|t||�� t||d d�}t|t �|�d�|�d��� ddddddg}t||d��dd�}t||d��dd�}t||�}t|jt� t||�}t|jt� t|ddddddgd��dd�}t|ddddddgd��dd�}t||dd�}t|jddgddgg� t||d d�}t|t �|�d�|�d��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t �|�d�|�d��� t|ddddddgd��dd�}t|ddddddgd��dd�}t||dd�}t|jddgddgg� t||�}t|t �|�d�|�d��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t �|�d�|�d��� t|ddddddgd��dd�}t|ddddddgd��dd�}t||dd�}t|jddgddgg� t||d d�}t|t �|�d�|�d��� t||dd�}t|jdddgdddgdddgg� t||d d�}t|t �|�d�|�d��� d S ) Nr; rP r )r< r9 ra T)�strictF) rE r r rp r r r< �filledr )rF �nr� r= r> r� rJ rJ rK �test_dot$ sz $ $ $ $ $z TestCompressFunctions.test_dotc C sn t �d�}t|�}ttt||��tk� ttt||��tk� ttt||��tk� ttt||��tk� d S )Nra )rE �eyer r r� r r )rF r= r> rJ rJ rK � test_dot_returns_maskedarrayj s z2TestCompressFunctions.test_dot_returns_maskedarrayc C sD t t�d��}t t�d��}t|||d�}t||k� t||� d S )Nra )ra ra )�out)r rE r� r r r r )rF r= r� �resrJ rJ rK � test_dot_outs s z"TestCompressFunctions.test_dot_outN) rh ri rj r� r� r� r� r� r� rJ rJ rJ rK r� ~ s p F r� c @ s e Zd Zdd� Zdd� ZdS )�TestApplyAlongAxisc C s@ t d��ddd�}dd� }t|d|�}t|ddgdd gg� d S ) Ng (@r9 ra c S s | d S )Nr; rJ )r> rJ rJ rK �myfunc� s z*TestApplyAlongAxis.test_3d.<locals>.myfuncr; r� rP rM )r rp r+ r )rF r= r� �xarJ rJ rK �test_3d} s zTestApplyAlongAxis.test_3dc C sF t d��ddd�}d dd�}t|d|dd�}t|dd gd d gg� d S ) Nr� r9 ra r c S s | d| S )Nr; rJ )r> �offsetrJ rJ rK r� � s z1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfuncr; )r� re r_ r� )r )r rp r+ r )rF r= r� r� rJ rJ rK �test_3d_kwargs� s z!TestApplyAlongAxis.test_3d_kwargsN)rh ri rj r� r� rJ rJ rJ rK r� { s r� c @ s e Zd Zdd� ZdS )�TestApplyOverAxesc C s� t d��ddd�}ttj|ddg�}t�dgdgdggg�}t||� t||d �tj �< ttj|ddg�}t�d gd gdggg�}t||� d S ) Nr� r9 ra r� r r� �\ �| r� r� ) r rp r* rE rU r r r r� r� )rF r= rG �ctrlrJ rJ rK � test_basic� s z TestApplyOverAxes.test_basicN)rh ri rj r� rJ rJ rJ rK 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 )Nr� )rl )rE r� r! �infr )rF r� rJ rJ rK � test_pytype� s $zTestMedian.test_pytypec C s� t jjt j�t jt jgt jt jgg�dd�}t|t j� t jjt j�t 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 )Nr� )rl T)r< )rE r� r! r r r r< )rF r� rJ rJ rK �test_inf� s zTestMedian.test_infc C sl t �d�}tt j�|�d� ttt j�|��tk � td�}tt j�|�d� ttt j�|��tk � d}tt j�|�d� ttt j�|��tk � t �d�� dd�}tt jj|dd �t j|dd �� tt jj|d d �t j|d d �� tt jj|d d �tk � t �d �� dd�}tt jj|dd �t j|dd �� tt jj|d d �t j|d d �� tt jj|d d �tk � d S ) Nr` g @r_ g @re g @�H r )rl r; g R@) rE r r r� r! r r� r rQ rp )rF rz rJ rJ rK �test_non_masked� s" zTestMedian.test_non_maskedc C s t t�d�dgd dgd d�}ttj�|�d� ttj�|�jdd� tttj�|��t k � t t�d �� d d �dgd dgd d�}ttj�|�d � ttj�|�jdd� tttj�|��t k � tjj|ddd�}t|ddg� t|jdd� tt|�t k� dS )z5test the examples given in the docstring of ma.medianr_ r r� r; )r< g �?rJ zshape mismatchrM r9 re r^ g @r� T)rl �overwrite_inputg @g @)r9 N) r rE r r r� r! r r r� r rp )rF rz Zma_xrJ rJ rK �test_docstring_examples� s "*z"TestMedian.test_docstring_examplesc C s d}x�t d�D ]�}x�dD ]�}td||d�}t�t | |�ddg�}xH|D ]@\}}ytjj|||d� W qH t|||||f ��Y qHX qHW t�|d |gddg�}xR|D ]J\}}ytjj|||d� W n tjk r� Y q�X t|||||f ��q�W qW qW d S ) Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sre )FTr; )�ndminr< FT)rl r ) rQ r � itertools�productrE r� r! �AssertionError� AxisError)rF �msgr r< rz �argsrl �overrJ rJ rK �test_axis_argument_errors� s" z$TestMedian.test_axis_argument_errorsc C sD t ddd�}ttj�|�d� t ddd�}ttj�|�tjj� d S )Nr; F)r< T)r r rE r� r! r )rF rz rJ rJ rK �test_masked_0d� s zTestMedian.test_masked_0dc C s� t t�d�dd�}ttj�|�tjj� ttj�|�jdd� tt tj�|��tjj j k� t t�d�dd�}ttj�|�d� ttj�|�jdd� tt tj�|��t k � t t�d�dd dddgd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � t t�d�dd d d d gd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � t t�d�dd d ddgd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � t t�d �dd d ddgd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � t t�d�dd d d d dgd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � t t�d�dd d d d dgd�}ttj�|�d � ttj�|�jdd� tt tj�|��t k � d S )Nre T)r< rJ zshape mismatchFg @r r; g @g g @g @r^ g @) r rE r r r� r! r r r r� �core�MaskedConstantr )rF rz rJ rJ rK �test_masked_1d� s@ zTestMedian.test_masked_1dc C sJ t tj�tdddgdddgd��jtj�tdddgdddgd��j� d S )Nr; r9 ra r )r< )r rE r� r! r r )rF rJ rJ rK �test_1d_shape_consistency s "z$TestMedian.test_1d_shape_consistencyc C s d\}}t t�dd|��}t |d d�< |dd �< t tj||ftd��}|d d � |d d �df< t�t|��}x2td|�D ]$}tj � |� || |d d �|f< qzW t t |d d �df �d� t t |�d� t t |dd �t� |�� t t |jdd �t� |�� d S ) N)�e r� g �g �?rM i����)r: r r; )rl )r rE �linspacer �emptyrD r r� rQ r� �shuffler r! r �T)rF r� �prz r� �idxrW rJ rJ rK r� " s zTestMedian.test_2dc C s� t t�d��dd��}t |d d�< |dd �< tt|�d� tttj �|��t k � tt|dd�ddd g� 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� rM ra r� g -@r )rl g +@g /@r; r� r� r� ) r rE r rp r r r! r r� r� r r< )rF rz rJ rJ rK � test_2d_waxis2 s & zTestMedian.test_2d_waxisc C s� t j�d��ddd�}t||d dk< tt|d�ddgdd gddgd d gg� d |_tt|d�d d gdd gddgg� t j�d��ddd�}t||d dk< tt|d�dd gddgddgg� d S )Nr� ra r� r9 r r� r` r^ r� r� )r� ra r9 �c rM r� r� r� re r_ r� r� )rE r� r rp r r r! r )rF rz rJ rJ rK r� > s ("zTestMedian.test_3dc C sL t t�d��dd��}t |d d�< |dd �< tt|dd�t|dd�� d S )Nr� rM ra r� r� )rl r; )r rE r rp r r r! )rF rz rJ rJ rK � test_neg_axisI s zTestMedian.test_neg_axisc C s� x�dD ]~}t t�|��}t |d d�< |dd �< t t�d��}t||d�}|dkr^t|d� n t|d� t||k� tt|�t k� qW d S ) N)r� g >@r� g ?@ra r� rJ )r� r� g -@g .@) r rE r r r r! r r r� r )rF �vrz r� r� rJ rJ rK � test_out_1dN s zTestMedian.test_out_1dc C s �x dD �]}t t�|��dd��}t |d d�< |dd �< t t�d��}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 )N)r� g D@r� g >@rM r� ra r� r; )rl r� r� g r� r� r� TFr� )r< g +@g �1@g �5@g �9@) r rE r rp r r r! r r r� r )rF r rz r� r� �erJ rJ rK �test_out\ s $ " zTestMedian.test_outc C sF ddgddgddgg}t j�|d�}ddg}tt jj|dd�|� d S )Ng �?g g @r )rl )rE r� � masked_equalr r! )rF rT Z masked_arrr� rJ rJ rK �$test_single_non_masked_value_on_axism s z/TestMedian.test_single_non_masked_value_on_axisc C s� t � ���}|�t� �x&dtjdtjd�fD �] }tj�dtjdgdddgg�}||_ tjj |d d�}t t� |�� t |tj� tjj |�� dd�}t t� |�� t |tj� tjj |dd�}tt|�t� t |dtjdg� tjj |dd�}tt|�t� t |tjdg� tjj |d d�}tt|�t� t |tjdg� q,W tj�dtjdgdddgg�}tjj|d d �df< t tjj |d d�tj� t tjj |dd�dtjdg� t tjj |dd�tjd g� td d � |jD �d gd � W d Q R X d S )NFr^ )r: r; ra r9 )rl r r� g �?c S s g | ]}|j tk�qS rJ )�category�RuntimeWarning)�.0rz rJ rJ rK � <listcomp>� s z'TestMedian.test_nan.<locals>.<listcomp>Tr� )r �recordr& rE r r� r� r �nanr< r! r �isscalarr �ravelr r� r r �log)rF �wr< �dmr� rJ rJ rK �test_nanv s6 zTestMedian.test_nanc C s� t jdd��� t �ddt� tj�t�d��}tj�t�d��}tj |d< tjj |d< t tjj |d |d �|� tj�t�d ��}t tjj |d |d �|� tj�t�d ��}t tjj ||d �|� W d Q R X d S )NT)r) �always� )r� )ra r� )r9 r; )r9 r9 r )r� )ra r; rJ ) �warnings�catch_warnings�filterwarningsr& rE r� r r r r* r r r! )rF �or� rJ rJ rK � test_out_nan� s zTestMedian.test_out_nanc C s� t j�t jdtd��}t jj|d d d�< t j|d< t� �d}|�t � t t j� |�t j� t t jj |dd�t j� t |j d jt k� t |j d jt k� W d Q R X t j�t jdtd��ddd��}t �|j�d dk|_|�� }t j|d < t j|d < t� �\}|�t � t�d d t � t t j� |�t j� t t �t j� |��� t |j d jt k� W d Q R X t jj |dd�}t j|d < t j|d< tjdd��6}t�d d t � tt j� |d�|� tt|�d� W d Q R X t jj |dd�}t j|d< t j|d< tjdd��6}t�d d t � tt j� |d�|� tt|�d� W d Q R X t jj |dd�}t j|d< t j|d< tjdd��6}t�d d t � tt j� |d�|� tt|�d� W d Q R X d S )Nr� )r: ra r9 r )rl r; r� )r; r9 ra )r; r; r9 r1 r2 )r9 ra )r; r9 T)r) )r; ra )r r9 )rE r� r r rD r r* r r) r& r r! r r- r% rp rf r< �copyr3 r5 r+ r4 r r� )rF r= r. Zaorigr> rJ rJ rK �test_nan_behavior� sR 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 j�|�d� d S ) Nra � )r: )r< r; )rl Fr )rE r �uint8r� r r r! r< )rF r= rJ rJ rK �test_ambigous_fill� s z TestMedian.test_ambigous_fillc C s �xt jt j gD �]�}t �|t jgt jt jgg�}t jj|t �|�d�}tt jj|dd�|t jg� tt jj|dd�|t jg� tt j�|�|� t �t jt j|gt jt j|gg�}t jj|t �|�d�}t t jj|dd�|� t t jj|dd�j d� t t jj|dd�|d � t t j�|�|� t �||g||gg�}tt j�|�|� tt jj|dd�|� tt jj|dd�|� t j|d| dgdt jt jd gd t jt j|ggt j d �}t jj|t �|�d�}|dk�rtt jj|dd�d d | dg� tt j�|�d� n2tt jj|dd�dd | dg� tt j�|�d� tt jj|dd�dd|g� x�t dd�D ]�}x�t dd�D ]�}t �t jg| |g| gd �}t jj|t �|�d�}tt j�|�|� tt jj|dd�|� tt jj|dd�t jg| |g| � �qpW �q`W qW d S )N)r< r )rl r; FrP i����i����re r� )r: g @g @g @g @g $�g "�g �g �rM r9 ) rE r r r* r� r �isnanr r! r r< �float32rQ )rF r r= rW rX rJ rJ rK � test_special� sF " "zTestMedian.test_specialc C st t j�t jg td��}t� �8}|�t� tt j� |�t j � t |j d j tk� W d Q R X t j�t jg tdd��}t� �F}|�t� t�ddt� tt j� |�t j � t |j d j tk� W d Q R X t j�t jg tdd��}tt jj |dd�|� tt jj |d d�|� t j�t jt j tdd��}tjd d ��<}t�ddt� tt jj |dd�|� t |d j tk� W d Q R X d S ) N)r: r ra )r: r r1 r2 r9 )rl r; T)r) )rE r� r r rD r r) r& r r! r* r r- r% r3 r5 r r4 )rF r= r. r> rJ rJ rK � test_empty s& zTestMedian.test_emptyc C sX t j�t �d��}ttt j�|�t���t � t j |d< ttt j�|�t���t � d S )Ng @r9 ) rE r� r r r r� r! r� �objectrD r* )rF r6 rJ rJ rK � test_object) s zTestMedian.test_objectN)rh ri rj r r r r r r r r r� r r� r r r" r$ r0 r7 r9 r< r? r@ rB rJ rJ rJ rK r� � s, & ! 4 + r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestCovc C s t tj�d��| _d S )Nr� )r rE r� r� rT )rF rJ rJ rK �setUp2 s z TestCov.setUpc C sZ | j }tt�|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� d S )NF)�rowvarT)rE �bias)rT r rE r )rF rz rJ rJ rK �test_1d_without_missing5 s z TestCov.test_1d_without_missingc C sb | j �dd�}tt�|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� d S )Nra r� F)rE T)rE rF )rT rp r rE r )rF rz rJ rJ rK �test_2d_without_missing= s z TestCov.test_2d_without_missingc C s@ | j }t|d< ||�� 8 }|�� }tt�|�t|�� ttj|dd�t|dd�� ttj|ddd�t|ddd�� yt|dd� W n tk r� Y nX |dd� }tt�||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 )Nr� F)rE T)rE rF )� allow_maskedr; )rT r �meanrV r rE r � ValueError)rF rz �nxrJ rJ rK �test_1d_with_missingE s$ , z TestCov.test_1d_with_missingc C s6 | j }t|d< |�dd�}t�t|���t�}t�||j �}||� d�d d �d f � d�}t t |�t� |�|jd d |d � t t |dd�tj |dd�|jd | � t�|j |�}||� d� � d�}t t |d d �tj |d d �|jd d |d � t t |d dd �tj |d dd �|jd | � d S ) Nr� ra r� r; r g �?T)rF F)rE )rE rF )rT r rp rE � logical_notr r� rR r r rJ r� r r r )rF rz �validZfrac�xfrJ rJ rK �test_2d_with_missing] s " &z TestCov.test_2d_with_missingN)rh ri rj rD rG rH rM rQ rJ rJ rJ rK rC 0 s rC c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestCorrcoefc C s( t tj�d��| _t tj�d��| _d S )Nr� )r rE r� r� rT �data2)rF rJ rJ rK rD u s zTestCorrcoef.setUpc C s� | j | j }}t�|�}t�||�}t� ��}t�d� ttt|dd� |� td� t tj|dd�t|dd�� t t|dd�|� t t||dd�|� t t|dd�|� t t||dd�|� W d Q R X d S )Nr1 r� )�ddofz bias and ddof have no effectr ra ) rT rS rE r r r3 � simplefilterr �DeprecationWarning�filterr )rF rz r{ r� Z expected2�suprJ rJ rK � test_ddofy s zTestCorrcoef.test_ddofc C s� | j | j }}t�|�}t� �b}t�d� ttt||dd� ttt||dd� ttt|dd� |� td� t t|dd�|� W d Q R X d S )Nr1 TF)rF z bias and ddof have no effectr; ) rT rS rE r r r3 rU r rV rW r )rF rz r{ r� rX rJ rJ rK � test_bias� s zTestCorrcoef.test_biasc C sx | j }tt�|�t|�� ttj|dd�t|dd�� t� �2}|�td� ttj|ddd�t|ddd�� W d Q R X d S )NF)rE z bias and ddof have no effectT)rE rF )rT r rE r r rW rV )rF rz rX rJ rJ rK rG � s z$TestCorrcoef.test_1d_without_missingc C s� | j �dd�}tt�|�t|�� ttj|dd�t|dd�� t� �2}|�td� ttj|ddd�t|ddd�� W d Q R X d S )Nra r� F)rE z bias and ddof have no effectT)rE rF )rT rp r rE r r rW rV )rF rz rX rJ rJ rK rH � s z$TestCorrcoef.test_2d_without_missingc C s� | j }t|d< ||�� 8 }|�� }tt�|�t|�� ttj|dd�t|dd�� t� �2}|�t d� ttj|ddd�t|ddd�� W d Q R X yt|dd� W n t k r� Y nX |dd� }tt�||d d d� �t||d d d� �� ttj||d d d� dd�t||d d d� dd�� t� �r}|�t d� tt�||d d d� �t||d d d� dd �� tt�||d d d� �t||d d d� d d �� W d Q R X d S ) Nr� F)rE z bias and ddof have no effectT)rE rF )rI r; )rF r9 )rT ) rT r rJ rV r rE r r rW rV rK )rF rz rL rX rJ rJ rK rM � s2 , z!TestCorrcoef.test_1d_with_missingc C s | j }t|d< |�dd�}t|�}t�|�}t|d d�d d�f |d d�d d�f � t� ��}|�td� tt|dd�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 ) Nr� ra r� z bias and ddof have no effectr� )rT r; )rF ) rT r rp r rE r r rW rV )rF rz rG rH rX rJ rJ rK rQ � s * z!TestCorrcoef.test_2d_with_missingN) rh ri rj rD rY rZ rG rH rM rQ rJ rJ rJ rK rR s s rR c @ s e Zd Zdd� Zdd� ZdS )�TestPolynomialc C s� t j�d�}t j�d��dd�}tt||d�t �||d�� |�t�}t|d< |�t�}t |d< |d< t||d d �df dd d �\}}}}}t j|d d � |d d �df � � dd d �\}} } } } x4t |||||f|| | | | f�D ]\} }t| |� q�W t||d d �df dd d �\}}}}}t j|d d� |d d�df dd d �\}} } } } x6t |||||f|| | | | f�D ]\} }t| |� �qnW t||dd d �\}}}}}t j|d d� |d d�d d �f dd d �\}} } } } x6t |||||f|| | | | f�D ]\} }t| |� �q�W t j�d�d }|� � }|d d� }|d d� }|d d� }t||dd |d �\}}}}}t j||dd |d �\}} } } } t ||� x6t |||||f|| | | | f�D ]\} }t| |� �q�W d S ) NrM r� r� r9 ra r )r r )r� r� T)rS r; )rS r. ) rE r� r� rp r r �viewr r rV �zipr8 r )rF rz r{ �C�R�K�S�Dr� r� rY r[ r� r= �a_r. Zwo�xs�ys�wsrJ rJ rK � test_polyfit� s< &"(&0(4( (zTestPolynomial.test_polyfitc C s� t j�d�}t j�d��dd�}t j|d< t j|d< |�t�}|�t�}t|d< t|d< t||ddd �\}}}}}t j|d d� |d d�d d �f ddd �\}} } } } x4t |||||f|| | | | f�D ]\} }t | |� q�W d S ) NrM r� r� r9 r )r� r� ra T)rS r; ) rE r� r� rp r* r\ r r r r] r )rF rz r{ r^ r_ r` ra rb r� r� rY r[ r� r= rc rJ rJ rK � test_polyfit_with_masked_NaNs s 4(z,TestPolynomial.test_polyfit_with_masked_NaNsN)rh ri rj rg rh rJ rJ rJ rK 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!S )"�TestArraySetOpsc C s� ddddddg}t |ddd�}| �t|d t�� t|d tdddgdddgd�� t|d dddg� t|d ddddddg� d S ) Nr; r9 ra T)� return_index�return_inverser )r< re )r# ro r� r r r )rF rT rG rJ rJ rK �test_unique_onlist s "z"TestArraySetOps.test_unique_onlistc C s t ddddddgddddddgd�}t|ddd�}t|d t ddddgddddgd�� t|d ddd dg� t|d ddddddg� d|_t ddddddgddddddgdd �}t|ddd�}t|d t ddddgddddgd�� t|d ddd dg� t|d ddddddg� d S ) Nr; r9 ra r )r< T)rj rk r� re )rT r< � fill_value)r r# r rm )rF rT rG rJ rJ rK �test_unique_onmaskedarray# 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 )Nr; T)r< )rj rk r r9 )r r# r r )rF rT rG rJ rJ rK �test_unique_allmasked3 s z%TestArraySetOps.test_unique_allmaskedc C sr t t�d�dddddgd�}tddddgddddgd�}t|�}t||� t|�d�|�d�� t|j|j� d S )Nre r; r )r< r� )r rE r r r) r r� r< )rF rz rH rG rJ rJ rK � test_ediff1dB s z TestArraySetOps.test_ediff1dc C s� t t�d�dddddgd�}t|td�}tdddddgdddddgd�}t||� t|�d�|�d�� t|j|j� t|dddgd�}tdddddddgdddddddgd�}t||� t|�d�|�d�� t|j|j� d S ) Nre r; r )r< )�to_beginr� r9 ra ) r rE r r) r r r r� r< )rF rz rG rH rJ rJ rK �test_ediff1d_tobeginK s ( z$TestArraySetOps.test_ediff1d_tobeginc C s� t t�d�dddddgd�}t|td�}tdddddgdddddgd�}t||� t|�d�|�d�� t|j|j� t|dddgd�}tdddddddgdddddddgd�}t||� t|�d�|�d�� t|j|j� d S ) Nre r; r )r< )�to_endr� r9 ra ) r rE r r) r r r r� r< )rF rz rG rH rJ rJ rK �test_ediff1d_toendZ s ( z"TestArraySetOps.test_ediff1d_toendc C s� t t�d�dddddgd�}t|ttd�}tddddddgddddddgd�}t||� t|�d�|�d�� t|j|j� t|dddgtd�}tddddddddgddddddddgd�}t||� t|�d�|�d�� t|j|j� d S ) Nre r; r )r< )rs rq r� r9 ra ) r rE r r) r r r r� r< )rF rz rG rH rJ rJ rK �test_ediff1d_tobegin_toendi s $ z*TestArraySetOps.test_ediff1d_tobegin_toendc C s� t �d�}t|�}tddddgddddgd�}t||� | �t|t�� t|�d�|�d�� t|j |j � t|t t d�}tddddddgddddddgd�}| �t|t�� t|�d�|�d�� t|j |j � d S )Nre r; r )r< )rs rq ) rE r r) r r ro r� r r� r< r )rF rz rG rH rJ rJ rK �test_ediff1d_ndarrayy s $z$TestArraySetOps.test_ediff1d_ndarrayc C sh t ddddgddddgd�}t ddddgddddgd�}t||�}t dddgdddgd�}t||� d S )Nr; ra r )r< r� )r r' r )rF rz r{ rG rH rJ rJ rK �test_intersect1d� s z TestArraySetOps.test_intersect1dc C s� t dddddgdddddgd�}t dddd ddgddddddgd�}t||�}t|t dd dg�� t dddddgdddddgd�}dddd dg}t||�}t|t dd ddgddddgd�� t dddg�}t d dd g�}t||�}tt|t�� t|dddd dd g� t dd ddgddddgd�}t d dd d gddddgd�}t||�}tt|t�� t|dddd dd g� tg tg g �� d S ) Nr; r9 re rP r� r )r< ra r� r^ r_ )r r$ r r r� r r )rF r= r> rG rJ rJ rK � test_setxor1d� s& $ " z TestArraySetOps.test_setxor1dc C s� t �d��dddg�}t �dddg�}d|d< t||d�}tdd d d ddd d dg dddddddddg d�}tdtd�}d|d< d|d< d|d< t||�}tt|t �� t ||� t �|||j �|j @ }t ||� d S )Nr� r9 ra r� r; )r; r9 r )r< r rM r� r� r� r� r� )rT r< )r9 ra r� )r: T)r r r )r r r; )r r9 ra ) rE r rp r r r� r6 r r� r r r< )rF r= r< r> �ecr� r� rJ rJ rK � test_isin� s zTestArraySetOps.test_isinc C s� t dddddgdddddgd�}t dddd ddgddddddgd�}t||�}t|d d d d d g� t dddddgdddddgd�}t dddgdddgd�}t||�}t|d d d d d g� tg tg g �� d S ) Nr; r9 re rP r� r )r< ra r� TF)r r( r r )rF r= r> rG rJ rJ rK � test_in1d� s $ zTestArraySetOps.test_in1dc C s� t dddddgdddddgd�}t dddd ddgddddddgd�}tt�t||��t||d d �� t dddddgdddddgd�}t dddgdddgd�}tt�t||��t||d d �� tg tg g d d �� d S ) Nr; r9 re rP r� r )r< ra r� T)�invert)r r rE r| r( r )rF r= r> rJ rJ rK �test_in1d_invert� s $ z TestArraySetOps.test_in1d_invertc C s� t ddddddgddddddgd�}t dddd ddgddddddgd�}t||�}t dddd dddgdddddddgd�}t||� tg tg g �� d S ) Nr; r9 re rP r� r )r< ra r� )r r& r r )rF r= r> rG rH rJ rJ rK � 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 )Nr^ re r� rP r; r9 r )r< ra r� rM r_ r` )r r% r r rE �uint32r: )rF r= r> rG rJ rJ rK �test_setdiff1d� s , z TestArraySetOps.test_setdiff1dc C s< t �dddg�}t �dddg�}tt||�t �dg�� d S )Nr= r> r� r[ )rE r r r% )rF r= r> rJ rJ rK �test_setdiff1d_char_array� s z)TestArraySetOps.test_setdiff1d_char_arrayN)rh ri rj rl rn ro rp rr rt ru rv rw rx rz r{ r} r~ r� r� rJ rJ rJ rK ri s ri 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 r; r9 )r< )r; ra )ra )r r r r r< rT )rF r= r> rJ rJ rK �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 �?)r; g @)r; r; )r; r; r; )r r r r< rT r r r7 )rF r> r= rJ rJ rK �test_shape_scalar s< z TestShapeBase.test_shape_scalarN)rh ri rj r� r� rJ rJ rJ rK r� � s r� �__main__)Q�__doc__� __future__r r r r3 r �numpyrE Z numpy.testingr r r r 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/ r0 r1 r2 r3 r4 r5 r6 r7 r� �extrasZmaer8 rk r� r� r� r� r� r� rC rR r[ ri r� rh rJ rJ rJ rK �<module> s< 4�i /= ~ Ck: a8