U
    *if                     @   sh   d dl Zd dlZd dlmZ dd Zdd Zdd Z	dd	 Z
d
d Zdd Zdd Zdd Zdd ZdS )    Nc                  C   sp   t jddddgddddgddddd	gd
} | jdd}t jdddgt jjddgddgdd}t|| d S )N         r   num_legs	num_wingsfalcondogcatantindexF)sort   )r   r   r   r   r   r   r   r   namesdatar   pd	DataFramevalue_countsSeries
MultiIndexZfrom_arraystmassert_series_equaldfresultexpected r!   _/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/frame/methods/test_value_counts.py%test_data_frame_value_counts_unsorted   s    
 r#   c                  C   sp   t jddddgddddgddddd	gd
} | jdd}t jdddgt jjddgddgdd}t|| d S )Nr   r   r   r   r   r   r	   r
   r   r   T)Z	ascendingr   )r   r   r   r   r   r   r   r   r   r   r!   r!   r"   &test_data_frame_value_counts_ascending   s    
 r$   c                  C   sl   t jddddgddddgddddd	gd
} |  }t jdddgt jjddgddgdd}t|| d S )Nr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r!   r!   r"   $test_data_frame_value_counts_default)   s    
 r'   c                  C   sp   t jddddgddddgddddd	gd
} | jdd}t jdddgt jjddgddgdd}t|| d S )Nr   r   r   r   r   r   r	   r
   r   r   T	normalizeg      ?g      ?r%   r&   r   r   r   r   r   r   r!   r!   r"   &test_data_frame_value_counts_normalize:   s    
 r*   c                  C   sX   t dddddgi} |  }t jdddgt jjdddggdgdd}t|| d S )Nr   r   r   r   r   r   r   r   r   r!   r!   r"   /test_data_frame_value_counts_single_col_defaultK   s    r+   c                  C   s0   t  } |  }t jg tjd}t|| d S )NZdtype)r   r   r   r   npZint64r   r   Z
df_no_colsr   r    r!   r!   r"   "test_data_frame_value_counts_emptyW   s    r/   c                  C   s4   t  } | jdd}t jg tjd}t|| d S )NTr(   r,   )r   r   r   r   r-   Zfloat64r   r   r.   r!   r!   r"   ,test_data_frame_value_counts_empty_normalize`   s    r0   c                 C   s^   t ddddgd| | dgd}| }t jddgt jjdd	gd
dgdd}t|| d S )NJohnAnneBethSmithLouise
first_namemiddle_namer   )r3   r1   )r5   r4   r7   r8   r   r   r   Znulls_fixturer   r   r    r!   r!   r"   (test_data_frame_value_counts_dropna_truei   s    

 r:   c              	   C   s   t ddddgd| | dgd}|jdd}t jd	d	d	d	gt jt dddgt dd| ggd
d	ddgdd
d	dggddgdd}t|| d S )Nr1   r2   r3   r4   r5   r6   F)Zdropnar   r   r   r7   r8   )Zlevelscodesr   r   )r   r   r   r   r   Indexr   r   r9   r!   r!   r"   )test_data_frame_value_counts_dropna_false|   s"    


r=   )numpyr-   Zpandasr   Zpandas._testingZ_testingr   r#   r$   r'   r*   r+   r/   r0   r:   r=   r!   r!   r!   r"   <module>   s   		