a
    nÝEbsº  ã                   @   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g d¢¡d d!„ ƒZ d"d#„ Z!d$d%„ Z"d&d'„ Z#d(d)„ Z$d*d+„ Z%d,d-„ Z&d.d/„ Z'd0d1„ Z(ej d2d dg¡d3d4„ ƒZ)ej d2d dg¡d5d6„ ƒZ*d7d8„ Z+ej d2d dg¡d9d:„ ƒZ,d;d<„ Z-d=d>„ Z.d?d@„ Z/ej dAdBdCg¡ejjddDdE„ dFdE„ gdGdHgdI�ej dJdKdLg¡ej d2d dg¡dMdN„ ƒƒƒƒZ0dOdP„ Z1dQdR„ Z2dSdT„ Z3dUdV„ Z4dWdX„ Z5dYdZ„ Z6d[d\„ Z7d]d^„ Z8d_d`„ Z9dadb„ Z:dcdd„ Z;dedf„ Z<dgdh„ Z=didj„ Z>dkdl„ Z?ej dme	dndogdpdqggƒedndpdrœdodqdrœgƒge	d dgdsdtggƒed dsdrœddtdrœgƒgg¡dudv„ ƒZ@dwdx„ ZAej dyddzg¡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‹dE„ 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 dyd­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 dyddsg¡dÁdÂ„ ƒZadÃdÄ„ Zbej dÅg dÆ¢ddsgg¡dÇdÈ„ ƒZcdÉdÊ„ ZddËdÌ„ ZedÍdÎ„ ZfdÏdÐ„ ZgdÑdÒ„ ZhdÓdÔ„ ZidÕdÖ„ Zjejjd×dØdE„ dÙdE„ dÚdE„ gg dÛ¢dI�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 d2d dg¡ej dódÆi fdôi fdõi fdöd÷dsifdøddsdùœfdødsddùœfdøddsdtdúœfg¡dûdü„ ƒƒZvej dýg dþ¢¡dÿ�d „ ƒZw�d�d„ Zxej �de	�ddg�d�dg�dœƒg¡ej �dg �d	¢¡�d
�d„ ƒƒZyej �dddzdK�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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sbJ ‚| jd }| jt jdd�}t  |  |¡¡}|| |ksœJ ‚|j| ju s¬J ‚W d   ƒ n1 sÀ0    Y  d S )NÚignore©ÚallÚAr   é   ©Úaxis)	ÚnpÚerrstateÚsqrtÚapplyÚtmÚassert_series_equalÚmeanÚindexÚxs)Úfloat_frameÚresultÚexpectedÚd© r   úc/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/pandas/tests/apply/test_frame_apply.pyÚ
test_apply   s    
r!   c                  C   s^   t g d¢g d¢dœƒ} |  dd„ ¡}|jdks2J ‚t|d jtƒsFJ ‚t|d jtƒsZJ ‚d S )	N)r   r   ÚBr"   )ÚCr#   ÚDr$   ©Ú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   ÚshapeÚ
isinstanceÚdtyper   ©Údfr   r   r   r    Útest_apply_categorical_func(   s
    r4   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   r5   r6   r7   c                 C   s<   t | g| ggddg|d�}|jdd„ dd�}t ||¡ d S )NÚa©r   r1   c                 S   s   | S r9   r   r:   r   r   r    r+   @   r,   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   ©r   Úaggr   r=   )Údatar1   r   r   r   r   r    Útest_agg_axis1_duplicate_index9   s    rF   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 r9   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    
þÿrO   Úfuncc                 C   s   t ƒ }| | ¡}|jsJ ‚d S r9   )r   r   Úempty)rP   Úempty_framer   r   r   r    Útest_apply_emptyQ   s    
rS   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 r9   ©r   r:   r   r   r    r+   \   r,   z(test_apply_float_frame.<locals>.<lambda>©r   c                 S   s   |   ¡ S r9   rT   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    rZ   c                  C   s.   t dgd�} | jdd„ dd�}t || ¡ d S )NrA   rU   c                 S   s   | d S )NrA   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 g d¢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ÐJ ‚d S )	Nr   Úexpand©r   Úresult_typeÚreduce©r1   rB   ©rA   ÚbÚc©rW   )r   r   Úappendr   r=   r   rN   ÚIndexÚobjectr   Úfloat64r   )rR   r;   r   r   Zempty_with_colsr   r   r    Útest_apply_with_reduce_emptym   s    rj   )ÚsumÚprodÚanyr   c                 C   s:   t g d¢d�}| tt| ƒ¡}t|| ƒƒ }t ||¡ d S )Nrb   re   )r   r   Úgetattrr   r   r   )rP   r3   r   r   r   r   r    Útest_apply_funcs_over_emptyƒ   s    ro   c                  C   s`   t g d¢d�} |  ¡ }td| jd�}t ||¡ | j ¡ }tg t g ¡t	j
d�}t ||¡ d S )Nrb   re   r   rU   rB   )r   Znuniquer   rW   r   r   ÚTrN   rg   r   ri   ©r3   r   r   r   r   r    Útest_nunique_empty�   s    
rr   c                  C   sv   t g d¢g d¢g d¢gg d¢d�} | jdd„ dd	�}tg d
¢g d¢ƒ}t ||¡ | jjdd„ dd	�}t ||¡ d S )N©r   r.   rH   )r-   é   é   )é   é   é	   )rA   rA   rd   rU   c                 S   s   | d S ©Nr   r   ©rK   r   r   r    r+   �   r,   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r-   rv   c                 S   s   | d S ry   r   rz   r   r   r    r+   ¡   r,   r   )r   r   r   r   r   rp   rq   r   r   r    Útest_apply_standard_nonuniqueš   s    "r{   c                 C   s4   | j tjdd�}t|  ¡ g| jd�}t ||¡ d S )NÚ	broadcast©r_   rU   )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>¯   r,   z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>)r   r   r   r   rW   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 r9   )ÚlistÚrangeÚlenrW   r:   ©r   r   r    r+   ¶   r,   z4test_apply_broadcast_lists_columns.<locals>.<lambda>r   r|   r^   ri   )r1   r   rW   )	r   r†   r‡   rˆ   rW   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 r9   )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„   Ê   r,   z4test_apply_broadcast_lists_index.<locals>.<dictcomp>ri   )r1   r   )	r   r†   r‡   rˆ   r   r   rW   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   g d¢S ©Nrs   r   r:   r   r   r    r+   Ô   r,   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r|   r^   ©r   r   r=   )Úint_frame_const_colr3   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 g d¢tdƒd�S )Nrs   ÚabcrU   )r   r†   r:   r   r   r    r+   Û   r,   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r|   r^   ©r   Úcopyr   r=   ©rŽ   r3   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J ‚| jdksJ ‚d S ©Nr   ©r0   r   ÚndarrayÚ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 r9   )Ú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J ‚| jdksJ ‚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:J ‚d S )Nr   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ç      ð?rM   r   r   r   Úint64ra   rU   )
r   r   r   r   r   rV   rN   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¥   rM   c                 S   s   | d S )Nr   r   r:   r   r   r    r+     r,   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   rU   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   rW   c                 C   s   | S r9   r   r:   r   r   r    r+   !  r,   r+   c                 C   s   |   ¡ S r9   rT   r:   r   r   r    r+   !  r,   Úidentityr   )Úidsr›   TFc           	   	   C   sè   t f i | g d¢i¤Ž}tjdd��² tjdd��. t dt¡ |tjg dd�ƒ}W d   ƒ n1 sd0    Y  t|tj	ƒ }|j
|||d�}|r¸| |¡}t|tƒs¨J ‚|j|u sÆJ ‚nt|t ƒsÆJ ‚W d   ƒ n1 sÚ0    Y  d S )	Nrb   r   r   T)ÚrecordZf8ra   rš   )r   r   r   ÚwarningsÚcatch_warningsÚsimplefilterÚRuntimeWarningÚarrayr0   r—   r   Z_get_agg_axisr   r   )	rª   rP   r›   r   r3   Ztest_resZis_reductionr   Zagg_axisr   r   r    Útest_apply_empty_infer_type  s    0
r³   c                  C   s2   t g d¢d�} | jdd„ dd�}t|t ƒs.J ‚d S )Nrb   rU   c                 S   s   |   ¡ S r9   rT   r:   r   r   r    r+   9  r,   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r|   r}   )r   r   r0   )rY   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 r9   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 r9   rT   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    rT   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 r9   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 r9   )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   rV   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.   rH   r-   r&   r'   Úi0Úi1©rW   r   r   r   )rÄ   rÅ   rU   r%   )r   r   Údictr   rW   r   r   r   )rE   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  r,   z2test_apply_differently_indexed.<locals>.<dictcomp>re   r   c                 S   s   i | ]\}}||  ¡ “qS r   rË   rÍ   r   r   r    r„   y  r,   )r   r   ÚrandomÚrandnr   r   rÌ   ÚitemsrW   r   r=   rp   r   rq   r   r   r    Útest_apply_differently_indexedq  s     rÓ   c               
   C   sØ   t g d¢g d¢g d¢g d¢g d¢g d¢gg d¢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g d¢d�} | j|dd�}t ||¡ d S )N)r   ÚABC0é2   )r   ÚYUM0rÉ   )r   ÚDEF0rÉ   )r.   ÚABC1rÕ   )r.   ÚYUM1rÉ   )r.   ÚDEF1rÉ   )rA   ÚmarketÚpositionre   c                 S   s   | d S )NrÛ   r   )Úrr   r   r    ÚfŒ  s    ztest_apply_bug.<locals>.fr   r   iÝ  rÔ   rÕ   r.   rÖ   rÉ   rH   r×   r-   rØ   rt   rÙ   ru   rÚ   )r   r   r   r   r   )Z	positionsrÞ   r   r   r   r   r    Útest_apply_bug}  s0    ú÷ú÷rß   c               	   C   sd   t g d¢g d¢g d¢tj d¡tj d¡tj d¡dœƒ} | jdd„ dd	�jd
