Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/__pycache__/test_crosstab.cpython-37.pyc
Назад
B �5�g�t � @ s� d dl Zd dlZd dlmZ d dlmZmZmZm Z m Z m Z d dl m Z G dd� d�Zej�dddg�ej�d ddg�d d � ��ZdS ) � N)�is_categorical_dtype)�CategoricalIndex� DataFrame�Index� MultiIndex�Series�crosstabc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zej� d/d0d1gd2d3gg�d4d5� �Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@S )A� TestCrosstabc C s� t dddddddddddg dddddddddddg dddddddddddg tj�d�tj�d�tj�d�d��}|j|d d �| _d S ) N�foo�bar�one�two�dull�shiny� )�A�B�C�D�E�FT)Z ignore_index)r �np�random�randn�append�df)�self�methodr � r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/test_crosstab.py� setup_method sL zTestCrosstab.setup_methodc C sL | j }t|d |d �}|�ddg��� �� }t�||�d��t j �� d S )Nr r r ) r r �groupby�size�unstack�tm�assert_frame_equal�fillna�astyper �int64)r r �result�expectedr r r �test_crosstab_singleD s z!TestCrosstab.test_crosstab_singlec C s� | j }t|d |d |d g�}|�dddg��� }|�d��d��d��tj�}t � ||� t|d |d g|d �}|�dddg��� }|�d��d��tj�}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 �test_crosstab_multipleJ s z#TestCrosstab.test_crosstab_multiplec C s� t jjdddd�}t jjdddd�}t jjdddd�}t|||d��}t|||gdgd d �}t|d |d |d g�}t�||� t||g|dgd d �}t|d |d g|d �}t�||� t| jd j| jd j�}|j j dks�t �|j j dks�t �d S )Nr � �d )r" � � )�a�b�cr1 )r2 r3 )�rownames�colnamesr2 r3 )r5 r4 r r �row_0�col_0) r r �randintr r r$ r% r �values�index�name�AssertionError�columns)r r1 r2 r3 r r) r* r r r �test_crosstab_ndarrayW s z"TestCrosstab.test_crosstab_ndarrayc C s� t dddgdddgd�}t dddddgdddd d gd�}t�dddg�}tddgddggtddgd d �tddgd d �d�}t||�}t�||� t||�}t�||� d S )Nr � r1 r2 r3 )r: r/ � �d�fr6 )r; r7 )r: r= )r r �arrayr r r r$ r% )r r1 r2 r3 r* r) r r r �test_crosstab_non_alignedk s z&TestCrosstab.test_crosstab_non_alignedc C sH t jjdddd�}t jjdddd�}t jjdddd�}t|||d��}t|||gdgd d d �}|jjd ksnt�|jjd dgks�t�|d }|� dg�� � � d�}t t |�gtdgdd�d�}|�|�}d|_t�||� |jd } |� d dg�� � � d�} | �t t |�gdgd��} d| _| �| j�} | �d�� t j�} t�| | � d S )Nr � r. )r" r/ r- )r1 r2 r3 r1 )r2 r3 T)r4 r5 �margins)r1 r2 r3 )�All� �i8rG )r; )r: )r r r8 r r r: �namesr< r= r! r" r' r �lenr r r; r$ �assert_series_equal�loc�reindexr& r( ) r r1 r2 r3 r r) �all_cols�exp_cols� exp_margin�all_rows�exp_rowsr r r �test_crosstab_margins} s( z"TestCrosstab.test_crosstab_marginsc C s� t jjdddd�}t jjdddd�}t jjdddd�}t|||d��}t|||gdgd d d d �}|jjd kspt�|jjddgks�t�|d }|� dg�� � � d�}t t |�gtd gdd�d�}|�|�}d|_t�||� |jd } |� ddg�� � � d�} | �t t |�gdgd��} d | _| �| j�} | �d�� t j�} t�| | � d} xLdd ddggD ]:} tjt| d�� t|||gdgd d | d � W d Q R X �qZW d S )Nr rE r. )r" r/ r- )r1 r2 r3 r1 )r2 r3 T�TOTAL)r4 r5 rF � margins_name)r1 r2 r3 )rU rH rI )r; )r: z&margins_name argument must be a stringi� )�match)r r r8 r r r: rJ r< r= r! r" r' r rK r r r; r$ rL rM rN r& r( �pytest�raises� ValueError) r r1 r2 r3 r r) rO rP rQ rR rS �msgrV r r r �%test_crosstab_margins_set_margin_name� sH z2TestCrosstab.test_crosstab_margins_set_margin_namec C s� t jjdddd�}t jjdddd�}t jjdddd�}t j�d�}t||g||t jddgd gd �}t||||d ��}|jd ddgd t jd �}t� ||� d S )Nr rE r. )r" r/ r- r r �baz)�aggfuncr4 r5 )r r r] r9 r9 )r: r= r^ ) r r r8 r r �sumr Z pivot_tabler$ r% )r r1 r2 r3 r9 �tabler r* r r r �test_crosstab_pass_values� s z&TestCrosstab.test_crosstab_pass_valuesc C s� t jdddddddgtd�}t jdddddddgtd�}t jdddddddgtd�}t|||gdgd d gd d �}tjd dddgd d gd�}t�|j|� d S )Nr r )�dtyper r r r r1 r2 r3 F)r4 r5 �dropna)r r )r r )r r )r r )rJ ) r rC �objectr r � from_tuplesr$ Zassert_index_equalr= )r r1 r2 r3 �res�mr r r �test_crosstab_dropna� s z!TestCrosstab.test_crosstab_dropnac C sj t dddgdddgd�}t dddgdddgd�}t||�}ttg dd d �tg dd d �d �}t�||� d S ) Nr? � r/ )r: r@ r- � r( r6 )rb r; r7 )r: r= )r r r r r$ r% )r �s1�s2�actualr* r r r �test_crosstab_no_overlap� s z%TestCrosstab.test_crosstab_no_overlapc C s� t dddddtjgddddddgd��}t|j|jddd�}t dddgdddgddd gg�}td d d gd d�|_tddd gdd�|_t � ||� d S )Nr? ri r/ r@ )r1 r2 T)rF rc r r- g �?g @rG r1 )r; r2 ) r r �nanr r1 r2 r r: r= r$ r% )r r rm r* r r r �test_margin_dropna� s ( z TestCrosstab.test_margin_dropnac C s� t dtjtjtjdtjgdtjddddgd��}t|j|jddd�}t dddgdddgdddgg�}td d d gd d �|_tddd gdd �|_t � ||� d S )Nr? ri r/ r@ )r1 r2 T)rF rc r g �?g @rG r1 )r; g @g @r2 ) r r ro r r1 r2 r r: r= r$ r% )r r rm r* r r r �test_margin_dropna2 s . z TestCrosstab.test_margin_dropna2c C s� t dtjtjtjtjdgddddddgd��}t|j|jddd�}t dddgdddgdddgg�}td d d gd d �|_tddd gdd �|_t � ||� d S )Nr? ri r/ r@ )r1 r2 T)rF rc r g �?g @rG r1 )r; r2 ) r r ro r r1 r2 r r: r= r$ r% )r r rm r* r r r �test_margin_dropna3 s , z TestCrosstab.test_margin_dropna3c C s� t dddddtjgddddddgd��}t|j|jddd�}t dd dgdddgddd gg�}td d d gdd�|_tddd gdd�|_t � ||� d S )Nr? ri r/ r@ )r1 r2 TF)rF rc r rj g �?g @rG r1 )r; r2 ) r r ro r r1 r2 r r: r= r$ r% )r r rm r* r r r �test_margin_dropna4 s ( z TestCrosstab.test_margin_dropna4c C s� t dtjtjtjdtjgdtjddddgd��}t|j|jddd�}t dd dgd ddgddd gg�}td d d gdd�|_tddd gdd�|_t � ||� d S )Nr? ri r/ r@ )r1 r2 TF)rF rc r rj g �?g @rG r1 )r; g @g @r2 ) r r ro r r1 r2 r r: r= r$ r% )r r rm r* r r r �test_margin_dropna5' s . z TestCrosstab.test_margin_dropna5c C s* t jdddddddgtd�}t jdddddt jdgtd�}t jdddddddgtd�}t|||gdgd d gd d d �}tjdddddgdddddggd d gd�}tdddddgdddddgdddddgg|d�}tdddgdd�|_ t � ||� t||g|dd gd gd d d �}tjdddddgdddddggdd gd�}tdddgdddgdddgdddgdddgg|d�}tdddgd d�|_ t � ||� t||g|dd gd gd d d �}tjdddddgdddddggdd gd�}tdddgdddgdddgdddgdddgg|d�}tdddgd d�|_ t � ||� d S )Nr r )rb r r r r r1 r2 r3 TF)r4 r5 rF rc rG rH )rJ r? r ri r- r/ rE )r= )r; )r: rj ) r rC rd ro r r � from_arraysr r r: r$ r% r= )r r1 r2 r3 rm rg r* r r r �test_margin_dropna61 s@ . 2 2z TestCrosstab.test_margin_dropna6c C sn t dddddgdddddgddtjddgd��}tddgdd�}tddgdd�}t d d gd d gg||d �}t d d gddgg||d �}t dd gdd gg||d �}t�t|j|jdd�|� t�t|j|jdd�|� t�t|j|jdd�|� t�t|j|jdd�|� t�t|j|jdd�t|j|jdd�� t�t|j|jd d�t|j|jdd�� t d d gddgdd ggtdddgddd�tddgddd�d �}t dd d gdd dggtddgddd�tdddgddd�d �}t d d d gd d dgdd dggtdddgddd�tdddgddd�d �} t�t|j|jddd�|� t�t|j|jddd�|� t�t|j|jddd�| � d S ) Nr? ri r/ r@ )r1 r2 r3 r1 )r; r2 g�������?r g333333�?)r: r= g �?g �?g �?g �?�all)� normalizeTr: r= g�������?rG rd )r; rb g�������?)rx rF ) r r ro r r$ r% r r1 r2 ) r r �rindexZcindexZ full_normalZ row_normalZ col_normalZrow_normal_marginsZcol_normal_marginsZall_normal_marginsr r r �test_crosstab_normalizeb sD .$z$TestCrosstab.test_crosstab_normalizec C st t dddddgdddddgddtjddgd��}tt�ddddg�t�ddddg�gt�ddddg�� t dddgddd gdddggtddd gd d d �tddd gdd�d�}t|j|j|jdddd�}t � ||� t dddddgdddddgddtjddgd��}t dddgdddgdddggtddd gd d d �tddd gdd d �d�}t|j|j|jtj ddd�}t � ||� d S )Nr? ri r/ r@ )r1 r2 r3 g �?r g �?g �?rG r1 rd )r; rb r2 )r; )r: r= �countrw T)r^ rx rF g g�������?g333333�?) r r ro r rC r r1 r2 r3 r$ r% r_ )r r Z norm_countsZ test_caseZnorm_sumr r r � test_crosstab_normalize_arrays� s( .6 .z+TestCrosstab.test_crosstab_normalize_arraysc C s t dddddgdddddgtjtjtjtjtjgd��}t ddgddggtddgddd �tddgd d �d �}x0d D ](}t|j|j|jd|d�}t� ||� qrW t dtjgddggtddgddd �tddgd d �d �}|r�|d � d�|d< t|j|j|jddd�}t� ||� d S )Nr? ri r/ r@ )r1 r2 r3 g r1 r( )r; rb r2 )r; )r: r= )Tr: r= r{ )r9 r^ rx F) r r ro r r r1 r2 r3 r$ r% r' )r Zusing_array_managerr �empty�iZ calculatedZnansr r r �test_crosstab_with_empties� s& z'TestCrosstab.test_crosstab_with_emptiesc C s& t dddddgdddddgddtjddgd��}d}tjt|d�� t|j|j|j d� W d Q R X d }tjt|d�� t|j|jtj d � W d Q R X d }tjt|d�� t|j|jd d � W d Q R X tjt|d�� t|j|jdd � W d Q R X d}tjt|d�� t|j|jddd� W d Q R X d S )Nr? ri r/ r@ )r1 r2 r3 z)values cannot be used without an aggfunc.)rW )r9 z%aggfunc cannot be used without values)r^ z Not a valid normalize argumentZ42)rx �* z Not a valid margins argumentrw )rx rF ) r r ro rX rY rZ r r1 r2 r3 Zmean)r r �errorr r r �test_crosstab_errors� s . z!TestCrosstab.test_crosstab_errorsc C s� t ddddddgddddddgd��}dddg}|d �d �j�|�|d< t|d |d �}tdddgd d �}t||d dd �}dddgdddgdddgg}t |||d�}t�||� d S )NZHondaZAcuraZTeslaZSedanZElectricZPickup)�MAKE�MODELr� �categoryr� )r; F)� categoriesZorderedr; ri r r? )r: r= ) r r' �catZset_categoriesr r r r$ r% )r r r� r) �expected_indexZexpected_columns� expected_datar* r r r �%test_crosstab_with_categorial_columns� s z2TestCrosstab.test_crosstab_with_categorial_columnsc C sp t ddddgd dddgd d d d d d d gd tj�d �tj�d �d ��}t|d |d g|d gdtj|d d�}tddddgddddggddddddddddg ddddddddddg gddgd�}td d dgddd�}t�dddgdddgdddgdtj dgtj ddgdtj dgtj ddgdtj dgtj ddgd d d gg �}t |||d �}|d � d �|d< t � ||� d S ) Nr r �threerj r r r � r r r@ � )r r r r r Tr )r: r= rF r^ r9 rG rH r? ri r/ r )�levels�codesrJ rd )rb r; g @g @g (@g 8@)r: r= r( ) r r r r r r" r r rC ro r' r$ r% )r r r) r� Zexpected_columnr� r* r r r � test_crosstab_with_numpy_size s@ . z*TestCrosstab.test_crosstab_with_numpy_sizec C s� t td�dd�}t tdd�dd�}t tdd�dd�}t td�dd�}ddi}t||�}t||�j|dd�}t�||� t||g|�}t||g|�j|d d �}t�||� t|||g�}t|||g�j|dd�}t�||� d S ) Nr/ r )r; r? r@ r Zwaldo)r= �axisr )r: r� )r �ranger Z rename_axisr$ r% )r rk Zs2_fooZs2_barZs3Zmapperr) r* r r r � test_crosstab_duplicate_names6 s z*TestCrosstab.test_crosstab_duplicate_namesrJ r1 )r2 r3 )r1 r2 r3 c C sv t td�|d d�}t tdd�|d d�}tjtd�tdd�g|d�}t d|d�jddd�}t||�}t�||� d S ) Nr/ r )r; r? r@ )rJ )r: )Z fill_value)r r� r ru r# r r$ r% )r rJ rk rl �mir* r) r r r �test_crosstab_tuple_nameR s z%TestCrosstab.test_crosstab_tuple_namec C sj t td�dd�}t td�dd�}ttjddd�ttd�dd�ttd�dd�d�}t||�}t�||� d S )Nr/ )r1 r2 )r; )r3 rA r( )rb )r: r= ) r r� r r �eyer r r$ r% )r rk rl r* r) r r r � test_crosstab_both_tuple_names] s z+TestCrosstab.test_crosstab_both_tuple_namesc C s� t dddgdddgd�dd d gd �}t|j|j|jg�}td d dgd d �}tjdddgddgd�}t dddgdddgdddgg||d�}t� ||� d S )Nr/ r? ri r- r@ rj )r2 r1 r r r )r: r6 )r; )r? r@ )ri rj )r/ r- r2 r1 )rJ r )r: r= ) r r r: r2 r1 r r re r$ r% )r r r) Ze_idxZ e_columnsr* r r r � test_crosstab_unsorted_orderj s $$z)TestCrosstab.test_crosstab_unsorted_orderc C s� t ddddgd dddgd d d d d d d gd d gd d gd d��}t|j|jg|j|jtjddd�}t tjd gd dg t d�� dd�t d d dgddd�t j dddd d d d d d!d"g ddgd#�d$�}t�||� d S )%Nr r r� rj r r r r� r r r@ r r� )r r r r r T)r9 r^ rx rF � r? )rb r0 r/ rG rd )rb r; )r r )r r )r r )r� r )r� r )r� r )r r )r r )r r )rG rH )rJ )r= r: )r r r r r r r r_ rC �floatZreshaper r re r$ r% )r r r) r* r r r �(test_crosstab_normalize_multiple_columnst s: z5TestCrosstab.test_crosstab_normalize_multiple_columnsc C sT t dddddddddg dddddddddg dddddddddg dddd d d d d d g dd d d d d dddg d��}t|j|jg|jdddd�}t ddgddgddgddgddgg�}tdddgdddggdddddgdddddggdd gd �|_tddgd d d �|_t � ||� t|j|jg|jdddd�}t d!d"d#gd!d"d#gdd"dgdd$d#gg�}tdddgd d d �|_tddgddggddddgddddggdd gd �|_t � ||� t|j|jg|jdddd�}t d%d%d#gd%d%d#gd#d%dgd&d#d#gdd'dgg�}tdddgd d d �|_tdddgdddggdddddgdddddggdd gd �|_t � ||� d S )(Nr r r r �small�larger? ri r/ r@ r- rj rE r� � )r r r r r Tz Sub-Totalr )rF rV rx g �?g]QJVU�?gF]k�SU�?g��9?�q�?g'c` ��?rH r r )r� r� rJ rd r )rb r; g �?g�������?g��9?�q�?g�������?g��9?�q�?g g#�G��?) r r r r r r r: r r= r$ r% )r r r) r* r r r �test_margin_normalize� sj $ z"TestCrosstab.test_margin_normalizec C s t dddddddddg dddddddddg dddddddddg dddd d d d d d g dd d d d d dddg d��}t|j|j|jgdddd�}t dddddgdddddgdddddggdddgd�}tdddgd ddggd d dddgddddd ggd d gd �|_d!|j_t � ||� d S )"Nr r r r r� r� r? ri r/ r@ r- rj rE r� r� )r r r r r T�margin)r: r= rF rV rx g��9?�q�?g��9?�q�?g g��9?�q�?g'c` ��?gF]k�SU�?g �?)r: rH r r r )r� r� rJ r ) r r r r r r r= r: r; r$ r% )r r r) r* r r r �&test_margin_normalize_multiple_columns� s> z3TestCrosstab.test_margin_normalize_multiple_columnsN)#�__name__� __module__� __qualname__r r+ r, r> rD rT r\ ra rh rn rp rq rr rs rt rv rz r| r r� r� r� r� rX �mark� parametrizer� r� r� r� r� r� r r r r r s: 2 2 13$ - " (Qr �a_dtyper� r( �b_dtypec C s� t j�d�}t|jdddd���| �}t|jdddd���|�}t||ddd �}tdd d gd d d�}tdd dd gd dd�}dddgdddgdddgdddgg}t|||�} t � || � d|j |d k< t |j �} | r�|�� j d dks�t�t||ddd �}dddgddt jgdddgdddgg}t|||�} | �sR| j ddd g } | d �d�| d < t|� t| � t| j ddd g � t � || � d S )Ni�v�r r/ r. )r" ri TF)rF rc r? rG rd r7 )rb r; r6 � � �" � �4 �0 �B r( )r r Z RandomStater r8 r' r r r r$ r% rM r rb Z value_countsr< ro �repr) r� r� �gr1 r2 r) r= r: r9 r* Za_is_catr r r �test_categoricals s, $ & r� )�numpyr rX Zpandas.core.dtypes.commonr Zpandasr r r r r r Zpandas._testingZ_testingr$ r r� r� r� r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка