Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/parser/__pycache__/test_compression.cpython-37.pyc
Назад
B �5�g � @ s d Z ddlZddlmZ ddlZddlZddlmZ ddlm Z ej ddgd�dd � �Z ej d d � �Z ej�d d ddg�dd� �Zej�d d dg�dd� �Zdd� Zdd� Zej�dddg�dd� �Zej�d dd d g�d d � �Zd!d"� Zej�d#d$d%d&g�d'd(� �Zd)d*� ZdS )+zZ Tests compressed data parsing functionality for all of the parsers defined in parsers.py � N)�Path)� DataFrameTF)�paramsc C s | j S )N)�param)�request� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/parser/test_compression.py�buffer s r c C s6 | }t |d��}|�� }W d Q R X |�|�}|||fS )N�rb)�open�read�read_csv)� all_parsers�csv1�parser�f�data�expectedr r r �parser_and_data s r � compression�zip�infer�zip2c C s� | \}}}t �d��t}tj|dd��}|�d|� W d Q R X |dkrjt|d��}|j|dd�}W d Q R X n|j||d�}t �||� W d Q R X d S ) Nz test_file.zip�w)�mode� test_filer r r )r )�tm� ensure_clean�zipfile�ZipFile�writestrr r �assert_frame_equal) r r r r r �path�tmpr �resultr r r �test_zip s r% c C s� | \}}}t �d��j}ddg}tj|dd�� }x|D ]}|�||� q4W W d Q R X tjtdd�� |j||d� W d Q R X W d Q R X d S ) Nzcombined_zip.zipr Z second_filer )r zMultiple files)�match)r ) r r r r r �pytest�raises� ValueErrorr ) r r r r r r" Zinner_file_namesr# � file_namer r r � test_zip_error_multiple_files1 s r+ c C sd | \}}}t �� �H}tj|dd�� W d Q R X tjtdd�� |j|dd� W d Q R X W d Q R X d S )Nr )r z Zero files)r& r )r )r r r r r' r( r) r )r r �_r" r r r �test_zip_error_no_files@ s r- c C sb | \}}}t �� �F}t|d��0}tjtjdd�� |j|dd� W d Q R X W d Q R X W d Q R X d S )Nr zFile is not a zip file)r& r )r )r r r r' r( r � BadZipfiler )r r r, r" r r r r �test_zip_error_invalid_zipK s r/ �filenamez test.{ext}c C s� | \}}}|}|dkrdn|}|d kr*|n |j |d�}|rH|rHt�d� tj|d��d} t�|| |� |rldn|} |r�t| d��} |j| | d�} W d Q R X n|j| | d�} t�| |� W d Q R X d S ) N�gzip�gz)�extz9Cannot deduce compression from buffer of compressed data.)r0 r r )r ) �formatr' �skipr r Zwrite_to_compressedr r r! ) r Zcompression_onlyr r0 r r r � compress_typer3 r" r r r$ r r r �test_compressionT s r7 r3 r2 �bz2c C s~ | }ddd�}|j |f|�}d|d< |rLt|��}|j |f|�}W d Q R X n"|rXd| nd}|j || f|�}t�||� d S )Nr T)Z index_colZ parse_datesr r �.� )r r r r! ) r r r r3 r �kwargsr r r$ r r r �test_infer_compressionl s r< c C s\ | }|� |�}tj�|d|� d��}|j||ddd�}tddgddgd ��}t�||� d S ) N�utfz _ex_small.zipr � )�encodingr �sepZ Venezuelau Hugo Chávez FríaszHenrique Capriles R.)ZCountryZTwitter)r4 �osr" �joinr r r r! ) r Z csv_dir_pathZ utf_valueZ encoding_fmtr r? r" r$ r r r r � test_compression_utf_encoding s rC �invalid_compressionZsfarkZbz3Zzipperc C s@ | }d|i}d|� �}t jt|d�� |jd|� W d Q R X d S )Nr z Unrecognized compression type: )r& � test_file.zip)rE )r' r( r) r )r rD r Zcompress_kwargs�msgr r r �test_invalid_compression� s rG c C s~ | }t dddgi�}t�d��V}t�d��@}|j|dd� t|��t|��� � t�|j|d d�|� W d Q R X W d Q R X d S ) N�ar � ztest.csvz test.csv.zipF)�index)r ) r r r Zto_csvr � write_text� read_textr! r )r r ZdfZpath_csvZpath_zipr r r �!test_ignore_compression_extension� s rM )�__doc__rA �pathlibr r r' Zpandasr Zpandas._testingZ_testingr Zfixturer r �markZ parametrizer% r+ r- r/ r7 r< rC rG rM r r r r �<module> s"
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка