Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_query_eval.cpython-37.pyc
Назад
B �5�g�� � @ sh d dl mZ d dlZd dlZd dlZd dlm mZ d dl Z d dl m Z m Z mZmZmZ d dlmZ d dlmZ dZdejde jd�fZejed d � d �d d � �Zejedd � d �dd� �Z dd� Z G dd� d�Z G dd� d�Z G dd� d�Z e jG dd� d��Z!e jG dd � d e!��Z"G d d � d e!�Z#G d d � d e"�Z$G d!d"� d"�Z%G d#d$� d$�Z&G d%d&� d&�Z'dS )'� )�StringION)� DataFrame�Index� MultiIndex�Series� date_range)�NUMEXPR_INSTALLED)�python�pandasr �numexpr)Zmarksc C s | S )N� )�xr r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_query_eval.py�<lambda> � r )�paramsZidsc C s | j S )N)�param)�requestr r r �parser s r c C s | S )Nr )r r r r r r c C s | j S )N)r )r r r r �engine s r c C s | dkr t �dt| �� �� d S )Nr z cannot evaluate with parser )�pytest�skip�repr)r r r r �skip_if_no_pandas_parser" s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � TestCompatc C s: t ddddgi�| _| j| jjdk | _| jjd | _d S )N�A� � � r )r �dfr � expected1� expected2)�self�methodr r r � setup_method( s zTestCompat.setup_methodc C s> | j }|�d�}t�|| j� |�d�}tj|| jdd� d S )NzA>0zA+1F)� check_names)r �query�tm�assert_frame_equalr �eval�assert_series_equalr! )r" r �resultr r r �test_query_default- s z TestCompat.test_query_defaultc C sF | j }|jdd d�}t�|| j� |jdd d�}tj|| jdd� d S )NzA>0)r zA+1F)r% )r r& r' r( r r) r* r! )r" r r+ r r r �test_query_None7 s zTestCompat.test_query_Nonec C sF | j }|jddd�}t�|| j� |jddd�}tj|| jdd� d S )NzA>0r )r zA+1F)r% )r r& r' r( r r) r* r! )r" r r+ r r r �test_query_python? s z TestCompat.test_query_pythonc C s� | j }trH|jddd�}t�|| j� |jddd�}tj|| jdd� nTd}t j t |d�� |jddd� W d Q R X t j t |d�� |jddd� W d Q R X d S ) NzA>0r )r zA+1F)r% z~'numexpr' is not installed or an unsupported version. Cannot use engine='numexpr' for query/eval if 'numexpr' is not installed)�match) r r r& r' r( r r) r* r! r �raises� ImportError)r" r r+ �msgr r r �test_query_numexprG s z TestCompat.test_query_numexprN)�__name__� __module__� __qualname__r$ r, r- r. r3 r r r r r ' s r c @ sd e Zd Zej�dddg�ej�dddddg�d d � ��Zd d � Zd d� Zdd� Z dd� Z dd� Z dS )�TestDataFrameEval�n� i� z op_str,op,rop)�+�__add__�__radd__)�-�__sub__�__rsub__)�*�__mul__�__rmul__)�/� __truediv__� __rtruediv__c C s� t dt|�td�d�}d|jd< |�� }t t�|j|��|d�td�d�}t d|� d ��}t d |� d ��} t � | |� |d kr�t ||�|�} t � | |� n"|d kr�t ||�|�} t � | |� d S ) Nr �abcd)�index�columnsr r ���)rH zbase z dfzm )r: r@ )r= rC ) r �range�list�ilocZmean�npZtile�valuesZreshaper) r' r( �getattr) r" Zop_str�opZropr8 r �m�base�expectedr+ r r r �test_ops^ s zTestDataFrameEval.test_opsc C sd t dtj�d�i�}tj|jdd�< dt�|jdd� � }dt�|� jdd� }t�||� d S )Nr i�a r � r � ) r rM �random�randn�nanrL �isnanr' r( )r" r rS r+ r r r � test_dataframe_sub_numexpr_path� s z1TestDataFrameEval.test_dataframe_sub_numexpr_pathc C sn t dddgdddgd��}d}tjt|d�� |�d d � � W d Q R X tjt|d�� |�d � W d Q R X d S ) Nr r r �a�b)r �Bz%expr must be a string to be evaluated)r/ c S s | j dkS )Nr] )r^ )r r r r r � r z6TestDataFrameEval.test_query_non_str.<locals>.<lambda>�o )r r r0 � ValueErrorr& )r" r r2 r r r �test_query_non_str� s z$TestDataFrameEval.test_query_non_strc C s> t ddddgi�}d}tjt|d�� |�d� W d Q R X d S )Nr r r r z expr cannot be an empty string)r/ � )r r r0 r` r& )r" r r2 r r r �test_query_empty_string� s z)TestDataFrameEval.test_query_empty_stringc C sz t tj�dd�td�d�}ddi}ddi}|jd||gd �|d |d ksPt�tjd||gd �|d |d ksvt�d S ) N� r �ab)rH r\ r r] za + b)� resolvers)r rM rW rX rK r) �AssertionError�pd)r" r Zdict1Zdict2r r r �test_eval_resolvers_as_list� s &z-TestDataFrameEval.test_eval_resolvers_as_listc C s@ t dddgi�}|�d�}t ddgddgd��}t�||� d S )N�a1�Y�Nzc = ((a1 == 'Y') & True)TF)rj �c)r r) r' r( )r" r �resrS r r r � test_eval_object_dtype_binop� s z.TestDataFrameEval.test_eval_object_dtype_binopN) r4 r5 r6 r �mark� parametrizerT r[ ra rc ri ro r r r r r7 [ s r7 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestDataFrameQueryWithMultiIndexc C s$ t |� tjjddgdd�}tjjddgdd�}tj||gddgd �}ttj�dd �|d �}t|j � d�j |dd �}|j d ||d�}|j d||d�} ||dk } t �|| � t �| | � |j d||d�}|j d||d�} ||dk } t �|| � t �| | � |j d||d�}|j d||d�} ||�dg� } t �|| � t �| | � |j d||d�}|j d||d�} ||�dg� } t �|| � t �| | � |j d||d�}|j d||d�} ||�dg� } t �|| � t �| | � |j d||d�}|j d||d�} ||�dg� } t �|| � t �| | � d S )N�red�greenrd )�size�eggs�ham�colorZfood)�namesr )rG )rG �namezcolor == "red")r r z"red" == colorzcolor != "red"z"red" != colorzcolor == ["red"]z["red"] == colorzcolor != ["red"]z["red"] != colorz["red"] in colorz"red" in colorz["red"] not in colorz"red" not in color)r rM rW �choicer � from_arraysr rX r rG �get_level_valuesrN r& r' r( �isin) r" r r r\ r] rG r �ind�res1�res2�expr r r � test_query_with_named_multiindex� sJ zATestDataFrameQueryWithMultiIndex.test_query_with_named_multiindexc C s� t |� tjjddgdd�}tjjddgdd�}t�||g�}ttj�dd�|d�}t|j � d �j |d�}|j d ||d �}|j d ||d �} ||dk } t �|| � t �| | � |j d ||d �}|j d||d �} ||dk } t �|| � t �| | � |j d||d �}|j d||d �} ||�dg� } t �|| � t �| | � |j d||d �}|j d||d �} ||�dg� } t �|| � t �| | � |j d||d �}|j d||d �} ||�dg� } t �|| � t �| | � |j d||d �}|j d||d �} ||�dg� } t �|| � t �| | � t|j � d�j |d�}|j d||d �}|j d||d �} ||dk } t �|| � t �| | � |j d||d �}|j d||d �} ||dk } t �|| � t �| | � |j d ||d �}|j d ||d �} ||�dg� } t �|| � t �| | � |j d ||d �}|j d ||d �} ||�dg� } t �|| � t �| | � |j d ||d �}|j d!||d �} ||�dg� } t �|| � t �| | � |j d"||d �}|j d#||d �} ||�dg� } t �|| � t �| | � d S )$Nrs rt rd )ru rv rw r )rG r zilevel_0 == "red")r r z"red" == ilevel_0zilevel_0 != "red"z"red" != ilevel_0zilevel_0 == ["red"]z["red"] == ilevel_0zilevel_0 != ["red"]z["red"] != ilevel_0z["red"] in ilevel_0z"red" in ilevel_0z["red"] not in ilevel_0z"red" not in ilevel_0r zilevel_1 == "eggs"z"eggs" == ilevel_1zilevel_1 != "eggs"z"eggs" != ilevel_1zilevel_1 == ["eggs"]z["eggs"] == ilevel_1zilevel_1 != ["eggs"]z["eggs"] != ilevel_1z["eggs"] in ilevel_1z"eggs" in ilevel_1z["eggs"] not in ilevel_1z"eggs" not in ilevel_1)r rM rW r{ r r| r rX r rG r} rN r& r' r( r~ ) r" r r r\ r] rG r r r� r� r� r r r �"test_query_with_unnamed_multiindex� s� zCTestDataFrameQueryWithMultiIndex.test_query_with_unnamed_multiindexc C sT t |� tjjddgdd�}t�d�}t�||g�}d dg|_ttj� dd�|d�}|j d||d �}t |j � d�j|dd �}||d k } t�|| � |j d ||d �}t |j � d�j|dd �}||d k } t�|| � |j d ||d �}t |j � d�j|d�}||dk } t�|| � |j d||d �}t |j � d�j|d�}||dk } t�|| � d S )Nrs rt rd )ru Zratingr )rG z rating == 1)r r )rG rz r z rating != 1zilevel_0 == "red"r zilevel_0 != "red")r rM rW r{ �aranger r| ry r rX r& r rG r} rN r' r( ) r" r r r\ r] rG r rn r r� r r r �*test_query_with_partially_named_multiindex< s0 zKTestDataFrameQueryWithMultiIndex.test_query_with_partially_named_multiindexc C s� t jdddddgd�}|�� }dd� }|j�� }|j|||jd�||jd�|d �}xV|�� D ]J\}}t|t�r�|� || �s�t �q\t|t �r�t � ||| � q\t d ��q\W d S ) Nrd r r �spamrv )Z r_idx_nlevelsZ r_idx_namesc S s | � |�}|�� }| |_|S )N)r} � to_seriesrG )�mi�levelZ level_values�sr r r r� a s z]TestDataFrameQueryWithMultiIndex.test_query_multiindex_get_index_resolvers.<locals>.to_series)rG rH r� rv ZC0z object must be a Series or Index) r' ZmakeCustomDataframeZ_get_index_resolversrH r� rG �items� isinstancer �is_rg r r* )r" r rf r� Z col_seriesrS �k�vr r r �)test_query_multiindex_get_index_resolvers[ s zJTestDataFrameQueryWithMultiIndex.test_query_multiindex_get_index_resolversN)r4 r5 r6 r� r� r� r� r r r r rr � s 2X rr c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5S )6� TestDataFrameQueryNumExprPandasc C s d| _ d| _d S )Nr r )r r )�clsr r r � setup_classz s z+TestDataFrameQueryNumExprPandas.setup_classc C s | ` | `d S )N)r r )r� r r r �teardown_class s z.TestDataFrameQueryNumExprPandas.teardown_classc C s� | j | j }}t|� ttj�dd��}tddd�|d< tddd�|d< tddd�|d < |jd ||d �}||j d k d |j k @ }t � ||� d S ) NrU r z1/1/2012)�periods�dates1z1/1/2013�dates2z1/1/2014�dates3z"@df.dates1 < 20130101 < @df.dates3)r r �20130101) r r r r rM rW rX r r& r� r� r' r( )r" r r r rn �expecr r r �%test_date_query_with_attribute_access� s zETestDataFrameQueryNumExprPandas.test_date_query_with_attribute_accessc C s� | j | j }}ttj�dd��}tddd�|d< tddd�|d< tddd�|d < |jd ||d �}||jd k d |j k @ }t � ||� d S ) NrU r z1/1/2012)r� r� z1/1/2013r� z1/1/2014r� zdates1 < 20130101 < dates3)r r r� ) r r r rM rW rX r r& r� r� r' r( )r" r r r rn r� r r r �#test_date_query_no_attribute_access� s zCTestDataFrameQueryNumExprPandas.test_date_query_no_attribute_accessc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< td|d�|d < tj|j tj� |�d kdf< tj|j tj� |�d kd f< |j d ||d �}||j d k d |j k @ }t�||� d S )Nrd r z1/1/2012)r� r� z1/1/2013r� z1/1/2014r� g �?zdates1 < 20130101 < dates3)r r r� )r r r rM rW rX r rh �NaT�loc�randr& r� r� r' r( )r" r r r8 r rn r� r r r �test_date_query_with_NaT� s z8TestDataFrameQueryNumExprPandas.test_date_query_with_NaTc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< |jdddd �}|d ks`t�|j d ||d �}||j d k d |j k @ }t � ||� d S ) Nrd r z1/1/2012)r� r� z1/1/2014r� T)�inplace�dropzindex < 20130101 < dates3)r r r� )r r r rM rW rX r � set_indexrg r& rG r� r' r( )r" r r r8 r � return_valuern r� r r r �test_date_index_query� s z5TestDataFrameQueryNumExprPandas.test_date_index_queryc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< tj|j d< |j dd d d �}|d kslt �|j d ||d �}||j d k d |jk @ }t�||� d S )Nrd r z1/1/2012)r� r� z1/1/2014r� )r r T)r� r� zindex < 20130101 < dates3)r r r� )r r r rM rW rX r rh r� rL r� rg r& rG r� r' r( )r" r r r8 r r� rn r� r r r � test_date_index_query_with_NaT� s z>TestDataFrameQueryNumExprPandas.test_date_index_query_with_NaTc C s� | j | j }}d}i }td|d�|d< td|d�|d< t|�}tj|jtj� |�dkdf< |j dddd �}|d ksvt �|j d ||d �}||j �� d k d |jk @ }t�||� d S ) Nrd z1/1/2012)r� r� z1/1/2014r� g �?T)r� r� zdates1 < 20130101 < dates3)r r r� )r r r r rh r� r� rM rW r� r� rg r& rG r� r� r' r( ) r" r r r8 �dr r� rn r� r r r �)test_date_index_query_with_NaT_duplicates� s zITestDataFrameQueryNumExprPandas.test_date_index_query_with_NaT_duplicatesc C s� | j | j }}d}ttd|d�t�|�d��}|jd||d�}t|�dksNt�|jd||d�}t � ||� d }x>d D ]6}t j t |d �� |jd |� d �||d� W d Q R X qtW d S )Nrd z1/1/2012)r� )�datesZnondatezdates == nondate)r r r zdates != nondatez=Invalid comparison between dtype=datetime64\[ns\] and ndarray)�<�>z<=z>=)r/ zdates z nondate)r r r r rM r� r& �lenrg r' r( r r0 � TypeError)r" r r r8 r r+ r2 rP r r r � test_date_query_with_non_date� s z=TestDataFrameQueryNumExprPandas.test_date_query_with_non_datec C s` | j | j }}ttd�tdd�tdd�d��}d}tjt|d�� |jd ||d � W d Q R X d S ) Nrd r � r9 � )�ir: �rzinvalid syntax)r/ zi - +)r r )r r r rJ r r0 � SyntaxErrorr& )r" r r r r2 r r r �test_query_syntax_error� s z7TestDataFrameQueryNumExprPandas.test_query_syntax_errorc C s� ddl m} | j| j }}t|� ttj�dd�t d�d�}d\}}|j d||d �}||j |j k }t �||� |j d ||d �}|||j k }t �||� tj|d d �� |j d ||d � W d Q R X tj|dd �� |j d||d � W d Q R X d S )Nr )�UndefinedVariableError� r re )rH )r r za > b)r r z@a > bz!local variable 'c' is not defined)r/ z @a > b > @czname 'c' is not definedz @a > b > c)�pandas.core.computation.opsr� r r r r rM rW rX rK r& r\ r] r' r( r r0 ) r" r� r r r r\ r] rn rS r r r �test_query_scope� s z0TestDataFrameQueryNumExprPandas.test_query_scopec C sp ddl m} | j| j }}d }}ttjj||dfd�td�d�}t j |dd �� |j d ||d � W d Q R X d S ) Nr )r� rd r )ru �abc)rH zname 'sin' is not defined)r/ zsin > 5)r r ) r� r� r r r rM rW �randintrK r r0 r& )r" r� r r r8 rQ r r r r � test_query_doesnt_pickup_local s z>TestDataFrameQueryNumExprPandas.test_query_doesnt_pickup_localc C s| ddl m} | j| j }}d }}ttjj||dfd�td�d�}d|j _ d }t j ||d �� |j d ||d � W d Q R X d S ) Nr )�NumExprClobberingErrorrd r )ru r� )rH �sinzVariables in expression.+)r/ zsin > 5)r r )Z pandas.core.computation.enginesr� r r r rM rW r� rK rG rz r r0 r& )r" r� r r r8 rQ r r2 r r r �test_query_builtin s z2TestDataFrameQueryNumExprPandas.test_query_builtinc C s� | j | j }}ttj�dd�dddgd�}t�|jd||d�||j |j k � t�|jd ||d�||j |j |j |j k � d S ) Nrd r r\ r] rm )rH za < b)r r z a + b > b * c) r r r rM rW rX r' r( r& r\ r] rm )r" r r r r r r � test_query s z*TestDataFrameQueryNumExprPandas.test_queryc C s� | j | j }}ttjjddd�ttd�dd�dddgd �}|jd ||d �}||j d k |j |j k @ }t � ||� |jd ||d �}||j |j k }t � ||� d S )Nrd )rd r )ru Zblob)rz r\ r] rm )rG rH z(blob < 5) & (a < b))r r rU zblob < b)r r r rM rW r� r rJ r& rG r\ r] r' r( )r" r r r rn r� r r r �test_query_index_with_name' s z:TestDataFrameQueryNumExprPandas.test_query_index_with_namec C s� | j | j }}ttjjddd�td�dddgd�}|jd||d �}||j|j k }t � ||� |jd ||d �}||jd k }t � ||� d S ) Nrd )rd r )ru r\ r] rm )rG rH z index < b)r r z index < 5rU ) r r r rM rW r� rJ r& rG r] r' r( )r" r r r rn r� r r r � test_query_index_without_name7 s z=TestDataFrameQueryNumExprPandas.test_query_index_without_namec C s� | j }| j}t|� ttj�dd��}ttj�dd��}||dk|dk@ }|jd||d�}t� ||� t j d||d�}t� ||� t j d||d�}||dk|dk@ ||dk dk@ }t� ||� t j d||d�}|jd ||d�}t� ||� d S ) NrU r r z(@df > 0) & (@df2 > 0))r r zdf[df > 0 and df2 > 0]z)df[df > 0 and df2 > 0 and df[df > 0] > 0]zdf[(df>0) & (df2>0)]z(@df>0) & (@df2>0)) r r r r rM rW rX r& r' r( rh r) )r" r r r �df2rS r+ r r r �test_nested_scopeI s" $ z1TestDataFrameQueryNumExprPandas.test_nested_scopec C sP ddl m} ttj�dd��}tj|dd�� |jd| j | j d� W d Q R X d S ) Nr )r� rU r zname 'df' is not defined)r/ zdf > 0)r r ) r� r� r rM rW rX r r0 r& r r )r" r� r r r r �*test_nested_raises_on_local_self_referencec s zJTestDataFrameQueryNumExprPandas.test_nested_raises_on_local_self_referencec C s� t | j� | j| j }}ttj�dd�td�d�}d}||j|k }|j d||d�}t � ||� ||j|j k }|j d||d�}t � ||� d S ) N�d rd Z abcdefghij)rH r za < @b)r r za < b) r r r r rM rW rX rK r\ r& r' r( r] )r" r r r r] �expectr+ r r r �test_local_syntaxl s z1TestDataFrameQueryNumExprPandas.test_local_syntaxc C s� t | j� | j| j }}td�}ttj�dt|��|d�}|j d||d�}|j |j k |j |j k @ |j � |j � @ |j � |j � @ }|| }t�||� d S )Nr� r� )rH z!a < b < c and a not in b not in c)r r )r r r rK r rM rW rX r� r& r\ r] rm r~ r' r( )r" r r �colsr rn r r� r r r �test_chained_cmp_and_inz s 8z7TestDataFrameQueryNumExprPandas.test_chained_cmp_and_inc C s� | j | j }}t|� ttjjddd�dd�}ttjjddd�dd�}t||d��}|j|j d � |� }|j d ||d �}t � ||� ttjjddd�dd�}|j|d � |� }|j d ||d �}t � ||� d S ) Nr � )ru r\ )rz rd r] )r\ r] r z b - 1 in a)r r z @b - 1 in a)r r r r rM rW r� r r� r] r~ r& r' r( )r" r r r\ r] r rS r+ r r r �test_local_variable_with_in� s z;TestDataFrameQueryNumExprPandas.test_local_variable_with_inc C s` | j | j }}t|� d}tdddddddgi�}|jd||d�}||jdk }t�||� d S )Nr r\ r] z@cz a == "@c")r r )r r r r r& r\ r' r( )r" r r rm r r+ rS r r r �test_at_inside_string� s z5TestDataFrameQueryNumExprPandas.test_at_inside_stringc C sj ddl m} | j| j }}t|� ttj�dd�t d�d�}t j |dd�� |j d ||d � W d Q R X d S ) Nr )r� rd r re )rH z!local variable 'c' is not defined)r/ za == @c)r r ) r� r� r r r r rM rW r� rK r r0 r& )r" r� r r r r r r �test_query_undefined_local� s z:TestDataFrameQueryNumExprPandas.test_query_undefined_localc C s: d}t jddd� }t|t j�|j�d��}d|j_|jd| j | j d�}||d dk }t � ||� t|t j�|j�d��}|jd | j | j d�}|j |j|jdk }t � ||� t|t j�|j�d ��}d |j_|jd | j | j d�}||jdk }t � ||� |jd| j | j d�}|j |j|jdk }t � ||� d S ) Nr r� �e )rG r] rG z index > 5)r r rU z ilevel_0 > 5)r\ r] r\ za > 5)rM Zr_r rW rX ru rG rz r& r r r' r( r� r\ )r" r8 r\ r r+ rS r r r �:test_index_resolvers_come_after_columns_with_the_same_name� s& zZTestDataFrameQueryNumExprPandas.test_index_resolvers_come_after_columns_with_the_same_namec C s� d}t tj�|�tj�|�d��}tj|jd d d�df< tjtjd�}xR|� � D ]F\}}d|� d�}|||j tj� }|j || j | j d�}t�||� qPW d S ) Nrd )r\ r] r r )z==z!=za z inf)r r )r rM rW r� �infr� �operator�eq�ner� r\ r& r r r' r( ) r" r8 r r� rP �f�qrS r+ r r r �test_inf� s z(TestDataFrameQueryNumExprPandas.test_infc C sf |}t ddd|dd�}t|d�}t|d�}|�d�}t�||� t|�}|�� �d�}t�||� d S )Nz 2019-01-01Z1drd �time)�start�freqr� �tzrz )rG z "2018-01-03 00:00:00+00" < time)r r r& r' r( Z reset_index)r" Ztz_aware_fixturer� Zdf_indexrS r r+ r r r � test_check_tz_aware_index_query� s z?TestDataFrameQueryNumExprPandas.test_check_tz_aware_index_queryN) r4 r5 r6 � classmethodr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� x s4 r� c sP e Zd Ze� fdd��Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z � Z S )� TestDataFrameQueryNumExprPythonc s t � �� d| _d| _d S )Nr r )�superr� r r )r� )� __class__r r r� � s z+TestDataFrameQueryNumExprPython.setup_classc C s� | j | j }}ttj�dd��}tddd�|d< tddd�|d< tddd�|d < |jd ||d �}||jd k d |j k @ }t � ||� d S ) NrU r z1/1/2012)r� r� z1/1/2013r� z1/1/2014r� z)(dates1 < 20130101) & (20130101 < dates3))r r r� ) r r r rM rW rX r r& r� r� r' r( )r" r r r rn r� r r r r� � s zCTestDataFrameQueryNumExprPython.test_date_query_no_attribute_accessc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< td|d�|d < tj|j tj� |�d kdf< tj|j tj� |�d kd f< |j d ||d �}||j d k d |j k @ }t�||� d S )Nrd r z1/1/2012)r� r� z1/1/2013r� z1/1/2014r� g �?z)(dates1 < 20130101) & (20130101 < dates3))r r r� )r r r rM rW rX r rh r� r� r� r& r� r� r' r( )r" r r r8 r rn r� r r r r� � s z8TestDataFrameQueryNumExprPython.test_date_query_with_NaTc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< |jdddd �}|d ks`t�|j d ||d �}||j d k d |j k @ }t � ||� d S ) Nrd r z1/1/2012)r� r� z1/1/2014r� T)r� r� z((index < 20130101) & (20130101 < dates3))r r r� )r r r rM rW rX r r� rg r& rG r� r' r( )r" r r r8 r r� rn r� r r r r� s z5TestDataFrameQueryNumExprPython.test_date_index_queryc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< tj|j d< |j dd d d �}|d kslt �|j d ||d �}||j d k d |jk @ }t�||� d S )Nrd r z1/1/2012)r� r� z1/1/2014r� )r r T)r� r� z((index < 20130101) & (20130101 < dates3))r r r� )r r r rM rW rX r rh r� rL r� rg r& rG r� r' r( )r" r r r8 r r� rn r� r r r r� s z>TestDataFrameQueryNumExprPython.test_date_index_query_with_NaTc C s� | j | j }}d}ttj�|d��}td|d�|d< td|d�|d< tj|j tj� |�dkdf< |j dd d d �}|d ks|t �d }t jt|d �� |jd ||d� W d Q R X d S )Nrd r z1/1/2012)r� r� z1/1/2014r� g �?T)r� r� z"'BoolOp' nodes are not implemented)r/ zindex < 20130101 < dates3)r r )r r r rM rW rX r rh r� r� r� r� rg r r0 �NotImplementedErrorr& )r" r r r8 r r� r2 r r r r� s zITestDataFrameQueryNumExprPython.test_date_index_query_with_NaT_duplicatesc C s( ddl m} | j}| j}d}tjd||d�}|dks8t�ttj � dd��}ttj � dd��}d }t j t |d �� |jd ||d� W d Q R X t j |d d �� |jd ||d� W d Q R X ||dk|dk@ } tjd||d�}t�| |� ||dk|dk@ ||dk dk@ } tjd||d�}t�| |� d S )Nr )r� r zx + 1)r r r rU r z5The '@' prefix is only supported by the pandas parser)r/ z(@df>0) & (@df2>0)zname 'df' is not definedz(df>0) & (df2>0)zdf[(df > 0) & (df2 > 0)]z+df[(df > 0) & (df2 > 0) & (df[df > 0] > 0)])r� r� r r rh r) rg r rM rW rX r r0 r� r& r' r( ) r" r� r r r r+ r r� r2 rS r r r r� , s( $ z1TestDataFrameQueryNumExprPython.test_nested_scope) r4 r5 r6 r� r� r� r� r� r� r� r� � __classcell__r r )r� r r� � s r� c s( e Zd Ze� fdd��Zdd� Z� ZS )� TestDataFrameQueryPythonPandasc s t � �� d| _d| _d S )Nr r )r� r� r r )r� )r� r r r� M s z*TestDataFrameQueryPythonPandas.setup_classc C sl | j | j }}d }}ttjj||dfd�td�d�}d|j_||jdk }|j d||d �}t � ||� d S ) Nrd r )ru r� )rH r� rU zsin > 5)r r ) r r r rM rW r� rK rG rz r& r' r( )r" r r r8 rQ r rS r+ r r r r� S s z1TestDataFrameQueryPythonPandas.test_query_builtin)r4 r5 r6 r� r� r� r� r r )r� r r� L s r� c s( e Zd Ze� fdd��Zdd� Z� ZS )� TestDataFrameQueryPythonPythonc s t � �� d | _| _d S )Nr )r� r� r r )r� )r� r r r� ` s z*TestDataFrameQueryPythonPython.setup_classc C sl | j | j }}d }}ttjj||dfd�td�d�}d|j_||jdk }|j d||d �}t � ||� d S ) Nrd r )ru r� )rH r� rU zsin > 5)r r ) r r r rM rW r� rK rG rz r& r' r( )r" r r r8 rQ r rS r+ r r r r� e s z1TestDataFrameQueryPythonPython.test_query_builtin)r4 r5 r6 r� r� r� r� r r )r� r r� _ s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestDataFrameQueryStringsc C s� t tj�dd�dgd�}ttd��|d< ||jdk }|dkr�d}d }|gd |gd }|d d d � }d \} } d | g| g } d } x`t|| |�D ]P\}} }|� d| � d|� �}tj t | d�� |j |||d|jid� W d Q R X q�W n�|j d||d�}t � ||� |j d||d�}t � ||� t � |||j�dg� � ||jdk }|j d||d�}t � ||� |j d||d�}t � ||� t � |||j�dg� � d S )Nrd r r] )rH � aabbccddee�stringsr\ r z"a"r rI )z==z!=z$'(Not)?In' nodes are not implemented� )r/ )r r Z local_dictz"a" == strings)r r zstrings == "a"zstrings != "a"z"a" != strings)r rM rW rX r rK r� �zipr r0 r� r& r' r( r~ )r" r r r r� �col�lst�lhs�rhsr� r� �opsr2 rP �exrn r r r �test_str_query_methodr s< z/TestDataFrameQueryStrings.test_str_query_methodc C sp t tj�dd�dgd�}ttd��|d< ||j�ddg� }|dkr�d}d }|gd |gd }|d d d � }d \} } d | g| g } d } x�t|| |�D ]H\}} }|� d| � d|� �}t j t | d�� |j |||d� W d Q R X q�W n�|j d||d�}t �||� |j d||d�}t �||� ||j�ddg� }|j d||d�}t �||� |j d||d�}t �||� d S )Nrd r r] )rH r� r� r\ r z ["a", "b"]r rI )z==z!=z$'(Not)?In' nodes are not implementedr� )r/ )r r zstrings == ["a", "b"]z["a", "b"] == stringszstrings != ["a", "b"]z["a", "b"] != strings)r rM rW rX r rK r� r~ r� r r0 r� r& r' r( )r" r r r r� r� r� r� r� r� r� r� r2 rP r� rn r r r �test_str_list_query_method� s0 z4TestDataFrameQueryStrings.test_str_list_query_methodc C s t td�td�tjjddd�tjjddd�d��}|dkr�|jd ||d �}||j�|j� }t � ||� |jd ||d �}||j�|j�|j |j k @ }t � ||� n\d }t jt|d �� |jd ||d � W d Q R X d}t jt|d �� |jd ||d � W d Q R X d S )N� aaaabbbbcccc� aabbccddeeffrU � )ru � )r\ r] rm r� r za in b)r r za in b and c < dz$'(Not)?In' nodes are not implemented)r/ z"'BoolOp' nodes are not implemented)r rK rM rW r� r& r\ r~ r] r' r( rm r� r r0 r� )r" r r r rn r� r2 r r r � test_query_with_string_columns� s$ z8TestDataFrameQueryStrings.test_query_with_string_columnsc C s� t td�td�tjjddd�tjjddd�d��}|jd||d �}||j|jk }t� ||� |jd ||d �}||j|jk }t� ||� d S ) Nr� r� rU r� )ru r� )r\ r] rm r� za == b)r r za != b) r rK rM rW r� r& r\ r] r' r( )r" r r r rn r� r r r �test_object_array_eq_ne� s z1TestDataFrameQueryStrings.test_object_array_eq_nec C sR t |� d}tjt|�dddgd�}||jdk }|jd||d�}t�||� d S ) NaI id event timestamp 1 "page 1 load" 1/1/2014 0:00:01 1 "page 1 exit" 1/1/2014 0:00:31 2 "page 2 load" 1/1/2014 0:01:01 2 "page 2 exit" 1/1/2014 0:01:31 3 "page 3 load" 1/1/2014 0:02:01 3 "page 3 exit" 1/1/2014 0:02:31 4 "page 1 load" 2/1/2014 1:00:01 4 "page 1 exit" 2/1/2014 1:00:31 5 "page 2 load" 2/1/2014 1:01:01 5 "page 2 exit" 2/1/2014 1:01:31 6 "page 3 load" 2/1/2014 1:02:01 6 "page 3 exit" 2/1/2014 1:02:31 z\s{2,}r � timestamp)�sepr Z parse_datesz "page 1 load"z'"page 1 load"' in event)r r )r rh Zread_csvr �eventr& r' r( )r" r r �rawr rS rn r r r � test_query_with_nested_strings� s z8TestDataFrameQueryStrings.test_query_with_nested_stringsc C sP t |� tdddgdddgd��}|jd||d �}||jdk }t�||� d S ) Nr\ r] z test & testr r r )r\ r] za == "test & test")r r )r r r& r\ r' r( )r" r r r rn r� r r r �(test_query_with_nested_special_character s zBTestDataFrameQueryStrings.test_query_with_nested_special_characterc C s� t tj�td�d��}t t�|j��}t||d��}tj tj tj tj d�}xH|� � D ]<\}}|jd|� d�||d�} |||jd� } t�| | � qTW d S ) NZabcder� )�Xrk )r� r� z<=z>=zX z "d")r r r� )r rM rW r{ rK r� ru r r� �lt�gt�le�ger� r& r� r' r( ) r" r r r\ r] r r� rP �funcrn rS r r r � test_query_lex_compare_strings s z8TestDataFrameQueryStrings.test_query_lex_compare_stringsc C s^ d}t jjddt|�fd��t�}t||d�}|jd||d�}||j|j @ }t � ||� d S )N)�bidZbidsize�askZasksizer r )ru )rH z bid & ask)r r ) rM rW r� r� Zastype�boolr r& r r r' r( )r" r r rH �datar rn rS r r r �"test_query_single_element_booleans s z<TestDataFrameQueryStrings.test_query_single_element_booleansc C sX t |� tddddgddddgd��}||jdk }d}|jd||d �}t�||� d S ) NzBUD USzIBM USg�����l[@g�G�zn[@g������f@g33333�f@)�SymbolZPricezSymbol == @symb)r r )r r r r& r' r( )r" r r r �eZsymbr� r r r �!test_query_string_scalar_variable s z;TestDataFrameQueryStrings.test_query_string_scalar_variableN) r4 r5 r6 r� r� r� r� r� r� r r r r r r r r� q s )# r� c @ sJ e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d d g�dd� �Z dS )�TestDataFrameEvalWithFramec C s t tj�dd�td�d�| _d S )Nrd r r� )rH )r rM rW rX rK �frame)r" r# r r r r$ + s z'TestDataFrameEvalWithFrame.setup_methodc C s | ` d S )N)r )r" r# r r r �teardown_method. s z*TestDataFrameEvalWithFrame.teardown_methodc C s2 | j jd||d�}| j j| j j }t�||� d S )Nza + b)r r )r r) r\ r] r' r* )r" r r rn r� r r r �test_simple_expr1 s z+TestDataFrameEvalWithFrame.test_simple_exprc C s> | j jd||d�}| j j| j jdk | j j }t�||� d S )Nz a[a < 1] + b)r r r )r r) r\ r] r' r* )r" r r rn r� r r r �test_bool_arith_expr6 s z/TestDataFrameEvalWithFrame.test_bool_arith_exprrP r: r= r@ rC c C sP t ddgddgd��}d}tjt|d�� |jd|� d �||d � W d Q R X d S ) Nr r rm r� )r\ r] z3unsupported operand type\(s\) for .+: '.+' and '.+')r/ za z b)r r )r r r0 r� r) )r" r r rP r r2 r r r �%test_invalid_type_for_operator_raises; s z@TestDataFrameEvalWithFrame.test_invalid_type_for_operator_raisesN) r4 r5 r6 r$ r r r r rp rq r r r r r r * s r c @ s� e Zd Zejdd�dd� �Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�!TestDataFrameQueryBacktickQuoting�class)�scopec c s� t dddgdddgdddgdddgdd d gd dd gdddgddd gdddgd d dgdddgdddgdddgdddgd d dgddd gdddgdddgd ddgdddgdddgddd gd ��V dS )z� Yields a dataframe with strings that may or may not need escaping by backticks. The last two columns cannot be escaped by backticks and should raise a ValueError. r r r r9 rU � � � r� rd � r� )r zB BzC CzC C�C_CzD_D DzE.EzF-F�1e1�defzA (x)zB(x)zB (x)u &^ :!€$?(} > <++*'' rb z Az zit'szthat'su ☺zfoo#barr N)r )r" r r r r E s. z$TestDataFrameQueryBacktickQuoting.dfc C s* |� d�}|d|d k }t�||� d S )Nz 1 < `B B`r zB B)r& r' r( )r" r rn r� r r r �#test_single_backtick_variable_queryg s zETestDataFrameQueryBacktickQuoting.test_single_backtick_variable_queryc C s6 |� d�}|d|d k d|d k @ }t�||� d S )Nz1 < `B B` and 4 < `C C`r zB Br9 zC C)r& r' r( )r" r rn r� r r r �!test_two_backtick_variables_queryl s zCTestDataFrameQueryBacktickQuoting.test_two_backtick_variables_queryc C s* |� d�}|d |d }t�||� d S )Nz A + `B B`r zB B)r) r' r* )r" r rn r� r r r �"test_single_backtick_variable_exprq s zDTestDataFrameQueryBacktickQuoting.test_single_backtick_variable_exprc C s* |� d�}|d |d }t�||� d S )Nz `B B` + `C C`zB BzC C)r) r' r* )r" r rn r� r r r � test_two_backtick_variables_exprv s zBTestDataFrameQueryBacktickQuoting.test_two_backtick_variables_exprc C s* |� d�}|d |d }t�||� d S )Nz `C_C` + Ar r )r) r' r* )r" r rn r� r r r � test_already_underscore_variable{ s zBTestDataFrameQueryBacktickQuoting.test_already_underscore_variablec C s* |� d�}|d |d }t�||� d S )Nz C_C + `C C`r zC C)r) r' r* )r" r rn r� r r r � test_same_name_but_underscores� s z@TestDataFrameQueryBacktickQuoting.test_same_name_but_underscoresc C s* |� d�}|d |d }t�||� d S )Nz A + `D_D D`r zD_D D)r) r' r* )r" r rn r� r r r �!test_mixed_underscores_and_spaces� s zCTestDataFrameQueryBacktickQuoting.test_mixed_underscores_and_spacesc C s* |� d�}|d |d }t�||� d S )Nz A + `C_C`r r )r) r' r* )r" r rn r� r r r �'test_backtick_quote_name_with_no_spaces� s zITestDataFrameQueryBacktickQuoting.test_backtick_quote_name_with_no_spacesc C s2 |� d�}|d |d |d }t�||� d S )Nz`E.E` + `F-F` - AzE.EzF-Fr )r) r' r* )r" r rn r� r r r �test_special_characters� s z9TestDataFrameQueryBacktickQuoting.test_special_charactersc C s* |� d�}|d |d }t�||� d S )Nz A + `1e1`r r )r) r' r* )r" r rn r� r r r �test_start_with_digit� s z7TestDataFrameQueryBacktickQuoting.test_start_with_digitc C s* |� d�}|d |d }t�||� d S )Nz A + `def`r r )r) r' r* )r" r rn r� r r r � test_keyword� s z.TestDataFrameQueryBacktickQuoting.test_keywordc C s* |� d�}||d dk }t�||� d S )Nz`A` > 2r r )r& r' r( )r" r rn r� r r r �test_unneeded_quoting� s z7TestDataFrameQueryBacktickQuoting.test_unneeded_quotingc C s* |� d�}||d dk }t�||� d S )Nz `A (x)` > 2zA (x)r )r& r' r( )r" r rn r� r r r �test_parenthesis� s z2TestDataFrameQueryBacktickQuoting.test_parenthesisc C s* |� d�}||d dk }t�||� d S )Nz`` > 5rb rU )r& r' r( )r" r rn r� r r r �test_empty_string� s z3TestDataFrameQueryBacktickQuoting.test_empty_stringc C s* |� d�}||d dk }t�||� d S )Nz `C C` > 5zC CrU )r& r' r( )r" r rn r� r r r �test_multiple_spaces� s z6TestDataFrameQueryBacktickQuoting.test_multiple_spacesc C s* |� d�}|d |d }t�||� d S )Nz ` A` + ` `z Az )r) r' r* )r" r rn r� r r r �test_start_with_spaces� s z8TestDataFrameQueryBacktickQuoting.test_start_with_spacesc C s* |� d�}||d dk }t�||� d S )Nu" ` &^ :!€$?(} > <++*'' ` > 4u &^ :!€$?(} > <++*'' r9 )r& r' r( )r" r rn r� r r r � test_lots_of_operators_string� s z?TestDataFrameQueryBacktickQuoting.test_lots_of_operators_stringc C s, d}t jt|d�� |�d� W d Q R X d S )Nz(module 'pandas' has no attribute 'thing')r/ z @pd.thing)r r0 �AttributeErrorr) )r" r �messager r r �test_missing_attribute� s z8TestDataFrameQueryBacktickQuoting.test_missing_attributec C s, d}t jt|d�� |�d� W d Q R X d S )Nz6(Could not convert ).*( to a valid Python identifier.))r/ z`it's` > `that's`)r r0 r� r& )r" r r2 r r r �test_failing_quote� s z4TestDataFrameQueryBacktickQuoting.test_failing_quotec C s, d}t jt|d�� |�d� W d Q R X d S )Nz6(Could not convert ).*( to a valid Python identifier.))r/ u `☺` > 4)r r0 r� r& )r" r r2 r r r �$test_failing_character_outside_range� s zFTestDataFrameQueryBacktickQuoting.test_failing_character_outside_rangec C s, d}t jt|d�� |�d� W d Q R X d S )NzFailed to parse backticks)r/ z `foo#bar` > 4)r r0 r� r& )r" r r2 r r r �test_failing_hashtag� s z6TestDataFrameQueryBacktickQuoting.test_failing_hashtagc C sd dd� }|g}|� d� tjtdd�� |� d� W dQ R X tjtdd�� |� d� W dQ R X dS ) z� Only attributes and variables ('named functions') can be called. .__call__() is not an allowed attribute because that would allow calling anything. https://github.com/pandas-dev/pandas/pull/32460 c W s dS )Nr r )�_r r r r � s zNTestDataFrameQueryBacktickQuoting.test_call_non_named_expression.<locals>.funcz@func()z"Only named functions are supported)r/ z @funcs[0]()Nz@funcs[0].__call__())r) r r0 r� )r" r r �funcsr r r � test_call_non_named_expression� s z@TestDataFrameQueryBacktickQuoting.test_call_non_named_expressionN) r4 r5 r6 r �fixturer r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r0 r1 r2 r3 r6 r r r r r D s. "r )(�ior r� �numpyrM r Z pandas.util._test_decorators�utilZ_test_decorators�tdr rh r r r r r Zpandas._testingZ_testingr' Z pandas.core.computation.checkr ZPARSERSr Z skip_if_no_neZENGINESr7 r r r r r7 rr r� r� r� r� r� r r r r r r �<module> s8 4V H il :
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.15 |
proxy
|
phpinfo
|
Настройка