U NZf@sLdZddlmZddlZddlZddlmZmZm Z ddl m Z ddl mZmZmZdddd d d d d ddg Zd gZejddZejddZejddZejdeejdddgejdddgejdddgddZdd Zd!d"Zd#d$Zejd%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6gd7d8Zd9d:Zd;d<ZdS)=zV test methods relating to generic function evaluation the so-called white/black lists )ascii_lowercaseN) DataFrameSeries date_range)groupby_other_methodsreduction_kernelstransformation_kernelssumprodminmaxmedianmeanskewstdvarsemc CsBtddddddddgddddddddgtjdtjddS)NfoobarZoneZtwoZthree)ABCD)rnprandomZrandnrrG/tmp/pip-unpacked-wheel-nbcvw55c/pandas/tests/groupby/test_allowlist.pydf&s  rcCsRttt}d}|tjdd|}t|dttj|t|d}|S)N r)floatsletters) rarraylistrtakerrandintrr)r"NZrandom_lettersrrrr df_letters2sr(cCs tdgS)Nr)rrrrr raw_frame@sr)opaxisskipnaTFsortc sdkr|}n|j}tkrx|jd|d}t|d}|jddfdd}|rj|jd}t||nT|jd|d}t|}|jddfdd}|r|jd}t||dS) Nr)levelr+r.)r-r/cst|dS)N)r+r-getattrhr+r*r-rrVz3test_regression_allowlist_methods..r+cst|dS)Nr8r1r3)r+r*rrr6^r7)TAGG_FUNCTIONS_WITH_SKIPNAgroupbyr2applyZ sort_indextmassert_frame_equal) r)r*r+r-r.frameZgroupedresultexpectedrr5r!test_regression_allowlist_methodsEs$    rBc Cs|}|j}ddddddddd d d d d g }ddt|D}|||D]}||fD]z}||j}dt|dt|jd}dt|jdt|d} |d| } tj t | dt ||W5QRXqTqHdS)NevalqueryabswheremaskZalignr;ZclipZastypeatcombineZ consolidateZconvert_objectscSsg|]}|dr|qS)to_ startswith).0methodrrr ws z*test_groupby_blocklist..z (?:^Cannot.+z.+'z!'.+try using the 'apply' method$)z(?:^'z' object has no attribute z$)|)match) r!dirextendr;r"reprtype__name__pytestZraisesAttributeErrorr2) r(rsZ blocklistZ to_methodsblobjgbZdefined_but_not_allowedZ not_definedmsgrrrtest_groupby_blocklistds8   r^cBCs|jdd}ddt|D}ddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFhB}||kstdS)GNsecondr0cSsh|]}|ds|qS_rK)rMvrrr s z&test_tab_completion..rrrZaggZ aggregater<ZboxplotfilterfirstZ get_groupgroupshistindiceslastr rr r ZngroupsZnthohlcZplotr sizerr Z transformrrcountZnuniqueheaddescribecummaxquantilerankcumprodtailresamplecumminfillnaZcumsumZcumcountZngroupallshiftrr% pct_changeanycorrZcorrwithZcovdtypesndimdiffidxmaxidxminffillbfillZrollingZ expandingpipesampleZewmZ value_counts)r;rRAssertionError)mframegrpresultsrArrrtest_tab_completions DrcCs2|jdd}dD]}t||}|j|kstqdS)Nr_r0)r r r r reri)r;r2rVr)rrnamefrrrtest_groupby_function_renames  rrNrlr{rorurrrnrqrpr~rxrwrzrrrrrycCsdtdt|d}||_|dgdg}|dg|d}t||}t||}t||dS)N2014Zperiodsrr)rlenindexr;r2r=r>)rrNrnggg_expresexprrr#test_groupby_selection_with_methodss  rcCstdt|d}d|j_||_|dgdg}|dg|d}t|d|dt|j |j t| dd| ddt| d  | d  t| d  | d  t|d d|d ddS) NrrrrrrcSs|S)N)r xrrrr6r7z6test_groupby_selection_other_methods..rcSs t|dkS)N)rrrrrr6r7)rrcolumnsrrr;r=r>rvr|r<rtrrjrd)rrrrrrr$test_groupby_selection_other_methodss    rcCs||jdddf}ddt|Dt|j}t|}|t8}|t8}|t8}tt@r`ttt@rlttt@rxt|rd|d}t|ttBtB}||krd||d}t|dS)NrcSsh|]}|ds|qSr`rK)rMrarrrrcs z/test_all_methods_categorized..z? There are uncategorized methods defined on the Grouper class: aG. Was a new method recently added? Every public method On Grouper must appear in exactly one the following three lists defined in pandas.core.groupby.base: - `reduction_kernels` - `transformation_kernels` - `groupby_other_methods` see the comments in pandas/core/groupby/base.py for guidance on how to fix this test. zI Some methods which are supposed to be on the Grouper class are missing: z. They're still defined in one of the lists that live in pandas/core/groupby/base.py. If you removed a method, you should update them ) r;ZilocrRsetrrrrr)rrnamesZ new_namesr]Zall_categorizedrrrtest_all_methods_categorizeds(    r) __doc__stringrZnumpyrrWZpandasrrrZpandas._testingZ_testingr=Zpandas.core.groupby.baserrrZ AGG_FUNCTIONSr:Zfixturerr(r)markZ parametrizerBr^rrrrrrrrrsp       ,J