Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/__pycache__/test_user_agent.cpython-37.pyc
Назад
B �5�g�* � @ s d Z ddlZddlZddlmZ ddlZddlZddlm m Z ddl Z ddlmZ G dd� dejj�ZG dd� de�ZG dd � d e�ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� dejj�Z ej � dee j dfee j dfee j!dfej"ee j!de j#d�ee j$dfee j%dfee j dfee j dfg�d d � �Z&ej � dee j dfee j dfee j!dfej"ee j!de j#d�ee j$dfee j%dfee j dfee j dfg�d d � �Z'ej � d e e j fg�d!d"� �Z(ej � d#ddg�d$d%� �Z)dS )&z9 Tests for the pandas custom headers in http(s) requests � N)�BytesIOc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �BaseUserAgentResponderz� Base class for setting up a server that can be set up to respond with a particular file format with accompanying content-type headers. The interfaces on the different io methods are different enough that this seemed logical to do. c C s, | � d� | jd | _t�d| jgi�}|S )z< shared logic at the start of a GET request �� z User-Agent�header)� send_response�headersZrequested_from_user_agent�pd� DataFrame)�self� response_df� r �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/test_user_agent.py�start_processing_headers s z/BaseUserAgentResponder.start_processing_headersc C s2 t � }tj|dd�}|�|� |�� |�� }|S )zQ some web servers will send back gzipped files to save bandwidth �w)�fileobj�mode)r �gzip�GzipFile�write�close�getvalue)r �response_bytes�bioZzipperr r r � gzip_bytes& s z!BaseUserAgentResponder.gzip_bytesc C s | j �|� dS )z: shared logic at the end of a GET request N)�wfiler )r r r r r �write_back_bytes1 s z'BaseUserAgentResponder.write_back_bytesN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c @ s e Zd Zdd� ZdS )�CSVUserAgentResponderc C s< | � � }| �dd� | �� |jdd��d�}| �|� d S )Nz Content-Typeztext/csvF)�indexzutf-8)r � send_header� end_headers�to_csv�encoder )r r r r r r �do_GET9 s z CSVUserAgentResponder.do_GETN)r r r r& r r r r r 8 s r c @ s e Zd Zdd� ZdS )� GzippedCSVUserAgentResponderc C sR | � � }| �dd� | �dd� | �� |jdd��d�}| �|�}| �|� d S )Nz Content-Typeztext/csvzContent-Encodingr F)r! zutf-8)r r" r# r$ r% r r )r r r r r r r& D s z#GzippedCSVUserAgentResponder.do_GETN)r r r r& r r r r r' C s r' c @ s e Zd Zdd� ZdS )�JSONUserAgentResponderc C s8 | � � }| �dd� | �� |�� �d�}| �|� d S )Nz Content-Typezapplication/jsonzutf-8)r r" r# �to_jsonr% r )r r r r r r r& Q s z JSONUserAgentResponder.do_GETN)r r r r& r r r r r( P s r( c @ s e Zd Zdd� ZdS )� GzippedJSONUserAgentResponderc C sN | � � }| �dd� | �dd� | �� |�� �d�}| �|�}| �|� d S )Nz Content-Typezapplication/jsonzContent-Encodingr zutf-8)r r" r# r) r% r r )r r r r r r r& \ s z$GzippedJSONUserAgentResponder.do_GETN)r r r r& r r r r r* [ s r* c @ s e Zd Zdd� ZdS )� ParquetPyArrowUserAgentResponderc C s8 | � � }| �dd� | �� |jddd�}| �|� d S )Nz Content-Typezapplication/octet-streamF�pyarrow)r! �engine)r r" r# � to_parquetr )r r r r r r r& i s z'ParquetPyArrowUserAgentResponder.do_GETN)r r r r& r r r r r+ h s r+ c @ s e Zd Zdd� ZdS )�$ParquetFastParquetUserAgentResponderc C sd | � � }| �dd� | �� dd l}|jdddd d� |�dd��}|�� }W d Q R X | �|� d S ) Nz Content-Typezapplication/octet-streamr z'memory://fastparquet_user_agent.parquetF� fastparquet)r! r- � compression�rb)r r" r# �fsspecr. �open�readr )r r r3 �fr r r r r& t s z+ParquetFastParquetUserAgentResponder.do_GETN)r r r r& r r r r r/ s s r/ c @ s e Zd Zdd� ZdS )�PickleUserAgentResponderc C sB | � � }| �dd� | �� t� }|�|� |�� }| �|� d S )Nz Content-Typezapplication/octet-stream)r r" r# r Z to_pickler r )r r r r r r r r& � s z PickleUserAgentResponder.do_GETN)r r r r& r r r r r7 � s r7 c @ s e Zd Zdd� ZdS )�StataUserAgentResponderc C sF | � � }| �dd� | �� t� }|j|dd� |�� }| �|� d S )Nz Content-Typezapplication/octet-streamF)Z write_index)r r" r# r Zto_statar r )r r r r r r r r& � s z StataUserAgentResponder.do_GETN)r r r r& r r r r r8 � s r8 c @ s e Zd ZdZdd� ZdS )�AllHeaderCSVResponderz? Send all request headers back for checking round trip c C sP t �| j�� �}| �d� | �dd� | �� |jdd��d�}| j � |� d S )Nr z Content-Typeztext/csvF)r! zutf-8) r r r �itemsr r" r# r$ r% r r )r r r r r r r& � s z AllHeaderCSVResponder.do_GETN)r r r r r& r r r r r9 � s r9 z&responder, read_method, parquet_enginer, r0 )Zmarksc C s� |d k r$t �|� |dkr$t �d� tj�d| ��d}tj|jd�}|�� |j }|d krh|d|� ��}n|d|� �|d�}|� � |� � |� � W d Q R X |j r�t�d S )Nr0 r3 )� localhostr )�targetzhttp://localhost:)r- )�pytest� importorskip�http�server� HTTPServer� threading�Thread� serve_forever�start� server_port�shutdown� server_close�join�empty�AssertionError)� responder� read_method�parquet_enginer@ � server_thread�port�df_httpr r r � test_server_and_default_headers� s rR c C s� |d k r$t �|� |dkr$t �d� d}t�d|gi�}tj�d| ��r}tj|j d�}|� � |j }|d kr�|d|� �d|id �}n|d|� �d|i|d �}|� � |� � |�� W d Q R X t�||� d S ) Nr0 r3 zSuper Cool Oner )r; r )r< zhttp://localhost:z User-Agent)�storage_options)rS r- )r= r> r r r? r@ rA rB rC rD rE rF rG rH rI �tm�assert_frame_equal) rL rM rN �custom_user_agent�df_truer@ rO rP rQ r r r � test_server_and_custom_headers� s, rX zresponder, read_methodc s� d}d}||d�� t j�d| ��L}tj|jd�}|�� |j}|d|� �� d�}|�� |� � |� � W d Q R X ||d � � � � � }|� dg��� }|dd g }t� � � �}t�|� fd d �|D �d ��} | � dg�} | �� jd gdd�} t�| |� d S )NzSuper Cool OnezSuper Secret One)z User-Agent�Auth)r; r )r< zhttp://localhost:)rS �0�1c s g | ] }� | �qS r r )�.0�k)rS r r � <listcomp>4 s z6test_server_and_all_custom_headers.<locals>.<listcomp>)rZ r[ r! � )Zaxis)r? r@ rA rB rC rD rE rF rG rH rI �isin�keysZ sort_valuesZ reset_index�listr r ZdroprT rU ) rL rM rV Zcustom_auth_tokenr@ rO rP rQ ra rW r )rS r �"test_server_and_all_custom_headers s, rc r- c C sV ddd�}t �| � t�ddgi�}d}t jt|d�� |jd|| d � W d Q R X d S ) NZcustomZ other_custom)z User-AgentrY Z column_nameZ column_valuezxstorage_options passed with file object or non-fsspec file path|storage_options passed with buffer, or non-supported URL)�matchz/tmp/junk.parquet)rS r- )r= r> r r Zraises� ValueErrorr. )r- r Ztrue_df�msgr r r �,test_to_parquet_to_disk_with_storage_options; s rg )*r r Z http.serverr? �ior rB r= Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr Zpandas._testingZ_testingrT r@ ZBaseHTTPRequestHandlerr r r' r( r* r+ r/ r7 r8 r9 �markZ parametrizeZread_csv� read_jsonZ read_parquet�paramZ&skip_array_manager_not_yet_implementedZ read_pickleZ read_statarR rX rc rg r r r r �<module> sj ' %%
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка