U
|
ý°df ã @ s d dl Z d dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z ej dddg¡ej dddg¡ej d d
|
d d gd
|
d d gd d d gdddgdd d gd d d gdddgdddgdddgdddgdddgdddgejejejgg ¡dd Zdd Zej dddg¡dd Zej dddg¡ej dddg¡ej d dddgdddgejejejgdejdgdejdgdejdgg¡d d Zej d d d!ejdged"ejejgd#d"fd$d ejejgd#d"ejdged"fd$d$ejejgd#d"ejejgd#d"fd!d#ejejgd#d"ejejgd#d"fg¡d%d& Zej dddg¡ej d'd$d!d#g¡ej dddg¡d(d) Zej d*dejejgdfdejdejgdfdejejgdfdejdejgdfg¡d+d, Zej dddg¡d-d. Zej dddg¡d/d0 Z dS )1é N)Ú DataFrameÚIndexÚSeriesÚisnaÚagg_funcÚanyÚallÚskipnaTFÚvalsÚfooÚbarÚbazÚ é é é ç ð?g @g @g c C s t dgd dgd |d d}tt| |}|rLtt|rL| dkrLd}t |gd dgtddgd d
|
d }t| d ¡| |d }t ||¡ d S ) NÚar Úbr )ÚkeyÚvalr Fr r )Úname©ÚcolumnsÚindex©r ) r ÚgetattrÚbuiltinsr r r ÚgroupbyÚtmÚassert_frame_equal)r r r
|
ÚdfÚexpZexp_dfÚresult© r$ úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_any_all.pyÚtest_groupby_bool_aggs s ""r& c C sv t dddgdtjdgdtjdggddd gd
|
} t d d gd d ggdd gddgd }d|j_| d¡ ¡ }t ||¡ d S )Nr r r r r r ÚAÚBÚC©r TFr ) r ÚnpÚnanr r r r r r )r! Úexpectedr# r$ r$ r% Útest_any3 s þ ÿr. Ú bool_agg_funcc C sP t ddggddgd}| dg¡}t|| }| t dg¡¡}t ||¡ d S )NTr r* r )r r r Zset_axisr+ Úarrayr r )r/ r! Zgrp_byr# r- r$ r$ r% Ú test_bool_aggs_dup_column_labels@ s
|
r1 Údatac C s\ t |dd}t|| |d}t |t dg¡dd}| dddg¡j| |d}t ||¡ d S )NÚboolean©Údtyper r ©r r5 )r r r+ r0 r Úaggr Zassert_series_equal)r/ r r2 ÚserZ expected_datar- r# r$ r$ r% Útest_masked_kleene_logicK s
|
r9 z dtype1,dtype2,exp_col1,exp_col2ÚfloatZFloat64r4 r3 ZInt64c C sn dt jg}ttj|| dtj||dd}| ddg¡jddd}t||dt dg¡d}t ||¡ d S ) Nr r4 )Zcol1Zcol2r r Fr )r ) r+ r, r Úpdr0 r r7 r r )Zdtype1Zdtype2Zexp_col1Zexp_col2r2 r! r# r- r$ r$ r% Útest_masked_mixed_typese s
|
ÿr< r5 c C sh |t jdg|d}d}|s(| dkr(t j}||gt dg¡dd}| ddg¡j| |d}t ||¡ d S )Nr r4 Tr r3 r6 r )r; ÚNAr+ r0 r r7 r Ú assert_equal)r/ r5 r Úframe_or_seriesÚobjÚ expected_resr- r# r$ r$ r% Ú test_masked_bool_aggs_skipna s rB z bool_agg_func,data,expected_resc C sN ||t d}| dgt| ¡ | ¡}||gt dg¡dd}t ||¡ d S )Nr4 r Úboolr6 )Úobjectr Úlenr7 r+ r0 r r> )r/ r2 rA r? r@ r# r- r$ r$ r% Ú test_object_type_missing_vals s rF c C sD t tjgtd}tjtdd | dg¡j| dd W 5 Q R X d S )Nr4 z boolean value of NA is ambiguous)Úmatchr Fr ) r r; r= rD ÚpytestZraisesÚ TypeErrorr r7 )r/ r8 r$ r$ r% Ú'test_object_NA_raises_with_skipna_false® s rJ c C sb | t krddginddi}| f |dti}t| |j¡| }| f |dti}t ||¡ d S )Nr r r r5 )r rD r r r rC r r> )r? r/ Úkwargsr@ r# r- r$ r$ r% Ú
|
test_empty¶ s
|
rL ) r Únumpyr+ rH Zpandasr; r r r r Zpandas._testingZ_testingr ÚmarkZ parametrizer, r& r. r1 r= r9 r0 rC r< rB ZNaTrF rJ rL r$ r$ r$ r% Ú<module> s¦ óþ
|
|
|
|
|
úþ üüüüíþ
|
üþ
|
|
|