d�}t || ¡ d S )N)r¤   r¤   r¤   r¤   Úbarrà   rà   rà   r¤   r¤   r¤   )Úonerá   rá   Útworá   rá   rá   râ   râ   râ   rá   )Údullrã   Úshinyrã   rã   rä   rä   rã   rä   rä   rä   é   )r   r"   r#   r$   ÚEÚFc                 S   s   | S r9   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   s    


Öÿ/ré   c                 C   s.   |   dd„ ¡}t| j| jd�}t ||¡ d S )Nc                 S   s   | j S r9   ©Únamer:   r   r   r    r+   Õ  r,   z(test_apply_attach_name.<locals>.<lambda>rU   )r   r   rW   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 r9   rê   r:   r   r   r    r+   Û  r,   z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   rU   )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 r9   ©r   Úrepeatrë   rˆ   r:   r   r   r    r+   â  r,   z6test_apply_attach_name_non_reduction.<locals>.<lambda>r   ©r   rW   )	r   r   r   ZtilerW   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 r9   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ˆ   rW   ©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  g d¢g d¢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 )N)rA   rA   rc   )rd   r   r   r   r.   rH   r-   rt   ru   Zcol1Z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   rK   r   r   r   r   r    Útest_apply_multi_indexô  s
    $$rú   z	df, dictsr¤   rà   ZspamÚeggs)r   r   r.   rH   c                 C   sV   dd„ }| j |dd�}| j |dd�}|   |¡}t ||¡ t || ¡ t ||¡ d S )Nc                 S   s   |   ¡ S r9   )Úto_dictr:   r   r   r    r+     r,   z!test_apply_dict.<locals>.<lambda>r`   r}   r]   )r   r   r   r=   )r3   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ƒsLJ ‚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 r9   r   r:   r   r   r    r+     r,   r   r   )Úapplymapr   r=   Útyper0   Útuple)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.J ‚d S )NrA   )rE   c                 S   s   | S r9   r   r:   r   r   r    r+      r,   z7test_applymap_float_object_conversion.<locals>.<lambda>r   )r   rÿ   Údtypesrh   )r  r3   r   r   r   r    Ú%test_applymap_float_object_conversion  s    r  c                  C   sP   t tj d¡ƒ} |  ¡ }g d¢}|| _| t¡}||_|  t¡}t ||¡ d S )N©rH   r-   )rA   rA   rA   rA   )	r   r   rÐ   r’   rW   rÿ   Ústrr   r=   )r3   Zdf2Úcolsr   r   r   r   r    Útest_applymap_str$  s    

r	  zcol, valr   rG   Ú	timedeltaz1 minc                 C   sJ   t tj d¡ƒ}||| < | t¡}|jd| f t|jd| f ƒksFJ ‚d S )Nr  r   )r   r   rÐ   rÿ   r  rX   )Úcolr  r3   r   r   r   r    Útest_applymap_datetimelike1  s    
r  r   ÚABCre   rU   ©r   r"   r#   c                 C   s   | S r9   r   r:   r   r   r    r+   F  r,   c                 C   s   |   |¡}t || ¡ d S r9   )rÿ   r   r=   )r   rP   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.   rH   r-   c                 S   s   | | S r9   r   ©r;   Úyr   r   r    r+   O  r,   z&test_applymap_kwargs.<locals>.<lambda>©r  rt   ru   )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 r9   ©rˆ   r  r:   r   r   r    r+   V  r,   z)test_applymap_na_ignore.<locals>.<lambda>r   r.   )Úsizer1   c                 S   s   t t| ƒƒS r9   r  r:   r   r   r    r+   [  r,   r   )Z	na_action)rÿ   r’   r   rÐ   Úrandintr/   ÚboolrN   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Ê   rI   c                 S   s   | j | j| jfS r9   )ÚhourÚdayÚmonthr:   r   r   r    rP   f  s    z*test_applymap_box_timestamps.<locals>.func)r   r	   r   rÿ   )ZserrP   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/Easternr7   z1 daysz2 daysÚM)Úfreq©rA   rc   rd   r   c                 S   s
   t | ƒjS r9   )r   Ú__name__r:   r   r   r    r+   ~  r,   z#test_applymap_box.<locals>.<lambda>r   Ú	TimedeltaÚPeriod)r   r   rN   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sZJ ‚d S )
Nr   ©ÚBDayÚx1iÌ  r   c                    s
   | ˆ ƒ  S r9   r   r:   r%  r   r    r+   �  r,   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>c                    s
   | ˆ ƒ  S r9   r   r:   r%  r   r    r+   �  r,   zM8[ns])Zpandas.tseries.offsetsr&  r   r   rÿ   r'  r1   r2   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-01rH   zEurope/Brussels)rJ   r8   c                 S   s   | S r9   r   r:   r   r   r    r+   ™  r,   z,test_apply_non_numpy_dtype.<locals>.<lambda>c                 S   s   | t  d¡ S )NZ1day)rN   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   s2   t dg d¢idd�} |  dd„ ¡}t || ¡ d S )Nr)  )rA   rc   rd   rA   r(   ra   c                 S   s   | S r9   r   r:   r   r   r    r+   ¥  r,   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>r[   r2   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.   rH   rA   re   r÷   rÆ   rC   ©r3   r   r   r   r   r    Útest_apply_dup_names_multi_agg©  s    r-  Úopr   rD   c                 C   s`   dd„ }t t d¡tdƒd�}t|| ƒ|dd�}tg d¢g d¢g d¢g d¢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-   ÚABCDre   r   r   )ç        r3  r3  )r   r   Zzerosr†   rn   r   r   r   )r.  r1  r3   r   r   r   r   r    Útest_apply_nested_result_axis_1²  s    ÿr4  c                  C   sR   t dtddd�gidd�} |  dd„ ¡}t || ¡ |  d	d„ ¡}t || ¡ d S )
