Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/computation/__pycache__/test_eval.cpython-37.pyc
Назад
B �5�g� � @ s0 U d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZ d dl Z d dl m Z d dl m mZ d dlmZmZmZ d dlZd dlmZmZmZmZ d dlm Z d dl m Z d d l m!Z!m"Z" d dl#m$ m% m&Z& d d l#m'Z'm(Z(m)Z) d d l*m+Z+m,Z, d d l-m.Z.m/Z/m0Z0m1Z1m2Z2 e j3d d� e!D �d�dd� �Z4e j3e&j5d�dd� �Z6dd� Z7e j3e7� d�dd� �Z8dd� Z9dd� Z:d d � Z;d d � Z<d d!� Z=d"d#� Z>e?e@e.��Ae/��ZBe j3eCeDd$��d�d%d&� �ZEeEZFeEZGejHG d'd(� d(��ZIejHG d)d*� d*eI��ZJG d+d,� d,eJ�ZKG d-d.� d.eK�ZLd/d0� ZMG d1d2� d2�ZNd3d4� ZOd5d6� ZPG d7d8� d8�ZQejHG d9d:� d:��ZRejHG d;d<� d<eR��ZSG d=d>� d>eS�ZTG d?d@� d@eR�ZUejHG dAdB� dB��ZVG dCdD� dDeV�ZWG dEdF� dFeV�ZXG dGdH� dHeV�ZYe jZ�[dI�Z\G dJdK� dK�Z]ejHdLdM� �Z^ejHdNdO� �Z_e)e j`e(dP�ZadQebdR< e jc�dde!�e jc�ddea�dSdT� ��ZedUdV� ZfdWdX� Zge jc�ddYdZd[d\g�d]d^� �Zhd_d`� Zidadb� Zjdcdd� Zkdedf� Zle jc�ddgdh�e jc�dd&emenf�e jc�ddiemenf�djdk� ���Zoe jc�ddldme jpdne jcjqejr dodp�dq�g�drds� �Zsdtdu� Ztdvdw� Zudxdy� ZvG dzd{� d{�ZwdS )|� )� annotations)�reduce)�productN)�PerformanceWarning)�is_bool� is_list_like� is_scalar)� DataFrame�Series�compat� date_range)�pytables)�ENGINES�NumExprClobberingError)�BaseExprVisitor�PandasExprVisitor�PythonExprVisitor)�NUMEXPR_INSTALLED� USE_NUMEXPR)�ARITH_OPS_SYMS�SPECIAL_CASE_ARITH_OPS_SYMS�_binary_math_ops�_binary_ops_dict�_unary_math_opsc c s> | ]6}t j|t jj|d ko t dt� dt� �d�d�V qdS )�numexprznumexpr enabled->z , installed->)�reason)�marksN)�pytest�param�markZskipifr r )�.0�engine� r" �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/computation/test_eval.py� <genexpr>6 s r$ )�paramsc C s | j S )N)r )�requestr" r" r# r! 4 s r! c C s | j S )N)r )r& r" r" r# �parserE s r' c C s t r tt�S g S )N)r �listr r" r" r" r# �_get_unary_fns_for_neJ s r) c C s | j S )N)r )r& r" r" r# �unary_fns_for_neN s r* c C s t | jS )N)r Z has_neg_frac)r! r" r" r# �engine_has_neg_fracS s r+ c C s^ t | }t|�rTy || |�S tk rR } zt|��d�r@tjS � W d d }~X Y nX || |�S )Nz6negative number cannot be raised to a fractional power)r r+ � ValueError�str� startswith�np�nan)�lhs�cmp1�rhsr! �c�er" r" r# �_eval_single_binW s r6 c C s@ t | t�r t |tj�r |jdkp>t |t�o>t | tj�o>| jdkS )N� )� isinstancer r/ �ndarray�ndim)r1 r3 r" r" r# �_series_and_2d_ndarraye s r; c C s( t | t�rt |t�p&t |t�o&t | t�S )N)r8 r r )r1 r3 r" r" r# �_series_and_framek s r< c C s t | t�ot |tjjj�S )N)r8 �bool�pd�coreZgenericZNDFrame)r1 r3 r" r" r# �_bool_and_frameq s r@ c C s t |ttjf�ot�| �S )N)r8 �complexr/ Zcomplexfloating�isnan)�result�expectedr" r" r# �_is_py3_complex_incompatu s rE � c C sj t tj�dd��}tj||dk< t tj�dd��ttj�d��tddtjtjdg�|tj�� f}|| j S )N� rF g �?r7 � )r r/ �random�randr0 �randnr r )r& Znan_df1�optsr" r" r# r1 } s r1 c @ s� e Zd ZU g Zded< g Zded< dZdZedd� �Z e dd � �Z e j jd d d d dddgddddddgd�e j jdddgddgd�e j �dej�dd� ���Ze j �d ej�d d � �Ze j �d e�d d!� �Zd"d#� Zd$d%� Zejd&d'� �Ze j �d ej�d(d)� �Ze j �d ej�d*d+� �Z e j �d ddg�e j �dddg�d,d-� ��Z d.d/� Z d0d1� Z d2d3� Z d4d5� Z!d6d7� Z"d8d9� Z#d:d;� Z$d<d=� Z%d>d?� Z&d@dA� Z'dBdC� Z(djdEdF�Z)dGdH� Z*dIdJ� Z+dKdL� Z,dMdN� Z-e j �dDe.e/j0�1dOdP��e.e/j0j2dOdQdR��e.e/j0�3dOdP�dSk�g�dTdU� �Z4e j �dDe5e/j0�1dO��e5e/j0j2dOdOdR��e5e/j0�3dO�dSk�g�dVdW� �Z6dXdY� Z7dZd[� Z8e j �d\e/j9e/j:g�d]d^� �Z;d_d`� Z<dadb� Z=dcdd� Z>dedf� Z?dgdh� Z@diS )k�TestEvalNumexprPandasz list[str]� exclude_cmp� exclude_boolr �pandasc C s dd l }|| _d S )Nr )r �ne)�clsrQ r" r" r# � setup_class� s z!TestEvalNumexprPandas.setup_classc s � fdd�t D �S )Nc 3 s | ]}|� j kr|V qd S )N)r! )r r! )�selfr" r# r$ � s z8TestEvalNumexprPandas.current_engines.<locals>.<genexpr>)r )rT r" )rT r# �current_engines� s z%TestEvalNumexprPandas.current_enginesr2 z!=z==z<=z>=�<�>rQ �eq�le�ge�lt�gt)Zids�cmp2�binopc C s~ || j krt�� t|||| j�}t|||| j�}t|||| j�}d|� d|� d|� d�} tj| | j| jd�} | �| |� d S )Nz(lhs z rhs) z (lhs z rhs))r! r' ) rO r �skipr6 r! r> �evalr' � check_equal) rT r2 r] r^ r1 r3 �lhs_new�rhs_newrD �exrC r" r" r# �test_complex_cmp_ops� s z*TestEvalNumexprPandas.test_complex_cmp_ops�cmp_opc C s� || j krt�� ttjdd��tt�d��t�� f}ttjdd��tt�d��t�� f}x$t||�D ]\}}| �|||� qfW d S )N)rG rF )�size)rF ) rN r r_ r �tmZrandboolr r �check_simple_cmp_op)rT rf Z bool_lhsesZ bool_rhsesr1 r3 r" r" r# �test_simple_cmp_ops� s z)TestEvalNumexprPandas.test_simple_cmp_ops�opc C s | � |||� d S )N)�check_binary_arith_op)rT rk r1 r3 r& r" r" r# �test_binary_arith_ops� s z+TestEvalNumexprPandas.test_binary_arith_opsc C s | � |d|� d S )N�%)� check_modulus)rT r1 r3 r" r" r# � test_modulus� s z"TestEvalNumexprPandas.test_modulusc C s | � |d|� d S )Nz//)�check_floor_division)rT r1 r3 r" r" r# �test_floor_division� s z)TestEvalNumexprPandas.test_floor_divisionc C s | � |d|� d S )Nz**)� check_pow)rT r1 r3 r" r" r# �test_pow� s z TestEvalNumexprPandas.test_powc C s" || j krt�� | �||� d S )N)rN r r_ �check_single_invert_op)rT rk r1 r" r" r# �test_single_invert_op� s z+TestEvalNumexprPandas.test_single_invert_opc C s$ || j krt�� | �|||� d S )N)rN r r_ �check_compound_invert_op)rT rk r1 r3 r" r" r# �test_compound_invert_op� s z-TestEvalNumexprPandas.test_compound_invert_opc C s | � |||||� d S )N)�check_chained_cmp_op)rT r2 r] r1 �midhsr3 r" r" r# �test_chained_cmp_op� s z)TestEvalNumexprPandas.test_chained_cmp_opc C sZ t |t�rt�||� n>t |t�r0t�||� n&t |tj�rJt�||� n ||ksVt �d S )N) r8 r rh �assert_frame_equalr �assert_series_equalr/ r9 �assert_numpy_array_equal�AssertionError)rT rC rD r" r" r# ra � s z!TestEvalNumexprPandas.check_equalc s� � fdd�}||||�}||||�}|d k r�|d k r�d|� d|� d�} d|� d|� d�} d|� d|� d �} t |d |� j�} x2| | | fD ]$} tj| � j� jd �}t�|| � q�W d S ) Nc s t | ||� j�S )N)r6 r! )�left�rightrf )rT r" r# �check_operands� s zBTestEvalNumexprPandas.check_chained_cmp_op.<locals>.check_operandszlhs z mid z rhsz mid and mid z(lhs z mid) & (mid z rhs)�&)r! r' )r6 r! r> r` r' rh �assert_almost_equal)rT r1 r2 �midr] r3 r� rb rc �ex1�ex2�ex3rD rd rC r" )rT r# ry � s z*TestEvalNumexprPandas.check_chained_cmp_opc C s� d|� d�}d}|dkrXt |�sXtjt|d��" tj|| j| j||d�d� W d Q R X n0t|||| j�}tj|| j| jd�}| � ||� d S ) Nzlhs z rhsz�only list-like( or dict-like)? objects are allowed to be passed to (DataFrame\.)?isin\(\), you passed a (\[|')bool(\]|')|argument of type 'bool' is not iterable)�inznot in)�match)r1 r3 )r! r' � local_dict)r! r' ) r r �raises� TypeErrorr> r` r! r' r6 ra )rT r1 r2 r3 rd �msgrD rC r" r" r# ri s z)TestEvalNumexprPandas.check_simple_cmp_opc C s� d|� d�}t j|| j| jd�}t|||| j�}t�||� d|� d|� d�}t j|| j| jd�}t|||| j�}| �||||� d S )Nzlhs z rhs)r! r' z rhs )r> r` r! r' r6 rh r� �check_alignment)rT r1 �arith1r3 rd rC rD �nlhsr" r" r# rl s z+TestEvalNumexprPandas.check_binary_arith_opc C sT y|� |�\}}W n tttfk r, Y n$X | j�d|� d��}t�|j|� d S )Nznlhs z ghs) �alignr, r� �AttributeErrorrQ �evaluaterh r� �values)rT rC r� �ghsrk rD r" r" r# r� & s z%TestEvalNumexprPandas.check_alignmentc C sz d|� d�}t j|| j| jd�}|| }t�||� | j�d|� d��}t|t t f�rft�|j |� nt�||� � � d S )Nzlhs z rhs)r! r' z expected ) r> r` r! r' rh r� rQ r� r8 r r r� �item)rT r1 r� r3 rd rC rD r" r" r# ro 8 s z#TestEvalNumexprPandas.check_modulusc C s~ d|� d�}| j dkr@tj|| j | jd�}|| }| �||� n:d}tjt|d��" tj|||d�| j | jd� W d Q R X d S ) Nzlhs z rhs�python)r! r' zGunsupported operand type\(s\) for //: 'VariableNode' and 'VariableNode')r� )r1 r3 )r� r! r' )r! r> r` r' ra r r� r� )rT r1 r� r3 rd �resrD r� r" r" r# rq D s z*TestEvalNumexprPandas.check_floor_divisionc C sn yt |d|| j�}W nT tk rh } z6t|��d�rV| jdkrNt�t|�� qXtj}n� W d d }~X Y nX |S )Nz**z6negative number cannot be raised to a fractional powerr� ) r6 r! r, r- r. r r_ r/ r0 )rT r1 r3 rD r5 r" r" r# �get_expected_pow_resultX s z-TestEvalNumexprPandas.get_expected_pow_resultc C s� d|� d�}| � ||�}tj|| j| jd�}t|�rrt|�rrt||�rrd}tjt |d�� t � ||� W d Q R X nRt � ||� d|� d|� d�}tj|| j| jd�}| � | � ||�|�}t � ||� d S )Nzlhs z rhs)r! r' z-(DataFrame.columns|numpy array) are different)r� z(lhs z rhs) ) r� r> r` r! r' r rE r r� r rh r~ r� )rT r1 r� r3 rd rD rC r� r" r" r# rs g s z TestEvalNumexprPandas.check_powc C s� y|� t�}W n$ tk r2 t�t|�g�}Y nX | }tjd| j| jd�}t � ||� x(| j D ] }t � |tjd|| jd�� qbW d S )Nz~elb)r! r' ) �astyper= r� r/ �arrayr> r` r! r' rh r� rU )rT �elemr2 ZelbrD rC r! r" r" r# ru ~ s z,TestEvalNumexprPandas.check_single_invert_opc C s ddg}d|� d�}d}t |�r`||kr`tjt|d��" tj|| j| j||d�d� W d Q R X n�t |�r�t |�r�td d � ||f�\}}t |||| j�}t |�r�| }n| }tj|| j| jd �}t � ||� x.| j D ]$} tj|| j| jd �} t � | |� q�W d S ) Nr� znot inz~(lhs z rhs)z�only list-like( or dict-like)? objects are allowed to be passed to (DataFrame\.)?isin\(\), you passed a (\[|')float(\]|')|argument of type 'float' is not iterable)r� )r1 r3 )r! r' r� c S s t �| g�S )N)r/ r� )�xr" r" r# �<lambda>� � z@TestEvalNumexprPandas.check_compound_invert_op.<locals>.<lambda>)r! r' ) r r r� r� r> r` r! r' �mapr6 rh r� rU ) rT r1 r2 r3 Z skip_theserd r� rD rC r! Zevr" r" r# rw � s* z.TestEvalNumexprPandas.check_compound_invert_opr1 c C s |� |� �S )Nr" )rT rk �var_namer" r" r# rd � s zTestEvalNumexprPandas.exc C s� | � d�}ttj�dd��}| jdkrZd}tjt|d�� t j || j| j d�}W d Q R X n2d}tjt |d�� t j || j| j d�}W d Q R X ttjj dd d ��}| jdkr�d }tjt|d�� t j || j| j d�}W d Q R X n&| }t j || j| j d�}t�||� ttj�dd�d k�}| }t j || j| j d�}t�||� td ddgtj�d�d kd��}| jdk�r�tjtdd�� t j || j| j d�}W d Q R X n2d}tjt |d�� t j || j| j d�}W d Q R X d S )N�~rF rH r z-couldn't find matching opcode for 'invert_dd')r� )r! r' z0ufunc 'invert' not supported for the input types)rF rH )rg z)couldn't find matching opcode for 'invertg �?�ar7 g @� )�br4 zunknown type objectz#bad operand type for unary ~: 'str')rd r r/ rI rK r! r r� �NotImplementedErrorr> r` r' r� �randintrh r| rJ r, )rT �exprr1 r� rC �expectr" r" r# �test_frame_invert� s8 z'TestEvalNumexprPandas.test_frame_invertc C s� | � d�}ttj�d��}| jdkrXd}tjt|d�� t j || j| j d�}W d Q R X n2d}tjt |d�� t j || j| j d�}W d Q R X ttjj ddd��}| jdkr�d }tjt|d�� t j || j| j d�}W d Q R X n&| }t j || j| j d�}t�||� ttj�d�d k�}| }t j || j| j d�}t�||� td d d g�}| jdk�r�tjtdd�� t j || j| j d�}W d Q R X n2d}tjt |d�� t j || j| j d�}W d Q R X d S )Nr� rF r z-couldn't find matching opcode for 'invert_dd')r� )r! r' z0ufunc 'invert' not supported for the input types)rg z)couldn't find matching opcode for 'invertg �?r� r7 g @zunknown type objectz#bad operand type for unary ~: 'str')rd r r/ rI rK r! r r� r� r> r` r' r� r� rh r} rJ r, )rT r� r1 r� rC r� r" r" r# �test_series_invert� s8 z(TestEvalNumexprPandas.test_series_invertc C s� | � d�}ttj�dd��}| }tj|| j| jd�}t � ||� ttjj ddd��}| }tj|| j| jd�}t � ||� ttj� dd�dk�}| jdkr�d }t jt|d �� tj|| j| jd�}W d Q R X n&| }tj|| j| jd�}t � ||� d S ) N�-rF rH )r! r' )rF rH )rg g �?r z*couldn't find matching opcode for 'neg_bb')r� )rd r r/ rI rK r> r` r! r' rh r| r� rJ r r� r� )rT r� r1 r� rC r� r" r" r# �test_frame_negate s" z'TestEvalNumexprPandas.test_frame_negatec C s� | � d�}ttj�d��}| }tj|| j| jd�}t � ||� ttjj ddd��}| }tj|| j| jd�}t � ||� ttj� d�dk�}| jdkr�d}t jt|d�� tj|| j| jd�}W d Q R X n&| }tj|| j| jd�}t � ||� d S ) Nr� rF )r! r' )rg g �?r z*couldn't find matching opcode for 'neg_bb')r� )rd r r/ rI rK r> r` r! r' rh r} r� rJ r r� r� )rT r� r1 r� rC r� r" r" r# �test_series_negate' s" z(TestEvalNumexprPandas.test_series_negaterF rH )rF rH )rg g �?c C s2 | � d�}|}tj|| j| jd�}t�||� d S )N�+)r! r' )rd r> r` r! r' rh r| )rT r1 r� r� rC r" r" r# �test_frame_posA s z$TestEvalNumexprPandas.test_frame_posc C s2 | � d�}|}tj|| j| jd�}t�||� d S )Nr� )r! r' )rd r> r` r! r' rh r} )rT r1 r� r� rC r" r" r# �test_series_posS s z%TestEvalNumexprPandas.test_series_posc C sr d}t jt|d�� tjd| j| jd� W d Q R X tjd| j| jd�dksNt�tjd| j| jd�d ksjt�tjd | j| jd�d ks�t�tjd | j| jd�d ks�t�tjd| j| jd�dks�t�tjd| j| jd�d ks�t�tjd| j| jd�d ks�t�tjd| j| jd�d k�st�tjd| j| jd�dk�s2t�tjd| j| jd�dk�sPt�tjd| j| jd�dk�snt�d S )Nz%bad operand type for unary ~: 'float')r� z~1.0)r! r' z-1.0)r' r! g �z+1.0g �?z~1���z-1���z+1r7 z~Truez~Falsez-Truez-Falser z+Truez+False)r r� r� r> r` r! r' r )rT r� r" r" r# �test_scalar_unarye s z'TestEvalNumexprPandas.test_scalar_unaryc C s: t �t�d�tjdddddddddd d d g tjd �� d S ) NzK[-True, True, ~True, +True,-False, False, ~False, +False,-37, 37, ~37, +37]r� Tr� r7 r Fi����% i���)�dtype)rh r~ r> r` r/ r� Zobject_)rT r" r" r# �test_unary_in_arrayv s" z)TestEvalNumexprPandas.test_unary_in_arrayr� c C s\ t dtjdg|d�i�}|�d�}|jt�dg�ks8t�|�d�}|jt�dg�ksXt�d S )Nr� r )r� zx < -0.1Fz-5 > x)r r/ r� r` r� r )rT r� �dfr� r" r" r# � test_float_comparison_bin_op� s z2TestEvalNumexprPandas.test_float_comparison_bin_opc C s� d}|d7 }|d7 }|d7 }|d7 }t j�d�ddtt j�dd��f\}}}}x>|D ]6}d }tjt|d �� tj|| j | j d � W d Q R X qTW d S ) N)z1 or 2z1 and 2)za and bza or b)z1 or 2 and (3 + 2) > 3)z2 * x > 2 or 1 and 2)z2 * df > 3 and 1 or ar� r7 rH z;cannot evaluate scalar only bool ops|'BoolOp' nodes are not)r� )r! r' ) r/ rI rK r r r� r� r> r` r! r' )rT �exprsr� r� r� r� rd r� r" r" r# � test_disallow_scalar_bool_ops� s * z3TestEvalNumexprPandas.test_disallow_scalar_bool_opsc C sf d}t jd| j| jd�}|dks$t�t|�s0t�d}t jd| j| jd�}|dksTt�t|�s`t�d}t jd| j| jd�}|r�t�t|�s�t�t|�s�t�t�dg�}t jd| j| jd�}t � |t�dg�� |j dks�t�t�dg�}t jd| j| jd�}t � |t�dg�� |j dk�s t�t�dg�}t jd| j| jd�}t � |t�dg�� |j dk�sbt�d S )Nr7 r� )r! r' g �?F)r7 ) r> r` r! r' r r r r/ r� rh r~ �shape)rT r� rC r" r" r# �test_identical� s2 z$TestEvalNumexprPandas.test_identicalc C s( d}t j|| j| jd�}|dks$t�d S )Nz1 + 2 * 5 - 1 + 2 )r! r' � )r> r` r! r' r )rT �exprC r" r" r# �test_line_continuation� s z,TestEvalNumexprPandas.test_line_continuationc C s� d}t j|| j| jd�}t�|�}||ks.t�tddddgi�}d}|�d|d ���}|j s`t�d }|�d |d ���}|j d d gd d �f }t � ||� d}|�d|d ���}|j d gd d �f }t � ||� d S )Nz1000000000.006)r! r' �Ag~ e��Ag $ e��Ag'1 e��Ag� e��AzA < z.4fg� e��AzA > r7 rH zA == ) r> r` r! r' r/ �float64r r �query�empty�locrh r| )rT r� rC rD r� �cutoff�exactr" r" r# �test_float_truncation� s z+TestEvalNumexprPandas.test_float_truncationc C sx t dddggdddgd�}d}tjt|d�� |�d� W d Q R X t � }d |j_tjt|d�� |�d � W d Q R X d S ) Nr Zfoo�bar�class)�columnsz4Python keyword not valid identifier in numexpr query)r� z class == 0�lambdaz lambda == 0)r r r� � SyntaxErrorr� �index�name)rT r� r� r" r" r# � test_disallow_python_keywords� s z3TestEvalNumexprPandas.test_disallow_python_keywordsN)r1 )A�__name__� __module__� __qualname__rN �__annotations__rO r! r' � classmethodrS �propertyrU r r � parametrizer� � BOOL_OPS_SYMSre � CMP_OPS_SYMSrj �_good_arith_opsrm rp rr �td�skip_if_windowsrt rv rx r{ ra ry ri rl r� ro rq r� rs ru rw rd r� r� r� r� r r/ rI rK r� rJ r� r r� r� r� �float32r� r� r� r� r� r� r� r" r" r" r# rM � sn # +/" !rM c s@ e Zd ZddgZddgZdZdZe� fdd��Zd d � Z � Z S ) �TestEvalNumexprPythonr� znot in�and�orr r� c s t � �� dd l}|| _d S )Nr )�superrS r rQ )rR rQ )� __class__r" r# rS � s z!TestEvalNumexprPython.setup_classc C sH d|� d|� d�}d}t jt|d�� tj|| j| jd� W d Q R X d S )Nzlhs z mid z rhsz"'BoolOp' nodes are not implemented)r� )r! r' )r r� r� r> r` r! r' )rT r1 r2 r� r] r3 r� r� r" r" r# ry s z*TestEvalNumexprPython.check_chained_cmp_op) r� r� r� rN rO r! r' r� rS ry � __classcell__r" r" )r� r# r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�TestEvalPythonPythonr� c C sT d|� d�}t j|| j| jd�}|| }t�||� t|||| j�}t�||� d S )Nzlhs z rhs)r! r' )r> r` r! r' rh r� r6 )rT r1 r� r3 rd rC rD r" r" r# ro s z"TestEvalPythonPython.check_modulusc C sN y|� |�\}}W n tttfk r, Y n X td|� d��}t�||� d S )Nznlhs z ghs)r� r, r� r� r` rh r� )rT rC r� r� rk rD r" r" r# r� s z$TestEvalPythonPython.check_alignmentN)r� r� r� r! r' ro r� r" r" r" r# r� s r� c @ s e Zd ZdZdZdd� ZdS )�TestEvalPythonPandasr� rP c C s t �| |||||� d S )N)rM ry )rT r1 r2 r� r] r3 r" r" r# ry ' s z)TestEvalPythonPandas.check_chained_cmp_opN)r� r� r� r! r' ry r" r" r" r# r� # s r� c O s t j�� S )N)r/ rI rK )�args�kwargsr" r" r# r� + r� r� c @ sB e Zd Zej�ddddddg�ej�dejejg�dd � ��Z d S ) �TestTypeCastingrk r� r� �*z**�/�dtc C s� t jddt|d�}d|� d�}tj|||d�}|jj|ks>t�|jj|ksNt�t �|t|�� d|� d�}tj|||d�}|jj|ks�t�|jj|ks�t�t �|t|�� d S ) NrF r� )� data_gen_fr� zdf z 3)r! r' z3 z df) rh �makeCustomDataframe�fr> r` r� r� r r| )rT r! r' rk r� r� �sr� r" r" r# �test_binop_typecasting3 s z&TestTypeCasting.test_binop_typecastingN) r� r� r� r r r� r/ r� r� r� r" r" r" r# r� 2 s r� c C s t | jjtj�S )N)� issubclassr� �typer/ Z datetime64)r� r" r" r# � _is_datetimeJ s r� c G s0 t tt�d�| �� }ttjtt| ��}|o.|S )NZ is_monotonic)�anyr� �operator� attrgetterr �xorr� )r� Znot_monoZ only_one_dtr" r" r# � should_warnN s r� c @ s� e Zd ZdddgZedg Zdd� Zej�de�ej�de�ej�d e�d d � ���Z ej�d e�ej�d e�d d� ��Z ej�de�ej�de�ej�de�ej�de�dd� ����Z ej�dddg�ej�d e�ej�d e�dd� ���Z ej�dddg�dd� �Z ej�d e�ej�d e�ej�dddg�ej�d d d g�d d � ����Zej�de�ej�de�ej�de�ej�de�d!d"� ����Zd#d$� Zd%S )&� TestAlignment�i�ur� r� c C s8 d}t jddtd�}tj|||d�}t �||d � d S )Nzdf * ~2rF r� )r� )r! r' ���)rh r� r� r> r` r| )rT r! r' r� r� r� r" r" r# �test_align_nested_unary_opY s z(TestAlignment.test_align_nested_unary_op� lr_idx_type� rr_idx_type� c_idx_typec C s� t jdd��� t �dt� tjddt||d�}tjddt||d�}t|j|j�rxt� t�� t j d||d�}W d Q R X nt j d||d�}t� ||| � W d Q R X d S ) NT)�record�alwaysrG )r� � r_idx_typer � zdf + df2)r! r' ) �warnings�catch_warnings� simplefilter�RuntimeWarningrh r� r� r� r� �assert_produces_warningr> r` r| ) rT r! r' r� r r r� �df2r� r" r" r# �test_basic_frame_alignment_ s z(TestAlignment.test_basic_frame_alignmentr c C st t jddt||d�}tjd||d�}t �||dk � ttjj |j � |j |j d�}tjd||d�}t �|||k � d S )NrG )r� r r zdf < 2)r! r' rH )r� r� zdf < df3) rh r� r� r> r` r| r r/ rI rK r� r� r� )rT r! r' r r r� r� �df3r" r" r# �test_frame_comparisonv s z#TestAlignment.test_frame_comparison�r1�c1�r2�c2c C s� t jdd��� t �dt� tjddt||d�}tjddt||d�}tjddt||d�} t|j|j| j�r�t� t�� t j d ||d �} W d Q R X nt j d ||d �} t� | || | � W d Q R X d S ) NT)r r r� rH )r� r r � rF zdf + df2 + df3)r! r' ) r r r r rh r� r� r� r� r r> r` r| ) rT r! r' r r r r r� r r r� r" r" r# �#test_medium_complex_frame_alignment� s z1TestAlignment.test_medium_complex_frame_alignment� index_namer� r� c sB � �fdd�}t jdd�� t �dt� ||||� W d Q R X d S )Nc s� t jddt| |d�}t||�}ttj�d�|d d� �}t|j |j �rnt � t �� t j d� �d�}W d Q R X nt j d� �d�}| dks�|dkr�� dkr�|�|�n|| }n|| }t �||� d S )NrG )r� r r rF zdf + s)r! r' r� r )rh r� r� �getattrr r/ rI rK r� r� r r r> r` �addr| )r r r r� r� r� r� rD )r! r' r" r# �testit� s z?TestAlignment.test_basic_frame_series_alignment.<locals>.testitT)r r )r r r r )rT r! r' r r r r r" )r! r' r# �!test_basic_frame_series_alignment� s z/TestAlignment.test_basic_frame_series_alignmentc s� � �fdd�}t dddgdddg�}tjdd��0 t�dt� x |D ]\}}||||� qDW W d Q R X t d gd g�}tjdd��0 t�dt� x |D ]\}}||||� q�W W d Q R X d S ) Nc s� t jddt| |d�}t||�}ttj�d�|d d� �}t|j |j �rnt � t �� t j d� �d�}W d Q R X nt j d� �d�}| dks�|dkr�� dkr�|�|�n|| }n|| }t �||� d S ) NrG � )r� r r rF zs + df)r! r' r� r )rh r� r� r r r/ rI rK r� r� r r r> r` r r| )r r r r� r� r� r� rD )r! r' r" r# r � s z?TestAlignment.test_basic_series_frame_alignment.<locals>.testitr� r� r� T)r �ignorer� )r r r r r )rT r! r' r r r� r r r" )r! r' r# �!test_basic_series_frame_alignment� s z/TestAlignment.test_basic_series_frame_alignmentrk r� r� c C s t jdd��� t �dt� tjddt||d�}t||�}tt j � d�|d d� �} d|� d�} d |� d �} t |j | j �r�t�t�� tj| ||d �} W d Q R X t�t�� tj| ||d �} W d Q R X n tj| ||d �} tj| ||d �} |d k�r|d k�r|d k�rt�| | � W d Q R X d S )NT)r r rG )r� r r rF zs z dfzdf z s)r! r' r� r )r r r r rh r� r� r r r/ rI rK r� r� r r> r` r| )rT r! r' r rk r r r� r� r� r1 r3 r� r� r" r" r# � test_series_frame_commutativity� s$ z-TestAlignment.test_series_frame_commutativityc C s� dd l }d}d} d| } tjdd���\ t�dt� |�dd g�} |�d d g�} tj| |t||d �} tj| |t||d �}t t � � | �| �}t t j �|�|d |� �}|d ks�|d kr�|dkr�|�|�}q�|| }n|| }|d ks�|d k�r |dk�r|�| �}n|| }n|| }t|j|j| j��rPt�t�� tjd||d�}W d Q R X ntjd||d�}|j|jk�srt�t�||� W d Q R X d S )Nr r� rF rH T)r r r� r� r� r )r� r r r� r zdf2 + ser + df)r! r' )rI r r r r �choicerh r� r� r �locals�getr r/ rK r r� r� r r> r` r� r r| )rT r! r' r r r r rI �n�m1�m2r �obj_namer� r r� ZserZ expected2rD r� r" r" r# �#test_complex_series_frame_alignment s< z1TestAlignment.test_complex_series_frame_alignmentc C s� t tj�dd��}ttj�d��}|dkr0t}nd}t�|�� tj d||d� W d Q R X ttj�d��}t�d�� tj d||d� W d Q R X t tj�dd��}ttj�d��}t�d�� tj d||d� W d Q R X t tj�dd��}ttj�d��}|dk}|�st}nd}t�|��n}tj d||d� |�s�t |�d k�sFt �t |d j �} t�|j|jd �} d | d �d �} | | k�s�t �W d Q R X d S )Ni� rG i' r Fzdf + s)r! r' r� r7 r z_Alignment difference on axis 1 is larger than an order of magnitude on term 'df', by more than z.4gz; performance may suffer)r r/ rI rK r r rh r r> r` �lenr r- �message�log10rg r� ) rT r! r' r� r� �seenZis_python_engineZwrn�wr� ZloggedrD r" r" r# �+test_performance_warning_for_poor_alignment2 s8 z9TestAlignment.test_performance_warning_for_poor_alignmentN)r� r� r� Z index_typesZlhs_index_typesr� r r r� r r r r r r r% r+ r" r" r" r# r� T s2 & /r� c @ s� e Zd ZU g Zded< dZdZedd� �Zdd� Z d d � Z e j � d d d g�e j � dd d g�e j � dej�dd� ���Ze j � d d d g�e j � dd d g�e j � dej�dd� ���Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!d4d5� Z"d6d7� Z#e j � d8d9d:d9d;ge$�%g �d<g�e j �&d=�d>d?� ��Z'e j � d8d9d:d<g�d@dA� �Z(e j � dBd9d:d9d;ge$�%g �d<d9d;ig�dCdD� �Z)dEdF� Z*dGdH� Z+dIdJ� Z,dKdL� Z-dMdN� Z.dOS )P�TestOperationsNumExprPandasz list[str]� exclude_arithr rP c s � fdd�t jt j D �� _d S )Nc s g | ]}|� j kr|�qS r" )r- )r rk )rR r" r# � <listcomp>m s z;TestOperationsNumExprPandas.setup_class.<locals>.<listcomp>)r� r r� � arith_ops)rR r" )rR r# rS j s z'TestOperationsNumExprPandas.setup_classc O s4 | j |d< | j|d< |�dd�d |d< tj||�S )Nr! r' �levelr r7 )r! r' �popr> r` )rT r� r� r" r" r# r` r s z TestOperationsNumExprPandas.evalc C s& dd� | j D �}�x|D �]}d|� d�}d|� d�}d|� d�}|dkr~d}tjt|d �� tj|| j| jd � W d Q R X qtd |d | j�}| j|| j| jd �}||ks�t �t||d | j�}| j|d |i| j| jd �} | |ks�t �td ||d | j�}| j|d |i| j| jd �} | |kst �qW d S )Nc s s | ]}|d kr|V qdS )z//Nr" )r rk r" r" r# r$ y s zDTestOperationsNumExprPandas.test_simple_arith_ops.<locals>.<genexpr>z1 z 1zx z (x + 1))r� znot inz&argument of type 'int' is not iterable)r� )r! r' r7 r� )r� r! r' ) r/ r r� r� r> r` r! r' r6 r ) rT �opsrk rd r� r� r� �expecr� �yr" r" r# �test_simple_arith_opsx s( z1TestOperationsNumExprPandas.test_simple_arith_opsr3 TFr1 rk c C s6 |� d|� d|� �}| � |�}t |�}||ks2t�d S )N� )r` r )rT r3 r1 rk rd r� r� r" r" r# �test_simple_bool_ops� s z0TestOperationsNumExprPandas.test_simple_bool_opsc C s6 |� d|� d|� �}| � |�}t |�}||ks2t�d S )Nr6 )r` r )rT r3 r1 rk rd r� r� r" r" r# � test_bool_ops_with_constants� s z8TestOperationsNumExprPandas.test_bool_ops_with_constantsc C sX t j�dddd�}tt j�d��}d}tjt|d�� | jd||d �d � W d Q R X d S ) Nr� r rF � rG z?N-dimensional objects, where N > 2, are not supported with eval)r� zx + y)r� r4 )r� )r/ rI rK r r r� r� r` )rT r� r4 r� r" r" r# �test_4d_ndarray_fails� s z1TestOperationsNumExprPandas.test_4d_ndarray_failsc C s | � d�}|dkst�d S )N�1r7 )r` r )rT r� r" r" r# � test_constant� s z)TestOperationsNumExprPandas.test_constantc C s4 t tj�dd��}| jdd|id�}t�||� d S )NrG rH r� )r� )r r/ rI rK r` rh r| )rT r� r r" r" r# �test_single_variable� s z0TestOperationsNumExprPandas.test_single_variablec C s` t �dg�}d}d|i}t�t�� | j|dd�}W d Q R X t�|t �dg�� t�t�� | j|dd�}W d Q R X t�|t �dg�� t�t�� | jddd�}W d Q R X d }||ks�t�t�t�� | jddd�}W d Q R X d }||ks�t�t�t�� | jd dd�}W d Q R X d }||k�s&t�t�t�� | jd dd�}W d Q R X d }||k�s\t�d S ) Nr7 zs / 1r� F)�truedivg �?Tz1 / 2g �?zs / 2)r/ r� rh r � FutureWarningr` r~ r )rT r� rd �dr� r3 r" r" r# � test_truediv� s2 z(TestOperationsNumExprPandas.test_truedivc C s: t tj�dd��}tjtdd�� | �d� W d Q R X d S )NrF r� zname 'x' is not defined)r� z df[x > 2] > 2)r r/ rI rK r r� � NameErrorr` )rT r� r" r" r# �&test_failing_subscript_with_name_error� s zBTestOperationsNumExprPandas.test_failing_subscript_with_name_errorc C sD t tj�dd��}| jdd|id�}|d |dk }t�||� d S )NrF r� z(df + 1)[df > 2]r� )r� r7 rH )r r/ rI rK r` rh r| )rT r� rC rD r" r" r# � test_lhs_expression_subscript� s z9TestOperationsNumExprPandas.test_lhs_expression_subscriptc C s� t tj�dd�td�d�}d}|j|jk }d}|j|j |j }d}|j|j |j|jdk }|||f}|||f} x0t|| �D ]"\} } t � | | j | d |id �� q�W d S ) NrF r� �abc)r� z df.a < df.bzdf.a + df.b + df.cz df.a + df.b + df.c[df.b < 0]r r� )r� ) r r/ rI rK r( r� r� r4 �ziprh r} r` ) rT r� �expr1Zexpec1Zexpr2Zexpec2Zexpr3Zexpec3r� Zexpecsr5 r3 r" r" r# �test_attr_expression� s z0TestOperationsNumExprPandas.test_attr_expressionc C sf t tj�dd�td�d�}t tj�dd��}d}d}tjt|d�� | j|||d�d � W d Q R X d S ) NrF r� rE )r� zdf = df2z%cannot assign without a target object)r� )r� r )r� ) r r/ rI rK r( r r� r, r` )rT r� r rG r� r" r" r# �test_assignment_fails� s z1TestOperationsNumExprPandas.test_assignment_failsc s� t tj�dd�td�d�� � �� �tjtdd�� � � d� W d Q R X d}tjt|d�� � � d � W d Q R X t j rzd }nd }tjt|d�� � � d � W d Q R X ��� }|d |d |d < ��� � � j ddd� t � � |� ��� }|d |d |d< ��� � � j ddd� t � � |� �fdd�}|� � ��� }d|d |d < t � � |� ��� � � fdd�}|� ��� � � j ddd� d}tjt|d�� � � d� W d Q R X ��� � | j dd� i� dd � ��� }|d |d |d< t � � |� d S ) NrF rH �ab)r� zinvalid syntax)r� z d c = a + bz5left hand side of an assignment must be a single namez d,c = a + bz cannot assign to function callz can't assign to function callz Timestamp("20131001") = a + br� r� z a = a + bT)�inplacer4 z c = a + bc s � � � } d}| jddd� | S )Nr7 z a = 1 + bT)rK )�copyr` )r� r� )�orig_dfr" r# r� s z=TestOperationsNumExprPandas.test_assignment_column.<locals>.fr7 c sJ d} � j �� }� jddd� |� j }tj|� j dd� |jd ksFt�d S )Nr7 z a = a + bT)rK F)� check_names)r� rL r` r� rh r} r� r )r� Zold_arC )r� r" r# r� ) s z#can only assign a single expressionz c = a = bzc = df.a + df.br� )r� �targetrK )r r/ rI rK r( rL r r� r� r` r �PY38rh r| )rT r� rD r� r" )r� rM r# �test_assignment_column� sP z2TestOperationsNumExprPandas.test_assignment_columnc C s6 t dgdgd��}|�d�}tdg�}t�||� d S )N� i��)r� r� za in [11, -32]T)r r` r rh r} )rT r� rC rD r" r" r# �test_column_inA s z*TestOperationsNumExprPandas.test_column_inc C s` t tj�dd�td�d�}|jddd�}|d k s4t�|�� }|d |d |d < t� ||� d S ) NrF rH rJ )r� z c = a + bF)rK r� r� r4 ) r r/ rI rK r( r` r rL rh r| )rT r� �actualrD r" r" r# �assignment_not_inplaceH s z2TestOperationsNumExprPandas.assignment_not_inplacec C s� t dddgdddgd��}|�� }|d |d |d < |d |d |d < |jd d d�}t�||� |d kspt�|d d |d< |d d |d< |jdd d�}t�||� |d ks�t�d}tjt|d�� |jddd� W d Q R X d S )Nr7 rH r� r rF r9 )r� r� r� r� r4 r@ z$ c = a + b d = c + bT)rK r5 z$ a = a - 1 e = a + 2z@Multi-line expressions are only valid if all expressions contain)r� z( a = b + 2 b - 2F) r rL r` rh r| r r r� r, )rT r� rD �answerr� r" r" r# �test_multi_line_expressionS s* z6TestOperationsNumExprPandas.test_multi_line_expressionc C s� t dddgdddgd��}|�� }|d |d |d < |d |d |d < |jd d d�}t�||� |d d |d< |d d |d< |jdd d�}t�||� d S )Nr7 rH r� r rF r9 )r� r� r� r� r4 r@ z$ c = a + b d = c + bF)rK r5 z$ a = a - 1 e = a + 2)r rL r` rh r| )rT r� rD r" r" r# �&test_multi_line_expression_not_inplacex s zBTestOperationsNumExprPandas.test_multi_line_expression_not_inplacec C sp t dddgdddgd��}|�� }d}|d | |d < |d | |d < |jd d d�}t�||� |d kslt�d S )Nr7 rH r� r rF r9 )r� r� r r� r4 r@ z? c = a * @local_var d = c + @local_var T)rK )r rL r` rh r| r )rT r� rD Z local_varrV r" r" r# �)test_multi_line_expression_local_variable� s zETestOperationsNumExprPandas.test_multi_line_expression_local_variablec C s� t dddgdddgd��}dd � }|�� }|d |dd � |d < |d |dd � |d < |jddd�}t�||� |d ks|t�d S )Nr7 rH r� r rF r9 )r� r� c S s |S )Nr" )r� r� r" r" r# � local_func� s zbTestOperationsNumExprPandas.test_multi_line_expression_callable_local_variable.<locals>.local_funcr� r r4 r@ zM c = a * @local_func(1, 7) d = c + @local_func(1, 7) T)rK )r rL r` rh r| r )rT r� rZ rD rV r" r" r# �2test_multi_line_expression_callable_local_variable� s zNTestOperationsNumExprPandas.test_multi_line_expression_callable_local_variablec C s� t dddgdddgd��}dd � }|�� }|d |d dd � |d < |d |d dd � |d< |jddd�}t�||� |d ks�t�d S )Nr7 rH r� r rF r9 )r� r� c S s |S )Nr" )r� r� r" r" r# rZ � s znTestOperationsNumExprPandas.test_multi_line_expression_callable_local_variable_with_kwargs.<locals>.local_funcr� r )r� r� r4 r@ zU c = a * @local_func(b=7, a=1) d = c + @local_func(b=7, a=1) T)rK )r rL r` rh r| r )rT r� rZ rD rV r" r" r# �>test_multi_line_expression_callable_local_variable_with_kwargs� s zZTestOperationsNumExprPandas.test_multi_line_expression_callable_local_variable_with_kwargsc C sZ t dddgdddgd��}|�� }d}tjt|d �� |�d � W d Q R X t�||� d S ) Nr7 rH r� r rF r9 )r� r� z%cannot assign without a target object)r� za = 1)r rL r r� r, r� rh r| )rT r� Zdf_origr� r" r" r# �test_assignment_in_query� s z4TestOperationsNumExprPandas.test_assignment_in_queryc C sx t dddgdddgd��}|�� }||d dk }|jd d d � t�||� i }ddi}| jd |d d � t�||� d S )Nr7 rH r� r rF r9 )r� r� r� za == 2T)rK z a = 1 + 2)rO rK )r rL r� rh r| r` Zassert_dict_equal)rT r� rD r" r" r# �test_query_inplace� s z.TestOperationsNumExprPandas.test_query_inplace�invalid_targetr7 �catrH )r7 r� zignore::FutureWarningc C sj d}d}t jt|d�� | j||dd� W d Q R X t|d�rft jt|d�� | j||dd� W d Q R X d S )Nz)Cannot assign expression output to targetz a = 1 + 2)r� T)rO rK rL F)r r� r, r` �hasattr)rT r_ r� � expressionr" r" r# �test_cannot_item_assign� s z3TestOperationsNumExprPandas.test_cannot_item_assignc C s6 d}d}t jt|d�� | j||dd� W d Q R X d S )Nz"Cannot return a copy of the targetz a = 1 + 2)r� F)rO rK )r r� r, r` )rT r_ r� rb r" r" r# �test_cannot_copy_item� s z1TestOperationsNumExprPandas.test_cannot_copy_itemrO c C sN d}| j ||dd�dks t�d}tjt|d�� | j ||dd� W d Q R X d S )Nz1 + 2F)rO rK r� z0Cannot operate inplace if there is no assignment)r� T)r` r r r� r, )rT rO rb r� r" r" r# �test_inplace_no_assignment� s z6TestOperationsNumExprPandas.test_inplace_no_assignmentc C sR t jddtddd�}|dk }| jdd|id�}|dk }t �||� t �||� d S )NrH �pr� )r� r r zdf < 2r� )r� )rh r� r� r` r| )rT r� r5 �rr� r" r" r# �*test_basic_period_index_boolean_expression s zFTestOperationsNumExprPandas.test_basic_period_index_boolean_expressionc C sB t jddtddd�}| jdd|id�}||dk }t �||� d S ) NrH rf r� )r� r r zdf[df < 2 + 3]r� )r� rF )rh r� r� r` r| )rT r� rg r5 r" r" r# �,test_basic_period_index_subscript_expression s zHTestOperationsNumExprPandas.test_basic_period_index_subscript_expressionc C sR t jddtddd�}| jdd|id�}|||dk dk |d }t �||� d S )NrH rf r� )r� r r zdf[df[df < 2] < 2] + df * 2r� )r� )rh r� r� r` r| )rT r� rg r5 r" r" r# �-test_nested_period_index_subscript_expression s zITestOperationsNumExprPandas.test_nested_period_index_subscript_expressionc C sZ t tj�dd��}tddd�|d< | jdd|i| j| jd�}|jd k }t j ||d d � d S ) NrF r� z1/1/2012)Zperiods�dates1zdf.dates1 < 20130101r� )r� r! r' Z20130101F)rN ) r r/ rI rK r r` r! r' rk rh r} )rT r� r� r3 r" r" r# �test_date_boolean s z-TestOperationsNumExprPandas.test_date_booleanc C sL | j dk�r,tjd| j| j d�}|s(t�tjd| j| j d�}|sDt�tjd| j| j d�}|r`t�tjd| j| j d�}|s|t�tjd| j| j d�}|s�t�tjd| j| j d�}|s�t�tjd | j| j d�}|s�t�tjd | j| j d�}|s�t�tjd | j| j d�}|�r t�tjd | j| j d�}|�sHt��n d }tjt|d�� tjd| j| j d� W d Q R X tjt|d�� tjd| j| j d� W d Q R X tjt|d�� tjd| j| j d� W d Q R X tjt|d�� tjd| j| j d� W d Q R X d}tjt|d�� tjd| j| j d� W d Q R X tjt|d�� tjd| j| j d� W d Q R X d S )Nr� z 1 in [1, 2])r! r' z 2 in (1, 2)z 3 in (1, 2)z3 not in (1, 2)z[3] not in (1, 2)z[3] in ([3], 2)z[[3]] in [[[3]], 2]z(3,) in [(3,), 2]z(3,) not in [(3,), 2]z[(3,)] in [[(3,)], 2]z 'In' nodes are not implemented)r� z[(3,)] in (1, 2, [(3,)])z!'NotIn' nodes are not implementedz[3] not in (1, 2, [[3]]))r' r> r` r! r r r� r� )rT r� r� r" r" r# �test_simple_in_ops# sN z.TestOperationsNumExprPandas.test_simple_in_opsN)/r� r� r� r- r� r! r' r� rS r` r5 r r r� r� r� r7 r8 r: r< r= rA rC rD rH rI rQ rS rU rW rX rY r[ r\ r] r^ r/ r� �filterwarningsrc rd re rh ri rj rl rm r" r" r" r# r, c sN # G % 0 r, c @ s� e Zd ZU ddgZded< dZdZdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z e j�dddg�e j�dddg�e j�dej�dd� ���Ze j�dddg�e j�dddg�e j�dej�dd� ���Zd S ) �TestOperationsNumExprPythonr� znot inz list[str]r- r r� c C s6 d}d� d�}d}tj|| j| jd�}||ks2t�d S )Nr7 z * Z!aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)r! r' )�joinr> r` r! r' r )rT r� r� rD r� r" r" r# �test_check_many_exprsb s z1TestOperationsNumExprPython.test_check_many_exprsc C sN t tj�dd��}d}tjt|d�� tjdd|i| j | j d� W d Q R X d S )NrF r� z"'BoolOp' nodes are not implemented)r� zdf > 2 and df > 3r� )r� r' r! ) r r/ rI rK r r� r� r> r` r' r! )rT r� r� r" r" r# �test_fails_andi s z*TestOperationsNumExprPython.test_fails_andc C sN t tj�dd��}d}tjt|d�� tjdd|i| j | j d� W d Q R X d S )NrF r� z"'BoolOp' nodes are not implemented)r� zdf > 2 or df > 3r� )r� r' r! ) r r/ rI rK r r� r� r> r` r' r! )rT r� r� r" r" r# � test_fails_ort s z)TestOperationsNumExprPython.test_fails_orc C sN t tj�dd��}d}tjt|d�� tjdd|i| j | j d� W d Q R X d S )NrF r� z 'Not' nodes are not implemented)r� z not df > 2r� )r� r' r! ) r r/ rI rK r r� r� r> r` r' r! )rT r� r� r" r" r# �test_fails_not s z*TestOperationsNumExprPython.test_fails_notc C sL t tj�dd��}d}d}tjt|d�� tj|| j | j d� W d Q R X d S )NrF r� z (df + 2)[df > 1] > 0 & (df > 0)z$cannot evaluate scalar only bool ops)r� )r' r! ) r r/ rI rK r r� r� r> r` r' r! )rT r� rd r� r" r" r# �test_fails_ampersand� s z0TestOperationsNumExprPython.test_fails_ampersandc C sL t tj�dd��}d}d}tjt|d�� tj|| j | j d� W d Q R X d S )NrF r� z (df + 2)[df > 1] > 0 | (df > 0)z$cannot evaluate scalar only bool ops)r� )r' r! ) r r/ rI rK r r� r� r> r` r' r! )rT r� rd r� r" r" r# �test_fails_pipe� s z+TestOperationsNumExprPython.test_fails_piper3 TFr1 rk c C sh |� d|� d|� �}|dkrFd}t jt|d�� | �|� W d Q R X n | �|�}t|�}||ksdt�d S )Nr6 )r� r� z"'BoolOp' nodes are not implemented)r� )r r� r� r` r )rT r1 r3 rk rd r� r� r� r" r" r# r8 � s z8TestOperationsNumExprPython.test_bool_ops_with_constantsc C st d|� d�}|dkrHd}t jt|d�� tj|| j| jd� W d Q R X n(tj|| j| jd�}t|�}||kspt�d S )Nzlhs z rhs)r� r� z"'BoolOp' nodes are not implemented)r� )r! r' )r r� r� r> r` r! r' r )rT r1 r3 rk rd r� r� r� r" r" r# r7 � s z0TestOperationsNumExprPython.test_simple_bool_opsN)r� r� r� r- r� r! r' rq rr rs rt ru rv r r r� r� r� r8 r7 r" r" r" r# ro [ s ro c @ s e Zd ZdZdZdS )�TestOperationsPythonPythonr� N)r� r� r� r! r' r" r" r" r# rw � s rw c @ s" e Zd ZU g Zded< dZdZdS )�TestOperationsPythonPandasz list[str]r- r� rP N)r� r� r� r- r� r! r' r" r" r" r# rx � s rx c @ sx e Zd ZdZdZdd� Zdd� Zej� de �dd � �Z d d � Z d d � Z dd� Zdd� Zejdd� �Zdd� Zdd� ZdS )�TestMathPythonPythonr� rP c O s4 | j |d< | j|d< |�dd�d |d< tj||�S )Nr! r' r0 r r7 )r! r' r1 r> r` )rT r� r� r" r" r# r` � s zTestMathPythonPython.evalc C sl t dtj�d�i�}|j}|}|� d�}| �|�}tjdd�� tt|�|�}W d Q R X tj ||dd� d S )Nr� rG z(a)r )�allF)rN ) r r/ rI rK r� r` �errstater rh r} )rT r* r� r� �fnr� �gotr� r" r" r# �test_unary_functions� s z)TestMathPythonPython.test_unary_functionsr| c C sz t tj�d�tj�d�d��}|j}|j}|� d�}| �|�}tjdd�� tt|�||�}W d Q R X t j ||dd� d S )NrG )r� r� z(a, b)r )rz F)rN ) r r/ rI rK r� r� r` r{ r rh r� )rT r| r� r� r� r� r} r� r" r" r# �test_binary_functions� s z*TestMathPythonPython.test_binary_functionsc C sd t tj�d�tj�d�d��}|jd| j| jdd� |j}t�t� |j �|j �}t j ||dd� d S )NrG )r� r� ze = arctan2(sin(a), b)T)r! r' rK F)rN )r r/ rI rK r` r! r' r5 Zarctan2�sinr� r� rh r} )rT r� r} r� r" r" r# �test_df_use_case� s z%TestMathPythonPython.test_df_use_casec C s` t tj�d�tj�d�d��}|jd| j| jdd� |j}t�|j |j �}t j ||dd� d S )NrG )r� r� ze = sin(a + b)T)r! r' rK F)rN ) r r/ rI rK r` r! r' r5 r� r� r� rh r} )rT r� r} r� r" r" r# � test_df_arithmetic_subexpression� s z5TestMathPythonPython.test_df_arithmetic_subexpressionc C s� t dtj�d��|�i�}|jj|ks*t�|jd| j | j dd� |j }t� |j�}|j|jksbt�||jkspt�t j||dd� d S )Nr� rG z b = sin(a)T)r! r' rK F)rN )r r/ rI rK r� r� r� r r` r! r' r� r� rh r} )rT r� Z expect_dtyper� r} r� r" r" r# �check_result_type� s z&TestMathPythonPython.check_result_typec C sD | � tjtj� | � tjtj� | � tjtj� | � tjtj� d S )N)r� r/ Zint32r� Zint64r� )rT r" r" r# �test_result_types s z&TestMathPythonPython.test_result_typesc C s | � tjtj� d S )N)r� r/ Z complex128)rT r" r" r# �test_result_complex128 s z+TestMathPythonPython.test_result_complex128c C sJ t dtj�d�i�}d}tjt|d�� |jd| j| j d� W d Q R X d S )Nr� rG z#"mysin" is not a supported function)r� zmysin(a))r! r' ) r r/ rI rK r r� r, r` r! r' )rT r� r� r" r" r# �test_undefined_func s z(TestMathPythonPython.test_undefined_funcc C sJ t dtj�d�i�}d}tjt|d�� |jd| j| j d� W d Q R X d S )Nr� rG z1Function "sin" does not support keyword arguments)r� zsin(x=a))r! r' ) r r/ rI rK r r� r� r` r! r' )rT r� r� r" r" r# �test_keyword_arg s z%TestMathPythonPython.test_keyword_argN)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# ry � s ry c @ s e Zd ZdZdZdS )�TestMathPythonPandasr� rP N)r� r� r� r! r' r" r" r" r# r� s r� c @ s e Zd ZdZdZdS )�TestMathNumExprPandasr rP N)r� r� r� r! r' r" r" r" r# r� $ s r� c @ s e Zd ZdZdZdS )�TestMathNumExprPythonr r� N)r� r� r� r! r' r" r" r" r# r� ) s r� rG c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestScopec C s$ d}t �td tj|||d�� d S )Nz _var_s * 2rH )r! r' )rh r~ �_var_sr> r` )rT r! r' r5 r" r" r# �test_global_scope2 s zTestScope.test_global_scopec C sD d}t � �� }tjd|||d� t � �� }|�d� ||ks@t�d S )Nr7 zx + 1)r� r! r' �lcls)r rL r> r` r1 r )rT r! r' r� r� Zlcls2r" r" r# �test_no_new_locals8 s z TestScope.test_no_new_localsc C s8 d}t � �� }tjd||d� t � �� }||ks4t�d S )Nr7 zx + 1)r! r' )�globalsrL r> r` r )rT r! r' r� ZgblsZgbls2r" r" r# �test_no_new_globals@ s z TestScope.test_no_new_globalsN)r� r� r� r� r� r� r" r" r" r# r� 1 s r� c C s8 d} t jt| d�� tjdddd�dd� W d Q R X d S ) Nz Invalid engine 'asdf' passed)r� zx + yr7 rH )r� r4 �asdf)r� r! )r r� �KeyErrorr> r` )r� r" r" r# �test_invalid_engineH s r� c C s8 d} t jt| d�� tjdddd�dd� W d Q R X d S ) Nz Invalid parser 'asdf' passed)r� zx + yr7 rH )r� r4 r� )r� r' )r r� r� r> r` )r� r" r" r# �test_invalid_parserO s r� )r� r rP z dict[str, type[BaseExprVisitor]]�_parsersc C sR t | }|d| |�}x8|jD ].}d}tjt|d�� t||�� W d Q R X q W d S )Nzx + 1znodes are not implemented)r� )r� Zunsupported_nodesr r� r� r )r! r' Z VisitorClass�instr2 r� r" r" r# �test_disallowed_nodes] s r� c C s2 d}t jtdd�� tj|| |d� W d Q R X d S )Nzs +zinvalid syntax)r� )r! r' )r r� r� r> r` )r! r' r5 r" r" r# �test_syntax_error_exprsj s r� c C s6 d}d}t jt|d�� tj|| |d� W d Q R X d S )Nzs + tzname 's' is not defined)r� )r! r' )r r� rB r> r` )r! r' r5 r� r" r" r# �test_name_error_exprsp s r� �expressza + @bz@a + bz@a + @bc C sj d\}}|dkr<t jtdd�� tj|| |d� W d Q R X n*t jtdd�� tj|| |d� W d Q R X d S )N)r7 rH rP zThe '@' prefix is only)r� )r! r' zThe '@' prefix is not)r r� r� r> r` )r! r' r� r� r� r" r" r# �%test_invalid_local_variable_referencew s r� c C sd d\}}| dkr@d}t jt|d�� tjd| |d� W d Q R X n tjd| |d�}||| ks`t�d S )N)r7 rH r zVariables in expression .+)r� zsin + dotted_line)r! r' )r r� r r> r` r )r! r' r� Z dotted_liner� r� r" r" r# �test_numexpr_builtin_raises� s r� c C s4 d}t jtdd�� tjd|| |d� W d Q R X d S )N)�* g @zResolver of type .+)r� z1 + 2)� resolversr! r' )r r� r� r> r` )r! r' Zcannot_resolver" r" r# �test_bad_resolver_raises� s r� c C s. t jtdd�� tjd| |d� W d Q R X d S )Nz expr cannot be an empty string)r� � )r! r' )r r� r, r> r` )r! r' r" r" r# �test_empty_string_raises� s r� c C s. t jtdd�� tjd| |d� W d Q R X d S )Nz#only a single expression is allowed)r� z 1 + 1; 2 + 2)r! r' )r r� r� r> r` )r! r' r" r" r# �$test_more_than_one_expression_raises� s r� �cmp)r� r� r3 c C s� t dd� ttjji}|| � }|| � } || � }d|� d|� d�}d|� d|� d�}d|� d|� d �} x@||| fD ]2} d } tjt| d �� tj | ||d � W d Q R X qtW d S ) Nc S s t j�d�S )NrG )r/ rI r� r" r" r" r# r� � r� z0test_bool_ops_fails_on_scalars.<locals>.<lambda>zlhs z mid z rhsz mid and mid z(lhs z mid) & (mid z rhs)z;cannot evaluate scalar only bool ops|'BoolOp' nodes are not)r� )r! r' ) �int�floatr/ rI rK r r� r� r> r` ) r1 r� r3 r! r' �genr� r� r� r� rd r� r" r" r# � test_bool_ops_fails_on_scalars� s r� �otherz'x'z...zGH-28116)r )r c C sN t ddddgi�}|�d| � ��}tdddgdd�}tr>d |_t�||� d S )Nr� r� r� r4 zA == F)r� )r r` r r r� rh r} )r� r� rC rD r" r" r# �test_equals_various� s r� c C s* d}t j}tj|| |d�}||ks&t�d S )Nzinf + 1)r! r' )r/ �infr> r` r )r! r' r� rD rC r" r" r# �test_inf� s r� c C s� d}t �t��}tjd| |dd� W d Q R X t|�dks<t�|t|d j�ksRt�t �t��}tjd| |dd� W d Q R X t|�dks�t�|t|d j�ks�t�d S )Nz0The `truediv` parameter in pd.eval is deprecatedz1+1T)r! r' r> r7 r F) rh r r? r> r` r& r r- r' )r! r' r� �mr" r" r# �test_truediv_deprecated� s r� c C s� t ddgddggddgd�}||jdk }|jd| |d �}t�||� |d kr�d }tjt|d �� |jd | |d � W d Q R X n |jd | |d �}t�||� d S )Nr rG r7 r r` �count)r� z ~(cat > 0))r! r' r� z 'Not' nodes are not implemented)r� z not (cat > 0))r r` r� rh r| r r� r� )r! r' r� rD rC r� r" r" r# �test_negate_lt_eq_le� s r� c @ s0 e Zd Zej�ddddddgdg�dd� �Zd S ) � TestValidate�valuer7 �TruerH r� g @c C s0 d}t jt|d�� tjd|d� W d Q R X d S )Nz8For argument "inplace" expected type bool, received type)r� z2+2)rK )r r� r, r> r` )rT r� r� r" r" r# �test_validate_bool_args� s z$TestValidate.test_validate_bool_argsN)r� r� r� r r r� r� r" r" r" r# r� � s r� )x� __future__r � functoolsr � itertoolsr r� r �numpyr/ r Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandas.core.dtypes.commonr r r rP r> r r r r Zpandas._testingZ_testingrh Zpandas.core.computationr Z pandas.core.computation.enginesr r Z pandas.core.computation.exprr? Z computationr� r r r Z#pandas.core.computation.expressionsr r Zpandas.core.computation.opsr r r r r Zfixturer! ZPARSERSr' r) r* r+ r6 r; r< r@ rE �sorted�set� differencer� r( �ranger1 r3 rz Z skip_if_no_nerM r� r� r� r� r� r� r� r� r, ro rw rx ry r� r� r� rI rK r� r� r� r� ZPyTablesExprVisitorr� r� r r� r� r� r� r� r� r� r� r� r� r� r� r ZxfailrP r� r� r� r� r� r" r" r" r# �<module> s� f {Y^ &
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.08 |
proxy
|
phpinfo
|
Настройка