zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
U
¬ý°dvã
@sÈddlZddlmZddlmZddlZddlmZ    ddl
m Z m Z ej dddZdd    „Zejje d
d ej d ejd dddfejd ddddfg¡dd„ƒƒZdd„Zdd„Zdd„Zdd„ZdS)éN)Úpa_version_under10p0)Ú PeriodDtype)Ú PeriodArrayÚ period_arrayZpyarrowz1.0.1)Z
minversioncCsvddlm}|dƒ}|dƒ}|dƒ}|jdks2t‚||ks>t‚||ksJt‚t|ƒt|ƒks^t‚t|ƒt|ƒksrt‚dS)Nr©ÚArrowPeriodTypeÚDÚM)Ú(pandas.core.arrays.arrow.extension_typesrÚfreqÚAssertionErrorÚhash)rÚp1Úp2Zp3©rúcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/period/test_arrow_compat.pyÚtest_arrow_extension_types   rzWrong behavior with pyarrow 10)Úreasonz
data, freqZ2017é)ÚperiodsrÚA)rr zA-DECc    CsÞddlm}t||d}t |¡}t|j|ƒs2t‚|jj|ksBt‚tj|j    dd}|j
  |¡sbt‚tj|t  ¡d}|  |¡s‚t‚d}t jt|dtj|ddW5QRXt jtd    dtj||d
ƒdW5QRXdS) Nrr©r Úint64©Útypez5Not supported to convert PeriodArray to 'double' type)ÚmatchÚfloat64zdifferent 'freq'ÚT)r
rrÚpaÚarrayÚ
isinstancerr r Zasi8ÚstorageÚequalsrÚpytestZraisesÚ    TypeError)Údatar rrÚresultÚexpectedÚmsgrrrÚtest_arrow_arrays      
r)cCszddlm}tdddgdd}tj|d<t |¡}t|j|ƒsBt    ‚|jj
dksRt    ‚tjdddgdd    }|j   |¡svt    ‚dS)
Nrréérrrrr) r
rrÚpdÚNaTrrr rr r r!r")rÚarrr&r'rrrÚtest_arrow_array_missing>s 
 
r/cCs¶ddlm}tdddgdd}tj|d<t d|i¡}t |¡}t|     d¡j
|ƒsVt ‚|  ¡}t|dj tƒsrt ‚t ||¡t ||g¡}|  ¡}tj||gd    d
}t ||¡dS) Nrrr*r+rrrÚaT)Z ignore_index)r
rrr,r-Ú    DataFramerÚtabler Úfieldrr Ú    to_pandasÚdtyperÚtmÚassert_frame_equalZ concat_tablesÚconcat)rr.Údfr2r&Ztable2r'rrrÚtest_arrow_table_roundtripKs 
 
 r:cCs–ddlm}tgdd}t d|i¡}t |¡}t| d¡j    |ƒsFt
‚tjtj g|  d¡j    dg|j d}| ¡}t|djtƒs†t
‚t ||¡dS)Nrrrrr0r)Úschema)r
rrr,r1rr2r r3rr Z chunked_arrayÚcolumnr;r4r5rr6r7)rr.r9r2r&rrrÚ test_arrow_load_from_zero_chunks^s  
ÿr=cCsxtdddgdd}tj|d<t d|i¡}t |¡}| ¡}|jjdksLt    ‚| 
¡}t |dj t ƒsht    ‚t ||¡dS)Nr*r+rÚHrr0)rr,r-r1rr2Zreplace_schema_metadatar;Úmetadatar r4r r5rr6r7)r.r9r2r&rrrÚ+test_arrow_table_roundtrip_without_metadataps
 
r@)r#Zpandas.compat.pyarrowrZpandas.core.dtypes.dtypesrZpandasr,Zpandas._testingZ_testingr6Zpandas.core.arraysrrZ importorskiprrÚmarkZxfailZ parametrizeZ
date_ranger)r/r:r=r@rrrrÚ<module>s&   þþ