U
    `»{j´À  ã                   @   s¢  d dl m Z  d dlZd dlZd dlZd dlmZ d dlZd dlm	Z	m
Z
mZmZmZ d dlmZ d dlmZ dd„ Zdd	„ Zd
d„ Zej dddedgƒfeddd�dfg¡dd„ ƒZdd„ Zej dejejg¡dd„ ƒZdd„ Zdd„ Zdd„ Zej ddd d!d"g¡d#d$„ ƒZ d%d&„ Z!d'd(„ Z"d)d*„ Z#d+d,„ Z$d-d.„ Z%d/d0„ Z&d1d2„ Z'd3d4„ Z(ej d5d dg¡d6d7„ ƒZ)ej d5d dg¡d8d9„ ƒZ*d:d;„ Z+ej d5d dg¡d<d=„ ƒZ,d>d?„ Z-d@dA„ Z.dBdC„ Z/ej dDdEdFg¡ejjddGdH„ dIdH„ gdJdKgdL�ej dMdNdOg¡ej d5d dg¡dPdQ„ ƒƒƒƒZ0dRdS„ Z1dTdU„ Z2dVdW„ Z3dXdY„ Z4dZd[„ Z5d\d]„ Z6d^d_„ Z7d`da„ Z8dbdc„ Z9ddde„ Z:dfdg„ Z;dhdi„ Z<djdk„ Z=dldm„ Z>dndo„ Z?ej dpe	dqdrgdsdtggƒedqdsduœdrdtduœgƒge	d dgdvdwggƒed dvduœddwduœgƒgg¡dxdy„ ƒZ@dzd{„ ZAej d|dd}g¡d~d„ ƒZBd€d�„ ZCej d‚dƒed„ƒgd…e Dd†¡gg¡d‡dˆ„ ƒZEej d‰e	ƒ e	eFdŠƒd‹�e	eFdŠƒdŒ�e	g g g d�œƒg¡ej deGdŽdH„ g¡d�d�„ ƒƒZHd‘d’„ ZId“d”„ ZJd•d–„ ZKd—d˜„ ZLd™dš„ ZMd›dœ„ ZNd�dž„ ZOdŸd „ ZPej d¡d¢d£g¡d¤d¥„ ƒZQd¦d§„ ZRd¨d©„ ZSdªd«„ ZTd¬d­„ ZUd®d¯„ ZVej d|d°d±dejWg¡d²d³„ ƒZXd´dµ„ ZYd¶d·„ ZZd¸d¹„ Z[dºd»„ Z\d¼d½„ Z]d¾d¿„ Z^dÀdÁ„ Z_dÂdÃ„ Z`ej d|ddvg¡dÄdÅ„ ƒZadÆdÇ„ Zbej dÈddvdwgddvgg¡dÉdÊ„ ƒZcdËdÌ„ ZddÍdÎ„ ZedÏdÐ„ ZfdÑdÒ„ ZgdÓdÔ„ ZhdÕdÖ„ Zid×dØ„ ZjejjdÙdÚdH„ dÛdH„ dÜdH„ gdÝdÞdßgdL�dàdá„ ƒZkdâdã„ Zldädå„ Zmdædç„ Zndèdé„ Zodêdë„ Zpdìdí„ Zqdîdï„ Zrej dðd£d¢g¡dñdò„ ƒZsej dðd£d¢g¡dódô„ ƒZtdõdö„ Zuej d5d dg¡ej d÷døi fdùi fdúi fdûdüdvifdýddvdþœfdýdvddþœfdýddvdwdÿœfg¡�d �d„ ƒƒZvej �ddvdw�dg¡�d�d„ ƒZw�d�d„ Zxej �de	�d	dg�d
�dg�dœƒg¡ej �d�d�ddg¡�d�d„ ƒƒZyej �ddd}dN�d	ejzg¡�d�d„ ƒZ{�d�d„ Z|�d�d„ Z}�d�d„ Z~�d�d„ Z�d�d„ Z€�d�d „ Z��d!�d"„ Z‚�d#�d$„ Zƒ�d%�d&„ Z„ej �d'ddvdwgdddgdþœddvdwgdgdþœfdvdvdvgdddgdþœdvgdgdþœfg¡�d(�d)„ ƒZ…dS (*  é    ©ÚdatetimeN)ÚCategoricalDtype)Ú	DataFrameÚ
MultiIndexÚSeriesÚ	TimestampÚ
date_range)Ú
zip_framesc              	   C   sº   t jdd��¤ t  | d ¡}|  t j¡d }t ||¡ |  t j¡d }t  | d ¡}||ksbt‚| jd }| jt jdd�}t  |  	|¡¡}|| |ksœt‚|j| jks¬t‚W 5 Q R X d S )NÚignore©ÚallÚAr   é   ©Úaxis)
ÚnpÚerrstateÚsqrtÚapplyÚtmÚassert_series_equalÚmeanÚAssertionErrorÚindexÚxs)Úfloat_frameÚresultÚexpectedÚd© r    úb/var/www/html/bwave/python/venv/lib/python3.8/site-packages/pandas/tests/apply/test_frame_apply.pyÚ
test_apply   s    
r"   c                  C   sf   t ddddgddddgdœƒ} |  dd„ ¡}|jdks:t‚t|d	 jtƒsNt‚t|d
 jtƒsbt‚d S )Nr   ÚBÚCÚD©Úc0Úc1c                 S   s
   |   d¡S )NÚcategory©Zastype)Útsr    r    r!   Ú<lambda>+   ó    z-test_apply_categorical_func.<locals>.<lambda>)é   é   r'   r(   )r   r   Úshaper   Ú
isinstanceÚdtyper   ©Údfr   r    r    r!   Útest_apply_categorical_func(   s
    r5   c                  C   s8   t dtddd�giƒ} | jdd„ dd�}t || ¡ d S )	Nr   ú
2013-01-01ÚUTC©Útzc                 S   s   | S ©Nr    ©Úxr    r    r!   r,   5   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_ea2   s    r@   zdata, dtype©r   Nr   r6   r7   r8   c                 C   s<   t | g| ggddg|d�}|jdd„ dd�}t ||¡ d S )NÚa©r   r2   c                 S   s   | S r:   r    r;   r    r    r!   r,   @   r-   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   ©r   Úaggr   r>   )Údatar2   r   r   r    r    r!   Útest_agg_axis1_duplicate_index9   s    rG   c                  C   sH   t tddd�tjt d¡dd�dœƒ} | jdd„ d	d
�}t || ¡ d S )NÚ20130101é   ©ÚperiodsÚs)Úunit©r   r#   c                 S   s   | S r:   r    r;   r    r    r!   r,   M   r-   z/test_apply_mixed_datetimelike.<locals>.<lambda>r   r   )	r   r	   ÚpdZto_timedeltar   Zaranger   r   r>   r?   r    r    r!   Útest_apply_mixed_datetimelikeD   s    
þÿrP   Úfuncc                 C   s   t ƒ }| | ¡}|jst‚d S r:   )r   r   Úemptyr   )rQ   Úempty_framer   r    r    r!   Útest_apply_emptyQ   s    
rT   c                 C   sz   | d d… }|  dd„ ¡}ttj| jd�}t ||¡ | jd d …g f }|j dd„ dd�}ttj| jd�}t ||¡ d S )Nr   c                 S   s   |   ¡ S r:   ©r   r;   r    r    r!   r,   \   r-   z(test_apply_float_frame.<locals>.<lambda>©r   c                 S   s   |   ¡ S r:   rU   r;   r    r    r!   r,   a   r-   r   r   )	r   r   r   ÚnanÚcolumnsr   r   Úlocr   )r   Zno_rowsr   r   Úno_colsr    r    r!   Útest_apply_float_frameZ   s    r[   c                  C   s.   t dgd�} | jdd„ dd�}t || ¡ d S )NrB   rV   c                 S   s   | d S )NrB   r    r;   r    r    r!   r,   i   r-   z/test_apply_empty_except_index.<locals>.<lambda>r   r   ©r   r   r   r>   r?   r    r    r!   Útest_apply_empty_except_indexf   s    r]   c                  C   sÖ   t ƒ } g }| j|jddd�}t || ¡ | j|jddd�}tg tjg td�t	j
d�}t ||¡ t ddd	gd
�}|j|jddd�}t ||¡ |j|jddd�}tg tjg td�t	j
d�}t ||¡ |g ksÒt‚d S )Nr   Úexpand©r   Úresult_typeÚreduce©r2   rC   rB   ÚbÚc©rX   )r   r   Úappendr   r>   r   rO   ÚIndexÚobjectr   Úfloat64r   r   )rS   r<   r   r   Zempty_with_colsr    r    r!   Útest_apply_with_reduce_emptym   s    rj   ÚsumÚprodÚanyr   c                 C   s<   t dddgd�}| tt| ƒ¡}t|| ƒƒ }t ||¡ d S )NrB   rc   rd   re   )r   r   Úgetattrr   r   r   )rQ   r4   r   r   r    r    r!   Útest_apply_funcs_over_emptyƒ   s    ro   c                  C   sb   t dddgd�} |  ¡ }td| jd�}t ||¡ | j ¡ }tg t g ¡t	j
d�}t ||¡ d S )NrB   rc   rd   re   r   rV   rC   )r   Znuniquer   rX   r   r   ÚTrO   rg   r   ri   ©r4   r   r   r    r    r!   Útest_nunique_empty�   s    
rr   c                  C   s‚   t dddgdddgddd	ggd
d
dgd�} | jdd„ dd�}tdddgd
d
dgƒ}t ||¡ | jjdd„ dd�}t ||¡ d S )Nr   r/   rI   r.   é   é   é   é   é	   rB   rd   rV   c                 S   s   | d S ©Nr   r    ©rL   r    r    r!   r,   �   r-   z/test_apply_standard_nonunique.<locals>.<lambda>r   c                 S   s   | d S rx   r    ry   r    r    r!   r,   ¡   r-   r   )r   r   r   r   r   rp   rq   r    r    r!   Útest_apply_standard_nonuniqueš   s    *rz   c                 C   s4   | j tjdd�}t|  ¡ g| jd�}t ||¡ d S )NÚ	broadcast©r`   rV   )r   r   r   r   r   r   r>   ©r   r   r   r    r    r!   Útest_apply_broadcast_scalars¥   s    r~   c                    sF   | j tjddd�}| jdd�‰ t‡ fdd„| jD ƒƒ}t ||¡ d S )Nr   r{   r_   r   c                    s   i | ]
}|ˆ “qS r    r    ©Ú.0rd   ©Úmr    r!   Ú
<dictcomp>¯   s      z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>)r   r   r   r   rX   r   r>   r}   r    r�   r!   Ú"test_apply_broadcast_scalars_axis1¬   s    r„   c                    sZ   ˆ j ‡ fdd„ddd�}tttˆ jƒƒƒ}t|gtˆ jƒ dˆ jˆ jd�}t ||¡ d S )Nc                    s   t ttˆ jƒƒƒS r:   )ÚlistÚrangeÚlenrX   r;   ©r   r    r!   r,   ¶   r-   z4test_apply_broadcast_lists_columns.<locals>.<lambda>r   r{   r_   ri   )r2   r   rX   )	r   r…   r†   r‡   rX   r   r   r   r>   )r   r   r‚   r   r    rˆ   r!   Ú"test_apply_broadcast_lists_columns³   s    
ýür‰   c                    sX   ˆ j ‡ fdd„dd�}tttˆ jƒƒƒ‰t‡fdd„ˆ jD ƒdˆ jd�}t ||¡ d S )	Nc                    s   t ttˆ jƒƒƒS r:   )r…   r†   r‡   r   r;   rˆ   r    r!   r,   Æ   r-   z2test_apply_broadcast_lists_index.<locals>.<lambda>r{   r|   c                    s   i | ]
}|ˆ “qS r    r    r   r�   r    r!   rƒ   Ê   s      z4test_apply_broadcast_lists_index.<locals>.<dictcomp>ri   )r2   r   )	r   r…   r†   r‡   r   r   rX   r   r>   r}   r    )r   r‚   r!   Ú test_apply_broadcast_lists_indexÄ   s    
 ÿýrŠ   c                 C   s(   | }|j dd„ ddd�}t ||¡ d S )Nc                 S   s
   dddgS ©Nr   r/   rI   r    r;   r    r    r!   r,   Ô   r-   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r{   r_   ©r   r   r>   )Úint_frame_const_colr4   r   r    r    r!   Ú%test_apply_broadcast_list_lambda_funcÑ   s    rŽ   c                 C   s0   | }|j dd„ ddd�}| ¡ }t ||¡ d S )Nc                 S   s   t dddgtdƒd�S )Nr   r/   rI   ÚabcrV   )r   r…   r;   r    r    r!   r,   Û   r-   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r{   r_   ©r   Úcopyr   r>   ©r�   r4   r   r   r    r    r!   Ú'test_apply_broadcast_series_lambda_funcØ   s    ýr“   r   c                 C   s   dd„ }| j ||dd� d S )Nc                 S   s"   t | tjƒst‚| jdkst‚d S ©Nr   ©r1   r   Úndarrayr   Úndimr;   r    r    r!   Ú_assert_rawå   s    z/test_apply_raw_float_frame.<locals>._assert_rawT©r   Úraw©r   )r   r   r˜   r    r    r!   Útest_apply_raw_float_frameã   s    rœ   c                 C   s4   | j tj|dd�}| j dd„ |d�}t ||¡ d S )NTr™   c                 S   s
   | j  ¡ S r:   )Úvaluesr   r;   r    r    r!   r,   ï   r-   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>r   )r   r   r   r   r   )r   r   r   r   r    r    r!   Ú!test_apply_raw_float_frame_lambdaì   s    rž   c                 C   s*   | j dd„ dd�}| d }t ||¡ d S )Nc                 S   s   | d S ©Nr/   r    r;   r    r    r!   r,   õ   r-   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>T)rš   r/   rŒ   r}   r    r    r!   Ú'test_apply_raw_float_frame_no_reductionó   s    r    c                 C   s   dd„ }| j ||dd� d S )Nc                 S   s"   t | tjƒst‚| jdkst‚d S r”   r•   r;   r    r    r!   r˜   ü   s    z4test_apply_raw_mixed_type_frame.<locals>._assert_rawTr™   r›   )Zmixed_type_framer   r˜   r    r    r!   Útest_apply_raw_mixed_type_frameú   s    r¡   c                 C   s>   | j d }| jtjdd�| }t |  |¡¡}||ks: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   sR   t dgdgdœƒ} | d d… jtjdd�}ttjtjg dd�d	�}t 	||¡ d S )
NÚfooç      ð?rN   r   r   r   Úint64rb   rV   )
r   r   r   r   r   rW   rO   rg   r   r   rq   r    r    r!   Útest_apply_mixed_dtype_corner  s    r¦   c                  C   sr   t dgdgdœƒ} | jdd„ dd�}tdgdgd	�}t ||¡ | jd
d„ dd�}tdgdgd	�}t ||¡ d S )Nr£   r¤   rN   c                 S   s   | d S )Nr   r    r;   r    r    r!   r,     r-   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   rV   c                 S   s   | d S )Nr#   r    r;   r    r    r!   r,     r-   ©r   r   r   r   r   rq   r    r    r!   Ú&test_apply_mixed_dtype_corner_indexing  s    r¨   Úaxr   rX   c                 C   s   | S r:   r    r;   r    r    r!   r,   !  r-   r,   c                 C   s   |   ¡ S r:   rU   r;   r    r    r!   r,   !  r-   Úidentityr   )Zidsrš   TFc           	   
   C   s¾   t f | dddgiŽ}tjdd��” tjdd��$ t dt¡ |tjg dd	�ƒ}W 5 Q R X t|tj	ƒ }|j
|||d
�}|r¢| |¡}t|tƒs’t‚|j|ks°t‚nt|t ƒs°t‚W 5 Q R X d S )NrB   rc   rd   r   r   T)ÚrecordZf8rb   r™   )r   r   r   ÚwarningsÚcatch_warningsÚsimplefilterÚRuntimeWarningÚarrayr1   r–   r   Z_get_agg_axisr   r   r   )	r©   rQ   rš   r   r4   Ztest_resZis_reductionr   Zagg_axisr    r    r!   Útest_apply_empty_infer_type  s    
r±   c                  C   s4   t dddgd�} | jdd„ dd�}t|t ƒs0t‚d S )	NrB   rc   rd   rV   c                 S   s   |   ¡ S r:   rU   r;   r    r    r!   r,   9  r-   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r{   r|   )r   r   r1   r   )rZ   r   r    r    r!   Ú%test_apply_empty_infer_type_broadcast7  s    r²   c                 C   s6   ddd„}| j |dd�}|   dd„ ¡}t ||¡ d S )	Nr   c                 S   s   | | S r:   r    ©r<   Úhowmuchr    r    r!   Úadd_some>  s    z4test_apply_with_args_kwds_add_some.<locals>.add_somer/   ©r´   c                 S   s   | d S rŸ   r    r;   r    r    r!   r,   B  r-   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>)r   rŒ   )r   rµ   r   r   r    r    r!   Ú"test_apply_with_args_kwds_add_some=  s    
r·   c                 C   s6   ddd„}| j |dd�}|   dd„ ¡}t ||¡ d S )	Nr   c                 S   s   |   ¡ | S r:   rU   r³   r    r    r!   Úagg_and_addG  s    z:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addr/   r¶   c                 S   s   |   ¡ d S rŸ   rU   r;   r    r    r!   r,   K  r-   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>)r   )r   r   r   )r   r¸   r   r   r    r    r!   Ú%test_apply_with_args_kwds_agg_and_addF  s    
r¹   c                 C   s8   d	dd„}| j |ddd�}|   dd„ ¡}t ||¡ d S )
Nr   c                 S   s   | | | S r:   r    )r<   ÚsubÚdivider    r    r!   Úsubtract_and_divideP  s    zJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide)r/   r/   )Úargsr»   c                 S   s   | d d S )Nç       @r    r;   r    r    r!   r,   T  r-   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>)r   rŒ   )r   r¼   r   r   r    r    r!   Ú-test_apply_with_args_kwds_subtract_and_divideO  s    
r¿   c                 C   s   |   t¡}t || ¡ d S r:   )r   r…   r   r>   )r   r   r    r    r!   Útest_apply_yield_listX  s    
rÀ   c                 C   s@   t j| d jd d d…< |  d¡}| jt jdd�}t ||¡ d S )Nr   r/   r   r   )r   rW   Zilocr   r   r   r   )r   r   r   r    r    r!   Útest_apply_reduce_Series]  s    
rÁ   c                  C   s–   t ddgddggddgddgd	�} | jtd
d�}tdddœdddœg| jd�}t ||¡ | jtdd�}tdddœdddœg| jd�}t ||¡ d S )Nr   r/   rI   r.   r'   r(   Úi0Úi1©rX   r   r   r   )rÂ   rÃ   rV   r&   )r   r   Údictr   rX   r   r   r   )rF   r   r   r    r    r!   Útest_apply_reduce_to_dictd  s    "rÆ   c                  C   sŠ   t tj dd¡ƒ} | jtjdd�}t dd„ |  ¡ D ƒ| jd�}t	 
||¡ | jtjdd�}t d	d„ | j ¡ D ƒ| jd�j}t	 
||¡ d S )
Né   é
   r   r   c                 S   s   i | ]\}}||  ¡ “qS r    ©Údescribe©r€   ÚiÚvr    r    r!   rƒ   u  s      z2test_apply_differently_indexed.<locals>.<dictcomp>re   r   c                 S   s   i | ]\}}||  ¡ “qS r    rÉ   rË   r    r    r!   rƒ   y  s      )r   r   ÚrandomÚrandnr   r   rÊ   ÚitemsrX   r   r>   rp   r   rq   r    r    r!   Útest_apply_differently_indexedq  s     rÑ   c               
   C   sè   t dddgdddgdddgdddgdd	dgdd
