a
    Pf                     @   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   sd   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Z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&   e/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/methods/test_value_counts.py%test_data_frame_value_counts_unsorted   s    r(   c                  C   sd   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 )Nr   r   r   r
   r   T)Z	ascending)r   r   r   )r   r   r   r   r   r	   r   r   r   r   r"   r&   r&   r'   &test_data_frame_value_counts_ascending   s    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    r-   c                  C   sd   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 )Nr   r   r   r
   r   T	normalize)g      ?      ?r0   r+   r,   r   r	   r   
proportionr   r   r"   r&   r&   r'   &test_data_frame_value_counts_normalize>   s    r2   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    r3   c                  C   s@   t  } |  }t jg tjdtjg tjdd}t	|| d S )Nr   dtyper5   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    r;   c                  C   sD   t  } | jdd}t jg tjdtjg tjdd}t	|| d S )NTr.   r1   r4   r6   )
r   r   r   r   r7   Zfloat64r8   r9   r    r!   r:   r&   r&   r'   ,test_data_frame_value_counts_empty_normalizeh   s    r<   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Anner>   BethSmithLouise
first_namemiddle_namer   )r@   r>   )rB   rA   rD   rE   r   r   r   r   nulls_fixturer#   r$   r%   r&   r&   r'   (test_data_frame_value_counts_dropna_trues   s    
rH   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 )Nr=   rA   rB   rC   F)Zdropna)r   r   r   r   )r?   r@   r>   )r   r   r   r   )r   r   r   r   rD   rE   )Zlevelscodesr   r   r   )
r   r   r   r   r   Indexr7   nanr    r!   rF   r&   r&   r'   )test_data_frame_value_counts_dropna_false   s$    
rL   columnsrD   rE   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   r=   r   rA   rB   r*   )r>   r?   r@   )r   r   r   )r   r   r   r   rJ   r    r!   )rG   rM   r#   r$   r%   r&   r&   r'   #test_data_frame_value_counts_subset   s    rN   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      categoryr4   r   )r   r5   r   r   r   )	r   r   r   r   r   r   rJ   r    r!   r"   r&   r&   r'   ,test_value_counts_categorical_future_warning   s    rS   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 )
NrO   )r   r   r   )r   r   rP   r   )
categories)r   r   r   r   )r   r   r   rP   )rT   r   r   rR   )
r   r   ZCategoricalr   r   r   r   ZCategoricalIndexr    r!   r"   r&   r&   r'   'test_value_counts_with_missing_category   s     rU   )numpyr7   ZpytestZpandasr   Zpandas._testingZ_testingr    r(   r)   r-   r2   r3   r;   r<   rH   rL   markZparametrizerN   rS   rU   r&   r&   r&   r'   <module>   s    
