3
���h� � @ s� d dl Zd dlZd dlmZ d dlmZmZ d dlZ d dl
j Z d dl
mZ dd� Zejjdejdg�dd � �Zd
d � Zejjdejdg�d d
� �Zejjde jdgd �e jdgd �g�dd� �ZdS )� N)�iNaT)�is_datetime64tz_dtype�needs_i8_conversion)� allow_na_opsc C s� | }t j|tdt|�d ��}|j� }ttj|j��}t |t
j �rdt
j j |�}|j
|_
tj||� nRt |t
j�r�t
j||jd�}t|j�r�|j� }tj||� nt j|�}tj||� d S )N� )�dtype)�np�repeat�range�len�unique�list�dict�fromkeys�values�
isinstance�pd�
MultiIndex� from_tuples�names�tm�assert_index_equal�Indexr r � normalize�array�assert_numpy_array_equal)�index_or_series_obj�obj�result�
unique_values�expected� r! �?/tmp/pip-build-5_djhm0z/pandas/pandas/tests/base/test_unique.py� test_unique
s
r# �null_objc C sV |}t |�stjd� n6t|�dk r0tjd� n t|tj�rNtjd| � d�� |j}t|j �rlt
|dd�< n | |dd�< t |�}t j
|tdt|�d ��}|||j d�}|j� }tj|j�}d d
� |D �}| g| } t|tj��r6tj| |j d�}
t|j ��r|j� }|
j� }
nt|tj��r(|
j|�}
tj||
� n t j| |j d�}
tj||
� d S ) Nz$type doesn't allow for NA operationsr z%Test doesn't make sense on empty datazMultiIndex can't hold '�'r � )r c S s g | ]}t j|�s|�qS r! )r Zisnull)�.0�valr! r! r" �
<listcomp>; s z$test_unique_null.<locals>.<listcomp>)r �pytest�skipr r r r r r r r �typer r r
r r r r r r �CategoricalIndexZset_categoriesr r r r ) r$ r r r �klass�repeated_valuesr Zunique_values_rawZunique_values_not_nullr r r! r! r" �test_unique_null" s8
r0 c C sB | }t j|tdt|�d ��}t|j� �}|jdd�|ks>t�d S )Nr F)�dropna)r r r
r r �nunique�AssertionError)r r r r! r! r" � test_nuniqueK s r4 c C s |}t |�stjd� n t|tj�r6tjd| � d�� |j}t|j�rTt |dd�< n | |dd�< t
|�}t j |t
dt|�d ��}|||jd�}t|tj�r�|j� t|j�ks�t�|jdd �t|j�d ks�t�nBt|j� �}|j� td|d �ks�t�|jdd �td|�k�st�d S )
Nz$type doesn't allow for NA operationszMultiIndex can't hold 'r% r r& r )r F)r1 )r r* r+ r r r r r r r r, r r r
r r- r2 �
categoriesr3 r �max)r$ r r r r. r/ Znum_unique_valuesr! r! r" �test_nunique_nullR s$
r7 �idx_or_series_w_bad_unicodeu �r& c C sV | }|j � }t|tj�r6tjdgtd�}tj||� n tjdgtd�}tj ||� d S )Nu �)r )
r r r r �objectr r r r r )r8 r r r r! r! r" �test_unique_bad_unicoden s r: )Znumpyr r* Z pandas._libsr Zpandas.core.dtypes.commonr r Zpandasr Zpandas._testingZ_testingr Zpandas.tests.base.commonr r# �markZ parametrize�nanr0 r4 r7 r ZSeriesr: r! r! r! r" �<module> s
) |