U
|
ý°dè$ ã @ s d Z ddlZddlZddlZddlm mZ ddl Z
|
ddl m Z m Z ddl mZ ddlmZmZmZ G dd dZG dd dZdS ) zP
|
Tests for statistical reductions of 2nd moment or higher: var, skew, kurt, ...
|
é N)Ú DataFrameÚSeries)Ú DatetimeArrayÚ PeriodArrayÚTimedeltaArrayc @ s~ e Zd Zej deeje g¡dd Z
|
ej deeje g¡ej dddddd g¡d
|
d Z ej deeje g¡d d ZdS )Ú TestDatetimeLikeStatReductionsÚboxc C sÀ |}t jdd|d}| dddddd d
|
d d d dg ¡}|j}||}| ¡ t jd|dks\t|jddt jd|dksztt j|d< ||}| ¡ t jd|dks¦t|jddt jks¼td S )Nú
|
2001-01-01é )ÚperiodsÚtzé é é é
|
é é é é r é é z
|
2001-01-06)r F©Zskipnaéþÿÿÿz2001-01-06 07:12:00)ÚpdÚ
|
date_rangeÚtakeÚ_dataÚmeanÚ TimestampÚAssertionErrorÚNaT)ÚselfZtz_naive_fixturer r ÚdtiZdtarrÚobj© r$ úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/reductions/test_stat_reductions.pyÚtest_dt64_mean s
|
z-TestDatetimeLikeStatReductions.test_dt64_meanÚfreqÚSÚHÚDÚWÚBc C sà t jddd}| dddddd d
|
d d d dg ¡}|j |¡}||}tjtdd | ¡ W 5 Q R X tjtdd |jdd W 5 Q R X t j |d< tjtdd | ¡ W 5 Q R X tjtdd |jdd W 5 Q R X d S )Nr r
|
©r r r r r r r r r r r r Z ambiguous©ÚmatchTr r )
|
r r r r Z to_periodÚpytestÚraisesÚ TypeErrorr r )r! r r' r" Zparrr# r$ r$ r% Útest_period_mean- s
|
z/TestDatetimeLikeStatReductions.test_period_meanc C sÀ t jddddddddddd g d
|
d }|j}||d d }| ¡ }t |¡ ¡ }||ksXtt j|d< |jd dt jksxt|jdd}||dd
¡ kst| d¡|d d d¡ks¼td S )Nr r r iùÿÿÿr r éÿÿÿÿr r r* )ÚunitF)Úcopyr TÚusg &@r ) r ZTimedeltaIndexr r ÚnpÚarrayr r Úround)r! r ZtdiZtdarrr# ÚresultÚexpectedZresult2r$ r$ r% Útest_td64_meanD s $
|
z-TestDatetimeLikeStatReductions.test_td64_meanN)Ú__name__Ú
|
__module__Ú __qualname__r0 ÚmarkZ parametrizer r ZIndexr r& r r3 r r= r$ r$ r$ r% r s
|
r c @ sz e Zd ZdddZdd Zdd Zdd Zd
|
d Zd d Zdd Z dd Z
|
dd Z e j dd Ze j dd Zdd ZdS ) ÚTestSeriesStatReductionsFc
|
C s t dd¡î tt|}tj|dd
< |dkrptt jddd}d | d
|
}tjt |d || W 5 Q R X t
|
||¡st t ||dd ¡st | ¡ } t || || j¡ t |||| j¡ |tj }
|
|rìt ||
|
¡sìt td ddd dg} || dg} | tdd¡ t| dd} t t|| t|| j¡ |r|tt jddd} || } || }| |ks|t |dkr°tjt d d |ttd W 5 Q R X d}tjt|d ||d d W 5 Q R X dt |¡jkrø||dd W 5 Q R X d S ) NZuse_bottleneckFr é )ÚmaxÚminr ÚmedianÚstdz1/1/2001r r- z does not support reduction 'ú'r. r r r r r l l è Úint64©Zdtypez1/1/2000)ÚsumrE rD Úabcz&No axis named 1 for object type Series)ZaxisÚ numeric_onlyT)rM ) r Zoption_contextÚgetattrr r8 ÚNaNr r0 r1 r2 Znotnar ÚisnaZdropnaÚtmÚassert_almost_equalÚvaluesÚnanÚisnanÚextendÚrangeÚfloatZ bdate_rangeÚlistÚ
|
ValueErrorÚinspectÚgetfullargspecÚargs)r! ÚnameZ alternateZstring_series_Ú check_objectsÚ check_allnaÚfZdsÚmsgZnonaZallnaÚsÚitemsÚresÚexpr$ r$ r% Ú_check_stat_op^ sF
|
|
|
z'TestSeriesStatReductions._check_stat_opc C s& t ¡ d¡}| jdtj|dd d S )NÚseriesrK F)r` )rQ ÚmakeStringSeriesÚrenamerg r8 rK ©r! Ú string_seriesr$ r$ r% Útest_sum s z!TestSeriesStatReductions.test_sumc C s" t ¡ d¡}| dtj|¡ d S )Nrh r )rQ ri rj rg r8 r rk r$ r$ r% Ú test_mean s z"TestSeriesStatReductions.test_meanc C sR t ¡ d¡}| dtj|¡ ttjdtdt dd}t
|
t |¡| ¡ ¡ d S )Nrh rF r rJ )Úindex) rQ ri rj rg r8 rF r ÚonesÚintrW rR )r! rl Zint_tsr$ r$ r% Ú test_median£ s z$TestSeriesStatReductions.test_medianc C s" t ¡ d¡}| dtj|¡ d S )Nrh Úprod)rQ ri rj rg r8 rs rk r$ r$ r% Ú test_prod« s z"TestSeriesStatReductions.test_prodc C s& t ¡ d¡}| jdtj|dd d S )Nrh rE T©r_ )rQ ri rj rg r8 rE rk r$ r$ r% Útest_min¯ s z!TestSeriesStatReductions.test_minc C s& t ¡ d¡}| jdtj|dd d S )Nrh rD Tru )rQ ri rj rg r8 rD rk r$ r$ r% Útest_max³ s z!TestSeriesStatReductions.test_maxc C sÜ t ¡ d¡}t ¡ d¡}dd }| d||¡ dd }| d||¡ |jdd }tj|jdd }t ||¡ |j dd }tj |jdd }t ||¡ |j
|
d
|
g }|j d d }t |¡s¾t |jd d }t |¡sØt d S ) Nrh Útsc S s t j| ddS ©Nr ©Zddof)r8 rG ©Úxr$ r$ r% Ú<lambda>» ó z7TestSeriesStatReductions.test_var_std.<locals>.<lambda>rG c S s t j| ddS ry )r8 Úvarr{ r$ r$ r% r} ¾ r~ r r rz r r )rQ ri rj ÚmakeTimeSeriesrg rG r8 rS rR r Úilocr rP r ©r! rl Zdatetime_seriesÚaltr; r< rc r$ r$ r% Ú test_var_std· s" z%TestSeriesStatReductions.test_var_stdc C s t ¡ d¡}t ¡ d¡}dd }| d||¡ |jdd}tj|jddt t
|
|j¡ }t ||¡ |j dg }|jd d}t |¡std S )
|
Nrh rx c S s t j| ddt t| ¡ S ry )r8 rG ÚsqrtÚlenr{ r$ r$ r% r} Õ r~ z3TestSeriesStatReductions.test_sem.<locals>.<lambda>Úsemr rz r r )rQ ri rj r rg r r8 rG rS r
r rR r r rP r r r$ r$ r% Útest_semÑ s ÿ z!TestSeriesStatReductions.test_semc sÆ ddl m t ¡ d¡} fdd}| d||¡ d}td|d D ]z}tt |¡}t
|
t ||f¡}||k rt | ¡ ¡st t | ¡ ¡ ¡ sÀt qFd| ¡ ks¬t | ¡ dk ¡ sFt qFd S ) Nr ©Úskewrh c s | ddS ©NF)Zbiasr$ r{ r r$ r% r} é r~ z4TestSeriesStatReductions.test_skew.<locals>.<lambda>r r r )Ú scipy.statsr rQ ri rj rg rW r r8 rp r rU r Úall)r! rl r Úmin_NÚirc Údfr$ r r% Ú test_skewã s z"TestSeriesStatReductions.test_skewc s8 ddl m t ¡ d¡} fdd}| d||¡ d S )Nr ©Úkurtosisrh c s | ddS r r$ r{ r r$ r% r} ÿ r~ z4TestSeriesStatReductions.test_kurt.<locals>.<lambda>Úkurt)r r rQ ri rj rg )r! rl r r$ r r% Ú test_kurtù s z"TestSeriesStatReductions.test_kurtc C s d}t d|d D ]z}tt |¡}tt ||f¡}||k rht | ¡ ¡sPtt | ¡ ¡ ¡ stqd| ¡ ksxt| ¡ dk ¡ stqd S )Nr r r ) rW r r8 rp r rU r r r )r! r r rc r r$ r$ r% Útest_kurt_corner s z)TestSeriesStatReductions.test_kurt_cornerN)FF)r> r? r@ rg rm rn rr rt rv rw r r ÚtdZskip_if_no_scipyr r r r$ r$ r$ r% rB Y s ÿ
|
=
|
|
rB )Ú__doc__r[ Únumpyr8 r0 Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr r r Zpandas._testingZ_testingrQ Zpandas.core.arraysr r r r rB r$ r$ r$ r% Ú<module> s A
|