Nr¤   Z2020r6   r7   zdatetime64[ns, UTC]ra   c                 S   s   | S r9   r   r:   r   r   r    r+   Å  r,   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   s   |   ¡ S r9   )r’   r:   r   r   r    r+   Ç  r,   r<   r>   r   r   r    Ú%test_apply_noreduction_tzaware_objectÀ  s    ÿr5  c                     sh   t dg d¢iƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ].}ˆ d d …= | j|dd� ˆ t| jƒks4J ‚q4d S )	NrA   rs   c                    s   ˆ   | j¡ d S r9   ©rf   rë   r/  ©Únamesr   r    Úreducing_functionÑ  s    z8test_apply_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | j¡ | S r9   r6  r/  r7  r   r    Únon_reducing_functionÔ  s    z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r†   r   ©r3   r9  r:  rP   r   r7  r    Útest_apply_function_runs_onceË  s    
r<  c                     sn   t dg d¢iƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ]4}ˆ d d …= | j|ddd	� ˆ t| j ¡ ƒks4J ‚q4d S )
NrA   rs   c                    s   ˆ   | ¡ d S r9   ©Úextendr/  ©rž   r   r    r9  å  s    z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | ¡ | S r9   r=  r/  r?  r   r    r:  è  s    
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r›   r   )r   r   r†   rA   Úto_listr;  r   r?  r    Ú!test_apply_raw_function_runs_onceß  s    
rA  c                     sd   t dg d¢iƒ} g ‰ ‡ fdd„}‡ fdd„}||fD ]*}ˆ d d …= |  |¡ ˆ | j ¡ ks4J ‚q4d S )NrA   rs   c                    s   ˆ   | ¡ d S r9   ©rf   ©r  r?  r   r    r9  ø  s    z;test_applymap_function_runs_once.<locals>.reducing_functionc                    s   ˆ   | ¡ | S r9   rB  rC  r?  r   r    r:  û  s    
z?test_applymap_function_runs_once.<locals>.non_reducing_function)r   rÿ   rA   r@  r;  r   r?  r    Ú test_applymap_function_runs_onceó  s    

