3
���hד � @ s� d dl mZ d dlZd dlZd dlZd dlZd dljj Z
d dl m Z d dl
Zd dl
mZmZ d dljZd dlmZ d dljjZejZeed�Z ej ddgd �d
d � �Z G d d
� d
�Z G dd� d�Z G dd� d�Z!G dd� d�Z"G dd� d�Z#G dd� d�Z$dd� Z%ej&j'dej(dfej)dfej*d fej+d fej,d fej-d fej.d fej/d fej0d fej1d fg
�d d � �Z2ej&j'd!ej3ej4ej)ej+ej-ej5ej6ej7ej8ej9ej1ej0ej:ej;ej<g�d"d#� �Z=dS )$� )�partialN)�is_integer_dtype)�Series�isna)�
DatetimeArrayZ
complex128TF)�paramsc C s | j S )z5
Fixture to pass skipna to nanops functions.
)�param)�request� r
�:/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_nanops.py�skipna s r c @ sV e Zd Zdd� Zdd� ZdSdd�ZdTd d
�ZdUd d �ZdVd
d�ZdWdd�Z e
j j de
jejfe
jejfg�dd� �Zdd� Zdd� Zdd� Ze
j j dejejejejeje edd�g�d d � �Z d d � Z e
j j d e d!��d"d#� �Z e
j j d e d!��d$d%� �Z!e"j#e
j j d e d!��d&d'� ��Z$e
j j de
j%ej&fe
j'ej(fg�d(d)� �Z)dXd*d+�Z*d,d-� Z+d.d/� Z,dYd0d1�Z-e"j#d2d3� �Z.e"j#d4d5� �Z/d6d7� Z0d8d9� Z1d:d;� Z2d<d=� Z3d>d?� Z4e"j#d@dA� �Z5e"j#dBdC� �Z6e"j#dDdE� �Z7dFdG� Z8dHdI� Z9e
j j dJe:j;e
j<fe:j=e
j>fe:j?e
j@fe:jAe
jBfe:jCe
jDfe:jEe
jFfg�dKdL� �ZGdMdN� ZHdOdP� ZIdQdR� ZJdS )Z�TestnanopsDataFramec
C s� t jjd� dt_d}t jj|� | _t jj|� | _| j| jd | _t jj dd|�| _
t jj dd|�dk| _ t j | j�j
d �| _t j | j�j
d
�| _t jj dd |�j
d �| _t jj dd |�j
d
�| _t jt j|�| _t j| j| jg�| _t j| j| jg�| _t j| j| jg�| _t j| j| jg�| _| jt j | _t j| j| jg�| _ t j| j| jg�| _ t j| j| j| jg�| _ t j| j| j| jg�| _ t j| jj
d�| j
j
d�| j j
d�| jj
d�| jj
d�| jj
d�| jj
d�| jj
d�g�| _ t j!dd��L | j| jd | _"t j| j| j"g�| _#| jd | _$t j| j| j$g�| _%W d Q R X | j| _&| j| _'| j| _(| j| _)| j| _*| j| _+| jd d �df | _,| jd d �df | _-| jd d �df | _.| jd d �df | _/| jd d �df | _0| jd d �df | _1d S )Ni�+ F� � y �?�
r � �S�Ui N zM8[ns]zm8[ns]�O�ignore)�invalid)r r i����)2�np�random�seed�nanops�_USE_BOTTLENECKZrandn� arr_float�
arr_float1� arr_complex�randint�arr_int�arr_bool�abs�astype�arr_str�arr_utf�arr_date�
arr_tdeltaZtile�nan�arr_nan�vstack�
arr_float_nan�arr_float1_nan�arr_nan_float1� arr_nan_nan�inf�arr_inf�
arr_float_inf� arr_nan_inf�arr_float_nan_inf�arr_nan_nan_inf�arr_obj�errstate� arr_nan_nanj�arr_complex_nan� arr_nan_infj�arr_complex_nan_infj� arr_float_2d�
arr_float1_2d�
arr_nan_2d�arr_float_nan_2d�arr_float1_nan_2d�arr_nan_float1_2d� arr_float_1d�
arr_float1_1d�
arr_nan_1d�arr_float_nan_1d�arr_float1_nan_1d�arr_nan_float1_1d)�self�methodZ arr_shaper
r
r � setup_method s`
z TestnanopsDataFrame.setup_methodc C s
t t_d S )N)�use_bnr r )rG rH r
r
r �teardown_method_ s z#TestnanopsDataFrame.teardown_methodTc C s: t |d|�}|dkrLt|d�rL|jrL|j|jkrLtj||jd gdd�d }ytj|||d� W n� tk
�r4 t|d�r�|j dkr�� t|d� s�|j j
d kr�� |j j
d kr�|j j
d kr�|j |j �}q�t r�d
nd }|j |�}|j |�}n|j j
d kr�� tjtj
|�tj
|�|d� tjtj|�tj|�|d� Y nX d S )
NZasm8r �shape)�axis)� check_dtype�dtypezm8[ns]�cr �c16�f8)rP r )�getattr�hasattr�ndimrL r �split�tm�assert_almost_equal�AssertionErrorrO �kindr# �has_c16�real�imag)rG �targ�resrM rN Z
cast_dtyper
r
r �
check_resultsb s.
z!TestnanopsDataFrame.check_resultsNc K sh �xt t|j��d g D ]�} |r$|n|}
|rR|rRt|
�j� rR||
fd| i|��} n||
fd| i|��} ||f| |d�|��} | j| | | |d� |r�||fd| i|��} | j| | | |d� | d kr�||fd|i|��} | j| | | |d� |o�| d kr||f|�} | j| | | |d� qW |jdk�r$d S tj|dd d�}
tj|dd
d�}| j|||
|f|||d�|�� d S ) NrM )rM r )rN r � r )rM )r rN �empty_targfunc���rc ) �list�rangerU r �allr` r �take�check_fun_data)rG �testfunc�targfunc� testarval� targarvalr rN rb �kwargsrM Z
targartempvalr^ r_ Z
testarval2Z
targarval2r
r
r rh � s: z"TestnanopsDataFrame.check_fun_datac
K sb |}|j d�r,t| |d d� �r,|d d� }t| |�}t| |�} | j|||| f||d�|�� d S )NZ_nan� )r rb ���ro )�endswithrT rS rh )
rG ri rj Ztestarr rb rm Ztargarrk rl r
r
r � check_fun� s
z TestnanopsDataFrame.check_func
K s� | j ||d|f| � | j ||d|f| � | j ||d|f| � | j ||d|f| � | jjd�| jjd�| jjd�g}
|r�| j ||d|f| � |r�| j ||d|f| � | j ||d|f| � |r�| j ||d |f| � |
| jjd�g7 }
|�r|| j� | j ||d
|f| � |
| jjd�g7 }
|�rhy|| j� W n tk
�r@ Y n(X | j ||d |f| � |
| jjd�g7 }
|�r�t j
|
�| _ |d k�r�t | j
||d
�}| j ||d|f| � d S )Nr r+ r r! r r) r r8 r7 r& r' �convert)�func�
allow_complexr5 )rq r r# r r! r r& r' � TypeErrorr r* r5 r � _badobj_wrap) rG ri rj r rt �
allow_all_nan�
allow_date� allow_tdelta� allow_objrm �objsr
r
r �
check_funs� s@
z TestnanopsDataFrame.check_funsc K s2 |j jdkr&|r |jd�}n
|jd�}||f|�S )Nr rQ rR )rO rZ r# )rG �valuers rt rm r
r
r rv s
z TestnanopsDataFrame._badobj_wrapz nan_op,np_opc C s | j |||ddd� d S )NF)rw rx )r| )rG �nan_op�np_opr r
r
r �test_nan_funcs s z"TestnanopsDataFrame.test_nan_funcsc C s | j tjtj|ddtjd� d S )NF)rx rN rb )r| r �nansumr �sum)rG r r
r
r � test_nansum s z TestnanopsDataFrame.test_nansumc C s | j tjtj|ddd� d S )NF)rz rx )r| r �nanmeanr �mean)rG r r
r
r � test_nanmean s z TestnanopsDataFrame.test_nanmeanc C s` xZd D ]R}t |td�tjd�}|j� }|jj� }||ks<t�||ksHt�|jtjkst�qW d S )
Nr �7 � �|�H%[<i� )�indexrO � l ��� )r� r� r� ) r re r �int64r� �valuesrY rO �float64)rG �a�s�resultZ np_resultr
r
r �test_nanmean_overflow# s
z)TestnanopsDataFrame.test_nanmean_overflowrO Zfloat128c C s� |d kr d S t td�|d�}dddddg}dd g}xL|| D ]@}t||�� }t|�rn||krn|jtjks|t�q<|j|ks<t�q<W d S )
Nr )rO r� �std�var�skew�kurt�min�max)r re rS r rO r r� rY )rG rO r� Zgroup_aZgroup_brH r� r
r
r �test_returned_dtype0 s z'TestnanopsDataFrame.test_returned_dtypec C sB t jdd��, t jdt� | jtjtj|dddd� W d Q R X d S )NT)�recordr Frr )rt rx rz ) �warnings�catch_warnings� simplefilter�RuntimeWarningr| r � nanmedianr �median)rG r r
r
r �test_nanmedianJ s z"TestnanopsDataFrame.test_nanmedian�ddof� c C s | j tjtj|ddd|d� d S )NFrr )rt rx rz r� )r| r �nanvarr r� )rG r� r r
r
r � test_nanvarV s z TestnanopsDataFrame.test_nanvarc C s | j tjtj|ddd|d� d S )NFrr )rt rx rz r� )r| r �nanstdr r� )rG r� r r
r
r � test_nanstdb s z TestnanopsDataFrame.test_nanstdc C sD ddl m} tjdd��" | jtj||dddd|d� W d Q R X d S )Nr )�semr )r Frr )rt rx ry rz r� )� scipy.statsr� r r6 r| r �nansem)rG r� r r� r
r
r � test_nansemn s z TestnanopsDataFrame.test_nansemc
C s: t jdd��$ t jdt� | j|||dd� W d Q R X d S )NT)r� r F)rz )r� r� r� r� r| )rG r~ r r r
r
r �test_nanops_with_warnings s z-TestnanopsDataFrame.test_nanops_with_warningsc C sX |||�}t j||�}t|�}|jr.d||< n&t|d�r@|j� sPt|d� rT|rTd}|S )Nra rf rc rc )r r� r rU rT rf )rG r} rM rs r_ ZnansZnullnanr
r
r �_argminmax_wrap� s
z#TestnanopsDataFrame._argminmax_wrapc
C sL t jdd��6 t jdt� t| jtjd�}| jt j
||dd� W d Q R X d S )NT)r� r )rs F)rz ) r� r� r� r� r r� r Zargmaxr| r � nanargmax)rG r rs r
r
r �test_nanargmax� s z"TestnanopsDataFrame.test_nanargmaxc
C sL t jdd��6 t jdt� t| jtjd�}| jt j
||dd� W d Q R X d S )NT)r� r )rs F)rz ) r� r� r� r� r r� r Zargminr| r � nanargmin)rG r rs r
r
r �test_nanargmin� s z"TestnanopsDataFrame.test_nanargminc C st t |jjtj�s|jd�}|||dd�}t |tj�rXd|tj||d�tj||d�k< |S tj|�tj|�krpdS |S )NrR F)rM Zbiasr )rM g ) �
isinstancerO �typer Zfloatingr# Zndarrayr� r� )rG r� rM rs r� r
r
r �_skew_kurt_wrap� s
z#TestnanopsDataFrame._skew_kurt_wrapc C sN ddl m} t| j|d�}tjdd�� | jtj||dddd� W d Q R X d S )Nr )r� )rs r )r F)rt rx ry ) r� r� r r� r r6 r| r �nanskew)rG r r� rs r
r
r � test_nanskew� s z TestnanopsDataFrame.test_nanskewc C sZ ddl m} t|dd�}t| j|d�}tjdd�� | jtj||dddd � W d Q R X d S )
Nr )�kurtosisT)Zfisher)rs r )r F)rt rx ry ) r� r� r r� r r6 r| r �nankurt)rG r r� Zfunc1rs r
r
r � test_nankurt� s z TestnanopsDataFrame.test_nankurtc C s | j tjtj|ddtjd� d S )NF)rx ry rb )r| r �nanprodr �prod)rG r r
r
r � test_nanprod� s z TestnanopsDataFrame.test_nanprodc K s~ || j | jf|�}|| j | jfdt| j �d i|��}tj||� tj||� || j| jf|�}|| j| jfdt| j �d i|��}tj||� tj||� tj} || j | jf|�}
|| j | j f|�} || j | j f|�} || j| j
f|�}
|| j| j
fdt| j �d i|��}|| j | jfdt| j �d i|��}tj| |
� tj| | � tj| | � tj| |
� tj| |� tj| |� d S )N� min_periodsra ) r; r< �lenrW rX r> r? r r( r= r@ )rG �checkfun�targ0�targ1rm �res00�res01�res10�res11�targ2�res20�res21�res22�res23�res24�res25r
r
r �check_nancorr_nancov_2d� sJ z+TestnanopsDataFrame.check_nancorr_nancov_2dc K s~ || j | jf|�}|| j | jfdt| j �d i|��}tj||� tj||� || j| jf|�}|| j| jfdt| j �d i|��}tj||� tj||� tj} || j | jf|�}
|| j | j f|�} || j | j f|�} || j| j
f|�}
|| j| j
fdt| j �d i|��}|| j | jfdt| j �d i|��}tj| |
� tj| | � tj| | � tj| |
� tj| |� tj| |� d S )Nr� ra ) rA rB r� rW rX rD rE r r( rC rF )rG r� r� r� rm r� r� r� r� r� r� r� r� r� r� r� r
r
r �check_nancorr_nancov_1d sJ z+TestnanopsDataFrame.check_nancorr_nancov_1dc C s� t j| j| j�d }t j| jj| jj�d }| jtj||� t j| j| j �d }t j| jj| j j�d }| j
tj||dd� d S ) Nr ra �pearson)rH )r ra )r ra )r ra )r ra ) r �corrcoefr; r< �flatr� r �nancorrrA rB r� )rG r� r� r
r
r � test_nancorr1 s z TestnanopsDataFrame.test_nancorrc C s� t j| j| j�d }t j| jj| jj�d }| jtj||dd� t j| j| j �d }t j| jj| j j�d }| j
tj||dd� d S ) Nr ra r� )rH )r ra )r ra )r ra )r ra ) r r� r; r< r� r� r r� rA rB r� )rG r� r� r
r
r �test_nancorr_pearson9 s z(TestnanopsDataFrame.test_nancorr_pearsonc C s� ddl m} || j| j�d }|| jj| jj�d }| jtj||dd� || j| j �d }|| jj| j j�d }| j
tj||dd� d S )Nr )�
kendalltauZkendall)rH ) r� r� r; r< r� r� r r� rA rB r� )rG r� r� r� r
r
r �test_nancorr_kendallA s z(TestnanopsDataFrame.test_nancorr_kendallc C s� ddl m} || j| j�d }|| jj| jj�d }| jtj||dd� || j| j �d }|| jj| j j�d }| j
tj||dd� d S )Nr )� spearmanrZspearman)rH ) r� r� r; r< r� r� r r� rA rB r� )rG r� r� r� r
r
r �test_nancorr_spearmanL s z)TestnanopsDataFrame.test_nancorr_spearmanc
C sb t j| j| j�d }t j| jj| jj�d }d}tjt|d�� | jt j
||dd� W d Q R X d S ) Nr ra z;Unknown method 'foo', expected one of 'kendall', 'spearman')�match�foo)rH )r ra )r ra ) r r� r; r< r� �pytest�raises�
ValueErrorr� r r� )rG r� r� �msgr
r
r �test_invalid_methodW s
z'TestnanopsDataFrame.test_invalid_methodc C s| t j| j| j�d }t j| jj| jj�d }| jtj||� t j| j| j �d }t j| jj| j j�d }| j
tj||� d S )Nr ra )r ra )r ra )r ra )r ra ) r Zcovr; r< r� r� r ZnancovrA rB r� )rG r� r� r
r
r � test_nancov_ s z TestnanopsDataFrame.test_nancovc C s4 | j }| j}| j}| j}| j}| j}| j} �x|j�r.|||�}
tj ||
� |jdkrft
j ||g�} nt
j ||g�} |||�} tj
| | dd� |}
||| �}tj
|
|dd� t
j|ddd�}t
j|ddd�}t
j|ddd�}t
j|dd d�}t
j|dd
d�}t
j|dd d�}t
j| dd d�} t
j|dd
d�}q.W d S )Nra F)rN r )rM rc rc rc rc rc rc rc rc )r r r) r. r+ r, r- rU rW rX r r* �hstackZassert_numpy_array_equalrg )rG r� r� r r r) r. r+ r, r- �res0r� Zres1r� Zres2r
r
r �
check_nancompg s4
z!TestnanopsDataFrame.check_nancompzop,nanopc C s || j | j�}| j||� d S )N)r r r� )rG �opZnanopr� r
r
r �test_nan_comparison� s z'TestnanopsDataFrame.test_nan_comparisonc C sR xLt |dd�rL||�}|r$|s.t�n
| s.t�t|d�s:P tj|ddd�}qW d S )NrU Tr ra )rM rc )rS rY rT r rg )rG rs r} �correctr� r
r
r �
check_bool� s
z TestnanopsDataFrame.check_boolc C s� dddddd d d d d g
}d!d"d#d$d%d&d'd(d)g }x*|D ]"\}}t | |�}| jtj||� q4W xV|D ]N\}}t | |�}| jtj||� | jtj|jd�|� | jtj|jd�|� q`W d S )*Nr Fr r! r$ r% r8 r7 r9 Tr: r r) r+ r. r1 r0 r2 r3 r4 Zf4�f2)r F)r F)r! F)r$ F)r% F)r F)r8 F)r7 F)r9 T)r: T)r F)r) F)r+ F)r. F)r1 T)r0 T)r2 T)r3 T)r4 T)rS r� r Z _has_infsr# )rG �pairsZ pairs_floatZarrr� �valr
r
r �test__has_infs� s6
z"TestnanopsDataFrame.test__has_infsc C s� t j| jjd�st�t j| jjd�s(t�t j| jjd�s<t�t j| jjd�sPt�t j| jjd�sdt�t j| j jd�sxt�t j| j
jd� s�t�t j| j jd� s�t�t j| j jd� s�t�d S )N�test)
r Z _bn_ok_dtyper rO rY r r r! r$ r% r& r' r5 )rG r
r
r �test__bn_ok_dtype� s z%TestnanopsDataFrame.test__bn_ok_dtype)T)TN)N)TTTTT)T)NN)NN)K�__name__�
__module__� __qualname__rI rK r` rh rq r| rv r� �mark� parametrizer �nananyr �any�nanallrf r� r� r� r� Zint16Zint32r� Zfloat32r� rS r� r� re r� r� �tdZskip_if_no_scipyr� �nanminr� �nanmaxr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �operator�eqZnaneq�neZnanne�gtZnangt�geZnange�ltZnanlt�leZnanler� r� r� r� r
r
r
r r
sv @
/
(
0
$
$
-- "
#r
c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestEnsureNumericc C s: t jd�dkst�t jd�dks$t�t jd�dks6t�d S )Nra g�������?y @y �? @y �? @)r �_ensure_numericrY )rG r
r
r �test_numeric_values� s z%TestEnsureNumeric.test_numeric_valuesc C s� t jdddg�}t jtj|�|�s&t�|jt�}t jtj|�|�sFt�t jdddgtd�}t jtj|�|�spt�t jdd d
gtd�}d }tj t
|d �� tj|� W d Q R X d S )
Nra r r� �1�2�3)rO r� �barZbazz Could not convert .* to numeric)r� ) r �array�allcloser r rY r# �objectr� r� ru )rG r� Zo_valuesZs_valuesr� r
r
r � test_ndarray� s
z TestEnsureNumeric.test_ndarrayc C sF t jtjd�d�st�t jtjd�d�s,t�t jtjd�d�sBt�d S ) Nr g �?z1.1g�������?z1+1jra y �?y �? �?)r r r r rY )rG r
r
r �test_convertable_values� s z)TestEnsureNumeric.test_convertable_valuesc C sx d}t jt|d�� tjd� W d Q R X d}t jt|d�� tji � W d Q R X t jt|d�� tjg � W d Q R X d S )Nz Could not convert foo to numeric)r� r� z%argument must be a string or a number)r� r� ru r r )rG r� r
r
r �test_non_convertable_values� s z-TestEnsureNumeric.test_non_convertable_valuesN)r� r� r� r r
r r r
r
r
r r � s
r c @ sX e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e dd� �Z dS )�TestNanvarFixedValuesc C s$ d | _ }| jj|d dd�| _d S )Ng @g �?i�� )Zscale�size)�variance�prngZnormal�samples)rG rH r r
r
r rI s
z"TestNanvarFixedValues.setup_methodc C s& | j }tj|�}tj|| jdd� d S )Ng{�G�z�?)�rtol)r r r� rW rX r )rG r �actual_variancer
r
r �test_nanvar_all_finite s
z,TestNanvarFixedValues.test_nanvar_all_finitec C sp t jt jd| jjd � }| j|d d d�< tj|dd�}tj|| j dd� tj|dd�}tj|t jdd� d S )Nr r T)r g{�G�z�?)r F)
r r( �onesr rL r r� rW rX r )rG r r r
r
r �test_nanvar_nans s z&TestNanvarFixedValues.test_nanvar_nansc C st t jt jd| jjd � }| j|d d d�< tj|dd�}tj|| j d dd� tj
|dd�}tj|t jdd� d S ) Nr r T)r g �?g{�G�z�?)r F) r r( r r rL r r� rW rX r r� )rG r Z
actual_stdr
r
r �test_nanstd_nans s z&TestNanvarFixedValues.test_nanstd_nansc C sV | j }| jj|jd d�}tj||g�}tj|dd�}tj |tj
| j d g�dd� d S )
Nr )r ra )rM g �?� g{�G�z�?)r gUUUUUU�?) r r �uniformrL r r* r r� rW rX r r )rG Z samples_normZ samples_unifr r r
r
r �test_nanvar_axis' s z&TestNanvarFixedValues.test_nanvar_axisc C s� d}| j jd|d fd�}tj|d d �df< tj|dddd�j� }tj|dddd�j� }tj|dddd�j� }d}tj||d d � tj||d | | d d � tj||d |d
| d d � d S )N� i' ra )r Tr )rM r r� r g �?r g{�G�z�?)r g @rc gUUUUUU�?) r r r r( r r� r� rW rX )rG �nr Z
variance_0Z
variance_1Z
variance_2r� r
r
r �test_nanvar_ddof2 s z&TestNanvarFixedValues.test_nanvar_ddofc C sj t jd"�}t jdddgdddgdd d
gg�|d d �d d �f< t j |d < |d d �d f< t jd d
dgdddgdddggdddgdddgdd d ggg�}xbtd �D ]V}xPtd �D ]D}tj|d ||d �}tj|d d � |||f � t j |d �s�t
�q�W q�W xltd �D ]`}xXtd �D ]L}tj |d ||d �}tj|d d � |||f d! � t j |d ��st
��qW �qW d S )#Nrn g*�f#�?g��| 9��?g�ֆ���?g�v���Z�?g���ן?g��צʺ�?g��6bڷ?gE��-�9�?g$mx�P�?r� gӞ����?g/�-�:Ŭ?g��?���?g-u��ll�?g(I�0쓵?g���M6�?g-u��ll�?gF���?gmɉ�M6�?g�
���^�?g)�܌ �?g쵇Z&�?g�@#�)G�?g/��TS��?g6���܃�?g�@#�)G�?g/��TS��?gSW_�܃�?r T)r rM r� g �?)rn rn ) r �emptyr r( re r r� rW rX �isnanrY r� )rG r r rM r� r� r� r
r
r �test_ground_truthE s.
z'TestNanvarFixedValues.test_ground_truthc C s@ t dtjd� �}x(td�D ] }|j|d�}|dks t�q W d S )Ni��-r r� )r� g )r r r re r� rY )rG �datar� r� r
r
r �test_nanstd_roundoffo s z*TestNanvarFixedValues.test_nanstd_roundoffc C s t jjd�S )Ni� )r r � RandomState)rG r
r
r r w s zTestNanvarFixedValues.prngN)
r� r� r� rI r r r r r r r" �propertyr r
r
r
r r
s
*r
c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
dd� �Z dS )�TestNanskewFixedValuesc C s t jt jddd��| _d| _d S )Nr ra �� g�5���?g�5��ȿ)r �sin�linspacer � actual_skew)rG rH r
r
r rI � s z#TestNanskewFixedValues.setup_methodc C s6 x0dD ](}|t jd� }tj|�}|dkst�qW d S )N�fffff�@皙����@� �@i, g )r* r+ r, )r r r r� rY )rG r� r! r� r
r
r �test_constant_series� s
z+TestNanskewFixedValues.test_constant_seriesc C s\ d\}}| j j||dd�}tj|�dk s,t�d\}}| j j||dd�}tj|�dksXt�d S )N�333333�?皙�����?�d )r r )r. r/ )r/ r. )r �betar r� rY )rG �alphar1 � left_tailed� right_tailedr
r
r �test_all_finite� s z&TestNanskewFixedValues.test_all_finitec C s t j| j�}tj|| j� d S )N)r r� r rW rX r) )rG r� r
r
r r � s z(TestNanskewFixedValues.test_ground_truthc C sN t j| jt jt jt| j�� g�}tj|dd�}tj |t j
| j t jg�� d S )Nra )rM ) r r* r r( r r� r r� rW rX r r) )rG r r� r
r
r � test_axis� s "z TestNanskewFixedValues.test_axisc C s2 t j| jt jg�}tj|dd�}t j|�s.t�d S )NF)r )r r� r r( r r� r rY )rG r r� r
r
r � test_nans� s z TestNanskewFixedValues.test_nansc C s2 t j| jt jg�}tj|dd�}tj|| j� d S )NT)r ) r r� r r( r r� rW rX r) )rG r r� r
r
r �test_nans_skipna� s z'TestNanskewFixedValues.test_nans_skipnac C s t jjd�S )Ni� )r r r# )rG r
r
r r � s zTestNanskewFixedValues.prngN) r� r� r� rI r- r5 r r6 r7 r8 r$ r r
r
r
r r% | s r% c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
dd� �Z dS )�TestNankurtFixedValuesc C s t jt jddd��| _d| _d S )Nr ra r& g|v�K�?g|v�K�)r r' r( r � actual_kurt)rG rH r
r
r rI � s z#TestNankurtFixedValues.setup_methodc C s6 x0dD ](}|t jd� }tj|�}|dkst�qW d S )N�fffff�@皙����@� �@i, g )r; r< r= )r r r r� rY )rG r� r! r� r
r
r r- � s
z+TestNankurtFixedValues.test_constant_seriesc C s\ d\}}| j j||dd�}tj|�dk s,t�d\}}| j j||dd�}tj|�dksXt�d S )N�333333�?皙�����?r0 )r r )r> r? )r? r> )r r1 r r� rY )rG r2 r1 r3 r4 r
r
r r5 � s z&TestNankurtFixedValues.test_all_finitec C s t j| j�}tj|| j� d S )N)r r� r rW rX r: )rG r� r
r
r r � s z(TestNankurtFixedValues.test_ground_truthc C sN t j| jt jt jt| j�� g�}tj|dd�}tj |t j
| j t jg�� d S )Nra )rM ) r r* r r( r r� r r� rW rX r r: )rG r r� r
r
r r6 � s "z TestNankurtFixedValues.test_axisc C s2 t j| jt jg�}tj|dd�}t j|�s.t�d S )NF)r )r r� r r( r r� r rY )rG r r� r
r
r r7 � s z TestNankurtFixedValues.test_nansc C s2 t j| jt jg�}tj|dd�}tj|| j� d S )NT)r ) r r� r r( r r� rW rX r: )rG r r� r
r
r r8 � s z'TestNankurtFixedValues.test_nans_skipnac C s t jjd�S )Ni� )r r r# )rG r
r
r r � s zTestNankurtFixedValues.prngN) r� r� r� rI r- r5 r r6 r7 r8 r$ r r
r
r
r r9 � s r9 c @ s& e Zd Zejjdddg�dd� �ZdS )�TestDatetime64NaNOps�tzN�UTCc C s� t jdd|d�}|d }x0|t|�t|�gD ]}tj|�}||ks,t�q,W |jdt j�}x0|t|�t|�gD ]}tj|�}||kslt�qlW d S )Nz
2016-01-01r� )ZperiodsrA ra ) �pdZ
date_ranger r r r� rY �insertZNaT)rG rA Zdti�expected�objr� Zdti2r
r
r r� � s
z!TestDatetime64NaNOps.test_nanmean)r� r� r� r� r� r� r� r
r
r
r r@ � s r@ c C sL t jrHtjdd� tjd�s t�tjdd� tjd� s<t�tjdt� d S )NZuse_bottleneckTF)r Z_BOTTLENECK_INSTALLEDrC Z
set_option�
get_optionrY rJ r
r
r
r �test_use_bottleneck� s rH znumpy_op, expectedr g @ra rn c C s& | t jddddg��}||ks"t�d S )Nra r r� rn )rC r rY )Znumpy_oprE r� r
r
r �test_numpy_ops� s rI � operationc C sF t jddtjdtjdg�}|j� }| |�}| ||d�}||ksBt�d S )Nra r r� rn )�mask)rC r r r( r rY )rJ r� rK Zmedian_expectedZ
median_resultr
r
r �%test_nanops_independent_of_mask_param s
rL )>� functoolsr r� r� Znumpyr r� Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandas.core.dtypes.commonr ZpandasrC r r Zpandas._testingZ_testingrW Zpandas.core.arraysr Zpandas.core.nanops�corer r rJ rT r[ Zfixturer r
r r
r% r9 r@ rH r� r� r� r� r� r� r� r� r� r� r� r� rI r� r� r� r� r� r� r� r� r� r� rL r
r
r
r �<module> sn
<0w11
|