U
    >vf;  ã                   @   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d}t ||¡ d S )Né   r   é
   )Úindex)ÚcolumnsZint64)Zdtype)
r   ÚcountÚ
isinstancer   ÚAssertionErrorÚranger   ÚtmÚassert_series_equalr   )ÚselfÚframeZct1Zct2ÚdfÚresultÚexpected© r   úI/tmp/pip-unpacked-wheel-vdrwu74i/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   