gL i"<ddlZddlZddlmZmZddlmZdZ dZ y)N) DataFrameIndexctjjd}tgd|j d|j dd}d}d}|j dj |j |}tdd gd }tjd d gd |}tj||y)N)foobarrrrrrr)ABCc|jj|jjjz SN)r maxr min)dfgbs d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/groupby/test_pipe.pyfztest_pipe..fs*vvzz|dffjjl..000c |dzS)Nr)srss rsquareztest_pipe..squares Av rr rrnameg8t @gt @r )rindex) nprandom default_rngrstandard_normalgroupbypiperpdSeriestmassert_series_equal) random_statedfrrresultrexpecteds r test_piper* s99((+L I--a0--a0  B1ZZ_ ! !! $ ) )& 1F 5%.s +Eyy+->?cQVWH8V,rcBtgdgdgdd}d}d}d}|jdj|d j|d j|d }td d gd}t j ddg|}t j||t j gd}|j|jd}t j gdtgdtj}t j||y)N)r r r r r )g?@g@r,g@)g$@gY@g@@gYg@)groupxycR|jfdd}|jdS)Nc>|jjkDSr)r/mean)grparg1s rz+test_pipe_args..f..9s355::<$+>rF)dropnar-)filterr )rr4filtereds ` rrztest_pipe_args..f8s';;>u;M((rcf|j|jjz |zSr)sum)rarg2s rgztest_pipe_args..g<s%xxzDHHJNN,,t33rc:|j|jz|z Sr)r.r/)r'arg3s rhztest_pipe_args..h?sttbdd{T!!rr-r dr r rg#SgbeS)r)rBrrrCcD|j|jzSr)r:count)r3s rr5z test_pipe_args..Lsswwy399;/Fr)r )rBrrC)dtype) rr r!rr"r#r$r%rint64)r'rr<r?r(rr)sers rtest_pipe_argsrK,s .*7  B)4"ZZ % %a + 0 0B 7 < rPs -B$-r