a
    d                     @   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ejdd d	g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/d0 Z'd1d2 Z(d3d4 Z)ejdd d	gd5d6 Z*ejdd d	gd7d8 Z+d9d: Z,ejdd d	gd;d< Z-d=d> Z.d?d@ Z/dAdB Z0ej1dCejdDdEdFgejjddGdH dIdH gdJdKgdLejdMdNdOgejdd d	gdPdQ Z2dRdS Z3dTdU Z4dVdW Z5dXdY Z6dZd[ Z7d\d] Z8d^d_ Z9d`da Z:dbdc Z;ddde Z<dfdg Z=dhdi Z>djdk Z?dldm Z@dndo ZAe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 ZBdzd{ ZCd|d} ZDd~d ZEejdddgdd ZFdd ZGdd ZHdd ZIdd ZJejddddejKgdd ZLdd ZMejddddH ide	ddvdwgifdddH idOe	ddvdwgifdddH ideddwifdddH idOeddwifddejNddH gide	dg dig ddfddejNddH gidOe	dg dig ddfdddH ide	dd	d	gifdddH idOedd	ifgdd ZOejddddH idddH iddejNddH gidddH igdd ZPdd ZQdd ZRdd ZSejdddH gde	ddvdwgifddH gdOe	ddvdwgifddH gde	ddwgidHgdfddH gdOe	ddwgidHgdfdejNddH gde	dg dig ddfdejNddH gdOe	dg dig ddfddH ddH gde	dvdwgdwdwggddgdHdHggdfddH ddH gdOe	ddvdwgidHdHgfgdd ZTejdddH gddH gdejNddH gddH ddH ggddƄ ZUddȄ ZVddʄ ZWdd̄ ZXdd΄ ZYejdd	dvgddЄ ZZdd҄ Z[ejdg dԢd	dvggddք Z\dd؄ Z]ddڄ Z^dd܄ Z_ddބ Z`dd Zadd Zbdd ZcejjdddH ddH ddH gg ddLdd Zddd Zedd Zfdd Zgdd Zhdd Zidd Zjdd Zkdd Zlejdddgdd Zmejdddgdd  Zndd Zoejdd d	gejddi fdi fdi fdddvifdd	dvd	fddvd	d	fdd	dvdwd
fgdd Zpejdg ddd Zqdd Zrejde	ddgddgdgejdg ddd Zsejdd	ddNdejKgdd Ztdd  Zud!d" Zvd#d$ Zwd%d& Zxd'd( Zyd)d* Zzd+d, Z{d-d. Z|d/d0 Z}d1d2 Z~ejd3g dԢg d4d	g dԢd	gd	fg d5g d4d	dvgd	gd	fgd6d7 Zd8d9 Zd:d; Zd<d= Zd>d? Zd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sJ |j| ju sJ W d    n1 s0    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   l/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pandas/tests/apply/test_frame_apply.py
test_apply   s    
r    r   r   c                 C   s,   | j dd |dd}| d }t|| d S )Nc                 S   s   | | S Nr   xyr   r   r   <lambda>*       z!test_apply_args.<locals>.<lambda>r   )argsr   r   r   assert_frame_equalr   r   r   r   r   r   r   test_apply_args(   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astype)tsr   r   r   r%   2   r&   z-test_apply_categorical_func.<locals>.<lambda>)      r1   r2   )r   r   shape
isinstancedtyper   dfr   r   r   r   test_apply_categorical_func/   s
    r>   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 r!   r   r#   r   r   r   r%   <   r&   z*test_apply_axis1_with_ea.<locals>.<lambda>r   r   r   r   r   r   r*   r   r   r   r   r   test_apply_axis1_with_ea9   s    rF   zdata, dtyper   Nr?   r@   rA   c                 C   s<   t | g| ggddg|d}|jdd dd}t|| d S )Na)r   r;   c                 S   s   | S r!   r   rC   r   r   r   r%   G   r&   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   r   aggr   r*   )datar;   r   r   r   r   r   test_agg_axis1_duplicate_index@   s    rL   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   rC   r   r   r   r%   T   r&   z/test_apply_mixed_datetimelike.<locals>.<lambda>r   r   )	r   r   pdZto_timedeltar   aranger   r   r*   rE   r   r   r   test_apply_mixed_datetimelikeK   s    
rV   funcc                 C   s   t  }|| }|jsJ d S r!   )r   r   empty)rW   empty_framer   r   r   r   test_apply_emptyX   s    
rZ   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   rC   r   r   r   r%   c   r&   z(test_apply_float_frame.<locals>.<lambda>r   c                 S   s   |   S r!   r[   rC   r   r   r   r%   h   r&   r   r   )	r   r   r   nancolumnsr   r   locr   )r   Zno_rowsr   r   no_colsr   r   r   test_apply_float_framea   s    ra   c                  C   s.   t dgd} | jdd dd}t||  d S )NrH   r\   c                 S   s   | d S )NrH   r   rC   r   r   r   r%   p   r&   z/test_apply_empty_except_index.<locals>.<lambda>r   r   r   r   r   r*   rE   r   r   r   test_apply_empty_except_indexm   s    rc   c                  C   s   t  } g }| j|jddd}t||  | j|jddd}tg tjd}t|| t g dd}|j|jddd}t|| |j|jddd}tg tjd}t|| |g ksJ d S )Nr   expandr   result_typereducer;   rH   bcr^   )	r   r   appendr   r*   r   r   float64r   )rY   r#   r   r   Zempty_with_colsr   r   r   test_apply_with_reduce_emptyt   s    ro   )sumprodanyr   c                 C   sL   t g dd}|tt| }t||  }| dv r<|t}t|| d S )Nri   rl   )rp   rq   )r   r   getattrr   r5   floatr   r   )rW   r=   r   r   r   r   r   test_apply_funcs_over_empty   s    
ru   c                  C   sX   t g dd} |  }td| jd}t|| | j }tg tjd}t|| d S )Nri   rl   r   r\   rh   )	r   Znuniquer   r^   r   r   Tr   rn   r=   r   r   r   r   r   test_nunique_empty   s    
rx   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   r8   rN   )r7         )      	   )rH   rH   rk   r\   c                 S   s   | d S Nr   r   rQ   r   r   r   r%      r&   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r7   r|   c                 S   s   | d S r   r   r   r   r   r   r%      r&   r   )r   r   r   r   r   rv   rw   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rf   r\   )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   re   r   c                    s   i | ]
}| qS r   r   .0rk   mr   r   
<dictcomp>   r&   z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>)r   r   r   r   r^   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lenr^   rC   r   r   r   r%      r&   z4test_apply_broadcast_lists_columns.<locals>.<lambda>r   r   re   rn   )r;   r   r^   )	r   r   r   r   r^   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   rC   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>rn   )r;   r   )	r   r   r   r   r   r   r^   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 Nry   r   rC   r   r   r   r%      r&   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r   re   r)   )int_frame_const_colr=   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 )Nry   abcr\   )r   r   rC   r   r   r   r%      r&   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r   re   r   copyr   r*   r   r=   r   r   r   r   r   'test_apply_broadcast_series_lambda_func   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 Nr   r:   r   ndarrayndimrC   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   rC   r   r   r   r%      r&   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>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 Nr8   r   rC   r   r   r   r%      r&   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>T)r   r8   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   rC   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      ?rS   r   r   r   int64rh   r\   )
r   r   r   r   r   r]   rT   Indexr   r   rw   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   rS   c                 S   s   | d S )Nr   r   rC   r   r   r   r%     r&   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   r\   c                 S   s   | d S )Nr-   r   rC   r   r   r   r%   #  r&   r   r   r   r   r   rw   r   r   r   &test_apply_mixed_dtype_corner_indexing  s    r   zignore::RuntimeWarningaxr   r^   c                 C   s   | S r!   r   rC   r   r   r   r%   +  r&   r%   c                 C   s   |   S r!   r[   rC   r   r   r   r%   +  r&   identityr   )idsr   TFc           	      C   s   t f i | g di}tjddz |tjg dd}t|tj }|j|||d}|r||}t|tspJ |j	|u sJ nt|t sJ W d    n1 s0    Y  d S )Nri   r
   r   Zf8rh   r   )
r   r   r   arrayr:   r   r   Z_get_agg_axisr   r   )	r   rW   r   r   r=   Ztest_resZis_reductionr   Zagg_axisr   r   r   test_apply_empty_infer_type(  s    
r   c                  C   s2   t g dd} | jdd dd}t|t s.J d S )Nri   r\   c                 S   s   |   S r!   r[   rC   r   r   r   r%   A  r&   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r   r   )r   r   r:   )r`   r   r   r   r   %test_apply_empty_infer_type_broadcast?  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_someF  s    z4test_apply_with_args_kwds_add_some.<locals>.add_somer8   r   c                 S   s   | d S r   r   rC   r   r   r   r%   J  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_someE  s    
r   c                 C   s6   ddd}| j |dd}|  dd }t|| d S )	Nr   c                 S   s   |   | S r!   r[   r   r   r   r   agg_and_addO  s    z:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addr8   r   c                 S   s   |   d S r   r[   rC   r   r   r   r%   S  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_addN  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_divideX  s    zJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide)r8   r8   )r(   r   c                 S   s   | d d S )N       @r   rC   r   r   r   r%   \  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_divideW  s    
r   c                 C   s   |  t}t||  d S r!   )r   r   r   r*   )r   r   r   r   r   test_apply_yield_list`  s    
r   c                 C   sH   t j| jd d d| jdf< | d}| jt jdd}t|| d S )Nr8   r   r   r   )	r   r]   ilocr^   Zget_locr   r   r   r   )r   r   r   r   r   r   test_apply_reduce_Seriese  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   r8   rN   r7   r1   r2   i0i1r^   r   r   r   )r   r   r\   r0   )r   r   dictr   r^   r   r   r   )rK   r   r   r   r   r   test_apply_reduce_to_dictl  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 )
Nr8   )   
   r   r   c                 S   s   i | ]\}}||  qS r   describer   ivr   r   r   r   }  r&   z2test_apply_differently_indexed.<locals>.<dictcomp>rl   r   c                 S   s   i | ]\}}||  qS r   r   r   r   r   r   r     r&   )r   r   randomdefault_rngstandard_normalr   r   r   itemsr^   r   r*   rv   r   rw   r   r   r   test_apply_differently_indexedy  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   ABC02   )r   YUM0r   )r   DEF0r   )r8   ABC1r   )r8   YUM1r   )r8   DEF1r   )rH   marketpositionrl   c                 S   s   | d S )Nr   r   )rr   r   r   f  s    ztest_apply_bug.<locals>.fr   r   i  r   r   r8   r   r   rN   r   r7   r   rz   r   r{   r   )r   r   r   r   r   )Z	positionsr   r   r   r   r   r   test_apply_bug  s0    r   c               	   C   sn   t g dg dg dtjddtjddtjddd} | jdd 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   r8      )r   r-   r.   r/   EFc                 S   s   | S r!   r   rC   r   r   r   r%     r&   z,test_apply_convert_objects.<locals>.<lambda>r   r   )r   r   r   r   r   r   r   r*   rE   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 r!   namerC   r   r   r   r%     r&   z(test_apply_attach_name.<locals>.<lambda>r\   )r   r   r^   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   rC   r   r   r   r%     r&   z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   r\   )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   rC   r   r   r   r%     r&   z6test_apply_attach_name_non_reduction.<locals>.<lambda>r   r   r^   )	r   r   r   Ztiler^   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   rC   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   r^   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)rH   rH   rj   )rk   r   r   r   r8   rN   r7   rz   r{   col1col2r   c                 S   s   t t| t| dS )N)minmax)r   r   r   rC   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   rQ   r   r   r   r   r   test_apply_multi_index  s
    $$r   z	df, dictsr   r   Zspameggs)r   r   r8   rN   c                 C   sV   dd }| j |dd}| j |dd}|  |}t|| t||  t|| d S )Nc                 S   s   |   S r!   )to_dictrC   r   r   r   r%     r&   z!test_apply_dict.<locals>.<lambda>rg   r   rd   )r   r   r   r*   )r=   ZdictsfnZreduce_trueZreduce_falseZreduce_noner   r   r   test_apply_dict  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-01rN   zEurope/Brussels)rP   rB   c                 S   s   | S r!   r   rC   r   r   r   r%     r&   z,test_apply_non_numpy_dtype.<locals>.<lambda>c                 S   s   | t d S )NZ1day)rT   Z	TimedeltarC   r   r   r   r%     r&   z
2015-01-02)r   r   r   r   r*   rw   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  )rH   rj   rk   rH   r3   rh   c                 S   s   | S r!   r   rC   r   r   r   r%   (  r&   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>rb   r<   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   r8   rN   rH   rl   r   r   rI   r=   r   r   r   r   r   test_apply_dup_names_multi_agg,  s    r  opr   rJ   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 )Nr8   r   r.   r-   r   rowr   r   r   
apply_list8  s    z3test_apply_nested_result_axis_1.<locals>.apply_list)r7   r7   ABCDrl   r   r   )        r  r  )r   r   zerosr   rs   r   r   r   )r  r
  r=   r   r   r   r   r   test_apply_nested_result_axis_15  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2020r@   rA   zdatetime64[ns, UTC]rh   c                 S   s   | S r!   r   rC   r   r   r   r%   H  r&   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   s   |   S r!   )r   rC   r   r   r   r%   J  r&   rD   rE   r   r   r   %test_apply_noreduction_tzaware_objectC  s    r  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 )	NrH   ry   c                    s     | j d S r!   rm   r   r  namesr   r   reducing_functionT  s    z8test_apply_function_runs_once.<locals>.reducing_functionc                    s     | j | S r!   r  r  r  r   r   non_reducing_functionW  s    z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r   r   r=   r  r  rW   r   r  r   test_apply_function_runs_onceN  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 )
NrH   ry   c                    s     |  d S r!   extendr  r   r   r   r  h  s    z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                    s     |  | S r!   r  r  r  r   r   r  k  s    
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r   r   )r   r   r   rH   Zto_listr  r   r  r   !test_apply_raw_function_runs_onceb  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   efghcolrl   )r^   r;   c                 S   s
   |  dS )Nobjectr4   rC   r   r   r   r%   |  r&   z-test_apply_with_byte_string.<locals>.<lambda>)r   r   r   r  r   r   r*   r  r   r   r   test_apply_with_byte_stringv  s    r  val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cderH   r3   rh   c                    s   |  kS r!   r   rC   r  r   r   r%     r&   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t |rtjn| kqS r   )rT   Zisnullr   r]   )r   r#   r!  r   r   
<listcomp>  r&   z1test_apply_category_equalness.<locals>.<listcomp>r   )r   r]   r   rH   r   r   r   r   )r  Z	df_valuesr=   r   r   r   r!  r   test_apply_category_equalness  s    r#  c                  C   sZ   t tjdd} | jtjjddj}|dks6J | jtjjddj}|dksVJ d S )Nr8   )r   r8   r   r   )r{   r8   )r   r   r   r   r   Zfftr9   Zrfftr<   r   r   r   test_infer_row_shape  s
    r$  zops, by_row, expectedrH   c                 C   s   | d S r   r   rC   r   r   r   r%     r&   compatc                 C   s   | d S r   r   rC   r   r   r   r%     r&   c                 C   s   |   S r!   rp   rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   rp   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   )rN   rN   rN   )rp   rp   r%   r\   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   dS r   r   rC   r   r   r   r%     r&   c                 C   s   dS r   r   rC   r   r   r   r%     r&   c                 C   s.   t dddgi}|j| |d}t|| d S NrH   r   r8   by_rowr   r   r   Zassert_equalopsr)  r   r=   r   r   r   r   test_dictlike_lambda  s    r-  r,  c                 C   s   | d S r   r   rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   dS r   r   rC   r   r   r   r%     r&   c                 C   sP   t dddgi}tjtdd |j| dd W d    n1 sB0    Y  d S NrH   r   r8   zby_row=True not allowedmatchTr(  r   pytestraises
ValueErrorr   r,  r=   r   r   r   test_dictlike_lambda_raises  s    r6  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   r8   rH   rj   rl   c                 S   s   d| d | d  iS NrQ   rH   rj   r   rC   r   r   r   r%     r&   z,test_with_dictlike_columns.<locals>.<lambda>r   c                 S   s   g | ]}d diqS )rQ   rN   r   r   r   r   r   r"    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 r7  r   rC   r   r   r   r%     r&   c                 S   s   d| iS )NrQ   r   rC   r   r   r   r%     r&   rQ   rN   )r   r   r   r   r   r   r   rw   r   r   r   test_with_dictlike_columns  s    r:  c                  C   sd   t  } g d| d< g d| d< tjg dd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:09T)Zdayfirstdatec                 S   s   i S r!   r   rC   r   r   r   r%     r&   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rT   to_datetimer   r   r   r   rw   r   r   r   (test_with_dictlike_columns_with_datetime  s    
rA  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   r8   rH   rj   rl   c                 S   s   d| d | d  iS r7  r   rC   r   r   r   r%     r&   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>rd   re   rQ   rN   r8  r9  r   c                 S   s   d| d | d  iS r7  r   rC   r   r   r   r%     r&   )r   r   r   r*   r   rw   r   r   r   %test_with_dictlike_columns_with_infer  s    rB  c                 C   s   | d S r   r   rC   r   r   r   r%     r&   )rH   r%   c                 C   s   | d S r   r   rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%      r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%     r&   c                 C   s   | d S r   r   rC   r   r   r   r%     r&   c                 C   s   dS NrN   r   rC   r   r   r   r%     r&   rl   c                 C   s   dS r   r   rC   r   r   r   r%     r&   c                 C   s   dS rC  r   rC   r   r   r   r%     r&   c                 C   s.   t dddgi}|j| |d}t|| d S r'  r*  r+  r   r   r   test_listlike_lambda  s    rD  c                 C   s   | d S r   r   rC   r   r   r   r%   !  r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%   "  r&   c                 C   s   |   S r!   r&  rC   r   r   r   r%   #  r&   c                 C   s   | d S r   r   rC   r   r   r   r%   $  r&   c                 C   s   dS rC  r   rC   r   r   r   r%   $  r&   c                 C   sP   t dddgi}tjtdd |j| dd W d    n1 sB0    Y  d S r.  r1  r5  r   r   r   test_listlike_lambda_raises  s    rE  c                  C   s   t ttjd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 )Nr8   r7   )rH   r   Zofwordsz
2016-10-01H)rP   freq)rH   rj   r6   rH   rj   r   r   c                 S   s   g | ]}|d d qS rG   r   r   r   r   r   r"  9  r&   z.test_with_listlike_columns.<locals>.<listcomp>r6   c                 S   s   g | ]}|d d qS rG   r   r   r   r   r   r"  =  r&   )r   r   r   r   r   r   r   r   tupler   r   r   rw   r   r   r   test_with_listlike_columns.  s    rJ  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 )NrH   rj   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   r"  F  r&   zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>r#   r   r  r   r  r   r%   F  r&   z;test_with_listlike_columns_returning_list.<locals>.<lambda>r   r   r\   )r   r   r   from_tuplesr   r   r   r   rw   r   r   r   )test_with_listlike_columns_returning_listA  s
    *rO  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!   rP  rQ  r  r   r   r   r%   X  r&   z1test_infer_output_shape_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}|j |jfqS r   rR  r   r   r   r   r"  Y  r&   z3test_infer_output_shape_columns.<locals>.<listcomp>r   r   r   r   r   r   r   rw   r   r   r   test_infer_output_shape_columnsK  s    
