o _~#g70 @slddlZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddlmZddlZddlmZddlmmZe jdejdgdgdggdd ejdgdd d ge jd gd ddZe jdejdgdgdggdd ejdgdd d ge jd ddgddZe jddddiejfdiejfdiej fgddZ!e jdddddejddifdiejddifdiej ifgd d!Z"d"d#Z#d$d%Z$d&d'Z%d(d)Z&e jj'd*d+Z(e jj'd,d-Z)e jdejdgdgdggdd ejdgdd d ge jd gd d.d/Z*e jdejdgdgdggdd ejdgdd d ge jd gd d0d1Z+e jdejdgdgdggdd ejdgdd d ge jd gd d2d3Z,e jd4d5e j-fd6ej.fgd7d8Z/d9d:Z0d;d<Z1d=d>Z2d?d@Z3dS)AN)Path)is_platform_windowsobjdg~Ϛ?g9DܜJ?g>D)?g.4i(@g33333@g̤AXYZcolumnsr namemethod) to_pickleto_jsonto_csvcCs|dkr ddd}t-}t||||dtj|}t|||ddtj|}||ks2JWddS1s=wYdS)Ntarw:gzrmode compressiontm ensure_cleangetattrospathgetsize)rrcompression_onlyrcompressed_sizeuncompressed_sizer$/var/www/static.ux5.de/https/Moving-Object-Detection-with-OpenCV/env/lib/python3.10/site-packages/pandas/tests/io/test_compression.pytest_compression_sizes    "r&rrc Cst:}tj||dkrdnd|d}t|||j|jjr"JWdn1s,wYtj |}Wdn1sAwYt;}tj|ddd}t|||j|jjrbJWdn1slwYtj |}||ks}JWddS1swYdS)Nrrwr) rricom get_handlerhandleclosedrrr )rrr!rhandlesr"r#r$r$r%test_compression_size_fh.s(   "r-z'write_method, write_kwargs, read_methodindexFrc Cstjgdgdggdd}||}td|}t|||fi||||d}Wdn1s6wYt||dS)N)g?r)g333333 @rr compressedr)pd DataFramerrrassert_frame_equal) write_method write_kwargs read_methodr!compression_to_extensioninput extensionroutputr$r$r%,test_dataframe_compression_defaults_to_inferKs r=z1write_method,write_kwargs,read_method,read_kwargsT)r.headersqueezetypseriesc Cstjgddd}||}td|6}t|||fi|d|vr9|} | d=||fd|i| d} n ||fd|i|} Wdn1sMwYtj| |dd dS) N)rr0 r rr2r?rr F) check_names)r3Seriesrrrcopyr?assert_series_equal) r6r7r8 read_kwargsr!r9r:r;rkwargsr<r$r$r%)test_series_compression_defaults_to_infer_s rJc Cstjdgdgdggdd}tF}tj|d|d%}tt|j|j |dWdn1s7wYWdn1sFwYWddSWddS1s^wYdS)Nrrrrr r'r) r3r4rrr(r)assert_produces_warningRuntimeWarningrr*)r!dfrr,r$r$r%test_compression_warnings  "rNc Cs tjdtddtjtdtdtjddtdDtdd }t 7}t |d d }|j |d |d | d Wdn1sFwYt |tj|d |dWdn1sawYt}|j |d |d | d t |tj|d |ddS)z? Binary file handles support compression. GH22555 皙?xABCDdtypecSg|]}d|qSzi-r$.0ir$r$r% z+test_compression_binary..rRr r.wb)r)rrrN) index_colr)r3r4nparangereshapeIndexlistobjectrangerropenrseekr5read_csvioBytesIO)r!rMrfiler$r$r%test_compression_binarys(    rncCstjdtddtjtdtdtjddtdDtdd }d d d }t ,}t |}|j ||d t d|}|j ||d ||ksPJWddS1s[wYdS)z Gzip should create reproducible archives with mtime. Note: Archives created with different filenames will still be different! GH 28103 rOrPrQrTrUcSrWrXr$rYr$r$r%r\r]z7test_gzip_reproducibility_file_name..rRr^gziprmtimer皙?N)r3r4rarbrcrdrerfrgrrrrtimesleep read_bytes)rMcompression_optionsrr<r$r$r%#test_gzip_reproducibility_file_names   "rxcCstjdtddtjtdtdtjddtdDtdd }d d d }t }|j ||d d| }t dt }|j ||d d|| ksQJdS)zL Gzip should create reproducible archives with mtime. GH 28103 rOrPrQrTrUcSrWrXr$rYr$r$r%r\r]z9test_gzip_reproducibility_file_object..rRr^rorprqr_)rrrsN)r3r4rarbrcrdrerfrgrkrlrgetvaluertru)rMrwbufferr<r$r$r%%test_gzip_reproducibility_file_objects  r{cC&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 executablePIPEcoder$r$r%test_with_missing_lzmasrcCr|)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~Nrrr$r$r%test_with_missing_lzma_runtimes rcCs|t0}t|||ddtj|}t|||ddddtj|}||ks,JWddS1s7wYdS)Nrorrpr compresslevelrrrrcompressed_size_defaultcompressed_size_fastr$r$r%test_gzip_compression_levels   "rcCstC}t|||ddtj|}t|||ddddtj|}||ks,J|dkr?tj|ddWddSWddS1sJwYdS)Nxzrrp)rpresetr)rrrrrr r3rjrr$r$r%test_xz_compression_level_reads    "rcCsFt}t|||ddddWddS1swYdS)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. bz2rprrN)rrr)rrrr$r$r%test_bzip_compression_level&s "rzsuffix,archive.zipz.tarc Cstj|dB}||dWdn1swYtjtdd t|Wdn1s3wYWddSWddS1sKwYdS)Nfilenamer'zZero files foundmatch)rrpytestraises ValueErrorr3rj)suffixarchiverr$r$r%test_empty_archive_zip:s  "rc CstjddO}t|d}|dd|ddWdn1s$wYtjtdd t |Wdn1s@wYWddSWddS1sXwYdS) Nrrr'a.csvzfoo,barb.csvz Multiple files found in ZIP filer) rrzipfileZipFilewritestrrrrr3rj)rrmr$r$r%test_ambiguous_archive_zipIs  "rcCs |d}t|ddd }|dWdn1swY|d}t|ddd }|dWdn1s;wY|d}t|d}||d||dWdn1sawYtjtdd t |WddS1s~wYdS) Nrr'zutf-8)encodingzfoo,bar rz archive.tarz#Multiple files found in TAR archiver) rhwritetarfileTarFileaddrrrr3rj)tmp_pathcsvAPathacsvBPathbtarpathrr$r$r%test_ambiguous_archive_tarRs     "rc Cstjddw}tjddggddgdj|dd d d d t|D}tj|d .}| }t |dks5J| |d d}trGd}nd}||ksOJWdn1sYwYWdn1shwYWddSWddS1swYdS)Nz.foor12foobarr rrrF)rr.)fileobjrprutf8zfoo,bar 1,2 z foo,bar 1,2 )rrr3r4rrorhrr getmemberslen extractfilereaddecoder)rm uncompressedrmemberscontentexpectedr$r$r%!test_tar_gz_to_different_filenamecs, "rc Cslt(}tj|ddWdn1swYWddSWddS1s/wYdS)Nr')rr)rkrlr( _BytesTarFile)rzr$r$r%test_tar_no_error_on_closews "r)4rorkrpathlibrrrrrrtrnumpyrar pandas.compatrpandasr3pandas._testing_testingrpandas.io.commoncommonr(mark parametrizer4rEr&r-rj read_json read_pickler=rJrNrnrxr{ single_cpurrrrrrrrrrrrr$r$r$r%s