
    A>i              
          S SK JrJr  S SKrS SKrS SKJs  Jr  S SK	r
S SKJr  S rS r\R                   \R"                  R$                  \R"                  R&                  S 5       5       5       rS rS rS rS	 r\R"                  R3                  S
S\R4                  " S\R6                  " S5      S9/5      \R"                  R3                  SSS/5      \R"                  R3                  SSS/5      S 5       5       5       rS rg)    )BytesIOStringIONc                    [         R                  " / SQ/ SQ/SS// SQS9nUR                  XS9  [        R                  " U[         R
                  " XS95        [        R                  " X5       nUR                  5       R                  S5      n[        U5      nS S S 5        [        R                  " U[         R
                  " W5      5        g ! , (       d  f       N:= f)	N)g~Ϛ?g9DܜJ?g>D)?)g.4i(@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         d/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/tests/io/json/test_compression.pytest_compression_roundtripr      s    		')GHCj
B JJyJ2"bll9NO 
		I	3r!!&) 
4 "bll401 
4	3s   /+C
Cc                     U " SSSS5      n[         R                  " U5      nU " SSSS5      n[         R                  " USS9n[        R                  " X$5        g )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)       sR     vv7JKll#45OtVV5LMOLLeDM/9    c                 r   [         R                  " [        S5      5      n[        R                  " 5        S3nUR                  X0S9  [        US5       nUR                  XVS9  S S S 5        [         R                  " SUR                   SU 3U US9n[        R                  " XG5        g ! , (       d  f       NK= f)	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	CBZZ\N%
 CJJyJ2	i	!###4 
 ll
 %%&au-O
 ". 
	s   B((
B6c                     [         R                  " [        S5      5      nUR                  USSU S9  [         R                  " USU S9n[        R
                  " X#5        g )Nr,   recordsTorientlinesr   )rA   r   )r   r   r   r   r   r   )r   r   r   r;   s       r   test_lines_with_compressionrB   >   sJ    	hAB	CBJJy$KJPll9DkRO".r*   c                 "   [         R                  " [        S5      5      nUR                  USSU S9  [         R                  " USSU S9 n[         R                  " U5      nS S S 5        [
        R                  " UW5        g ! , (       d  f       N&= f)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   sq    	hMN	OBJJy$KJP	
	))C.
 "o.	
 
s   	B  
Bc                     [         R                  " [        S5      5      nSn[        R                  " [
        US9   UR                  U SS9  S S S 5        g ! , (       d  f       g = f)Nr,   *Unrecognized compression type: unsupportedmatchunsupportedr   )r   r   r   pytestraises
ValueErrorr   )r   r   msgs      r   'test_write_unsupported_compression_typerR   P   sE    	hAB	CB
6C	z	-


9-
8 
.	-	-s   A
A#c                     Sn[         R                  " [        US9   [        R                  " U SS9  S S S 5        g ! , (       d  f       g = f)NrJ   rK   rM   r   )rN   rO   rP   r   r   )r   rQ   s     r   &test_read_unsupported_compression_typerT   W   s/    
6C	z	-
YM: 
.	-	-s	   ;
A	infer_stringFTpyarrow)marksto_infer
read_inferc                 N   [         R                  " SU5         U nUSX6    3-  n[         R                  " SS/05      nU(       a  SOUn	U(       a  SOUn
UR                  XyS9  [         R                  " XzS9n[
        R                  " X5        S S S 5        g ! , (       d  f       g = f)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*%$,+&07k


8
8hE
f) 
@	?	?s   A5B
B$c                     [         R                  " SS/05      n[        5        nUR                  X S9  S S S 5        g ! , (       d  f       g = f)Nr   rD   r   )r   r   r   r   )r   expectedbuffers      r   test_to_json_compression_moderg   {   s8    ||S1#J'H	f9 
s	   =
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; UFLLR]]95MNO dE]3e}5* 6 4
*2:r*   