Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/groupby/__pycache__/test_apply.cpython-37.pyc
Назад
B �5�gy� � @ sx d dl mZm Z d dlmZ d dlZd dlZd dlm m Z d dl Z d dl m Z mZmZmZmZ d dlmZ dd� Zdd� Zd d � Ze jd d � �Zejjd e dddddgdddddgd��dddgfe d d ddgd dd dgd��d dgfe ddgi�dgfe ddddddddge d�d��ddgfe ddddddgddddddgd��dddgfe e d �ddddddddddg dd ddddddddg d ��dddgfe dddgdddggdddgd �ddgfgd d d!d"d#d$d%gd&�d'd(� �Z d)d*� Z e jejj d+d,�d-d.� ��Z!e jej�d/d0d1� ej"d2d1� ejj d+d,�d3�d4d1� ej"d5d1� ejj d+d,�d3�g�d6d7� ��Z#d8d9� Z$d:d;� Z%d<d=� Z&d>d?� Z'd@dA� Z(dBdC� Z)dDdE� Z*dFdG� Z+dHdI� Z,dJdK� Z-dLdM� Z.dNdO� Z/dPdQ� Z0dRdS� Z1dTdU� Z2dVdW� Z3dXdY� Z4ej�dZd[d\g�d]d^� �Z5d_d`� Z6dadb� Z7dcdd� Z8dedf� Z9dgdh� Z:didj� Z;dkdl� Z<dmdn� Z=dodp� Z>dqdr� Z?dsdt� Z@dudv� ZAej�dwe �B� e �B� �� e �B� �C� g�dxdy� �ZDdzd{� ZEej�d|d\d[g�d}d~� �ZFdd�� ZGej�d�e �He d���e �Id d�e jJd�dd�d��e�Kd�d�d�g�g�d�d�� �ZLej�d�d�d1� d dgddggfd�d1� d dhddhgfd�d1� d�d�gfd�d1� d dd��ddd��gfd�d1� d d iddigd diddiggfg�d�d�� �ZMd�d�� ZNej�d�d�d1� d�d1� g�d�d�� �ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTej�d�d[d\g�d�d�� �ZUej�d�d\e dddgdddggedddgeVd��d �gd[eddgejKd�d�gddgd��d��gg�d�d�� �ZWd�d�� ZXd�d�� ZYdS )�� )�date�datetime)�StringION)� DataFrame�Index� MultiIndex�Series� bdate_rangec C s� d} t jt| �d dddgddggd�}|�d�}|�|jj��� }|�|jj��dd� �}t � ||� t jt| �d dddgd �}t d d d gt dd �}t dddg|d�}|�d��dd� �}t �||� d S )Na+ 2011.05.16,00:00,1.40893 2011.05.16,01:00,1.40760 2011.05.16,02:00,1.40750 2011.05.16,03:00,1.40649 2011.05.17,02:00,1.40893 2011.05.17,03:00,1.40760 2011.05.17,04:00,1.40750 2011.05.17,05:00,1.40649 2011.05.18,02:00,1.40893 2011.05.18,03:00,1.40760 2011.05.18,04:00,1.40750 2011.05.18,05:00,1.40649r �time�value)�header�namesZ parse_dates� date_timec S s | � � S )N)�idxmax)�x� r �U/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/groupby/test_apply.py�<lambda>0 � z#test_apply_issues.<locals>.<lambda>)r r z 2011.05.16z 2011.05.17z 2011.05.18)�dtype�namez00:00z02:00)�indexc S s | d | d � � S )Nr r )r )r r r r r : r )�pdZread_csvr � set_index�groupbyr r r �apply�tm�assert_frame_equalr �objectr �assert_series_equal)�s�df�expected�resultZexp_idxr r r �test_apply_issues s r$ c s� t dddddgdddddgd�d d gd �� tj� jd d � � jd d � gd d dgd�} � jdd� � jD �d d��� fdd��}t�|| � d S )N�a�bg �?g @g @g @g @)�key�datar' r( )�columns� �float64r )�axis�keysc S s g | ] }t |��qS r )�str)�.0r r r r � <listcomp>F s z&test_apply_trivial.<locals>.<listcomp>)r, c s � j dd � S )Nr* )�iloc)r )r! r r r G r z$test_apply_trivial.<locals>.<lambda>) r r �concatr1 r �dtypesr r r )r" r# r )r! r �test_apply_trivial> s ,r4 c sz t dddddgdddddgd�d d gd �� tj� � gd d dgd�} � jdd� � jD �d d��� fdd��}t�|| � d S )Nr% r& g �?g @g @g @g @)r' r( r' r( )r) r* r+ r )r, r- c S s g | ] }t |��qS r )r. )r/ r r r r r0 T s z+test_apply_trivial_fail.<locals>.<listcomp>)r, c s � S )Nr )r )r! r r r T r z)test_apply_trivial_fail.<locals>.<lambda>)r r r2 r r3 r r r )r" r# r )r! r �test_apply_trivial_failM s (r5 c C s� d} t jjdd| d�}t jjdd| d�}t||t j�| �dddd g| d d ��}d d � }|�ddg�}|j}|j|j|j d�}|� � }|j } |� || |�\} } | r�t �d S )Ni� r i� )�size� �foo�bar�baz�qux� )r' �key2�value1�value2c S s dS )Nr* r )�gr r r �fk s ztest_fast_apply.<locals>.fr' r= )r, )�np�random�randintr �randnr �grouperZ _get_splitterZ _selected_objr, Z_get_group_keysZ sorted_dataZ fast_apply�AssertionError) �N�labelsZlabels2r! rA r@ rF Zsplitter� group_keysZsdata�valuesZmutatedr r r �test_fast_applyY s rL zdf, group_namesr* � r7 r% r& �c)r% r& � r< � � � � )r% �twoZ aaabbbcccc)r% �B�C)r) ZGH2936zGH7739 & GH10519ZGH10519ZGH2656ZGH12155ZGH20084ZGH21417)Zidsc s� g � � fdd�}� fdd�}� fdd�}� fdd�}� fd d �}x<|||||gD ]*}� d d �= | � d ��|� � |ksPt�qPW d S ) Nc s � � | j� | �� S )N)�appendr �copy)�group)r r r �f_copy� s z/test_group_apply_once_per_group.<locals>.f_copyc s � � | j� | S )N)rW r )rY )r r r �f_nocopy� s z1test_group_apply_once_per_group.<locals>.f_nocopyc s � � | j� dS )Nr )rW r )rY )r r r �f_scalar� s z1test_group_apply_once_per_group.<locals>.f_scalarc s � � | j� d S )N)rW r )rY )r r r �f_none� s z/test_group_apply_once_per_group.<locals>.f_nonec s � � | j� tdgdgd��S )Nr* )r% r& )rW r r )rY )r r r � f_constant_df� s z6test_group_apply_once_per_group.<locals>.f_constant_dfr% )r r rG )r! Z group_namesrZ r[ r\ r] r^ �funcr )r r � test_group_apply_once_per_group{ s ' r` c C sz d}t ddddddddgdddddd d d gd �dddddd d d gd �}|�d��dd� � | �� j�d�}||ksvt�d S )NrM r r* �0�2�4�6�8�10Z12Z14)�group_by_columnZ test_column)r rg c S s t d�S )N�function_called)�print)r! r r r r � r z2test_group_apply_once_per_group2.<locals>.<lambda>rh )r r r Z readouterr�out�countrG )Zcapsysr" r! r# r r r � test_group_apply_once_per_group2� s rl zGH-34998)�reasonc C sX t dddgtd�d��} dd� }dd� }| �d ��|�}| �d ��|�}t�||� d S ) Nr r* r7 )�Ar& c S s | S )Nr )rY r r r �slow� s z,test_apply_fast_slow_identical.<locals>.slowc S s | � � S )N)rX )rY r r r �fast� s z,test_apply_fast_slow_identical.<locals>.fastrn )r �ranger r r r )r! ro rp Zfast_dfZslow_dfr r r � test_apply_fast_slow_identical� s rr r_ c C s | S )Nr )r r r r r � r r c C s | d d � S )Nr )r r r r r � r )Zmarksc C s | j dd�S )NF)�deep)rX )r r r r r � r c C s | j dd�S )NT)rs )rX )r r r r r � r c C sH t ddddgddddgddddgd ��}|�d ��| �}t�||� d S ) Nr* rM r7 r< rP rQ rR rO )r@ r% r& r@ )r r r r r )r_ r! r# r r r �5test_groupby_apply_identity_maybecopy_index_identical� s (rt c C s� t tj�d�ddddddgd��} | jdd� dd �j}| j}t�||� t d dd ddd gi�} | jd | d< | � d�� � � � j }| j ddd�� � j }t�||� d S )NrQ �onerT �three)�foo1�foo2c S s | S )Nr )r r r r r r z-test_apply_with_mixed_dtype.<locals>.<lambda>r* )r, �c1rM rO g @�c2F)�as_index) r rB rC rE r r3 r r ry r �mean� reset_indexrz )r! r# r" �result1�result2r r r �test_apply_with_mixed_dtype s r� c C sP t ddddddgddddddgtd�d��} | jd d d �}| jd d d �}|�d�j}|�d�j}td dddg�}t�||� t�||� |�dd� �j}|�dd� �j}t � ddddg�}ddddg} t j | d d gd�} t�|| � t�||� tt d��} t ddgddgddgdd gddgg| d �} | jd d d ��d d� �j} t�| | � d S ) Nr& r% rN r* rM r7 rQ )Zitem_id�user_idr r� T)r{ Fr r< c S s | � d�S )NrM )�head)r r r r r / r z-test_groupby_as_index_apply.<locals>.<lambda>c S s | � d�S )NrM )r� )r r r r r 0 r )r r )r rM )r* r* )rM r< )r* r )r* rM )rM r* )r7 r< )r �abcderP )r c S s | S )Nr )r r r r r = r ) r rq r r� r r r �assert_index_equalr r � from_tuples�list) r! Zg_asZg_not_asZres_asZ res_not_as�expZ res_as_applyZres_not_as_applyZexp_not_as_apply�tpZ exp_as_apply�ind�resr r r �test_groupby_as_index_apply s, *r� c C sx | � ddg�}dd� }dd� }dd� }|�|�}|jjd ks@t�|�|�}|jjd ksZt�|�|�}|jjd kstt�d S ) Nrn rU c S s | � � }d|j_|S )N�stat)�describer r )rY r# r r r �descD s z.test_apply_concat_preserve_names.<locals>.descc S s$ | � � }d|j_|d t| �� }|S )Nr� )r� r r �len)rY r# r r r �desc2I s z/test_apply_concat_preserve_names.<locals>.desc2c S s0 | � � }dt| �d��|j_|d t| �� }|S )NZstat_�d)r� r� r r )rY r# r r r �desc3P s z/test_apply_concat_preserve_names.<locals>.desc3)rn rU r� )rn rU N)r r r r rG )Z three_group�groupedr� r� r� r# r Zresult3r r r � test_apply_concat_preserve_namesA s r� c C sb dd� } t ddd�}ttj�d�|d�}|�dd� �}|�| �}t|t�sNt �t � |j |j � d S ) Nc S s: t jdd�� t �| �}W d Q R X t| | | �� |d��S )N�ignore)�invalid)r Zdemeaned�logged)rB Zerrstate�logr r| )Zpiecer� r r r rA e s z%test_apply_series_to_frame.<locals>.fz1/1/2000�d )�periods)r c S s | j S )N)�month)r r r r r o r z,test_apply_series_to_frame.<locals>.<lambda>) r r rB rC rE r r � isinstancer rG r r� r )rA Zdr�tsr� r# r r r �test_apply_series_to_framed s r� c C s4 | � ddg�d �t�}|jjd d� dks0t�d S )Nrn rU rV rM )rn rU )r r r� r r rG )r! r# r r r � test_apply_series_yield_constantv s r� c C sl | � ddg��t�}t|t�s"t�|jd ks0t�| � ddg�ddg �t�}t|t�sZt�|jd ksht�d S )Nrn rU rV �D)r r r� r� r rG r )r! r# r r r � test_apply_frame_yield_constant{ s r� c C sH | � ddg�}|�t�}|�� d }t�|j|j� t�|j|j� d S )Nrn rU rV ) r r r� rk r r� r �assert_numpy_array_equalrK )r! r� r# r" r r r �test_apply_frame_to_series� s r� c C s^ | j ddgdd�}|�t�}|�� jdtjid�jdd�}t� |j |j � t� |j |j � d S )Nrn rU F)r{ rV )r) r� ) r r r� rk �renamerB �nan�dropr r� r r� rK )r! r� r# r" r r r �)test_apply_frame_not_as_index_column_name� s r� c s� dd� } � fdd�}t tj�ddd�tj�ddd�tj�d�d��� � �d ��| �}� �d �d �|�}tj||d d � |j d ks�t �d S ) Nc S s | � d�d �� �� d d� S )NrU rV rM )r �sum� sort_values)rY r r r �trans� s z-test_apply_frame_concat_series.<locals>.transc s* | � � �| j�d �}|�� �� d d� S )NrU rM )r �reindexr r� r� )rY r� )r! r r �trans2� s z.test_apply_frame_concat_series.<locals>.trans2r rP i� )rn rU rV rn rV F)Z check_names) r rB rC rD rE r r r r r rG )r� r� r# r� r )r! r � test_apply_frame_concat_series� s r� c C s: | � dd� �}|�dd� �}|�dd� �}t�||� d S )Nc S s | j S )N)r� )r r r r r � r z&test_apply_transform.<locals>.<lambda>c S s | d S )NrM r )r r r r r � r c S s | d S )NrM r )r r r r r � r )r r � transformr r )r� r� r# r" r r r �test_apply_transform� s r� c C sT | � dd� dd� g�}dd� }|�|�}x&|D ] \}}t�|j| ||�� q.W d S )Nc S s | j S )N)�year)r r r r r � r z,test_apply_multikey_corner.<locals>.<lambda>c S s | j S )N)r� )r r r r r � r c S s | � d�dd � S )Nrn ���)r� )rY r r r rA � s z%test_apply_multikey_corner.<locals>.f)r r r r �loc)�tsframer� rA r# r' rY r r r �test_apply_multikey_corner� s r� c C sb t dddddddddg td�d��} | jddd��d d � �}| �d ddd d dg�}t�||� d S )Nr* rM r7 rS )r' r r' F)rJ c S s | d d� S )NrM r )r r r r r � r z'test_apply_chunk_view.<locals>.<lambda>r r< rQ rR )r rq r r Ztaker r )r! r# r" r r r �test_apply_chunk_view� s $r� c C s` t ddddddddddg ddddddddddg tddd�d��} | �ddg�}|�d d � � d S ) Nr* rM r rS ���)r �name2r r r� c S s | j ddd�S )Nr T)�inplace)r� )r r r r r � r z4test_apply_no_name_column_conflict.<locals>.<lambda>)r rq r r )r! r� r r r �"test_apply_no_name_column_conflict� s r� c C sx t ddddddgt�dddgd�t�dd�d��} d d � }| �d ��|�}| �� }t�d d dgd�|d< t�||� d S )Ng �?g @r% r& rN rM g @)r� rN �vc S s, | d }||� � |�� |� � | d< | S )Nr� �v2)�min�max)rY r� r r r rA � s z#test_apply_typecast_fail.<locals>.fr� g g �?r* r� ) r rB �tile�aranger r rX r r )r! rA r# r" r r r �test_apply_typecast_fail� s r� c C s� t �ddddddgddddddgg�} tddddddgt�ddd gd�t�dd �d �| d �}d d� }|�d��|�}|�� }t�dddgd�|d< t � ||� d S )Nr r* rM r7 g �?g @r% r& rN g @)r� rN r� )r c S s, | d }||� � |�� |� � | d< | S )Nr� r� )r� r� )rY r� r r r rA � s z%test_apply_multiindex_fail.<locals>.fr� g g �?r� ) r � from_arraysr rB r� r� r r rX r r )r r! rA r# r" r r r �test_apply_multiindex_fail� s &r� c C s0 | � dd� ��dd� �}| d }t�||� d S )Nc S s | j S )N)r� )r r r r r r z#test_apply_corner.<locals>.<lambda>c S s | d S )NrM r )r r r r r r rM )r r r r )r� r# r" r r r �test_apply_corner s r� c C sh t ddddgddddgddd d gd ��} d d � }dd� }| �d��|�}| �d��|�}t�||� d S )Nr� �� i, r% r& rN r* rM r7 r< )�id_field�categoryr c S s( | j d dkr| �� S | | jdk S d S )Nr r* rN )�shaperX r� )r r r r �filt1 s z&test_apply_without_copy.<locals>.filt1c S s$ | j d dkr| S | | jdk S d S )Nr r* rN )r� r� )r r r r �filt2 s z&test_apply_without_copy.<locals>.filt2r� )r r r r r )r( r� r� r" r# r r r �test_apply_without_copy s r� � test_seriesTFc C s� t ddgddgddggddgdddgd�}| rp|�d�d }|jd d ��d d � �}|�� }|�� }t�||� n4|�d��d d � �}|�d�}|�d�}t�||� d S )Nr �p�o�X�Yr* rM )r) r r )�levelc S s | S )Nr )r r r r r 0 r z<test_apply_with_duplicated_non_sorted_axis.<locals>.<lambda>c S s | S )Nr )r r r r r 7 r ) r r r r � sort_indexr r r� r )r� r! Zserr# r" r r r �*test_apply_with_duplicated_non_sorted_axis( s ( r� c C sj ddddg} ddddg}t ddgd | d�|d�}t| |d d �}d d � }|�d �j�|�}t�||� d S )Nr* rM r7 r< ZGroup1ZGroup2)rY r )r r )r r c S s" | � t�| j�� | j�� d ��S )Nr* )r� rB r� r r� r� )r r r r �reindex_helperI s z1test_apply_reindex_values.<locals>.reindex_helperrY )r r r r r r r )rK �indicesr! r" r� r# r r r �test_apply_reindex_values? s r� c C sj d} t jjdd| d�}t|t j�| �ddddg| d d ��}|�d �}d d � }|�|�}d|ksft�d S )Ni� r r� )r6 r8 r9 r: r; r< )r' r> r? r' c S s | d d | d< | S )Nr> rM �value3r )r@ r r r rA ` s z"test_apply_corner_cases.<locals>.fr� )rB rC rD r rE r r rG )rH rI r! r� rA r# r r r �test_apply_corner_casesQ s r� c C sN t ddgdgd ddgd��} | �dg��dd � �}t�| j�| _| �dg��d d � �}t�|d |d � t d d dgdddgt�d�gd d��} dd� }| �d��|�d }| j }| j |_ t�||� dd� }t ddddgddddgd d d d gd d d!d!gd"��}|� � }t�|j �|_ |�d#��|�j}|�d#��|�j}t�||� d S )$Nr* rM z 2017-03-02r8 �inf)�Number�Date�Strr� c S s | j d S )Nr )r1 )r r r r r r r z;test_apply_numeric_coercion_when_datetime.<locals>.<lambda>c S s | j d S )Nr )r1 )r r r r r t r r� � � � �3rc z12:31:22r7 )rn rU �Tc S s | j d dg S )Nr rU )r1 )r@ r r r �get_B| s z8test_apply_numeric_coercion_when_datetime.<locals>.get_Brn rU c S s� t dddgtd�}dt| j�kr>t| | jdk jjd �|d< dt| j�kr�t| | jdk jjd �|d< t| | jdk jjd �|d< |S )N�p1�p2ZuseTime)r r �step1r �step2)r r r� �Stater. �MachinerK �oTime)Ztoolrj r r r � predictions� s z>test_apply_numeric_coercion_when_datetime.<locals>.predictionsr� r� � z2016-09-19 05:24:33z2016-09-19 23:59:04Z23Z36LZ36R)�Keyr� r� r� r� )r r r r � to_datetimer� r r � TimestamprU rn r rX r� r� )r! r" r# r� r� �df1�df2r r r �)test_apply_numeric_coercion_when_datetimeh s0 & r� c C s� t dddgt�d�gd d��} | j| j | d< | �d��d d � �}t dddgt�d d �gd t�d�gd d ���d�}t� ||� d S )Nrn rU rV z2017-02-01 00:00:00r7 )�clientidr �time_delta_zeror� c S s t | j�� | j�� d��S )N)� clientid_ager )r r� r� r )Zddfr r r r � s z?test_apply_aggregating_timedelta_and_datetime.<locals>.<lambda>r r� )r� r� r ) r rB Z datetime64r r r Z timedelta64r r r )r! r# r" r r r �-test_apply_aggregating_timedelta_and_datetime� s r� c C s� ddgddgddgddgddgg} t | d d gt�d d �d �}|�d ��� }t dddgdddgd��}|jd dd� t�||� d S )Nrn r� rU r� r� rV �( �2 �Name�Valuez 2020-09-01z 2020-09-05)r) r �Z )r� r� T)r� )r r � date_ranger r� r r r )r( r! r# r" r r r � test_apply_groupby_datetimeindex� s "r� c C s� t ddd� td�D �d��} dd� }dd � }| jd gd ��|�}t d d idgd�}d |j_| jd gd ��|�}t tddd�d d�dgd�}d |j_t�||� t�||� d S )Nr* c S s g | ] }t �� �qS r )r �now)r/ �nnr r r r0 � s z'test_time_field_bug.<locals>.<listcomp>r� )r% r& c S s t ddi�S )NrN rM )r )�batchr r r �func_with_no_date� s z.test_time_field_bug.<locals>.func_with_no_datec S s t tddd�dd��S )Ni� r* rM )r& rN )r r )r� r r r �func_with_date� s z+test_time_field_bug.<locals>.func_with_dater% )�byrN rM )r i� )r& rN ) r rq r r r r r r r )r! r� r Zdfg_no_conversionZdfg_no_conversion_expectedZdfg_conversionZdfg_conversion_expectedr r r �test_time_field_bug� s r c s� t ddddddddddddg ddddddddddddg dddd d d dd d d d dg dd ddddd dd d d dg d��} | �ddg�} | jddgdd�}dd� � |�� fdd �� d S ) Nr% r& rN r� �eg�������?rM r7 r< rP rQ rO r* gffffff @rS r� � � rR )�group1�group2�weightr r r T)r� �sortc S s t �| | ��d�}|S )Nr7 )rB �array�repeat)r r rj r r r �noddy� s z7test_gb_apply_list_of_unequal_len_arrays.<locals>.noddyc s � | j | j�S )N)r r )r )r r r r � r z:test_gb_apply_list_of_unequal_len_arrays.<locals>.<lambda>)r r r r )r! Z df_groupedr )r r �(test_gb_apply_list_of_unequal_len_arrays� s " r c C sL t ddddgddddgd��} dd � }| �d ��|�}t � }t�||� d S ) Nr r* rO rR r< rP )�groupsZ random_varsc S s d S )Nr )r r r r � test_func s z.test_groupby_apply_all_none.<locals>.test_funcr )r r r r r )Ztest_dfr r# r" r r r �test_groupby_apply_all_none� s r c C s� t ddddgddddgd��} t ddddgddddgd��}dd� }| �d��|�}|�d��|�}tjddgddggdd gd �}tjddgddggdd gd �}t ddgddgd�|d �}t ddgddgd�|d �}t�||� t�||� d S ) Nr* rM r r7 )r �varsc S s | j d dk rd S | jddg S )Nr rM r� )r� r1 )r r r r r s z0test_groupby_apply_none_first.<locals>.test_funcr )r )r )r r r r r� r r ) Ztest_df1Ztest_df2r r~ r Zindex1�index2Z expected1Z expected2r r r � test_groupby_apply_none_first s r c C sl t ddgddgd��} | �d�}|�dd� �}tdgd tjddgdggdd gd ��d�d �}t�||� d S ) Nr r* Zfilled�empty)r rY rY c S s | | j dk d S )Nr* r )r )rY r r r r # r z7test_groupby_apply_return_empty_chunk.<locals>.<lambda>r )r )r r ) r r r r r � from_productr� r r )r! r r# r" r r r �%test_groupby_apply_return_empty_chunk s r c C s| t d�� dddgdddgd��} | �d �}|�d d � �}t d d dgdddgd��}t�||� |�dd � �}t�||� d S )Nza a br* rM r7 r< rQ rP )rn rU rV rn c S s | | � � S )N)r� )r r r r r 3 r z-test_apply_with_mixed_types.<locals>.<lambda>gUUUUUU�?gUUUUUU�?g�������?g333333�?g �?)rU rV c S s | | � � S )N)r� )r r r r r 7 r )r �splitr r� r r r )r! r@ r# r" r r r �test_apply_with_mixed_types. s r c C sp t dddgit�ddg�d�} | �d��dd� �}tt�dg�t�dg�gtjddgdd�d�}t�||� d S )Nr% r* rM )r c S s | j S )N)r )r@ r r r r > r z*test_func_returns_object.<locals>.<lambda>)r )r r Z Int64Indexr r r r r )r! r# r" r r r �test_func_returns_object; s ,r �group_column_dtlikec C sT t dg| gd��}|�d��dd� �}t dgtdgddd�d gd �}t�||� d S ) Nr8 )r% r& r% c S s t dgdgd�S )N�spam�* )r )r )r r r r r Q r z+test_apply_datetime_issue.<locals>.<lambda>r r )r r r )r) )r r r r r r )r r! r# r" r r r �test_apply_datetime_issueF s r c C s� t t�d�t�d�t�d�t�d�t�d�d�dddddd�dddddd�d��} dd� }| �d ��|�d }tdgtjd gd d �d d �}t�||� d S )Nz2015-02-24 00:00:00)r r* rM r7 r< zsome UA stringzanother UA stringZ17661101)�dayZ userAgent�userIdc S s t dd� | �� D ��S )Nc S s i | ]\}}|� � jd |�qS )r )Z value_countsr )r/ rN r r r r � <dictcomp>v s zYtest_apply_series_return_dataframe_groups.<locals>.most_common_values.<locals>.<dictcomp>)r � iteritems)r! r r r �most_common_valuesu s zEtest_apply_series_return_dataframe_groups.<locals>.most_common_valuesr r z 2015-02-24)r )r r ) r r r� r r r � DatetimeIndexr r )Ztdfr" r# r" r r r �)test_apply_series_return_dataframe_groupsY s* r$ r� c C s� ddgd }| r@t j|dddgd�}t jddgdddgdd�}ntddgdd�}tt�d �|ttd ��ttd ��d ��� d dg�}|� d�� d d � �}tddgddgd�|d�}t � ||� |jjd dgks�t�d S )Nr* rM rP r7 )� categoriesrU )r% r )r r� )rn rU rV r� rn c S s | � � S )N)r� )r r r r r � r z-test_apply_multi_level_name.<locals>.<lambda>r� � )rV r� )r )r Z Categorical�CategoricalIndexr r rB r� r� rq r r r r r r r rG )r� r& Zexpected_indexr! r# r" r r r �test_apply_multi_level_name s & r( c C s� t jt�ddd�ddddft�ddd�ddd d ft�d dd�d d ddft�ddd�d dddfgdddddgd�} | �d��dd� �j}tt� d �t t tj t gdddddgd �}t � ||� d S ) Ni� r* �redZdarkre i� �greenZstormyrM �9i� �blue�brightr7 rf i� Zcalmr< �potatoZ observation�colorZmoodZ intensityZscore)r) c S s | j d S )Nr )r1 )r@ r r r r � r z;test_groupby_apply_datetime_result_dtypes.<locals>.<lambda>zdatetime64[ns])r )r Z from_recordsr r� r r r3 r rB r r Zint64r r )r( r# r" r r r �)test_groupby_apply_datetime_result_dtypes� s r0 r �abcZ2020r� )r� �freq)r% r )r% r* )r& r c C sB t dddgdddgd�| d�}|�d��dd� �}t�||� d S ) Nr* rM r )rY r )r rY c S s | S )Nr )r r r r r � r z8test_apply_index_has_complex_internals.<locals>.<lambda>)r r r r r )r r! r# r r r �&test_apply_index_has_complex_internals� s r3 zfunction, expected_valuesc C s | j �� S )N)r �to_list)r r r r r � r c C s t | j�� �S )N)�setr r4 )r r r r r � r c C s t | j�� �S )N)�tupler r4 )r r r r r � r )r r* )rM r7 c C s dd� t | j�� �D �S )Nc S s i | ]\}}||�qS r r )r/ �n�ir r r r � s z <lambda>.<locals>.<dictcomp>)� enumerater r4 )r r r r r � r c C s dd� t | j�� �D �S )Nc S s g | ]\}}||i�qS r r )r/ r7 r8 r r r r0 � s z <lambda>.<locals>.<listcomp>)r9 r r4 )r r r r r � r c C sN t ddddgdgd�}|�d��| �}t|tddgdd�d�}t�||� d S )Nrn rU r )r) )r )r )r r r r r r r )�functionZexpected_valuesr! r# r" r r r �1test_apply_function_returns_non_pandas_non_scalar� s r; c C sr dd� } t ddddgdddtjgd ��}|�d ��| �}td d gd gtjggtdddgd d�d�}t�||� d S )Nc S s | d j �� S )NrU )rK �flatten)rY r r r �fct� s z4test_apply_function_returns_numpy_array.<locals>.fctr% r& �noner* rM r7 )rn rU rn g �?g @g @)r )r ) r rB r� r r r r r r )r= r! r# r" r r r �'test_apply_function_returns_numpy_array� s (r? r: c C s | j S )N)r )�grr r r r � r c C s | j d d S )Nr* )r )r@ r r r r � r c C s� t ddddddddddg dgd�}|�d��| �}ttdddd g�tdddd g�td d g�gtdddgdd �d�}t�||� d S )Nr* rM r7 �id)r) r r< rR rS rP rQ rO )r )r )r r r r r r r )r: r! r# r" r r r � test_apply_function_index_return� s "(rB c C s� t ddddddgddddddgd ��} tjtd d �� | jd d d��dd� �}W d Q R X t dddgdddgd ��}t�||� d S )Nru rT rv r* rM r< rP rQ )rw rx zSelect only valid)�matchrw F)r{ c S s | � � S )N)r| )r r r r r � r zAtest_apply_function_with_indexing_return_column.<locals>.<lambda>g @g @)r r Zassert_produces_warning� FutureWarningr r r )r! r# r" r r r �/test_apply_function_with_indexing_return_column� s "rE c C s� dgd dgd dgd } t �| �}t j| dd�}tttd��d td�|d ��}tttd��d td�|d ��}|jd d d ��d d� �}|jd d d ��dd� �}t�||� d S )Nz 2001-01-01rM z 2001-01-02z 2001-01-03�UTC)�tzr7 rQ )r �y�tr F)rJ c S s | ddg � � S )Nr rH )rX )r! r r r r � r z1test_apply_with_timezones_aware.<locals>.<lambda>c S s | ddg � � S )Nr rH )rX )r! r r r r � r ) r r# r r� rq r r r r )�datesZ index_no_tzZindex_tzr� r� r~ r r r r � test_apply_with_timezones_aware� s rK c C s� t ddddddgddddddgd d d d d dgd��}t ddgddgddgd�tddgdd�d�}|jdd�}|�t�}t�||� |jdd�}dg|gd��| g �}t|| �|� }|�t�}t�||� d S )N�c �X r* rM r7 r< rP rQ r� r� r� r� r� �< )r% r& rN i i) � � r% )r )r )r r )ZnthZcorrwith) r r r r r� r r �get�getattr)Zreduction_funcr! r" �grpr# �args�_r r r �;test_apply_is_unchanged_when_other_methods_are_called_first s rV c C s� t ddddgtddd�tddd�tddd�tddd�gddddgd �td d d d gdd�d�} | �ddg�}|�dd� �}| jdddg }|�� }t�|dddg �|_ |j dd�}t � ||� x$|j j d D ]}t|�tks�t�q�W d S )Nr% r& i� r* r� rM r7 r< )rn rU rV r� �e �f �g �idx)r )r rn rU c S s | � d�S )Nr* )r� )r r r r r 1 r zRtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp.<locals>.<lambda>r )r) )r r r r r r1 r} r Z from_framer r� r r Zlevels�typerG )r! rS r# r" �valr r r �@test_apply_with_date_in_multiindex_does_not_convert_to_timestamp s" r] c C sz t tj�ddg�t�ddgddgg�d�} | jjddd ��d d � �}| jddd ��d d � �}t� ||j� t� || � d S ) NrQ r< rn rU r* rM )r) r )r, r� c S s | j ddd�S )Nr )r, r� )� droplevel)r r r r r I r zDtest_apply_by_cols_equals_apply_by_rows_transposed.<locals>.<lambda>c S s | j ddd�S )Nr* r )r, r� )r^ )r r r r r K r ) r rB rC r r r� r r r r )r! Zby_rowsZby_colsr r r �2test_apply_by_cols_equals_apply_by_rows_transposed= s r_ �dropnac C sv t dddddgdtjtjddgd�td �d �}|jd | d ��d d� �}| rR|�� n|jdddddg }t� ||� d S )Nr* rM r7 r< rP r% r& )�colrY Zxxyxz)r rY )r` c S s | S )Nr )r r r r r \ r z5test_apply_dropna_with_indexed_same.<locals>.<lambda>r ) r rB r� r� r r r` r1 r r )r` r! r# r" r r r �#test_apply_dropna_with_indexed_sameQ s rb zas_index, expected)r )r* r* )rM rM )r )r c C sT t ddddgddddgddddgd��}|jddg| d��dd� �}t�||� d S ) Nr* rM )r% r& rN r% r& )r{ c S s dS )Nr* r )r r r r r u r z5test_apply_as_index_constant_lambda.<locals>.<lambda>)r r r r Z assert_equal)r{ r" r! r# r r r �#test_apply_as_index_constant_lambdaa s ( rc c C s� t dddddgdddd d gdddddgd �td�d �} | �d ��dd� �}ttdd�tjdddddgd d gd�dd�}t�||� d S )Nr* rM r7 r< rP rQ rR rO rS r )rn rU rV )r rV c S s | j �� S )N)rn r� )r r r r r r z(test_sort_index_groups.<locals>.<lambda>)r* r )r* r* )r* rM )rM r7 )rM r< )r rn )r r ) r rq r r r r r� r r )r! r# r" r r r �test_sort_index_groupsy s ( rd c C sT t tjdddd�td�td�d��} | �| j| jjjg�� dd� �}t � || � d S ) Nz 2010-01-01Z12HrP )r2 r� r� )r �vals�letc S s | j dd � S )Nr )r1 )r r r r r � r z;test_positional_slice_groups_datetimelike.<locals>.<lambda>) r r r� rq r� r rf r �dtr r r )r" r# r r r �)test_positional_slice_groups_datetimelike� s rh )Zr r �ior �numpyrB ZpytestZ pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r r r Zpandas._testingZ_testingr r$ r4 r5 Z&skip_array_manager_not_yet_implementedrL �markZ parametrizerq r� r` rl Zxfailrr �paramrt 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 r r r r �todayr r r$ r( r0 r' Zinterval_rangeZ period_ranger� r3 r; r? rB rE rK rV r] r_ rb r rc rd rh r r r r �<module> s� ' "*$(. *0&%# 5 (& . " (.
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка