a nEb! @sddlZddlZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z ddlmmZejde jdgdgdggdde jdgdd d gejd gd d dZejde jdgdgdggdde jdgdd d gejd ddgddZejddddie jfdie jfdie jfgddZejddddde jddifdie jddifdie jifgdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Z d+d,Z!ejde jdgdgdggdde jdgdd d gejd gd d-d.Z"ejde jdgdgdggdde jdgdd d gejd gd d/d0Z#dS)1N)Pathobjdg~Ϛ?g9DܜJ?g>D)?g.4i(@g33333@g̤AXYZcolumnsrnamemethod) to_pickleto_jsonto_csvcCsttX}t||||dtj|}t|||ddtj|}||ksRJWdn1sf0YdS)N compressiontm ensure_cleangetattrospathgetsize)rrcompression_onlyrcompressed_sizeuncompressed_sizer`/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/io/test_compression.pytest_compression_sizes   r!rrc Csth}tj|d|d,}t|||j|jjr8JWdn1sL0Ytj |}Wdn1sv0Ytt}tj|ddd,}t|||j|jjrJWdn1s0Ytj |}||ksJWdn1s0YdS)Nwr) rricom get_handlerhandleclosedrrr)rrrrhandlesrrrrr test_compression_size_fh%s ** * r(z'write_method, write_kwargs, read_methodindexFrcCstjgdgdggdd}tj|}td|2}t|||fi||||d}Wdn1sn0Yt||dS)N)g?r)g333333 @rr compressedr)pd DataFramer#_compression_to_extensionrrrassert_frame_equal) write_method write_kwargs read_methodrinput extensionroutputrrr ,test_dataframe_compression_defaults_to_infer>s  *r8z1write_method,write_kwargs,read_method,read_kwargsT)r)headersqueezetypZseriesc Cstjgddd}tj|}td|l}t|||fi|d|vrt|}|d=||fd|i|d} n||fd|i|} Wdn1s0Ytj | |dd dS) N)rr+ rr r-r:rr F)Z check_names) r.Seriesr#r0rrrcopyr:Zassert_series_equal) r2r3r4Z read_kwargsrr5r6rkwargsr7rrr )test_series_compression_defaults_to_inferRs  2rAc Cstjdgdgdggdd}tz}tj|d|dJ}tt |j|j |dWdn1sn0YWdn1s0YWdn1s0YdS)Nrrrrr r"r) r.r/rrr#r$Zassert_produces_warningRuntimeWarningrr%)rdfrr'rrr test_compression_warningms  rDc Cst}tn}t|dd*}|j|d|d|dWdn1sN0Yt|tj|d|dWdn1s0Yt }|j|d|d|dt|tj|d|ddS)z? Binary file handles support compression. GH22555 wb)mode)rFrrN)Z index_colr) r makeDataFrameropenrseekr1r.read_csvioBytesIO)rrCrfilerrr test_compression_binaryzs (" rNcCst}ddd}tV}t|}|j||dtd|}|j||d||ksbJWdn1sv0YdS)z Gzip should create reproducible archives with mtime. Note: Archives created with different filenames will still be different! GH 28103 gziprmtimerr,N)rrGrrrtimesleep read_bytes)rCcompression_optionsrr7rrr #test_gzip_reproducibility_file_names   rWcCsht}ddd}t}|j||dd|}tdt}|j||dd||ksdJdS)zL Gzip should create reproducible archives with mtime. GH 28103 rOrPrQrE)rrFr,N)rrGrKrLrgetvaluerSrT)rCrVbufferr7rrr %test_gzip_reproducibility_file_objects  rZcCs&td}tjtjd|gtjddS)z6Tests if import pandas works when lzma is not present.zT import sys sys.modules['lzma'] = None import pandas -cstderrNtextwrapdedent subprocess check_outputsys executablePIPEcoderrr test_with_missing_lzmasrhcCs&td}tjtjd|gtjddS)z\Tests if RuntimeError is hit when calling lzma without having the module available. z import sys import pytest sys.modules['lzma'] = None import pandas as pd df = pd.DataFrame() with pytest.raises(RuntimeError, match='lzma module'): df.to_csv('foo.csv', compression='xz') r[r\Nr^rfrrr test_with_missing_lzma_runtimes ricCszt^}t|||ddtj|}t|||ddddtj|}||ksXJWdn1sl0YdS)NrOrrPr compresslevelr)rrrZcompressed_size_defaultZcompressed_size_fastrrr test_gzip_compression_levels   rlcCsDt(}t|||ddddWdn1s60YdS)zGH33196 bzip needs file size > 100k to show a size difference between compression levels, so here we just check if the call works when compression is passed as a dict. bz2rPrjrN)rrr)rrrrrr test_bzip_compression_levels rn)$rKrpathlibrrarcr_rSZpytestZpandasr.Zpandas._testingZ_testingrZpandas.io.commoncommonr#markZ parametrizer/r>r!r(rJ read_jsonZ read_pickler8rArDrNrWrZrhrirlrnrrrr s