3
���h�6 � @ s\ d dl Z d dlZd dlZd dlmZ d dlZd dljZ d dl
m Z m Z m
Z
d dlmZ e edd�ed�dd �Ze ed
d�ed�dd �Ze ed j� ed jd
�ed jd�ed jd�d��Ze ed j� ed jd
�ed jd�ed jd�d��Ze ejjdd
dd�ed�dd �Ze ejjdd
d d�ed�dd �Zej j ej dd�G dd� d��Z dS ) � N)�randn)� DataFrame�Index�Series)� expressionsi' � �ABCDZfloat64)�columns�dtype�d �A�BZfloat32�CZint64�DZint32)r r
r r � �' )�size�e znot using numexpr)�reasonc
@ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd[d
d �Zej j
d e e e e
jjd
de
je �d� eeeeg�dd� �Zdd� Zej j
dd\d]d^d_d`g�ej j
d eefeefg�d d!� ��Zej j
ddadbdcdddedfg�ej j
d eefeefg�d.d/� ��Zej j
d0d d1g�ej j
d eeeeg�d2d3� ��Zej j
d4dgdhdig�d7d8� �Zej j
d4djdkdlg�d9d:� �Zej j
d;e d
d<dd=gd
d<dd=ggd>d?d@dAgdB�e d1d1gd1d1ggd>dAgdB�fe d
dCdd=gd
dDdd=gd
d<d<dEggd>d?d@dAgdB�e d1d1gd1d1gd1d1ggd>dAgdB�fg�dFdG� �Z ej j
dHdm�ej j
dJdn�dKdL� ��Z ej j
dMdNej dOej j dPdQ�dR�dSdTg�ej j
dUe e!e"g�ej j
dVdodWg�dXdY� ���Z#dZS )p�TestExpressionsc C s4 t j� | _tj� | _tj� | _tj� | _t j
| _
d S )N) �_frame�copy�frame�_frame2�frame2�_mixed�mixed�_mixed2Zmixed2�expr�
_MIN_ELEMENTS)�self�method� r" �?/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_expressions.py� setup_method( s
z TestExpressions.setup_methodc C s | j t_ d S )N)r r )r r! r" r" r# �teardown_method0 s z TestExpressions.teardown_methodc s� dt _ddddddg}x�dD ]�}x�|D ]�� |rB� fd
d �}� |_n
tt� �}t jd � |||�}t jd� |||�}� dkr�|jd kr�|jjd
ks�t �nt
dd� |j j D ��s�t �t
j||� q&W q W d S )Nr �add�sub�mul�mod�truediv�floordivTFc s t | � �|�S )N)�getattr)�x�y)�arithr" r# �<lambda>: s z0TestExpressions.run_arithmetic.<locals>.<lambda>r �fc s s | ]}|j d kV qdS )r1 N)�kind)�.0r- r" r" r# � <genexpr>G s z1TestExpressions.run_arithmetic.<locals>.<genexpr>)TF)r r �__name__r, �operator�set_use_numexpr�ndimr
r2 �AssertionError�all�dtypes�values�tm� assert_equal)r �df�other�
operations� test_flex�op�expected�resultr" )r/ r# �run_arithmetic3 s"
z TestExpressions.run_arithmeticc s� dt _t jd� ddddddg}x�dD ]�}x~|D ]v� |rL� fd
d �}� |_n
tt� �}t jd � |||�}t jd� t j� |||�}t j� }|s�td ��t j
||� q0W q&W d
S )z�
tests solely that the result is the same whether or not numexpr is
enabled. Need to test whether the function does the correct thing
elsewhere.
r T�gt�lt�ge�le�eq�neFc s t | � �|�S )N)r, )r- r. )r/ r" r# r0 W s z,TestExpressions.run_binary.<locals>.<lambda>z Did not use numexpr as expected.N)TF) r r Z
set_test_moder5 r, r6 r7 Zget_test_resultr9 r= r> ) r r? r@ rA rB rC rD rE Z used_numexprr" )r/ r# �
run_binaryJ s"
zTestExpressions.run_binaryTc C s| | j ||� |r8tjd� |d }tjd� | j||� x>tt|j��D ],}| j |jd d �|f |jd d �|f � qHW d S )NFr T)rF r r7 rM �range�lenr �iloc)r r? r@ rM Z binary_comp�ir" r" r# � run_framee s
zTestExpressions.run_framer? r � )r c C s2 dd� |j jD �}t|�dk}| j|||d� d S )Nc S s h | ]
}|j �qS r" )r2 )r3 r- r" r" r# � <setcomp>� s z2TestExpressions.test_arithmetic.<locals>.<setcomp>r )rM )r; r< rO rR )r r? ZkindsZshouldr" r" r# �test_arithmetict s z TestExpressions.test_arithmeticc C s� t jtjd | j| jd�}| s"t�t jtjd| j| jd�}| sDt�t jtjd| j| jd�}| sft�t jtjd| j| jd�}|s�t�d S )N�evaluate�+)r �_can_use_numexprr6 r& r r9 r r )r rE r" r" r# � test_invalid� s
z TestExpressions.test_invalidz
opname,op_strr& rW r' �-r( �*r* �/�pow�**z
left,rightc sN � ���fdd�}t jd� |� t jd� t jd� |� t j� |� d S )Nc s� �dkr d S t t��} tj| �� � d�}|� jks6t�tj| � � dd�}tj| � � dd�}t|t�rrt j
||� nt j ||j � tj| ���d�}| s�t�d S )Nr] rV T)� use_numexprF)
r, r6 r rX �_is_mixed_typer9 rV �
isinstancer r= �assert_frame_equal�assert_numpy_array_equalr< )rC rE rD )�left�op_str�opname�rightr" r# �testit� s
z/TestExpressions.test_binary_ops.<locals>.testitFTr )r r7 �set_numexpr_threads)r rf re rd rg rh r" )rd re rf rg r# �test_binary_ops� s
z TestExpressions.test_binary_opsrG �>rH �<rI �>=rJ �<=rK �==rL �!=c sN � ���fdd�}t jd� |� t jd� t jd� |� t j� |� d S )Nc s� � d } �d }t t��}tj|�� | d�}|� jks:t�tj|� | dd�}tj|� | dd�}t|t�rvt j
||� nt j ||j � tj|��|d�}| s�t�d S )Nr rV T)r_ F)
r, r6 r rX r` r9 rV ra r r= rb rc r< )Zf12Zf22rC rE rD )rd re rf rg r" r# rh � s
z3TestExpressions.test_comparison_ops.<locals>.testitFTr )r r7 ri )r rf re rd rg rh r" )rd re rf rg r# �test_comparison_ops� s
z#TestExpressions.test_comparison_ops�condFc sJ � �fdd�}t jd� |� t jd� t jd� |� t j� |� d S )Nc sX t j�jt jd�} | j� � tj| �j�jd �}t j| �j�jd �}tj ||� d S )N)r
r )
�np�empty�shapeZbool_�fillr �wherer< r= rc )�crE rD )rr r? r" r# rh � s
z*TestExpressions.test_where.<locals>.testitFTr )r r7 ri )r rr r? rh r" )rr r? r# �
test_where� s
zTestExpressions.test_wherez
op_str,opname�//r+ c - C s. t tjjd�dktjjd�dkd��}dt|�� d�}tt|�}tj|�}t j
t |d�� |||� W d Q R X t j
t |d�� ||j |j
� W d Q R X t j
t |d�� ||j d� W d Q R X t j
t |d�� |d|j � W d Q R X t j
t |d�� |d|� W d Q R X t j
t |d�� ||d� W d Q R X d S ) N�
g �?)�a�bz operator z not implemented for bool dtypes)�matchTF)r rs �random�rand�reprr, r6 �re�escape�pytestZraises�NotImplementedErrorr| r} )r re rf r? �msgr1 �err_msgr" r" r# �!test_bool_ops_raise_on_arithmetic s &
z1TestExpressions.test_bool_ops_raise_on_arithmeticc 4 C s� d}t tjj|�dktjj|�dkd��}dddd�}dd d
d �}tt|�}tt||| �}|d krjd S tjd
dd���f tjdd��& |||�} |||�}
tj | |
� W d Q R X tjdd��. ||j
|j �} ||j
|j �}
tj | |
� W d Q R X tjdd��* ||j
d
�} ||j
d
�}
tj | |
� W d Q R X tjdd��* |d|j
�} |d|j
�}
tj | |
� W d Q R X tjdd��& |d|�} |d|�}
tj | |
� W d Q R X tjdd��& ||d
�} ||d
�}
tj | |
� W d Q R X W d Q R X d S )Nr{ g �?)r| r} �|�&�^)rW r[ rZ �or_�and_�xor)r� r� r� rZ T� )Z min_elementsF)Zcheck_stacklevel)
r rs r r� r, r6 r= r_ Zassert_produces_warningrb r| r} Zassert_series_equal) r re rf �nr? �subsZ sub_funcsr1 Zfe�r�er" r" r# � test_bool_ops_warn_on_arithmetic sB &
z0TestExpressions.test_bool_ops_warn_on_arithmeticztest_input,expectedr Zaar| r} rx r
)r � r Zbbc C s> |j d d �ddgf j|j d d �ddgf �}tj||� d S )Nr| r
)�locrL r= rb )r Z
test_inputrD rE r" r" r# � test_bool_ops_column_name_dtypeN s .z/TestExpressions.test_bool_ops_column_name_dtyper/ r) �axisc C s� | j }|dkr$| j jdd d �f }n| j jd d �df }dt_t||�}tjd� |||d�}tjd� |||d�}tj||� d S )Nr r F)r� T)r rP r r r, r7 r= rb )r r� r/ r? r@ Zop_funcrD rE r" r" r# �test_frame_series_axish s
z&TestExpressions.test_frame_series_axisrC �__mod__�__rmod__zGH-36552)r )Zmarks� __floordiv__�
__rfloordiv__�box�scalarr� c C s� dt _tjdd�}||�}t||�}||�}t jd� ||�}t jd� tj||� xlt|�D ]`\} }
|t kr~|j
| df } n|| } ytt |
�|�|�}W n t k
r� Y q^X | |ks^t
�q^W d S )Nr �2 FTi���)r r rs Zaranger, r7 r= r> � enumerater rP �int�ZeroDivisionErrorr9 ) r rC r� r� �data�objr! rE rD rQ �elemZ
scalar_resultr" r" r# �,test_python_semantics_with_numexpr_installed� s$
z<TestExpressions.test_python_semantics_with_numexpr_installedN)T)r& rW )r' rZ )r( r[ )r* r\ )r] r^ )rG rk )rH rl )rI rm )rJ rn )rK ro )rL rp )r\ r* )rz r+ )r^ r] )rW r& )r[ r( )rZ r' )r& r' r( r) r* r+ )r r ���)$r5 �
__module__� __qualname__r$ r% rF rM rR r� �markZ parametrize�_integer� _integer2rs r �randintru r r r r rU rY rj rq ry r� r� r r� r� �paramZxfailr r r� r" r" r" r# r & sp
$ $ /$
r )r r )r r ) r6 r� Znumpyrs Z numpy.randomr r� Zpandas._testingZ_testingr= Zpandas.core.apir r r Zpandas.core.computationr r �listr r r Zastyper r r r� r� r� r� ZskipifZ _USE_NUMEXPRr r" r" r" r# �<module> s0
|