o i) @sdZddlmZddlZddlZddlmZmZm Z ddl m Z ddl mZmZmZgdZddgZgd Zejed d d Zgd Zejed ddZejddZejddZejdeegddZddZddZddZejddZ ejdeejd dd!gejd"dd!gejd#d$d%gejd&d$d%gd'd(Z!d)d*Z"d+d,Z#d-d.Z$ejd/gd0d1d2Z%ej&d3d4d5Z'd6d7Z(d8d9Z)dS):zV test methods relating to generic function evaluation the so-called white/black lists )ascii_lowercaseN) DataFrameSeries date_range)groupby_other_methodsreduction_kernelstransformation_kernels) sumprodminmaxmedianmeanskewmadstdvarsemrr)quantilefillnartakeidxmaxidxmintshiftrplothistdtypescorrwithcorrcovdiff)paramscC|jSNparamrequestr(^/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/tests/groupby/test_allowlist.pydf_allowlist_fixture9r*)rrrrrrrrrrdtyperrr uniquenlargest nsmallestis_monotonic_increasingis_monotonic_decreasingcCr"r#r$r&r(r(r)s_allowlist_fixtureUr+r2cCs*tgdgdtjdtjddS)N)foobarr3r4r3r4r3r3)oner5twothreer6r6r5r7)ABCD)rnprandomrandnr(r(r(r)dfZs  r@cCsRttt}d}|tjdd|}t|dttj|t|d}|S)N r)floatsletters) r=arraylistrrr>randintrr)rDNrandom_lettersr@r(r(r) df_lettersfsrJ allowlistcCs@|}|tkr |}n|d}||j}t|t|jksJdS)NrC) df_allowlistgroupbyrDset_apply_allowlist)rJrKr@objgbr(r(r)test_groupby_allowlistts  rRcCsx||j}tt||}z|j}Wn tyYdSw||ks#Jz|j}Wn ty2YdSw||s:JdSr#)rMrDgetattrtype__name__AttributeError __qualname__endswith)rPr@mrQfnr(r(r)check_allowlists      r\cCs|}|}t|j||dSr#)r\rD)rJr2rYr@r(r(r)test_groupby_series_allowlistsr]cCs|}|}t|||dSr#)r\)rJr*rYr@r(r(r)test_groupby_frame_allowlistsr^cCs0|}tj|jdddgf<tj|jdddgf<|S)Nr)r=naniloc) multiindex_dataframe_random_datar@r(r(r) raw_framesreoplevelr_axisskipnaTFsortc CsV|dkrtnd}|dkr|}n|j}|tvrn|j|||d}tj|dddt|||d} Wdn1s9wYttt|||||d} Wdn1sXwY|rf| j||d } t| | dS|j|||d}ttt||} t||||d } Wdn1swY|r| j||d } t| | dS) Nrr)rgrhrjzThe 'mad' method is deprecatedF)matchraise_on_extra_warnings)ri)rgrhri)rhrg)rgrh) FutureWarningTAGG_FUNCTIONS_WITH_SKIPNArMtmassert_produces_warningrS sort_indexassert_frame_equal) rerfrgrhrirjwarnframegroupedresultexpectedr(r(r)!test_regression_allowlist_methodss2    ryc Cs|}|j}gd}ddt|D}|||D]N}||fD]G}||j}dt|dt|jd}dt|jdt|d } |d | } tj t | d  t ||Wdn1sawYqqdS) N) evalqueryabswheremaskalignrMclipastypeatcombine consolidateconvert_objectscSsg|] }|dr|qS)to_ startswith).0methodr(r(r) z*test_groupby_blocklist..z (?:^Cannot.+z.+'z!'.+try using the 'apply' method$)z(?:^'z' object has no attribute z$)|rk) rCdirextendrMrDreprrTrUpytestraisesrVrS) rJr@s blocklist to_methodsblrPrQdefined_but_not_allowed not_definedmsgr(r(r)test_groupby_blocklists&    rcCs6|jdd}ddt|D}hd}||ksJdS)NsecondrgcSh|] }|ds|qS_r)rvr(r(r) rz&test_tab_completion..>Fr9r:r;aggallanyrewmrr r nthpadrrr rrr headrlastrndimohlcpiperr ranksizertailrapplybfillcountffillfirstshiftcummaxcummincumsumrrfiltergroupsrrr ngroupsamplerboxplotcumprodindicesngroupsnuniquerollingbackfillrcumcountdescriberresample aggregate expanding get_group transform pct_change value_counts)rMr)mframegrpresultsrxr(r(r)test_tab_completions HrcCs2|jdd}dD]}t||}|j|ksJqdS)Nrr)r r r r rr)rMrSrU)rrnamerZr(r(r)test_groupby_function_renameKs  rr)rrrrrrrrr rrrrrrrrcCsdtdt|d}||_|dgdg}|dg|d}t||}t||}t||dS)N2014periodsr9r;)rlenindexrMrSrprs)r@rrnggg_expresexpr(r(r)#test_groupby_selection_with_methodsRs  rz)ignore:tshift is deprecated:FutureWarningcCsdtdt|d}||_|dgdg}tjtdd |WddS1s+wYdS)Nrrr9r;zFreq was not set in the indexr)rrrrMrr ValueErrorr)r@rrr(r(r)$test_groupby_selection_tshift_raisesxs  "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) Nrrr3r9r;rcSs|Sr#)r xr(r(r)sz6test_groupby_selection_other_methods..r<cSs t|dkS)N)rrr(r(r)rs )rrcolumnsrrrMrprsrrrrrrr)r@rrrr(r(r)$test_groupby_selection_other_methodss   rcCs||jdddf}ddt|Dt|j}t|}|t8}|t8}|t8}tt@r0Jtt@r6Jtt@r.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 ) rMrcrrNrrrrAssertionErrorprint)rrnames new_namesrall_categorizedr(r(r)test_all_methods_categorizeds.    r)*__doc__stringrnumpyr=rpandasrrrpandas._testing_testingrppandas.core.groupby.baserrr AGG_FUNCTIONSrorLfixturer* s_allowlistr2r@rJmark parametrizerRr\r]r^reryrrrrfilterwarningsrrrr(r(r(r)sX              ,N