o[we@ddlZddlZddlmZdZdZdZ dZ dZ dS)Nc<tjgdgddd}|dgdd}|dgdd }t j||dS) N)AliceBobCarl)r)nameager r F group_keysc|SNgroups Flib/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 Dtjdd 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) rrrangenprandom default_rngintegersrrrassert_series_equal)rr@rB grpby_copy grpby_no_copys rtest_mutate_groupsrKsP EAI )EAIeaieaieaieai  eai  eai 43eArll3339((++44Sr4BB   B/// ///F##))&11JJJv&&,,Y77M:}55555rc(tjgdtdd}|ddd}|ddd}t j||dS) N) r1r1r1r(r(r(rMrM )keyvaluerOTr c |ddjSrrOr?s rrz/test_no_mutate_but_looks_like..=s111rc|jSrrRr?s rrz/test_no_mutate_but_looks_like..>sr)rrrCrrrrH)rresult1result2s rtest_no_mutate_but_looks_likerV7s 999E!HHMM N NBjj4j00667I7IJJGjj4j0066GGG7G,,,,,rc2tjgdgdd}d}|dgd|}tjgdtjgd d }tj||dS) N)ArXrXBrYrY)r1r(rMr&)col1col2c>d|j|jddf<|jS)Nrr])locindexr]r?s rfnz-test_apply_function_with_indexing..fnHs %&agbk6!"v rr\F)as_index)r1r(rrZr[r))rr)rr1)rr()r1rM)r1rZ)r1r[r])rar ) rrrrSeries MultiIndex from_tuplesrrH)rrbresultexpecteds r!test_apply_function_with_indexingriBs ///9K9K9KLL  BZZ5Z 1 1 7 7 ; ;Fym'' < < <   H68,,,,,rc tjgdgdgdgdgdgddtjgd}d}d}t jt |5|dd }dddn #1swxYwY||}tjgd gd gd gtjgd }t j ||dS)N)r1r(rM)CjulianrYgeoffreyrXrmrYrmrXrorlro)rprrrqrnrkrs)columnscb|jdd}|d|d|f<|S)Nrr1)axissum)rtrw)groupedr s r add_columnz8test_apply_mutate_columns_multiindex..add_columnns4q!!$&{{{22t rz+DataFrame.groupby with axis=1 is deprecated)matchr1)levelrv)r1r1r1rMr1r1r1rM)r(r(r(r&r(r(r(r&)rMrMrMrNrMrMrMrN))rorXro)rorYro)rorlro)rorwro)rmrXrm)rmrYrm)rmrlrm)rmrwrm) rrrerfrassert_produces_warning FutureWarningrrr)rrymsggbrgrhs r$test_apply_mutate_columns_multiindexrWs &YY(yy&YY&YY(yy(yy    ))         B* 8C #M = = =)) ZZaaZ ( ())))))))))))))) XXj ! !F| $ $ $ $ $ $     ))      H8&(+++++s)B  BB) numpyrDpandasrpandas._testing_testingrr!rKrVrirrrrrs : : : 666@------*<,<,<,<,<,r