U
    MZf                     @   s   d dl Z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ejdddgd dgfdd ZdS )    Nc                  C   sr   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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countdatar   namepd	DataFramevalue_countsSeries
MultiIndexZfrom_arraystmassert_series_equaldfresultexpected r#   P/tmp/pip-unpacked-wheel-nbcvw55c/pandas/tests/frame/methods/test_value_counts.py%test_data_frame_value_counts_unsorted   s    
 r%   c                  C   sr   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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#   r$   &test_data_frame_value_counts_ascending   s    
 r&   c                  C   sn   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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#   r$   $test_data_frame_value_counts_default,   s    
 r)   c                  C   sr   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d}t|| d S )Nr   r   r   r   r   r   r	   r
   r   r   T	normalizeg      ?g      ?r'   r(   r   r   r   
proportionr   r   r   r#   r#   r$   &test_data_frame_value_counts_normalize>   s    
 r-   c                  C   sZ   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d}t|| d S )	Nr   r   r   r   r   r   r   r   r   r   r#   r#   r$   /test_data_frame_value_counts_single_col_defaultP   s    r.   c                  C   s@   t  } |  }t jg tjdtjg tjdd}t	|| d S )Nr   dtyper0   r   r   )
r   r   r   r   npZint64arrayintpr   r   Z
df_no_colsr!   r"   r#   r#   r$   "test_data_frame_value_counts_empty]   s       r6   c                  C   sD   t  } | jdd}t jg tjdtjg tjdd}t	|| d S )NTr*   r,   r/   r1   )
r   r   r   r   r2   Zfloat64r3   r4   r   r   r5   r#   r#   r$   ,test_data_frame_value_counts_empty_normalizeh   s       r7   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d}t|| d S )NJohnAnneBethSmithLouise
first_namemiddle_namer   )r:   r8   )r<   r;   r>   r?   r   r   r   r   nulls_fixturer    r!   r"   r#   r#   r$   (test_data_frame_value_counts_dropna_trues   s    

 rB   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d}t|| d S )Nr8   r9   r:   r;   r<   r=   F)Zdropnar   r   r   r>   r?   )levelscodesr   r   r   )r   r   r   r   r   Indexr   r   r@   r#   r#   r$   )test_data_frame_value_counts_dropna_false   s$    


rF   columnsr>   r?   r   c              	   C   sr   t |d ddddg|d d| | dgi}||d }t jdddgt jdddg|d d	d
d}t|| d S )Nr   r8   r9   r:   r   r;   r<   r   )r   r   r   )r   r   r   r   rE   r   r   )rA   rG   r    r!   r"   r#   r#   r$   #test_data_frame_value_counts_subset   s     
 
rH   )Znumpyr2   ZpytestZpandasr   Zpandas._testingZ_testingr   r%   r&   r)   r-   r.   r6   r7   rB   rF   markZparametrizerH   r#   r#   r#   r$   <module>   s   