df@ddlZddlZddlmZdZdZdZ dZ dZ dS)Nc<tjgdgddd}|dgdd}|dgdd }t j||dS) N)AliceBobCarl)r)nameager r F group_keysc|SNgroups b/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/groupby/test_apply_mutate.pyz$test_group_by_copy..sRWc*|Sr)copyrs rrz$test_group_by_copy..sejjllr)pd DataFrame set_indexgroupbyapplytmassert_frame_equal)dfgrp_by_same_value grp_by_copys rtest_group_by_copyr!s ,,,<<     i  E7u ==CCDWDWXX**eW*77==""K+[99999rc tjdgdzdgdzzdgdzdgdzzdgdzzd gdzzdgdzzdgdzzdgdzzd td d Dtjd dd}d}d}|d|}|d|}tj ||dS)Nabcdefcg|]}d|S)gr).0xs r z&test_mutate_groups..$s333WWW333rd)size)cat1cat2cat3valc|}|jd|d<|ddSNmin)methodrankr7)rr9r>rr<r/s rf_copyz"test_mutate_groups..f_copy)sK FFHHEJJeJ,,& yy  (,,...rc|jd|d<|ddSr;)r9r>rr<r?s r f_no_copyz%test_mutate_groups..f_no_copy.s>EJJeJ,,& yy  (,,...rr6) rrrangenprandomrandintrrrassert_series_equal)rr@rB grpby_copy grpby_no_copys rtest_mutate_groupsrJsB EAI )EAIeaieaieaieai  eai  eai 43eArll3339$$Sr$22   B/// ///F##))&11JJJv&&,,Y77M:}55555rc(tjgdtdd}|ddd}|ddd}t j||dS) N) r1r1r1r(r(r(rLrL )keyvaluerNTr c |ddjSrrNr?s rrz/test_no_mutate_but_looks_like..=s111rc|jSrrQr?s rrz/test_no_mutate_but_looks_like..>sr)rrrCrrrrG)rresult1result2s rtest_no_mutate_but_looks_likerU7s 999E!HHMM N NBjj4j00667I7IJJGjj4j0066GGG7G,,,,,rc2tjgdgdd}d}|dgd|}tjgdtjgd d }tj||dS) N)ArWrWBrXrX)r1r(rLr&)col1col2c>d|j|jddf<|jS)Nrr\)locindexr\r?s rfnz-test_apply_function_with_indexing..fnHs %&agbk6!"v rr[F)as_index)r1r(rrYrZr))rr)rr1)rr()r1rL)r1rY)r1rZr\)r`r ) rrrrSeries MultiIndex from_tuplesrrG)rraresultexpecteds r!test_apply_function_with_indexingrhBs ///9K9K9KLL  BZZ5Z 1 1 7 7 ; ;Fym'' < < <   H68,,,,,rc tjgdgdgdgdgdgddtjgd}d}|dd|}tjgdgd gd gtjgd }t j||dS) N)r1r(rL)CjulianrXgeoffreyrWrlrXrlrWrnrkrn)rorqrprmrjrr)columnscb|jdd}|d|d|f<|S)Nrr1)axissum)rsrv)groupedr s r add_columnz8test_apply_mutate_columns_multiindex..add_columnns4q!!$&{{{22t rr1)levelru)r1r1r1rLr1r1r1rL)r(r(r(r&r(r(r(r&)rLrLrLrMrLrLrLrM))rnrWrn)rnrXrn)rnrkrn)rnrvrn)rlrWrl)rlrXrl)rlrkrl)rlrvrl)rrrdrerrrr)rrxrfrgs r$test_apply_mutate_columns_multiindexrzWs# &YY(yy&YY&YY(yy(yy    ))         B* ZZaaZ ( ( . .z : :F| $ $ $ $ $ $     ))      H8&(+++++r) numpyrDpandasrpandas._testing_testingrr!rJrUrhrzrrrrs : : : 666@------*9,9,9,9,9,r