U b(@sdZddlmZddlZddlZddlmZmZm Z ddl m Z ddl mZmZmZdddd d d d d dddg Zd d gZddd ddddd dddddddgZejeddd Zddd ddddd ddd!dddd"d#d$d%d&gZejedd'd(Zejd)d*Zejd+d,Zejd-eegd.d/Zd0d1Zd2d3Zd4d5Zejd6d7Z ejd8eejd9dd:gejd;dd:gejdgejd?d=d>gd@dAZ!dBdCZ"dDdEZ#dFdGZ$ejdHdIddJdKdLdMdNdddOdPdQdddRdSdTgdUdVZ%ej&dWdXdYZ'dZd[Z(d\d]Z)dS)^zV test methods relating to generic function evaluation the so-called white/black lists )ascii_lowercaseN) DataFrameSeries date_range)groupby_other_methodsreduction_kernelstransformation_kernelssumprodminmaxmedianmeanskewmadstdvarsemquantilefillnatakeidxmaxidxmintshiftplothistdtypescorrwithcorrcovdiff)paramscCs|jSNparamrequestr'G/tmp/pip-unpacked-wheel-ck39h295/pandas/tests/groupby/test_allowlist.pydf_allowlist_fixture9sr)Zdtypeuniquenlargest nsmallestZis_monotonic_increasingZis_monotonic_decreasingcCs|jSr"r#r%r'r'r(s_allowlist_fixtureUsr-c CsBtddddddddgddddddddgtjdtjddS)NfoobarZoneZtwoZthree)ABCD)rnprandomZrandnr'r'r'r(dfZs  r7cCsRttt}d}|tjdd|}t|dttj|t|d}|S)N r)floatsletters) r5arraylistrrr6randintrr)r;NZrandom_lettersr7r'r'r( df_lettersfsr@ allowlistcCs@|}|tkr|}n|d}||j}t|t|jkss z*test_groupby_blocklist..z (?:^Cannot.+z.+'z!'.+try using the 'apply' method$)z(?:^'z' object has no attribute z$)|match) r:dirextendrCr;reprrJrKpytestraisesrLrI) r@r7sZ blocklistZ to_methodsblrFrGZdefined_but_not_allowedZ not_definedmsgr'r'r(test_groupby_blocklists8   rcFCs|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@dAdBdCdDdEdFdGdHdIdJhF}||kstdS)KNsecondr\cSsh|]}|ds|qS_rq)rsvr'r'r( s z&test_tab_completion..r1r2r3ZaggZ aggregateapplyZboxplotfilterfirstZ get_groupgroupsrindiceslastr rr r ZngroupsZnthohlcrr sizerr Z transformrrcountZnuniqueheaddescribecummaxrrankcumprodtailresamplecumminrZcumsumZcumcountZngroupallshiftrrr pct_changeanyrrrrrndimr rrffillbfillpadZbackfillZrollingZ expandingpipesampleZewmZ value_counts)rCryrE)mframegrpresultsrgr'r'r(test_tab_completions HrcCs2|jdd}dD]}t||}|j|kstqdS)Nrr)r r r r rr)rCrIrKrE)rrnamerPr'r'r(test_groupby_function_renameGs  rrtrrrrrrrrrrrrcCsdtdt|d}||_|dgdg}|dg|d}t||}t||}t||dS)N2014Zperiodsr1r3)rlenindexrCrIrbrd)r7rtrnggg_expresexpr'r'r(#test_groupby_selection_with_methodsNs  rz)ignore:tshift is deprecated:FutureWarningc CsNtdt|d}||_|dgdg}tjtdd|W5QRXdS)Nrrr1r3zFreq was not set in the indexrw)rrrrCr|r} ValueErrorr)r7rrr'r'r($test_groupby_selection_tshift_raisests 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) Nrrr.r1r3rcSs|Sr")r xr'r'r(z6test_groupby_selection_other_methods..r4cSs t|dkS)N)rrr'r'r(rr)rrcolumnsrrrCrbrdrrrrrrr)r7rrrr'r'r($test_groupby_selection_other_methodss    rcCs||jdddf}ddt|Dt|j}t|}|t8}|t8}|t8}tt@r`ttt@rlttt@rxt|rd|d}t|ttBtB}t |t |||ksd||d}t|dS)NrcSsh|]}|ds|qSrrq)rsrr'r'r(rs z/test_all_methods_categorized..z? There are uncatgeorized 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 ) rCrYryrDrrrrrEprint)rrnamesZ new_namesrZall_categorizedr'r'r(test_all_methods_categorizeds,    r)*__doc__stringrZnumpyr5r|ZpandasrrrZpandas._testingZ_testingrbZpandas.core.groupby.baserrrZ AGG_FUNCTIONSrarBZfixturer)Z s_allowlistr-r7r@markZ parametrizerHrRrSrTrZrhrrrrfilterwarningsrrrr'r'r'r(s            ,N