a nEb<@sddlmZddlZddlmmZddlZddl m Z ddl mm ZddZddZejddZd d Zd d Zd dZddZejdddgejdddgddZddZdS))BytesIONc Cstjgdgdgddggdd}t}|j||dt|tj||dt||}| d}Wdn1s0Yt|t|Wdn1s0YdS) N)g~Ϛ?g9DܜJ?g>D)?)g.4i(@g33333@g̤AAB)XYZ)indexcolumns compressionutf8) pd DataFrametm ensure_cleanto_jsonassert_frame_equal read_jsonZdecompress_filereaddecode)r dfpathfhresultre/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/io/json/test_compression.pytest_compression_roundtrip s ,rcCsD|dddd}t|}|dddd}tj|dd}t||dS)Niojsondataztsframe_v012.jsonztsframe_v012.json.zipzipr )r rrr)datapathZuncompressed_pathZuncompressed_dfZcompressed_pathZ compressed_dfrrrtest_read_zipped_jsons  r"c Cstd}t\}|j||dt|d$}|djd|dWdn1sV0YWdn1st0Ytjd||d}t||dS) N {"a": [1, 2, 3], "b": [4, 5, 6]}r rbz pandas-testztest-1)KeyZBodyzs3://pandas-test/test-1)r Zstorage_options) r rrrropenZBucketZ put_objectr)r Z s3_resourceZs3sorrfroundtripped_dfrrrtest_with_s3_url(s   Pr)cCsdtH}td}|j|dd|dtj|d|d}t||Wdn1sV0YdS)Nr#recordsTZorientlinesr )r,r )rrr rrr)r rrr(rrrtest_lines_with_compression9s   r-c Cstt}td}|j|dd|dtj|dd|d}t|}Wdn1sX0Yt||Wdn1s0YdS)Nz,{"a": ["foo", "bar", "baz"], "b": [4, 5, 6]}r*Tr+)r, chunksizer )rrr rrconcatr)r rrresr(rrrtest_chunksize_with_compressionBs  (r2c Csvtd}tP}d}tjt|d|j|ddWdn1sJ0YWdn1sh0YdS)Nr#*Unrecognized compression type: unsupportedmatch unsupportedr )r rrrpytestraises ValueErrorr)rrmsgrrr'test_write_unsupported_compression_typeOs   r;c CsltP}d}tjt|dtj|ddWdn1s@0YWdn1s^0YdS)Nr3r4r6r )rrr7r8r9r r)rr:rrr&test_read_unsupported_compression_typeWs r<to_inferTF read_inferc Cs|}d}|tj|7}tddgi}|r.dn|}|r:dn|}t|8}|j||dtj||d} t| |Wdn1s0YdS)Nztest.rr.Zinferr ) icomZ_compression_to_extensionr rrrrrr) Zcompression_onlyr>r=r filenamerZto_compressionZread_compressionrrrrrtest_to_json_compression^s   rAcCsHtddgi}t}|j||dWdn1s:0YdS)Nrr.r )r rrr)r expectedbufferrrrtest_to_json_compression_modessrD)rrr7Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr Zpandas._testingZ_testingrZpandas.io.commoncommonr?rr"Zskip_if_not_us_localer)r-r2r;r<markZ parametrizerArDrrrrs"