o ۭj`@sddlmZddlZddlZddlZddlZddlmZmZm Z ddl Z ddl Z ddl m Z mZddlmZddlZddlmZerRddlmZmZddlmZmZ ddddZddddZddddZdd ddZdS)!) annotationsN) TYPE_CHECKINGAnyCallable) get_bz2_file get_lzma_file)import_optional_dependency) ensure_clean)FilePathReadPickleBuffer) DataFrameSeriesobjrpath"FilePath | ReadPickleBuffer | NonereturnDataFrame | SeriescCs^|}|durdtd}t|}t||t|WdS1s(wYdS)a Pickle an object and then read it again. Parameters ---------- obj : any object The object to pickle and then re-read. path : str, path object or file-like object, default None The path where the pickled object is written and then read. Returns ------- pandas object The original object that was pickled and then re-read. N__z __.pickle)uuiduuid4r pd to_pickle read_pickle)rr_path temp_pathrP/var/www/html/CropPilot/venv/lib/python3.10/site-packages/pandas/_testing/_io.pyround_trip_pickle's  $r str | NonecCsXtj}|dur d}t|}||||||}Wd|S1s%wY|S)a Write an object to file specified by a pathlib.Path and read it back Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. N ___pathlib___)pathlibPathr )writerreaderrr!rrrrround_trip_pathlibAs   r$cCsfddl}|dj}|durd}t|}||||||}Wd|S1s,wY|S)a Write an object to file specified by a py.path LocalPath and read it back. Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. rNzpy.path___localpath___)pytest importorskiplocalr )r"r#rr& LocalPathrrrrround_trip_localpath\s    r*testdeststrNonec Cs|f}d}d}|dkrtj}d}||f}d}nI|dkr7tj}d}tj|d}t|} t||_|| f}d}n)|d kr?t j }n!|d krGt }n|d krQt d j }n|d krYt}ntd||||d} t| ||WddS1sywYdS)a Write data to a compressed file. Parameters ---------- compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd'} The compression type to use. path : str The file path to write the data. data : str The data to write. dest : str, default "test" The destination file (for ZIP only) Raises ------ ValueError : An invalid compression value was passed in. wbwritezipwwritestrtar)nameaddfilegzipbz2zstd zstandardxzzUnrecognized compression type: )modeN)zipfileZipFiletarfileTarFileTarInfoioBytesIOlensizer7GzipFilerropenr ValueErrorgetattr) compressionrdatar,argsr<methodcompress_methodfilebytesfrrrwrite_to_compressedys8    "rR)N)rrrrrr)rr)r+)r,r-rr.) __future__rr7rBr r?typingrrrrr= pandas.compatrrpandas.compat._optionalrpandasrpandas._testing.contextsr pandas._typingr r r r rr$r*rRrrrrs(