OwgI <ddlZddlZddlmZmZmZddlmZ dZ dZ dZ ejjdgdgdfdgdgfgejjd dge jejjd d d gd Zy)N) MultiIndexSeries date_rangectgd}ttd}|j|}|jd}tgdt j tdgdg}t j||tgd}|j|}tgd t j tdgd g}t j|jdd |y) N  r aaaaabbbbbr )r r r rr raaabbb)r r rr r index rrr r rr r r rr)r r rr r r )r r rrr r lastkeep)rlistgroupbynlargestr from_arraystmassert_series_equalabgbres k/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_nlargest_nsmallest.py test_nlargestr' s./At%&'A 1B AA$$d8n6H%IJ A1a -.A 1B$$d8n6H%IJ A2;;qv;6:c`tjjd}tdd}|ddgg}t j |ddg }t |jd | }|jdjd }t j|d |d df|d |d df|d|ddf|d|ddf|d|ddf|d|ddf|d|ddf|d|ddf|d|ddf|d|ddfg gd }gd}t || }tj||ddy)Nr 20180101r)periodsonetwofirstsecond)namesrrrr rr rr rr )r.r.r/) g.2?gooڿg?g?gSL [?g ;H?gjs տg1Au?g f+Qs?g;?FgMbP?) check_exactrtol) nprandom default_rngrr from_productrstandard_normalrr from_tuplesrr) nprdts iterablesidxsresultexp_idx exp_valuesexpecteds r&test_nlargest_mi_grouperrC sn ))   "C Z ,Cuen%I  ! !)GX3F GCs""2&c2A YYw  ( ( +F$$ VSVU # VSVU # VSVU # VSVU # VSVU # VSVU # VSVU # VSVU # VSVU # VSVU # +G Jj0H68TJr(ctgd}ttd}|j|}|jd}tgdt j tdgdg}t j||tgd}|j|}tgd t j tdgd g}t j|jdd |y) Nrrr )rr r rrrr)rrrrr rrr)rrrrrr )rrrr rr rr)rrr nsmallestrrrrr s r&test_nsmallestrFMs./At%&'A 1B QA$$d8n6H%IJ A1a -.A 1B$$d8n6H%IJ A2<<<7;r(z data, groups)rrr r )rrrrdtypemethodrrEc|tj||}|dk(rtt|}t |d}t |j ||d}t|trtj|tn|}t |tj||jgd}tj||y)N)rGrr!)namer )n)rrJ)r4arrayrreversedrgetattrr isinstanceintrrrrr)datagroupsrGrHserr?expidxrBs r&test_nlargest_and_smallest_nooprUas xxE* HTN# C C 1WS[[(& 1A 6F,6vt,DRXXfC (&Fd*"8"8&#))9L"MTWXH68,r()numpyr4pytestpandasrrrpandas._testing_testingrr'rCrFmark parametrizeALL_INT_NUMPY_DTYPESrUr(r&r_s  ;(*KZ<(L!QC!:.4"B"*A*A"BCJ #<= ->D  -r(