rD  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   )rW   r1   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    rE  Ú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 )NrF  rG  ZcderA   r(   ra   c                    s   | ˆ kS r9   r   r:   rC  r   r    r+     r,   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t  |¡rtjn|ˆ k‘qS r   )rN   Zisnullr   ÚNaN)r�   r;   rC  r   r    Ú
<listcomp>  r,   z1test_apply_category_equalness.<locals>.<listcomp>rê   )r   rH  r   rA   r   r   r   r   )r  Z	df_valuesr3   r   r   r   rC  r    Útest_apply_category_equalness  s    ÿrJ  c                  C   sV   t tj dd¡ƒ} | jtjjdd�j}|dks2J ‚| jtjjdd�j}|dksRJ ‚d S )NrÊ   r.   r   r   )rÊ   r.   )ru   r.   )r   r   rÐ   Zrandr   Zfftr/   Zrfftr2   r   r   r    Útest_infer_row_shape"  s
    rK  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.   rA   rc   re   c                 S   s   d| d | d  iS ©NrK   rA   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 )rK   rH   r   rò   r   r   r    rI  1  r,   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 rL  r   r:   r   r   r    r+   8  r,   c                 S   s   d| iS )NrK   r   r:   r   r   r    r+   <  r,   rK   rH   )r   r   r   rõ   r   r   r   rq   r   r   r    Útest_with_dictlike_columns-  s    þrO  c                  C   s`   t ƒ } g d¢| d< g d¢| d< t g d¢¡| d< | jdd„ d	d
�}ti i i gƒ}t ||¡ d S )N)ÚXÚYÚZÚauthor)ZBBCZNBCZN24Z	publisher)z17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09Údatec                 S   s   i S r9   r   r:   r   r   r    r+   I  r,   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rN   Úto_datetimer   r   r   r   rq   r   r   r    Ú(test_with_dictlike_columns_with_datetimeA  s    ÿrV  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.   rA   rc   re   c                 S   s   d| d | d  iS rL  r   r:   r   r   r    r+   Q  r,   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>r]   r^   rK   rH   rM  rN  r   c                 S   s   d| d | d  iS rL  r   r:   r   r   r    r+   Y  r,   )r   r   r   r=   r   rq   r   r   r    Ú%test_with_dictlike_columns_with_inferN  s    þrW  c                  C   s®   t ttj d¡ƒg d¢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-   )rA   r†   ZofÚwordsz
2016-10-01ÚH)rJ   r  )rA   rc   r*   rA   rc   r   r   c                 S   s   g | ]}|d d… ‘qS r@   r   rò   r   r   r    rI  h  r,   z.test_with_listlike_columns.<locals>.<listcomp>r*   c                 S   s   g | ]}|d d… ‘qS r@   r   rò   r   r   r    rI  l  r,   )r   r   r   rÐ   rÑ   r	   r   r  rõ   r   r   rq   r   r   r    Útest_with_listlike_columns]  s    ýÿrZ  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 )NrA   rc   ÚqÚzró   r  )rÄ   Zj0)rÅ   Zj1c                    s   ‡ fdd„ˆ d D ƒS )Nc                    s   g | ]}|ˆ d  v r|‘qS r  r   )r�   Úelr/  r   r    rI  u  r,   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   rU   )r   r   r   Úfrom_tuplesr   r   r   r   rq   r   r   r    Ú)test_with_listlike_columns_returning_listp  s
    *r_  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 r9   ©r`  ra  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   rb  rò   r   r   r    rI  ˆ  r,   z3test_infer_output_shape_columns.<locals>.<listcomp>©r   r   r   r   rõ   r   r   rq   r   r   r    Útest_infer_output_shape_columnsz  s    þýÿ
rd  c                  C   s†   t tj dd¡g d¢d�} | jdd„ dd�}td	d
„ |  ¡ D ƒƒ}t ||¡ | jdd„ dd�}tdd
„ |  ¡ D ƒƒ}t ||¡ d S )Nru   rH   r  re   c                 S   s   g d¢S rŒ   r   r:   r   r   r    r+   ‘  r,   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}g d ¢‘qS )rs   r   rò   r   r   r    rI  ’  r,   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    rI  –  r,   ©	r   r   rÐ   rÑ   r   r   rõ   r   r   rq   r   r   r    Ú(test_infer_output_shape_listlike_columnsŒ  s    rh  c                    s^   t dg d¢itdƒd�}|j‡ fdd„dd�}t‡ fd	d
„| ¡ D ƒ|jd�}t ||¡ d S )NrA   rs   r�   rU   c                    s
   t  ˆ ¡S r9   ©r   Zonesr/  rC  r   r    r+   Ÿ  r,   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>r   r   c                    s   g | ]}t  ˆ ¡‘qS r   ri  rò   rC  r   r    rI     r,   zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r†   r   r   rõ   r   r   r   )r  r3   r   r   r   rC  r    Ú0test_infer_output_shape_listlike_columns_np_funcš  s     rj  c                  C   sr   t tdƒtdƒtdƒtdƒgg d¢g d¢g d¢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-06)rx   rt   r-   rH   )rt   rH   r-   r.   )r   r.   rH   r-   r   c                 S   s   dS )Nrf  r   r:   r   r   r    Úfun´  s    zDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.funr   r   c                 S   s   g | ]}d ‘qS )rf  r   rò   r   r   r    rI  ¸  r,   zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>rc  )r3   rk  r   r   r   r   r    Ú7test_infer_output_shape_listlike_columns_with_timestamp¤  s    ü÷ÿrl  Úlstrs   c                    sZ   t tj dd¡g d¢d�}|j‡ fdd„dd�}t‡ fd	d
„| ¡ D ƒƒ}t ||¡ d S )Nr-   rH   r  re   c                    s   ˆ S r9   r   r:   ©rm  r   r    r+   Â  r,   z3test_consistent_coerce_for_shapes.<locals>.<lambda>r   r   c                    s   g | ]}ˆ ‘qS r   r   rò   rn  r   r    rI  Ã  r,   z5test_consistent_coerce_for_shapes.<locals>.<listcomp>rg  )rm  r3   r   r   r   rn  r    Ú!test_consistent_coerce_for_shapes¼  s    ro  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 g d¢g d¢d�S )Nrs   )ÚtestÚotherr  rU   ©r   r:   r   r   r    r+   Ì  r,   z'test_consistent_names.<locals>.<lambda>r   r   rp  rq  r  r  re   c                 S   s   t ddgddgd�S )Nr   r.   rp  rq  rU   rr  r:   r   r   r    r+   Ó  r,   )r   Úrenamer   r=   r“   r   r   r    Útest_consistent_namesÇ  s    ÿ
ÿrt  c                 C   s:   | }|j dd„ ddd�}| ¡ }g d¢|_t ||¡ d S )Nc                 S   s   g d¢S rŒ   r   r:   r   r   r    r+   Ý  r,   z"test_result_type.<locals>.<lambda>r   r]   r^   ©r   r   r.   ©r   r’   rW   r   r=   r“   r   r   r    Útest_result_typeØ  s
    
rw  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 re  r   r:   r   r   r    r+   ç  r,   z/test_result_type_shorter_list.<locals>.<lambda>r   r]   r^   r   r"   r   rv  r“   r   r   r    Útest_result_type_shorter_listã  s
    
rx  c                 C   s0   | }|j dd„ ddd�}| ¡ }t ||¡ d S )Nc                 S   s   g d¢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    ry  c                    s<   | }g d¢‰ |j ‡ fdd„ddd�}| ¡ }t ||¡ d S )N©rq  r  r8  c                    s   t g d¢ˆ d�S ©Nrs   rU   rr  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    ÿr|  c                 C   s.   | }|j dd„ dd�}| ¡ }t ||¡ d S )Nc                 S   s   t g d¢| jd�S r{  )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    r}  c                    s@   | }g d¢‰ |j ‡ fdd„dd�}| ¡ }ˆ |_t ||¡ d S )Nrz  c                    s   t g d¢ˆ d�S r{  rr  r:   re   r   r    r+     r,   z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   rv  r“   r   re   r    Ú*test_result_type_series_result_other_index  s    r~  Úboxc                 C   s   t | ƒS r9   )r†   r:   r   r   r    r+     r,   c                 C   s   t | ƒS r9   )r  r:   r   r   r    r+     r,   c                 C   s   t j| dd�S )Nr¦   ra   )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 re  r   r:   ©r  r   r    r+   "  r,   z,test_consistency_for_boxed.<locals>.<lambda>r   r   c                    s   g | ]}ˆ d dgƒ‘qS rf  r   rò   r€  r   r    rI  #  r,   z.test_consistency_for_boxed.<locals>.<listcomp>c                    s   ˆ ddgƒS re  r   r:   r€  r   r    r+   &  r,   r]   r^   r   r"   r   rM   re   )r   r   rõ   r   r   rs  r=   )r  rŽ   r3   r   r   r   r€  r    Útest_consistency_for_boxed  s    r�  c                 C   s<  | dv rdnd}t jdd��� t  |¡}t  |¡}| ¡ }|jt j| d�}t ||¡ |jt jg| d�}| ¡ }| dv r�t 	|j
dgg¡|_
nt 	|jdgg¡|_t ||¡ |jt jt jg| d�}t||g|d�}| dv röt 	|j
ddgg¡|_
nt 	|jddgg¡|_t ||¡ W d   ƒ n1 �s.0    Y  d S )	N¾   r   r   r   r   r   r   r   r   Úabsolute)r   r   Úabsr   r’   r   r   r=   r   Zfrom_productrW   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 )
Nrt   rM   r÷   rø   r   r-   r   r"   rÆ   )r   r‡   rD   r   r=   rq   r   r   r    Ú	test_demoQ  s    ÿr‡  c                  C   sl   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g d¢d�}t | |¡|¡ d S )Nrt   rM   r÷   rø   rk   ç      @r3  ç      @g      9@r   r"   )rø   r÷   rk   rÆ   )r   r‡   rD   r   rV   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   rD   r   r   r   )rE   r3   r   r   r   r   r    Ú!test_agg_with_name_as_column_nameh  s    rŒ  c                  C   s  t g d¢g d¢g d¢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 d   ƒ n1 sˆ0    Y  t 	||¡ tjtdd��& | g d¢  ddg¡}W d   ƒ n1 sØ0    Y  |g d¢  
ddg¡}t 	||¡ d S )Nrs   ©r¥   rÀ   ç      @©r¤   rà   ZbazrG   rH   rI   ©r   r"   r#   r$   r   ru   r¥   ç      @rà   Ú	foobarbazr5   r÷   rk   rU   ú&\['D'\] did not aggregate successfully©Úmatch)r$   r#   r"   r   )r   r	   r   rN   ZNaTr   Úassert_produces_warningÚFutureWarningrD   r=   Zreindex)Zmdfr   r   r   r   r    Ú"test_agg_multiple_mixed_no_warningx  s4    
üÿüù
ÿ,ÿ4r˜  c              	   C   s   | dv rdnd}|j |  ¡ d d…  ¡ \}}tj|j| d�|j| d�|j| d�gdd�}g d¢|_| dv rr|j	n|}|j
g d¢| 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
v �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
v �r|j	n|}t ||¡ d S )Nr‚  r   r   r.   r   )r   rø   rk   r   rk   rU   >   rW   r   rø   )ZaxesÚuniqueZsort_valuesrN   Úconcatr   rø   rk   rW   rp   rD   r   r=   r   rX   r   r   )r   r   r…  Zname1Zname2r   r   rP   r   r   r    Útest_agg_reducež  sf    


