U
    Mäfw  ã                   @   s<   d dl Zd dlmZ d dlmZ dd„ Zdd„ Zdd„ Z	dS )	é    N)Ú	DataFramec                 C   s  |t krdnd}t| ƒf| }tjj|Ž }||| d}t | ¡ |jd d… ¡ t | 	¡ |jdd … ¡ t | d¡|jdd… ¡ t | 	d¡|jdd… ¡ t | t|ƒd ¡|¡ t | 	t|ƒd ¡|¡ t | d¡| t| ƒd ¡¡ t | 	d¡| 	t| ƒd ¡¡ d S )	Né   é   )Úindexé   éûÿÿÿr   éýÿÿÿé   )
r   ÚlenÚnpÚrandomZrandnÚtmZassert_equalÚheadÚilocÚtail)r   Zframe_or_seriesÚndimÚshapeÚvalsÚobj© r   úM/tmp/pip-unpacked-wheel-eb6vo0j3/pandas/tests/frame/methods/test_head_tail.pyÚtest_head_tail_generic   s     r   c                 C   sŒ  t  |  ¡ | d d… ¡ t  |  ¡ | dd … ¡ t  |  d¡| dd… ¡ t  |  d¡| dd… ¡ t  |  d¡| d d… ¡ t  |  d¡| dd … ¡ t  |  d¡| d d… ¡ t  |  d¡| dd … ¡ |  ¡ }t t| ƒ¡d |_t  | ¡ |j	d d… ¡ t  | ¡ |j	dd … ¡ t  | d¡|dd… ¡ t  | d¡|dd… ¡ t  | d¡|j	d d… ¡ t  | d¡|j	dd … ¡ d S )Nr   r   r   éÿÿÿÿr   gš™™™™™¹?)
r   Úassert_frame_equalr   r   Úcopyr   Zaranger
   r   r   )Zfloat_frameZdfr   r   r   Útest_head_tail   s     r   c                  C   s*   t ƒ } t |  ¡ | ¡ t |  ¡ | ¡ d S )N)r   r   r   r   r   )Zempty_dfr   r   r   Útest_head_tail_empty5   s    r   )
Znumpyr   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   r   r   Ú<module>   s
   