o
    ŀg                     @   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d Zdd ZdS )    Nc                  C   d   t jg dg ddg dd} | jdd}t jg dt jjd	d
gddgddd}t|| d S )N      r      r   r   r   r   num_legs	num_wingsfalcondogcatantindexF)sort)   r   r   )r   r   r   r   r   r   r	   r
   namescountdatar   namepd	DataFramevalue_countsSeries
MultiIndexfrom_arraystmassert_series_equaldfresultexpected r(   `/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/frame/methods/test_value_counts.py%test_data_frame_value_counts_unsorted      r*   c                  C   r   )Nr   r   r   r   r   T)	ascending)r   r   r   )r   r   r   r   r	   r
   r   r   r   r   r$   r(   r(   r)   &test_data_frame_value_counts_ascending   r+   r-   c                  C   s`   t jg dg ddg dd} |  }t jg d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)   $test_data_frame_value_counts_default,   s   r1   c                  C   r   )Nr   r   r   r   r   T	normalize)g      ?      ?r4   r/   r0   r	   r
   r   
proportionr   r   r$   r(   r(   r)   &test_data_frame_value_counts_normalize>   r+   r6   c                  C   sR   t dg di} |  }t jg dt jjg dgdgddd}t|| d S )Nr	   r   r.   r/   r   r   r   r   r$   r(   r(   r)   /test_data_frame_value_counts_single_col_defaultP   s   r7   c                  C   s@   t  } |  }t jg tjdtjg tjdd}t	|| d S )Nr   dtyper9   r   r   )
r   r   r   r   npint64arrayintpr"   r#   
df_no_colsr&   r'   r(   r(   r)   "test_data_frame_value_counts_empty]   s   rA   c                  C   sD   t  } | jdd}t jg tjdtjg tjdd}t	|| d S )NTr2   r5   r8   r:   )
r   r   r   r   r;   float64r=   r>   r"   r#   r?   r(   r(   r)   ,test_data_frame_value_counts_empty_normalizeh   s   rC   c                 C   s\   t g d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AnnerE   BethSmithLouise
first_namemiddle_namer   )rG   rE   )rI   rH   rK   rL   r   r   r   r   nulls_fixturer%   r&   r'   r(   r(   r)   (test_data_frame_value_counts_dropna_trues   s   
rO   c              	   C   s   t g dd| | dgd}|jdd}t jg dt jt g dt ddtjggg d	g d
gddgddd}t	|| d S )NrD   rH   rI   rJ   F)dropna)r   r   r   r   )rF   rG   rE   )r   r   r   r   )r   r   r   r   rK   rL   )levelscodesr   r   r   )
r   r   r   r   r    Indexr;   nanr"   r#   rM   r(   r(   r)   )test_data_frame_value_counts_dropna_false   s$   
rU   columnsrK   rL   r   c              	   C   sj   t |d g d|d d| | dgi}||d }t jg dt jg d|d dd	d
}t|| d S )Nr   rD   r   rH   rI   r.   )rE   rF   rG   )r   r   r   )r   r   r   r   rS   r"   r#   )rN   rV   r%   r&   r'   r(   r(   r)   #test_data_frame_value_counts_subset   s   rW   c               	   C   sX   t jdg didd} |  }t jdt jt jg ddddgdd}t|| d S )	Na)r   r      categoryr8   r   )r   r9   r   r   r   )	r   r   r   r   r    r!   rS   r"   r#   r$   r(   r(   r)   ,test_value_counts_categorical_future_warning   s   r\   c               	   C   sj   t dt jg dg ddi} |  }t jg dt jt jg dg dddgdd	}t	|| d S )
NrX   )r   r   r   )r   r   rY   r   )
categories)r   r   r   r   )r   r   r   rY   )r]   r   r   r[   )
r   r   Categoricalr   r   r    r!   CategoricalIndexr"   r#   r$   r(   r(   r)   'test_value_counts_with_missing_category   s    r`   )numpyr;   pytestpandasr   pandas._testing_testingr"   r*   r-   r1   r6   r7   rA   rC   rO   rU   markparametrizerW   r\   r`   r(   r(   r(   r)   <module>   s"    
