3
���h9& �
@ s� d dl mZ d dlZd dlZd dlZd dlZd dlj Z
d dl m Z ej
ejejgZejejgZddgZddgZddgZejdd � �Zejjd
e�ejjdeed �d d
� ��Zejjd
e�ejjdeed �ejjdddgddgd �dd� ���Z ejjd
e�ejjdeed �ejjdddgddgd �dd� ���Z ejjd
e�ejjdeed �ejjdddgddgd �ejjdddgddgd �dd� ����Z ejjd
e�ejjdeed �ejjdddg�dd� ���Z ejjd
ej g�ejjdeed �ejjde�ejj!d �d d � ����Z"ejjdeed �d d � �Z#ejjdeed �ejjd
e�d!d"� ��Z$d#d$� Z%ejjd%ej&d&d'd(gd)d*�ej&d&d+d gd,d*�ej'd-d.d/g�ej'd-d.d/g�j(d0�ej'd-d.d/g�j)d1d2�g�d3d4� �Z*ejjd5e+ee,g�d6d7� �Z-d8d9� Z.d:d;� Z/dS )<� )�dequeN)� SparseArrayTF�sparseZdensec C sL t jjddddd�} t jjddddd�}d| ddd�< d|ddd�< | |fS ) zO
A pair of random, length-100 integer-dtype arrays, that are mostly 0.
r �
�d �int64)�dtypeN� � )�np�random�randint)�a1�a2� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/series/test_ufunc.py�arrays_for_binary_ufunc s
r �ufunc)�idsc C s� t jjddddd�}d|d d d�< |r:t|tjdd�d�}ttjd d� �}d}tj |||d�}| |�}tj | |�||d�}t
j ||� d S )Nr r r )r � �name)�indexr ) r r r
r �pd� SparseDtype�list�string�
ascii_letters�Series�tm�assert_series_equal)r r �arrayr r �series�result�expectedr r r �test_unary_ufunc s r$ �flipZflippedZstraightc
C s� |\}}|r4t |tjdd�d�}t |tjdd�d�}d}tj||d�}|}||f} ||f}
| rnt| �} t|
�}
tj|| � |d�} ||
� } tj| | � d S )Nr r )r r )r )r r r r �reversedr r )
r% r r r r r r r! �other�
array_args� series_argsr# r" r r r � test_binary_ufunc_with_array0 s r* c
C s� |\}}|r4t |tjdd�d�}t |tjdd�d�}d}tj||d�}tj||d�jd�}||f} ||f}
| r~t| �} t|
�}
tj|| � |d�} ||
� } tj| | � d S )Nr r )r r )r ) r r r r �IndexZastyper& r r )
r% r r r r r r r! r' r( r) r# r" r r r � test_binary_ufunc_with_indexJ s r, �shuffleZ unalignedZalignedc C s� |\}}|r4t |tjdd�d�}t |tjdd�d�}d}tj||d�}tj||d�} tjjt|��}
|r�| j|
�} | r�| j |�d j
} q�|j | �d j
} n|j
} ||f} || f}
| r�t t | ��} t t |
��}
tj|| � | |d�}||
� }t
j||� d S )Nr r )r r )r )r r )r r r r r r Z permutation�lenZtakeZalignr �tupler& r r )r% r- r r r r r r r! r' �idxr r( r) r# r" r r r � test_binary_ufunc_with_seriesf s,
r1 c C s| |\}}|rt |�}d}tj|dd�}||f}||f} |rRtt|��}tt| ��} tj| | � dd�}
| |� } tj| |
� d S )Nr r )r )r r r r/ r& r r ) r r r% r r �_r' r! r) r( r# r" r r r �test_binary_ufunc_scalar� s r3 z$ignore:divide by zero:RuntimeWarningc
C s� |r| t jkrtjd� |\}}d||dk< d||dk< |rdt|tjdd�d�}t|tjdd�d�}tj|�}tj|�}|r�|jdd�}| ||�}t |t
�s�t �| ||�} t | t
�s�t �t j
| d tj|d �� t j
| d tj|d �� d S )Nz sparse divmod not implemented.� r r )r )�frac)r �divmod�pytest�skipr r r r �sample�
isinstancer/ �AssertionErrorr r )
r r r- r r r �s1�s2r# r" r r r �"test_multiple_output_binary_ufuncs� s$
r> c C s� |\}}| rt |�}tj|dd�}tj|�}tj|�}t|t�sDt�t|t�sRt�tj |d tj|d dd�� tj |d tj|d dd�� d S )Nr )r r r4 )
r r r r �modfr: r/ r; r r )r r r r2 r! r" r# r r r �test_multiple_output_ufunc� s
r@ c C s@ |\}}t j|dd�}t j|dd�}| ||�}|jd ks<t�d S )N�a)r �b)r r r r; )r r r r r r< r= r" r r r �#test_binary_ufunc_drops_series_name� s
rC c C s~ G dd� d�} t j| d�| d�g�}tj|�}tjt j||�tjt j||��� tjt j|| d��tjt j|| d���� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z$test_object_series_ok.<locals>.Dummyc S s
|| _ d S )N)�value)�selfrD r r r �__init__� s z-test_object_series_ok.<locals>.Dummy.__init__c S s | j |j S )N)rD )rE r' r r r �__add__� s z,test_object_series_ok.<locals>.Dummy.__add__N)�__name__�
__module__� __qualname__rF rG r r r r �Dummy� s rK r r4 )r r r r r r �add)rK ZarrZserr r r �test_object_series_ok� s
"rM �valuesr4 r r r )r r z Sparse[int]�2000Z2010Z2001ZCET�D)�freqc C s& t j| �}tjj|�| d ks"t�d S )Nr4 )r r r �maximum�reducer; )rN rA r r r � test_reduce� s
rT �type_c C sX t jdddgdd�}| dddg�}tj||�}t jtj|j� |�dd�}tj||� d S )Nr4 r r r )r r
� )r r r rL Zto_numpyr r )rU rA rB r" r# r r r � test_binary_ufunc_other_types s
rW c s^ G � fdd�d�� t j� d�� d�g�} tj| � d��}t j� d�� d�g�}tj||� d S )Nc s< e Zd Zdd� Zdd� Zed�� fdd� Zed�dd �Zd
S ) z#test_object_dtype_ok.<locals>.Thingc S s
|| _ d S )N)rD )rE rD r r r rF s z,test_object_dtype_ok.<locals>.Thing.__init__c S s t |d|�}t| �| j| �S )NrD )�getattr�typerD )rE r' r r r rG s z+test_object_dtype_ok.<locals>.Thing.__add__)�returnc s t |�� ko| j|jkS )N)rY rD )rE r' )�Thingr r �__eq__ s z*test_object_dtype_ok.<locals>.Thing.__eq__c S s d| j � d�S )NzThing(�))rD )rE r r r �__repr__! s z,test_object_dtype_ok.<locals>.Thing.__repr__N) rH rI rJ rF rG �boolr\ �strr^ r )r[ r r r[ s r[ r4 r r )r r r rL r r )�sr" r# r )r[ r �test_object_dtype_ok s
rb c
C sH t jdddg�} tjdddg�}tjt�� tjj| |� W d Q R X d S )Nr4 r r ) r r r r r7 Zraises�NotImplementedError�subtract�outer)ra �or r r �
test_outer* s rg )0� collectionsr r Znumpyr r7 Zpandasr Zpandas._testingZ_testingr Z
pandas.arraysr Zpositive�floor�expZ UNARY_UFUNCSrL Z logaddexpZ
BINARY_UFUNCSZSPARSEZ
SPARSE_IDSZSHUFFLEZfixturer �markZ parametrizer$ r* r, r1 r3 r6 �filterwarningsr> r@ rC rM r Z to_datetimeZ tz_localizeZ to_periodrT r r/ rW rb rg r r r r �<module> sZ
& & (' ! $
|