Ë
|
oñúh ã
|
ó. d dl Zd dlZd dlZd dlmZ ej j d« Z
|
ej d« Z d dl mZ ej D cg c] } ej g d¢| ¬« c} Zeej$ D cg c] } ej g d¢| ¬« c} z Ze ej g d¢d ¬« gz Z ej& eeD cg c] }|j( j* c}¬
|
« d « Zd Zd Zd Zd Zd Zd Z ej& d « Z d Z ej j? de jA d« e jC e jA d« e jA d« g« g« d « Z"d Z#yc c} w c c} w c c}w )é Nz=ignore:Passing a BlockManager to DataFrame:DeprecationWarningÚpyarrow)Ú pyarrow_array_to_numpy_and_mask)é é é N©Údtype)g¹?gÉ?g333333Ó?N)TFTNÚboolean)ÚparamsÚidsc ó | j S )zi
|
Fixture returning parametrized array from given dtype, including integer,
|
float and boolean
|
)Úparam)Úrequests ú_H:\Change_password\venv_build\Lib\site-packages\pandas/tests/arrays/masked/test_arrow_compat.pyÚdatar s ð ==Ðó c óú t j | « }t j | j t d ¬« t j | j
|
j « ¬« }|j |« sJ y )N)Úna_value©Útype)ÚpaÚarrayÚto_numpyÚobjectÚfrom_numpy_dtyper Ú numpy_dtypeÚequals)r ÚarrÚexpecteds r Útest_arrow_arrayr s^ Ü
|
((4.CÜxxØ f t Ó,Ü × Ñ §¡×!7Ñ!7Ó 8ð ó Hð ::hÔ Ð Ñ r c ó^ t j d| i« }t j |« }|j d« j
|
t | j j « k( sJ |j « }|d j | j k( sJ t j ||« y )NÚa) ÚpdÚ DataFramer ÚtableÚfieldr Ústrr r Ú to_pandasÚtmÚassert_frame_equal©r Údfr% Úresults r Útest_arrow_roundtripr. ' s
Ü sDkÓ "BÜ HHRLEØ ;;sÓ × Ñ ¤C¨¯
|
©
|
×(>Ñ(>Ó$?Ò ?Ð?Ð ?à __Ó FØ #;× Ñ §
|
¡
|
Ò *Ð*Ð *Ü×Ñ& "Õ%r c ón d } t j g d¢t j « ¬« }t j g d¢t j « ¬« }t j g d¢t j « ¬« }t j
|
j |||gg d¢« }|j | ¬« }t j g d¢d¬ « }t j g d¢d
|
¬ « }t j g d¢d
|
¬ « }t j |||d« } t j || « y ) Nc óÐ t j j | « rt j « S t j j | « rt j « S y )N)r ÚtypesÚ
|
is_booleanr# Ú BooleanDtypeÚ
|
is_integerÚ
|
Int64Dtype)Ú
|
arrow_types r Ú types_mapperz<test_dataframe_from_arrow_types_mapper.<locals>.types_mapper2 sD Ü 88× Ñ zÔ *Ü??Ó$Ð $Ü XX× Ñ Ô ,Ü==?Ð "ð -r )TNFr )r Nr )éÿÿÿÿr é )ÚboolsÚintsÚ
|
small_ints)r7 r
|
r ÚInt64) r r Úbool_Úint64Úint8Ú RecordBatchÚ from_arraysr( r# ÚSeriesr$ r) r* )
|
r7 Ú bools_arrayÚ
|
ints_arrayÚsmall_ints_arrayÚ record_batchr- r: r; r< r s
|
r Ú&test_dataframe_from_arrow_types_mapperrH 1 sà ò#ô ((Ò.´R·X±X³Z(Ó@KÜ,¬R¯X©X«ZÓ8JÜxx¢
|
´·±³xÓ;ÐÜ>>×-Ñ-Ø jÐ"2Ð3Ò5TóLð × #Ñ #° Ð #Ó >FÜ IIÒ)°Ô ;EÜ 99\¨Ô 1DÜ:¨WÔ5JÜ|| e°TÈÑ TÓUHÜ×Ñ& (Õ+r c ó t j d| dd i« }t j |« }|j d« j
|
t | j j « k( sJ t j t j g |j d« j
|
¬« g|j ¬« }|j « }|d j | j k( sJ t j ||« y )Nr" r r )Úschema)r# r$ r r% r& r r' r r Ú chunked_arrayrJ r( r) r* r+ s r Ú test_arrow_load_from_zero_chunksrL F sÎ ô
|
sD 1IÐ&Ó 'BÜ HHRLEØ ;;sÓ × Ñ ¤C¨¯
|
©
|
×(>Ñ(>Ó$?Ò ?Ð?Ð ?Ü HHÜ × Ñ " 5§;¡;¨sÓ#3×#8Ñ#8Ð Ó 9Ð:À5Ç<Á<ð ó Eð __Ó FØ #;× Ñ §
|
¡
|
Ò *Ð*Ð *Ü×Ñ& "Õ%r c óÚ t j « } | j t j g d¢d¬« « }t j g d¢d¬« }t j ||« y )N)r r r é Nr? r ÚUInt32r )r# Ú UInt32DtypeÚ__from_arrow__r r r) Ú assert_extension_array_equal)r r- r s r Útest_arrow_from_arrow_uintrS T sO ô NNÓ EØ × !Ñ !¤"§(¡(Ò+=ÀG (Ó"LÓ MFÜxxÒ*°(Ô;Hä×#Ñ# F¨HÕ5r c ó t j d| i« }t j |« }|j dd « j « }|j dd j d¬« }t j ||« |j | d « }t j |« }|j dd « j « }|j dd j d¬« }t j ||« y )Nr" r T)Údropr ) r# r$ r r% Úslicer( ÚilocÚ reset_indexr) r* Úfillna)r r, r% r- r Údf2s r Útest_arrow_slicedr[ _ sÐ ô
|
sDkÓ "BÜ HHRLEØ [[DÓ !× +Ñ +Ó -FØwwqr{×&Ñ&¨DÐ&Ó1HÜ×Ñ& (Ô+ð ))DGÓ
|
CÜ HHSMEØ [[DÓ !× +Ñ +Ó -FØxx|×'Ñ'¨TÐ'Ó2HÜ×Ñ& (Õ+r c óô t j | « }t j |« }t j g d¢|¬« }t j g d¢|¬« }t j g d¢« }||||fS )zt
|
Fixture returning actual and expected dtype, pandas and numpy arrays and
|
mask from a given numpy dtype
|
)r r r Nr )r r r r )TTTF)Únpr r r r )Úany_real_numpy_dtypeÚnp_dtypeÚpa_typeÚpa_arrayÚ np_expectedÚ mask_expecteds r Únp_dtype_to_arraysrd p sg ô xxÐ,Ó-HÜ×!Ñ! (Ó+Gô xx¨gxÓ6Hô ((9¨HÔ5KÜHHÒ 6Ó7MØ X{¨MÐ 9Ð9r c óª | \ }}}}t ||« \ }}t j |dd |« t j ||« |j « d }|j « d }|j « d j « } t
|
j | dz « }
|
t
|
j j |j t |« ||
|
g|j ¬« } | j « t | |« \ }}t j |dd |« t j ||« d|j j dz z } t
|
j | | z « } t
|
j d« }t
|
j j |j t |« || g|j dz ¬« }|j « t ||« \ }}t j |dd |« t j ||« t j g |¬ « }t j g t j ¬ « }t
|
j j |j d||g|j ¬« }|j « t ||« \ }}t j |dd |« t j ||« y)
|
a&
|
Test conversion from pyarrow array to numpy array.
|
|
Modifies the pyarrow buffer to contain padding and offset, which are
|
considered valid buffers by pyarrow.
|
|
Also tests empty pyarrow arrays with non empty buffers.
|
See https://github.com/pandas-dev/pandas/issues/40896
|
Nr r r ó )r ÚlengthÚbuffersÚoffseté ó r )r r) Úassert_numpy_array_equalrh Ú
|
to_pybytesr Ú py_bufferÚArrayÚ from_buffersr Úlenri ÚvalidateÚ bit_widthr] r r> )rd r_ ra rb rc r ÚmaskÚ mask_bufferÚ data_bufferÚdata_buffer_bytesÚdata_buffer_trailÚpa_array_trailri Údata_buffer_offsetÚmask_buffer_offsetÚpa_array_offsetÚnp_expected_emptyÚmask_expected_emptys r Ú$test_pyarrow_array_to_numpy_and_maskr s} ð 6HÑ2Hh ]Ü0°¸8ÓDJD$Ü× Ñ R a ¨+Ô6Ü× Ñ mÔ4à×"Ñ"Ó$ QÑ'KØ×"Ñ"Ó$ QÑ'KØ ×(Ñ(Ó*¨1Ñ-×8Ñ8Ó:Ðô Ð%6¸Ñ%@ÓAÐÜXX×*Ñ*Ø ]]Ü8}Ø Ð /Ð0Øð +ó Nð ×ÑÔ Ü0° ÀÓJJD$Ü× Ñ R a ¨+Ô6Ü× Ñ mÔ4ð ×/Ñ/°1Ñ4Ñ 5FÜ fÐ/@Ñ&@ÓAÐÜ gÓ.ÐÜhh×+Ñ+Ø ]]Ü8}Ø#Ð%7Ð8Ø Ñ"ð ,ó Oð × Ñ Ô Ü0° À(ÓKJD$Ü× Ñ R a ¨+Ô6Ü× Ñ mÔ4ô ¨8Ô4ÐÜ (( 2¬R¯X©XÔ6Ðähh×+Ñ+Ø ]]ØØ kÐ*Øð ,ó Oð × Ñ Ô Ü0° À(ÓKJD$Ü× Ñ R a Ð*;Ô<Ü× Ñ Ð&9Õ:r r é
|
rN é c ó | j j |« }|j « j « sJ t |« dk( sJ y )Nr )r rQ ÚisnaÚallrq )r r Úress r Útest_from_arrow_nullr Á s> ð **×
|
#Ñ
|
# CÓ
|
(CØ 88:>>Ô ÐÐ Ü s8r>Ð>r c óâ t j | « j d« }t j t
|
d ¬« 5 | j j |« d d d « y # 1 sw Y y xY w)NÚstring)Úmatch)r r ÚcastÚpytestÚraisesÚ TypeErrorr rQ )r r s r Útest_from_arrow_type_errorr Ê sS ô ((4.×
|
Ñ
|
hÓ
|
'CÜ y¨Ô -ñ 'ð
|
|
×!Ñ! #Ô&÷'÷ 'ñ 'ús Á A%Á%A.)$Únumpyr] r Úpandasr# Úpandas._testingÚ_testingr) ÚmarkÚfilterwarningsÚ
|
pytestmarkÚ importorskipr Ú%pandas.core.arrays.arrow._arrow_utilsr ÚALL_INT_EA_DTYPESr ÚarraysÚFLOAT_EA_DTYPESÚfixturer Únamer r r. rH rL rS r[ rd r Ú parametrizeÚnullsrK r r )r r" s 00r ú<module>r s ðÛ Û ã Ý à [[× 'Ñ 'ØCó
|
ð
|
V×ÑÓ#å Qà>@×>RÑ>RÖ S°U("((?¨%Ö
|
0Ò SØ ÀR×EWÑEWÖ
|
X¸E8288Ò)°Ö 7Ò
|
XÑ XØ 8288Ò-°YÔ ?Ð
|
@Ñ @ð v¸&Ö#A°Q A§G¡G§L£LÒ#AÔBñó Cðò ò&ò,ò* &ò 6ò,ð" ñ:ó ð:ò"<;ð~ ×ÑØ BHHRL"×*Ñ*¨B¯H©H°Q«K¸¿¹À!»Ð+EÓFÐ Góñóðó 'ùòu
|
TùÚ
|
Xùò $Bs Á FÂ F Ã F
|