rT  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 )Nr8   )r{   rN   r   r-   r.   rl   c                 S   s   g dS r   r   rC   r   r   r   r%   d  r&   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}g d qS )ry   r   r   r   r   r   r"  e  r&   z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S   s   ddgS Nr   r8   r   rC   r   r   r   r%   h  r&   c                 S   s   g | ]}d dgqS r   r8   r   r   r   r   r   r"  i  r&   
r   r   r   r   r   r   r   r   r   r   rw   r   r   r   (test_infer_output_shape_listlike_columns]  s    rY  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 )NrH   ry   r   r\   c                    s
   t  S r!   r   Zonesr  r!  r   r   r%   r  r&   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>r   r   c                    s   g | ]}t  qS r   rZ  r   r!  r   r   r"  s  r&   zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r   r   r   r   r   r   r   )r  r=   r   r   r   r!  r   0test_infer_output_shape_listlike_columns_np_funcm  s     r[  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)r~   rz   r7   rN   )rz   rN   r7   r8   )r   r8   rN   r7   )rH   rj   rk   r   c                 S   s   dS )NrW  r   rC   r   r   r   fun  s    zDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.funr   r   c                 S   s   g | ]}d qS )rW  r   r   r   r   r   r"    r&   zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>rS  )r=   r\  r   r   r   r   r   7test_infer_output_shape_listlike_columns_with_timestampw  s    r]  lstry   c                    s^   t tjddg dd}|j fdddd}t fd	d
| D }t	|| d S )Nr8   )r7   rN   rU  rl   c                    s    S r!   r   rC   r^  r   r   r%     r&   z3test_consistent_coerce_for_shapes.<locals>.<lambda>r   r   c                    s   g | ]} qS r   r   r   r_  r   r   r"    r&   z5test_consistent_coerce_for_shapes.<locals>.<listcomp>rX  )r^  r=   r   r   r   r_  r   !test_consistent_coerce_for_shapes  s    r`  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 )Nry   )testothercolsr\   r   rC   r   r   r   r%     r&   z'test_consistent_names.<locals>.<lambda>r   r   ra  rb  rc  rU  rl   c                 S   s   t ddgddgdS )Nr   r8   ra  rb  r\   rd  rC   r   r   r   r%     r&   )r   renamer   r*   r   r   r   r   test_consistent_names  s    
rf  c                 C   s:   | }|j dd ddd}| }g d|_t|| d S )Nc                 S   s   g dS r   r   rC   r   r   r   r%     r&   z"test_result_type.<locals>.<lambda>r   rd   re   r   r   r8   r   r   r^   r   r*   r   r   r   r   test_result_type  s
    