dggdddgd�} dd„ }| j|dd�}t tdddƒddgtdddƒddgtdddƒddgtdddƒddgtdddƒd	dgtdddƒd
dggdddgd�} | j|dd�}t ||¡ d S )Nr   ZABC0é2   ZYUM0rÇ   ZDEF0r/   ZABC1ZYUM1ZDEF1rB   ÚmarketÚpositionre   c                 S   s   | d S )NrÓ   r    )Úrr    r    r!   ÚfŒ  s    ztest_apply_bug.<locals>.fr   iÝ  rI   r.   rs   rt   )r   r   r   r   r   )Z	positionsrÖ   r   r   r    r    r!   Útest_apply_bug}  s0    ú÷ú÷r×   c                  C   sš   t dddddddddddgdddddddddddgdddddddddddgtj d¡tj d¡tj d¡dœƒ} | jd	d
„ dd�jdd�}t || ¡ d S )Nr£   ÚbarÚoneÚtwoZdullZshinyé   )r   r#   r$   r%   ÚEÚFc                 S   s   | S r:   r    r;   r    r    r!   r,   Ð  r-   z,test_apply_convert_objects.<locals>.<lambda>r   r   Tr   )r   r   rÎ   rÏ   r   Ú_convertr   r>   r?   r    r    r!   Útest_apply_convert_objects   sX    õõõ


Öÿ/rß   c                 C   s.   |   dd„ ¡}t| j| jd�}t ||¡ d S )Nc                 S   s   | j S r:   ©Únamer;   r    r    r!   r,   Õ  r-   z(test_apply_attach_name.<locals>.<lambda>rV   )r   r   rX   r   r   r}   r    r    r!   Útest_apply_attach_nameÔ  s    râ   c                 C   s2   | j dd„ dd�}t| j| jd�}t ||¡ d S )Nc                 S   s   | j S r:   rà   r;   r    r    r!   r,   Û  r-   z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   rV   )r   r   r   r   r   r}   r    r    r!   Útest_apply_attach_name_axis1Ú  s    rã   c                 C   sD   |   dd„ ¡}tt | jt| jƒdf¡| j| jd�}t ||¡ d S )Nc                 S   s   t  | jt| ƒ¡S r:   ©r   Úrepeatrá   r‡   r;   r    r    r!   r,   â  r-   z6test_apply_attach_name_non_reduction.<locals>.<lambda>r   ©r   rX   )	r   r   r   ZtilerX   r‡   r   r   r>   r}   r    r    r!   Ú$test_apply_attach_name_non_reductionà  s    ýrç   c                    sD   ˆ j dd„ dd�}t‡ fdd„ˆ  ¡ D ƒƒ}ˆ j|_t ||¡ d S )Nc                 S   s   t  | jt| ƒ¡S r:   rä   r;   r    r    r!   r,   ì  r-   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>r   r   c                 3   s$   | ]}t  |d  tˆ jƒ¡V  qdS )r   N)r   rå   r‡   rX   ©r€   Útrˆ   r    r!   Ú	<genexpr>í  s    z=test_apply_attach_name_non_reduction_axis1.<locals>.<genexpr>)r   r   Ú
itertuplesr   r   r   r}   r    rˆ   r!   Ú*test_apply_attach_name_non_reduction_axis1ë  s    ÿrì   c                  C   s†   t  dddgdddgg¡} tddgddgd	d
gg| ddgd�}| dd„ d¡}tddgddgd	d
gg| ddgd�}tj||dd� d S )NrB   rc   rd   r   r   r/   rI   r.   rs   rt   Úcol1Úcol2ræ   c                 S   s   t t| ƒt| ƒdœƒS )N)ÚminÚmax)r   rï   rð   r;   r    r    r!   r,   ÷  r-   z(test_apply_multi_index.<locals>.<lambda>rï   rð   T©Z
check_like)r   Zfrom_arraysr   r   r   r>   )r   rL   r   r   r    r    r!   Útest_apply_multi_indexô  s
    $$rò   z	df, dictsr£   rØ   ZspamÚeggs)r   r   r/   rI   c                 C   sV   dd„ }| j |dd�}| j |dd�}|   |¡}t ||¡ t || ¡ t ||¡ d S )Nc                 S   s   |   ¡ S r:   )Úto_dictr;   r    r    r!   r,     r-   z!test_apply_dict.<locals>.<lambda>ra   r|   r^   )r   r   r   r>   )r4   ZdictsÚfnZreduce_trueZreduce_falseZreduce_noner    r    r!   Útest_apply_dictü  s    
rö   c                 C   sP   |   dd„ ¡}t || d ¡ |   t¡ |   dd„ ¡d d }t|tƒsLt‚d S )Nc                 S   s   | d S rŸ   r    r;   r    r    r!   r,     r-   ztest_applymap.<locals>.<lambda>r/   c                 S   s   | | fS r:   r    r;   r    r    r!   r,     r-   r   r   )Úapplymapr   r>   Útyper1   Útupler   )r   Zappliedr   r    r    r!   Útest_applymap  s
    
rú   Úvalr¤   c                 C   s2   t | dgd�}| dd„ ¡jd }|tks.t‚d S )NrB   )rF   c                 S   s   | S r:   r    r;   r    r    r!   r,      r-   z7test_applymap_float_object_conversion.<locals>.<lambda>r   )r   r÷   Údtypesrh   r   )rû   r4   r   r    r    r!   Ú%test_applymap_float_object_conversion  s    rý   c                  C   sT   t tj d¡ƒ} |  ¡ }ddddg}|| _| t¡}||_|  t¡}t ||¡ d S )N©rI   r.   rB   )	r   r   rÎ   r‘   rX   r÷   Ústrr   r>   )r4   Zdf2Úcolsr   r   r    r    r!   Útest_applymap_str$  s    

r  zcol, valr   rH   Ú	timedeltaz1 minc                 C   sJ   t tj d¡ƒ}||| < | t¡}|jd| f t|jd| f ƒksFt‚d S )Nrþ   r   )r   r   rÎ   r÷   rÿ   rY   r   )Úcolrû   r4   r   r    r    r!   Útest_applymap_datetimelike1  s    
r  r   ÚABCre   rV   ©r   r#   r$   c                 C   s   | S r:   r    r;   r    r    r!   r,   F  r-   c                 C   s   |   |¡}t || ¡ d S r:   )r÷   r   r>   )r   rQ   r   r    r    r!   Útest_applymap_empty=  s    
r  c                  C   sF   t ddgddggƒjdd„ dd�} t ddgdd	ggƒ}t | |¡ d S )
Nr   r/   rI   r.   c                 S   s   | | S r:   r    ©r<   Úyr    r    r!   r,   O  r-   z&test_applymap_kwargs.<locals>.<lambda>©r	  rs   rt   )r   r÷   r   r>   )r   r   r    r    r!   Útest_applymap_kwargsM  s    "r  c                 C   sj   |   dd„ ¡}|  ¡ }tjjdd| jtd�}tj||< |j dd„ dd�}| ¡ }tj||< t	 
||¡ d S )	Nc                 S   s   t t| ƒƒS r:   ©r‡   rÿ   r;   r    r    r!   r,   V  r-   z)test_applymap_na_ignore.<locals>.<lambda>r   r/   )Úsizer2   c                 S   s   t t| ƒƒS r:   r  r;   r    r    r!   r,   [  r-   r   )Z	na_action)r÷   r‘   r   rÎ   Úrandintr0   ÚboolrO   ZNAr   r>   )r   Zstrlen_frameZfloat_frame_with_naÚmaskZstrlen_frame_na_ignoreZstrlen_frame_with_nar    r    r!   Útest_applymap_na_ignoreT  s    
 ÿ
r  c                  C   s*   t tddd�ƒ} dd„ }t| ƒ |¡ d S )Nz1/1/2000rÈ   rJ   c                 S   s   | j | j| jfS r:   )ÚhourÚdayÚmonthr;   r    r    r!   rQ   f  s    z*test_applymap_box_timestamps.<locals>.func)r   r	   r   r÷   )ZserrQ   r    r    r!   Útest_applymap_box_timestampsb  s    r  c               	   C   sš   t tdƒtdƒgtddd�tddd�gt d¡t d¡gtjddd�tjddd�gd	œƒ} |  d
d„ ¡}t ddgddgddgddgd	œƒ}t ||¡ d S )Nz
2011-01-01z
2011-01-02z
US/Easternr8   z1 daysz2 daysÚM)Úfreq©rB   rc   rd   r   c                 S   s
   t | ƒjS r:   )rø   Ú__name__r;   r    r    r!   r,   ~  r-   z#test_applymap_box.<locals>.<lambda>r   Ú	TimedeltaÚPeriod)r   r   rO   r  r  r÷   r   r>   rq   r    r    r!   Útest_applymap_boxm  s(    

þþùÿüÿr  c                     s^   ddl m‰  tdtdddƒgiƒ} |  ‡ fdd„¡} |  ‡ fdd„¡} | jj}|d	ksZt‚d S )
Nr   ©ÚBDayÚx1iÌ  r   c                    s
   | ˆ ƒ  S r:   r    r;   r  r    r!   r,   �  r-   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>c                    s
   | ˆ ƒ  S r:   r    r;   r  r    r!   r,   �  r-   zM8[ns])Zpandas.tseries.offsetsr  r   r   r÷   r  r2   r   r3   r    r  r!   Ú(test_frame_apply_dont_convert_datetime64Š  s    r   c                  C   sd   t dtdddd�iƒ} |  dd„ ¡}t || ¡ |  dd„ ¡}t dtd	ddd�iƒ}t ||¡ d S )
NÚdtz
2015-01-01rI   zEurope/Brussels)rK   r9   c                 S   s   | S r:   r    r;   r    r    r!   r,   ™  r-   z,test_apply_non_numpy_dtype.<locals>.<lambda>c                 S   s   | t  d¡ S )NZ1day)rO   r  r;   r    r    r!   r,   œ  r-   z
2015-01-02)r   r	   r   r   r>   rq   r    r    r!   Útest_apply_non_numpy_dtype–  s    ÿr"  c                  C   s6   t dddddgidd�} |  dd„ ¡}t || ¡ d S )	Nr!  rB   rc   rd   r)   rb   c                 S   s   | S r:   r    r;   r    r    r!   r,   ¥  r-   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>r\   r3   r    r    r!   Ú#test_apply_non_numpy_dtype_category£  s    r#  c                  C   sR   t ddgddggddgd�} t ddggddgdgd�}|  dg¡}t ||¡ d S )	Nr   r   r/   rI   rB   re   rï   rÄ   rD   ©r4   r   r   r    r    r!   Útest_apply_dup_names_multi_agg©  s    r%  Úopr   rE   c                 C   sh   dd„ }t t d¡tdƒd�}t|| ƒ|dd�}tdddgdddgdddgdddggƒ}t ||¡ d S )	Nc                 S   s"   d| d  d| d  d| d  gS )Nr/   r   r$   r#   r    ©Úrowr    r    r!   Ú
apply_listµ  s    z3test_apply_nested_result_axis_1.<locals>.apply_list)r.   r.   ZABCDre   r   r   ç        )r   r   Zzerosr…   rn   r   r   r   )r&  r)  r4   r   r   r    r    r!   Útest_apply_nested_result_axis_1²  s    "ÿr+  c                  C   sR   t dtddd�gidd�} |  dd„ ¡}t || ¡ |  d	d„ ¡}t || ¡ d S )
Nr£   Z2020r7   r8   zdatetime64[ns, UTC]rb   c                 S   s   | S r:   r    r;   r    r    r!   r,   Å  r-   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   s   |   ¡ S r:   )r‘   r;   r    r    r!   r,   Ç  r-   r=   r?   r    r    r!   Ú%test_apply_noreduction_tzaware_objectÀ  s     ÿr,  c                     sj   t ddddgiƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ].}ˆ d d …= | j|dd	� ˆ t| jƒks6t‚q6d S )
NrB   r   r/   rI   c                    s   ˆ   | j¡ d S r:   ©rf   rá   r'  ©Únamesr    r!   Úreducing_functionÑ  s    z8test_apply_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | j¡ | S r:   r-  r'  r.  r    r!   Únon_reducing_functionÔ  s    z<test_apply_function_runs_once.<locals>.non_reducing_functionr   )r   r   r…   r   r   ©r4   r0  r1  rQ   r    r.  r!   Útest_apply_function_runs_onceË  s    
r3  c                     sp   t ddddgiƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ]4}ˆ d d …= | j|d	dd
� ˆ t| j ¡ ƒks6t‚q6d S )NrB   r   r/   rI   c                    s   ˆ   | ¡ d S r:   ©Úextendr'  ©r�   r    r!   r0  å  s    z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | ¡ | S r:   r4  r'  r6  r    r!   r1  è  s    
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionT)rš   r   )r   r   r…   rB   Úto_listr   r2  r    r6  r!   Ú!test_apply_raw_function_runs_onceß  s    
r8  c                     sf   t ddddgiƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ]*}ˆ d d …= |  |¡ ˆ | j ¡ ks6t‚q6d S )	NrB   r   r/   rI   c                    s   ˆ   | ¡ d S r:   ©rf   ©rû   r6  r    r!   r0  ø  s    z;test_applymap_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | ¡ | S r:   r9  r:  r6  r    r!   r1  û  s    
z?test_applymap_function_runs_once.<locals>.non_reducing_function)r   r÷   rB   r7  r   r2  r    r6  r!   Ú test_applymap_function_runs_onceó  s    

r;  c                  C   sP   t t ddg¡dgd�} t t ddg¡dgtd�}|  dd„ ¡}t ||¡ d S )Ns   abcds   efghr  re   )rX   r2   c                 S   s
   |   d¡S )Nrh   r*   r;   r    r    r!   r,     r-   z-test_apply_with_byte_string.<locals>.<lambda>)r   r   r°   rh   r   r   r>   r$  r    r    r!   Útest_apply_with_byte_string  s    r<  Úasdé   c                    s`   dd dddt jg}td|idd�}|j ‡ fdd„¡}t‡ fd	d
„|D ƒdd�}t ||¡ d S )Nr=  r>  ZcderB   r)   rb   c                    s   | ˆ kS r:   r    r;   r:  r    r!   r,     r-   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t  |¡rtjn|ˆ k‘qS r    )rO   Zisnullr   ÚNaN)r€   r<   r:  r    r!   Ú
<listcomp>  s     z1test_apply_category_equalness.<locals>.<listcomp>rà   )r   r?  r   rB   r   r   r   r   )rû   Z	df_valuesr4   r   r   r    r:  r!   Útest_apply_category_equalness  s     ÿrA  c                  C   sV   t tj dd¡ƒ} | jtjjdd�j}|dks2t‚| jtjjdd�j}|dksRt‚d S )NrÈ   r/   r   r   )rÈ   r/   )rt   r/   )	r   r   rÎ   Zrandr   Zfftr0   r   Zrfftr3   r    r    r!   Útest_infer_row_shape"  s
    rB  c                  C   sÀ   t ddgddggddgd�} | jdd„ dd�}td	d
„ |  ¡ D ƒƒ}t ||¡ tdƒtdƒg| d< | jdd„ dd�}t ||¡ | d | d   dd„ ¡}tddiddigƒ}t ||¡ d S )Nr   r/   rB   rc   re   c                 S   s   d| d | d  iS ©NrL   rB   rc   r    r;   r    r    r!   r,   0  r-   z,test_with_dictlike_columns.<locals>.<lambda>r   c                 S   s   g | ]}d di‘qS )rL   rI   r    rè   r    r    r!   r@  1  s     z.test_with_dictlike_columns.<locals>.<listcomp>ú2017-05-01 00:00:00ú2017-05-02 00:00:00r   c                 S   s   d| d | d  iS rC  r    r;   r    r    r!   r,   8  r-   c                 S   s   d| iS )NrL   r    r;   r    r    r!   r,   <  r-   rL   rI   )r   r   r   rë   r   r   r   rq   r    r    r!   Útest_with_dictlike_columns-  s    þrF  c                  C   sf   t ƒ } dddg| d< dddg| d< t d	d
dg¡| d< | jdd„ dd�}ti i i gƒ}t ||¡ d S )NÚXÚYÚZÚauthorZBBCZNBCZN24Z	publisherz17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09Údatec                 S   s   i S r:   r    r;   r    r    r!   r,   I  r-   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rO   Úto_datetimer   r   r   r   rq   r    r    r!   Ú(test_with_dictlike_columns_with_datetimeA  s    ÿrM  c                  C   s„   t ddgddggddgd�} | jdd„ ddd	�}t d
ddgiƒ}t ||¡ tdƒtdƒg| d< | jdd„ ddd	�}t ||¡ d S )Nr   r/   rB   rc   re   c                 S   s   d| d | d  iS rC  r    r;   r    r    r!   r,   Q  r-   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>r^   r_   rL   rI   rD  rE  r   c                 S   s   d| d | d  iS rC  r    r;   r    r    r!   r,   Y  r-   )r   r   r   r>   r   rq   r    r    r!   Ú%test_with_dictlike_columns_with_inferN  s    þrN  c                  C   s²   t ttj d¡ƒddddgtdddd�d	œƒ} | dd
g jtdd�}tdd„ | dd
g  ¡ D ƒƒ}t	 
||¡ | ddg jtdd�}tdd„ | ddg  ¡ D ƒƒ}t	 
||¡ d S )Nr.   rB   r…   ZofÚwordsz
2016-10-01ÚH)rK   r  )rB   rc   r+   rc   r   r   c                 S   s   g | ]}|d d… ‘qS rA   r    rè   r    r    r!   r@  h  s     z.test_with_listlike_columns.<locals>.<listcomp>r+   c                 S   s   g | ]}|d d… ‘qS rA   r    rè   r    r    r!   r@  l  s     )r   r   r   rÎ   rÏ   r	   r   rù   rë   r   r   rq   r    r    r!   Útest_with_listlike_columns]  s    
ýÿrQ  c                  C   sp   t tddgdggƒtdgddggƒdœƒ} t ddg¡| _| jd	d
„ dd�}tg dgg| jd�}t ||¡ d S )NrB   rc   ÚqÚzré   r  )rÂ   Zj0)rÃ   Zj1c                    s   ‡ fdd„ˆ d D ƒS )Nc                    s   g | ]}|ˆ d  kr|‘qS r
  r    )r€   Úelr'  r    r!   r@  u  s      zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>r<   r    r'  r    r'  r!   r,   u  r-   z;test_with_listlike_columns_returning_list.<locals>.<lambda>r   r   rV   )r   r   r   Úfrom_tuplesr   r   r   r   rq   r    r    r!   Ú)test_with_listlike_columns_returning_listp  s
    *rV  c                  C   s\   t ddgddgtdƒtdƒgdœƒ} | jdd	„ d
d�}tdd„ |  ¡ D ƒƒ}t ||¡ d S )Nr¤   r¾   r£   rØ   z2017-11-29 03:30:00z2017-11-29 03:45:00)ÚnumberÚstringr   c                 S   s   | j | jfS r:   ©rW  rX  r'  r    r    r!   r,   ‡  r-   z1test_infer_output_shape_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}|j |jf‘qS r    rY  rè   r    r    r!   r@  ˆ  s     z3test_infer_output_shape_columns.<locals>.<listcomp>©r   r   r   r   rë   r   r   rq   r    r    r!   Útest_infer_output_shape_columnsz  s    þýÿ
r[  c                  C   sˆ   t tj dd¡dddgd�} | jdd„ d	d
�}tdd„ |  ¡ D ƒƒ}t ||¡ | jdd„ d	d
�}tdd„ |  ¡ D ƒƒ}t ||¡ d S )Nrt   rI   r   r#   r$   re   c                 S   s
   dddgS r‹   r    r;   r    r    r!   r,   ‘  r-   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}d ddg‘qS ©r   r/   rI   r    rè   r    r    r!   r@  ’  s     z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S   s   ddgS ©Nr   r/   r    r;   r    r    r!   r,   •  r-   c                 S   s   g | ]}d dg‘qS ©r   r/   r    rè   r    r    r!   r@  –  s     ©	r   r   rÎ   rÏ   r   r   rë   r   r   rq   r    r    r!   Ú(test_infer_output_shape_listlike_columnsŒ  s    r`  c                    s`   t ddddgitdƒd�}|j‡ fdd„dd	�}t‡ fd
d„| ¡ D ƒ|jd�}t ||¡ d S )NrB   r   r/   rI   r�   rV   c                    s
   t  ˆ ¡S r:   ©r   Zonesr'  r:  r    r!   r,   Ÿ  r-   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>r   c                    s   g | ]}t  ˆ ¡‘qS r    ra  rè   r:  r    r!   r@     s     zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r…   r   r   rë   r   r   r   )rû   r4   r   r   r    r:  r!   Ú0test_infer_output_shape_listlike_columns_np_funcš  s     rb  c                  C   s~   t tdƒtdƒtdƒtdƒgddddgdddd	gd
d	ddgdœƒ} dd„ }| j|d
d�}tdd„ |  ¡ D ƒƒ}t ||¡ d S )Nz
2010-02-01z
2010-02-04z
2010-02-05z
2010-02-06rw   rs   r.   rI   r/   r   r  c                 S   s   dS )Nr^  r    r;   r    r    r!   Úfun´  s    zDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.funr   c                 S   s   g | ]}d ‘qS )r^  r    rè   r    r    r!   r@  ¸  s     zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>rZ  )r4   rc  r   r   r    r    r!   Ú7test_infer_output_shape_listlike_columns_with_timestamp¤  s    ü


÷ÿrd  Úlstc                    s\   t tj dd¡dddgd�}|j‡ fdd„d	d
�}t‡ fdd„| ¡ D ƒƒ}t ||¡ d S )Nr.   rI   r   r#   r$   re   c                    s   ˆ S r:   r    r;   ©re  r    r!   r,   Â  r-   z3test_consistent_coerce_for_shapes.<locals>.<lambda>r   r   c                    s   g | ]}ˆ ‘qS r    r    rè   rf  r    r!   r@  Ã  s     z5test_consistent_coerce_for_shapes.<locals>.<listcomp>r_  )re  r4   r   r   r    rf  r!   Ú!test_consistent_coerce_for_shapes¼  s    rg  c                 C   sd   | }|j dd„ dd�}| jddddœd	�}t ||¡ |j d
d„ dd�}|ddg }t ||¡ d S )Nc                 S   s   t dddgdddgd�S )Nr   r/   rI   ÚtestÚotherr   rV   ©r   r;   r    r    r!   r,   Ì  r-   z'test_consistent_names.<locals>.<lambda>r   r   rh  ri  r   r  re   c                 S   s   t ddgddgd�S )Nr   r/   rh  ri  rV   rj  r;   r    r    r!   r,   Ó  r-   )r   Úrenamer   r>   r’   r    r    r!   Útest_consistent_namesÇ  s     ÿ
ÿrl  c                 C   s<   | }|j dd„ ddd�}| ¡ }dddg|_t ||¡ d S )Nc                 S   s
   dddgS r‹   r    r;   r    r    r!   r,   Ý  r-   z"test_result_type.<locals>.<lambda>r   r^   r_   r   r/   ©r   r‘   rX   r   r>   r’   r    r    r!   Útest_result_typeØ  s
    rn  c                 C   sB   | }|j dd„ ddd�}|ddg  ¡ }ddg|_t ||¡ d S )	Nc                 S   s   ddgS r]  r    r;   r    r    r!   r,   ç  r-   z/test_result_type_shorter_list.<locals>.<lambda>r   r^   r_   r   r#   r   rm  r’   r    r    r!   Útest_result_type_shorter_listã  s
    
ro  c                 C   s0   | }|j dd„ ddd�}| ¡ }t ||¡ d S )Nc                 S   s
   dddgS r‹   r    r;   r    r    r!   r,   ò  r-   z,test_result_type_broadcast.<locals>.<lambda>r   r{   r_   r�   r’   r    r    r!   Útest_result_type_broadcastí  s    rp  c                    s>   | }dddg‰ |j ‡ fdd„ddd�}| ¡ }t ||¡ d S )	Nri  r  r/  c                    s   t dddgˆ d�S ©Nr   r/   rI   rV   rj  r;   re   r    r!   r,   ý  r-   z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r{   r_   r�   r’   r    re   r!   Ú&test_result_type_broadcast_series_func÷  s    

  ÿrr  c                 C   s.   | }|j dd„ dd�}| ¡ }t ||¡ d S )Nc                 S   s   t dddg| jd�S rq  )r   r   r;   r    r    r!   r,     r-   z0test_result_type_series_result.<locals>.<lambda>r   r   r�   r’   r    r    r!   Útest_result_type_series_result  s    rs  c                    sB   | }dddg‰ |j ‡ fdd„dd�}| ¡ }ˆ |_t ||¡ d S )Nri  r  r/  c                    s   t dddgˆ d�S rq  rj  r;   re   r    r!   r,     r-   z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   rm  r’   r    re   r!   Ú*test_result_type_series_result_other_index  s    
rt  Úboxc                 C   s   t | ƒS r:   )r…   r;   r    r    r!   r,     r-   c                 C   s   t | ƒS r:   )rù   r;   r    r    r!   r,     r-   c                 C   s   t j| dd�S )Nr¥   rb   )r   r°   r;   r    r    r!   r,     r-   r…   rù   r°   c                    s‚   |}|j ‡ fdd„dd�}t‡ fdd„| ¡ D ƒƒ}t ||¡ |j ‡ fdd„ddd	�}|d
dg jdddœd�}t ||¡ d S )Nc                    s   ˆ ddgƒS r]  r    r;   ©ru  r    r!   r,   "  r-   z,test_consistency_for_boxed.<locals>.<lambda>r   r   c                    s   g | ]}ˆ d dgƒ‘qS r^  r    rè   rv  r    r!   r@  #  s     z.test_consistency_for_boxed.<locals>.<listcomp>c                    s   ˆ ddgƒS r]  r    r;   rv  r    r!   r,   &  r-   r^   r_   r   r#   r   rN   re   )r   r   rë   r   r   rk  r>   )ru  r�   r4   r   r   r    rv  r!   Útest_consistency_for_boxed  s    rw  c              	   C   s$  | dkrdnd}t jdd��þ t  |¡}t  |¡}| ¡ }|jt j| d�}t ||¡ |jt jg| d�}| ¡ }| dkrŽt 	|j
dgg¡|_
nt 	|jdgg¡|_t ||¡ |jt jt jg| d�}t||g|d�}| dkrôt 	|j
ddgg¡|_
nt 	|jddgg¡|_t ||¡ W 5 Q R X d S )	N¾   r   r   r   r   r   r   r   r   Úabsolute)r   r   Úabsr   r‘   r   r   r>   r   Zfrom_productrX   r   r
   )r   r   Ú
other_axisZf_absZf_sqrtr   r   r    r    r!   Útest_agg_transform+  s.    

ÿÿr|  c                  C   sT   t tdƒddœƒ} |  ddg¡}t ddgddgdœddgddgd	�}t ||¡ d S )
Nrs   rN   rï   rð   r   r.   r   r#   rÄ   )r   r†   rE   r   r>   rq   r    r    r!   Ú	test_demoQ  s      ÿr}  c                  C   sn   t tdƒddœƒ} |  ddgddgdœ¡}t ddtjgdtjd	gdœd
dgdddgd�}t | |¡|¡ d S )Nrs   rN   rï   rð   rk   ç      @r*  ç      @g      9@r   r#   rÄ   )r   r†   rE   r   rW   r   r>   Zreindex_likerq   r    r    r!   Útest_demo_dict_agg\  s    ýr€  c                  C   sl   dddgi} t | ƒ}| ddi¡}tddiƒ}t ||¡ |d  ddi¡}tddidd�}t ||¡ d S )Nrá   r£   rØ   Úcountr/   rà   )r   rE   r   r   r   )rF   r4   r   r   r    r    r!   Ú!test_agg_with_name_as_column_nameh  s    r‚  c               	   C   sî   t dddgdddgddd	gtd
dd�dœƒ} t ddgddgddgtdƒtjgdœddgd�}tjtdd�� |  ddg¡}W 5 Q R X t 	||¡ tjtdd��  | ddddg  ddg¡}W 5 Q R X |ddddg  
ddg¡}t 	||¡ d S )Nr   r/   rI   r¤   r¾   ç      @r£   rØ   ÚbazrH   rJ   ©r   r#   r$   r%   rt   ç      @Ú	foobarbazr6   rï   rk   rV   ú&\['D'\] did not aggregate successfully©Úmatchr%   r$   r#   r   )r   r	   r   rO   ZNaTr   Úassert_produces_warningÚFutureWarningrE   r>   Zreindex)Zmdfr   r   r    r    r!   Ú"test_agg_multiple_mixed_no_warningx  s8    
üÿüù
 ÿ ÿ$r�  c              	   C   s$  | dkrdnd}|j |  ¡ d d…  ¡ \}}tj|j| d�|j| d�|j| d�gdd�}dddg|_| dkrt|j	n|}|j
dddg| d�}t ||¡ |d|di}|j
|| d�}t| |¡|  ¡ | |¡|  ¡ g||gd	�}t ||¡ |dg|dgi}|j
|| d�}t|t| |¡|  ¡ gdgd	�|t| |¡|  ¡ gdgd	�iƒ}| d
k�r`|j	n|}t ||¡ |ddg|ddgi}|j
|| d�}tj|t| |¡|  ¡ | |¡|  ¡ gddgd	�|t| |¡|  ¡ | |¡|  ¡ gddgd	�idd�}| d
k�r|j	n|}t ||¡ d S )Nrx  r   r   r/   r   r   rð   rk   rV   >   r   rX   )ZaxesÚuniqueZsort_valuesrO   Úconcatr   rð   rk   rX   rp   rE   r   r>   r   rY   r   r   )r   r   r{  Zname1Zname2r   r   rQ   r    r    r!   Útest_agg_reducež  sn    


ýúþû  þÿ þû þûøïr�  c               	   C   s0  t dddgdddgddd	gtd
dd�dœƒ} |  d¡}tdddtd
ƒg| jd�}t ||¡ |  dg¡}t dddtd
ƒggdg| jd�}t ||¡ tj	t
dd�� |  d¡}W 5 Q R X tdddgdddgd�}t ||¡ tj	t
dd�� |  dg¡}W 5 Q R X t dddggdgdddgd�}t ||¡ d S )Nr   r/   rI   r¤   r¾   rƒ  r£   rØ   r„  rH   rJ   r…  rï   rV   ræ   úSelect only validr‰  rk   rt   r†  r‡  r   r#   r$   rˆ  )r   r	   rE   r   r   rX   r   r   r>   r‹  rŒ  rq   r    r    r!   Útest_nuiscance_columnsã  s@    
üÿ	
ý ÿ
  ÿr’  Úhowc                 C   s  t d ddgdtjdgdd dgdœƒ}t|| ƒdd	iƒ}tddiƒ}t ||¡ t|| ƒdd
