Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_grouping.cpython-37.pyc
Назад
B �5�gx� � @ s� d Z ddlZddlZddlZddlmZmZmZm Z m Z m Z m Z ddl mZ ddlmZ G dd� d�ZG dd� d�ZG d d � d �ZG d d � d �ZdS ) zG test where we are determining what we are grouping, or getting groups � N)�CategoricalIndex� DataFrame�Index� MultiIndex�Series� Timestamp� date_range)�Groupingc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� TestSelectionc C sn t ddggddgd�}|�d�}tjtdd�� |dg W d Q R X tjtd d�� |ddg W d Q R X d S ) N� � �A�B)�columnsz"Columns not found: 'C'")�match�Cz^[^A]+$)r �groupby�pytest�raises�KeyError)�self�df�g� r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_grouping.py�test_select_bad_cols s z"TestSelection.test_select_bad_colsc C s� t ddddgtd�tdd�tddd�gd �}d }tjt|d �� |�d� W d Q R X tjt|d �� |�ddg� W d Q R X |�d�}|�� }|jjd ks�t �|jj d ks�t �d S )Nr r r � r � r � )r �dataz!Grouper for 'A' not 1-dimensional)r r � ) r �ranger r � ValueErrorr �countr Znlevels�AssertionError�size)r r �msg�grouped�cr r r �'test_groupby_duplicated_column_errormsg# s * z5TestSelection.test_groupby_duplicated_column_errormsgc C sf |� d�j�� }|� d�d �� }t�||� d|d< |� d��� }|� d��tj�}t�||� d S )Nr r g �?�mean) r r �sum�tm�assert_series_equalr* �agg�np�assert_frame_equal)r r �result�expectedr r r �test_column_select_via_attr4 s z)TestSelection.test_column_select_via_attrc C s� t ddddddddgddddddddgtj�d�tj�d�tj�d�d��}|�d�d d g �� }|�d�|jd d � �� }|jd d �dd d gf �d��� }t� ||� t� ||� d S ) N�foo�bar�one�two�threer )r r r �D�Er r r9 r r ) r r/ �random�randnr r* r �locr, r0 )r r r1 �result2r2 r r r � test_getitem_list_of_columns> s " z*TestSelection.test_getitem_list_of_columnsc C s� t td�d tj�d�tj�d�tj�d�d��}|�d�|jdd� �� }|�d�ddg �� }|jd d �dddgf �d��� }t � ||� t � ||� t � t �� |�d�d �� W d Q R X d S ) N�abcdr r )r r r r r r r r )r r ) r �listr/ r; r< r r r* r= r, r0 �assert_produces_warning� FutureWarning)r r r1 r> r2 r r r �!test_getitem_numeric_column_namesQ s " z/TestSelection.test_getitem_numeric_column_namesc C s, t �t�� |�d�d �� W d Q R X d S )Nr )r r9 )r, rB rC r r* )r r r r r �#test_getitem_single_list_of_columnsg s z1TestSelection.test_getitem_single_list_of_columnsc C s� t ddddddddgddddddddgtj�d�tj�d�tj�d�d��}|�d�d �� }|jd d �dd gf �d��� }|jd d �d f }|}t� ||� d S ) Nr4 r5 r6 r7 r8 r )r r r r9 r: r r r ) r r/ r; r< r r* r= �ilocr, r- )r r r1 Zas_frameZ as_seriesr2 r r r �test_getitem_single_columnl s z(TestSelection.test_getitem_single_columnc sN t ddd� dD �i�� � �d�}� �� fdd��}|j}|j}t�||� d S )N�Tuplesc s s, | ]$}t j�d dd�D ]}||fV qqdS )r � N)r/ r; �randint)�.0�x�yr r r � <genexpr>� s zJTestSelection.test_indices_grouped_by_tuple_with_lambda.<locals>.<genexpr>)r r c s � j | df S )Nr )rF )rL )r r r �<lambda>� � zITestSelection.test_indices_grouped_by_tuple_with_lambda.<locals>.<lambda>)r r �indicesr, �assert_dict_equal)r �gb� gb_lambdar2 r1 r )r r �)test_indices_grouped_by_tuple_with_lambda s z7TestSelection.test_indices_grouped_by_tuple_with_lambdaN) �__name__� __module__� __qualname__r r) r3 r? rD rE rG rU r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z ej�dddddgd d d d gd d!� ed"�D �g�d#d$� �Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zej�d3d4d5g�d6d7� �Zd8d9� Zej�d3d4d5g�d:d;� �Z d<d=� Z ej�d>d4ddddddddddg gd5ddddddddddg gg�d?d@� �Z dAdB� Z dCdD� Z ej�dEdFe!de"j#e$g �dG�fdHe!de"j#e%j&g ddI�dG�fdJe!de"j#e%j&g ddI�dG�fg�dKdL� �Z'dMdN� Z(dOdP� Z)dQdR� Z*dSS )T� TestGroupingc C s� t t�d��dd�td�d�}xvtjtjtjtj tj tj gD ]V}|t |��|_ |�td���dd� � tt|j �� ��|_ |�td���d d� � q<W d S ) N� rI r ZAB)r Zabcdec S s | S )Nr )rL r r r rO � rP z7TestGrouping.test_grouper_index_types.<locals>.<lambda>c S s | S )Nr )rL r r r rO � rP )r r/ �arange�reshaperA r, ZmakeFloatIndexZmakeStringIndexZmakeUnicodeIndexZ makeIntIndexZ makeDateIndexZmakePeriodIndex�len�indexr �apply�reversed�tolist)r r r^ r r r �test_grouper_index_types� s z%TestGrouping.test_grouper_index_typesc C s ddl m}m} |�� |dd� }t||�� �}tj||gddgd�}ttj � ddd �|d �}|� � � t jdd d �t jdd d �g��� }tdgd d�|_|� t jdd d�t jdd d�g��� }t�||� |� t jdd d�t jdd d�g��� }t�||� d S )Nr )�date� timedelta� )�daysr4 r5 )�names�d �� )r^ �W)�key�freq�int64)�dtype)�levelrl r )�datetimerc rd �todayr r � from_productr r/ r; rJ � reset_indexr �pd�Grouperr+ r r r, r0 ) r rc rd Zd0�datesZ date_indexr r2 r1 r r r � test_grouper_multilevel_freq� s "" "z)TestGrouping.test_grouper_multilevel_freqc C s| t ddddddgddddddgd��}|�d �}|�� }|�tjd d ��}|�� }t�||� |�tjd dd ��}|�� }t�||� |�d d � �}dddg|d < |jd d �d dgf }t�||� t ddddddgddddddgddddddgd��}|�d ��� }|�tjd d �g�}|�� }t�||� |�d dg��� }|�tjd d �tjdd �g�}|�� }t�||� |�d tjdd �g�}|�� }t�||� |�tjd d �dg�}|�� }t�||� t t j ddd�t j td�td�tddd�gdddgd�d�}|�tjddd ���� }t d gtjtd �gddd �d�}t�||� |�tjdd ���� }|jdd ��� }t�||� d S )!Nr r r r r rI r )r r r )rk )rk �axisc S s | � � S )N)r+ )rL r r r rO � rP z8TestGrouping.test_grouper_creation_bug.<locals>.<lambda>r )r r r r rm )rn �ab�20130101)�periodsr6 r7 r8 )rg )r^ �M)ro rl � z 2013-01-31)rl �name)ro )r r r+ rt ru r, r0 r_ r= r r/ r[ r rr rA r! r � DatetimeIndexr r- )r r r r2 r1 �sr r r �test_grouper_creation_bug� sT & 2 z&TestGrouping.test_grouper_creation_bugc C sB t �ddddddg�}ddg|_tt�d �d d d d d d gd �|d �}|�dtjdd�g�� � }|� � �ddg�� � }t � ||� |�tjdd�dg�� � }|� � �ddg�� � }t � ||� |� d�}|�dtjdd�g�� � }|� � �ddg�� � }t � ||� |�tjdd�dg�� � }|� � �ddg�� � }t � ||� d S )N)�ar )r� r )r� r )�br )r� r )r� r �outer�innerr r6 r7 )r r )r^ r )ro ) r � from_tuplesrg r r/ r[ r rt ru r* rs r, r0 )r �idxZdf_multir1 r2 Z df_singler r r � test_grouper_column_and_index s& z*TestGrouping.test_grouper_column_and_indexc C sz ddg}t jddddg|d�}tt�d��d d �|d �}|j|d ��� }|�� �|��� }|j � tj �|_ t � ||� d S ) NrL rM )r r )r r )r r )rI r )rg � ���r )r^ )ro )r r� r r/ r[ r\ r r* rs r Zastyperm r, r0 )r Z idx_namesr� r Z by_levelsZ by_columnsr r r � test_groupby_levels_and_columns0 s z,TestGrouping.test_groupby_levels_and_columnsc C s ddddg}ddg}t �ddddgddddgddddgddddgddddggt�}t||dd�}t||d�}|jdd|d ��� }t �d dgd dgd dgd dgd dggt�}t||dd�} t|| d�} t�|| � t|j |d �}|jdd|d ��� }t|j | d �} t�|| � d S ) Nr r r r T)� categoriesZordered)r r r )rx ro �observedr )r^ )r r^ ) r/ �array�intr r r r+ r, r0 �T) r r� r r� r Z cat_columnsr r1 Z expected_dataZexpected_columnsr2 r r r �*test_groupby_categorical_index_and_columns> s : * z7TestGrouping.test_groupby_categorical_index_and_columnsc C s� t dditjtd�tddd�gddgd �d �}|�tjdd �tjdd d �g��� }t dddddddgitjtd�tdd dd�gddgd �d �}t � ||� d S )Nr r ry rz �P )r{ r6 r7 )rg )r^ )ro r| )ro rl � r} � r )rl r{ ) r r rr rA r r rt ru r+ r, r0 )r r r1 r2 r r r �#test_grouper_getting_correct_binnerU s " z0TestGrouping.test_grouper_getting_correct_binnerc C s t |�d�j�ddgks t�d S )Nr r5 r4 )�sortedr �grouperr$ )r r r r r �test_grouper_iterl s z TestGrouping.test_grouper_iterc C s( t jtdd�� |�g � W d Q R X d S )NzNo group keys passed!)r )r r r" r )r r r r r �test_empty_groupso s z TestGrouping.test_empty_groupsc C s2 |� d�}|� |j��� }|�� }t�||� d S )Nr )r r� r* r, r0 )r r r' r1 r2 r r r �test_groupby_groupert s z!TestGrouping.test_groupby_grouperc C s� ddl m} |ddi�}|�ddi��t�}|�dg��t�}t�||� |dddd gtd �d �}ddd d d �}|�|��� }|�|��t j�}|�ddd d g��� }|�ddd d g��� }t�||� t�||� t�||� d S )Nr )r ZT1rI ZT2g �?g @g @g @r@ )r^ r )r� r� r( �d) �pandasr r r. r+ r, r- rA r* r/ )r r r� r1 r2 �mappingr> Z expected2r r r �test_groupby_dict_mapping{ s z&TestGrouping.test_groupby_dict_mappingr^ r r r r r� r� r( r� c C s g | ]}t d dd| ��qS )i� � r} )r )rK �ir r r � <listcomp>� s zTestGrouping.<listcomp>r c C s\ |ddddg|d�}t ddddg|dd�}|�|��� }|ddg�}d|j_t�||� d S ) Nr r r r )r^ r )r� r� )r^ r~ )r r �lastr^ r~ r, Z assert_equal)r Zframe_or_seriesr^ �obj�groupsr1 r2 r r r �$test_groupby_series_named_with_tuple� s z1TestGrouping.test_groupby_series_named_with_tuplec C sR t dddd�}ttj�d�|d�}d}tjt|d�� |�dd � � W d Q R X d S ) Nz 01-Jan-2013r� ZMS)r{ rl )r^ z4Grouper result violates len\(labels\) == len\(data\))r c S s | dd� S )Nr r r )rk r r r rO � rP zDTestGrouping.test_groupby_grouper_f_sanity_checked.<locals>.<lambda>) r r r/ r; r< r r r$ r )r rv �tsr& r r r �%test_groupby_grouper_f_sanity_checked� s z2TestGrouping.test_groupby_grouper_f_sanity_checkedc C s6 d}t jt|d�� t|j|ddg � W d Q R X d S )NzEGrouper for '<class 'pandas.core.frame.DataFrame'>' not 1-dimensional)r r )r r r" r r^ )r r r& r r r �%test_grouping_error_on_multidim_input� s z2TestGrouping.test_grouping_error_on_multidim_inputc C sX t dddgdddgddd gg�}t�d d d g�|_|jdd dgd��� }t�||� d S )Nr r r r rI r r� r � )r r )r r )r r r )rx ro )r r r� r r �firstr, r0 )r r r1 r r r �test_multiindex_passthru� s z%TestGrouping.test_multiindex_passthruc C s� |j dd��� }|j dd��� }t�||� |j dd��� }|j dd��� }t�||� |j ddgd��� }|}t�||� |j ddgd��� }|j ddgd��� }t�||� d S )Nr� )ro �second���r� )r r+ r, r0 )r �mframer1 r2 r r r � test_multiindex_negative_level� s z+TestGrouping.test_multiindex_negative_levelc C s4 t �t|j��|_|jddd�d �dt ji� d S )Nr F)Zas_indexr �Q)r/ r[ r] r r r. r* )r r r r r �&test_multifunc_select_col_integer_cols� s z3TestGrouping.test_multifunc_select_col_integer_colsc C s� ddgddgg}t �|�}tddgg|d�}|�d�j}|d dgksJt�|�d g�j}|d dgksjt�tddgd d gg|d�}|�d�j}|�d g�j}||ks�t�tddgd dgg|d�}|�d�j}|�d g�j}t�||� d S ) Nr# �valuesz to filter� r r )r r )z to filterr� r r )r r� r r r� r$ r, rR )r �lstZmidxr r' r2 r1 r r r �#test_multiindex_columns_empty_level� s z0TestGrouping.test_multiindex_columns_empty_levelc C s� t ddddgddddgddddggt�dddd gddddgg�d �}|�d g�j}|�d �j}t�||� t |jt�dddd gd d d d gg�d �}|�dg�j}|�d �j}t�||� t |jdddd gd �}|�dg�j}|�d �j}t�||� d S )Nr r r r rI r r� r� r( )r )r� r r� �e)r� r� )r� r� )r� r� )r r � from_arraysr r� r, rR r� )r r r2 r1 Zdf2Zdf3r r r � test_groupby_multiindex_tuple� s" " z*TestGrouping.test_groupby_multiindex_tuple�sortTFc C s� |}|� � }|jd|d��� }|jd|d��� }|j|d j|d��� }|j|d j|d��� } d|j_d| j_|jjdks�t�|jjdks�t�t�||� t�|| � |jj|jj d ks�t�|jj|jj d ks�t�|jd|d��� }|jd|d��� }t�||� t�|| � |j jdd|d��� }|j jdd|d��� }t�||j � t�|| j � d} t j t | d �� |jdd � W d Q R X d S ) Nr )ro r� r r� )r� r� )ro rx r� z2level > 0 or level < -1 only valid with MultiIndex)r )ro )rs r r+ r� r^ r~ r$ r, r0 rg r� r r r" ) r r� r� r �frameZ deleveledZresult0�result1Z expected0Z expected1r& r r r �test_groupby_level s2 z TestGrouping.test_groupby_levelc C s� t dgd dgd td�d���d�}|dkr6|j}|jd|d� d |�|�� �}tjt|d �� |jd |d� W d Q R X d S ) Nr r r r )�expZvar1r� )r r )ro rx z&level name foo is not the name of the )r r4 ) r r! � set_indexr� r Z_get_axis_namer r r" )r rx r r& r r r � test_groupby_level_index_names2 s "z+TestGrouping.test_groupby_level_index_namesc C s t ddgddddggddddddddgddddddddggd�}tt�d�|d�}|jd|d��� }td d gddgd�}t�||� t ddgddddggddddd dddgddddddddggd�}tt�d�|d�}|jd|d��� }td d gd dgd�}t�||� d S )Nr r r r )�levels�codesg @)r^ )ro r� g @g 6@r� g 2@g g �?)r r r/ r[ r r+ r, r- )r r� r^ r� r1 r2 r r r �test_groupby_level_with_nas> s , ,z(TestGrouping.test_groupby_level_with_nasc C sZ |}d}t jt|d�� |�� W d Q R X d}t jt|d�� |jd d d� W d Q R X d S )Nz*You have to supply one of 'by' and 'level')r )Zbyro )r r � TypeErrorr )r r� r� r& r r r �test_groupby_argsW s z TestGrouping.test_groupby_argsz sort,labelsc C s4 |j d|d�}t�|tj�}t�|jjd |� d S )Nr )ro r� )r r/ r� �intpr, �assert_almost_equalr� r� )r r� �labelsr� r' � exp_labelsr r r �test_level_preserve_orderc s z&TestGrouping.test_level_preserve_orderc C sN |� |j�d��}tjddddddddddg tjd�}t�|jj d |� d S )Nr r r r )rn ) r r^ Zget_level_valuesr/ r� r� r, r� r� r� )r r� r' r� r r r �test_grouping_labelsp s $z!TestGrouping.test_grouping_labelsc C s� t dtdddd�i�}tj|jd< tjddd�}|�|g�}td �tt t d ���i}t � |j |� |�|�}td �di}t � |j |� d S ) Nrc z1/1/2011im r9 )r{ rl r� ZAS)rk rl z 2011-01-01il )r r rt ZNaTrF ru r r r rA r! r, rR r� )r r r� r1 r2 r r r �test_list_grouper_with_natu s z'TestGrouping.test_list_grouper_with_natz func,expectedZ transform)r~ rn r^ r. )r~ r_ c C s> t g g d��}|�d�}t|d |�dd� �}t�||� d S )N)r r r r c S s | S )Nr )rL r r r rO � rP z>TestGrouping.test_evaluate_with_empty_groups.<locals>.<lambda>)r r �getattrr, r- )r �funcr2 r r r1 r r r � test_evaluate_with_empty_groups� s z,TestGrouping.test_evaluate_with_empty_groupsc C s� t g ddd�}|�g �}|�� }t�||� t|jj�dks@t�t� |jj d t j g t � t j�d�� t� |jj d t j g t � d�d�� |jj d dks�t�|�|�jjdgks�t�d S ) Nr~ �float64)r~ rn r r )rn r� r )r r r* r, r- r] r� � groupingsr$ Zassert_numpy_array_equalZ group_infor/ r� rn r� rg )r r� �grr1 r r r �test_groupby_empty� s " z TestGrouping.test_groupby_emptyc C s� t dtjdgd tjdggdddgd��ddg�}|jddgd��� }t g ttdgd d �tg d d �gg g gddgd �dgd d�}t� ||� d S )NrL rZ � r r r )r )ro �object)rn r� )r� r� rg rm )r r^ r rn ) r r/ �nanr� r r+ r r r, r0 )r r r1 r2 r r r �%test_groupby_level_index_value_all_na� s $ z2TestGrouping.test_groupby_level_index_value_all_nac C sv t dddgdddggdddgd �}|�ddg�}||jd k }|�d��� }t d dgtjg dd �d �}t�||� d S )N�{ r� g �?r� g @�id�category�value)r r r� )r~ )rn r r^ ) r r� r� r r+ rt Z Int64Indexr, r0 )r r �emptyr1 r2 r r r �#test_groupby_multiindex_level_empty� s z0TestGrouping.test_groupby_multiindex_level_emptyN)+rV rW rX rb rw r� r� r� r� r� r� r� r� r� r �markZ parametrizer! r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r/ r� r rt Z Float64Indexr� r� r� r� r r r r rY � sX "F" ) ( (rY c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestGetGroupc C s� t t�ddddddg�ddddddgddddd d gd ��}|�d �}t|j�d }|�|�}|�t|��� �}|�t t|���}t � ||� t � ||� |�d dg�}t|j�d }|�|�}|�t|d ��� |d f�}|�t t|d ��|d f�}t � ||� t � ||� d}t j t|d�� |�d� W d Q R X t j t|d�� |�d� W d Q R X d}t j t|d�� |�d� W d Q R X d S )Nz 10-Oct-2013z 11-Oct-2013r4 r5 r r r r rI r )�DATE�labelZVALr� r r� z<must supply a tuple to get_group with multiple grouping keys)r zHmust supply a same-length tuple to get_group with multiple grouping keys)r4 r5 �baz)r rt � to_datetimer rA r� � get_groupr Z to_pydatetime�strr, r0 r r r" )r r r rk r� r> Zresult3r& r r r �test_get_group� s@ zTestGetGroup.test_get_groupc C s� t ddddg�}ddddg}|jt�|d |�|d �}|�t�dd��}t ddgddgd �}t�||� d }tj t |d �� |�t�dd�� W d Q R X d S ) Nr r r� r r rI rZ � )r� )r^ z"Interval\(10, 15, closed='right'\))r ) r r rt �cutr� ZIntervalr, r0 r r r )r r� r� Zbinsr r1 r2 r& r r r �test_get_group_empty_bins s z&TestGetGroup.test_get_group_empty_binsc C s� t ddddggdgd�j}|�d�}t dddgiddgd�}|�d�}t�||� t�ddddg�}t dd d � |D �i�}|�d�}|�d �}t d|d f|d fgiddgd�}t�||� d S ) N)r )r r Zids)r^ r r z 2010-01-01z 2010-01-02c S s g | ] }|f�qS r r )rK rL r r r r� $ s z@TestGetGroup.test_get_group_grouped_by_tuple.<locals>.<listcomp>)z 2010-01-01)r r� r r� r, r0 rt r� )r r r� r2 r1 �dtr r r � test_get_group_grouped_by_tuple s $z,TestGetGroup.test_get_group_grouped_by_tuplec sr t ddd� dD �i�� � �d�}� �� fdd��}|�t|j�� �d �}|�t|j�� �d �}t�||� d S )NrH c s s, | ]$}t j�d dd�D ]}||fV qqdS )r rI N)r/ r; rJ )rK rL rM r r r rN - s zKTestGetGroup.test_get_group_grouped_by_tuple_with_lambda.<locals>.<genexpr>)r r c s � j | df S )Nr )rF )rL )r r r rO 1 rP zJTestGetGroup.test_get_group_grouped_by_tuple_with_lambda.<locals>.<lambda>r )r r r� rA r� �keysr, r0 )r rS rT r2 r1 r )r r �+test_get_group_grouped_by_tuple_with_lambda* s z8TestGetGroup.test_get_group_grouped_by_tuple_with_lambdac C sL t �d�}d}t||td�}t jdd�}|�|�}tt|�d �d ksHt�d S )Nr )rn r9 )rl ) rt r r r� ru r �next�iterr$ )r r^ r Zseriesr� r' r r r �test_groupby_with_empty8 s z$TestGetGroup.test_groupby_with_emptyc C s� t dtd�i�}t�|�d��d�|jddg � t tdddgdd�d�}t�|�d��� |� t�|�d�� � |� t�|�d�� d �|� d S ) Nr� Zabssbabr rI r� r� )r~ )r^ r ) r rA r, r0 r r� rF r r# r+ Znth)r r r� r r r � test_groupby_with_single_column@ s "z,TestGetGroup.test_groupby_with_single_columnc C s| t ddddgddddgddddggd d d d gd �}|�d d g�}|�d d d g��� }|jd dksft�|jd dksxt�d S )Nr4 r5 r r r r� r r r� r� �thirdr6 )r )r4 r5 r )r4 r� r )r r� r r% r= r$ )r r r r r � test_gb_key_len_equal_axis_lenI s z+TestGetGroup.test_gb_key_len_equal_axis_lenN) rV rW rX r� r� r� r� r� r� r� r r r r r� � s / r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� TestIterationc C s� |� dg�}|j}||jks t�x0|j�� D ]"\}}|j| d |k�� s,t�q,W |� ddg�}|j}||jkstt�xR|j�� D ]D\}}|j| d |d k�� s�t�|j| d |d k�� s�t�q�W d S )Nr r r r )r r� r$ �itemsr= �all)r r r' r� �k�vr r r � test_groups\ s zTestIteration.test_groupsc C s0 |� dd� dd� g�}x|jjd D ]}q$W d S )Nc S s | � � S )N)�weekday)rL r r r rO o rP z9TestIteration.test_grouping_is_iterable.<locals>.<lambda>c S s | j S )N)�year)rL r r r rO o rP r )r r� r� )r Ztsframer' r r r r �test_grouping_is_iterablel s z'TestIteration.test_grouping_is_iterablec C s� t t�d��}t�ddddddg�}t�ddddddg�}|�||g�}t|�}dd|ddg fdd|dg fdd|d g fdd|d d g fg}xNt|�D ]B\}\\}} } || \} } } | |ks�t�| | ks�t�t� | | � q�W d S ) Nr r� r� �1�2r r r r r rI ) r r/ r[ r� r rA � enumerater$ r, r- )r r� �k1�k2r' �iteratedr2 r� r6 r7 r8 �e1�e2�e3r r r �test_multi_iteru s z TestIteration.test_multi_iterc C s� t �ddddddg�}t �ddddddg�}tt j�d�t j�d�||d�ddd d d d gd �}|�ddg�}t|�}|j}dd|j|dg fdd|j|ddg fdd|j|ddg fdd|j|dg fg}xRt |�D ]F\} \\} } } || \} }}| | k�st �|| k�st �t � | |� q�W t �ddddddg�|d< t �ddddddg�|d< |�ddg�}dd� |D �}t |�dk�s�t �|�dddg��� }|jjddd �}x|D ] \}}�q�W d S ) Nr� r� r� r� r )Zv1Zv2r� r r6 r7 r8 �four�five�six)r^ r� r r r rI r r r c S s i | ]\}}||�qS r r )rK rk �gpr r r � <dictcomp>� s z7TestIteration.test_multi_iter_frame.<locals>.<dictcomp>r r r )r r )rx ro )r/ r� r r; r< r rA r^ r= r� r$ r, r0 r] r* r� )r Z three_groupr� r r r' r r� r2 r� r6 r7 r8 r r r r� Z three_levelsrk �groupr r r �test_multi_iter_frame� s4 z#TestIteration.test_multi_iter_framec C s� t t|�d��� t t|�ddg��� t t|d �|d ��� t t|d �|d |d g��� t t|�d�d �� t t|�ddg�d �� d S )Nr r r )�dictr� r )r r r r r � test_dictify� s "zTestIteration.test_dictifyc C s t ddgddgd�t�ddg�d�}|�tjdd �d g�}t|j�d ksLt�|jd ksZt�t d �df|jkspt�t d �df|jks�t�|� t d �df�}t � ||j dgd d �f � |� t d �df�}t � ||j dgd d �f � t dddgdddgd�t�dddg�d�}|�tjdd �d g�}t|j�d k�s6t�|jd k�sFt�t d �df|jk�s^t�t d �df|jk�svt�|� t d �df�}t � ||j dd gd d �f � |� t d �df�}t � ||j dgd d �f � t dddgdddgd�t�dddg�d�}|�tjdd �d g�}t|j�dk�s(t�|jdk�s8t�t d �df|jk�sPt�t d �df|jk�sht�t d�df|jk�s�t�|� t d �df�}t � ||j dgd d �f � |� t d �df�}t � ||j dgd d �f � |� t d�df�}t � ||j d gd d �f � d S )N�starti� i. )�eventZchangez 2014-09-10z 2013-10-10)r^ r| )rl r r z 2014-09-30z 2013-10-31r r i�# z 2014-09-15z 2014-08-05r z 2014-08-31)r rt r r ru r] r� r$ Zngroupsr r� r, r0 rF )r r r' �resr r r � test_groupby_with_small_elem� sN z*TestIteration.test_groupby_with_small_elemc C s\ t �td�td�g�}tdddgg|d�}|�|d �}|jjd �� }d }||ksXt�d S ) NZAABZabar r r )r )r r� r zGrouping(('A', 'a'))) r r� rA r r r� r� �__repr__r$ )r �mir r� r1 r2 r r r �test_grouping_string_repr� s z'TestIteration.test_grouping_string_reprN) rV rW rX r� r� r r r r r r r r r r� [ s &4r� )�__doc__�numpyr/ r r� rt r r r r r r r Zpandas._testingZ_testingr, Zpandas.core.groupby.grouperr r rY r� r� r r r r �<module> s $ { M
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка