3
Udc                 @   s@   d dl Zd dlZd dlmZmZmZ d dljZ	G dd dZ
dS )    N)Categorical
MultiIndexSeriesc               @   s   e Zd Zdd Zdd ZdS )TestSeriesCountc          
   C   s
  |j  t|ksttj|d d d< |j  tj|j ks>ttjt	ddddtj
ddgg}ttjt||d}|j dd}tdddgddtj
gd}tj|| tj
|jdddg< tj|j dd|d  tjd	d
$ tjtjdgj  dkstW d Q R X d S )N   Zaabbcc   )index)level   r      Zuse_inf_as_naTz1990/1/1)countlenAssertionErrornpNaNisfinitesumr   Zfrom_arrayslistnanr   ZarangetmZassert_series_equalZilocpdZoption_context	Timestamp)selfZdatetime_seriesmitsleftright r   Z/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/series/methods/test_count.py
test_count	   s     zTestSeriesCount.test_countc             C   s@   t ttjddtjgdddddgdd}|j }|dks<td S )Nr   r   r      r
   T)
categoriesZordered)r   r   r   r   r   r   )r   Zserresultr   r   r   test_count_categorical   s
    $z&TestSeriesCount.test_count_categoricalN)__name__
__module____qualname__r   r#   r   r   r   r   r      s   r   )Znumpyr   Zpandasr   r   r   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   
