Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/__pycache__/test_gbq.cpython-37.pyc
Назад
B �5�gn � @ sB d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dlZ d dl Z d dl Z d dl Z d dl mZ d dlmZ e �d�Ze �d�Ze �d�Ze �d�ZdZdZdZe�� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Z!d'e"e#d�dd�Z$dd� Z%dd � Z&d d � Z'd d � Z(e j)�*d!dd"g�d#d$� �Z+e j)j,G d%d&� d&��Z-dS )(� )� ExitStack)�datetimeN)� DataFramezgoogle.api_core.exceptionszgoogle.cloud.bigqueryz google.oauth2.service_account� pandas_gbqc C s t � st�d� d S )Nz1Cannot run integration tests without a project id)�_get_project_id�pytest�skip� r r �N/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/test_gbq.py�_skip_if_no_project_id s r c C s t � st�d� d S )NzACannot run integration tests without a private key json file path)�_get_private_key_pathr r r r r r � _skip_if_no_private_key_path! s r c C s t ptj�d�S )NZGBQ_PROJECT_ID)� PROJECT_ID�os�environ�getr r r r r & s r c C s t } | stj�d�} | S )NZ"GBQ_GOOGLE_APPLICATION_CREDENTIALS)�PRIVATE_KEY_JSON_PATHr r r )�private_key_pathr r r r * s r c C s t � } | rtj�| �S d S )N)r �service_account� CredentialsZfrom_service_account_file)r r r r �_get_credentials1 s r c C s t � } t� }tj| |d�S )N)�project� credentials)r r �bigqueryZClient)� project_idr r r r � _get_client7 s r � )�length�returnc C s d� tjtj| d��S )N� )�k)�join�random�choices�string�ascii_lowercase)r r r r �generate_rand_str= s r& c C s� t jjdd| fd��t�}t j�d| �}t jjddd| fd�}t jjddd| fd��t�}dd� t| �D �}t|d |d |d |d |d d�t| �d �S ) N� � )�sizer c S s g | ]}t �t�d ���qS )z US/Arizona)r �now�pytz�timezone)�.0�tr r r � <listcomp>G s z+make_mixed_dataframe_v2.<locals>.<listcomp>r )�bools�flts�ints�strs�times)�index) �npr" �randintZastype�boolZrandn�str�ranger )� test_sizer0 r1 r2 r3 r4 r r r �make_mixed_dataframe_v2A s r< c sB i � � fdd�}| � d|� t�d� d� ks2t�d� ks>t�d S )Nc s � � |� tdgg�S )Ng �?)�updater )�sql�kwargs)�captured_kwargsr r � mock_read_gbqW s z>test_read_gbq_without_deprecated_kwargs.<locals>.mock_read_gbqzpandas_gbq.read_gbqzSELECT 1�verboseZ private_key)�setattr�pd�read_gbq�AssertionError)� monkeypatchrA r )r@ r �'test_read_gbq_without_deprecated_kwargsT s rH c sH i � � fdd�}| � d|� tjdddd� � d s8t�� d sDt�d S ) Nc s � � |� tdgg�S )Ng �?)r= r )r> r? )r@ r r rA e s z4test_read_gbq_with_new_kwargs.<locals>.mock_read_gbqzpandas_gbq.read_gbqzSELECT 1Tr( )�use_bqstorage_api� max_resultsrI rJ )rC rD rE rF )rG rA r )r@ r � test_read_gbq_with_new_kwargsb s rK c sB i � � fdd�}| � d|� t�d� d� ks2t�d� ks>t�d S )Nc s � � |� tdgg�S )Ng �?)r= r )r> r? )r@ r r rA s s z7test_read_gbq_without_new_kwargs.<locals>.mock_read_gbqzpandas_gbq.read_gbqzSELECT 1rI rJ )rC rD rE rF )rG rA r )r@ r � test_read_gbq_without_new_kwargsp s rL � progress_barZfooc s: i � � fdd�}| � d|� tjd|d� d� ks6t�d S )Nc s � � |� tdgg�S )Ng �?)r= r )r> r? )r@ r r rA � s z<test_read_gbq_progress_bar_type_kwarg.<locals>.mock_read_gbqzpandas_gbq.read_gbqzSELECT 1)�progress_bar_typerN )rC rD rE rF )rG rM rA r )r@ r �%test_read_gbq_progress_bar_type_kwarg~ s rO c @ s` e Zd Ze�� dd� �Zdd� Zej�ddde � fdd e � e j j�fd d e � fg�d d � �ZdS )�-TestToGBQIntegrationWithServiceAccountKeyPathc c sp t � t� dt� }t� | _| j�|�| _| j�t�| j�� t� }|� d|� �}|V | jj | jdd� d S )NZpydata_pandas_bq_testing_�.T)Zdelete_contents) r r r& r �clientZdatasetZcreate_datasetr ZDatasetZdelete_dataset)�selfZ dataset_idZ table_name�destination_tabler r r � gbq_dataset� s z9TestToGBQIntegrationWithServiceAccountKeyPath.gbq_datasetc C sZ |}d}t |�}|j|t� d t� d� tjd|� �t� t� dd�}|d d |ksVt�d S )Ni!N )� chunksizer z!SELECT COUNT(*) AS num_rows FROM �standard)r r �dialect�num_rowsr )r< �to_gbqr r rD rE rF )rS rU rT r; �df�resultr r r �test_roundtrip� s z<TestToGBQIntegrationWithServiceAccountKeyPath.test_roundtripz)if_exists, expected_num_rows, expectation�appendi, Zfail�� �replace�d c C s� |}d}t |�}|j|t� d t� d� |�( |jd d� j|t� |d t� d� W d Q R X tjd|� �t� t� dd�}|d d |ks�t�d S ) Nr_ )rV r ra )� if_existsrV r z!SELECT COUNT(*) AS num_rows FROM rW )r r rX rY r )r< rZ r r ZilocrD rE rF ) rS rb Zexpected_num_rowsZ expectationrU rT r; r[ r\ r r r �test_gbq_if_exists� s* z@TestToGBQIntegrationWithServiceAccountKeyPath.test_gbq_if_existsN)�__name__� __module__� __qualname__r ZfixturerU r] �mark� parametrize�does_not_raise�tmZexternal_error_raisedr ZgbqZTableCreationErrorrc r r r r rP � s rP )r ).� contextlibr ri r r �platformr" r$ �numpyr6 r r+ ZpandasrD r Zpandas._testingZ_testingrj Z importorskipZapi_exceptionsr r r r r ZPRIVATE_KEY_JSON_CONTENTS�python_version�VERSIONr r r r r r �intr9 r& r< rH rK rL rg rh rO ZsinglerP r r r r �<module> s@
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка