3
���h3 �
@ s� d dl Z d dlZd dlZd dlmZ d dlmZmZm Z m
Z
d dl Z d dl m
Z
mZmZmZ d dljZejjdd)d*d+d,d-d.d/g�ejjdee
g�dd� ��Zdd� Zdd� Zejjedd �d d � �Zd d � Zd!d"� Z d#d$� Z d%d&� Z d'd(� Z dS )0� N)�PYPY)�is_categorical_dtype�is_datetime64_dtype�is_datetime64tz_dtype�is_object_dtype)� DataFrame�Index�
IntervalIndex�Seriesz op_name, op�add�+�sub�-�mul�*�mod�%�pow�**�truediv�/�floordiv�//�klassc C s^ | j j� }d}dj|||g�}|t| |�jks2t�dj|||g�}|t| d| �jksZt�d S )N�other� �r)�__name__�lower�join�getattr�__doc__�AssertionError)r Zop_name�opZoperand1Zoperand2Z expected_str� r$ �=/tmp/pip-build-5_djhm0z/pandas/pandas/tests/base/test_misc.py�test_binary_ops_docstring s
r& c C s� | }t |jt�rtjd� t|�dk r0tjd� tj|d< |d k}|jd sRt �|jd sbt �|d k}|jd sxt �|jd s�t �d |k}|jd s�t �|jd s�t �d |k}|jd s�t �|jd s�t �t
|j �s�t |j ��r2d}tj
t|d�� d |k W d Q R X tj
t|d�� |d k W d Q R X nXd |k}|jd �sLt �|jd �s^t �|d k }|jd �sxt �|jd �s�t �d S )Nz(IntervalIndex doesn't support assignment� z%Test doesn't make sense on empty datar zInvalid comparison)�match)�
isinstance�indexr �pytest�skip�len�np�nanZiatr" r �dtyper �raises� TypeError)�series_with_simple_index�series�result�msgr$ r$ r% �test_none_comparison. s<
r7 c
C s� | }x d
D ]}t ||d �d k s
t�q
W x dD ]}t||� s,t�q,W d
}tjt|d �� |j� W d Q R X |jd ksxt�|jt |�ks�t�t
d g�j� d ks�t�t d g�j� d ks�t�d S )N�shaper0 �T�nbytes�flags�strides�itemsize�base�dataz6can only convert an array of size 1 to a Python scalar)r( r' )r8 r0 r9 r: )r; r< r= r> r? ) r r" �hasattrr+ r1 �
ValueError�item�ndim�sizer- r r
)�index_or_series_obj�obj�pr6 r$ r$ r% � test_ndarray_compat_properties^ s
rH znot relevant for PyPy)�reasonc C s� | }|j � }|j dd�}t|�p2t|t�o2t|j�}t|j�pRt|t�oRt|jj�}t|�dkr~|| kordkn s�t�n"|s�|r�||ks�t�n ||ks�t�|t j
|� }t |�dk s�t�d S )NT)�deepr �d ) � memory_usager r) r
r* r r0 r- r" �sys� getsizeof�abs)rE rF �resZres_deepZ is_objectZis_categoricalZdiffr$ r$ r% �test_memory_usaget s
rQ c C s: | }|j dd�}|j dd�}|jj � }||| ks6t�d S )NT)r* F)rL r* r" )r3 r4 � total_usage�non_index_usage� index_usager$ r$ r% �#test_memory_usage_components_series� s
rU c C s: | }|j dd�}|j dd�}|jj � }||| ks6t�d S )NT)r* F)rL r* r" )Z
narrow_seriesr4 rR rS rT r$ r$ r% �*test_memory_usage_components_narrow_series� s
rV c C s� | }t |tj�rtjd� t|dd�}tj||�}d| koHt|�kn sRt �tj||t
t|��d�}d| ko�t|�kn s�t �d S )Nz-np.searchsorted doesn't work on pd.MultiIndexr )�default)Zsorter) r) �pd�
MultiIndexr+ r, �maxr. Z searchsortedr- r" �range)rE rF Zmax_objr* r$ r$ r% �test_searchsorted� s
r\ c C s� t | �dkrtjd� nt| tj�r.tjd� tj| �}| d |jd ksNt�| d |jd ksdt�| d
|jd kszt�t | �}| d | |d ks�t�d|� d|� �}tj t
|d�� | | W d Q R X d }tj t
|d�� |j| W d Q R X d S )
Nr z%Test doesn't make sense on empty dataz(Can't instantiate Series from MultiIndex� r' zindex z' is out of bounds for axis 0 with size )r( z*single positional indexer is out-of-bounds���r^ r^ ) r- r+ r, r) rX rY r
Zilocr" r1 �
IndexError)r* r4 rD r6 r$ r$ r% �test_access_by_position� s
r` c C sX t jddg�jt jdg��\} }tjtjdgtjd�| � tjtjdgtjd�|� d S )N�A�Br r' )r0 r^ ) rX r Zget_indexer_non_unique�tmZassert_numpy_array_equalr. �arrayZintpZint64)�indexes�missingr$ r$ r% �*test_get_indexer_non_unique_dtype_mismatch� s rg )r r )r
r )r r )r r )r r )r r )r r ) rM Znumpyr. r+ Z
pandas.compatr Zpandas.core.dtypes.commonr r r r ZpandasrX r r r r
Zpandas._testingZ_testingrc �markZ parametrizer& r7 rH ZskipifrQ rU rV r\ r` rg r$ r$ r$ r% �<module> s2
0 |