iƒ}tddiƒ}t ||¡ t|| ƒd	d
gƒ}t|| ƒd	d
gd	d
gd	d
gdœƒ}t dddœdddœdddœdœƒ}tj||dd� tj||dd� t|| ƒd	ƒ}| ¡ }t ||¡ d S )Nr/   rI   r¤   rƒ  r£   rØ   r  r   r�  r  )r�  r  Trñ   )	r   r   rW   rn   r   r   r   r>   r�  )r“  r4   r   r   Zresult1Zresult2r    r    r!   Útest_non_callable_aggregates
  s0    ÿÿýÿr”  c                 C   s€   t d ddgdtjdgdd dgdœƒ}t|| ƒd|d	�}|d
ksF|dkr\t|jd
 |jd�}nt|jd |jd�}t 	||¡ d S )Nr/   rI   r¤   rƒ  r£   rØ   r  r  r   r   r   rV   r   )
r   r   rW   rn   r   r0   rX   r   r   r   )r“  r   r4   r   r   r    r    r!   Útest_size_as_str8  s    ÿr•  c                  C   sŒ   t dddgdtjdgdd dgdœƒ} dd„ }|  |¡}tddgdgddggd	d
dgd�}t ||¡ |  |g¡}| d¡j}t 	||¡ d S )Nr/   rI   g      ø?r£   rØ   r  c                 S   s   t |  ¡  ¡ ƒS r:   )r…   ZdropnarŽ  )Z	group_colr    r    r!   rQ   L  s    z&test_agg_listlike_result.<locals>.funcr   r#   r$   rV   rQ   )
r   r   rW   rE   r   r   r   Zto_framerp   r>   )r4   rQ   r   r   r    r    r!   Útest_agg_listlike_resultH  s    $
"r–  zargs, kwargsr\  )rv   ru   é   r^  )r   rc   r    ©rB   rc   ©rB   rc   rd   c                 C   sd   ddd„}t ddgddggƒ}| dkr4tdd	gƒ}ntd
dgƒ}|j|| f|ž|Ž}t ||¡ d S )NrI   c                 S   s   |   ¡ || |  S r:   ©rk   )r<   rB   rc   rd   r    r    r!   rÖ   f  s    ztest_agg_args_kwargs.<locals>.fr   r/   r.   r   r  g      @r~  g       @)rI   ©r   r   rE   r   r   )r   r½   ÚkwargsrÖ   r4   r   r   r    r    r!   Útest_agg_args_kwargsX  s    
r�  Únum_colsrs   c                 C   sL   t  dddg¡}| ¡ }td|t| ƒd�}| dd„ ¡ |j|jksHt‚d S )Nz
1950-06-30z
1952-10-24z
1953-05-29r   ræ   c                 S   s   | S r:   r    r;   r    r    r!   r,   {  r-   z,test_frequency_is_original.<locals>.<lambda>)rO   ZDatetimeIndexr‘   r   r†   r   r  r   )rž  r   Úoriginalr4   r    r    r!   Útest_frequency_is_originalu  s
    r   c                  C   sb   t ddd�t ddd�t ddd�g} tdddg| d	�}|jd
d„ dd�}t| | d�}t ||¡ d S )Nz2019-03-15 12:34:31.909000+0000r7   r8   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r   r   r/   )rF   r   c                 S   s   | j S r:   rà   r;   r    r    r!   r,   ˆ  r-   z.test_apply_datetime_tz_issue.<locals>.<lambda>r   )r   rF   )r   r   r   r   r   r   )Z
timestampsr4   r   r   r    r    r!   Útest_apply_datetime_tz_issue  s    


ýr¡  r4   rB   rd   r   rN   Úmethodrï   rð   c              	   C   sR   t jtdd��. t| ddg |ƒƒ }t| ddg |ƒƒ }W 5 Q R X t  ||¡ d S )Nr‘  r‰  r   r#   )r   r‹  rŒ  rn   r   )r4   r¢  Znone_in_first_column_resultZnone_in_second_column_resultr    r    r!   Ú=test_consistency_of_aggregates_of_columns_with_missing_valuesŽ  s    r£  r  c                 C   s:   t d| ggddgd�}| dd„ ¡}|j}t ||¡ d S )Nr¤   rB   rc   re   c                 S   s   | j S r:   rb   r;   r    r    r!   r,   �  r-   z"test_apply_dtype.<locals>.<lambda>)r   r   rü   r   r   )r  r4   r   r   r    r    r!   Útest_apply_dtype™  s    r¤  c                 C   s€   t tdƒtddƒdœƒ}| ¡ }dd„ }| ¡ }|d  d7  < |j|dd�}t ||¡ |sb| rpt ||¡ nt ||¡ d S )	Néd   éÈ   r˜  c                 S   s*   | j }| jd  d7  < | j |k	s&t‚| S )NrB   r   )Z_mgrrY   r   )r(  Zmgrr    r    r!   rQ   ¨  s    z!test_apply_mutating.<locals>.funcrB   r   r   )r   r†   r‘   r   r   r>   )Zusing_array_managerZusing_copy_on_writer4   Zdf_origrQ   r   r   r    r    r!   Útest_apply_mutating£  s    r§  c                  C   sb   t ddgddgddgddgd	d
ggddgd�} | jdd„ dd�}tg g dœtd�}t ||¡ d S )Nr   r/   rI   r.   rs   rt   ru   rv   rw   rÈ   rB   rc   re   c                 S   s   g S r:   r    r;   r    r    r!   r,   Á  r-   z.test_apply_empty_list_reduce.<locals>.<lambda>ra   r|   r˜  rb   )r   r   r   rh   r   r   rq   r    r    r!   Útest_apply_empty_list_reduce½  s    .r¨  c                  C   sf   t ddggd ddgd�} |  ddd	„ dd	„ g¡}t d
d
d
gdddgdœdddgd�}t ||¡ d S )Nr.   rw   rI   r   r#   re   rk   c                 S   s   |   ¡ S r:   rš  r;   r    r    r!   r,   É  r-   z,test_apply_no_suffix_index.<locals>.<lambda>r>  é   rN   r,   rV   r\   )Zpdfr   r   r    r    r!   Útest_apply_no_suffix_indexÆ  s     ÿrª  c                  C   s@   t dddgiƒ} | jdd„ ddd�}tddgƒ}t ||¡ d S )	Nr   ZaaÚbbbc                 S   s   | d S rx   r    r;   r    r    r!   r,   Ô  r-   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr™   r§   rq   r    r    r!   Útest_apply_raw_returns_stringÑ  s    r¬  c               	   C   s    t ddddddgdd�} d	d
„ }d|ddg}tjtdd�� |  |¡}W 5 Q R X t dtjddgddddgddddgddddgdœddddgd�}t ||¡ d S )N)Ú1r   r   r   )Ú2r/   r   r   )Ú3rI   r   r   )Ú4r.   rs   r.   )Ú5rs   rt   rt   )Ú6rt   ru   ru   )ÚitemZatt1Zatt2Zatt3re   c                 S   s   |   ¡ d S rŸ   rš  ry   r    r    r!   r£   è  s    z/test_aggregation_func_column_order.<locals>.foork   r�  rï   z)\['item'\] did not aggregate successfullyr‰  Z123456rt   r­  g      5@g      %@r†  r¤   g      2@g      "@r*  g      1@g      !@r£   rV   )r   r   r‹  rŒ  rE   r   rW   r>   )r4   r£   Zaggsr   r   r    r    r!   Ú"test_aggregation_func_column_orderÙ  s4    ú÷ ÿ


ü
ù	r´  c                  C   sR   t dddgdddgdœƒ} | ddg jdd„ dd	�}tddd
gƒ}t ||¡ d S )Nr   r   r/   rI   r˜  rB   c                 S   s   | d | d  S )Nr   r   r    r;   r    r    r!   r,   ÿ  r-   z+test_apply_getitem_axis_1.<locals>.<lambda>r   r.   r§   rq   r    r    r!   Útest_apply_getitem_axis_1ü  s    rµ  c               	   C   sD   dd„ } t ddddgiƒ}tjtdd�� | | g¡ W 5 Q R X d S )	Nc                 S   s   t  d¡ |  ¡ S )NúHello, World!)r¬   Úwarnrk   r;   r    r    r!   r£     s    
z7test_nuisance_depr_passes_through_warnings.<locals>.foorB   r   r/   rI   r¶  r‰  )r   r   r‹  ÚUserWarningrE   )r£   r4   r    r    r!   Ú*test_nuisance_depr_passes_through_warnings  s    r¹  c                  C   s¾   t ddtgdtdddƒtjgdœddd	gd
�} |  t¡}t tttgtttgdœddd	gd
�}t	 
||¡ | jtdd�}tttdœƒ}t	 ||¡ | jtdd�}ttttdœƒ}t	 ||¡ d S )NrI   rX  g      Ð?iä  r   )rí   rî   rB   rc   rd   rV   r   r   r™  )r   Úfloatr   r   rW   r÷   rø   Úintrÿ   r   r>   r   r   r   rq   r    r    r!   Útest_apply_type  s     þ
þr¼  c                  C   sL   t ddgddgdœƒ} |  d¡jdd„ dd�}tg tjd	�}t ||¡ d S )
Nr   r/   rI   r   r˜  c                 S   s   t | d | d ƒS )NrB   rc   )rð   r;   r    r    r!   r,   2  r-   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   rb   )r   Úheadr   r   r   ri   r   r   rq   r    r    r!   Útest_apply_on_empty_dataframe/  s    r¾  ztest, constantc                 C   s>   t | ƒ}t|ddgdd�}dddœ}| |¡}t ||¡ d S )NrB   rc   rh   )rF   r   r2   rŽ  r˜  r›  )rh  ZconstantZdf1r   Zaggregationr   r    r    r!   Útest_unique_agg_type_is_series7  s
    	

r¿  )†r   r¬   Únumpyr   ZpytestZpandas.core.dtypes.dtypesr   ZpandasrO   r   r   r   r   r	   Zpandas._testingZ_testingr   Zpandas.tests.frame.commonr
   r"   r5   r@   ÚmarkZparametrizerG   rP   r   r   rT   r[   r]   rj   ro   rr   rz   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…   Úroundr  r  r  r  r  r   r"  r#  r%  r+  r,  r3  r8  r;  r<  r?  rA  rB  rF  rM  rN  rQ  rV  r[  r`  rb  rd  rg  rl  rn  ro  rp  rr  rs  rt  rw  r|  r}  r€  r‚  r�  r�  r’  r”  r•  r–  r�  r   r¡  r£  rW   r¤  r§  r¨  rª  r¬  r´  rµ  r¹  r¼  r¾  r¿  r    r    r    r!   Ú<module>   st  
þ


	


		  ÿ			#4	þ*ûþ



þ
üþ		




	




ý
&&E'
-

ùþ	*			#&"þþ