Ë
    ž…jë  ã            
       ó  — d dl mZmZ d dlZd dlZd dlmc mZ d dl	Z
d dlmZ d„ Zd„ Zej                   ej"                  j$                  ej"                  j&                  d„ «       «       «       Zd„ Zd„ Zd„ Zd	„ Zej"                  j3                  d
d ej4                  d ej6                  d«      ¬«      g«      ej"                  j3                  dddg«      ej"                  j3                  dddg«      d„ «       «       «       Zd„ Zy)é    )ÚBytesIOÚStringIONc                 óÀ  — t        j                  g d¢g d¢gddgg d¢¬«      }|j                  || ¬«       t        j                  |t        j
                  || ¬«      «       t        j                  || «      5 }|j                  «       j                  d«      }t        |«      }d d d «       t        j                  |t        j
                  «      «       y # 1 sw Y   Œ3xY w)	N)g¿¶~úÏš¿?g9DÜœJÎ?g>éD‚)â?)gÜ.4×i¤(@g33333þ@gÍÌÌÌ¤œAÚAÚB)ÚXÚYÚZ)ÚindexÚcolumns©ÚcompressionÚutf8)
ÚpdÚ	DataFrameÚto_jsonÚtmÚassert_frame_equalÚ	read_jsonÚdecompress_fileÚreadÚdecoder   )r   Ú	temp_fileÚdfÚfhÚresultÚdatas         úf/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/io/json/test_compression.pyÚtest_compression_roundtripr      s´   € Ü	�‰Ú	'Ò)GÐHØ�CˆjÚô
€Bð ‡J�Jˆy k€JÔ2Ü×Ñ˜"œbŸl™l¨9À+ÔNÔOô 
×	Ñ	˜I {Ó	3ð  °rØ—‘“×!Ñ! &Ó)ˆÜ˜Óˆ÷ ô ×Ñ˜"œbŸl™l¨4Ó0Õ1÷ ð  ús   Á7+CÃCc                 ó´   —  | dddd«      }t        j                  |«      } | dddd«      }t        j                  |d¬«      }t        j                  ||«       y )NÚioÚjsonr   ztsframe_v012.jsonztsframe_v012.json.zipÚzipr   )r   r   r   r   )ÚdatapathÚuncompressed_pathÚuncompressed_dfÚcompressed_pathÚcompressed_dfs        r   Útest_read_zipped_jsonr)       sT   € Ù   v¨vÐ7JÓKÐÜ—l‘lÐ#4Ó5€Oá˜t V¨VÐ5LÓM€OÜ—L‘L ¸eÔD€Mä×Ñ˜/¨=Õ9ó    c                 óx  — t        j                  t        d«      «      }t        j                  «       › d�}|j                  || ¬«       t        |d«      5 }|j                  ||¬«       d d d «       t        j                  d|j                  › d|› �| |¬«      }t        j                  ||«       y # 1 sw Y   ŒHxY w)	Nú {"a": [1, 2, 3], "b": [4, 5, 6]}z.jsonr   Úrb)ÚKeyÚBodyzs3://ú/)r   Ústorage_options)r   r   r   ÚuuidÚuuid4r   ÚopenÚ
put_objectÚnamer   r   )r   Ús3_bucket_publicÚs3sor   r   ÚkeyÚfÚroundtripped_dfs           r   Útest_with_s3_urlr<   *   s®   € ô
 
�‰”hÐAÓBÓ	C€Bä�Z‰Z‹\ˆN˜%Ð
 €CØ‡J�Jˆy k€JÔ2Ü	ˆi˜Ó	ð 5 !Ø×#Ñ#¨°!Ð#Ô4÷5ô —l‘lØ
Ð ×%Ñ%Ð& a¨ uÐ-ØØô€Oô
 ×Ñ˜"˜oÕ.÷5ð 5ús   ÁB0Â0B9c                 óÆ   — t        j                  t        d«      «      }|j                  |dd| ¬«       t        j                  |d| ¬«      }t	        j
                  ||«       y )Nr,   ÚrecordsT©ÚorientÚlinesr   )rA   r   )r   r   r   r   r   r   )r   r   r   r;   s       r   Útest_lines_with_compressionrB   >   sL   € Ü	�‰”hÐAÓBÓ	C€BØ‡J�Jˆy °$ÀK€JÔPÜ—l‘l 9°DÀkÔR€OÜ×Ñ˜"˜oÕ.r*   c                 ó  — t        j                  t        d«      «      }|j                  |dd| ¬«       t        j                  |dd| ¬«      5 }t        j                  |«      }d d d «       t        j                  |«       y # 1 sw Y   Œ xY w)Nz,{"a": ["foo", "bar", "baz"], "b": [4, 5, 6]}r>   Tr?   é   )rA   Ú	chunksizer   )r   r   r   r   Úconcatr   r   )r   r   r   Úresr;   s        r   Útest_chunksize_with_compressionrH   E   sv   € Ü	�‰”hÐMÓNÓ	O€BØ‡J�Jˆy °$ÀK€JÔPä	�‰Ø˜¨¸ô
ð )à	ÜŸ)™) C›.ˆ÷)ô ×Ñ˜"˜oÕ.÷	)ð )ús   ÁBÂBc                 óÊ   — t        j                  t        d«      «      }d}t        j                  t
        |¬«      5  |j                  | d¬«       d d d «       y # 1 sw Y   y xY w)Nr,   ú*Unrecognized compression type: unsupported©ÚmatchÚunsupportedr   )r   r   r   ÚpytestÚraisesÚ
ValueErrorr   )r   r   Úmsgs      r   Ú'test_write_unsupported_compression_typerR   P   sN   € Ü	�‰”hÐAÓBÓ	C€BØ
6€CÜ	�‰”z¨Ô	-ñ 9Ø
�
‰
�9¨-ˆ
Ô8÷9÷ 9ñ 9ús   ¼AÁA"c                 ó–   — d}t        j                  t        |¬«      5  t        j                  | d¬«       d d d «       y # 1 sw Y   y xY w)NrJ   rK   rM   r   )rN   rO   rP   r   r   )r   rQ   s     r   Ú&test_read_unsupported_compression_typerT   W   s8   € Ø
6€CÜ	�‰”z¨Ô	-ñ ;Ü
�‰�Y¨MÕ:÷;÷ ;ñ ;ús	   ž?¿AÚinfer_stringFTÚpyarrow)ÚmarksÚto_inferÚ
read_inferc                 ó<  — t        j                  d|«      5  | }|d||   › �z  }t        j                  ddgi«      }|rdn|}	|rdn|}
|j                  ||	¬«       t        j                  ||
¬«      }t        j                  ||«       d d d «       y # 1 sw Y   y xY w)Nzfuture.infer_stringztest.r   rD   Úinferr   )r   Úoption_contextr   r   r   r   r   )Úcompression_onlyrY   rX   Úcompression_to_extensionrU   Útmp_pathr   Úfilenamer   Úto_compressionÚread_compressionr   s               r   Útest_to_json_compressionrc   ]   s¡   € ô 
×	Ñ	Ð0°,Ó	?ñ *à&ˆð  Ð&>¸{Ñ&KÐ%LÐMÑMˆä�\‰\˜3  ˜*Ó%ˆá$,™°+ˆÙ&0™7°kÐà
�
‰
�8¨ˆ
Ô8Ü—‘˜hÐ4DÔEˆÜ
×Ñ˜f bÔ)÷*÷ *ñ *ús   —A2BÂBc                 ó˜   — t        j                  ddgi«      }t        «       5 }|j                  || ¬«       d d d «       y # 1 sw Y   y xY w)Nr   rD   r   )r   r   r   r   )r   ÚexpectedÚbuffers      r   Útest_to_json_compression_moderg   {   sE   € ä�|‰|˜S 1 #˜JÓ'€Hä	‹ð :�fØ×Ñ˜¨[ÐÔ9÷:÷ :ñ :ús   £A Á A	)r!   r   r   r2   rN   Úpandas.util._test_decoratorsÚutilÚ_test_decoratorsÚtdÚpandasr   Úpandas._testingÚ_testingr   r   r)   Úskip_if_not_us_localeÚmarkÚ
single_cpuÚnetworkr<   rB   rH   rR   rT   ÚparametrizeÚparamÚ
skip_if_norc   rg   © r*   r   ú<module>rw      s   ð÷ó ã ç )Ð )ã Ý ò2ò":ð ×ÑØ‡�×ÑØ‡�×Ññ/ó ó ó ð/ò"/ò/ò9ò;ð ‡�×ÑØ�U˜L˜FŸL™L¨°]°R·]±]À9Ó5MÔNÐOóð ‡�×Ñ˜ d¨E ]Ó3Ø‡�×Ñ˜¨¨e }Ó5ñ*ó 6ó 4óð
*ó2:r*   