a
    Pªf  ã                   @   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 u rdnd}t| ƒf| }tj d¡ |¡}||| 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default_rngZstandard_normalÚtmZassert_equalÚheadÚilocÚtail)r   Zframe_or_seriesÚndimÚshapeÚvalsÚobj© r   úb/var/www/ai-form-bot/venv/lib/python3.9/site-packages/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   )
Únumpyr   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   r   r   Ú<module>   s
   