U
    *if                     @   s6   d dl Zd dlmZmZ d dlmZ G dd dZdS )    N)CategoricalSeriesc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )
TestUniquec                 C   s   t t dddg   d S )Nacb)r   uniqueZsort_values)self r
   Z/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/series/methods/test_unique.pytest_unique_data_ownership   s    z%TestUnique.test_unique_data_ownershipc                 C   st   t dgd }tj|d d d< | }t|dks6tt dgd dd}tj|d d d< | }t|dksptd S )Ngn?d      Zf4Zdtyper   npnanr   lenAssertionErrorr	   serresultr
   r
   r   test_unique   s    zTestUnique.test_uniquec                 C   s>   t dgd dd}tj|d d d< | }t|dks:td S )NZfoor   Or   r   r   r   r
   r
   r   test_unique_nan_object_dtype   s    z'TestUnique.test_unique_nan_object_dtypec                 C   sF   t dddd d d gtd}| }tjdddd gtd}t|| d S )N   r      r   )r   objectr   r   arraytmZassert_numpy_array_equal)r	   r   r   expectedr
   r
   r   test_unique_none#   s    zTestUnique.test_unique_nonec                 C   sP   t g }t|}| }t|| t tjg}t|}| }t|| d S )N)r   r   r   r   Zassert_categorical_equalr   r   )r	   catr   r   r
   r
   r   test_unique_categorical*   s    z"TestUnique.test_unique_categoricalN)__name__
__module____qualname__r   r   r   r!   r#   r
   r
   r
   r   r   
   s
   r   )	numpyr   Zpandasr   r   Zpandas._testingZ_testingr   r   r
   r
   r
   r   <module>   s   