ýú
þûþÿþûþûøïr›  c                  C   sL  t g d¢g d¢g d¢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 d   ƒ n1 s¼0    Y  tg d¢g d¢d�}t ||¡ tj	t
dd�� |  dg¡}W d   ƒ n1 �s0    Y  t g d¢gdgg d¢d�}t ||¡ d S )Nrs   r�  r�  rG   rH   rI   r�  r÷   r   r¥   rà   rU   rð   úSelect only validr”  rk   )ru   r‘  r’  r  r“  )r   r	   rD   r   r   rW   r   r   r=   r–  r—  rq   r   r   r    Útest_nuiscance_columnsã  s:    
üÿ	
ý(ÿ,ÿr�  Úhowc                 C   s  t g d¢dtjdgg d¢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 )N©Nr.   rH   r¥   rŽ  ©r¤   Nrà   r  r   r‹  r.   r  rH   )r‹  r  Trù   )	r   r   rV   rn   r   r   r   r=   r‹  )rž  r3   r   r   Zresult1Zresult2r   r   r    Útest_non_callable_aggregates
  s0    ÿÿýÿr¡  c                 C   s|   t g d¢dtjdgg d¢dœƒ}t|| ƒd|d�}|dksB|d	krXt|jd |jd
�}nt|jd |jd
�}t 	||¡ d S )NrŸ  r¥   rŽ  r   r  r  r   r   r   rU   r   )
r   r   rV   rn   r   r/   rW   r   r   r   )rž  r   r3   r   r   r   r   r    Útest_size_as_str8  s    ÿr¢  c                  C   s†   t g d¢dtjdgg d¢dœƒ} dd„ }|  |¡}tddgdgd	d
ggg d¢d�}t ||¡ |  |g¡}| d¡j}t 	||¡ d S )N)r.   r.   rH   g      ø?r   r  c                 S   s   t |  ¡  ¡ ƒS r9   )r†   Zdropnar™  )Z	group_colr   r   r    rP   L  s    z&test_agg_listlike_result.<locals>.funcr.   rH   r¤   rà   rU   rP   )
r   r   rV   rD   r   r   r   Zto_framerp   r=   )r3   rP   r   r   r   r   r    Útest_agg_listlike_resultH  s     
 r£  zargs, kwargs)rw   rv   é   rf  )r   rc   r   ©rA   rc   rb   c                 C   sj   ddd„}t ddgddggƒ}| dkr4tdd	gƒ}ntd