ri  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 rV  r   rC   r   r   r   r%     r&   z/test_result_type_shorter_list.<locals>.<lambda>r   rd   re   r   r-   r   rh  r   r   r   r   test_result_type_shorter_list  s
    
rj  c                 C   s0   | }|j dd ddd}| }t|| d S )Nc                 S   s   g dS r   r   rC   r   r   r   r%     r&   z,test_result_type_broadcast.<locals>.<lambda>r   r   re   r   r   r   r   r   test_result_type_broadcast  s    rk  c                    s<   | }g d |j  fddddd}| }t|| d S )Nrb  r  r  c                    s   t g d dS Nry   r\   rd  rC   rl   r   r   r%     r&   z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r   re   r   r   r   rl   r   &test_result_type_broadcast_series_func  s    rn  c                 C   s.   | }|j dd dd}| }t|| d S )Nc                 S   s   t g d| jdS rm  )r   r   rC   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    ro  c                    s@   | }g d |j  fdddd}| } |_t|| d S )Nrl  c                    s   t g d dS rm  rd  rC   rl   r   r   r%     r&   z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   rh  r   r   rl   r   *test_result_type_series_result_other_index  s    rp  boxc                 C   s   t | S r!   )r   rC   r   r   r   r%     r&   c                 C   s   t | S r!   )rI  rC   r   r   r   r%     r&   c                 C   s   t j| ddS )Nr   rh   )r   r   rC   r   r   r   r%     r&   )r   rI  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 rV  r   rC   rq  r   r   r%     r&   z,test_consistency_for_boxed.<locals>.<lambda>r   r   c                    s   g | ]} d dgqS rW  r   r   rr  r   r   r"    r&   z.test_consistency_for_boxed.<locals>.<listcomp>c                    s    ddgS rV  r   rC   rr  r   r   r%     r&   rd   re   r   r-   r   rS   rl   )r   r   r   r   r   re  r*   )rq  r   r=   r   r   r   rr  r   test_consistency_for_boxed  s    rs  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 rt	|j
dgg|_
nt	|jdgg|_t|| |jt jt jg| d}t||g|d}| dv rt	|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_productr^   r   r	   )r   r   
other_axisZf_absZf_sqrtr   r   r   r   r   test_agg_transform   s.    

rx  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 )
Nrz   rS   r   r   r   r7   r   r-   r   )r   r   rJ   r   r*   rw   r   r   r   	test_demo%  s    ry  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 )Nrz   rS   r   r   rp         @r        @g      9@r   r-   )r   r   rp   r   )r   r   rJ   r   r]   r   r*   Zreindex_likerw   r   r   r   test_demo_dict_agg0  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   countr8   r   )r   rJ   r   r   r   )rK   r=   r   r   r   r   r   !test_agg_with_name_as_column_name<  s    r~  c                  C   s   t g dg dg dd} t ddgddgd	d
gdddgd}| ddg}t|| | g d ddg}|g d ddg}t|| d S )Nry   r   r         @r   r   ZbazrU  r   r{   r         @r   	foobarbazr   rp   r\   )r.   r-   r   )r   rJ   r   r*   Zreindex)mdfr   r   r   r   r   test_agg_multiple_mixedL  s$    	r  c                  C   s   t g dg dg dtdddd} d}tjt|d	 | d
dg W d    n1 s\0    Y  tjt|d	& | g d dd
g W d    n1 s0    Y  d S )Nry   r  r  rM   rN   rO   r   r-   r.   r/   does not support reductionr/  r   rp   )r/   r.   r-   r   )r   r   r2  r3  	TypeErrorrJ   )r  msgr   r   r   test_agg_multiple_mixed_raisesh  s    

,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 )Nrt  r   r   r8   r   )r   r   rp   r   rp   r\   >   r^   r   r   )ZaxesuniqueZsort_valuesrT   concatr   r   rp   r^   rv   rJ   r   r*   r   r_   r   r   )r   r   rw  Zname1Zname2r   r   rW   r   r   r   test_agg_reduce|  sf    



r  c                  C   s@  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|| d}t	j
t|d | d W d    n1 s0    Y  | g d d}tg dg dd}t|| d}t	j
t|d | dg W d    n1 s20    Y  d S )Nry   r  r  rM   rN   rO   r  r   r   r   r   r\   r   r  r/  rp   rU  )r{   r  r  )r   r   rJ   r   r   r^   r   r   r*   r2  r3  r  )r=   r   r   r  r   r   r   test_nuiscance_columns  s4    
	
