Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/pytables/__pycache__/test_select.cpython-37.pyc
Назад
B �5�g�� � @ sJ d dl mZ d dlZd dlZd dlmZ d dlZd dlm Z m Z m Z m Z m Z mZmZmZmZmZmZ d dlmZmZmZ d dlmZ ejj Z dd� Z d d � Z d d � Z d d� Z!dd� Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'dd � Z(d d � Z)d d � Z*d!d"� Z+d#d$� Z,d%d&� Z-d'd(� Z.d)d*� Z/d+d,� Z0ej�1d-d.d/d0g dgg�d1d2� �Z2dS )3� )�catch_warningsN)� Timestamp) � DataFrame�HDFStore�Index� MultiIndex�Series�_testing� bdate_range�concat� date_range�isna�read_hdf)� _maybe_remove�ensure_clean_path�ensure_clean_store)�Termc C s t ddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd�}ttj�dd �|dddgd�}t| ��N}|jd|dd� |dg }t�|j ddgd�|� t�|j ddd�|� W d Q R X t tj�d�|dd�}t| ��,}|jd|dd� t� |j ddd�|� W d Q R X d S ) N�foo�barZbazZqux�one�two�threer � � � Zfoo_nameZbar_name)Zlevels�codes�names� �A�B�C)�index�columns�df�table)�format)r" z columns=['A'])�where)r! �name�s) r r �np�random�randnr �put�tm�assert_frame_equal�selectr Zassert_series_equal)� setup_pathr! r# �store�expectedr( � r3 �Z/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/pytables/test_select.py� test_select_columns_in_where! s . r5 c C s^ t tj�dd�ddddgd�}tdddd�|_t| ���}|�d |� |�d �}|}t j ||d d � |jd |j d�}|}t j ||d d � |jd dgd�}|j d d �dgf }t � ||� W d Q R X t t tj�dd�ddddgd�t tjjd dd d��dd�ddgd�gdd�}tdddd�|_t| ���}|�d |� |�d �}|}t j ||d d � |jd |j d�}|}t j ||d d � |j d d �dgf }|jd dgd�}t j ||d d � |j d d �ddgf }|jd ddgd�}t j ||d d � W d Q R X t| ��b}|�d |� |�d |� |j d d �ddgf }t ||g�}|jd ddgd�}t j ||d d � W d Q R X d S )Nr � r r )r" z 20130101 9:30�T)�periods�freqr# T)Z by_blocksr � )�sizer r r )�axis)r r) r* r+ r r! r �appendr/ r- r. r" �locr �randintZreshape)r0 r# r1 �resultr2 r3 r3 r4 �test_select_with_dups> sP $ rA c C s� t | ����}tdd���� t�� }t|d� |�d|� |jdddgd�}|jddgd�}t�||� |�ddg�}|jddgd�}t�||� t|d� |jd|dgd� |jdd gddgd�}||j d k jddgd�}t�||� t|d� |jd|dd� |jdd gddgd�}||j d k jddgd�}t�||� t|d� |jd|dgd� |jdd gd d gd�}||j d k jd d gd�}t�||� W d Q R X W d Q R X d S ) NT)�recordr# r r )r" zcolumns=['A', 'B'])� data_columnszA > 0r r �D) r r r- �makeTimeDataFramer r= r/ �reindexr. r )r0 r1 r# r@ r2 r3 r3 r4 � test_select} s4 rG c C s� t | ���H}ttddd�tj�d�d��}t|d� |jd|ddgd� |�dd �}||j t d �k }t � ||� ttj�d d �dd gd�}d|d< d|j dd �df< |d dk|d< t|d� |jd|dd� ||jdk jddgd�}x2dD ]*}|jdd|� �ddgd�}t � ||� q�W ||jdk jddgd�}x4dD ],}|jdd|� �ddgd�}t � ||� �q>W ttj�d�tj�d�d��}t|d � |�d |� |�d d �}|jt|j�dd � dgd �}t � ||� ttj�d�tj�d�tjdd d!�d"��}t|d#� |�d#|� |�d#d$�}|jt|j�dd � dgd �}t � ||� W d Q R X t | ���.}ttd%�td%�d&�d'd!�}|d( d �t�|d(< |jd)|dd� |jd)d*d+�}||d, d-k }t � ||� tj|jd< ||d, d-k }|jd.|ddd/� |jd.d*d+�}t � ||� ttd%�td%�d&�d'd!�}|d( d �t�|d(< tj|jd0< ||d, d-k }|jd1|dd� |jd1d*d+�}t � ||� W d Q R X t | ��T}t �� }||d dk }|jd|dd� t�d�}|jdd2gd+�}t � ||� W d Q R X d S )3Nz 2012-01-01i, )r8 )�tsr r# rH r )rC zts>=Timestamp('2012-02-01')z 2012-02-01� r r )r" r �objectr r6 r �boolvT)T�truer z boolv == F)F�falser r: )r r Zdf_intzindex<10 and columns=['A']r )r! r" Zf8)�dtype)r r r! Zdf_floatz index<10.0 and columns=['A']� )�cols�values�float64rP �df1z values>2.0)r& rQ g @�df2)rC r! r �df4z A>np_zero)r r r r) r* r+ r r= r/ rH r r- r. r> rK rF Zrand�listr! Zarange�range�apply�str�nan�ilocZ makeDataFramerR )r0 r1 r# r@ r2 �vZnp_zeror3 r3 r4 �test_select_dtypes� s� r] c C s� t | ����}ttddd�tj�d�td�dgd dgd dgd d d � td�D � d ��}t|d � |jd |d dddgd� |� d d�}||j t d�k }t � ||� |� d d�}||j t d�k|j�dddg�@ }t � ||� dddgdd � td�D � }|� d d�}||j t d�k|j�|�@ }t � ||� tdd�}|� d d�}||j�|� }t � ||� t|�dk�s|t�t|j dd� j�}|� d d�}||j �|j� }t � ||� t|�dk�s�t�W d Q R X d S ) Nz 2012-01-01i, )r8 �a�2 �b�c�d c S s g | ]}d |d���qS )r^ �03dr3 )�.0�ir3 r3 r4 � <listcomp> s z0test_select_with_many_inputs.<locals>.<listcomp>)rH r r �usersr# rH r r rg )rC zts>=Timestamp('2012-02-01')z 2012-02-01z1ts>=Timestamp('2012-02-01') & users=['a','b','c']c S s g | ]}d |d���qS )r^ rc r3 )rd re r3 r3 r4 rf 0 s �< z.ts>=Timestamp('2012-02-01') and users=selector�� z B=selectorr z ts=selector)r r r r) r* r+ rW r r= r/ rH r r- r. rg �isinr �len�AssertionErrorr rQ )r0 r1 r# r@ r2 �selectorr3 r3 r4 � test_select_with_many_inputs s: 6 " rn c C s^ t | ���}t�d�}t|d� |�d|� |�d�}t|jddd��}t|�}t�||� t|jddd��}t |�dks|t �t|�}t�||� t|jddd��}t|�}t�||� W d Q R X t | ��p}t�d�}|� |d � d }t jt|d �� t|d dd� W d Q R X t jt|d �� t|d dd� W d Q R X W d Q R X t | ��j}t�d�}|j |dd d � tt|ddd��}t|�}t |�dk�s�t �t�||� t�|t|d�� W d Q R X t | ���}t�d�}|jd|dd� t�d�jdjd�} d| d< |�d| � t|| gdd�}|jddgdd�}t|jddgddd��}t|�}t�||� W d Q R X d S )Ni� r# T)�iteratorrb )� chunksizerI � Z df_non_tablez0can only use an iterator or chunksize on a table)�matchr$ )r% rS )rC z{}_2)r" r r rT r )r< )rm )rm rp )r r- rE r r= r/ rV r r. rk rl r �to_hdf�pytest�raises� TypeErrorr �renamer% �select_as_multiple) r0 r1 r# r2 �resultsr@ �path�msgrS rT r3 r3 r4 �test_select_iteratorC sV " r| c C s� d}t | ���}t�dd�}t|d� |�d|� |jd }|jd }|�d�}t�||� d|� d�}|jd|d �}t�||� d |� d�}|jd|d �}t�||� d|� d |� d�}|jd|d �}t�||� W d Q R X t | ���}t�dd�}t|d� |�d|� |jd }|jd }t|jd|d ��}t |�}t�||� d|� d�}t|jd||d ��}t |�}t�||� d |� d�}t|jd||d ��}t |�}t�||� d|� d |� d�}t|jd||d ��}t |�}t�||� W d Q R X d S )Ng ��@i� �Sr# r ���z index >= '�')r& z index <= 'z' & index <= ')rp )r& rp ) r r- rE r r= r! r/ r. rV r ) r0 rp r1 r2 �beg_dt�end_dtr@ r& ry r3 r3 r4 �"test_select_iterator_complete_8014� sN r� c C s� d}t | ���}t�dd�}t|d� |�d|� |jd }|jd }d|� d�}t|jd||d ��}t|�}||j|k } t� | |� d |� d�}t|jd||d ��}t|�}||j|k } t� | |� d|� d |� d�}t|jd||d ��}t|�}||j|k|j|k@ } t� | |� W d Q R X t | ��d}t�dd�}t|d� |�d|� |jd }d |� d�}t|jd||d ��}dt |�k�s�t �W d Q R X d S )Ng ��@i� r} r# r ���z index >= 'r )r& rp z index <= 'z' & index <= 'r~ z index > 'r ) r r- rE r r= r! rV r/ r r. rk rl ) r0 rp r1 r2 r� r� r& ry r@ � rexpectedr3 r3 r4 �&test_select_iterator_non_complete_8014� s< r� c C s� d}t | ���p}t�dd�}t|d� |�d|� |jd }|j|d }d|� d�}t|jd||d ��}t|�}||j|k } t� | |� d |� d�}t|jd||d ��}t |�dks�t �t|�}||j|k } t� | |� d|� d |� d�}t|jd||d ��}t |�dk�st �t|�}||j|k|j|k@ } t� | |� d |� d |� d�}t|jd||d ��}t |�dk�szt �W d Q R X d S ) Ni' i�� r} r# r r z index >= 'r )r& rp z index <= 'z' & index <= 'z' & index >= ') r r- rE r r= r! rV r/ r r. rk rl ) r0 rp r1 r2 r� r� r& ry r@ r� r3 r3 r4 �&test_select_iterator_many_empty_frames s6 r� c C s t �� }t| ���}|jd|dd� |jt|�d }td�}|jjd |ksPt �d}d}|� d||g�}|j |d �d d gf }t � ||� |� d|g�}|j d d �d gf }t � ||� t �� }|� d |� d } tjt| d �� |� d d� W d Q R X W d Q R X d S )N�framer$ )r% r z index>=date�datezcolumns=['A', 'D']z columns=Ar rD Zdf_timez%could not convert string to Timestamp)rr zindex>0)r- rE r r, r! rk r �env�scoperl r/ r> r. r= rt ru � ValueError) r0 r# r1 r� Zcrit1Zcrit2Zcrit3r@ r2 r{ r3 r3 r4 �test_frame_select= s& r� c C s t �� }d|d< d|j|jdd� df< t| ����}|jd|ddgd� |�dd �}|j|j|jd k|jdk@ }t �||� |�dd �}|j|j|jd k|jdk@ }t �||� |�dd �}|j|j|jd k|jdkB }t �||� |�dd �}|j|j|jd k|j|jd k@ |jdkB }t �||� |�dd�}|j|jdk }t �||� d}t j t |d�� |�dd� W d Q R X |�dd�}|jd d �|j � ddg�f }t �||� |�dd�}|j|j|jd k jddgd�}t �||� W d Q R X d S )Nr �stringr r r6 r# r$ )r% rC z index>df.index[3] & string="bar"r z index>df.index[3] & string="foo"z index>df.index[3] | string="bar"z7(index>df.index[3] & index<=df.index[6]) | string="bar"� z string!="bar"z6cannot use an invert condition when passing to numexpr)rr z~(string="bar")z~(columns=['A','B'])r r z(index>df.index[3] & columns in ['A','B'])r" )r- rE r> r! r r, r/ r� r. rt ru �NotImplementedErrorr" � differencerF )r0 r# r1 r@ r2 r{ r3 r3 r4 �test_frame_select_complex` s@ "r� c C s� t ddg����}|\}}tddddddgi�}|j|ddd dgd � t|dd d �}ttj�d d�dgtjdd� t d�D �ddgd�d�}|j|ddd d� t|ddd �}|j � � }t |�} | j ddd �} t�| |� | �� t|ddd �} t�| |� |j } t|ddd �} t�| |� t|ddd �} t�| |� t|ddd �} t�| |� t|dd d �} t�| |� t |�} | j ddd �} t�| |� | j ddd �} t�| |� | j ddd �} t�| |� | j dd d �} t�| |� | �� W d Q R X d S ) Nz params.hdfzhist.hdfr r r r r# �wr$ )�moder% rC zA=[2,3])r& � �datac S s" g | ]}t d �D ] }||f�qqS )rI )rW )rd re �jr3 r3 r4 rf � s z.test_frame_select_complex2.<locals>.<listcomp>rI �l1�l2)r )r" r! )r� r% z l1=[2, 3, 4]zl1=lzl1=indexzl1=selection.indexzl1=selection.index.tolist()zl1=list(selection.index))r r rs r r) r* r+ r � from_tuplesrW r! �tolistr r/ r- r. �close) r0 �paths�pp�hh�paramsZ selection�histr2 �lr1 r@ r! r3 r3 r4 �test_frame_select_complex2� sL r� c C s� t �� }t| ��f}|jd|dd� d}tjt|d�� |�dd� W d Q R X tjt|d�� |�dd� W d Q R X W d Q R X d S )Nr# r$ )r% z unable to collapse Joint Filters)rr z columns=['A'] | columns=['B']z!columns=['A','B'] & columns=['C'])r- rE r r, rt ru r� r/ )r0 r# r1 r{ r3 r3 r4 �test_invalid_filtering� s r� c C st t | ���`}t�� }d|d< d|j|jdd� df< |jd|dgd� |�dd�}||jdk }t�||� |�dd �}||jdk }t�||� |� � }t j |j|jdkdf< |jd |dgd� |�d d �}|t |j� }t�||� d |d < d|j|jdd� d f< |jd |d gd� |�d d�}||j dk }t�||� |�d d�}||j dk }t�||� W d Q R X d S )N�none�x� r � r# )rC zx=nonezx!=nonerT r �int�df3zint=2zint!=2)r r- rE r> r! r= r/ r� r. �copyr) rZ r r� )r0 r1 r# r@ r2 rT r3 r3 r4 �test_string_select� s4 r� c C s� t �� }t �� jdjd�}d|d< t| ���v}d}tjt|d�� |jd ddgd d � W d Q R X |j d |d d gd � |� d|� tjt|d�� |jd ddgd d � W d Q R X tjt|d�� |jd gddgd d � W d Q R X d}tjt |d�� |jd dgddgd d � W d Q R X tjt |d�� |jdgddgd d � W d Q R X tjt dd�� |jd dgddgdd � W d Q R X |� d ddg�}|jd gddgd d �}t � ||� |jd ddgd d �}t � ||� |jd dgddgd d �}t ||gdd�}||jdk|jdk@ }t j ||dd� |jd dgddd �}t ||gdd�}|dd � }t � ||� |� dt jdd�� d }tjt|d�� |jd dgddgd d � W d Q R X W d Q R X d S ) Nz{}_2)r" r r zkeys must be a list/tuple)rr zA>0zB>0rS )r& rm r r )rC rT z!'No object named df3 in the file'r� z!'No object named df4 in the file'rU r )r< r F)Z check_freqzindex>df2.index[4]rI r_ )Znperz,all tables must have exactly the same nrows!)r- rE rw r% r rt ru rv rx r= �KeyErrorr/ r. r r r r� )r0 rS rT r1 r{ r@ r2 r3 r3 r4 �test_select_as_multiple sV r� c C s� t | ���}ttd�td�d�dd�}|d d �t�|d< tj|jd< tdd d gd d d gd�dddgd�}|jd|ddgd� |j ddd�}t � ||� W d Q R X d S )Nr� )rP rQ rR )rN rP r r z13.0z14.0z15.0g @g @g @r r6 rI )r! r# T)rC r! z values>2.0)r& ) r r rW rX rY r) rZ r[ r= r/ r- r. )r0 r1 r# r2 r@ r3 r3 r4 �test_nan_selection_bug_4858] s r� c C s~ t ddddddddgdddd d d d d gd��}||jdk }t| ��$}|jd|ddd� |�dd�}W d Q R X t�||� d S )Nr^ ra r` z test & test�er r r r6 rI r� r� � )r^ r` �testr$ T)r% rC za = "test & test")r r^ r r= r/ r- r. )r0 r# r2 r1 r@ r3 r3 r4 �(test_query_with_nested_special_characterp s r� c C s� t ddddgi�}t| ���}|jd|ddd� d }|�dd |d ���}|jsPt�d }|�dd |d ���}|jddgd d �f }t�||� d}|�dd|d ���}|jdgd d �f }t�||� W d Q R X d S )Nr g~ e��Ag $ e��Ag'1 e��Ar� r$ T)r% rC g� e��AzA < z.4fg� e��AzA > r r zA == ) r r r= r/ �emptyrl r> r- r. )r0 r# r1 �cutoffr@ r2 �exactr3 r3 r4 � test_query_long_float_literal~ s r� c C s t ddgtddd�ddgddgd�d d d d gd �}t| ����}|jd|ddd� td�}|jddd�}|jdgd d �f }t�||� �xldD �]b}xjddtd�t � dd�gD ]N}d|� d�}d|� dt |�� d�} t j t| d �� |jd|d� W d Q R X q�W d }xJd D ]B} | � d |� d�}d } t j t| d �� |jd|d� W d Q R X �q W x�td!d"dgd d d g�D ]�\}} | � d |� d�}|jd|d�}|d#k�r�|jd$gd d �f }n2|d%k�r�|jdgd d �f }n|jg d d �f }t�||� �qhW q�W W d Q R X d S )&Nz 2014-01-01z 2014-01-02r )r8 g�������?g333333�?r )r� � real_date�floatr� r� r� r� r� )r" r� r$ T)r% rC zreal_date > ts)r& )�<�>z==g������ @r( zdate z vzCannot compare z of type z to string column)rr r^ )r� r� r� � z could not convert string to �1z1.1z==r r� )r r r r= r r/ r> r- r. �pdZ Timedelta�typert ru rv r� �zip) r0 r# r1 rH r@ r2 �opr\ �queryr{ �colr3 r3 r4 � test_query_compare_column_type� sD r� r& r� r3 )Nc C sb t dddg�}td��B}t|��.}|�d|d� t|d| d�}t�||� W d Q R X W d Q R X d S )Nr r r zempty_where.h5r# �t)r& )r r r r, r r- r. )r& r# rz r1 r@ r3 r3 r4 �test_select_empty_where� s r� )3�warningsr �numpyr) rt Zpandas._libs.tslibsr Zpandasr� r r r r r r r- r r r r r Z pandas.tests.io.pytables.commonr r r Zpandas.io.pytablesr �markZsingleZ pytestmarkr5 rA rG r] rn r| r� r� r� r� r� r� r� r� r� r� r� r� r� Z parametrizer� r3 r3 r3 r4 �<module> s8 4 ?)k2EH49#6A+H3
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка