
    Pi              
          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                            d
d ej        d ej        d                    g          ej                            dddg          ej                            dddg          d                                     Zd ZdS )    )BytesIOStringIONc                    t          j        g dg dgddgg d          }|                    ||            t          j        |t          j        ||                      t          j        ||           5 }|                                                    d          }t          |          }d d d            n# 1 swxY w Y   t          j        |t          j        |                     d S )	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         y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/io/json/test_compression.pytest_compression_roundtripr      s)   		'	'	')G)G)GHCj
 
 
B JJykJ222"bl9+NNNOOO 
	I{	3	3  r!!&))                              "bl40011111s   77B::B>B>c                      | dddd          }t          j        |          } | dddd          }t          j        |d          }t          j        ||           d S )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)       sl     vv7JKKl#455OhtVV5LMMOLeDDDM/=99999    c                    t          j        t          d                    }t          j                     d}|                    ||            t          |d          5 }|                    ||           d d d            n# 1 swxY w Y   t          j        d|j         d| | |          }t          j
        ||           d S )	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B	C	CBZ\\
 
 
 CJJykJ222	i		 5!##!#4445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 l- %----  O
 "o.....s   BB
Bc                     t          j        t          d                    }|                    |dd|            t          j        |d|           }t	          j        ||           d S )Nr,   recordsTorientlinesr   )rA   r   )r   r   r   r   r   r   )r   r   r   r;   s       r   test_lines_with_compressionrB   >   sc    	hABB	C	CBJJy$KJPPPl9DkRRRO"o.....r*   c                 ,   t          j        t          d                    }|                    |dd|            t          j        |dd|           5 }t          j        |          }d d d            n# 1 swxY w Y   t          j        ||           d S )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   s    	hMNN	O	OBJJy$KJPPP	
 
 
 )	)C..) ) ) ) ) ) ) ) ) ) ) ) ) ) ) "o.....s   A44A8;A8c                     t          j        t          d                    }d}t          j        t
          |          5  |                    | d           d d d            d S # 1 swxY w Y   d S )Nr,   *Unrecognized compression type: unsupportedmatchunsupportedr   )r   r   r   pytestraises
ValueErrorr   )r   r   msgs      r   'test_write_unsupported_compression_typerR   P   s    	hABB	C	CB
6C	z	-	-	- 9 9


9-
8889 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s   A$$A(+A(c                     d}t          j        t          |          5  t          j        | d           d d d            d S # 1 swxY w Y   d S )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A	Ainfer_stringFTpyarrow)marksto_infer
read_inferc                 L   t          j        d|          5  | }|d||          z  }t          j        ddgi          }|rdn|}	|rdn|}
|                    ||	           t          j        ||
          }t          j        ||           d d d            d S # 1 swxY w Y   d S )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,	?	? * *& M&>{&KMMM\3*%%$,=+&0A77k


8
888h4DEEE
fb)))* * * * * * * * * * * * * * * * * *s   A6BB Bc                     t          j        ddgi          }t                      5 }|                    ||            d d d            d S # 1 swxY w Y   d S )Nr   rD   r   )r   r   r   r   )r   expectedbuffers      r   test_to_json_compression_moderg   {   s    |S1#J''H	 :f[999: : : : : : : : : : : : : : : : : :s   A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 2 2": : : / /   /"/ / // / /9 9 9; ; ; ULFL]R]95M5MNNNO  dE]33e}55* * 65 43 
*2: : : : :r*   