dgƒ}|j|| g|¢R i |¤Ž}t ||¡ d S )NrH   c                 S   s   |   ¡ || |  S r9   ©rk   )r;   rA   rc   rd   r   r   r    rÞ   f  s    ztest_agg_args_kwargs.<locals>.fr   r.   r-   r   r‰  g      @rˆ  g       @)rH   )r   r   rD   r   r   )r   r¿   ÚkwargsrÞ   r3   r   r   r   r   r    Útest_agg_args_kwargsX  s    
r¨  Únum_cols)r.   rH   rt   c                 C   sJ   t  g d¢¡}| ¡ }td|t| ƒd�}| dd„ ¡ |j|jksFJ ‚d S )N)z
1950-06-30z
1952-10-24z
1953-05-29r   rð   c                 S   s   | S r9   r   r:   r   r   r    r+   {  r,   z,test_frequency_is_original.<locals>.<lambda>)rN   ZDatetimeIndexr’   r   r‡   r   r  )r©  r   Úoriginalr3   r   r   r    Útest_frequency_is_originalu  s
    r«  c                  C   s`   t ddd�t ddd�t ddd�g} tg d¢| d�}|jdd	„ d
d�}t| | d�}t ||¡ d S )Nz2019-03-15 12:34:31.909000+0000r6   r7   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000ru  )rE   r   c                 S   s   | j S r9   rê   r:   r   r   r    r+   ˆ  r,   z.test_apply_datetime_tz_issue.<locals>.<lambda>r   r   )r   rE   )r   r   r   r   r   r   )Z
timestampsr3   r   r   r   r   r    Útest_apply_datetime_tz_issue  s    


ýr¬  r3   rA   rd   r   rM   Úmethod)r÷   rø   rk   c                 C   sf   t jtdd��8 t| ddg |ƒƒ }t| ddg |ƒƒ }W d   ƒ n1 sL0    Y  t  ||¡ d S )Nrœ  r”  r   r"   )r   r–  r—  rn   r   )r3   r­  Znone_in_first_column_resultZnone_in_second_column_resultr   r   r    Ú=test_consistency_of_aggregates_of_columns_with_missing_valuesŽ  s    2r®  r  c                 C   s:   t d| ggddgd�}| dd„ ¡}|j}t ||¡ d S )Nr¥   rA   rc   re   c                 S   s   | j S r9   ra   r:   r   r   r    r+   �  r,   z"test_apply_dtype.<locals>.<lambda>)r   r   r  r   r   )r  r3   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 ||¡ | slt ||¡ nt ||¡ d S )	Néd   éÈ   r¥  c                 S   s*   | j }| jd  d7  < | j |us&J ‚| S )NrA   r   )Z_mgrrX   )r0  Zmgrr   r   r    rP   ¨  s    z!test_apply_mutating.<locals>.funcrA   r   r   )r   r‡   r’   r   r   r=   )Zusing_array_managerr3   Zdf_origrP   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.   rH   r-   rt   ru   rv   rw   rx   rÊ   rA   rc   re   c                 S   s   g S r9   r   r:   r   r   r    r+   À  r,   z.test_apply_empty_list_reduce.<locals>.<lambda>r`   r}   r¥  ra   )r   r   r   rh   r   r   rq   r   r   r    Útest_apply_empty_list_reduce¼  s    .r³  c                  C   s`   t ddggd ddgd�} |  ddd	„ dd	„ g¡}t g d
¢g d¢dœg d¢d�}t ||¡ d S )Nr-   rx   rH   r   r"   re   rk   c                 S   s   |   ¡ S r9   r¦  r:   r   r   r    r+   È  r,   z,test_apply_no_suffix_index.<locals>.<lambda>)rG  rG  rG  )é   r´  r´  rM   )rk   r+   r+   rU   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 ry   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 g d¢dd�} dd„ }d|ddg}tjtd	d
�� |  |¡}W d   ƒ n1 sR0    Y  t dtjddgg d¢g d¢g d¢dœg d¢d�}t ||¡ d S )N))Ú1r   r   r   )Ú2r.   r   r   )Ú3rH   r   r   )Ú4r-   rt   r-   )Ú5rt   ru   ru   )Ú6ru   rv   rv   )ÚitemZatt1Zatt2Zatt3re   c                 S   s   |   ¡ d S r    r¦  rz   r   r   r    r¤   ç  s    z/test_aggregation_func_column_order.<locals>.foork   r‹  r÷   z)\['item'\] did not aggregate successfullyr”  Z123456ru   r¸  )g      5@g      %@r‘  r¥   )g      2@g      "@r‘  r3  )g      1@g      !@r‘  r3  )rk   r¤   r‹  r÷   rU   )r   r   r–  r—  rD   r   rV   r=   )r3   r¤   Zaggsr   r   r   r   r    Ú"test_aggregation_func_column_orderØ  s&    ÷ÿ(üù	r¿  c                  C   sL   t g d¢g d¢dœƒ} | ddg jdd„ dd�}tg d	¢ƒ}t ||¡ d S )
Nru  rs   r¥  rA   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   r.   r-   r¨   rq   r   r   r    Útest_apply_getitem_axis_1û  s    rÀ  c                  C   sV   dd„ } t dg d¢iƒ}tjtdd�� | | g¡ W d   ƒ n1 sH0    Y  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>.foorA   rs   rÁ  r”  )r   r   r–  ÚUserWarningrD   )r¤   r3   r   r   r    Ú*test_nuisance_depr_passes_through_warnings  s    rÄ  )ƒr   r®   Únumpyr   ZpytestZpandas.core.dtypes.dtypesr   ZpandasrN   r   r   r   r   r	   Zpandas._testingZ_testingr   Zpandas.tests.frame.commonr
   r!   r4   r?   ÚmarkZparametrizerF   rO   r   r   rS   rZ   r\   rj   ro   rr   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†   Úroundr  r  r  r  r$  r(  r*  r+  r-  r4  r5  r<  rA  rD  rE  rH  rJ  rK  rO  rV  rW  rZ  r_  rd  rh  rj  rl  ro  rt  rw  rx  ry  r|  r}  r~  r�  r†  r‡  rŠ  rŒ  r˜  r›  r�  r¡  r¢  r£  r¨  r«  r¬  r®  rV   r¯  r²  r³  rµ  r·  r¿  rÀ  rÄ  r   r   r   r    Ú<module>   s^  
þ


	


		ÿ			#4	þ*ûþ

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




ý
&&E'
-

ùþ	*			#