U
|
ý°d] ã @ s( d dl Z d dlZd dlmZ e j dg dfdddgdfdd d
|
gd fd d dgdfd d dgdfdddgdfdddgdfdddgdfdddgdfdd dgdfe d d¡e dd¡e dd¡gdfg ¡e j ddd d gdddgd d d gg¡e j d e ej
|
ej ej g¡e j d dd g¡d d Z d!d" ZdS )#é Nz values, dtypeÚobjecté é é Úint64g ð?g @g @Úfloat64ÚaÚbÚcÚstringzdatetime64[ns]zdatetime64[ns, CET]ztimedelta64[ns]Ú2000Z2001Z2002z Period[D]ZSparseé ÚintervalÚmaskTFÚ indexer_classÚframec
|
C s* dddgd t |
}|d t |
}tj| ||d}|r^t | dkrVtj||d}n| ¡ }|tjkrxtj|dd}n&|tjkrtj||jdd}n||}|| }|| }t ||¡ |tjkrúd } t j
|
t | d
|
|j | }t ||¡ W 5 Q R X n|j | }t ||¡ |j | }t ||¡ d S ) Nr r r
|
)ÚdtypeÚindexr Úboolean©r )r r zBiLocation based boolean indexing cannot use an indexable as a mask)Úmatch)ÚlenÚpdÚSeriesZ DataFrameZto_frameÚarrayr ÚtmÚ assert_equalÚpytestZraisesÚ
|
ValueErrorZilocÚloc)
|
Úvaluesr r r r r ÚobjÚexpectedÚresultÚmsg© r% ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/test_na_indexing.pyÚtest_series_mask_boolean s0
|
|
|
|
|
|
r' c C sP | dddg}t jddd gdd}||| }||| d¡ }t ||¡ d S )Nr r r TFr r )r r Zfillnar r )Zframe_or_seriesZ indexer_slir! r r# r" r% r% r& Útest_na_treated_as_falseB s
|
r( )r Zpandasr Zpandas._testingZ_testingr ÚmarkZ parametrizeZIntervalÚlistr ZIndexr r' r( r% r% r% r& Ú<module> s2 $õþ ÿ&
|
