Ë
    AúYj|  ã            	       óN  — d dl Z d dlmZ d dlZd dlmZ d dlm	Z	m
Z
 e j                  j                  d«      Z e j                  d«      Zd„ Ze j                  j#                  d ej$                  dd	¬
«      df ej$                  dd	d¬«      dfg«      d„ «       Zd„ Zd„ Zd„ Zd„ Zy)é    N)ÚPeriodDtype)ÚPeriodArrayÚperiod_arrayz=ignore:Passing a BlockManager to DataFrame:DeprecationWarningÚpyarrowc                  óâ   — ddl m}   | d«      } | d«      } | d«      }|j                  dk(  sJ ‚||k(  sJ ‚||k7  sJ ‚t        |«      t        |«      k(  sJ ‚t        |«      t        |«      k7  sJ ‚y )Nr   ©ÚArrowPeriodTypeÚDÚM)Ú(pandas.core.arrays.arrow.extension_typesr	   ÚfreqÚhash)r	   Úp1Úp2Úp3s       úw/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/pandas/tests/arrays/period/test_arrow_compat.pyÚtest_arrow_extension_typer      su   € ÝHá	˜Ó	€BÙ	˜Ó	€BÙ	˜Ó	€Bà�7‰7�cŠ>Ðˆ>Ø�Š8€Oˆ8Ø�Š8€Oˆ8Ü�‹8”t˜B“xÒÐÐÜ�‹8”t˜B“xÒÐÑó    z
data, freqÚ2017é   )Úperiodsr
   ÚYE)r   r   zY-DECc                 ó@  — ddl m} t        | |¬«      }t        j	                  |«      }t        |j                  |«      sJ ‚|j                  j                  |k(  sJ ‚t        j	                  |j                  d¬«      }|j                  j                  |«      sJ ‚t        j	                  |t        j                  «       ¬«      }|j                  |«      sJ ‚d}t        j                  t        |¬«      5  t        j	                  |d¬«       d d d «       y # 1 sw Y   y xY w)	Nr   r   )r   Úint64©Útypez5Not supported to convert PeriodArray to 'double' type)ÚmatchÚfloat64)r   r	   r   ÚpaÚarrayÚ
isinstancer   r   Úasi8ÚstorageÚequalsr   ÚpytestÚraisesÚ	TypeError)Údatar   r	   r   ÚresultÚexpectedÚmsgs          r   Útest_arrow_arrayr,   "   sã   € õ Iä˜4 dÔ+€GÜ�X‰X�gÓ€FÜ�f—k‘k ?Ô3Ð3Ð3Ø�;‰;×Ñ˜tÒ#Ð#Ð#Ü�x‰x˜Ÿ™¨7ˆxÓ3€HØ�>‰>× Ñ  Ô*Ð*Ð*ô �X‰X�g¤B§H¡H£JˆXÓ/€FØ�=‰=˜Ô"Ð"Ð"ð B€CÜ	�‰”y¨Ô	,ñ *Ü
�‰�˜yˆÔ)÷*÷ *ñ *ús   Ã3DÄDc                  óP  — ddl m}  t        g d¢d¬«      }t        j                  |d<   t
        j                  |«      }t        |j                  | «      sJ ‚|j                  j                  dk(  sJ ‚t
        j                  g d¢d	¬
«      }|j                  j                  |«      sJ ‚y )Nr   r   ©é   é   r   ú	period[D]©Údtyper/   r
   )r/   Nr   r   r   )r   r	   r   ÚpdÚNaTr   r    r!   r   r   r#   r$   )r	   Úarrr)   r*   s       r   Útest_arrow_array_missingr7   =   s�   € ÝHä
’i {Ô
3€CÜ�V‰V€Cˆ�Fä�X‰X�c‹]€FÜ�f—k‘k ?Ô3Ð3Ð3Ø�;‰;×Ñ˜sÒ"Ð"Ð"Ü�x‰xš¨7ˆxÓ3€HØ�>‰>× Ñ  Ô*Ð*Ñ*r   c                  ó0  — ddl m}  t        g d¢d¬«      }t        j                  |d<   t        j
                  d|i«      }t        j                  |«      }t        |j                  d«      j                  | «      sJ ‚|j                  «       }t        |d   j                  t        «      sJ ‚t        j                  ||«       t        j!                  ||g«      }|j                  «       }t        j"                  ||gd¬	«      }t        j                  ||«       y )
Nr   r   r.   r1   r2   r/   ÚaT)Úignore_index)r   r	   r   r4   r5   Ú	DataFramer   Útabler!   Úfieldr   Ú	to_pandasr3   r   ÚtmÚassert_frame_equalÚconcat_tablesÚconcat)r	   r6   Údfr<   r)   Útable2r*   s          r   Útest_arrow_table_roundtriprE   J   sØ   € ÝHä
’i {Ô
3€CÜ�V‰V€Cˆ�FÜ	�‰�s˜C�jÓ	!€Bä�H‰H�R‹L€EÜ�e—k‘k #Ó&×+Ñ+¨_Ô=Ð=Ð=Ø�_‰_Ó€FÜ�f˜S‘k×'Ñ'¬Ô5Ð5Ð5Ü×Ñ˜& "Ô%ä×Ñ˜u e˜nÓ-€FØ×ÑÓ€FÜ�y‰y˜"˜b˜°Ô5€HÜ×Ñ˜& (Õ+r   c                  óú  — ddl m}  t        g d¬«      }t        j                  d|i«      }t
        j                  |«      }t        |j                  d«      j                  | «      sJ ‚t
        j                  t
        j                  g |j                  d«      j                  ¬«      g|j                  ¬«      }|j                  «       }t        |d   j                  t        «      sJ ‚t!        j"                  ||«       y )Nr   r   r1   r2   r9   r   )Úschema)r   r	   r   r4   r;   r   r<   r!   r=   r   Úchunked_arrayÚcolumnrG   r>   r3   r   r?   r@   )r	   r6   rC   r<   r)   s        r   Ú test_arrow_load_from_zero_chunksrJ   ]   sÇ   € õ Iä
�b Ô
,€CÜ	�‰�s˜C�jÓ	!€Bä�H‰H�R‹L€EÜ�e—k‘k #Ó&×+Ñ+¨_Ô=Ð=Ð=Ü�H‰HÜ	×	Ñ	˜" 5§<¡<°£?×#7Ñ#7Ð	Ó	8Ð9À%Ç,Á,ð ó €Eð �_‰_Ó€FÜ�f˜S‘k×'Ñ'¬Ô5Ð5Ð5Ü×Ñ˜& "Õ%r   c                  óz  — t        g d¢d¬«      } t        j                  | d<   t        j                  d| i«      }t        j                  |«      }|j                  «       }|j                  j                  �J ‚|j                  «       }t        |d   j                  t        «      sJ ‚t        j                  ||«       y )Nr.   z	period[h]r2   r/   r9   )r   r4   r5   r;   r   r<   Úreplace_schema_metadatarG   Úmetadatar>   r!   r3   r   r?   r@   )r6   rC   r<   r)   s       r   Ú+test_arrow_table_roundtrip_without_metadatarN   p   s”   € Ü
’i {Ô
3€CÜ�V‰V€Cˆ�FÜ	�‰�s˜C�jÓ	!€Bä�H‰H�R‹L€Eà×)Ñ)Ó+€EØ�<‰<× Ñ Ð(Ð(Ð(à�_‰_Ó€FÜ�f˜S‘k×'Ñ'¬Ô5Ð5Ð5Ü×Ñ˜& "Õ%r   )r%   Úpandas.core.dtypes.dtypesr   Úpandasr4   Úpandas._testingÚ_testingr?   Úpandas.core.arraysr   r   ÚmarkÚfilterwarningsÚ
pytestmarkÚimportorskipr   r   ÚparametrizeÚ
date_ranger,   r7   rE   rJ   rN   © r   r   ú<module>r[      s³   ðÛ å 1ã Ý ÷ð
 �[‰[×'Ñ'ØCó€
ð
 €V×Ñ˜Ó#€ò ð ‡�×ÑØà	ˆ�‰�v qÔ	)¨3Ð/Ø	ˆ�‰�v q¨tÔ	4°gÐ>ðóñ*óð*ò(
+ò,ò&&ó&&r   