U Zj<@s&ddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddZddZdd Zd d Zd d ZddZejddgddgddidddgddZddZddZdd Zd!d"Zd#d$Z ejd%d&d'ej!ej!ej!ej!gd(d)d*d)gd*d+d,d+gej!ej!ej!ej!gd)d+d,d+gd,d-d.d-gd.d/d0d1ggfd2dd3d3d)d)gd(d)d*d)gd)d+d+d)gd3d3d+d+gd)d+d,d+gd4d-d5d+gd-d/d.d+ggfgd6d7Z"d8d9Z#d:d;Z$ej%d<d=d>Z&d?d@Z'dAdBZ(dCdDZ)ejdEdFdGdHgejdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWgdXdYZ*dS)ZN)SpecificationErrorUnsupportedFunctionCall) DataFrameIndex MultiIndexPeriodSeries Timestampconcat date_rangetimedelta_rangecCsttjdd}|jd|d}t|jj|dd|j|jd|dd}|jj |dd|jdksnt |jd|dd}t|jj|dd|jddg|jd|dddg}t|jj|dd|jddgdS)Nwindowstep)rr) rnprandomZrandnrollingtmassert_index_equalZ _selected_objcolumnsnameAssertionError)rframerr_/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/window/test_api.py test_getitems $rc Csptddggddgd}|jdd}tjtdd |d gW5QRXtjtd d |dd gW5QRXdS) NrABrr rzColumns not found: 'C'matchCz^[^A]+$)rrpytestraisesKeyError)dfgrrrtest_select_bad_cols's  r-c Csdtddggddgd}|jdd}t|j|dd}tjt|d  |j W5QRXdS) Nrr r!r"r#r r$z%'Rolling' object has no attribute 'F'r%) rrrassert_series_equalr!sumr(r)AttributeErrorF)r+rmsgrrrtest_attribute_access2s  r3cCsttdtdddd}|jd|d}|ddg}ttjtjdd d gtjtjd d d gdtdddd|}t||dS)Nr foor!r"r'rrr!r" r!r"ABr#) rrangerr/rnanlistrassert_frame_equalrr+rresultexpectedrrrtests_skip_nuisance<s$rEc Csttdtdddd}|jd|d}d}tjt|d|}W5QRXttjtjdd d gtjtjd d d gdt dddd|}t ||dS)Nr r4r5r6rrzGnuisance columns.*Dropped columns were Index\(\['C'\], dtype='object'\)r%r7r8r9r:r;r<r=r#) rr>rrassert_produces_warning FutureWarningr/rr?r@rA)rr+rr2rCrDrrrtest_skip_sum_object_raisesHs$rHc Csttdtdddd}|jd|d}|d}|d}|d}|d }|d }|tjtjg}t||||gd d } t dd gd d gg| _ t || |tjtjd}t||gd d } t j || dd|dd d gi}t||gd d } t ddg| _ t || |dd dg}t||gd d } d dg| _ t || d} tjt| d|dd ddiW5QRXtjt| d"|d ddd dddW5QRX|d d gd d gd}t||||gd d } ddddg} t | | _ t j || dddS)Nr rr4r r<rrr!r"raxismeanstdTZ check_liker!rK)r!rLr/nested renamer is not supportedr%)rKr/)Zmean2Zsum2)r"rK)r"rL)rr>rrKrLr/ aggregaterr r from_productrrrA from_tuplesr(r)r) rr+rZa_meanZa_stda_sumZb_meanZb_stdrCrDr2Zexp_colsrrrtest_aggVsD           rTfuncminrKmaxbr/prodmedian)rXcc CsXtdddgdddgdddgd }|jddd }tjtd d ||W5QRXdS) Nrr rr r7)arXr[)rrJz"axis other than 0 is not supportedr%)rrr(r)NotImplementedErroragg)rUr+rrrrtest_multi_axis_1_raisess"rbcCsttdtdddd}|jdd}|d}|tjd d d}|d jd d |d }t||gdd}tj ||dddS)Nr rr4r r<rr$r!cSstj|ddSNr)ZddofrrLxrrrz test_agg_apply..r"cSstj|ddSrcrdrerrrrgrh)rawrrITrM) rr>rr/rarapplyr rrA)rir+rrSrCZrcustomrDrrrtest_agg_applys  rkcCsttdtdddd}|jd|d}|tjtjgj}t t dd d gg}t |||d tjtjgj}t d d g}t |||d tjtjgij}td d g}t ||dS)Nr rr4r r<rrr=r/rKr!)r!r/rN)rr>rrarr/rKrrrQr@rrrrRrBrrrtest_agg_consistencys   rlc Cs,ttdtdddd}|jdd}d}tjt|d &|d d d gid d d gidW5QRXt|d |d  |d |d  gdd}t ddddg|_ tjt|d .|d d g dd dgidd dgidW5QRXtjt|d &| dd dgidd dgidW5QRXdS)Nr rr4r r<rr$rOr%r!rKr/r")r1r2rrI)rarK)rorL)rbrK)rprLrorLrp)rr>rr(r)rrPr rKrLrrRrra)r+rr2rDrrrtest_agg_nested_dictss  ** 2rqcCs~ddddddddd d d g }td td tdg}tdddgdddgtdtd ddtddddtdtdtdgddtjgddtjgddtjg|tdtdtdgd |d!}tdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgdddgd |d!dd|}|j dd"|d# }t |||j dd"|d$ }| tdd|}t ||dS)%Nintfloatstringdatetime timedeltaperiodsfl_inffl_nanstr_nandt_nat periods_natZ20170101Z20170203rr rg@g@g@abc)rwz1 ss)rwfreqz2012-01z2012-02z2012-03g?g@ZaaZbb) rrrsrtrurvrwrxryrzr{r|r#rr min_periodsr)rr)r rr@r r rrZInfNaNrcountrrAZnotnaZastypers)rcolsZ dt_nat_colr+rDrCrrrtest_count_nonnumeric_typessp        rcCsNttddd}|d}|d}|jdks@gA@gD@gN@gT@g@U@P expandingg$@gRB:@gI@c Cstdddgdddgdddgdddgdddgdddgdddggd d d gd }t|d |}|rh||}n|}tjd ddddddgd dgd}tddddg}t|||d}|ddgddgd} t| |dS)Nr!r4rr(r"rZZstocklowhighr#)r!r)r!r)r!r )r"r)r"r\)r"r )r"r7)names)rrK)rrW)rrK)rrV)indexrrKrWrV)rr)rgetattrgroupbyrrRrarrA) rUZ window_sizeZ expected_valsr+frrrrDrCrrrtest_multiple_agg_funcss2!    rc Cs6ttdd}tt|jr(tW5QRXdSNr)rr>rrrFrGZis_datetimelikerr~rrrtest_is_datetimelike_deprecatedWs rc Cs<ttdd}tt|dks.tW5QRXdSr)rr>rrrFrGvalidaterrrrrtest_validate_deprecated]s rz ignore:min_periods:FutureWarningcs`ttdjd||||dfddjD}t|fddjD}||ks\tdS)Nr)centerclosedrrcsi|]}|t|qSrr.0attrZroll_objrr ksz=test_dont_modify_attributes_after_methods..csi|]}|t|qSrrrrrrrms)rr>r _attributesrr)Zarithmetic_win_operatorsrrrrrDrCrrr)test_dont_modify_attributes_after_methodscs  rc Csttdjddd|dd}tjt|d&ttdjddd|dW5QRXttd jddd|dttd jddd|dd }tjt|d&ttd jddd |dW5QRXdS) Nr4rTr)rrrJrz&No axis named 1 for object type Seriesr%r)r4r4z)No axis named 2 for object type DataFramer ) rronesrrKr(r) ValueErrorr)rr2rrrtest_centered_axis_validationqs2 *  rc Cstdddddg}|jdd|d}ttt|dd|}t||d}tj t |d $tdddgjdd|dW5QRXdS) Nrr rr\r rr!min_periods 5 must be <= window 3r%) rrrVrrlenrr.r(r)rrr_rCrDr2rrrtest_rolling_min_min_periodss rc Cstdddddgtjd}|jdd|d}|dd|}t||d }tjt |d $tdddgjdd|dW5QRXdS) Nrr rr\r )dtyperrrr%) rrfloat64rrWrZassert_almost_equalr(r)rrrrrtest_rolling_max_min_periodss rzroll_type, class_name)rZRolling)rZ Expanding)ewmZExponentialMovingWindowzkernel, has_args, raises)r/TT)rWTT)rVTT)rKTT)rZFF)rLTT)varTT)ZskewFF)semTT)ZkurtFF)quantileFF)ZrankFF)covFF)corrFFc Csttdddg|d}d}|dkr*d}nd}|dkrn|d krntjtd |d d t||W5QRXnd |d|}tjt|d R|rtjt|d t|||dt j iW5QRXnt|||dt j iW5QRX|rd|d|}tjt|d `|rR|dks$|dkrRtjt|d t|||dW5QRXnt|||dW5QRXd|d|}tjt|d \|rtjt|d  t|||ddt j iW5QRXnt|||ddt j iW5QRXdS)Nr r\r7z2numpy operations are not valid with window objectsr)g?rr)r/rKrLrrrzhas no attribute ''r%zPassing additional kwargs to .rzPassing additional args to rrrrz&Passing additional args and kwargs to )rr rr\)rr rr\)rr rr\)rr rr\) rrr(r)r0rrFrGrrr) Z roll_type class_nameZkernelZhas_argsr)r error_msgZ required_argsZwarn_msgrrrtest_args_kwargs_deprs6 " &r)+numpyrr(Z pandas.errorsrrZpandasrrrrrr r r r Zpandas._testingZ_testingrrr-r3rErHrTmarkZ parametrizerbrkrlrqrrr?rrrfilterwarningsrrrrrrrrrs,    0 C              #