(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Nr8   rN   r   r  r   Nr   rU  r   r}  r8   sizerN   )r}  r  Tr   )	r   r   r]   rs   r   r   r   r*   r}  )r  r=   r   r   Zresult1Zresult2r   r   r   test_non_callable_aggregates  s0    r  c                 C   st   t g ddtjdgg dd}t|| d|d}|dv rPt|jd	 |jd
}nt|jd |jd
}t	|| d S )Nr  r   r  r  rU  r  r   )r   r   r   r\   r   )
r   r   r]   rs   r   r9   r^   r   r   r   )r  r   r=   r   r   r   r   r   test_size_as_str  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)r8   r8   rN   g      ?r  rU  c                 S   s   t |   S r!   )r   Zdropnar  )Z	group_colr   r   r   rW   &  s    z&test_agg_listlike_result.<locals>.funcr8   rN   r   r   r\   rW   )
r   r   r]   rJ   r   r   r   Zto_framerv   r*   )r=   rW   r   r   r   r   r   test_agg_listlike_result"  s     
 r  zargs, kwargs)r}   r|      rW  r'   rj   r   rH   rj   ri   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 )NrN   c                 S   s   |   || |  S r!   r&  )r#   rH   rj   rk   r   r   r   r   @  s    ztest_agg_args_kwargs.<locals>.fr   r8   r7   r   r{  g      @rz  g       @)rN   r   r   rJ   r   r   )r   r(   kwargsr   r=   r   r   r   r   r   test_agg_args_kwargs2  s    
r  num_cols)r8   rN   rz   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 r!   r   rC   r   r   r   r%   U  r&   z,test_frequency_is_original.<locals>.<lambda>)rT   ZDatetimeIndexr   r   r   r   rH  )r  r   originalr=   r   r   r   test_frequency_is_originalO  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+0000r@   rA   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000rg  )rK   r   c                 S   s   | j S r!   r   rC   r   r   r   r%   b  r&   z.test_apply_datetime_tz_issue.<locals>.<lambda>r   r   )r   rK   )r   r   r   r   r   r   )Z
timestampsr=   r   r   r   r   r   test_apply_datetime_tz_issueY  s    


r  r=   rk   r   rS   method)r   r   rp   c                 C   sP   |dkrd}nd}t jt|d t| |  W d    n1 sB0    Y  d S )Nrp   z-can only concatenate str \(not "int"\) to strz4not supported between instances of 'str' and 'float'r/  )r2  r3  r  rs   )r=   r  r  r   r   r   test_mixed_column_raisesh  s
    r  r  r   c                 C   s:   t d| ggddgd}|dd }|j}t|| d S )Nr   rH   rj   rl   c                 S   s   | j S r!   rh   rC   r   r   r   r%   x  r&   z"test_apply_dtype.<locals>.<lambda>)r   r   Zdtypesr   r   )r  r=   r   r   r   r   r   test_apply_dtypet  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 )	Nd      r  c                 S   s*   | j }| jd  d7  < | j |us&J | S )NrH   r   )Z_mgrr_   )r	  Zmgrr   r   r   rW     s    z!test_apply_mutating.<locals>.funcrH   r   r   )r   r   r   r   r   r*   )Zusing_array_managerZusing_copy_on_writer=   Zdf_origrW   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   r8   rN   r7   rz   r{   r|   r}   r~   r   rH   rj   rl   c                 S   s   g S r!   r   rC   r   r   r   r%     r&   z.test_apply_empty_list_reduce.<locals>.<lambda>rg   r   r  rh   )r   r   r   r  r   r   rw   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 )Nr7   r~   rN   r   r-   rl   rp   c                 S   s   |   S r!   r&  rC   r   r   r   r%     r&   z,test_apply_no_suffix_index.<locals>.<lambda>)r   r   r   )   r  r  rS   )rp   r%   r%   r\   rb   )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 r   r   rC   r   r   r   r%     r&   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr   r   rw   r   r   r   test_apply_raw_returns_string  s    r  c                  C   sb   t g ddd} dd }d|ddg}| |}t g d	g d
g ddg dd}t|| d S )N))r   r   r   )r8   r   r   )rN   r   r   )r7   rz   r7   )rz   r{   r{   )r{   r|   r|   )Zatt1Zatt2Zatt3rl   c                 S   s   |   d S r   r&  r   r   r   r   sum_div2  s    z4test_aggregation_func_column_order.<locals>.sum_div2rp   r}  r   )g      5@g      %@r  r   )g      2@g      "@r  r  )g      1@g      !@r  r  )rp   r  r}  r   r\   rI   )r=   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 )
Nrg  ry   r  rH   c                 S   s   | j d | j d  S )Nr   r   )r   rC   r   r   r   r%     r&   z+test_apply_getitem_axis_1.<locals>.<lambda>r   r   )r   r8   r7   r   rw   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!)warningswarnrp   rC   r   r   r   expected_warning  s    
zDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warningrH   ry   r  r/  )r   r   assert_produces_warningUserWarningrJ   )r  r=   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g dd} | jtd	d
}tttd}t	|| | jtdd
}ttttd}t	|| d S )NrN   rQ  g      ?i  r   )r   r   ri   r\   r   r   )
r   rt   r   r   r]   r   typer   r   r   rw   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   r8   rN   r   r  c                 S   s   t | d | d S )NrH   rj   )r   rC   r   r   r   r%     r&   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   rh   )r   headr   r   r   rn   r   r   rw   r   r   r   test_apply_on_empty_dataframe  s    r  c                  C   sN   t ddgddgd} | dd }t ddggddggd}t|| d S )Nr   r8   rN   r  c                 S   s   | j gS r!   r  rC   r   r   r   r%     r&   z(test_apply_return_list.<locals>.<lambda>rb   rw   r   r   r   test_apply_return_list  s    r  ztest, constant)r   r   r   )r8   r8   r8   c                 C   s>   t | }t|ddgdd}ddd}||}t|| d S )NrH   rj   r  )rK   r   r;   r  r  r  )ra  ZconstantZdf1r   Zaggregationr   r   r   r   test_unique_agg_type_is_series  s
    	

r  c                  C   sr   t g dg dg dd} tg d}t| jdd| | jddd}t|| | dd}t|| d S )	N)r   r8   r   )r   r8   r   )r   r   r   rU  )TTFr   r   rr   )r   r   r   r   rr   r   r  r   r   r   /test_any_apply_keyword_non_zero_axis_regression  s    r  c                  C   s0  t dg di} ddd}ddd	}| j|dd
dd}| d }t|| d}tjt|d& | j||gdd
dd}W d    n1 s0    Y  t ddgddgddggddgddggd}t|| tjt|d& | jd|idd
dd}W d    n1 s0    Y  t g ddgd}t|| d S )Nr#   ry   r   r   c                 S   s   | | | S r!   r   )r#   rH   rk   r   r   r   foo10  s    z.test_agg_mapping_func_deprecated.<locals>.foo1r8   c                 S   s   | | | S r!   r   )r#   rj   rk   r   r   r   foo23  s    z.test_agg_mapping_func_deprecated.<locals>.foo2rN   r7   )rk   r|   z+using .+ in Series.agg cannot aggregate andr/  r}   r~   r   r  r  rl   )r8   rN   r7   )r   r   )r8   r   )r   rJ   r   r*   r  FutureWarning)r=   r  r  r   r   r  r   r   r    test_agg_mapping_func_deprecated,  s"    

4"6r  c                  C   s   t tdddddgd} tjtdd | tj}W d    n1 sN0    Y  t	d	d	d
t
d}t|| tjtdd | tjg}W d    n1 s0    Y  t d	d	d
dgd}t|| d S )Nr{   rN   r8   r   r-   rl   zusing DataFrame.stdr/  r   rS   rh   zusing Series.stdstdr\   )r   r   rU   Zreshaper   r  r  rJ   r  r   rt   r   r*   rw   r   r   r   test_agg_stdK  s    *,r  c                  C   sT   t g ddtjdgg dd} g d| _| ddi}| d  }t|| d S )	Nr  r   r  r  rU  )r   r   r.   r   r}  )r   r   r]   r^   rJ   r}  r   r   rw   r   r   r   (test_agg_dist_like_and_nonunique_columnsY  s    
r  )r   r  numpyr   r2  Zpandas.core.dtypes.dtypesr   ZpandasrT   r   r   r   r   r   Zpandas._testingZ_testingr   Zpandas.tests.frame.commonr	   r    markZparametrizer,   r>   rF   rL   rV   r   r   rZ   ra   rc   ro   ru   rx   r   r   r   r   r   r   r   r   r   r   r   r   r   r   filterwarningsr   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$  rp   r-  r6  r:  rA  rB  rD  rE  rJ  rO  rT  rY  r[  r]  r`  rf  ri  rj  rk  rn  ro  rp  rs  rx  ry  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   <module>   s  







		
			"4	*

	






	
"
	


	




%E$
,	(
		&&