ioGdZddlmZddlmZddlZddlZddlZddl Z ddl m Z ddl Z e jjdZe j"j%dd gdd ggd d g Zd d gifZd gdiefZge j,dgfej.dgfe j,dgfej.ddgfe j,dgfej.ddfe j,dgfej.dfe j,dgfej.ddfe j,dgfej.dfe j,ddgfej.dfe j,ddgfej.dfe j,ddgfej.dfe j,ddgfej.ddfe j,ddgfej.ddfe j,ddgfej.ddd gfe j,ddgfej.dd dgfe j,ddgfej.ddgfe j,e j0de j2gfej.d dfe j,ddgfej.d!dd ife j,ddgfej.d"fe j,ddgfej.d#dd gfe j,ddgfej.d$ddfe j,ddgfej.d%fe j,ddgfej.d&fe j,ddgfej.d'fe j,ddgfej.d(fe j,d gfej.d)e j,d gfe j,dge j4d*d +fej.d,fe j,dge j6d*d +fej.d-fe j8e j:eej.d.e j:d g/fe jjd01e j:eej.d2fe j:eej.dd fe j:eej.dd gfe j:eej.dej0dgfe j:d3d gifej.dd gfe j:eej.d4d5fe j:eej.d6d7d89fe j:eej.d:d;<fe j:eej.d=d >fe j:eej.dd gfe j:eej.ddd gfe j:eej.dd g?fe j:eej.ddg/fe j:eej.dd di?fe j:eej.dd@/fe j:eej.d d fe j:eej.dAd fe j:eej.dfe j:eej.d%fe j:eej.d&fe j:eej.d'fe j:eej.d(fe j:eej.dBfe j:eej.dfe j:eej.dfe j:eej.dCd Dfe j:eej.dEfe j:eej.dFd d fe j:eej.dGd d fe j:eej.dHfe j:eej.d)e j:efe j:eej.dIe j:eej<fe j:eej.dJe j:efe j8e j:eej.dKe j:efe1e j:eej.dLd ?fe j:d gd gdMfej.dNd ?fe j:d gd gdMfej.dNd dOdPgQfe j:eej.dRfe j:eej.dSd fe j:eej.dTfe j:gdUgdVgdWdXfej.dYd gd gZfe j:eej.d[d\fe j:eej.ddfe j:eej.d]fe j8e j:eej.d^fe jj?d_g1e j:eej.d`e j:efe j:eej.dafe j:eej.dbfe j:eej.dcfe j:eej.ddfe j:eej.defe j,dgej.defe j:eej.dffe j:eej.dgdhfe j:eej.dgdidjgdkfe j:d e j@d le j@dlgifej.dgdmhfe j:d ejBdnejBdogifej.dgdhfe j:d d gie jDd*dpgfej.d,fe j:d d gie jFd*gfej.d-dpqfe j:eej.d#d gfe j:eej.d#e j,d gfe j:eej.d#e j:d d gife j:eej.drd fe j:eej.dsd fe j:eej.dtduvfe j,d dgfej.dtfe j,d dgfej.dwd/fe j:eej.dwd?fe j:eejHfe j,d gejHfe j:eejJfe j,d gejJfe j:eejLfe j,d gejLfe j:ee'fe j,d ge'fe j:ee(fe j,d ge(fe j:eej.dddgfe j:eej.dxdfe j,d efej.dxdfe j:eej.dyd fe j:eej.dze j:d gd{ife j,eej.dze j,gd|fe j:eej.d}d~fe j:eej.dd~fe j,d dd gfej.d}d~fe j,d dd gfej.dd~fe j,ddgfej.dCfe j,d gdzfej.dfe j:d d gdzifej.dfe j,d gdzfej.dfe j:d d gdzifej.dfe j,d dgfej.dddfe j:efej.dddfe j,d dgfej.de)fe j:eej.de)fe j,d dgfej.dfe j:eej.dfe j,d dgde*fej.dfe j:d ej0d dge*ifej.dfe j,d dgfej.dfe j:eej.dfe j,gdfej.dfe j:d gdifej.dfe j,d dgfej.dd fe j:eej.dd fe j,d e j6d*d+fej.ddfe j:d gdie j6d*d+fej.ddfe j,d e j6d*d+fej.dedfe j:d gdie j6d*d+fej.dedfe j,d e j6d*d+fej.dddfe j:d gdie j6d*d+fej.dddfe j,d dgfej.dfe j:eej.dfe j,d dgfej.dej0ddmgfe j:eej.dej0dggfe j,d dgfej.dej0ddmgfe j:eej.dej0dggfe j,d dgfej.ddfe j:eej.ddfe j,d e j6d*ddfej.ddfe j:d gdie j6d*ddfej.ddfe j,d e j6d*d+fej.ddfe j:d gdie j6d*d+fej.ddfe j,d dgfej.dfe j:eej.dfe j,d dgfej.dfe j:eej.dfe j,d gfej.ddfe j:eej.ddfe j,d gfej.ddfe j:eej.ddfe j,d gfej.dfe j:eej.dfe j,d gfej.dfe j:eej.dfe j,d gfej.dfe j:eej.dfe j,d gfej.dfe j:eej.dfe j:eej.dfe j:eej.dfe j:eej.dfe j:eej.dfe j:eej.dPfe j:eej.dfe j:eej.dOfe j:eej.dfe j:eej.dfe j:eej.dfe j:eej.dfZ+dZ,e jj[de+ddZ.edZ/e jj[dgde jj[dd e j,d gfd e j:d d gife j,d gd fe j:d d gid fe j,d ge j,d gfe j:d d gie j:d d gife j,d ge j:d d gife j:d d gie j,d gfgddZ0e jj[de j,e j:ge jj[de j,e j:gdĄZ1e jj[dgej.dƫej.dǫej.ddgej.ddej.dadej.dd˫ej.ddej.ddΫej.dddmϫej.ddej.ddej.dydej.ddej.dӫej.ddej.dիej.d֫ej.ddثej.ddګej.ddej.ddej.ddej.d!dd ej.ddej.ddej.ddej.ddej.dej.ddej.dd dej.ddej.dej.dej.ddd iej.dej.ddej.ddej.dej.dej.dej.dej.dej.dej.dej.dej.dej.de,dZ2e jj[dej.d-ej.ddej.d٫ej.ddej.ddej.ddej.ddej.dej.dg e,dZ3e jj[dgddZ4e jj[dgddZ5e jj[dej.dgdZ6e jj[dej.ddgej.dej.dd ej.d dej.d ej.d d d d ej.dd dgej.dd d gg edZ7e jj[de j,ddge j:dd gd dgdMge jj[dej.dPddddgdZ8e jj[de j,ddge j:dd gd dgdMge jj[dŐddddddgedZ9dZ:e jj[dd ggd!e jj[d"gd#d"e de;d e;fd$Zd'Z?e jj[gd(dmdifddmifdmdmifggd)d*e;d+e;d,e@fd-ZAy(.zG An exhaustive list of pandas methods exercising NDFrame.__finalize__. )deepcopy)timeN)MergeHowznot implemented)reasonabAB)names)r take __getitem__Trepeatr reset_index)dropto_framedrop_duplicates duplicatedroundrenamec |dzS)Nr xs w/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pandas/tests/generic/test_finalize.pyr/sQUnameset_axisreindexrfillnareplaceshiftisinbetweenisnaisnullnotnanotnulladd2000periods to_timestamp to_perioddot)indexzImplement binary finalize)marks transpose)r rqueryzA == 1evalA + 1pythonengine select_dtypesint)includeassign)r)columnsc|SNrrs rrr_sqr set_indexdropna sort_values)by sort_indexnlargest nsmallest swaplevelcombine combine_firstupdatepivot)r r pivot_tablemeansum)rAaggfuncstackexplodeunstack)rrc)r r)r r)r r Cmelt)id_vars value_varsmapc|SrCrrs rrrsar corrcovzignore::RuntimeWarningcorrwithcountnuniqueidxminidxmaxmodemedianquantile) numeric_onlyg?g?)qrj)daysFz 2022-01-01z 2022-01-02D)freq droplevelpopsqueezerAaxis rename_axisxsget reindex_like)r r r)rr r add_prefix_ add_suffixr headtailsample)nr%astypecopy infer_objectsdtypeconvert_dtypes)r Nr interpolateclip)lowerrasfreqh)r r r r at_time between_timez12:00z13:00rankwheremasktruncate)beforeUTC)r0tz tz_convertCET tz_localizedescribe pct_change transformc(||jz SrCminrs rrrMsQ[r c(||jz SrCrrs rrrQsQ[r applyc|SrCrrs rrrSsr c|SrCrrs rrrTs1r cumsumcummincummaxcumprodanyallrmaxstdprodsemskewkurtctjd}|jt|}|r|j dSt|S)Nz'(.*)?'r )recompilesearchstrgroup)rxprms ridfnrms< **Z C 3q6Awwqz1v r ndframe_methodct|dS)N)rrs rrrvs tAbE{r )idscb|\}}}||}ddi|_||}|jddik(sJy)Nrr attrs)rcls init_argsmethodndframeresults rtest_finalize_calledrvsA+CF9oG!HGM G_F <rrrr >rrrr) isinstancerrSeriesrr>rskipoperatoreqnegtgeltlealign)requestrrall_binary_operatorsrris_cmprs r test_binopsrs"KD%$ryy12 %",, 23 6js3 ?@7z%5 @A##JtS,A1X $$Zs-CAh !      &F*T2<<0Zryy5QjjQj/ e JtRYY/Jubll4Skk$Qk/ t !$ .F < "--23AAhAG QTT4 F <r"r?s rtest_timedelta_propertyrGHsP "$$VQ78AAhAG QTT4 F <..{sAr )rrs rrr{s !''+&r c$|jdS)NrRaggrs rrr| !%%,r c$|jdS)NrQr]rs rrr} !%%-r c$|jdS)Nrhr]rs rrr~s !%%/r crddi|_||jddgd}|jddik(sJy)Nrr rF) group_keysrgroupbyrXrrs rtest_groupby_finalizerhts@aCI CKKA5K9 :F <Esv6Es+%2%E~" << / /3Q QQ Qr ctjddgij|}tjddgij|}tj||}|xr|}|jj |k(sJy)Nrr r)rrr merge_asofrr)rwrxrrrrs r9test_merge_asof_correctly_sets_duplication_allowance_flagrs <<! & 0 0 0 WD LL&1# ' 1 1 . 2 E]]4 'F%2%E~" << / /3Q QQ Qr cdddgi}tjddgi}||_tjddgi}t||_|j |dd}|j|k(sJ|j|jusJ|j|jusJ|jd|jdusJ|jd|jdusJy)Nrr r rrzr)rrrrr)metadatarrrs r8test_merge_propagates_metadata_from_equal_input_metadatarsaV}H <<! &DDJ LL&1# 'E8$EK ZZ7vZ 6F <<8 ## # <rsn+  #{{((0A(B]]#saV 4S#JGQCj] |$b) EYY!---fb9:EYY!---mdVDEEYY!---h:;EYY!---m<= E YY!---m$GH E YY!---j9: EYY!Q 08001BCDEYY!Q 0800>?EYY!Q 08009:EYY!Q 0800?KLEYY!Q 08006BCEYY!Q 0800c3ZHIEYY!Q 0800QFCDEYY!Q 0800!=>EYY1bee*%')>)>)>x)KLE YY!Q 0800QFCD!E"YY!Q 08009:#E$YY!Q 0800!Q@A%E&YY!Q 0800AqAB'E(YY!Q 080089)E*YY!Q 0800:;+E,YY!Q 08009:-E.YY!Q 0800;)>)>}se)TUaEb\\:4x44WhGHcEd\\:4x44VWXVWeEf\\:4x44_eTUgEh\\:4x44XCDiEj\\:4x44Z#GHkEl\\:4x44YAGHmEn\\:4x44VcUKLoEp\\:4x44VA3GHqEr\\:4x44XSzRSsEt\\:4x44X[QRuEv\\:4x44XsCDwEx\\:4x44[#FGyEz\\:4x44]CD{E|\\:4x44V<=}E~\\:4x44X>?E@\\:4x44W=>AEB\\:4x44Y?@CED\\:4x44X>?EEF\\:4x445FGHGEH\\:4x44\BCIEJ\\:4x44]sKLKEL\\:4x44\BCMEN\\:4x44ZCHIOEP\\:4x44[!SIJQER\\="7("7"7 "DESEV e\R\\:%>?UEb iz)BHLLQaEl o|r||Z/HIkEtFLL LL  !H ! !(LBLL*,E F # uED\\:4x44WcJKEEH s! mS9GER s! mS65/RQEZ\\:4x44W=>[E\\\:4x44YDE]E^\\="7("7"7 "BC_Eb YY ?AfseFaEj\\:4x44UKHIkEl\\:4x44Wa@AmEn\\:4x44V<=oEpFLL z#88#8#8#?@ KK & &'? @ qE~ j,",, *CD}EF\\:4x44W=>GEH\\:4x44Y?@IEJ\\:4x44X>?KEL\\:4x44X>?MEN\\:4x44V<=OEPYY*X**623QER\\:4x44X>?SEV jt<UE` jT4LtL_Ej  !$lbll&:; <>ju=iEt  l+]R]]<-HI JLjt<sE~ sibii,-.n-}EH slbll6*+,k4GEP\\="7("7"7"DEQER\\="7("7"7  1#"OPSEV flbllC!:&>?UE^\\="7("7"7 S"IJ_E`\\:4x44UC@AaEd\\:4x44YYOPeEfYY!Q 0800;?h,_Ej MBMM&! ,-ib2iEt | mbmmFA>?ib2sE~ MBMM&! ,-ngw?}EH | mbmmFA>?ngw?GEPYY!Q 080089QER\\:4x44V<=SETYY!Q 0800("((D%=:QRSUEV\\:4x44Whbhhx>PQRWEXYY!Q 08004-9PQRYEZ\\:4x44VXRXXvh=OPQ[E\YY!Q 0800AFG]E^\\:4x44ZJK_Eb MBMM&! 67lE2aEl | mbmmFA%HIlE2kEv MBMM&! ,-mU3uE@ | mbmmFA>?mU3EH YY!Q 0800<=I EJ \\:4x44Z@AK EL YY!Q 0800>?M EN \\:4x44\BCO EP YY!---k;PQRQ ET k+@AS E\ YY!---g{CD] E^ \\="7("7"7"MN_ Eb YY!---h78c Ed \\:4x44X>?e Ef YY!---h78g Eh \\:4x44X>?i Ej YY!---h78k El \\:4x44X>?m En YY!---i89o Ep \\:4x44Y?@q Et \\:4x44U;? IBIIqcN LBLL#s $% A3 sQCj !1% A3A3( sQCj !<2<<qc #;< A3sQCj12 sQCj !9299aS>2  C !$ @!$H"))R\\!:;299bll";<<=<< 0l+0j)0 ecU+0 j#. 0 gs+ 0 h0 0 j#.0 i50 iUC0 fc*0 i-0 eQ'0 gs+0 e$0 gq)0 g&!0" h'#0$ gu-%0& k51'0( eQ')0* k3/+0, h*-0. ic2/00 gs+102 h,304 gq)506 lC0708 h'90: gq);0< oqs;=0> lC0?0@ g&A0B j)C0D kC:6E0F g&G0H fa(I0J gq)K0L i(M0N i(O0P i(Q0R i(S0T i(U0V i(W0X i(Y0Z k*[0\ k*]0^ m,_0b g4j$k4j$ k*mU3k*j$/gs+gs+fc*l+j)  $$ <$=<$ L$$$9H$9$9/$J#KL$M$ .6l+n-13789133GH2S#J?.c ;  $ $ IBIIq!f |r||1a&1v,FG H e$&!  $ $  IBIIq!f |r||1a&1v,FG H )  $ $ &'@  R RRR   R&&'@ R R R  R5$; b ub r 4$$$$$r