Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/__pycache__/test_gcs.cpython-37.pyc
Назад
B �5�gd � @ s� d dl mZ d dlZd dlZd dlZd dlmZmZm Z m Z m Z m Z d dl mZ d dlmZ ejdd� �Ze�d�ej�dd d d d d g�dd� ��Zeed�dd�Ze�d�ej�dddg�dd� ��Ze�d�e�d�dd� ��Z e� d�dd � �Z dS ) � )�BytesION)� DataFrame� date_range�read_csv� read_excel� read_json� read_parquet)�_test_decoratorsc sN ddl m}m} |j�� t� � dd� � _G � fdd�d|�}| �d|� � S )z"Emulate GCS using a binary buffer.r )�AbstractFileSystem�registryc S s dS )NT� r r r �N/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/test_gcs.py�<lambda> � z gcs_buffer.<locals>.<lambda>c s e Zd Z� fdd�Zdd� ZdS )z%gcs_buffer.<locals>.MockGCSFileSystemc s � � d� � S )Nr )�seek)�args�kwargs)� gcs_bufferr r �open! s z*gcs_buffer.<locals>.MockGCSFileSystem.openc [ s |dd�gS )N�file)�name�typer )�self�pathr r r r �ls% s z(gcs_buffer.<locals>.MockGCSFileSystem.lsN)�__name__� __module__� __qualname__r r r )r r r �MockGCSFileSystem s r zgcsfs.GCSFileSystem)�fsspecr r �target�clearr �close�setattr)� monkeypatchr r r r )r r r s r Zgcsfs�format�csv�json�parquet�excel�markdownc C s ddl m} |j�� tddgdtjgddgtdd d �d ��}d |� �}|d krp|j|dd� t |dgdd�}n�|dkr�d}|� |� t |dgdd�}nj|dkr�|� |� t |dgd�}nH|dkr�t�d� |�|� t|�}n"|dk�rt�d� |�|� |}t�||� dS )zE Test that many to/read functions support GCS. GH 33987 r )r � � g @�t�sz 2018-06-18� )�periods)�int�float�str�dtzgs://test/test.r& T)�indexr4 )Z parse_dates� index_colr) zgs://test/test.xlsr' )Z convert_datesr( Zpyarrowr* �tabulateN)r r r r! r �np�nanr �to_csvr Zto_excelr �to_jsonr �pytestZ importorskip� to_parquetr Z to_markdown�tm�assert_frame_equal)r r% r �df1r Zdf2r r r �test_to_read_gcs. s6 rA )�result�expectedc C s< | dd� |dd� ks t �| dd� |dd� ks8t �dS )z� We would like to assert these are equal, but the 10th and 11th bytes are a last-modified timestamp, which in some builds is off-by-one, so we check around that. See https://en.wikipedia.org/wiki/ZIP_(file_format)#File_headers N� � )�AssertionError)rB rC r r r �assert_equal_zip_safe[ s rG �encodingzutf-8�cp1251c C s ddl m} |j�� t�� }d|i}|dkr6d|d< t� }|j|||dd� d }|j|||d � | �� }|�� } t || � t |d||d �} t� || � |dkr�d }d |d< |d|� �7 }|j|||d � | �� }|�� } t || � t |dd |d �} t� || � dS )z� Compression and encoding should with GCS. GH 35677 (to_csv, compression), GH 26124 (to_csv, encoding), and GH 32392 (read_csv, encoding) r )r �method�gzipr+ �mtime�wb)� compressionrH �modezgs://test/test.csv)rN rH )r6 rN rH �gzZinfer�.N) r r r r! r> Z makeDataFramer r: �getvaluerG r r? ) r Zcompression_onlyrH r ZdfrN �bufferZpath_gcs�resrC Zread_dfr r r �$test_to_csv_compression_encoding_gcsg s4 rU � fastparquetc sx ddl m}m} |j�� tddgdtjgddgtdd d �d ��}G � fd d �d |�}| � d|� |j ddddd� dS )zCRegression test for writing to a not-yet-existent GCS Parquet file.r )r r r+ r, g @r- r. z 2018-06-18r/ )r0 )r1 r2 r3 r4 c s e Zd Zd� fdd� ZdS )z7test_to_parquet_gcs_new_file.<locals>.MockGCSFileSystem�rc s d|kr t �ttj�� d�|�S )N�wz test.parquet)�FileNotFoundErrorr �osr �join)r r rO r )�tmpdirr r r � s z<test_to_parquet_gcs_new_file.<locals>.MockGCSFileSystem.openN)rW )r r r r r )r\ r r r � s r zgcsfs.GCSFileSystemzgs://test/test.csvTrV N)r5 ZenginerN ) r r r r r! r r8 r9 r r# r= )r$ r\ r r r@ r r )r\ r � test_to_parquet_gcs_new_file� s r] c C s" t �t�� td� W d Q R X d S )Nzgs://test/test.csv)r> Zexternal_error_raised� ImportErrorr r r r r � test_gcs_not_present_exception� s r_ ) �ior rZ �numpyr8 r< Zpandasr r r r r r Zpandas._testingZ_testingr> Z pandas.utilr �tdZfixturer Z skip_if_no�markZ parametrizerA �bytesrG rU r] Zskip_if_installedr_ r r r r �<module> s ", /
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка