Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_groupby_dropna.cpython-37.pyc
Назад
B �5�g�- � @ sN d dl Zd dlZd dlZd dlmZ ej� ddddgddggddgddgdd gd �fd ddgdej gddggdd dgdd dgddd gd �fg�dd� �Z ej� ddddgddggddgddgdd gd �fd ddgdej gddgej dggdddd gdddd gddd d gd �fg�dd� �Z ej� ddddgddgddgd dgd�fd ddej gddd gddd gd ddgd�fg�dd� �Z ej� dddddej gejd d gddgd �fd dddej gejd d d gddej gd �fg�d d � �Zej� d dejd!d"gddgd#d$�fd ejd!d"d%gddej gd#d$�fg�d&d'� �Zej� d(d)�d*d+� �Zej� d,de�d d-�dd.d.d/gie�d d �fded0�dd.d.d/gied1�fdejjd2d3d4d5gd6d7gd8�dd.d.d/giejjd2d3d4gd6d7gd8�fd e�d d-�dd.d.d/d/gie�d d-�fd ed0�dd.d.d/d/gied0�fd ejjd2d3d4d5gd6d7gd8�dd.d.d/d/giejjd2d3d4d5gd6d7gd8�fg�d9d:� �Zej� ddddgddggddgddgd d gd �fd ddgdej gddggdd dgdd dgd dd gd �fg�d;d<� �Zejjej� d=e�d>�e�d?�fe�d@�e�dA�fe�d>�e�d?�fg�ej� dBddCd gfd dCd dDgfg�dEdF� ���Z ej� dGej d dddej gdHdHdIdJgdK�dLd d/d d giddej gdMdN�ej ddddej gdHdHdIdJgdK�dLd d/d giddOdN�ej d ddddPgdHdHdIdJgdK�dLd d/d d giddQdN�ej dddddPgdHdHdIdJgdK�dLd d/d d giddRdN�g�dSdT� �Z dUdV� Z dS )W� Nzdropna, tuples, outputsT�A�Bg *@g ��Q��^@g �^@g �?)�c�d�eFg������(@g m@g (@c C s� dddddgd|dddgdddddgddddd gg}t j|d d d d dgd�}|jd d g| d��� }t jj|td�d�}| s�|jddtj gd d�}t j||d�}t � ||� d S )Nr r � g������(@g m@g ��Q��^@�{ � g �?�a�br r r )�columns)�dropna�ab)�names)�level)�index) �pd� DataFrame�groupby�sum� MultiIndex� from_tuples�list� set_levels�np�nan�tm�assert_frame_equal) r �tuples�outputs� nulls_fixture�df_list�df�grouped�mi�expected� r& �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_groupby_dropna.py�:test_groupby_dropna_multi_index_dataframe_nan_in_one_group s r( g������*@g @m@c C s� dddddgd|dddgdddddg|dddd gd|ddd gg}t j|d d d d dgd�}|jd d g| d��� }t jj|td�d�}| s�|�ddtj gddtj gg�}t j||d�} t � || � d S )Nr r r g������(@g m@g ��Q��^@r r g �?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 Znulls_fixture2r! r" r# r$ r% r& r& r' �;test_groupby_dropna_multi_index_dataframe_nan_in_two_groups3 s r) zdropna, idx, outputs)r r r c C s~ ddddgd dddgddddgdddd gg}t j|d d d d gd�}|jd | d��� }t j|t j|dd d�d�}t�||� d S )Nr r g������(@g m@r g ��Q��^@r r g �?r r r r )r )r �object)�dtype�name)r )r r r r �Indexr r )r �idxr r! r" r# r% r& r& r' �*test_groupby_dropna_normal_index_dataframe_ s r/ zdropna, idx, expectedr r � )r c C s8 t jddddg|d�}|jd| d��� }t�||� d S )Nr � r0 )r r )r r )r �Seriesr r r �assert_series_equal)r r. r% �ser�resultr& r& r' � test_groupby_dropna_series_level~ s r6 zdropna, expectedg @j@g �u@z Max Speed)r r, g 4@c C sL t jddddgddddgdd�}|jd d d tjg| d ��� }t�||� d S ) Ng `x@g �u@g >@g 4@ZFalconZParrotz Max Speed)r r, r r )r )r r2 r r r Zmeanr r3 )r r% r4 r5 r& r& r' � test_groupby_dropna_series_by� s r7 r )FTc C sB t �dddd gdddd gd��}|jd| d�}|jj| ks>t�d S )Nr r r1 r0 )r r r )r )r r r Zgrouperr �AssertionError)r r" �gbr& r& r' � test_grouper_dropna_propagation� s r: z/dropna,input_index,expected_data,expected_index� r1 r �abcd�abc)r �R)r r )r1 r> )r1 r �num�col)r c C s� t jdddd gdddd gd�|d�}|jd| d�}|�t�}t j||d�}t�||� |d g �t�}t j||d�}t�||� |d �t�}t j|d |d d �}t�||� d S ) Nr r r1 r0 )r r )r r )r r )r r, ) r r r Z transform�lenr r r2 r3 )r Z input_indexZ expected_dataZexpected_indexr" r9 r5 r% r& r& r' �+test_groupby_dataframe_slice_then_transform� s "$ rB c C s� dddddgdd dddgdddddgddddd gg}t j|d d d d dgd�}ttdd�}|jd d g| d��|�}t jj|td�d�}| s�|j ddt j gd d�}t j||d�}t � ||� d S )Nr r r g������(@g m@g ��Q��^@r r g �?r r r r r )r �min)r r r )r r )r )r )r )r r r �maxr �aggr r r r r r r r ) r r r r! r" Zagg_dictr# r$ r% r& r& r' �-test_groupby_dropna_multi_index_dataframe_agg� s rF zdatetime1, datetime2z 2020-01-01z 2020-02-01z-2 daysz-1 dayszdropna, valuesr � c C s� t �ddddddg||||||gd��}| r6||g}n ||tjg}|jd| d ��d ti�}t jd |it j|dd �d �} t� || � d S ) Nr r1 r0 r; � rG )�values�dtrJ )r rI )r, )r ) r r r r r rE r r- r r ) r rI Z datetime1Z datetime2Zunique_nulls_fixtureZunique_nulls_fixture2r" Zindexesr# r% r& r& r' �&test_groupby_dropna_datetime_like_data s rK z#dropna, data, selected_data, levels� � � )�groupsrI rI Zdropna_false_has_nan)�idZdropna_true_has_nanr Zdropna_false_no_nanZdropna_true_no_nanc C s� t �|�}|jd| d�}|�dd� �}tt|d |d ��}t jj|dd gd�}| sf|rf|j|dd�}t j||d�} t � || � d S ) NrO )r c S s t �dtt| ��i�S )NrI )r r �rangerA )�grpr& r& r' �<lambda>X � z@test_groupby_apply_with_dropna_for_multi_index.<locals>.<lambda>rI )r )r )r ) r r r �apply�tuple�zipr r r r r ) r �dataZ selected_dataZlevelsr" r9 r5 Z mi_tuplesr$ r% r& r& r' �.test_groupby_apply_with_dropna_for_multi_index0 s & rY c C s� dt jddt jgdddddgd�} t�| �}|jd d d �}|j}t j}dt jddg|d �dt jdg|d �t jt jddg|d �i}x*t|� � |� � �D ]\}}t � ||� q�W t � t |�� �d �s�t�t |�� �dd� ddgks�t�d S ) N�g1�g2r r r1 r0 r; )�groupr r\ F)r )r+ )r r r r r �indicesZintp�arrayrW rI r Zassert_numpy_array_equal�isnanr �keysr8 )rX r" r# r5 r+ r% Z result_valuesZexpected_valuesr& r& r' �test_groupby_nan_includede s " ra ) �numpyr ZpytestZpandasr Zpandas._testingZ_testingr �markZ parametrizer r( r) r/ r2 r6 r7 r: Z RangeIndexr r r rB rF Zarm_slow� TimestampZ TimedeltaZPeriodrK �paramrY ra r& r& r& r' �<module> s� $ * ,$ & $ 0
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка