U
|
®ý°dÓB ã @ sf d dl m Z d dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z d dl mZ G dd dZdS )é )ÚdatetimeN)Ú DataFrameÚIndexÚSeriesÚ bdate_range)Úopsc @ s, e Zd Zej dejejej g¡dd Z
|
dd Z dd Z dd Z d
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zej d ejejej g¡d d Zd d Zej d ejejg¡d!d" Z ej d#eje d$d$gfeje d%d%gfej e d%d%gfg¡d&d' Z d(d) Z d*d+ Z!d,S )-ÚTestSeriesLogicalOpsÚbool_opc C s t tdddtd}tj|d d d
< | ¡ }| |d ¡}|||d k ||d k}|||d k ||d k}d ||< t ||¡ d S )
|
Nz1/1/2000é
|
)Zperiods©Zdtypeé r é é F) r r ÚobjectÚnpÚnanZisnaZfillnaÚtmÚassert_series_equal)Úselfr ÚserÚmaskZfilledÚresultÚexpected© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/test_logical_ops.pyÚ test_bool_operators_with_nas s z1TestSeriesLogicalOps.test_bool_operators_with_nasc C sl t d}tdddg|d}tdddg|d}tg td}||@ }|}t ||¡ ||B }|}t ||¡ d S )NÚbcaTF©Úindexr )Úlistr r r r )r r Ús_tftÚs_fffZs_emptyÚresr r r r Ú,test_logical_operators_bool_dtype_with_empty! s zATestSeriesLogicalOps.test_logical_operators_bool_dtype_with_emptyc C sì t tddd}t dgd }t dgd }||@ }t tddd}t ||¡ ||B }t tdddd}t ||¡ t dgd dd}||@ }t ddddgdd}t ||¡ | tj¡| tj¡B }t ddddgd d}t ||¡ d S )
|
Né Úint64r r é é Úint8r Úint32)r Úranger r Zastyper Úint16r) )r Ús_0123Zs_3333Zs_4444r" r Ús_1111r r r Ú/test_logical_operators_int_dtype_with_int_dtype1 s zDTestSeriesLogicalOps.test_logical_operators_int_dtype_with_int_dtypec C sZ t tddd}|d@ }t dgd }t ||¡ |d@ }t ddddg}t ||¡ d S )Nr$ r% r r r' ©r r* r r )r r, r" r r r r Ú0test_logical_operators_int_dtype_with_int_scalarI s zETestSeriesLogicalOps.test_logical_operators_int_dtype_with_int_scalarc C sê t tddd}d}tjt|d |tj@ W 5 Q R X tjt|d |d@ W 5 Q R X d}tjt|d |dddd g@ W 5 Q R X tjt|d |t dddd g¡@ W 5 Q R X tjt|d |t ddd
|
d g@ W 5 Q R X d S ) Nr$ r% r ú7Cannot perform.+with a dtyped.+array and scalar of type©Úmatchg
ëQ¸ @z unsupported operand type.+for &:g¹?r g
ëQ¸ À)r r* ÚpytestÚraisesÚ TypeErrorr ÚNaNÚarray)r r, Úmsgr r r Ú+test_logical_operators_int_dtype_with_floatU s z@TestSeriesLogicalOps.test_logical_operators_int_dtype_with_floatc C sf t dgd dd}d}tjt|d |d@ W 5 Q R X tjtdd |dd d
|
d g@ W 5 Q R X d S ) Nr' r$ r( r z=Cannot perform 'and_' with a dtyped.+array and scalar of typer2 Úazunsupported operand.+for &ÚbÚcÚd)r r4 r5 r6 )r r- r9 r r r Ú)test_logical_operators_int_dtype_with_strf s z>TestSeriesLogicalOps.test_logical_operators_int_dtype_with_strc C s t tddd}t dgd }|d@ }t ||¡ |dg@ }t ||¡ |d@ }t ||¡ |dA }t ddddg}t ||¡ d S )Nr$ r% r F)FTr/ )r r, r r r r r Ú*test_logical_operators_int_dtype_with_booln s
|
z?TestSeriesLogicalOps.test_logical_operators_int_dtype_with_boolc C sx t tddd}|t dtjddg@ }t dgd }t ||¡ t ddtjdg}tjtdd ||@ W 5 Q R X d S )
|
Nr$ r% r Fr; r< r> z unsupported.* 'int' and 'str'r2 ) r r* r r7 r r r4 r5 r6 )r r, r r Zs_abNdr r r Ú,test_logical_operators_int_dtype_with_object s zATestSeriesLogicalOps.test_logical_operators_int_dtype_with_objectc C s` t d}tdddg|d}tdddg|d}|d@ }|}t ||¡ |d@ }|}t ||¡ d S )Nr TFr r r' )r r r r )r r r r! r" r r r r Ú*test_logical_operators_bool_dtype_with_int s z?TestSeriesLogicalOps.test_logical_operators_bool_dtype_with_intc C sv t dddddg}ddd dtjg}t dddddg}||@ }t ||¡ |t |¡@ }t ||¡ |t|@ }t ||¡ |t |@ }t ||¡ t dddddg}||B }t ||¡ |t |¡B }t ||¡ |t|B }t ||¡ |t |B }t ||¡ t dddddg}||A }t ||¡ |t |¡A }t ||¡ |t|A }t ||¡ |t |A }t ||¡ d S ©NTF)r r r r r r8 r )r ÚleftÚrightr r r r r Ú(test_logical_ops_bool_dtype_with_ndarray s: z=TestSeriesLogicalOps.test_logical_ops_bool_dtype_with_ndarrayc C s t d}tdddg|d}tdddg|d}tdddg|d}ttddd}tdgd d d
|
d d d ddgd}||@ }t ||¡ tdgd d d
|
d d d ddgd}||@ }t ||¡ td
|
gt d}||@ }| t d¡}t ||¡ ||B }| t d¡}t ||¡ d S )Nr TFr r$ r% r é r r' r r r; r< r= Úabc)r r r* r r Zreindex) r r r Zs_tffr, r r Zs_a0b1c0r" r r r Ú<test_logical_operators_int_dtype_with_bool_dtype_and_reindex¾ s$ zQTestSeriesLogicalOps.test_logical_operators_int_dtype_with_bool_dtype_and_reindexc C s² t ddddddddd g }d
|
}tjt|d |td d d @ W 5 Q R X t ddddddddtd d d g }tj|d d d
< t d|jd}d|d d d
< |t|@ }t
|
||¡ d S )Nr r r$ é é rG r& r r
|
r1 r2 éÕ r' Tr F) r r4 r5 r6 r r r r r r r )r Úsr9 r r r r r Ú"test_scalar_na_logical_ops_cornersÞ s " z7TestSeriesLogicalOps.test_scalar_na_logical_ops_cornersc C s t ddddddddtd d
|
d
|
g }tj|d d d
< td |i}td tdd gttd d }||@ }t ||¡ ||@ }t ||¡ d S )Nr r r$ rJ rK rG r& r rL r' ÚAF)r Úcolumns) r r r r r r* r r Úassert_frame_equal)r rM r> r r r r r Ú)test_scalar_na_logical_ops_corners_alignsí s " z>TestSeriesLogicalOps.test_scalar_na_logical_ops_corners_alignsÚopc s¨ t ddddgtddddg tddddgt fddttD } }t ||¡ t fddttD td}}t ||¡ d S ) NTFr' r c s g | ]}| | qS r r ©Ú.0Ún)Úidx1rS r r r Ú
|
<listcomp> s zDTestSeriesLogicalOps.test_logical_ops_with_index.<locals>.<listcomp>c s g | ]}| | qS r r rT )Úidx2rS r r r rX s r )r r r* Úlenr r Úbool)r rS r r r )rW rY rS r r Útest_logical_ops_with_indexú s "
|
&
|
z0TestSeriesLogicalOps.test_logical_ops_with_indexc C sp t ddddg}tddddgtd}tddddg}t ddddg}||A }t ||¡ ||A }t ||¡ d S )NTFr r' r )r r r r r )r r rW rY r r r r r Ú+test_reversed_xor_with_index_returns_series s
|
ÿ z@TestSeriesLogicalOps.test_reversed_xor_with_index_returns_seriesc C s t ddddg}tddddg}tddddg}t ||j|j}|||}t ||¡ ||t |}|||}t ||¡ d S )NTFr' r )r r Úvaluesr r )r rS r rW rY r r r r r Ú2test_reversed_logical_op_with_index_returns_series s
|
|
zGTestSeriesLogicalOps.test_reversed_logical_op_with_index_returns_seriesz op, expectedFTc C s2 t ddg}tddg}|||}t ||¡ d S rC )r r r r )r rS r r Úidxr r r r Útest_reverse_ops_with_index/ s
|
z0TestSeriesLogicalOps.test_reverse_ops_with_indexc
|
C sô t dddgtd}t dddgtd}t dddgtd}||@ }t ||¡ t dddgtd}||B }t ||¡ t dddgtd}||A }t ||¡ t dddgtd}t ddddgtd}t ddddgtd}||@ }t ||¡ t ddddgtd}||B }t ||¡ t g td}|| ¡ @ }t dddgtd}t ||¡ || ¡ B }t dddgtd}t ||¡ |t dgdg@ }t ddddgtd }t ||¡ |t dgdgB }t ddddgtd }t ||¡ | ¡ t dgdgt tj|jt tj|jfD ]"}|||B }t ||| ¡ qt dgfD ]"}|||B }t ||| ¡ qHtd}t dddg}d
|
D ]:} t dddg|d | B }t dddg|d }t ||¡ qd }
|
tjd fD ](} t j
|
t |
|
d || B W 5 Q R X qÐdD ]:} t dddg|d | B }t dddg|d }t ||¡ qþdD ]:} t dddg|d | @ }t dddg|d }t ||¡ q>dD ]:} t dddg|d | @ }t dddg|d }t ||¡ q~d }
|
tjfD ](} t j
|
t |
|
d || @ W 5 Q R X qÆd S )NTFr rH Zabcdr r' ÚzZabcz)Tr' r r r1 Zfoor2 )Fr )Tr' ) r r r r r Úcopyr r r r4 r5 r6 ) r r; r< r r ÚemptyÚer ÚtÚvr9 r r r Ú test_logical_ops_label_basedA s ü z1TestSeriesLogicalOps.test_logical_ops_label_basedc C s t dddgtddd}t dddgtddd}t ddddgtddd}t ||@ |¡ t ||@ |¡ t ddddgtddd}t ||B |¡ t ddddgtddd}t ||B |¡ t | ¡ | ¡ @ | ¡ ¡ t | ¡ | ¡ @ | ¡ ¡ tdddtjtjgitdd}t | ¡ | ¡ B | ¡ ¡ t | ¡ | ¡ B | ¡ ¡ t dddgtddd}t ddddgtddd}t ddddgtddd}t ||@ |¡ t ||@ |¡ t ddddgtddd}t ||B |¡ t ddddgtddd}t ||B |¡ t | ¡ | ¡ @ | ¡ ¡ t | ¡ | ¡ @ | ¡ ¡ t | ¡ | ¡ B | ¡ ¡ t | ¡ | ¡ B | ¡ ¡ d S ) NTFÚABCÚx)r ÚnameZABDÚABCDr ) r r r r rQ Zto_framer r r )r Ús1Ús2ÚexpZexp_or1Zexp_orZs3Zs4r r r Útest_logical_ops_df_compat¦ s6 z/TestSeriesLogicalOps.test_logical_ops_df_compatN)"Ú__name__Ú
|
__module__Ú __qualname__r4 ÚmarkZ parametrizeÚoperatorÚand_Úor_Úxorr r# r. r0 r: r? r@ rA rB rF rI rN rR r\ r] r Zrand_Zror_r_ r Zrxorra rh rp r r r r r sD
|
#
|
þþ
|
ýþ
|
|
er )r ru Únumpyr r4 Zpandasr r r r Zpandas._testingZ_testingr Z pandas.corer r r r r r Ú<module> s
|
