U
    sVÞcÈ  ã                   @   s,   d dl Zd dlmZmZ dd„ Zdd„ ZdS )é    N)ÚCategoricalÚSeriesc                  C   sB   t tj d¡ƒ} tj| dd…< d| dd…< |  ¡ }|dks>t‚d S )Niô  é   iˆ  é
   é   )r   ÚnpÚrandomZrandnÚnanÚnuniqueÚAssertionError)ZseriesÚresult© r   úL/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/series/methods/test_nunique.pyÚtest_nunique	   s
    r   c                  C   s@   t tg ƒƒ} |  ¡ dkst‚t ttjgƒƒ} |  ¡ dks<t‚d S )Nr   )r   r   r
   r   r   r	   )Zserr   r   r   Útest_nunique_categorical   s    r   )Znumpyr   Zpandasr   r   r   r   r   r   r   r   Ú<module>   s   	