o d@sHdZddlZddlZddlmZmZddlmZmZddZ ddZ dS) zY Test the consistency of the groupby API, both internally and with other pandas objects. N) DataFrameSeries)DataFrameGroupBy SeriesGroupByc Cs|dvrd}|jtjj|d|dvr"d}|jtjj|d|dkr/tt|r-JdStt|}tt|}t t |j }|dkrIdh}nt t |j }t t }}|d vrahd }nl|d vrjd d h}nc|dvrrd h}n[|dvrhd}hd}nN|dvrhd}ddh}nA|dvrhd}n8|dvrhd}n/|dvrddh}d h}n#|dvrdh}n|dvrdh}d h}n|dvrd h}n|dvrd d h}||@|ksJ||@|ksJ||8}||8}||ksJdS)!Nfirstlastz?first and last are entirely different between frame and groupbyreason)cumcountzDataFrame has no such methodngroupsizeselfanyall>kwargs bool_onlyaxis)count numeric_onlyr)Znuniquemaxmin>skipnarrZ min_count engine_kwargsenginemeanZstdsumvarrrZmedianprodZsemZbackfillZbfillZffillpadZdowncastrZinplaceZcummaxZcumminrargsZcumprodZcumsumZ pct_changerZrankZquantilemethod) node add_markerpytestmarkxfailhasattrrgetattrrsetinspect signature parameters) request groupby_funcmsgZ frame_method gb_methodresultexpectedexclude_expectedexclude_resultrA]/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/groupby/test_api_consistency.pytest_frame_consistencysb        rCc Cs|dvrd}|jtjj|d|dvr"d}|jtjj|d|dkr/tt|r-JdStt|}tt|}t t |j }|dkrIdh}nt t |j }t t }}|d vrahd }nk|d vrid h}nc|d vrvhd}hd}nV|dvrhd}ddh}nI|dvrhd}n@|dvrhd}n7|dvrddh}dh}n+|dvrdh}n#|dvrdh}d h}n|dvrdh}n|dvrddh}n|dvrdh}||@|ksJ||@|ksJ||8}||8}||ksJdS) Nrz@first and last are entirely different between Series and groupbyr )r ZcorrwithzSeries has no such methodr r rr>rrr)diffrr>rrrrrrrr"r$r&r'rr(rr)r*rr+)ZidxminZidxmaxr,) r.r/r0r1r2r3rr4rr5r6r7r8) r9r:r;Z series_methodr<r=r>r?r@rArArBtest_series_consistencyRsb        rE) __doc__r6r0ZpandasrrZpandas.core.groupby.genericrrrCrErArArArBs ?