Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/numpy/lib/tests/__pycache__/test_nanfunctions.cpython-37.pyc
Назад
B �5�g�� � @ s� d dl Z d dlZd dlZd dlmZmZ d dlmZm Z m Z m Z m Z m Z mZ 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� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG d d � d e�ZG d d � d e�ZG d d!� d!e�Zd"Z e ej d# �Z e d$d%� e D �7 Z G d&d'� d'�Z G d(d)� d)�Z!G d*d+� d+�Z"ej#�$d,e�ejd-ejej%g�e�d.d/d.d/g�fejd0d1d2d3gej&d4�d/fe�d.d/d.d/g�d/fejejd-gejej%ggej'd4�e�d.d/gd.d/gg�fg�d5d6� �Z(d7d8� Z)dS )9� N)� _nan_mask� _replace_nan)�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 �W/home/digitalm-up/venv/lib/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 �maZ fix_invalidr Z_maskr �nanminr r �isinf�any)r r Zmskr 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 G dd� dt j�}t �d��|�}xt| jD ]j}||dd�}tt||�� t|jdk� ||dd�}tt||�� t|jdk� ||�}t|jdk� q*W t j|d< �x`| jD �]T}t j d d ��P}t � d � ||dd�}tt||�� tt � t � |�� � tt|�dk� W d Q R X t j d d ���}t � d � ||dd�}tt||�� tt � |d ��oxt � |d � �oxt � |d � � tt|�dkd � tt|d jt�� W d Q R X t j d d ��D}t � d � ||�}t|jdk� t|t jk� tt|�dk� W d Q R X q�W d S )Nc @ s e Zd ZdS )z8TestNanFunctions_MinMax.test_subclass.<locals>.MyNDArrayN)�__name__� __module__� __qualname__r r r r � MyNDArrayu s rM r r )r )r r r T)r4 r5 � zno warning raised)r �ndarrayr �viewr r � isinstance�shaper7 r9 r: r; rG r< r> r? r@ rA )r rM �miner r" rB r r r � test_subclasst sB $ z%TestNanFunctions_MinMax.test_subclassc 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)r4 r5 r zno warning raised)r r6 r7 �objectr rE r9 r: r; �listr r> r? r@ rA )r �arrrB r r r �test_object_array� s & z)TestNanFunctions_MinMax.test_object_arrayN)rJ rK rL r rE �nanmaxr �min�maxr r r# r' r- r2 rC rH rI rT rX 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 Zgreater�lessr r �filterrA r r< rG �equal)r r Zfcmp�row�sup�ind�valr r r r2 � 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 )Nr3 r )Nr r )r )r r6 r7 r8 r r � ValueError)r r r r r r r rC � 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 rd 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 rI � s z)TestNanFunctions_ArgminArgmax.test_scalarc C s� G dd� dt j�}t �d��|�}xt| jD ]j}||dd�}tt||�� t|jdk� ||dd�}tt||�� t|jdk� ||�}t|jdk� q*W d S ) Nc @ s e Zd ZdS )z>TestNanFunctions_ArgminArgmax.test_subclass.<locals>.MyNDArrayN)rJ rK rL r r r r rM � s rM r r )r )r r r )r rO r rP r r rQ rR )r rM rS r r" r r r rT � s z+TestNanFunctions_ArgminArgmax.test_subclassN) rJ rK rL r � nanargmin� nanargmaxr r r2 rC re rI rT 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 rZ r rp r rE )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 rp r rY )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 Zargminr rp r rf )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 Zargmaxr rp r rg )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 rp 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 rp 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 rp 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 rp 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 rp 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 rp 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 rp r �nanstd)r r! r r r r � test_nanstd+ s z%TestNanFunctions_IntTypes.test_nanstdN)rJ rK rL r Zint8Zint16�int32�int64Zuint8Zuint16Zuint32Zuint64rn r6 r rp rq rr rs rt rw rz r} r� r� r� r� r r r r rh � s rh 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 6 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# = 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' E 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� r^ �ComplexWarningr) r* r ) r r r+ r r r, ra r! r" r r r �test_dtype_from_dtypeN 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� r^ r� r) r* r ) r r r+ r r r, ra r! r" r r r �test_dtype_from_char_ 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 r- p 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 )r. r/ )r r r r0 s zCSharedNanFunctionsTestsMixin.test_result_values.<locals>.<listcomp>r )r )r r r r1 r r )r r r! r" r )r r r2 } 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 rI � s z(SharedNanFunctionsTestsMixin.test_scalarc C s� G dd� dt j�}t �d�}|�|�}x�| jD ]�}||dd�j}||dd�}tt||�� t|j|k� ||dd�j}||dd�}tt||�� t|j|k� ||�j}||�}tt||�� t|j|k� q.W d S )Nc @ s e Zd ZdS )z=SharedNanFunctionsTestsMixin.test_subclass.<locals>.MyNDArrayN)rJ rK rL r r r r rM � s rM r r )r r )r rO r rP r rR r rQ )r rM r6 rS r Zexpected_shaper" r r r rT � s z*SharedNanFunctionsTestsMixin.test_subclassN) rJ rK rL r r# r' r� r� r- r2 rI rT r r r r r� 5 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)r4 r5 r )r r zresult is not 0zwarning raisedzunwanted warning raised)r9 r: r; r rv r7 r r> )r rB r" r r r rC � 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 rv ry r% r )r r � tgt_valuer r! r" r r r re � s z#TestNanFunctions_SumProd.test_emptyN) rJ rK rL r rv ry r ru rx r rC re r r r r r� � s r� c @ sL e Zd ZejejgZejejgZ dd� Z dd� Z dd� Z dd� Z d 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 r7 �onesr Z array_equal)r r r� r" r! r r r rC � 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 re � 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� �randomZ RandomStater7 �randrR r �arange) r r �gr r r! r" r/ �rsr r r r# � s z)TestNanFunctions_CumSumProd.test_keepdimsc 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 r2 � 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)rJ rK rL r r| r r r{ r~ r rC re r# r2 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 rR 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 )r. r/ )r� r r r r0 s z9TestNanFunctions_MeanVarStd.test_ddof.<locals>.<listcomp>r )r r� ) r r� r� r� r� r r1 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 )r> )r. r/ r r r r0 # s zATestNanFunctions_MeanVarStd.test_ddof_too_big.<locals>.<listcomp>r� c s g | ] }� |k�qS r r )r. r/ )r� r r r0 ) s r )r r� r )r r� r� r� r� r1 r �ranger r4 rA r^ r� r r r< rG r r> �log) r r r Zdsizer r ra 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 ) Nr3 r )Nr r T)r4 r5 )r r r rN )r r6 r7 r8 r r9 r: r; r r< r= r> r? r@ rA )r r r r rB r r r rC 1 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)r4 r5 )r r r )r )r r% r r9 r: r; r r< r= r> r? r@ rA r )r r r r rB r r r re ? s " z&TestNanFunctions_MeanVarStd.test_emptyN)rJ rK rL r r� r� r� r r� r� r� r r� r� r� r� rC re r r r r r� s r� ) �Y�M�W�D�h�m�s�ms�us�nsZps�fs�asZAllFloatc C s g | ]}d |� d��qS )zm8[�]r )r. �unitr r r r0 U s r0 c @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d dg�ej � de �dd� ��Z dd� Z dd� Zdd� Zdd� Zd S )�TestNanFunctions_Medianc C s t �� }t�|� t|t � d S )N)r r r � nanmedianr )r r r r r r Z 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 rN r )r r r )r r r� r )r r �medianr� r r r� r� r6 rR ro �intpr7 �tupler r^ rA r )r r r r! r" r/ rB ra r r r r# ` 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 rN r )r )r r$ r )r r ) r r� r� �insertr7 r% r� r� r )r r �nan_matr& r! r" r r r r' { 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� Zrandn�randintr� r7 Zravel�compressr< �appendr� r r� )r r� r/ rB 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� )r. r/ r r r r0 � s z>TestNanFunctions_Median.test_result_values.<locals>.<listcomp>r )r )r1 r r� r r )r r! r" r r r r2 � s z*TestNanFunctions_Median.test_result_valuesr Nr r r) c C s� t �dt j��|�}t� ��}|�t� t j||d�}|j|jksDt �t � |�� � sVt �|d krrt t |j�dk� nt t |j�dk� t �t j��|�d }t �|�}|j|jks�t �t � |�s�t �|d kr�t t |j�dk� nt t |j�dk� W d Q R X d S )N)r r )r r r r rN r� )r �fullr7 ro r r4 rA r� r) �AssertionErrorr< r= r r> r� r6 )r r) r r ra �outputZscalarZ output_scalarr r r rC � 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)r4 r5 )r r r )r )r r% r9 r: r; r r< r� r= r> r? r@ rA r )r r r rB r r r re � s " z"TestNanFunctions_Median.test_emptyc C s t t�d�dk� d S )Ng )r r r� )r r r r rI � 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� rd )r r/ 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 �� rN ) r r^ rA r �infr6 r7 r r� �float32r� )r ra r� �a�i�jr r r �test_float_special� sH "$z*TestNanFunctions_Median.test_float_special)rJ rK rL r r# r' r� r2 �pytest�mark� parametrize� _TYPE_CODESrC re rI r� r� r r r r r� X 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 rN r )r r r )r r r� r )r r � percentiler� r r r� r� r6 rR ro r� r7 r� r r^ rA r )r r r r! r" r/ rB ra 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 rN r )r �* )r r$ r )r r ) r r� r� r� r7 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� )r. r/ r r r r0 = s zBTestNanFunctions_Percentile.test_result_values.<locals>.<listcomp>r� r )r c S s g | ]}t �|d ��qS ))r� �b )r r� )r. r/ r r r r0 A s )r� r� )r1 r r� r r Z transpose)r r! r" r r r r2 < 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 ) Nr3 r )Nr r T)r4 r5 �< )r r r rN r� )r r6 r7 r8 r9 r: r; r r< r� r= r> r? r@ rA )r r r rB r r r rC E 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)r4 r5 �( )r r r )r )r r% r9 r: r; r r< r� r= r> r? r@ rA r )r r r rB r r r re X 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 Zisscalar)r r� �rr r r rI f 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� rd )r r/ r r r r� m 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 rN r� r )Nr r )FTzAll-NaN slice encountered)r r )r r� r� r� )r rN )r )rN r r� ) r r� r7 r r^ rA r� r� r rR ) r �percr r� Z large_matr Zkeepdimra rc Znan_valZmegamatr r r �test_multiple_percentilesu s* " z5TestNanFunctions_Percentile.test_multiple_percentilesN) rJ rK rL r r# r' r2 rC re rI r� r� r r r r r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestNanFunctions_Quantilec C s� t �d��ddd��t�}t j|d d< tt j|dd�t j|d d�� tt j|ddd �t j|d dd �� tt j|ddd �t j|d dd �� tt j|dgdd �t j|d gdd �� tt j|d dd gdd �t j|d d dgdd �� d S )N� rN r r� r r g �?)r� r� )r� r g �?g �?� �K ) r r� r8 ro �floatr7 r � nanquantiler� )r �arr r r �test_regression� s z)TestNanFunctions_Quantile.test_regressionc C sH t �d�d }tt �|d�d� tt �|d�d� tt �|d�d� d S )N� g �?r g r g @g �?)r r� r r� )r r� r r r � test_basic� s z$TestNanFunctions_Quantile.test_basicc C sp t �dddddg�}|�� }t jt �d�|dd� t||� |�� }|�� }t jt �d�|dd� t||� d S ) Nr g �?g �?g �?g �?g Y@Zmidpoint)� interpolation)r r6 r r� r� r �tolist)r �p0�pr r r �test_no_p_overwrite� s z-TestNanFunctions_Quantile.test_no_p_overwriteN)rJ rK rL r� r� r� r r r r r� � s r� z arr, expectedg @FTr r� r� r3 )r) c C sT xNd t j| jt jd�gD ]4}t| |d�}t||� t|�t jk r|dkst�qW d S )N)r) )r$ T) r r� rR r� r r r* rO r� )rW �expectedr$ �actualr r r �test__nan_mask� s r c C s xPt jt jt jgD ]<} t jddg| d�}t|d�\}}|dksBt�||kst�qW x�t jt jgD ]�} t jddg| d�}t|d�\}}|dk� � s�t�||k s�t�t ||� t jddt j g| d�}t|d�\}}t |t �dddg�� ||k s�t�t |t �dddg�� t � |d �s`t�q`W dS ) z] Test that _replace_nan returns the original array if there are no NaNs, not a copy. r r )r) NrN FTr� ) r r� r� r� r6 r r� r� Zfloat64r= r r7 r< )r) rW �result�maskZarr_nanZ result_nanZmask_nanr r r �test__replace_nan� s" r )*r9 r� �numpyr Znumpy.lib.nanfunctionsr r Z numpy.testingr r r r r r r r6 r7 r r1 r� r� r r\ rh r� r� r� r� Z _TIME_UNITSrV � typecodesr� r� r� r� r� r� r� r� Z complex64r r r r r r �<module> s` $ >Ki"GI . #
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка