a < bO@sddlmZddlmZddlZddlmZddlZddl Z ddl m Z ddl m mZddlmZddlZddlmZmZmZmZmZmZmZmZmZmZmZddlm Z!ddl"m#m$Z$ddl%m#m&Z&d8d d Z'd9ddZ(d:ddZ)d;ddZ*GdddZ+GdddZ,GdddZ-ddZ.ddZ/e j01dgde j01d d!d"gd#d$Z2d%d&Z3e j0j4d'd(d)d*Z5e j01d+gd,d-d.Z6e j01d/ed0ej7gd1d2Z8d3d4Z9e j01d gd5d6d7Z:dS)<) timedelta)DecimalN)tzlocal)is_platform_windowsis_categorical_dtype) Categorical DataFrameIndex MultiIndexSeries Timestamp date_rangeisnanotna to_datetime to_timedeltaTFh㈵>:0yE>c st||} |r|dvrtnd} tdtdddi} t| t| |} Wdn1s^0Yt| tsvJtt | | d<t| t| |} Wdn1s0Yt| tsJt | sJ|rDfdd } t |}| d d d }| d d d }tj || | |||dtj ||j | d d||dn}| d d}| d d}tj || ||||d|dvr|j |d d}tj ||d ||d|r|j j}||jksJ||jksJtjtdd| ddWdn1s0Y|r|tj}t||d d}t||d d}|dvr|dkrRd nd }t||j|jd}t ||t||j|jd}t ||dS)a Check that operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame alternative : function Function that opname is tested against; i.e. "frame.opname()" should equal "alternative(frame)". frame : DataFrame The object that the tests are executed on has_skipna : bool, default True Whether the method "opname" has the kwarg "skip_na" check_dtype : bool, default True Whether the dtypes of the result of "frame.opname()" and "alternative(frame)" should be checked. check_dates : bool, default false Whether opname should be tested on a Datetime Series rtol : float, default 1e-5 Relative tolerance. atol : float, default 1e-8 Absolute tolerance. skipna_alternative : function, default None NaN-safe version of alternative )meanmedianNbz1/1/2001periodsacs |jSNvaluesx alternativer/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/frame/test_reductions.pywrapperWsz$assert_stat_op_calc..wrapperrFaxisskipna) check_dtypertolatolr')r+r,sumprodNo axis named 2matchr0indexdtype)getattr FutureWarningr rtmassert_produces_warning isinstancer rangelenZ_make_skipna_wrapperassert_series_equalapplyrr6pytestraises ValueErrornpNaNr5)opnamer"frame has_skipnar* check_datesr+r,skipna_alternativefZexpected_warningdfresultr%skipna_wrapperresult0result1expectedZ lcd_dtypeall_nar0r1unitr#r!r$assert_stat_op_calc!sr%  * *          *   rUcCspt||ddt||dd|rlt||dddt||dddt||dddt||ddddS)a Check that API for operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame float_frame : DataFrame DataFrame with columns of type float float_string_frame : DataFrame DataFrame with both float and string columns has_numeric_only : bool, default False Whether the method "opname" has the kwarg "numeric_only" rr-r)Tr' numeric_onlyFN)r7)rE float_framefloat_string_framehas_numeric_onlyr#r#r$assert_stat_op_apisr[c s`t||}|rhfdd}fdd}|ddd}|ddd}t|||t||j|dd n}}|dd }|dd }t|||tj||j|dd dd tjtd d |d d Wdn1s0Y|r\|tj} t| |dd } t| |dd } |dkr@| r0J| r\Jn| sNJ| s\JdS)a Check that bool operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame alternative : function Function that opname is tested against; i.e. "frame.opname()" should equal "alternative(frame)". frame : DataFrame The object that the tests are executed on has_skipna : bool, default True Whether the method "opname" has the kwarg "skip_na" cs|j}|Sr)dropnar)r Znonar!r#r$rMs z+assert_bool_op_calc..skipna_wrappercs |jSrrrr!r#r$r%sz$assert_bool_op_calc..wrapperrFr&r)r-)r*r1r2rNany) r7r9r>r?r@rArBrCrDr]all) rEr"rFrGrJrMr%rNrOrQrRrSr#r!r$assert_bool_op_calcs6       (  r_cCs|}tjt|dk|d<t||ddt||dd|rt||dddt||dddt||dddt||dddd S) a Check that API for boolean operator opname works as advertised on frame Parameters ---------- opname : str Name of the operator to test on frame float_frame : DataFrame DataFrame with columns of type float float_string_frame : DataFrame DataFrame with both float and string columns has_bool_only : bool, default False Whether the method "opname" has the kwarg "bool_only" g?Z_bool_rr-r)T)r' bool_onlyFN)rCrandomrandnr=r7)rEbool_frame_with_narY has_bool_onlymixedr#r#r$assert_bool_op_apisrfc-@seZdZejdddZejdejddZ ddZ ejdd Z ejd d d Z ej d gdddZej dgdddZddZddZej dddgddZej dddgddZd d!Zd"d#Zej d$gd%d&d'Zd(d)Zejd*d+Zej d,d-d.gd/gd0gd1ged1gd1gd2ed3ged4gd5fd6d.gd/gejgejejged7eejgd1gd2e j!gee j!gd5fd-d8d9ejejgd8d9ejejgd:ejejejged1ejejejgd1gd2egd;egd<gd=d>fd6d8d9ejejgd8d9ejejgd:ejejejgeejd1ejejgd1gd2egd?egd@gd=d>fgdAdBZ"dCdDZ#dEdFZ$dGdHZ%dIdJZ&dKdLZ'ej dMdNdOgej dPgdQdRdSZ(dTdUZ)dVdWZ*dXdYZ+dZd[Z,d\d]Z-d^d_Z.d`daZ/dbdcZ0dddeZ1dfdgZ2dhdiZ3djdkZ4dldmZ5dndoZ6ej dpdqdrdsgfdtdrdugfgdvdwZ7dxdyZ8ej dzd{d|gd}d~Z9ej dzd{d|gddZ:ddZ;ej ddrd:gej dd{d|gej dd-d6gddZej dej?id6fej@id-fej?dgid6fej@dgid-fej?dd6d6gid6fej@dd6d6gid6fej?dd-d6gid-fej@dd-d6gid6fej?dd-d-gid-fej@dd-d-gid-fej?d6gd6gdd6fej@d6gd6gdd6fej?d6d6gd6d-gdd-fej@d6d6gd6d-gdd6fej@deAdd0gdd7id6fej?deAdd0gdd7id-fej@deAdrd:geBd7id6fej?deAdrd:geBd7id-feCej@deAdrd:gdd7id6eCej@deAdrd:gdd7id6eCej?deAdrd:gdd7id-eCej?deAdrd:gdd7id-eCej@deAd:dgdd7id-eCej@deAd:dgdd7id-eCej?deAd:dgdd7id-eCej?deAd:dgdd7id-eCej@deAdrd:gdd7id6eCej?deAdrd:gdd7id-eCej@deAd:dgdd7id-eCej?deAd:dgdd7id-ej@deAdrd:gdd7id-fej?deAdrd:gdd7id6fej@deAd:dgdd7id-fej?deAd:dgdd7id6feCej@eAddgdd7eAddgdd7dd-g#ddZDddZEddZFej d d{d|gddZGddZHdS)TestDataFrameAnalyticsz)ignore:Dropping of nuisance:FutureWarningcCstd||td||td||ddtd||td||td||td ||td ||td ||ddtd ||td ||td||td||dS)Ncountr/nuniqueF)rZrproductrminmaxmadvarstdsemr[selfrXrYr#r#r$test_stat_op_api s"          z'TestDataFrameAnalytics.test_stat_op_apicCstd||td||dS)Nskewkurtrqrrr#r#r$test_stat_op_api_skew_kurt s z1TestDataFrameAnalytics.test_stat_op_api_skew_kurtc Csdd}dd}dd}dd}d d }d d }td ||ddddtdtj|ddddtdtj|tjdtdtj|ddtdtj|tjdtd||td||td||td||td||dddddS)NcSs t|Sr)rr/sr#r#r$rh'sz7TestDataFrameAnalytics.test_stat_op_calc..countcSstt|Sr)r= algorithmsZunique1dr\rxr#r#r$ri*sz9TestDataFrameAnalytics.test_stat_op_calc..nuniquecSst||Sr)rCabsrrr#r#r$rm-sz5TestDataFrameAnalytics.test_stat_op_calc..madcSstj|ddSNr)Zddof)rCrnrr#r#r$rn0sz5TestDataFrameAnalytics.test_stat_op_calc..varcSstj|ddSr|)rCrorr#r#r$ro3sz5TestDataFrameAnalytics.test_stat_op_calc..stdcSstj|ddtt|Sr|)rCrosqrtr=rr#r#r$rp6sz5TestDataFrameAnalytics.test_stat_op_calc..semriFT)rGr*rHr/Zfloat32gMbP?)r*r+)rIrrHrjrmrnrorprh)rUrCr/astypeZnansumrr0Znanprod) rsfloat_frame_with_naZmixed_float_framerhrirmrnrorpr#r#r$test_stat_op_calc&sP       z(TestDataFrameAnalytics.test_stat_op_calccCs,dd}dd}td||td||dS)NcSs*ddlm}t|dkrtjS||ddS)Nr)ruFZbias) scipy.statsrur=rCnan)r rur#r#r$skewnesses  zHTestDataFrameAnalytics.test_stat_op_calc_skew_kurtosis..skewnesscSs*ddlm}t|dkrtjS||ddS)Nr)kurtosisFr)rrr=rCr)r rr#r#r$rvls  zDTestDataFrameAnalytics.test_stat_op_calc_skew_kurtosis..kurtrurvrU)rsrrrvr#r#r$test_stat_op_calc_skew_kurtosiscs z6TestDataFrameAnalytics.test_stat_op_calc_skew_kurtosiszignore::RuntimeWarningcCs.dd}td||ddtd||ddddS)NcSst|rtjSt|Sr)rr]rCrrrr#r#r$r%zs z3TestDataFrameAnalytics.test_median..wrapperrTrF)r*rHr)rsr int_framer%r#r#r$ test_medianxs  z"TestDataFrameAnalytics.test_medianmethod)r/rr0rnrorurkrlcCsgdgdgdd}t|gddd}ttjdgtjd gtjd gd td }||fD]L}|jjtjksnJt||d }t|d|d }|dvrXt ||qXdS)N)gȃw@Ba@gCVZg ?Z.F?)rr)g94c4?g@}jX?gNrrcfoobarbazOr4rrrrr)rr6r)Zf8r.) r rCrobjectrr6Zobject_r7rr9r>)rsrdatadf1df2rKrLrPr#r#r$%test_stat_operators_attempt_obj_arrays & z)rsrKtestZaltr#r#r$test_reduce_mixed_frames z.TestDataFrameAnalytics.test_reduce_mixed_framecCstgdgddtjdgd}t|tddddt|jddtddddt|jdd tdddd t|jddd tdddd dS) Nr)r)r)r)rrr)rABCrFr\r-r)r'r\)r rCrr9r>rir )rsrKr#r#r$ test_nuniques  z#TestDataFrameAnalytics.test_nuniquetzNUTCcCsrtddgtd|dgdd}tt|}Wdn1sH0Ytdgdgd}t||dS) Nr)2000rrrrrrr5)r r r9r:r8rr r>rsrrKrLrPr#r#r$ test_mean_mixed_datetime_numerics   &z7TestDataFrameAnalytics.test_mean_mixed_datetime_numericcCshtdtd|dgdi}tt|}Wdn1sB0Yttjd}t ||dS)Nrrrrr) r r r9r:r8rr rCfloat64r>rr#r#r$test_mean_excludes_datetimess  & z3TestDataFrameAnalytics.test_mean_excludes_datetimesc Csddtddddtddddtddddtddd dtd ddd td dddtd dd dtddddtddddtddg }t|}tjtdd|}Wdn1s0Ytddgddgd}t||dS)Nrz628.00rr)z383.00rz651.00z575.00rz1114.00TESTz241.00z572.00z609.00z820.00z1223.00rr2g@gL@rrr)rr r9r:r8rr r>)rsrrKrLrPr#r#r$test_mean_mixed_string_decimals$ &z5TestDataFrameAnalytics.test_mean_mixed_string_decimalcCs|jdd}|dd}t|||jdd}|dd}t||ttjddd}tj |dd }|dk rJt d d .tj |dd }|dk rJWdn1s0YdS) Nrr}cSs |jddSNrr})rorr#r#r$z5TestDataFrameAnalytics.test_var_std..cSs |jddSr)rnrr#r#r$r rr)rrr-rF) ror?r9assert_almost_equalrnrCrepeatrananopsZnanvarr]rrrsZdatetime_framerLrParrr#r#r$ test_var_stds    z#TestDataFrameAnalytics.test_var_stdmeth)rprnrocCsBttjddgdd}d|jd<ttjddgdd}d|jd<t||dd d }t|d d g|dd }t||t||dd d }t|d d g|dd }t||d}tj t |d"t||ddd Wdn1s0Yd}tj t |d"t||ddd Wdn1s40YdS)NrrrrZ100)rrrr)TrVrrr-z6unsupported operand type\(s\) for -: 'float' and 'str'r2Fz&could not convert string to float: 'a') r rCrarblocr7r9r>r@rA TypeError)rsrrrrLrPmsgr#r#r$test_numeric_only_flags     0z-TestDataFrameAnalytics.test_numeric_only_flagcCs|jdd}|dd}t||ttjddd}tj|dd}|dk rZJt d d .tj|dd}|dk rJWdn1s0YdS) Nrr}cSs|jddtt|Sr)rorCr~r=rr#r#r$r2rz1TestDataFrameAnalytics.test_sem..rrrr-rF) rpr?r9rrCrrarZnansemr]rrrr#r#r$test_sem0s  zTestDataFrameAnalytics.test_semcCstdggdddgggdgdgdgd}ttjd d |d }|}tt"|jdd  d}Wdn1s~0Ytj ||d d|j dusJ|j dksJdS)Nr)onetwoZthreerr))rrrrrr)rr)rrr)r)rr)rr)rr))ZlevelscodesrrlevelF)Z check_names) r r rCrarbrvr9r:r8xsr>name)rsr5rKrvZkurt2r#r#r$ test_kurt=s 0z TestDataFrameAnalytics.test_kurtzdropna, expectedT g$@rr) categories2000-1-21 days)rrrDEFGFr r))rNaTrrrrrr)rr)rr)HIJKLMNrrrr)rrrrcCstgdddtjdgdtjtjtjgtjtjdtjgttjtjdtjgtgdtgdgdgd ddtjtjgtdtjdtjgtgd tgd tjd d dd}|t|j |d}t|}t ||dS)N)rr rrr)rrr)rrrr)rrrr)rrrr)rrrrrint64r)rrrrrrrrrrrrrrr) r rCrrrrarangesortedkeysmoder9assert_frame_equal)rsr\rPrKrLr#r#r$test_mode_dropnaLs(7      z'TestDataFrameAnalytics.test_mode_dropnacCstdtjtjddgi}tddtjgi}tt0|jdd}|jddjdd}Wdn1sj0Yt ||dS)NrrFr)ZbyT)Zdrop) r rCrr9r: UserWarningrZ sort_valuesZ reset_indexr)rsrKrPrLr#r#r$test_mode_sortwarnings   2z,TestDataFrameAnalytics.test_mode_sortwarningcCsBtgddgd}|}tgddgtgtdd}t||dS)Nrrrr)rr5)r rr rr9rrsrKrLrPr#r#r$test_mode_empty_dfsz)TestDataFrameAnalytics.test_mode_empty_dfcCs"ttddddtddddtdtdddd }t|d |d |d |d d }|}|d|jdksxJ|d|jdksJ|jdd}||jdksJ|}|d|jdksJ|d|jdksJ|jdd}||d ksJ|}t|}t|d |d |d |d d }t ||t ||| }d|d <d|d<d|d<td|d<|}t t tddt tdddddtdg|jd}t ||t jtdd |jdd}Wdn1s0Yt gd!gd"d}t |||d d gd}t tddgd}t |||d d g}t tddtddgd d gd}t ||ttd#dd$td%dd$d&}|d'|d(|d)<|d)jd*ksJ|d(|d'|d+<||d)jd*ks J|d+jd*ksJdS),Nz2012-1-1rrrfreqz2012-1-2Z20120101r)minutessecondsrrrrrr)rrr))rrr-)rr)rrrrrZ20130101ri1r )daysrSelect only validr2)r)r)rrZ20130102rZ20130105)timetime2rrZoff1timedelta64[ns]Zoff2)r rr rrkrr^rlr{r9rcopyr r Timedeltarr>r:r8r6_consolidate_inplace)rsrKZdiffsrLZresult2rPrer#r#r$test_operators_timedelta64s|  &  &    ,     z1TestDataFrameAnalytics.test_operators_timedelta64cCstjddd}t||d}|}tj|jd<|jdd}t|dtjgdd gd d }t |||jd dd }tt dgdtjt dg}t ||dS)Nz1 Dayrrr)rFr(rrrr4r)r&rr) rtimedelta_ranger rrilocror r9r>r)rsZtdirKrLrPr#r#r$!test_std_timedelta64_skipna_falses   $z8TestDataFrameAnalytics.test_std_timedelta64_skipna_falsecCsZt}|d}|d}t|ts(Jt|ts6Jt|dksFJt|dksVJdS)Nrr))r r/r;r r=)rsZ empty_frameZaxis0Zaxis1r#r#r$test_sum_corners  z&TestDataFrameAnalytics.test_sum_cornerz method, unit)r/r)r0r)rW)NTFcCsgd}t||g|tjgtjtjgd}t|||d}t|||g|dd}t||t|||dd}t||tjg|d}t||t|||dd}t|||g|dd}t||t|jdd||dd}t|tjtjg|d}t||t|gd |gd tjgd d }t|||d d}t|d d gd}t||t|||dd}t|d d gd}t||dS)NrrWrr4r))rW min_countrrrrrrrr)r rCrr7r r9r>r)rsrrTrWidxrKrLrPr#r#r$test_sum_prod_nanops#s*"    & z+TestDataFrameAnalytics.test_sum_prod_nanopscCsgd}tddgdtjgtjtjgd}|t}|}tgdd|d}t|||jdd}t|||jdd}tddtjgd|d}t||dS)Nrr)rrrm8[ns])r6r5r r)) r rCrr?rr/r r9r>)rsr!rKrrLrPr#r#r$test_sum_nanops_timedeltaEs"     z0TestDataFrameAnalytics.test_sum_nanops_timedeltacCsJtgdgdd}|jdd}ttjtjgddgd}t||dS) Nrrrr)r yrr$r r'r)r r/r rCrr9r>rr#r#r$test_sum_nanops_min_countZs z0TestDataFrameAnalytics.test_sum_nanops_min_countcCs6|jt}t||j|jd}|td}|dS)Nr5rr))rrrr r5rrr/)rsrXrrFZdeltasr#r#r$test_sum_objectas  z&TestDataFrameAnalytics.test_sum_objectcCs"t|}|d|ddS)Nr)r)rCisnanr/)rsrXboolsr#r#r$ test_sum_boolgs  z$TestDataFrameAnalytics.test_sum_boolcCsvttdddgddgd}tjtdd|}Wdn1sP0Ytd d i}t||dS) Nrrrrr)rrrrr2rg@) r rrr9r:r8r/r r>rr#r#r$test_sum_mixed_datetimems& z.TestDataFrameAnalytics.test_sum_mixed_datetimecCstjtdd|jdd}Wdn1s00Y|jddd}t|j|jt|jt|jkspJtjtdd|jdd}Wdn1s0Y|jddd}t|j|j|ddk|d <|d}|d |d j ksJdS) Nrr2rr-TrVr)rbool) r9r:r8rr/Zassert_index_equalr5r=rr)rsrXrYZthe_meanZthe_sumZmeansr#r#r$test_mean_cornerxs** z'TestDataFrameAnalytics.test_mean_cornerc Csttdtdddtjdddtjddddd}|jd d }tdd i}t ||t t |}Wdn1s0Ytd |j d d }t ||dS)Nr 2016-01-01r1D2016rr )rrrrTrrr)r)rr)r rCrrrr period_rangerr r9r>r:r8rrr#r#r$test_mean_datetimelikes      &z-TestDataFrameAnalytics.test_mean_datetimelikecCsttdtdddtjdddd}|jdd}td|jd |jd d}t ||tj d dd d |d<t j tdd|jddWdn1s0YdS)Nrr1rr2rFrr))r)rr4r3rr rz"mean is not implemented for Periodr2)r rCrrrrrr rr9r>r5r@rArrr#r#r$)test_mean_datetimelike_numeric_only_falses    z@TestDataFrameAnalytics.test_mean_datetimelike_numeric_only_falsecCsDtjjddd}t|dd}|jdd}t|}t||dS)Nr)rr)sizeInt64rTr)rCrarandintr rr9r>)rsrrKrLrPr#r#r$*test_mean_extensionarray_numeric_only_trues    zATestDataFrameAnalytics.test_mean_extensionarray_numeric_only_truecCsZtjtdd8|d|d|d|dWdn1sL0YdS)Nrr2r))r9r:r8rornrru)rsrYr#r#r$test_stats_mixed_types   z,TestDataFrameAnalytics.test_stats_mixed_typecCs8ttdtdd}t|}|jddddks4JdS)Nr)rr)r-r)r r<rr/)rsrKr,r#r#r$test_sum_boolssz%TestDataFrameAnalytics.test_sum_boolsc Cs|}tj|jdd<tj|jddddf<dD]H}dD]>}||fD]0}|j||d}|jtj||d}t||qDq8q0d } tj t | d |jd d Wdn1s0YdS Nrrrr)TF)rr)r&z)No axis named 2 for object type DataFramer2rr-) rCrridxminr?r r9r>r@rArB rsrXrrFr(r'rKrLrPrr#r#r$ test_idxmins z"TestDataFrameAnalytics.test_idxminc Cs|}tj|jdd<tj|jddddf<dD]H}dD]>}||fD]0}|j||d}|jtj||d}t||qDq8q0d } tj t | d |jd d Wdn1s0YdSr>) rCrridxmaxr?r r9r>r@rArBrAr#r#r$ test_idxmaxs z"TestDataFrameAnalytics.test_idxmaxcCsPtddd}tgdtdddd|jddd}|}tgd gdd }t|||}tgd gdd }t||t j |j d <|}tgd gdd }t|||}tgdgdd }t|||ddd|d <| |}tgdgdd }t|||}tgdgdd }t||dS)Nr1rr)rrr)rT)deepr)r)rrr)rrr)rrr)r)rrrr)rrr)r) rr r<rrCr r9r>r@rrrrrsZdtirKrLrPr#r#r$test_idxmax_mixed_dtypes. *      z.TestDataFrameAnalytics.test_idxmax_mixed_dtypezop, expected_valuerCrrr@rcCs^tgdgdddd}|d}t||}td|itdd gddd d }t||dS) N)drHrHrIrI)rrrr)rr)IDvaluer9rrJrKrHrI)rr6r)r groupbyr7r r9r)rsrZexpected_valuerKrLrPr#r#r$!test_idxmax_idxmin_convert_dtypess  z8TestDataFrameAnalytics.test_idxmax_idxmin_convert_dtypescCstddd}t||dddd}|}tj|jd<||jdd}tgd }t |||j dd}tgd }t ||dS) Nr1rrr)rr)rrr)r-)rrr)rrr) rr rrrrrrCr r9r>r@rFr#r#r$"test_idxmax_dt64_multicolumn_axis1)s       z9TestDataFrameAnalytics.test_idxmax_dt64_multicolumn_axis1rEr]r^cCst|||dddS)NT)rd)rf)rsrErcrYr#r#r$ test_any_all?sz#TestDataFrameAnalytics.test_any_allcCs$|d}t|tt||dddS)NT)rG)Zfillnar_r7rC)rsrErcrKr#r#r$test_any_all_bool_frameEs z.TestDataFrameAnalytics.test_any_all_bool_framecCstgdgdgddgdd}|ddgd }tgdgdd}t|||ddgjd d d }t|||d }tgdgdd}t|||jd d d }t|||jdd }|d usJ|jdd }|d usJ|dgjdd }|d usJdS)N)TFF)TTF)TTTrrrrrr)Tr`r-Fr)r r]r r9r>r^itemrr#r#r$test_any_all_extraLs.       z)TestDataFrameAnalytics.test_any_all_extrar' bool_agg_funcr(cCsttdtjtjdgtjdtjdgtjtjtjdgtjtjdtjggd}t||||d}tgd}t||dS)Nr)Tr5rr&)TTTT)r rCrr7r r9r>)rsr'rTr(rKrLrPr#r#r$test_any_all_object_dtypems z0TestDataFrameAnalytics.test_any_all_object_dtypecCs\dtjdtjg}tdtdtjtjg}t||d}|d}tgd}t ||dS)Nr)rz 1960-02-15z 1960-02-16r)TTTF) rCrr rrr r]r r9r>)rsZ float_dataZ datetime_datarKrLrPr#r#r$test_any_datetime~s  z(TestDataFrameAnalytics.test_any_datetimecCstgdgdgdd}|jdd}ttjd}t||tgdgdgdgdd }|jdd}td d i}t||dS) Nrr&)NNN)col1col2col3TrQr)FFT)rYrZr[col4r\F)r r^r rCbool_r9r>rr#r#r$test_any_all_bool_onlys      z-TestDataFrameAnalytics.test_any_all_bool_onlyzfunc, data, expectedrrrrzM8[ns]z M8[ns, UTC]rr#categoryrrcCst|}d}tdd|jDr$t}tj|ddd||}Wdn1sR0Yt|tjslJ| |us|Jtj|ddd&t t||j dd}Wdn1s0Yt|tjsJ| |usJdS)Ncss|]}t|VqdSrr).0r r#r#r$ rz>TestDataFrameAnalytics.test_any_all_np_func..rFr3Zcheck_stacklevelr-) r r]Zdtypesr8r9r:r;rCr]rRr7__name__)rsfuncrrPwarnrLr#r#r$test_any_all_np_funcs 5&4z+TestDataFrameAnalytics.test_any_all_np_funccCsLttddgd}|dus$Jttddgd}|dusHJdS)NrrrTF)rCr^r rRr])rsrLr#r#r$test_any_all_objects z*TestDataFrameAnalytics.test_any_all_objectcCs0tddgddgdt}|tddg|d<|dd|d<|}|d dg}t|||jdd d }tddgd dgd }t |||d dgjdd d }t |||jddd rJ|j dd d }tddgd dgd }t |||d dgj dd d }t |||j ddd s,JdS) NrrTFrrr_rrr)r`r'r) r rrrr Z_get_bool_datar9rr^r>r])rsrKresrPr#r#r$test_any_all_object_bool_onlys&      z4TestDataFrameAnalytics.test_any_all_object_bool_onlyc Cstdditjddgddggddgdd }d }tjt|d Ltt"t ||ddd Wdn1sp0YWdn1s0YdS) Nrr)rrroutin)namesrz.Must specify 'axis' when aggregating by level.r2)r'r) r r Z from_productr@rArBr9r:r8r7)rsrrKZxprr#r#r$#test_any_all_level_axis_none_raisess z:TestDataFrameAnalytics.test_any_all_level_axis_none_raisescCstgd}tdtjdtjgi}tgd}ttjtjdg}td:|j|dddD]}t|||dd q^Wdn1s0YdS) N)rrrrrrr)r)lowerr')ltlegtgeeqner-)r rCrr r9r:Zclipr7)rsrKZdf_nanryZs_nanrr#r#r$test_series_broadcasting*s   z/TestDataFrameAnalytics.test_series_broadcasting)Irc __module__ __qualname__r@markfilterwarningsrttdZskip_if_no_scipyrwrrr parametrizerrrrrrrrrrrrrrrCrrrrrrrr rrrr"r%r(r*r-r.r0r6r7r;r<r=rBrDrGrMrNrOrPrSrWrXr^r]r^r rparamrfrgrirmrur#r#r#r$rgsH    =                   5  X    &   !  3 " rgc@seZdZddZddZddZddZej d d d gej d d dge dddZ ddZ ddZ ddZddZddZdS)TestDataFrameReductionscCstdtjtjtdgi}|}ttdgdgd}t|||}ttdgdgd}t||tdtjtjgi}|}ttjgdgd}t|||}ttjgdgd}t||dS)Nrz 2012-05-01r) r rrr rkr r9r>rl)rsrKrhexpr#r#r$test_min_max_dt64_with_NaT9s   z2TestDataFrameReductions.test_min_max_dt64_with_NaTcCs|}t|trtrtdttd|dtd|dgtd|dtjgd}|j ddd }t |j d tjg}|j |d j ksJt |||jddd }t |j d tjg}|j |d j ksJt ||dS) NzSGH#37659 OSError raised within tzlocal bc Windows chokes in times before 1970-01-01z2020-01-01 08:00:00rz1920-02-01 09:00:00z2020-02-01 08:00:00rrr)Fr&)rrr)rr)r;rrr@skipr r rrrkr rr6r9r>rl)rsrequestZtz_naive_fixturerrKrhrPr#r#r$'test_min_max_dt64_with_NaT_skipna_falsePs(    z?TestDataFrameReductions.test_min_max_dt64_with_NaT_skipna_falsecCstdtgi}ttg}|jddjtju|tjuksBJ|jddjtju|tjukshJt |jdd|t |jdd|dS)Nr rr-r)) r rr rkr rrrlr9r>)rsrKZexpected_dt_seriesr#r#r$*test_min_max_dt64_api_consistency_with_NaTns  &&zBTestDataFrameReductions.test_min_max_dt64_api_consistency_with_NaTcCstdgi}tgtd}t|jddjt|ks>Jt|jddjt|ksdJt |jdd|t |jdd|dS)Nr rrr-r)) r r rrCr+rkr rlr9r>)rsrKZexpected_float_seriesr#r#r$*test_min_max_dt64_api_consistency_empty_df}s   &&zBTestDataFrameReductions.test_min_max_dt64_api_consistency_empty_dfinitialz2018-10-08 13:36:45+00:00z2018-10-08 13:36:45+03:00rrkrl)rcCs<t|}t|g}t|g}t||dd}t||dS)Nr)r-)rr r r7r9r>)rsrrZ initial_dtrPrKrLr#r#r$test_preserve_timezones   z.TestDataFrameReductions.test_preserve_timezonecCstdgdigdgdgd}tjtdd|jdd }Wdn1sP0Ytdd d gid d gd}t||tjtdd|jdd }Wdn1s0Ytdd d gid d gd}t||dS)Nr)FFTFTFT)rrrrrrr)rr)rrr)rrrzUsing the levelr2rrFTrr)r r9r:r8r]rr^)rsrKrLexr#r#r$test_frame_any_all_with_levels * *z5TestDataFrameReductions.test_frame_any_all_with_levelcCsttddgttddtddgd}|jdd}tddgd d gd }t|||jdd}tddgd }t||dS) Nrryr)ms)rtr-FTrr)rr5rV)r r rr]r9r>rr#r#r$test_frame_any_with_timedeltas    z5TestDataFrameReductions.test_frame_any_with_timedeltacCsJ|gd}tjtdd|jddWdn1s<0YdS)Nrr(r2r)r9r:r8rm)rsframe_or_seriesobjr#r#r$'test_reductions_deprecation_skipna_nones z?TestDataFrameReductions.test_reductions_deprecation_skipna_nonecCsd|gdtgdgdgd}tjtdd t||ddWdn1sV0YdS)Nrr&rrr2rr)r Z from_arraysr9r:r8r7)rsrreduction_functionsrr#r#r$*test_reductions_deprecation_level_arguments zBTestDataFrameReductions.test_reductions_deprecation_level_argumentcCsd|dvrtd|gd}d}tjt|d t||ddWdn1sV0YdS)N)rhrmz)rsrserrKrLrPr#r#r$.test_any_all_categorical_dtype_nuisance_columns&*,.0 2zBTestNuisanceColumns.test_any_all_categorical_dtype_nuisance_columncCstdtgdi}|d}tjtdd|Wdn1sH0Ytjtdd|jddWdn1s0Ytjtdd|}Wdn1s0Yt gt gt j d}t |||dt|d <tjtdd|jddWdn1s&0Ytjtdd|}Wdn1s^0Yt d gd gd }t ||dS) Nr)r)rrrrrr2Frrr4rrr)r rr@rArrr9r:r8r r rCrr>rr)rsrKrrLrPr#r#r$-test_median_categorical_dtype_nuisance_columns*&*& ,(zATestNuisanceColumns.test_median_categorical_dtype_nuisance_columnrkrlcCsTtgddd}t|}|d}tjtddt||Wdn1sR0Ytjtddtt||Wdn1s0Ytjtdd t||ddWdn1s0Ytj t ddt||}Wdn1s0Ytgt gtj d }t ||tj t ddd tt||}Wdn1sh0Yt |||dt|d <tj t ddt||}Wdn1s0Y|d krtd gd gd}ntdgd gd}t ||tj t ddd tt||}Wdn1s:0Yt ||dS)N)rrrrF)Zorderedrzis not ordered for operationr2rrr4rbrrkrrr)rr rr@rArr7rCr9r:r8r rr>rr)rsrcatrrKrLrPr#r#r$:test_min_max_categorical_dtype_non_ordered_nuisance_columnsD *,., . ,  .zNTestNuisanceColumns.test_min_max_categorical_dtype_non_ordered_nuisance_columncCstgdgddtd}tjtdd|}Wdn1sF0Ytdgdgd }t|||dt d |d <tjtdd|}Wdn1s0Ytdd gdd gd }t||dS) Nrrrrrr2rrrrrg@) r rr9r:r8rr r>rrrr#r#r$3test_reduction_object_block_splits_nuisance_columnsFs& &zGTestNuisanceColumns.test_reduction_object_block_splits_nuisance_columnsN) rcrvrwr@rxr{rrrrr#r#r#r$rs  # /rcCstdtjddd}d|d<t|}|jdd}tt j d d t j g}t |||jd dd }t |||jd dd }tt j d d t j dd t j dd t j g}t ||dS)Nrzm8[s]rrZNat)rrFrrrrr&r)rr)rCrrrviewZreshaper r/r rrrr9r>)rrKrLrPr#r#r$!test_sum_timedelta64_skipna_false]s"       rcCsPtddggtdd}|ddi}|}tddgddgd}t||dS)Nrr)abrrr9r)r listrr/r r9r>)rKrLrPr#r#r$!test_mixed_frame_with_integer_sumws rrW)TFNrrkrlcCsntd}t|jd|jgtd}td|i}t|||d}tt||gt dgddd}t ||dS)Nrrr9rrr) rCZiinfor rlrkrZ Int64Dtyper r7r r9r>)rrWZ int64_inforrKrLrPr#r#r$test_minmax_extensionarrays  rcCsttjddtj}||}|}|}t|||jdd}|jdd}t||t j |j ddddf<|}|}|j ddddf|d<t|||jdd}|jdd}|j j gddddf|ddd<t||dS)NrHrr)r-r)rrr)r rCrarbrrrmr9r>rNArrrZany_signed_int_ea_dtyperKrrLrPr#r#r$test_mad_nullable_integers"        (rz GH#42895 caused by lack of 2D EA)reasoncCsttjddtj}||}tj|jdddf<tj tjgt ||d|jdddf<| }| }tj|d<t ||dS)NrHrr)r)r rCrarbrrrrrrr=rmr9r>rr#r#r$ test_mad_nullable_integer_all_nas ( rr)rlrkr/rrcCsttjddtjddgddgd}|}tt t||dd}Wdn1s^0Yt||t|j dd|d d }t||dS) Nr)hoursg333333(@g*@rrrFr) r rrrr9r:r8r7rrL)rrKrPrLr#r#r$*test_groupby_regular_arithmetic_equivalents  . rts_valuez 2000-01-01cCsttdgdgdg|gd}tjtdd|}Wdn1sF0Ytgdtdd }t||dS) Nr)皙?rrzDropping of nuisancer2)r)rrabcr)r r9r:r8r/r rr>)rrKrLrPr#r#r$.test_frame_mixed_numeric_object_with_timestamps &rcCs~tgd}|jdddd}tdg}t||td}tjt |d |j ddddWdn1sp0YdS) N)r)rTrr)F)r'r rWrz2unsupported operand type(s) for +: 'int' and 'str'r2) r r0r r9r>reescaper@rArr/)rKrLrPrr#r#r$$test_prod_sum_min_count_mixed_objects    r)rkrlrrrurvcCsttjdd}t||}d|d}tjt|d|dd}Wdn1sV0Ytd|}Wdn1s0Yt||t||dddS)Nrzscalar z over the entire DataFramer2r-r) r rCrarbr7r9r:r8r>)rrKrrrhrPr#r#r$$test_reduction_axis_none_deprecations  ( $ r)TTFrrN)T)T)F);datetimerdecimalrrZ dateutil.tzrZnumpyrCr@Z pandas.compatrZpandas.util._test_decoratorsutilZ_test_decoratorsrzZpandas.core.dtypes.commonrZpandasrrr r r r r rrrrrZpandas._testingZ_testingr9Zpandas.core.algorithmscorerzZpandas.core.nanopsrrUr[r_rfrgr}rrrrxr{rrZxfailrrrrrrr#r#r#r$sh     4  o  > :