Edit File: test_nanfunctions.cpython-37.pyc
B ��FdE� � @ s d dl mZmZmZ d dlZd dlZd dlmZm Z m Z m Z m Z m Z mZmZmZ e�dejddejdgdd ejd d d gejejejd ejdgdejejddejgg�Ze�ddddg�e�dd d d d g�e�d dg�e�dddg�gZe�ddddddgdd dd d d gdddd ddgddddddgg�Ze�ddddddgdd dd d d gdddd ddgddddddgg�ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de�ZG d d � d e e�Z G d d � d e e�Z G d d!� d!e e�Z G d"d#� d#e �Z G d$d%� d%e �Z e!d&k�re� dS )'� )�division�absolute_import�print_functionN) �run_module_suite�TestCase�assert_� assert_equal�assert_almost_equal�assert_no_warnings� assert_raises�assert_array_equal�suppress_warningsg��s���?g&S��:�?g�����?g�ʡE��?g� � �?g�����g�z�G��?gL� F%u�?g6<�R�!�?g[B>�٬�?g�:M� �g�S㥛�?gz�):���?g�� h"�?g �?g c @ st e Zd ZejejgZejejgZ dd� Z dd� Z dd� Z dd� Z d d � Zd d � Zd d� Zdd� Zdd� Zdd� ZdS )�TestNanFunctions_MinMaxc C s. t �� }x | jD ]}||� t|t � qW d S )N)�_ndat�copy�nanfuncsr )�self�ndat�f� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_nanfunctions.py� test_mutation+ s z%TestNanFunctions_MinMax.test_mutationc C sd t �d�}xTt| j| j�D ]B\}}x8dD ]0}|||dd�}|||dd�}t|j|jk� q(W qW d S )N� )Nr � T)�axis�keepdims)�np�eye�zipr �stdfuncsr �ndim)r �mat�nf�rfr �tgt�resr r r � test_keepdims2 s z%TestNanFunctions_MinMax.test_keepdimsc C sb t �d�}xRt| j| j�D ]@\}}t �d�}||dd�}||d|d�}t||� t||� qW d S )Nr r )r )r �out)r r r r r �zerosr )r r! r" r# �resoutr$ r% r r r �test_out: s z TestNanFunctions_MinMax.test_outc C s� d}x�t | j| j�D ]|\}}xr|D ]j}tjd|d�}||dd�jj}||dd�jj}t||k� ||d d�jj}||d d�jj}t||k� q"W qW d S )N�efdgFDGr )�dtyper )r )r r r r r r, �typer )r �codesr" r# �cr! r$ r% r r r �test_dtype_from_inputC s z-TestNanFunctions_MinMax.test_dtype_from_inputc sH xBt | j| j�D ]0\}� � fdd�tD �}|tdd�}t||� qW d S )Nc s g | ] }� |��qS r r )�.0�d)r# r r � <listcomp>R s z>TestNanFunctions_MinMax.test_result_values.<locals>.<listcomp>r )r )r r r �_rdatr r )r r" r$ r% r )r# r �test_result_valuesP s z*TestNanFunctions_MinMax.test_result_valuesc C s� t �t jgd ��dd�}x�| jD ]�}xndD ]f}tjdd��P}t�d� tt � |||d��� � � tt |�dkd � tt |d j t�� W d Q R X q,W tjdd��J}t�d� tt � |t j��� tt |�dkd � tt |d j t�� W d Q R X q"W d S ) N� r )Nr r T)�record�always)r r zno warning raisedr )r �array�nan�reshaper �warnings�catch_warnings� simplefilterr �isnan�all�len� issubclass�category�RuntimeWarning)r r! r r �wr r r � test_allnansV s " z$TestNanFunctions_MinMax.test_allnansc C sn t j�t�}|j�� }xRt jgD ]F}||dd�}|tdd�}t||� t|j|� tt � |�� � � q W d S )Nr )r ) r �ma� fix_invalidr �_maskr �nanminr r �isinf�any)r r! �mskr r% r$ r r r � test_maskedf s z#TestNanFunctions_MinMax.test_maskedc C s$ x | j D ]}t|d�dk� qW d S )Ng )r r )r r r r r � test_scalarp s z#TestNanFunctions_MinMax.test_scalarc C s t �t �d��}xx| jD ]n}||dd�}tt|t j�� t|jdk� ||dd�}tt|t j�� t|jdk� ||�}tt �|�� qW t j|d< �xd| jD �]X}t j dd��R}t � d � ||dd�}tt|t j�� tt � t � |�� � tt|�dk� W d Q R X t j dd���}t � d � ||dd�}tt|t j�� tt � |d ��ont � |d � �ont � |d � � tt|�dkd � tt|d jt�� W d Q R X t j dd��D}t � d � ||�}tt �|�� t|t jk� tt|�dk� W d Q R X q�W d S )Nr r )r )r r r )r r T)r7 r8 )r r )r r )� r zno warning raised)r �matrixr r r � isinstance�shape�isscalarr: r< r= r> rL r? rA rB rC rD )r r! r r% rE r r r � test_matricest s@ $ z%TestNanFunctions_MinMax.test_matricesc C s� t jddgt jdgt jt jggtd�}tt �|�d� tt j|dd�ddg� tjdd��V}t�d � tt t j|d d��ddt jg� t t |�d kd � t t |d j t�� W d Q R X d S ) Ng �?g @g @)r, r )r T)r7 r8 r zno warning raised)r r9 r: �objectr rJ r< r= r> �listr rA rB rC rD )r �arrrE r r r �test_object_array� s & z)TestNanFunctions_MinMax.test_object_arrayN)�__name__� __module__� __qualname__r rJ �nanmaxr �min�maxr r r& r* r0 r5 rF rN rO rU rY r r r r r & s 'r c @ sH e Zd ZejejgZdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z d S )� TestNanFunctions_ArgminArgmaxc C s. t �� }x | jD ]}||� t|t � qW d S )N)r r r r )r r r r r r r � s z+TestNanFunctions_ArgminArgmax.test_mutationc C s� x�t | jtjtjg�D ]�\}}x|tD ]t}t� �d}|�td� ||�}|| }t t� |� � t |||�� � � t t� ||d |� �� � � W d Q R X q$W qW d S )Nz invalid value encountered in) r r r �greater�lessr r �filterrD r r? rL �equal)r r Zfcmp�row�sup�ind�valr r r r5 � s z0TestNanFunctions_ArgminArgmax.test_result_valuesc C sZ t �t jgd ��dd�}x:| jD ]0}x dD ]}tt|||d� q,W tt|t j� q"W d S )Nr6 r )Nr r )r )r r9 r: r; r r � ValueError)r r! r r r r r rF � s z*TestNanFunctions_ArgminArgmax.test_allnansc C sf t �d�}xV| jD ]L}x dD ]}tt|||d� q W x(dD ] }|||d�}t|t �d�� q:W qW d S )N)r r )r N)r )r r )r r( r r ri r )r r! r r r% r r r � test_empty� s z(TestNanFunctions_ArgminArgmax.test_emptyc C s$ x | j D ]}t|d�dk� qW d S )Ng )r r )r r r r r rO � s z)TestNanFunctions_ArgminArgmax.test_scalarc C s� t �t �d��}xx| jD ]n}||dd�}tt|t j�� t|jdk� ||dd�}tt|t j�� t|jdk� ||�}tt �|�� qW d S )Nr r )r )r r r )r r )r rQ r r r rR rS rT )r r! r r% r r r rU � s z+TestNanFunctions_ArgminArgmax.test_matricesN) rZ r[ r\ r � nanargmin� nanargmaxr r r5 rF rj rO rU r r r r r` � s r` c @ s� e Zd Zejejejejejej ej ej fZ e� dddddg�Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestNanFunctions_IntTypes� �' �] �W �. c c s" x | j D ]}| j�|�V qW d S )N)� int_typesr! �astype)r r, r r r �integer_arrays� s z(TestNanFunctions_IntTypes.integer_arraysc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r r^ r! ru r rJ )r r$ r! r r r � test_nanmin� s z%TestNanFunctions_IntTypes.test_nanminc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r r_ r! ru r r] )r r$ r! r r r � test_nanmax� s z%TestNanFunctions_IntTypes.test_nanmaxc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �argminr! ru r rk )r r$ r! r r r �test_nanargmin� s z(TestNanFunctions_IntTypes.test_nanargminc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �argmaxr! ru r rl )r r$ r! r r r �test_nanargmax� s z(TestNanFunctions_IntTypes.test_nanargmaxc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �sumr! ru r �nansum)r r$ r! r r r � test_nansum s z%TestNanFunctions_IntTypes.test_nansumc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �prodr! ru r �nanprod)r r$ r! r r r � test_nanprod s z&TestNanFunctions_IntTypes.test_nanprodc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �cumsumr! ru r � nancumsum)r r$ r! r r r �test_nancumsum s z(TestNanFunctions_IntTypes.test_nancumsumc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �cumprodr! ru r � nancumprod)r r$ r! r r r �test_nancumprod s z)TestNanFunctions_IntTypes.test_nancumprodc C s2 t �| j�}x | �� D ]}tt �|�|� qW d S )N)r �meanr! ru r �nanmean)r r$ r! r r r � test_nanmean s z&TestNanFunctions_IntTypes.test_nanmeanc C sf t �| j�}x | �� D ]}tt �|�|� qW t j|dd�}x$| �� D ]}tt j|dd�|� qFW d S )Nr )�ddof)r �varr! ru r �nanvar)r r$ r! r r r � test_nanvar s z%TestNanFunctions_IntTypes.test_nanvarc C sh t �| j�}x | �� D ]}tt �|�|� qW t j| jdd�}x$| �� D ]}tt j|dd�|� qHW d S )Nr )r� )r �stdr! ru r �nanstd)r r$ r! r r r � test_nanstd$ s z%TestNanFunctions_IntTypes.test_nanstdN)rZ r[ r\ r �int8�int16�int32�int64�uint8�uint16�uint32�uint64rs r9 r! ru rv rw ry r{ r~ r� r� r� r� r� r� r r r r rm � s rm c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� SharedNanFunctionsTestsMixinc C s. t �� }x | jD ]}||� t|t � qW d S )N)r r r r )r r r r r r r / s z*SharedNanFunctionsTestsMixin.test_mutationc C sd t �d�}xTt| j| j�D ]B\}}x8dD ]0}|||dd�}|||dd�}t|j|jk� q(W qW d S )Nr )Nr r T)r r )r r r r r r r )r r! r" r# r r$ r% r r r r& 6 s z*SharedNanFunctionsTestsMixin.test_keepdimsc C sb t �d�}xRt| j| j�D ]@\}}t �d�}||dd�}||d|d�}t||� t||� qW d S )Nr r )r )r r' )r r r r r r( r )r r! r" r# r) r$ r% r r r r* > s z%SharedNanFunctionsTestsMixin.test_outc C s� t �d�}d}x�t| j| j�D ]�\}}x�|D ]�}t� ��}|t jt jhkr\|dkr\|�t j � ||t � |�dd�j j }||t � |�dd�j j }t ||k� ||t � |�d d�j j }||t � |�d d�j j }t ||k� W d Q R X q,W q W d S )Nr r+ �FDGr )r, r ) r r r r r r r� r� rc �ComplexWarningr, r- r ) r r! r. r" r# r/ rf r$ r% r r r �test_dtype_from_dtypeG s z2SharedNanFunctionsTestsMixin.test_dtype_from_dtypec C s� t �d�}d}x�t| j| j�D ]�\}}x�|D ]�}t� ��}|t jt jhkr\|dkr\|�t j � |||dd�j j }|||dd�j j }t ||k� |||d d�j j }|||d d�j j }t ||k� W d Q R X q,W q W d S )Nr r+ r� r )r, r ) r r r r r r r� r� rc r� r, r- r ) r r! r. r" r# r/ rf r$ r% r r r �test_dtype_from_charX s z1SharedNanFunctionsTestsMixin.test_dtype_from_charc C s� d}x�t | j| j�D ]�\}}x||D ]t}tjd|d�}||dd�jj}||dd�jj}t||kd||f � ||d d�jj}||d d�jj}t||k� q"W qW d S )Nr+ r )r, r )r zres %s, tgt %s)r r r r r r, r- r )r r. r" r# r/ r! r$ r% r r r r0 i s z2SharedNanFunctionsTestsMixin.test_dtype_from_inputc sH xBt | j| j�D ]0\}� � fdd�tD �}|tdd�}t||� qW d S )Nc s g | ] }� |��qS r r )r1 r2 )r# r r r3 x s zCSharedNanFunctionsTestsMixin.test_result_values.<locals>.<listcomp>r )r )r r r r4 r r )r r" r$ r% r )r# r r5 v s z/SharedNanFunctionsTestsMixin.test_result_valuesc C s$ x | j D ]}t|d�dk� qW d S )Ng )r r )r r r r r rO | s z(SharedNanFunctionsTestsMixin.test_scalarc C s� t �t �d��}xx| jD ]n}||dd�}tt|t j�� t|jdk� ||dd�}tt|t j�� t|jdk� ||�}tt �|�� qW d S )Nr r )r )r r r )r r )r rQ r r r rR rS rT )r r! r r% r r r rU � s z*SharedNanFunctionsTestsMixin.test_matricesN) rZ r[ r\ r r& r* r� r� r0 r5 rO rU r r r r r� . s r� c @ s4 e Zd ZejejgZejejgZ dd� Z dd� Z dS )�TestNanFunctions_SumProdc C s� t jdd���}t �d� tjtjgd d d�}t|dkd� tt|�dkd� t�tj�}t|dkd� tt|�dkd� tjdgd d d� tt|�dkd � W d Q R X d S ) NT)r7 r8 r )r r zresult is not 0zwarning raisedzunwanted warning raised)r< r= r> r r} r: r rA )r rE r% r r r rF � s z%TestNanFunctions_SumProd.test_allnansc C s� x�t tjtjgddg�D ]f\}}t�d�}|gd }||dd�}t||� g }||dd�}t||� |}||d d�}t||� qW d S )Nr r )r r r )r )r r r} r� r( r )r r � tgt_valuer! r$ r% r r r rj � s z#TestNanFunctions_SumProd.test_emptyN) rZ r[ r\ r r} r� r r| r r rF rj r r r r r� � s r� c @ sT e Zd ZejejgZejejgZ dd� Z dd� Z dd� Z dd� Z d d � Zd d � Zd S )�TestNanFunctions_CumSumProdc C s� x�t | jddg�D ]�\}}t� �~ |tjgd d d�}|t�d� }tt�||�d| � |tj�}|t�d� }tt�||�d| � |dgd d d� W d Q R X qW d S )Nr r r )r z result is not %s * np.ones((3))z result is not %s * np.ones((1)))r r r r r: �onesr � array_equal)r r r� r% r$ r r r rF � s z(TestNanFunctions_CumSumProd.test_allnansc C s� x�t | jddg�D ]p\}}t�d�}|t�d� }||dd�}t||� |}||dd�}t||� t�d�}||d d�}t||� qW d S )Nr r )r r )r )r r r r( r� r )r r r� r! r$ r% r r r rj � s z&TestNanFunctions_CumSumProd.test_emptyc C s� x^t | j| j�D ]L\}}t�d�}x8dD ]0}|||d d�}|||d d�}t|j|jk� q(W qW x~| jD ]t}t�d�}tj� d�}tj ||j |j � dk < ||d d�}t |j d� x*t�d �D ] }|||d�}t |j d� q�W qhW d S ) Nr )Nr r )r r' )r � � � r g �?)r )i� � )r r r r r r r r� �random� RandomStater: �randrS r �arange) r r �gr! r r$ r% r2 �rsr r r r& � s z)TestNanFunctions_CumSumProd.test_keepdimsc C sx t �t �d��}xb| jD ]X}x<t �d�D ].}|||d�}tt|t j�� t|jdk� q(W ||�}t|jdk� qW d S )Nr rP )r )r r )r r6 )r rQ r r r� r rR rS )r r! r r r% r r r rU � s z)TestNanFunctions_CumSumProd.test_matricesc C s^ xXdD ]P}t jt|d�}t jt|d�}t||� t jt|d�}t jt|d�}t||� qW d S )N)������r r N)r ) r r� � _ndat_onesr� r r r� � _ndat_zerosr� )r r r$ r% r r r r5 � s z.TestNanFunctions_CumSumProd.test_result_valuesc C sp t �d�}x`t| j| j�D ]N\}}t �d�}x:dD ]2}|||d�}||||d�}t||� t||� q2W qW d S )Nr )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* � s z$TestNanFunctions_CumSumProd.test_outN)rZ r[ r\ r r� r� r r� r� r rF rj r& rU r5 r* r r r r r� � s r� c @ s\ e Zd ZejejejgZejej ej gZ dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd S )�TestNanFunctions_MeanVarStdc C s@ x:| j D ]0}x*tjtjtjgD ]}tt|td|d� q W qW d S )Nr )r r, )r r �bool_�int_�object_r � TypeErrorr )r r r, r r r �test_dtype_error s z,TestNanFunctions_MeanVarStd.test_dtype_errorc C sT xN| j D ]D}x>tjtjtjgD ]*}tjtjd |d�}tt |td|d� q W qW d S )Nr )r, r )r r' ) r r r� r� r� �emptyr rS r r� )r r r, r' r r r �test_out_dtype_error s z0TestNanFunctions_MeanVarStd.test_out_dtype_errorc sn t jt jg}t jt jg}xPt||�D ]B\}�x8dD ]0� � �fdd�tD �}|td� d�}t||� q2W q$W d S )N)r r c s g | ]}�|� d ��qS ))r� r )r1 r2 )r� r# r r r3 s z9TestNanFunctions_MeanVarStd.test_ddof.<locals>.<listcomp>r )r r� ) r r� r� r� r� r r4 r r )r r r r" r$ r% r )r� r# r � test_ddof s z%TestNanFunctions_MeanVarStd.test_ddofc s� t jt jg}t jt jg}dd� tD �}x�t||�D ]�\}}x�td�D ]�� t� �z}|� t � |� t j � � fdd�|D �}|t d� d�}tt �|�|� t|�r�tt|j�dk� ntt|j�dk� W d Q R X qDW q2W d S )Nc S s g | ] }t |��qS r )rA )r1 r2 r r r r3 s zATestNanFunctions_MeanVarStd.test_ddof_too_big.<locals>.<listcomp>r� c s g | ] }� |k�qS r r )r1 r2 )r� r r r3 & s r )r r� r )r r� r� r� r� r4 r �ranger r7 rD rc r� r r r? rL r rA �log) r r r Zdsizer" r# rf r$ r% r )r� r �test_ddof_too_big s z-TestNanFunctions_MeanVarStd.test_ddof_too_bigc C s� t �t jgd ��dd�}x�| jD ]�}x�dD ]�}tjdd���}t�d� tt � |||d��� � � tt |�dk� tt |d j t�� tt � |t j��� tt |�d k� tt |d j t�� W d Q R X q,W q"W d S ) Nr6 r )Nr r T)r7 r8 )r r r rP )r r9 r: r; r r< r= r> r r? r@ rA rB rC rD )r r! r r rE r r r rF . s z(TestNanFunctions_MeanVarStd.test_allnansc C s� t �d�}x�| jD ]�}xldD ]d}tjdd��N}t�d� tt �|||d���� � tt |�dk� tt |d j t �� W d Q R X q W xVd D ]N}tjdd��8}t�d� t |||d�t �g �� tt |�dk� W d Q R X q�W qW d S ) N)r r )r NT)r7 r8 )r r r )r )r r( r r< r= r> r r? r@ rA rB rC rD r )r r! r r rE r r r rj < s " z&TestNanFunctions_MeanVarStd.test_emptyN)rZ r[ r\ r r� r� r� r r� r� r� r r� r� r� r� rF rj r r r r r� s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�TestNanFunctions_Medianc C s t �� }t�|� t|t � d S )N)r r r � nanmedianr )r r r r r r N s z%TestNanFunctions_Median.test_mutationc C s\ t �d�}x@dD ]8}t j||d dd�}t j||d dd�}t|j|jk� qW t �d�}t j�d�t �|j �d d �d f }|� t j �}t j |t |�< t� ��}|�t� t j|d dd�}t|j d � t j|d dd�}t|j d � t j|d dd�}t|j d � t j|ddd�}t|j d� t j|ddd�}t|j d � t j|ddd�}t|j d� W d Q R X d S )Nr )Nr r F)r r' �overwrite_input)r r� r� r� )r� �� T)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 rP r )r r r )r r r� r )r r �medianr� r r r� r� r9 rS rt �intpr: �tupler rc rD r )r r! r r$ r% r2 rE rf r r r r& T s. $ z%TestNanFunctions_Median.test_keepdimsc C s� t j�dd�}t j|ddgt jdd�}t �d�}t j|dd�}t j|d|d�}t||� t||� t �d�}t j|d d�}t j|d |d�}t||� t||� t j|d|d�}t||� t||� d S ) Nr r rP r )r )r r' r )r r ) r r� r� �insertr: r( r� r� r )r r! �nan_matr) r$ r% r r r r* o s z TestNanFunctions_Median.test_outc C s� x�dD ]�}t j�d|�}t jjd|j|jd d�}t j|�� |< d|d d �df< g }x4|D ],}t �t �|� |�}|� t j |dd�� qZW t t j |d d �|� qW d S ) N)r� � �3 r� i� r� r r� )�sizeg �?T)r� r� )r ) r r� �randn�randintr� r: �ravel�compressr? �appendr� r r� )r �sr2 rE r$ �xZnonanr r r �test_small_large� s z(TestNanFunctions_Median.test_small_largec C s* dd� t D �}tjtdd�}t||� d S )Nc S s g | ]}t �|��qS r )r r� )r1 r2 r r r r3 � s z>TestNanFunctions_Median.test_result_values.<locals>.<listcomp>r )r )r4 r r� r r )r r$ r% r r r r5 � s z*TestNanFunctions_Median.test_result_valuesc C s� t �t jgd ��dd�}x�dD ]�}t� ��}|�t� tt �t j ||d��� � � |d krntt |j �dk� ntt |j �dk� tt �t � t j��� |d kr�tt |j �dk� ntt |j �dk� W d Q R X q W d S )Nr6 r )Nr r )r r rP r� ) r r9 r: r; r r7 rD r r? r� r@ rA r� )r r! r rf r r r rF � s z$TestNanFunctions_Median.test_allnansc C s� t �d�}xndD ]f}tjdd��P}t�d� tt �t j||d���� � tt |�dk� tt |d j t �� W d Q R X qW xXd D ]P}tjdd��:}t�d� t t j||d�t �g �� tt |�dk� W d Q R X q�W d S ) N)r r )r NT)r7 r8 )r r r )r )r r( r< r= r> r r? r� r@ rA rB rC rD r )r r! r rE r r r rj � s " z"TestNanFunctions_Median.test_emptyc C s t t�d�dk� d S )Ng )r r r� )r r r r rO � s z#TestNanFunctions_Median.test_scalarc C sp t �d�}tt jt j|dd� tt jt j|dd� tt jt j|dd� tt jt j|dd� ttt j|dd� d S )N)r r� r� r� ���)r )r r� r� )r r� )r r )r r� r � AxisErrorr� ri )r r2 r r r �test_extended_axis_invalid� s z2TestNanFunctions_Median.test_extended_axis_invalidc C s� t � ���}|�t� �x�tjtj gD �]�}t�|tjgtjtjgg�}ttj|dd�|tjg� ttj|dd�|tjg� tt�|�|� t�tjtj|gtjtj|gg�}tt�|�|� ttj|dd�tjtj|g� ttj|dd�|� t�||g||gg�}ttj|dd�|� tj|d| dgdtjtjdgdtjtj|ggtj d �}|dk�r�ttj|dd�d d | d g� tt�|�d � n.ttj|dd�dd | dg� tt�|�d� ttj|dd�dd|g� �x t dd�D ]�}x�t dd�D ]�}t�tjg| |g| gd �}tt�|�|� ttj|dd�|� ttj|dd�tjg| |g| � t�tjg| | g| gd �}tt�|�| � ttj|dd�| � ttj|dd�tjg| | g| � �q�W �q�W q&W W d Q R X d S )Nr )r r r� i����i����r� r� )r, g @g @g @g @g $�g "�g �r� g �� rP ) r rc rD r �infr9 r: r r� �float32r� )r rf r� �a�i�jr r r �test_float_special� sH "$z*TestNanFunctions_Median.test_float_specialN) rZ r[ r\ r r& r* r� r5 rF rj rO r� r� r r r r r� L s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestNanFunctions_Percentilec C s" t �� }t�|d� t|t � d S )N� )r r r � nanpercentiler )r r r r r r � s z)TestNanFunctions_Percentile.test_mutationc C sl t �d�}xDdD ]<}t j|d|d dd�}t j|d|d dd�}t|j|jk� qW t �d�}t j�d�t �|j �d d �d f }|� t j �}t j |t |�< t� ��}|�t� t j|dd d d �}t|j d � t j|dd d d �}t|j d � t j|ddd d �}t|j d� t j|ddd d �}t|j d� t j|ddd d �}t|j d � t j|ddd d �}t|j d� W d Q R X d S )Nr )Nr r �F F)r r' r� )r r� r� r� )r� r� �Z T)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 rP r )r r r )r r r� r )r r � percentiler� r r r� r� r9 rS rt r� r: r� r rc rD r )r r! r r$ r% r2 rE rf r r r r& � s2 $ z)TestNanFunctions_Percentile.test_keepdimsc C s� t j�dd�}t j|ddgt jdd�}t �d�}t j|ddd�}t j|dd|d�}t||� t||� t �d�}t j|dd d�}t j|dd |d�}t||� t||� t j|dd |d�}t||� t||� d S ) Nr r rP 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* s z$TestNanFunctions_Percentile.test_outc C sZ dd� t D �}tjtddd�}t||� t�dd� t D ��}tjtddd�}t||� d S )Nc S s g | ]}t �|d ��qS )� )r r� )r1 r2 r r r r3 ( s zBTestNanFunctions_Percentile.test_result_values.<locals>.<listcomp>r� r )r c S s g | ]}t �|d ��qS ))r� �b )r r� )r1 r2 r r r r3 , s )r� r� )r4 r r� r r � transpose)r r$ r% r r r r5 ' s z.TestNanFunctions_Percentile.test_result_valuesc C s t �t jgd ��dd�}x�dD ]�}tjdd���}t�d� tt �t j |d|d��� � � |d krttt |�d k� ntt |�dk� tt |d j t�� tt �t � t jd��� |d kr�tt |�d k� ntt |�d k� tt |d j t�� W d Q R X q W d S ) Nr6 r )Nr r T)r7 r8 �< )r r r rP r� )r r9 r: r; r< r= r> r r? r� r@ rA rB rC rD )r r! r rE r r r rF 0 s z(TestNanFunctions_Percentile.test_allnansc C s� t �d�}xpdD ]h}tjdd��R}t�d� tt �t j|d|d���� � tt |�dk� tt |d j t �� W d Q R X qW xZd D ]R}tjdd��<}t�d� t t j|d|d�t �g �� tt |�d k� W d Q R X q�W d S ) N)r r )r NT)r7 r8 �( )r r r )r )r r( r< r= r> r r? r� r@ rA rB rC rD r )r r! r rE r r r rj C s " z&TestNanFunctions_Percentile.test_emptyc C sH t t�dd�d� t�d�}tj|ddd�}t |d� tt�|�� d S )Ng �d � �2 r )r g @)r r r� r� r rT )r r� �rr r r rO Q s z'TestNanFunctions_Percentile.test_scalarc C sz t �d�}tt jt j|ddd� tt jt j|ddd� tt jt j|ddd� tt jt j|ddd� ttt j|ddd� d S ) N)r r� r� r� r� r� )�qr )r r� r� )r r� )r r )r r� r r� r� ri )r r2 r r r r� X s z6TestNanFunctions_Percentile.test_extended_axis_invalidc C s$ ddg}t �d�}t j| }t �d�}d|d d �ddd�d d �f< |d d �d d �dd �f d9 < x�d D ]�}x�d D ]�}t� �r}|�td � t j||||d �}t j||||d �} t| j |j � t j||||d �}t j||||d �} t| |� W d Q R X qtW qjW t �d �} tt j| |dd�j d� d S )Nr� r� )r� r )r r� r� r rP r� r )Nr r )FTzAll-NaN slice encountered)r r )r r� r� r� )r rP )r )rP r r� ) r r� r: r rc rD r� r� r rS ) r �percr! r� Z large_matr �keepdimrf rh Znan_valZmegamatr r r �test_multiple_percentiles` s* " z5TestNanFunctions_Percentile.test_multiple_percentilesN) rZ r[ r\ r r& r* r5 rF rj rO r� r� r r r r r� � s r� �__main__)"� __future__r r r r< �numpyr Z numpy.testingr r r r r r r r r r9 r: r r4 r� r� r r` rm rV r� r� r� r� r� r� rZ r r r r �<module> sB , ;Ka"SH %