U
    sVc9                     @   s.   d dl mZmZ d dlmZ G dd dZdS )    )	DataFrameSeriesNc                   @   s   e Zd Zdd Zdd ZdS )TestDataFrameCountc                 C   s   t  }|d}t|tst|d}t|ts6tt tdd}|d}td|jd}t|| t tdd}|d}td|j	d}t|| t  }| }tdg d}t|| d S )N   r   
   )index)columns)
r   count
isinstancer   AssertionErrorranger   tmassert_series_equalr   )selfframeZct1Zct2dfresultexpected r   I/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/frame/methods/test_count.py
test_count	   s"    



zTestDataFrameCount.test_countc                 C   sD   t |j}t |j}t| |  t|d|d d S )Nr   )r   Z_seriesr   r   r	   )r   Zfloat_string_framedmr   r   r   r   test_count_objects"   s    

z%TestDataFrameCount.test_count_objectsN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   )Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   