U
|
¬ý°dÀ1 ã @ s´ d dl Zd dlZd dlZd dlmZ d dlm Z d dl
|
m Z dd Z dd Z dd Zd
|
d Zej d d ddejejgd ddddgfd ejgd dgfd ejgd dgfejejgddgfejejejgedddgfg¡dd Zej dddgddgddgddgejddd e dg¡e ddg¡e ddg¡ejd digg ¡d d Zd d! Zd"d# Zd$d% Zd&d' Z d(d) Z d*d+ Z d,d- Z d.d/ Z ejjd0d dgd1d2gd3d4d5 Z!d6d7 Z"dS )8é N)Ú BooleanArray)Úcoerce_to_arrayc C sR t jddddgdd} t jddddgdd}t| |}tjdddd gdd}t ||¡ tjtdd t| ¡ | W 5 Q R X tjtdd t| | ¡ W 5 Q R X tjtdd t|
|
t ¡| W 5 Q R X tjtdd t| d W 5 Q R X tjt d d t| d
|
d ¡| W 5 Q R X tjt d d t| | d
|
d ¡ W 5 Q R X d S ) NTFÚbool©ÚdtypeÚbooleanz$values should be boolean numpy array©Úmatchz"mask should be boolean numpy arrayz"values.shape must match mask.shapeé éÿÿÿÿ)ÚnpÚarrayr ÚpdÚtmÚ assert_extension_array_equalÚpytestÚraisesÚ TypeErrorÚtolistZastypeÚintÚ
|
ValueErrorÚreshape©ÚvaluesÚmaskÚresultÚexpected© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/boolean/test_construction.pyÚ test_boolean_array_constructor
|
s"
|
r c C s t jddddgdd} t jddddgdd}t| |}|j| ksDt|j|ksRtt| |dd}|j| k snt|j|k s|td S )NTFr r )Úcopy)r r r Ú_dataÚAssertionErrorÚ_mask)r r r r r r Ú#test_boolean_array_constructor_copy% s
|
r$ c C s t t dddg¡t dddg¡} tjdddgdd}t || ¡ tjt dddg¡dd}t || ¡ tjtjdddgtddd}t || ¡ t t dddg¡t dddg¡} tjddd gdd}t || ¡ tjtjddd gtddd}t || ¡ d S )NTFr r ©r r r r r r Úobject©r r r r r Útest_to_boolean_array2 s$ ÿ ÿ r( c C sp t t dddg¡t dddg¡} tjd d d gdd}t || ¡ tjtjd d d gtddd}t || ¡ d S )NTr r r% r' r r r Ú test_to_boolean_array_all_noneI s
|
" r) za, bTFr c C s, t j| dd}t j|dd}t ||¡ d S )Nr r ©r r r r )ÚaÚbr r r r r Ú(test_to_boolean_array_missing_indicatorsR s r- r ZfooÚbarÚ1Ú2r
|
é ç ð?g @Z20130101)Zperiodsr+ c C s0 d}t jt|d tj| dd W 5 Q R X d S )Nz Need to pass bool-like valuer r r )r r r r r )r Úmsgr r r Útest_to_boolean_array_errorb s r4 c C s t jt ddddg¡dd} t jddddgdd}t | |¡ t jt dddd g¡dd} t jdddd gdd}t | |¡ d S ©Nr
|
r r r TF)r r r r r ©r r r r r Ú(test_to_boolean_array_from_integer_arrayx s r7 c C s t jt ddddg¡dd} t jddddgdd}t | |¡ t jt dddtjg¡dd} t jdddd gdd}t | |¡ d S )Nr2 g r r TF)r r r r r Únanr6 r r r Ú&test_to_boolean_array_from_float_array s r9 c C st t jddddgdd} t jddddgdd}t | |¡ t jdddd gdd} t jdddd gdd}t | |¡ d S r5 r* r6 r r r Ú"test_to_boolean_array_integer_like s r: c C sø t jddddgdd} t jddddgdd}tt| |d }t| |}t ||¡ |j| ks`t|j|ksnttt| |dd }t| |}t ||¡ |j| k s¤t|j|k s²tddd dg} t jddddgdd}tt| |d }tt ddddg¡t ddddg¡}t ||¡ ttt j| t d|d }t ||¡ tt| |
|
¡ d }t ||¡ t jddddgdd} t jddddgdd}t| dd¡ t j td d
|
t| dd¡|d W 5 Q R X t j td d
|
t| | dd¡d W 5 Q R X d S ) NTFr r ©r )r r r
|
r z&values.shape and mask.shape must matchr )r r r r r r r! r" r# r& r r r r r r r r r Útest_coerce_to_array s>
|
|
ÿ r< c C sÖ t jddddgdd} t jddddgdd}t| |}tt| }t ||¡ |j|jks^t|j|jksnttt|ddi}t ||¡ |j|jk st|j|jk s¬tt j
|
t dd t||d W 5 Q R X d S ) NTFr r r z'cannot pass mask for BooleanArray inputr r; ) r r r r r r r! r" r# r r r )r r Úarrr r r r Ú'test_coerce_to_array_from_boolean_arrayÄ s
|
r> c C sð t jddd gdd} t | ¡}tjddt jgdd}t ||¡ t jdddgdd} t | ¡}tjdddgdd}t ||¡ tj| dd}tjdddgdd}t ||¡ t jddd gdd} d}tjt|d tj| dd W 5 Q R X d S ) NTFr r r& r zqcannot convert to 'bool'-dtype NumPy array with missing values. Specify an appropriate 'na_value' for this dtype.r ) r r r ÚNAr Úassert_numpy_array_equalr r r )r= r r r3 r r r Útest_coerce_to_numpy_arrayØ s
|
|
ÿrA c C sf t tjddddddtjgtd¡} t t dd ddd d d g¡t d d d d d d dg¡}t | |¡ d S )
|
NÚTrueÚFalser/ z1.0Ú0z0.0r TF)r Ú_from_sequence_of_stringsr r r8 r& r r r6 r r r Ú"test_to_boolean_array_from_stringsó s ÿþrF c C s* t jtdd t dg¡ W 5 Q R X d S )Nzcannot be castr Zdonkey)r r r r rE r r r r Ú1test_to_boolean_array_from_strings_invalid_stringÿ s rG ÚboxZseriesr )Úidsc C sb | r
|
t jnt j}|dddgdd}| ¡ }tjdddgdd}t ||¡ |ddd gdd}| ¡ }tjddt jgdd}t ||¡ |ddd gdd}|jdd}tjddt jgtj dd}t ||¡ |dddgdd}|jdd}tjdddgdd}t ||¡ |ddd gdd}t j
|
t d d
|
|jdd}W 5 Q R X |ddd gdd}|jt d d }tjddd gdd}t ||¡ |jt dd }tjdddgdd}t ||¡ |jd d d }tjddd gd d}t ||¡ |jdtjd }tjddtjgdd}t ||¡ t j
|
t dd
|
|jd d W 5 Q R X t j
|
t dd
|
|jdd W 5 Q R X d S )NTFr r r& ÚstrZU5r z cannot convert to 'bool'-dtyper )r Zna_valueÚint64iÿÿÿr
|
r Úfloat64z cannot convert to 'int64'-dtypez!cannot convert to 'float64'-dtype)r ZSeriesr Úto_numpyr r r@ r? ZENDIANr r r r& r r8 )rH Úconr= r r r r r Ú test_to_numpy sJ rO c C s t jdddgdd} | jtd}d|d< t | t jdddgdd¡ t jdddgdd} | jtdd}d|d< t | t jdddgdd¡ d S )NTFr r r )r r )r r rM r r r )r= r r r r Útest_to_numpy_copy: s ÿrP )#Únumpyr r Zpandasr Zpandas._testingZ_testingr Z pandas.arraysr Zpandas.core.arrays.booleanr r r$ r( r) ÚmarkZ parametrizer8 r? r Úfloatr- Z
|
date_ranger4 r7 r9 r: r< r> rA rF rG rO rP r r r r Ú<module> sX ûþ
|
|
|
öþ
|
*
|
5
|
