Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/__pycache__/test_expressions.cpython-37.pyc
Назад
B �5�g�2 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlm Z m Z m Z d dl m Z e ej�dd�ed�dd�Ze ej�d d�ed�dd�Ze ed �� ed �d �ed �d�ed �d�d��Ze ed �� ed �d �ed �d�ed �d�d��Ze ejjdd dd�ed�dd�Ze ejjdd dd�ed�dd�Zed j�� Z ed j�� Z ed j�� Z ed j�� Z ej j!ej" dd�G dd� d��Z#dS )� N)� DataFrame�Index�Series)� expressionsi' � ZABCDZfloat64)�columns�dtype�d �A�BZfloat32�CZint64�DZint32)r r r r � )i' r )�size)�e r znot using numexpr)�reasonc @ s� e Zd Zdd� Zdd� Zeeed�dd��Zed�d d �Z ed�d d �Z ed�d d�Z e j �deeeejjdde�e�d� eeeeg�e j �dddg�dd� ��Zdd� Ze j �ddd d d d g�e j �d ee fe e fg�d!d"� ��Z e j �dd#d$d%d&d'd(g�e j �d ee fe e fg�d)d*� ��Z e j �d+ddg�e j �deeeeg�d,d-� ��Z!e j �d.d/d0d1g�d2d3� �Z"e j �d.d4d5d6g�d7d8� �Z#e j �d9e$dd:dd;gdd:dd;ggd<d=d>d?gd@�e$ddgddggd<d?gd@�fe$ddAdd;gddBdd;gdd:d:dCggd<d=d>d?gd@�e$ddgddgddggd<d?gd@�fg�dDdE� �Z%e j �dFdG�e j �dHdI�dJdK� ��Z&e j �dLdMe j'dNe j j(dOdP�dQ�dRdSg�e j �dTe$e)e*g�e j �dUdVdWg�dXdY� ���Z+dZS )[�TestExpressionsc C s4 t �� | _t�� | _t�� | _t�� | _t j | _ d S )N) �_frame�copy�frame�_frame2Zframe2�_mixed�mixed�_mixed2Zmixed2�expr� _MIN_ELEMENTS)�self�method� r �S/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/test_expressions.py� setup_method0 s z TestExpressions.setup_methodc C s | j t_ d S )N)r r )r r r r r �teardown_method8 s z TestExpressions.teardown_method)�flex�opnamec sZ |r� fdd�}� |_ n tt� �}t�d� || |�}t�d� t�� || |�}||fS )Nc s t | � �|�S )N)�getattr)�x�y)r# r r �<lambda>> � z)TestExpressions.call_op.<locals>.<lambda>FT)�__name__r$ �operatorr �set_use_numexpr�get_test_result)�df�otherr" r# �op�expected�resultr )r# r �call_op; s zTestExpressions.call_op)r" c C s� dt _ddddddg}xj|D ]b}| �||||�\}}|dkrr|jdkrX|jjd ksrt�ntd d � |jj D ��srt�t � ||� q W d S ) Nr �add�sub�mul�mod�truediv�floordivr �fc s s | ]}|j d kV qdS )r9 N)�kind)�.0r% r r r � <genexpr>V s z1TestExpressions.run_arithmetic.<locals>.<genexpr>) r r r2 �ndimr r: �AssertionError�allZdtypes�values�tm� assert_equal)r r- r. r" � operations�arithr1 r0 r r r �run_arithmeticL s z TestExpressions.run_arithmeticc C sf dt _t �d� ddddddg}x@|D ]8}| �||||�\}}t �� }|sRtd ��t�||� 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�nez Did not use numexpr as expected.N)r r Z set_test_moder2 r, r> rA rB ) r r- r. r" rC rD r1 r0 Z used_numexprr r r � run_binaryY s zTestExpressions.run_binaryc C s~ | � |||� t�d� |d }t�d� | �|||� x@tt|j��D ].}| � |jd d �|f |jd d �|f |� qHW d S )NFr T)rE r r+ rL �range�lenr �iloc)r r- r. r" Z binary_comp�ir r r � run_framej s zTestExpressions.run_framer- r � )r r" TFc C s | � |||� d S )N)rQ )r r- r" r r r �test_arithmeticy s z TestExpressions.test_arithmeticc C sp t j�d�}t j�d�}t�tjd ||d�}|r4t�t�tjd||d�}|rPt�t�tjd||d�}|slt�d S )NiAB r �evaluate�+)�np�random�randnr �_can_use_numexprr* r3 r> )r �arrayZarray2r1 r r r � test_invalid� s z TestExpressions.test_invalidz opname,op_str)r3 rU )r4 �-)r5 �*)r7 �/)�powz**z left,rightc sN � ���fdd�}t �d� |� t �d� t �d� |� t �� |� d S )Nc sd �dkr d S t t��} tj| � � dd�}tj| � � dd�}t�||� t�| ���d�}|r`t�d S )Nr_ T)� use_numexprFrT )r$ r* r rT rA �assert_numpy_array_equalrY r> )r/ r1 r0 )�left�op_strr# �rightr r �testit� s z/TestExpressions.test_binary_ops.<locals>.testitFTr )r r+ �set_numexpr_threads)r r# rc rb rd re r )rb rc r# rd r �test_binary_ops� s z TestExpressions.test_binary_ops)rF �>)rG �<)rH z>=)rI z<=)rJ z==)rK z!=c sN � ���fdd�}t �d� |� t �d� t �d� |� t �� |� d S )Nc sh � d } �d }t t��}tj|� | dd�}tj|� | dd�}t�||� t�|��|d�}|rdt�d S )Nr T)r` FrT )r$ r* r rT rA ra rY r> )Zf12Zf22r/ r1 r0 )rb rc r# rd r r re � s z3TestExpressions.test_comparison_ops.<locals>.testitFTr )r r+ rf )r r# rc rb rd re r )rb rc r# rd r �test_comparison_ops� s z#TestExpressions.test_comparison_ops�condc sJ � �fdd�}t �d� |� t �d� t �d� |� t �� |� d S )Nc sX t j�jt jd�} | �� � t�| �j�jd �}t �| �j�jd �}t� ||� d S )N)r r ) rV �empty�shapeZbool_�fillr �wherer@ rA ra )�cr1 r0 )rk r- r r re � s z*TestExpressions.test_where.<locals>.testitFTr )r r+ rf )r rk r- re r )rk r- r � test_where� s zTestExpressions.test_wherez op_str,opname)r^ r7 )z//r8 )z**r_ c C s* t tj�d�dktj�d�dkd��}d|� d�}tt|�}t�|�}tj t |d�� |||� W d Q R X tj t |d�� ||j |j � W d Q R X tj t |d�� ||j d� W d Q R X tj t |d�� |d|j � W d Q R X tj t |d�� |d|� W d Q R X tj t |d�� ||d� W d Q R X d S ) N� g �?)�a�bz operator 'z!' not implemented for bool dtypes)�matchTF) r rV rW �randr$ r* �re�escape�pytestZraises�NotImplementedErrorrs rt )r rc r# r- �msgr9 �err_msgr r r �!test_bool_ops_raise_on_arithmetic� s & z1TestExpressions.test_bool_ops_raise_on_arithmetic)rU r3 )r] r5 )r\ r4 c C s� d}t tj�|�dktj�|�dkd��}dddd�}dd d d �}tt|�}tt||| �}|d krjd S tjd dd���f tjdd��& |||�} |||�} t� | | � W d Q R X tjdd��. ||j |j �} ||j |j �} t� | | � W d Q R X tjdd��* ||j d �} ||j d �} t� | | � W d Q R X tjdd��* |d|j �} |d|j �} t� | | � W d Q R X tjdd��& |d|�} |d|�} t� | | � W d Q R X tjdd��& ||d �} ||d �} t� | | � W d Q R X W d Q R X d S )Nrr g �?)rs rt �|�&�^)rU r] r\ �or_�and_�xor)r~ r r� r\ T� )Z min_elementsF)Zcheck_stacklevel) r rV rW rv r$ r* rA r` Zassert_produces_warning�assert_frame_equalrs rt Zassert_series_equal) r rc r# �nr- �subsZ sub_funcsr9 Zfe�r�er r r � test_bool_ops_warn_on_arithmetic sB & z0TestExpressions.test_bool_ops_warn_on_arithmeticztest_input,expectedr Zaars rt rp r )r � r Zbbc C s> |j d d �ddgf �|j d d �ddgf �}t�||� d S )Nrs r )�locrK rA r� )r Z test_inputr0 r1 r r r � test_bool_ops_column_name_dtype; s .z/TestExpressions.test_bool_ops_column_name_dtyperD )r3 r4 r5 r6 r7 r8 �axis)r r c C s� | j }|dkr$| j jdd d �f }n| j jd d �df }dt_t||�}t�d� |||d�}t�d� |||d�}t�||� d S )Nr r F)r� T)r rO r r r$ r+ rA r� )r r� rD r- r. Zop_funcr0 r1 r r r �test_frame_series_axisU s z&TestExpressions.test_frame_series_axisr/ �__mod__�__rmod__zGH-36552)r )Zmarks� __floordiv__� __rfloordiv__�box�scalar���r� c C s� dt _t�dd�}||�}t||�}||�}t �d� ||�}t �d� t�||� xlt|�D ]`\} } |t kr~|j | df } n|| } ytt | �|�|�}W n t k r� Y q^X | |ks^t �q^W d S )Nr i����2 FT)r r rV Zaranger$ r+ rA rB � enumerater rO �int�ZeroDivisionErrorr> ) r r/ r� r� �data�objr r1 r0 rP �elemZ scalar_resultr r r �,test_python_semantics_with_numexpr_installedm s$ z<TestExpressions.test_python_semantics_with_numexpr_installedN),r) � __module__� __qualname__r r! � staticmethod�bool�strr2 rE rL rQ ry �markZ parametrize�_integer� _integer2rV rW �randintrm r r r r rS r[ �_array�_array2� _array_mixed� _array_mixed2rg rj rq r} r� r r� r� �paramZxfailr r r� r r r r r . sz /$ r )$r* rw �numpyrV ry Zpandas._testingZ_testingrA Zpandas.core.apir r r Zpandas.core.computationr r rW rX �listr r r Zastyper r r� r� r� r@ r� r� r� r� r� ZskipifZ USE_NUMEXPRr r r r r �<module> s6
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка