Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/apply/__pycache__/test_frame_apply.cpython-37.pyc
Назад
B �5�gҿ � @ s� d dl m Z d dlmZ d dlZd dlZd dlZd dlmZ d dl Z d dl m Z m Z m Z mZmZ d dlmZ d dlmZ dd� Zd d � Zej�d d d ed g�feddd�dfg�dd� �Zdd� Zdd� Zdd� Z ej�dddd d g�d d � �Z d d!� Z d"d#� Z ej�ddd$d%d&d'g�ej�d(ej g i d)d*�ej d gi d+d*�ej g d,d id-d*�ej g d.d/id0d*�ej d dgd.d/id1d*�g�ej�d2d3d4g�d5d6� ���Z!d7d8� Z"d9d:� Z#d;d<� Z$d=d>� Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+dKdL� Z,dMdN� Z-dOdP� Z.dQdR� Z/dSdT� Z0dUdV� Z1dWdX� Z2dYdZ� Z3d[d\� Z4d]d^� Z5d_d`� Z6dadb� Z7dcdd� Z8ej�ded4d3g�dfdg� �Z9dhdi� Z:djdk� Z;dldm� Z<dndo� Z=dpdq� Z>ej�drdsdtdej?g�dudv� �Z@dwdx� ZAdydz� ZBd{d|� ZCd}d~� ZDdd�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZIejjd�d�d�� d�d�� d�d�� gd�d�d�gd��d�d�� �ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPej�d2d3d4g�d�d�� �ZQej�d2d3d4g�d�d�� �ZRd�d�� ZSej�d�ee�Te � de d�d��fd&e d�d��fd%e d�d��fd e eUd��fd e eUd��fd$e d�d��fde d�d��fd'e d�d��fd�e d�d��fd�e d�d��fg �e�Te ejVd gd d�gg�de d�d�g�fd&e d�d�g�fd%e d�d g�fd e d/d/g�fd e d/d/g�fd$e d d�g�fde d�d�g�fd'e ejVd�g�fd�e ejVd�g�fd�e d d�g�fg ���d�d�� �ZWej�d�ee�Te � d�e � fd�e � fg�e�Te ejVd gd d�gg�d�e ejVd gd d�gg�fd�e ejVd gd d�gg�fg���d�d�� �ZXej�d,d d g�ej�d�d�i fd�i fd�i fd�d�d�ifd�d d�d��fd�d�d d��fd�d d�d�d��fg�d�d�� ��ZYej�d�d�d�d�g�d�dń �ZZd�dDŽ Z[ej�d�e d�dgd�d�gd̜�g�ej�d�d%d&dg�d�dτ ��Z\ej�d�d d�d/d�ejVg�d�d҄ �Z]d�dԄ Z^d�dք Z_d�d Z`d�dڄ Zaej�ded�d�d�d�d�d�d�d�g�ej�d2d�d4g�d�d� ��Zbej�ded$d�d'd�g�ej�d2d3d4g�d�d� ��Zcd�d� Zdd�d� ZedS )�� )�datetime)�chainN)�CategoricalDtype)� DataFrame� MultiIndex�Series� Timestamp� date_range)� zip_framesc C s t jdd��� t �| d �}| �t j�d }t�||� | �t j�d }t �| d �}||ksbt�| jd }| jt jdd�}t �| � |��}|| |ks�t�|j| jks�t�W d Q R X t ddddgddd d gd ��}|�d d � �}|j d ks�t�t |d j t��st�t |d j t��s t�d S )N�ignore)�all�Ar � )�axis�B�C�D)�c0�c1c S s | � d�S )N�category)�astype)�ts� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/apply/test_frame_apply.py�<lambda>* � z test_apply.<locals>.<lambda>)� � r r )�np�errstate�sqrt�apply�tm�assert_series_equal�mean�AssertionError�index�xsr �shape� isinstance�dtyper )� float_frame�result�expected�d�dfr r r � test_apply s" r0 c C s8 t dtddd�gi�} | jdd� dd�}t�|| � d S ) Nr z 2013-01-01�UTC)�tzc S s | S )Nr )�xr r r r 4 r z*test_apply_axis1_with_ea.<locals>.<lambda>r )r )r r r! r" �assert_frame_equal)r- r, r r r �test_apply_axis1_with_ea1 s r5 z data, dtype)r Nr z 2013-01-01r1 )r2 c C s< t | g| ggddg|d�}|jdd� dd�}t�||� d S )N�a)r& r* c S s | S )Nr )r3 r r r r ? r z0test_agg_axis1_duplicate_index.<locals>.<lambda>r )r )r �aggr" r4 )�datar* r- r, r r r � test_agg_axis1_duplicate_index8 s r9 c C sH t tddd�tjt�d�dd�d��} | jdd� d d �}t�|| � d S ) N�20130101� )�periods�s)�unit)r r c S s | S )Nr )r3 r r r r L r z/test_apply_mixed_datetimelike.<locals>.<lambda>r )r ) r r �pdZ to_timedeltar Zaranger! r" r4 )r- r, r r r � test_apply_mixed_datetimelikeC s r@ c C s� t � }|�tj�}|js t�|�tj�}|js2t�| d d� }|�dd� �}ttj| j d�}t � ||� | j d d �g f }|jdd� dd�}ttj| j d�}t � ||� t dgd�}|jd d� dd�}t �||� d S ) Nr c S s | � � S )N)r$ )r3 r r r r [ r z"test_apply_empty.<locals>.<lambda>)r& c S s | � � S )N)r$ )r3 r r r r ` r r )r r6 c S s | d S )Nr6 r )r3 r r r r f r )r r! r r �emptyr% r$ r �nan�columnsr" r# �locr& r4 )r+ � empty_framer, Zno_rowsr- �no_colsr r r �test_apply_emptyP s rG c C s� t � } g }| j|jddd�}t�|| � | j|jddd�}tg tjg td�t j d�}t� ||� t ddd gd �}|j|jddd�}t�||� |j|jddd�}tg tjg td�t j d�}t� ||� |g ks�t �d S ) Nr �expand)r � result_type�reduce)r* )r& r* r6 �b�c)rC ) r r! �appendr" r4 r r? �Index�objectr �float64r# r% )rE r3 r, r- Zempty_with_colsr r r � test_apply_with_reduce_emptyj s rQ �func�sum�prod�anyr c C s< t dddgd�}|�tt| ��}t|| �� }t�||� d S )Nr6 rK rL )rC )r r! �getattrr r" r# )rR r/ r, r- r r r �test_apply_funcs_over_empty� s rW c C sb t dddgd�} | �� }td| jd�}t�||� | j�� }tg t�g �t j d�}t�||� d S )Nr6 rK rL )rC r )r& )r& r* ) r Znuniquer rC r" r# �Tr? rN r rP )r/ r, r- r r r �test_nunique_empty� s rY c C s� t dddgdddgddd ggd d d gd �} | jd d� dd�}tdddgd d d g�}t�||� | jjdd� dd�}t�||� d S )Nr r r; r � � � � � r6 rL )r& c S s | d S )Nr r )r= r r r r � r z/test_apply_standard_nonunique.<locals>.<lambda>)r c S s | d S )Nr r )r= r r r r � r r )r r! r r" r# rX )r/ r, r- r r r � test_apply_standard_nonunique� s * r_ r$ �min�max�stdz args,kwdsZno_args_or_kwds)�idZaxis_from_argsr Zaxis_from_kwdsZ numeric_onlyTZ optional_kwdsZ args_and_kwds�howr7 r! c C sb t |�dkr,|dkr,| j�tjjtdd�� t||�|f|�|�}t||�||�}t� ||� d S )Nr r7 z>agg/apply signature mismatch - agg passes 2nd argument to func)Zraises�reason) �len�nodeZ add_marker�pytest�markZxfail� TypeErrorrV r" r# )�requestr+ rR �args�kwdsrd r, r- r r r � test_apply_with_string_funcs� s rn c sp � j tjdd�}t� �� g� jd�}t�||� � j tjddd�}� jdd��t�fdd�� jD ��}t�||� � j � fd d �ddd�}tt t � j����t�gt � j� d � j� jd �}t�||� � j � fd d �dd�}tt t � j����t�fdd�� jD �d � jd�}t�||� |}|j dd � ddd�}t�||� |}|j dd � ddd�}|� � }t�||� d S )N� broadcast)rI )r& r )r rI )r c s i | ] }� |�qS r r )�.0rL )�mr r � <dictcomp>� s z(test_apply_broadcast.<locals>.<dictcomp>c s t tt� j���S )N)�list�rangerf rC )r3 )r+ r r r � r z&test_apply_broadcast.<locals>.<lambda>rP )r* r& rC c s t tt� j���S )N)rs rt rf r& )r3 )r+ r r r � r c s i | ] }� |�qS r r )rp rL )rq r r rr � s )r* r& c S s dddgS )Nr r r; r )r3 r r r r � r c S s t dddgtd�d�S )Nr r r; �abc)r& )r rs )r3 r r r r � r ) r! r r$ r r& r" r4 rC rs rt rf �copy)r+ �int_frame_const_colr, r- r/ r )r+ rq r �test_apply_broadcast� sH rx c C s� dd� }| j |dd� | j |ddd� | j tjdd�}| � dd� �}t�||� | j tjddd�}| j d d� dd �}t�||� | j d d� dd�}| d }t�||� |j |dd� |j |ddd� d S ) Nc S s" t | tj�st�| jdks t�d S )Nr )r) r �ndarrayr% �ndim)r3 r r r � _assert_raw� s z#test_apply_raw.<locals>._assert_rawT)�rawr )r r| c S s | j �� S )N)�valuesr$ )r3 r r r r � r z test_apply_raw.<locals>.<lambda>c S s | j �� S )N)r} r$ )r3 r r r r � r )r c S s | d S )Nr r )r3 r r r r r r )r! r r$ r" r# r4 )r+ Zmixed_type_framer{ r, r- r r r �test_apply_raw� s r~ c C s> | j d }| jtjdd�| }t�| �|��}||ks:t�d S )Nr r )r )r& r! r r$ r' r% )r+ r. r, r- r r r �test_apply_axis1 s r c C s� t dgdgd��} | d d� jtjdd�}ttjtjg dd�d �}t� ||� t dgdgd��} | jd d � dd�}tdgdgd �}t� ||� | jd d � dd�}tdgdgd �}t� ||� d S ) N�foog �?)r r r r )r �int64)r* )r& c S s | d S )Nr r )r3 r r r r r z/test_apply_mixed_dtype_corner.<locals>.<lambda>c S s | d S )Nr r )r3 r r r r r ) r r! r r$ r rB r? rN r" r# )r/ r, r- r r r � test_apply_mixed_dtype_corner s r� c C s� t dddgd�} t dddgd�}dd� }tjdd ��> || d d � � || d d � � ||d d � � ||dd � � W d Q R X | jdd � dd�}t|t �s�t�d S )Nr6 rK rL )r& )rC c s� t jdd��$ t �dt� �tjg dd��}W d Q R X t|tj� �d� ��fdd � }|� |d d � |dd � |ddd � d S )NT)�recordr Zf8)r* r Fc sN � j �| |d�}�r<� �| �}t|t�s,t�|j|ksJt�nt|t�sJt�d S )N)r r| )r! Z _get_agg_axisr) r r% r& r )r r| r, Zagg_axis)r/ �f� is_reductionr r �_checkit. s z=test_apply_empty_infer_type.<locals>._check.<locals>._checkitr )r )r| )r r| )r F)�warnings�catch_warnings� simplefilter�RuntimeWarningr �arrayr) ry )r/ r� Ztest_resr� r )r/ r� r� r �_check( s z+test_apply_empty_infer_type.<locals>._checkr )r c S s | S )Nr )r3 r r r r = r z-test_apply_empty_infer_type.<locals>.<lambda>c S s | � � S )N)r$ )r3 r r r r > r c S s | S )Nr )r3 r r r r ? r c S s | � � S )N)r$ )r3 r r r r @ r c S s | � � S )N)r$ )r3 r r r r B r ro )rI )r r r r! r) r% )rF �no_indexr� r, r r r �test_apply_empty_infer_type$ s r� c C s� ddd�}ddd�}ddd�}| j |d d �}| � d d � �}t�||� | j |d d �}| � d d � �}t�||� | j |dd d�}| � dd � �}t�||� d S )Nr c S s | | S )Nr )r3 �howmuchr r r �add_someG s z+test_apply_with_args_kwds.<locals>.add_somec S s | � � | S )N)r$ )r3 r� r r r � agg_and_addJ s z.test_apply_with_args_kwds.<locals>.agg_and_addr c S s | | | S )Nr )r3 �sub�divider r r �subtract_and_divideM s z6test_apply_with_args_kwds.<locals>.subtract_and_divider )r� c S s | d S )Nr r )r3 r r r r Q r z+test_apply_with_args_kwds.<locals>.<lambda>c S s | � � d S )Nr )r$ )r3 r r r r U r )r )rl r� c S s | d d S )Ng @r )r3 r r r r Y r )r )r )r )r! r" r4 r# )r+ r� r� r� r, r- r r r �test_apply_with_args_kwdsF s r� c C s | � t�}t�|| � d S )N)r! rs r" r4 )r+ r, r r r �test_apply_yield_list] s r� c C s@ t j| d jd d d�< | �d�}| jt jdd�}t�||� d S )Nr r r )r )r rB Zilocr$ r! r" r# )r+ r- r, r r r �test_apply_reduce_Seriesb s r� c C s� t ddgddggddgddgd �} | jtd d �}tddd �ddd �g| jd �}t�||� | jtdd �}tddd�ddd�g| jd �}t�||� d S )Nr r r; r r r �i0�i1)rC r& r )r )r� r� )r& )r r )r r! �dictr rC r" r# r& )r8 r, r- r r r �test_apply_reduce_to_dicti s " r� c C s� t tj�dd��} | jtjdd�}t dd� | �� D �| jd�}t � ||� | jtjdd�}t d d� | j �� D �| j d�j }t � ||� d S ) N� � r )r c S s i | ]\}}|� � |�qS r )�describe)rp �i�vr r r rr z s z2test_apply_differently_indexed.<locals>.<dictcomp>)rC r c S s i | ]\}}|� � |�qS r )r� )rp r� r� r r r rr ~ s ) r r �random�randnr! r r� �itemsrC r" r4 rX r& )r/ r, r- r r r � test_apply_differently_indexedv s r� c C s� t dddgdddgdddgdddgdd dgdd dggd d d gd�} dd� }| j|dd�}t tddd�ddgtddd�ddgtddd�ddgtddd�ddgtddd�d dgtddd�d dggd d d gd�} | j|dd�}t�||� d S )Nr ZABC0�2 ZYUM0r� ZDEF0r ZABC1ZYUM1ZDEF1r6 �market�position)rC c S s | d S )Nr� r )�rr r r r� � s ztest_apply_bug.<locals>.f)r i� r; r rZ r[ )r r! r r" r# )Z positionsr� r- r, r r r �test_apply_bug� s( r� c C s� t dddddddddddg dddddddddddg dddddddddddg tj�d�tj�d�tj�d�d��} | jd d � d d �jd d�}t�|| � d S )Nr� �bar�one�twoZdullZshiny� )r r r r �E�Fc S s | S )Nr )r3 r r r r � r z,test_apply_convert_objects.<locals>.<lambda>r )r T)r )r r r� r� r! �_convertr" r4 )r- r, r r r �test_apply_convert_objects� sN r� c s� � � dd� �}t� j� jd�}t�||� � j dd� dd�}t� j� jd�}t�||� � � dd� �}tt�� jt � j�df�� j� jd�}t� ||� � j d d� dd�}t� fd d �� � � D ��}� j|_t�||� d S ) Nc S s | j S )N)�name)r3 r r r r � r z(test_apply_attach_name.<locals>.<lambda>)r& c S s | j S )N)r� )r3 r r r r � r r )r c S s t �| jt| ��S )N)r �repeatr� rf )r3 r r r r � r )r& rC c S s t �| jt| ��S )N)r r� r� rf )r3 r r r r � r c 3 s$ | ] }t �|d t� j��V qdS )r N)r r� rf rC )rp �t)r+ r r � <genexpr>� s z)test_apply_attach_name.<locals>.<genexpr>) r! r rC r" r# r& r r Ztilerf r4 � itertuples)r+ r, r- r )r+ r �test_apply_attach_name� s" r� c C s� t �dddgdddgg�}tddgddgd d gg|d d gd �}|�dd� d�}tddgddgd d gg|ddgd �}tj||dd� d S )Nr6 rK rL r. r r r; r rZ r[ Zcol1Zcol2)r& rC c S s t t| �t| �d��S )N)r` ra )r r` ra )r3 r r r r � r z(test_apply_multi_index.<locals>.<lambda>r` ra T)� check_like)r Z from_arraysr r! r" r4 )r+ r& r= r, r- r r r �test_apply_multi_index� s $$r� c C s� t ddgddgg�} tddd�ddd�g�}t ddgdd gg�}tddd�dd d�g�}d d � }xf| |f||fgD ]R\}}|j|d d �}|j|dd �}|�|�} t�||� t�||� t�| |� qrW d S )Nr� r� Zspam�eggs)r r r r r r; c S s | � � S )N)�to_dict)r3 r r r r r z!test_apply_dict.<locals>.<lambda>rJ )rI rH )r r r! r" r# r4 ) r ZA_dictsr ZB_dicts�fnr/ ZdictsZ reduce_trueZ reduce_falseZ reduce_noner r r �test_apply_dict� s r� c C s� | � dd� �}t�|| d � | � t� | � dd� �d d }t|t�sLt�tddgd �}|� d d� �jd }|t kszt�td dgd �}|� d d� �jd }|t ks�t�tt j � d ��}|� � }ddddg}||_ |� t�}||_ |� t�}t�||� td�|d< t�d�|d< |� t�}x4dD ],}|jd|f t|jd|f �k�s"t��q"W t� ttd�d�ttd�d�tg g g d��g}x<|D ]4}x,tdd� gD ] } |� | �}t�||� �q�W �q�W d S )Nc S s | d S )Nr r )r3 r r r r r z test_applymap.<locals>.<lambda>r c S s | | fS )Nr )r3 r r r r r r r r r6 )r8 c S s | S )Nr )r3 r r r r r g �?c S s | S )Nr )r3 r r r r r )r; r r: r z1 min� timedelta)r r� �ABC)rC )r& )r r r c S s | S )Nr )r3 r r r r : r )�applymapr" r4 �typer) �tupler% r �dtypesrO r r� rv rC �strr r? � TimedeltarD rs �round) r+ Zappliedr, r/ Zdf2�colsr- r� Z empty_framesrR r r r � test_applymap s@ , r� c C sF t ddgddgg�jdd� dd�} t ddgdd gg�}t�| |� d S ) Nr r r; r c S s | | S )Nr )r3 �yr r r r A r z&test_applymap_kwargs.<locals>.<lambda>)r� rZ r[ )r r� r" r4 )r, r- r r r �test_applymap_kwargs? s "r� c C sj | � dd� �}| �� }tjjdd| jtd�}tj||< |j dd� dd�}|�� }tj||< t � ||� d S ) Nc S s t t| ��S )N)rf r� )r3 r r r r H r z)test_applymap_na_ignore.<locals>.<lambda>r r )�sizer* c S s t t| ��S )N)rf r� )r3 r r r r M r r )Z na_action) r� rv r r� �randintr( �boolr? ZNAr" r4 )r+ Z strlen_frameZfloat_frame_with_na�maskZstrlen_frame_na_ignoreZstrlen_frame_with_nar r r �test_applymap_na_ignoreF s r� c C s* t tddd��} dd� }t| ��|� d S )Nz1/1/2000r� )r< c S s | j | j| jfS )N)�hour�day�month)r3 r r r rR X s z*test_applymap_box_timestamps.<locals>.func)r r r r� )ZserrR r r r � test_applymap_box_timestampsT s r� c C s� t td�td�gtddd�tddd�gt�d�t�d�gtjddd�tjddd�gd ��} | �d d � �}t d d gd d gd d gddgd ��}t�||� d S )Nz 2011-01-01z 2011-01-02z US/Eastern)r2 z1 daysz2 days�M)�freq)r6 rK rL r. c S s t | �jS )N)r� �__name__)r3 r r r r p r z#test_applymap_box.<locals>.<lambda>r r� �Period)r r r? r� r� r� r" r4 )r/ r, r- r r r �test_applymap_box_ s r� c s^ ddl m� tdtddd�gi�} | �� fdd��} | �� fdd��} | jj}|d ksZt�d S ) Nr )�BDay�x1i� r c s | � � S )Nr )r3 )r� r r r � r z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>c s | � � S )Nr )r3 )r� r r r � r zM8[ns])Zpandas.tseries.offsetsr� r r r� r� r* r% )r/ r, r )r� r �(test_frame_apply_dont_convert_datetime64| s r� c C s� t dtdddd�i�} | �dd� �}t�|| � | �dd� �}t dtd ddd�i�}t�||� t dd d d d gid d�} | �dd� �}t�|| � d S )N�dtz 2015-01-01r; zEurope/Brussels)r< r2 c S s | S )Nr )r3 r r r r � r z,test_apply_non_numpy_dtype.<locals>.<lambda>c S s | t �d� S )NZ1day)r? r� )r3 r r r r � r z 2015-01-02r6 rK rL r )r* c S s | S )Nr )r3 r r r r � r )r r r! r" r4 )r/ r, r- r r r �test_apply_non_numpy_dtype� s r� c C sR t ddgddggddgd�} t ddggddgdgd�}| �dg�}t�||� d S ) Nr r r r; r6 )rC r` )rC r& )r r7 r" r4 )r/ r- r, r r r � test_apply_dup_names_multi_agg� s r� �opc C sh dd� }t t�d�td�d�}t|| �|dd�}tdddgdddgdddgdddgg�}t�||� d S ) Nc S s" d| d d| d d| d gS )Nr r r r r )�rowr r r � apply_list� s z3test_apply_nested_result_axis_1.<locals>.apply_list)r r ZABCD)rC r )r g )r r Zzerosrs rV r r" r# )r� r� r/ r, r- r r r � test_apply_nested_result_axis_1� s &r� c C sR t dtddd�gidd�} | �dd� �}t�|| � | �d d� �}t�|| � d S ) Nr� Z2020r1 )r2 zdatetime64[ns, UTC])r* c S s | S )Nr )r3 r r r r � r z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c S s | � � S )N)rv )r3 r r r r � r )r r r! r" r4 )r- r, r r r �%test_apply_noreduction_tzaware_object� s r� c sn t ddddgi�} g � � fdd�}� fdd�}x:||gD ].}� d d �= | j|dd � � t| j�ks8t�q8W d S ) Nr6 r r r; c s � � | j� d S )N)rM r� )r� )�namesr r �reducing_function� s z8test_apply_function_runs_once.<locals>.reducing_functionc s � � | j� | S )N)rM r� )r� )r� r r �non_reducing_function� s z<test_apply_function_runs_once.<locals>.non_reducing_function)r )r r! rs r& r% )r/ r� r� rR r )r� r � test_apply_function_runs_once� s r� c st t ddddgi�} g � � fdd�}� fdd�}x@||gD ]4}� d d �= | j|d dd � � t| j�� �ks8t�q8W d S ) Nr6 r r r; c s � � | � d S )N)�extend)r� )r} r r r� � s z<test_apply_raw_function_runs_once.<locals>.reducing_functionc s � � | � | S )N)r� )r� )r} r r r� � s z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionT)r| r )r r! rs r6 �to_listr% )r/ r� r� rR r )r} r �!test_apply_raw_function_runs_once� s r� c sj t ddddgi�} g � � fdd�}� fdd�}x6||gD ]*}� d d �= | �|� � | j�� ks8t�q8W d S ) Nr6 r r r; c s � � | � d S )N)rM )�val)r} r r r� � s z;test_applymap_function_runs_once.<locals>.reducing_functionc s � � | � | S )N)rM )r� )r} r r r� � s z?test_applymap_function_runs_once.<locals>.non_reducing_function)r r� r6 r� r% )r/ r� r� rR r )r} r � test_applymap_function_runs_once� s r� c C sP t t�ddg�dgd�} t t�ddg�dgtd�}| �dd� �}t�||� d S )Ns abcds efgh�col)rC )rC r* c S s | � d�S )NrO )r )r3 r r r r � r z-test_apply_with_byte_string.<locals>.<lambda>)r r r� rO r! r" r4 )r/ r- r, r r r �test_apply_with_byte_string� s r� r� �asd� c s` dd dddt jg}td|idd�}|j�� fdd��}t� fd d �|D �dd �}t�||� d S ) Nr� r� Zcder6 r )r* c s | � kS )Nr )r3 )r� r r r r z/test_apply_category_equalness.<locals>.<lambda>c s$ g | ] }t �|�rtjn|� k�qS r )r? Zisnullr �NaN)rp r3 )r� r r � <listcomp> s z1test_apply_category_equalness.<locals>.<listcomp>)r� )r r� r r6 r! r r" r# )r� Z df_valuesr/ r, r- r )r� r � test_apply_category_equalness s r� c C sV t tj�dd��} | jtjjdd�j}|dks2t�| jtjjdd�j}|dksRt�d S )Nr� r r )r )r� r )r[ r ) r r r� Zrandr! Zfftr( r% Zrfft)r/ r, r r r �test_infer_row_shape s r� c C s" t ddgddggddgd�} | jdd� dd�}td d � | �� D ��}t�||� td �td �g| d < | jdd� dd�}t�||� | d | d �dd� �}tddiddig�}t�||� t � } dddg| d< dddg| d< t�ddd g�| d < | jd d� dd�}ti i i g�}t�||� d S ) Nr r r6 rK )rC c S s d| d | d iS )Nr= r6 rK r )r3 r r r r r z,test_with_dictlike_columns.<locals>.<lambda>)r c S s g | ] }d di�qS )r= r; r )rp r� r r r r� ! s z.test_with_dictlike_columns.<locals>.<listcomp>z2017-05-01 00:00:00z2017-05-02 00:00:00r" c S s d| d | d iS )Nr= r6 rK r )r3 r r r r ( r c S s d| iS )Nr= r )r3 r r r r , r r= r; �X�Y�Z�authorZBBCZNBCZN24Z publisherz17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09�datec S s i S )Nr )r3 r r r r 7 r ) r r! r r� r" r# r r? � to_datetime)r/ r, r- r r r �test_with_dictlike_columns s& r� c C s� t ddgddggddgd�} | jdd� ddd �}t d d d gi�}t�||� td �td �g| d< | jdd� ddd �}t�||� d S )Nr r r6 rK )rC c S s d| d | d iS )Nr= r6 rK r )r3 r r r r ? r z7test_with_dictlike_columns_with_infer.<locals>.<lambda>rH )r rI r= r; z2017-05-01 00:00:00z2017-05-02 00:00:00r" c S s d| d | d iS )Nr= r6 rK r )r3 r r r r G r )r r! r" r4 r )r/ r, r- r r r �%test_with_dictlike_columns_with_infer< s r� c C s t ttj�d��ddddgtdddd�d ��} | dd g jtd d �}td d� | dd g �� D ��}t � ||� | ddg jtd d �}tdd� | ddg �� D ��}t � ||� t tdd gdgg�tdgddgg�d��} t � ddg�| _ | jdd� d d �}tg dgg| j d�}t � ||� d S )Nr r6 rs Zof�wordsz 2016-10-01�H)r< r� )r6 rK r rK r )r c S s g | ]}|d d� �qS )r Nr )rp r� r r r r� V s z.test_with_listlike_columns.<locals>.<listcomp>r c S s g | ]}|d d� �qS )r Nr )rp r� r r r r� Z s �q�zr� )r3 r� )r� Zj0)r� Zj1c s � fdd�� d D �S )Nc s g | ]}|� d kr|�qS )r� r )rp �el)r� r r r� a s z@test_with_listlike_columns.<locals>.<lambda>.<locals>.<listcomp>r3 r )r� r )r� r r a r z,test_with_listlike_columns.<locals>.<lambda>)r& )r r r r� r� r r! r� r� r" r# r � from_tuplesr& )r/ r, r- r r r �test_with_listlike_columnsK s *r c C s\ t ddgddgtd�td�gd��} | jdd � d d �}td d � | �� D ��}t�||� d S )Ng �?g @r� r� z2017-11-29 03:30:00z2017-11-29 03:45:00)�number�stringr c S s | j | jfS )N)r r )r� r r r r s r z1test_infer_output_shape_columns.<locals>.<lambda>r )r c S s g | ]}|j |jf�qS r )r r )rp r� r r r r� t s z3test_infer_output_shape_columns.<locals>.<listcomp>)r r r! r r� r" r# )r/ r, r- r r r � test_infer_output_shape_columnsf s r c C s� t tj�dd�dddgd�} | jdd� d d �}td d � | �� D ��}t�||� | jd d� d d �}tdd � | �� D ��}t�||� t dd ddgit d�d�} | jdd� d d �}tdd � | �� D �| j d�}t�||� | jdd� d d �}tdd � | �� D �| j d�}t�||� t t d�t d�t d�t d�gdd d dgd dd dgd ddd gd ��} d d � }| j|d d �}td!d � | �� D ��}t�||� d S )"Nr[ r; r r r )rC c S s dddgS )Nr r r; r )r3 r r r r } r z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r )r c S s g | ]}d ddg�qS )r r r; r )rp r� r r r r� ~ s z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c S s ddgS )Nr r r )r3 r r r r � r c S s g | ] }d dg�qS )r r r )rp r� r r r r� � s r6 r ru )r& c S s t �d�S )Nr )r �ones)r� r r r r � r c S s g | ]}t �d ��qS )r )r r )rp r� r r r r� � s c S s t �d�S )Nr )r r )r� r r r r � r c S s g | ]}t �d ��qS )r )r r )rp r� r r r r� � s z 2010-02-01z 2010-02-04z 2010-02-05z 2010-02-06r^ rZ r )r6 rK rL r. c S s dS )N)r r r )r3 r r r �fun� s z5test_infer_output_shape_listlike_columns.<locals>.func S s g | ]}d �qS ))r r r )rp r� r r r r� � s ) r r r� r� r! r r� r" r# rs r& r )r/ r, r- r r r r �(test_infer_output_shape_listlike_columnsx s4 r c C s� t tj�dd�dddgd�} | jdd� d d �}td d � | �� D ��}t�||� | jd d� d d �}tdd � | �� D ��}t�||� d S )Nr r; r r r )rC c S s dddgS )Nr r r; r )r3 r r r r � r z3test_consistent_coerce_for_shapes.<locals>.<lambda>r )r c S s g | ]}d ddg�qS )r r r; r )rp r� r r r r� � s z5test_consistent_coerce_for_shapes.<locals>.<listcomp>c S s ddgS )Nr r r )r3 r r r r � r c S s g | ] }d dg�qS )r r r )rp r� r r r r� � s ) r r r� r� r! r r� r" r# )r/ r, r- r r r �!test_consistent_coerce_for_shapes� s r c C sd | }|j dd� dd�}| jdddd�d �}t�||� |j d d� dd�}|ddg }t�||� d S ) Nc S s t dddgdddgd�S )Nr r r; �test�otherr� )r& )r )r3 r r r r � r z'test_consistent_names.<locals>.<lambda>r )r r r r� )r r r )rC c S s t ddgddgd�S )Nr r r r )r& )r )r3 r r r r � r )r! �renamer" r4 )rw r/ r, r- r r r �test_consistent_names� s r c s4 | }|j dd� ddd�}|�� }dddg|_t�||� |j dd� ddd�}|d d g �� }ddg|_t�||� |j d d� dd d�}|�� }t�||� d ddg� |j � fdd�dd d�}|�� }t�||� |j dd� dd�}|�� }t�||� d ddg� |j � fdd�dd�}|�� }� |_t�||� d S )Nc S s dddgS )Nr r r; r )r3 r r r r � r z"test_result_type.<locals>.<lambda>r rH )r rI r r c S s ddgS )Nr r r )r3 r r r r � r r r c S s dddgS )Nr r r; r )r3 r r r r � r ro r r� r� c s t dddg� d�S )Nr r r; )r& )r )r3 )rC r r r � r c S s t dddg| jd�S )Nr r r; )r& )r r& )r3 r r r r � r )r c s t dddg� d�S )Nr r r; )r& )r )r3 )rC r r r � r )r! rv rC r" r4 )rw r/ r, r- r )rC r �test_result_type� s2 r �boxc C s t | �S )N)rs )r3 r r r r � r r c C s t | �S )N)r� )r3 r r r r � r c C s t j| dd�S )Nr� )r* )r r� )r3 r r r r � r rs r� r� )Zidsc s� |}|j � fdd�dd�}t� fdd�|�� D ��}t�||� |j � fdd�ddd �}|d d g jd dd �d�}t�||� d S )Nc s � ddg�S )Nr r r )r3 )r r r r � r z,test_consistency_for_boxed.<locals>.<lambda>r )r c s g | ]}� d dg��qS )r r r )rp r� )r r r r� � s z.test_consistency_for_boxed.<locals>.<listcomp>c s � ddg�S )Nr r r )r3 )r r r r � r rH )r rI r r r )r r )rC )r! r r� r" r# r r4 )r rw r/ r, r- r )r r �test_consistency_for_boxed� s r c C s$ | dkr dnd}t jdd��� t �|�}t �|�}|�� }|jt j| d�}t�||� |jt jg| d�}|�� }| dkr�t� |j dgg�|_ nt� |j dgg�|_ t�||� |jt jt jg| d�}t ||g|d�}| dkr�t� |j ddgg�|_ nt� |j ddgg�|_ t�||� W d Q R X d S ) N> r r& r r r )r )r r �absolute) r r �absr rv r! r" r4 r Z from_productrC r& r )r r+ � other_axisZf_absZf_sqrtr- r, r r r �test_agg_transform s* r c C s� t td�dd��} | �ddg�}t ddgddgd�ddgddgd �}t�||� | �ddgd dgd��}t d d tjgd tjdgd�ddgddd gd �}t�|�|�|� d S )NrZ )r r r` ra r r r r )rC r& rS g @g g @g 9@)r rt r7 r" r4 r rB Z reindex_like)r/ r, r- r r r � test_demo& s " r c C sl dddgi} t | �}|�ddi�}tddi�}t�||� |d �ddi�}tddidd�}t�||� d S )Nr� r� r� �countr )r� )r r7 r r" r# )r8 r/ r, r- r r r �!test_agg_with_name_as_column_name9 s r c C s� t dddgdddgddd gtd dd �d ��} t dd gddgddgtd�tjgd �ddgd�}t�d �� | �ddg�}W d Q R X t�||� t�d �� | ddddg �ddg�}W d Q R X |ddddg � ddg�}t�||� d S )Nr r r; g �?g @g @r� r� �bazr: )r< )r r r r r[ g @� foobarbazz 2013-01-01r` rS )r& r r r r ) r r r r? ZNaTr" �assert_produces_warningr7 r4 Zreindex)Zmdfr- r, r r r �"test_agg_multiple_mixed_no_warningI s$ $r c C s$ | dkr dnd}|j | �� d d� �� \}}tj|j| d�|j| d�|j| d�gdd�}dddg|_| dkrt|j n|}|j dddg| d�}t � ||� |d|di}|j || d�}t |�|�| �� |�|�| �� g||gd �}t �||� |dg|dgi}|j || d�}t|t |�|�| �� gdgd �|t |�|�| �� gdgd �i�}| d k�r`|j n|}t � ||� |ddg|ddgi}|j || d�}tj|t |�|�| �� |�|�| �� gddgd �|t |�|�| �� |�|�| �� gddgd �idd�}| d k�r|j n|}t � ||� d S ) N> r r& r r r )r r$ ra rS )r& > rC r )Zaxes�uniqueZ sort_valuesr? �concatr$ ra rS rC rX r7 r" r4 r rD r# r )r r+ r Zname1Zname2r- r, rR r r r �test_agg_reducek sN $ r c C s t dddgdddgddd gtd dd �d ��} | �d �}tdddtd �g| jd�}t�||� | �d g�}t dddtd �ggd g| jd�}t�||� tj t ddd�� | �d�}W d Q R X tdddgdddgd�}t�||� | �dg�}t dddggdgdddgd�}t�||� d S )Nr r r; g �?g @g @r� r� r r: )r< )r r r r r` )r& )r& rC zSelect only validF)�matchZcheck_stacklevelrS r[ g @r r r r ) r r r7 r r rC r" r# r4 r � FutureWarning)r/ r, r- r r r �test_nuiscance_columns� s. r" c C s t d ddgdtjdgdd dgd��}t|| �dd i�}tddi�}t�||� t|| �dd i�}tddi�}t�||� t|| �d d g�}t|| �d d gd d gd d gd��}t ddd �ddd �ddd �d��}tj||d d � tj||d d � t|| �d �}|�� }t�||� d S )Nr r; g �?g @r� r� )r r r r r r� )r r� T)r� ) r r rB rV r r" r# r4 r )rd r/ r, r- Zresult1Zresult2r r r � test_non_callable_aggregates� s( " r# c C s� t d ddgdtjdgdd dgd��}t|| �d|d �}|d ksF|d kr\t|jd |jd �}nt|jd |jd �}t� ||� d S )Nr r; g �?g @r� r� )r r r r� )r r r& )r& r ) r r rB rV r r( rC r& r" r# )rd r r/ r, r- r r r �test_size_as_str s "r$ c C s� t dddgdtjdgdd dgd��} dd� }| �|�}tddgdgddggd d d gd �}t�||� | �|g�}|�d �j}t� ||� d S )Nr r; g �?r� r� )r r r c S s t | �� �� �S )N)rs Zdropnar )Z group_colr r r rR s z&test_agg_listlike_result.<locals>.funcr r r )r& rR ) r r rB r7 r r" r# Zto_framerX r4 )r/ rR r, r- r r r �test_agg_listlike_result s $ " r% zdf, func, expectedrP )r* �varZmedianr g �?r; g �?g���۞��?g �?c C s | j ||d�}t�||� d S )N)r )r7 r" r# )r/ rR r- r r, r r r �test_agg_cython_table$ s 'r' ZcumprodZcumsumc C s8 |dks|dkr|� d�}| j||d�}t�||� d S )NrC r rP )r )r r7 r" r4 )r/ rR r- r r, r r r � test_agg_cython_table_transformO s r( z args, kwargs)r r r; )r] r\ � )r r )r rK r )r6 rK )r6 rK rL c C sd d dd�}t ddgddgg�}| dkr4tdd g�}n td d g�}|j|| f|�|�}t�||� d S ) Nr; c S s | � � || | S )N)rS )r3 r6 rK rL r r r r� x s z test_agg_args_kwargs.<locals>.fr r r r g @g @g @g @)r; )r r r7 r" r# )r rl �kwargsr� r/ r- r, r r r �test_agg_args_kwargsj s r+ �num_colsrZ c C sL t �dddg�}|�� }td|t| �d�}|�dd� � |j|jksHt�d S )Nz 1950-06-30z 1952-10-24z 1953-05-29r )r& rC c S s | S )Nr )r3 r r r r � r z,test_frequency_is_original.<locals>.<lambda>)r? Z DatetimeIndexrv r rt r! r� r% )r, r& �originalr/ r r r �test_frequency_is_original� s r. c C sb t ddd�t ddd�t ddd�g} tdddg| d �}|jd d � dd �}t| | d �}t�||� d S )Nz 2019-03-15 12:34:31.909000+0000r1 )r2 z 2019-03-15 12:34:34.359000+0000z 2019-03-15 12:34:34.660000+0000r r r )r8 r& c S s | j S )N)r� )r3 r r r r � r z.test_apply_datetime_tz_issue.<locals>.<lambda>)r )r& r8 )r r r! r r" r# )Z timestampsr/ r, r- r r r � test_apply_datetime_tz_issue� s r/ r/ r6 rL r. )r r �methodc C sR t jtdd��. t| ddg |�� }t| ddg |�� }W d Q R X t �||� d S )NzSelect only valid)r r r )r" r r! rV r# )r/ r0 Znone_in_first_column_resultZ none_in_second_column_resultr r r �=test_consistency_of_aggregates_of_columns_with_missing_values� s r1 r� c C s: t d| ggddgd�}|�dd� �}|j}t�||� d S )Ng �?r6 rK )rC c S s | j S )N)r* )r3 r r r r � r z"test_apply_dtype.<locals>.<lambda>)r r! r� r" r# )r� r/ r, r- r r r �test_apply_dtype� s r2 c C s| t td�tdd�d��}|�� }dd� }|�� }|d d7 < |j|dd�}t�||� | slt�||� n t�||� d S ) N�d �� )r6 rK c S s* | j }| jd d7 < | j |k s&t�| S )Nr6 r )Z_mgrrD r% )r� Zmgrr r r rR � s z!test_apply_mutating.<locals>.funcr6 r )r )r rt rv r! r" r4 )Zusing_array_managerr/ Zdf_origrR r- r, r r r �test_apply_mutating� s r5 c C sb t ddgddgddgddgd d ggd d gd �} | jdd� dd�}tg g d�td�}t�||� d S )Nr r r; r rZ r[ r\ r] r^ r� r6 rK )rC c S s g S )Nr )r3 r r r r � r z.test_apply_empty_list_reduce.<locals>.<lambda>rJ )rI )r6 rK )r* )r r! r rO r" r# )r/ r, r- r r r � test_apply_empty_list_reduce� s .r6 c C sf t ddggd ddgd�} | �ddd � dd � g�}t d d d gd d d gd �dd d gd�}t�||� d S )Nr r^ r; r r )rC rS c S s | � � S )N)rS )r3 r r r r � r z,test_apply_no_suffix_index.<locals>.<lambda>r� � )r r z<lambda>)r& )r r! r" r4 )Zpdfr, r- r r r �test_apply_no_suffix_index� s "r8 c C s@ t dddgi�} | jdd� ddd�}tddg�}t�||� d S ) Nr Zaa�bbbc S s | d S )Nr r )r3 r r r r � r z/test_apply_raw_returns_string.<locals>.<lambda>r T)r r| )r r! r r" r# )r/ r, r- r r r � test_apply_raw_returns_string� s r: r �ceil�cos�exp�logr ZsquareZ transformc C s, t | |�|�}t t|�| �}t�||� d S )N)rV r r" r4 )r+ r� rd r, r- r r r �test_apply_np_transformer� s r? c C sl t ddgddgd��} t| |�|�}|dkr4ddini }ttt|�| fdd i|��| jd �}t�||� d S ) Nr r r; r )r6 rK )rb r& Zddofr r )r& )r rV r r rC r" r# )r+ r� rd r, r* r- r r r �test_apply_np_reducer� s "r@ c C s� t ddddddgdd�} d d � }d |d d g}| �|�}t dtjddgddddgddddgddddgd�d dd d gd�}t�||� d S ) N)�1r r r )�2r r r )�3r; r r )�4r rZ r )�5rZ r[ r[ )�6r[ r\ r\ )�itemZatt1Zatt2Zatt3)rC c S s | � � d S )Nr )rS )r= r r r r� s z/test_aggregation_func_column_order.<locals>.foorS r r` Z123456r[ rA g 5@g %@g @g �?g 2@g "@g g 1@g !@r� )r& )r r7 r rB r" r4 )r/ r� Zaggsr, r- r r r �"test_aggregation_func_column_order s$ rH c C sR t dddgdddgd��} | ddg jdd� dd �}tddd g�}t�||� d S ) Nr r r r; )r6 rK r6 c S s | d | d S )Nr r r )r3 r r r r & r z+test_apply_getitem_axis_1.<locals>.<lambda>)r r )r r! r r" r# )r/ r, r- r r r �test_apply_getitem_axis_1# s rI )fr � itertoolsr r� �numpyr rh Zpandas.core.dtypes.dtypesr Zpandasr? r r r r r Zpandas._testingZ_testingr" Zpandas.tests.frame.commonr r0 r5 ri Z parametrizer9 r@ rG rQ rW rY r_ �paramrn rx 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� r� r� r� r r r r r r r r r r r r r" r# r$ r% Zget_cython_table_paramsr� rB r' r( r+ r. r/ r1 r2 r5 r6 r8 r: r? r@ rH rI r r r r �<module> s * 5" #41 /(&"E&. * "
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка