o
    ҷh                     @   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X t | d }| t jd }t|| | t jd }t | d }||ks1J | jd }| jt jdd}t | |}|| |ksNJ |j| ju sVJ W d    d S 1 saw   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   V/var/www/html/venv/lib/python3.10/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 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sJ t|d jts#J t|d jts-J d S )	N)r   r   Br.   )Cr/   Dr0   c0c1c                 S   
   |  dS )Ncategoryastype)tsr   r   r   r&   2      
 z-test_apply_categorical_func.<locals>.<lambda>)      r2   r3   )r   r   shape
isinstancedtyper   dfr   r   r   r   test_apply_categorical_func/   s
   rA   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 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   rK   zdata, dtyper   NrB   rC   rD   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   rF   r"   r   rG   r   r   r   r&   G   rH   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   r   aggr   r+   )datar>   r   r   r   r   r   test_agg_axis1_duplicate_index@   s   rQ   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   rF   r"   r   rG   r   r   r   r&   T   rH   z/test_apply_mixed_datetimelike.<locals>.<lambda>r   r   )	r   r   pdto_timedeltar   aranger   r   r+   rJ   r   r   r   test_apply_mixed_datetimelikeK   s   
r\   funcc                 C   s   t  }|| }|jsJ d S r"   )r   r   empty)r]   empty_framer   r   r   r   test_apply_emptyX   s   
r`   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 r"   r   rG   r   r   r   r&   c   r'   z(test_apply_float_frame.<locals>.<lambda>r   c                 S   ra   r"   rb   rG   r   r   r   r&   h   r'   r   r   )	r   r   r   nancolumnsr   r   locr   )r   no_rowsr   r   no_colsr   r   r   test_apply_float_framea   s   ri   c                  C   s.   t dgd} | jdd dd}t||  d S )NrM   rc   c                 S      | d S )NrM   r   rG   r   r   r   r&   p   r'   z/test_apply_empty_except_index.<locals>.<lambda>r   r   r   r   r   r+   rJ   r   r   r   test_apply_empty_except_indexm   s   rl   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s\J d S )Nr   expandr   result_typereducer>   rM   bcre   )	r   r   appendr   r+   r   r   float64r   )r_   r$   r   r   empty_with_colsr   r   r   test_apply_with_reduce_emptyt   s   ry   )sumprodanyr   c                 C   sL   t g dd}|tt| }t||  }| dv r|t}t|| d S )Nrr   ru   )rz   r{   )r   r   getattrr   r7   floatr   r   )r]   r@   r   r   r   r   r   test_apply_funcs_over_empty   s   
r   c                  C   sX   t g dd} |  }td| jd}t|| | j }tg tjd}t|| d S )Nrr   ru   r   rc   rq   )	r   nuniquer   re   r   r   Tr   rw   r@   r   r   r   r   r   test_nunique_empty   s   
r   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;   rS   )r:         )      	   )rM   rM   rt   rc   c                 S   rj   Nr   r   rV   r   r   r   r&      r'   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r:   r   c                 S   rj   r   r   r   r   r   r   r&      r'   r   )r   r   r   r   r   r   r   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ro   rc   )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   rn   r   c                       i | ]}| qS r   r   .0rt   mr   r   
<dictcomp>       z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>)r   r   r   r   re   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                       t tt jS r"   )listrangelenre   rG   r   r   r   r&      r   z4test_apply_broadcast_lists_columns.<locals>.<lambda>r   r   rn   rw   )r>   r   re   )	r   r   r   r   re   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                    r   r"   )r   r   r   r   rG   r   r   r   r&      r   z2test_apply_broadcast_lists_index.<locals>.<lambda>r   r   c                    r   r   r   r   r   r   r   r      r   z4test_apply_broadcast_lists_index.<locals>.<dictcomp>rw   )r>   r   )	r   r   r   r   r   r   re   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      g dS Nr   r   rG   r   r   r   r&      r'   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r   rn   r*   )int_frame_const_colr@   r   r   r   r   %test_apply_broadcast_list_lambda_func   s   r   c                 C   0   | }|j dd ddd}| }t|| d S )Nc                 S   s   t g dtddS )Nr   abcrc   )r   r   rG   r   r   r   r&          z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r   rn   r   copyr   r+   r   r@   r   r   r   r   r   'test_apply_broadcast_series_lambda_func   s   r   c                 C      dd }| j ||dd d S )Nc                 S   "   t | tjsJ | jdksJ d S Nr   r=   r   ndarrayndimrG   r   r   r   _assert_raw      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   rG   r   r   r   r&      r9   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 Nr;   r   rG   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   r   )Nc                 S   r   r   r   rG   r   r   r   r     r   z4test_apply_raw_mixed_type_frame.<locals>._assert_rawTr   r   )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      ?rX   r   r   r   int64rq   rc   )
r   r   r   r   r   rd   rY   Indexr   r   r   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   rX   c                 S   rj   )Nr   r   rG   r   r   r   r&     r'   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   rc   c                 S   rj   )Nr.   r   rG   r   r   r   r&   #  r'   r   r   r   r   r   r   r   r   r   &test_apply_mixed_dtype_corner_indexing  s   r   zignore::RuntimeWarningaxr   re   c                 C   rF   r"   r   rG   r   r   r   r&   +  rH   r&   c                 C   ra   r"   rb   rG   r   r   r   r&   +  r'   identityr   )idsr   TFc           	      C   s   t di | g di}tjddF |tjg dd}t|tj }|j|||d}|r@||}t|ts8J |j	|u s?J nt|t sGJ W d    d S W d    d S 1 sZw   Y  d S )Nrr   r
   r   f8rq   r   r   )
r   r   r   arrayr=   r   r   _get_agg_axisr   r   )	r   r]   r   r   r@   test_resis_reductionr   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 )Nrr   rc   c                 S   ra   r"   rb   rG   r   r   r   r&   A  r'   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r   r   )r   r   r=   )rh   r   r   r   r   %test_apply_empty_infer_type_broadcast?  s   r   c                 C   6   ddd}| j |dd}|  dd }t|| d S )	Nr   c                 S   r!   r"   r   r$   howmuchr   r   r   add_someF     z4test_apply_with_args_kwds_add_some.<locals>.add_somer;   r   c                 S      | d S r   r   rG   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     
r   c                 C   r   )	Nr   c                 S   s   |   | S r"   rb   r   r   r   r   agg_and_addO     z:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addr;   r   c                 S   s   |   d S r   rb   rG   r   r   r   r&   S      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  r   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  r   zJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide)r;   r;   )r)   r   c                 S   s   | d d S )N       @r   rG   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 )Nr;   r   r   r   )	r   rd   ilocre   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   r;   rS   r:   r2   r3   i0i1re   r   r   r   )r   r   rc   r1   )r   r   dictr   re   r   r   r   )rP   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 )
Nr;   )   
   r   r   c                 S      i | ]	\}}||  qS r   describer   ivr   r   r   r   }      z2test_apply_differently_indexed.<locals>.<dictcomp>ru   r   c                 S   r   r   r   r   r   r   r   r     r   )r   r   randomdefault_rngstandard_normalr   r   r   itemsre   r   r+   r   r   r   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   )r;   ABC1r   )r;   YUM1r   )r;   DEF1r   )rM   marketpositionru   c                 S   rj   )Nr  r   )rr   r   r   f  r   ztest_apply_bug.<locals>.fr   r   i  r   r   r;   r   r   rS   r   r:   r   r   r   r   r   )r   r   r   r   r   )	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  r;      )r   r.   r/   r0   EFc                 S   rF   r"   r   rG   r   r   r   r&     rH   z,test_apply_convert_objects.<locals>.<lambda>r   r   )r   r   r   r   r   r   r   r+   rJ   r   r   r   test_apply_convert_objects  s   /r  c                 C   s.   |  dd }t| j| jd}t|| d S )Nc                 S      | j S r"   namerG   r   r   r   r&         z(test_apply_attach_name.<locals>.<lambda>rc   )r   r   re   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   r  r"   r  rG   r   r   r   r&     r  z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   rc   )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      t | jt| S r"   r   repeatr  r   rG   r   r   r   r&     r   z6test_apply_attach_name_non_reduction.<locals>.<lambda>r   r   re   )	r   r   r   tilere   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   r  r"   r  rG   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   re   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)rM   rM   rs   )rt   r   r   r   r;   rS   r:   r   r   col1col2r  c                 S   s   t t| t| dS )N)minmax)r   r#  r$  rG   r   r   r   r&         z(test_apply_multi_index.<locals>.<lambda>r#  r$  T
check_like)r   from_arraysr   r   r   r+   )r   rV   r   r   r   r   r   test_apply_multi_index  s
   $$r)  z	df, dictsr   r  spameggs)r   r   r;   rS   c                 C   sV   dd }| j |dd}| j |dd}|  |}t|| t||  t|| d S )Nc                 S   ra   r"   )to_dictrG   r   r   r   r&     r'   z!test_apply_dict.<locals>.<lambda>rp   r   rm   )r   r   r   r+   )r@   dictsfnreduce_truereduce_falsereduce_noner   r   r   test_apply_dict  s   
r2  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-01rS   zEurope/Brussels)rU   rE   c                 S   rF   r"   r   rG   r   r   r   r&     rH   z,test_apply_non_numpy_dtype.<locals>.<lambda>c                 S   s   | t d S )N1day)rY   	TimedeltarG   r   r   r   r&         z
2015-01-02)r   r   r   r   r+   r   r   r   r   test_apply_non_numpy_dtype  s   r7  c                  C   s2   t dg didd} | dd }t||  d S )Nr3  )rM   rs   rt   rM   r5   rq   c                 S   rF   r"   r   rG   r   r   r   r&   (  rH   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>rk   r?   r   r   r   #test_apply_non_numpy_dtype_category&  s   r8  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;   rS   rM   ru   r#  r   rN   r@   r   r   r   r   r   test_apply_dup_names_multi_agg,  s   r:  opr   rO   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_list8  s   "z3test_apply_nested_result_axis_1.<locals>.apply_list)r:   r:   ABCDru   r   r   )        r@  r@  )r   r   zerosr   r}   r   r   r   )r;  r>  r@   r   r   r   r   r   test_apply_nested_result_axis_15  s   rB  c                  C   sR   t dtdddgidd} | dd }t||  | d	d }t||  d S )
Nr   2020rC   rD   zdatetime64[ns, UTC]rq   c                 S   rF   r"   r   rG   r   r   r   r&   H  rH   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   ra   r"   )r   rG   r   r   r   r&   J  r'   rI   rJ   r   r   r   %test_apply_noreduction_tzaware_objectC  s   rD  c                     sh   t dg di} g   fdd} fdd}||fD ]} d d = | j|dd  t| jks1J qd S )	NrM   r   c                    s     | j d S r"   rv   r  r<  namesr   r   reducing_functionT     z8test_apply_function_runs_once.<locals>.reducing_functionc                    s     | j | S r"   rE  r<  rF  r   r   non_reducing_functionW  s   z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r   r   r@   rH  rJ  r]   r   rF  r   test_apply_function_runs_onceN  s   
rL  c                     sn   t dg di} g   fdd} fdd}||fD ]} d d = | j|ddd	  t| j ks4J qd S )
NrM   r   c                    s     |  d S r"   extendr<  r   r   r   rH  h  s   z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                    s     |  | S r"   rM  r<  rO  r   r   rJ  k  s   
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r   r   )r   r   r   rM   to_listrK  r   rO  r   !test_apply_raw_function_runs_onceb  s   
rQ  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colru   )re   r>   c                 S   r4   )Nobjectr6   rG   r   r   r   r&   |  r9   z-test_apply_with_byte_string.<locals>.<lambda>)r   r   r   rS  r   r   r+   r9  r   r   r   test_apply_with_byte_stringv  s   rT  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 )NrV  rW  cderM   r5   rq   c                    s   |  kS r"   r   rG   rU  r   r   r&     r'   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t |rtjn| kqS r   )rY   isnullr   rd   )r   r$   rY  r   r   
<listcomp>  s   $ z1test_apply_category_equalness.<locals>.<listcomp>r  )r   rd   r   rM   r   r   r   r   )rU  	df_valuesr@   r   r   r   rY  r   test_apply_category_equalness  s   r]  c                  C   sZ   t tjdd} | jtjjddj}|dksJ | jtjjddj}|dks+J d S )Nr;   )r   r;   r   r   )r   r;   )r   r   r   r   r   fftr<   rfftr?   r   r   r   test_infer_row_shape  s
   r`  zops, by_row, expectedrM   c                 C   r   r   r   rG   r   r   r   r&     r'   compatc                 C   r   r   r   rG   r   r   r   r&     r'   c                 C   ra   r"   rz   rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   rz   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   )rS   rS   rS   )rz   rz   r&   rc   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C      dS r   r   rG   r   r   r   r&     rH   c                 C   rc  r   r   rG   r   r   r   r&     rH   c                 C   .   t dddgi}|j| |d}t|| d S NrM   r   r;   by_rowr   r   r   assert_equalopsrg  r   r@   r   r   r   r   test_dictlike_lambda  s   rl  rk  c                 C   r   r   r   rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C   rc  r   r   rG   r   r   r   r&     rH   c                 C   R   t dddgi}tjtdd |j| dd W d    d S 1 s"w   Y  d S NrM   r   r;   zby_row=True not allowedmatchTrf  r   pytestraises
ValueErrorr   rk  r@   r   r   r   test_dictlike_lambda_raises     "rv  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;   rM   rs   ru   c                 S      d| d | d  iS NrV   rM   rs   r   rG   r   r   r   r&     r   z,test_with_dictlike_columns.<locals>.<lambda>r   c                 S   s   g | ]}d diqS )rV   rS   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   rx  ry  r   rG   r   r   r   r&     r   c                 S   s   d| iS )NrV   r   rG   r   r   r   r&     r'   rV   rS   )r   r   r   r  r   r   r   r   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)BBCNBCN24	publisher)z17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09T)dayfirstdatec                 S   s   i S r"   r   rG   r   r   r   r&     rH   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rY   to_datetimer   r   r   r   r   r   r   r   (test_with_dictlike_columns_with_datetime  s   
r  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;   rM   rs   ru   c                 S   rx  ry  r   rG   r   r   r   r&     r   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>rm   rn   rV   rS   rz  r{  r   c                 S   rx  ry  r   rG   r   r   r   r&     r   )r   r   r   r+   r   r   r   r   r   %test_with_dictlike_columns_with_infer  s   r  c                 C   r   r   r   rG   r   r   r   r&     r'   )rM   r&   c                 C   r   r   r   rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&      r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C   ra   r"   rb  rG   r   r   r   r&     r'   c                 C   r   r   r   rG   r   r   r   r&     r'   c                 C   rc  NrS   r   rG   r   r   r   r&     rH   ru   c                 C   rc  r   r   rG   r   r   r   r&     rH   c                 C   rc  r  r   rG   r   r   r   r&     rH   c                 C   rd  re  rh  rj  r   r   r   test_listlike_lambda  s   r  c                 C   r   r   r   rG   r   r   r   r&   !  r'   c                 C   ra   r"   rb  rG   r   r   r   r&   "  r'   c                 C   ra   r"   rb  rG   r   r   r   r&   #  r'   c                 C   r   r   r   rG   r   r   r   r&   $  r'   c                 C   rc  r  r   rG   r   r   r   r&   $  rH   c                 C   rm  rn  rq  ru  r   r   r   test_listlike_lambda_raises  rw  r  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 )Nr;   r:   )rM   r   ofwordsz
2016-10-01H)rU   freq)rM   rs   r8   rM   rs   r   r   c                 S      g | ]}|d d qS rL   r   r  r   r   r   r[  9      z.test_with_listlike_columns.<locals>.<listcomp>r8   c                 S   r  rL   r   r  r   r   r   r[  =  r  )r   r   r   r   r   r   r   r   tupler  r   r   r   r   r   r   test_with_listlike_columns.  s   r  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 )NrM   rs   qzr  r#   )r   j0)r   j1c                    s    fdd d D S )Nc                    s   g | ]
}| d  v r|qS )r%   r   )r   elr<  r   r   r[  F  s    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   rc   )r   r   r   from_tuplesr   r   r   r   r   r   r   r   )test_with_listlike_columns_returning_listA  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 r"   r  r  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   r  r  r   r   r   r[  Y  r  z3test_infer_output_shape_columns.<locals>.<listcomp>r   r   r   r   r  r   r   r   r   r   r   test_infer_output_shape_columnsK  s   
r  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 )Nr;   )r   rS   r   r.   r/   ru   c                 S   r   r   r   rG   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 )r   r   r  r   r   r   r[  e  r   z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S      ddgS Nr   r;   r   rG   r   r   r   r&   h  r'   c                 S   s   g | ]}d dgqS r   r;   r   r  r   r   r   r[  i  r   
r   r   r   r   r   r   r   r  r   r   r   r   r   r   (test_infer_output_shape_listlike_columns]  s   r  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 )NrM   r   r   rc   c                    s
   t  S r"   r   onesr<  rY  r   r   r&   r  r9   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>r   r   c                    s   g | ]}t  qS r   r  r  rY  r   r   r[  s  r%  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r   r   r   r  r   r   r   )rU  r@   r   r   r   rY  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   r   r:   rS   )r   rS   r:   r;   )r   r;   rS   r:   )rM   rs   rt   r   c                 S   rc  )Nr  r   rG   r   r   r   fun  s   zDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.funr   r   c                 S   s   g | ]}d qS )r  r   r  r   r   r   r[        zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>r  )r@   r  r   r   r   r   r   7test_infer_output_shape_listlike_columns_with_timestampw  s   r  lstr   c                    s^   t tjddg dd}|j fdddd}t fd	d
| D }t	|| d S )Nr;   )r:   rS   r  ru   c                    s    S r"   r   rG   r  r   r   r&     rH   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>r  )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 )Nr   )testothercolsrc   r   rG   r   r   r   r&     r   z'test_consistent_names.<locals>.<lambda>r   r   r  r  r  r  ru   c                 S   s   t ddgddgdS )Nr   r;   r  r  rc   r  rG   r   r   r   r&     r   )r   renamer   r+   r   r   r   r   test_consistent_names  s   
r  c                 C   s:   | }|j dd ddd}| }g d|_t|| d S )Nc                 S   r   r   r   rG   r   r   r   r&     r'   z"test_result_type.<locals>.<lambda>r   rm   rn   r   r   r;   r   r   re   r   r+   r   r   r   r   test_result_type  s
   
r  c                 C   sB   | }|j dd ddd}|ddg  }ddg|_t|| d S )	Nc                 S   r  r  r   rG   r   r   r   r&     r'   z/test_result_type_shorter_list.<locals>.<lambda>r   rm   rn   r   r.   r   r  r   r   r   r   test_result_type_shorter_list  s
   
r  c                 C   r   )Nc                 S   r   r   r   rG   r   r   r   r&     r'   z,test_result_type_broadcast.<locals>.<lambda>r   r   rn   r   r   r   r   r   test_result_type_broadcast  s   r  c                    s<   | }g d |j  fddddd}| }t|| d S )Nr  rR  rG  c                       t g d dS Nr   rc   r  rG   ru   r   r   r&     r  z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r   rn   r   r   r   ru   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   rG   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 )Nr  c                    r  r  r  rG   ru   r   r   r&     r  z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   r  r   r   ru   r   *test_result_type_series_result_other_index  s   r  boxc                 C      t | S r"   )r   rG   r   r   r   r&     r'   c                 C   r  r"   )r  rG   r   r   r   r&     r'   c                 C   s   t j| ddS )Nr   rq   )r   r   rG   r   r   r   r&     r6  )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                        ddgS r  r   rG   r  r   r   r&     r   z,test_consistency_for_boxed.<locals>.<lambda>r   r   c                    s   g | ]} d dgqS r  r   r  r  r   r   r[    r  z.test_consistency_for_boxed.<locals>.<listcomp>c                    r  r  r   rG   r  r   r   r&     r   rm   rn   r   r.   r   rX   ru   )r   r   r  r   r   r  r+   )r  r   r@   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 rGt	|j
dgg|_
n
t	|jdgg|_t|| |jt jt jg| d}t||g|d}| dv rzt	|j
ddgg|_
nt	|jddgg|_t|| W d    d S 1 sw   Y  d S )	N   r   r   r   r   r
   r   r   r   absolute)r   r   absr   r   r   r   r+   r   from_productre   r   r	   )r   r   
other_axisf_absf_sqrtr   r   r   r   r   test_agg_transform   s0   

"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 )
Nr   rX   r#  r$  r   r:   r   r.   r   )r   r   rO   r   r+   r   r   r   r   	test_demo%  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 )Nr   rX   r#  r$  rz         @r@        @g      9@r   r.   )r$  r#  rz   r   )r   r   rO   r   rd   r   r+   reindex_liker   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  countr;   r  )r   rO   r   r   r   )rP   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 )Nr   r   r         @r   r  bazr  r   r   r         @r  	foobarbazr#  rz   rc   )r/   r.   r   )r   rO   r   r+   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.w   Y  tjt|d	 | g d dd
g W d    d S 1 sQw   Y  d S )Nr   r  r  rR   rS   rT   r   r.   r/   r0   does not support reductionro  r#  rz   )r0   r/   r.   r   )r   r   rr  rs  	TypeErrorrO   )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 r9|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$  rz   r   rz   rc   >   re   r   r$  )axesuniquesort_valuesrY   concatr   r$  rz   re   r   rO   r   r+   r   rf   r   r   )r   r   r  name1name2r   r   r]   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 s`w   Y  | g d d}tg dg dd}t|| d}t	j
t|d | dg W d    d S 1 sw   Y  d S )Nr   r  r  rR   rS   rT   r  r#  r   r   r  rc   r  r  ro  rz   r  )r   r  r  )r   r   rO   r   r   re   r   r   r+   rr  rs  r  )r@   r   r   r  r   r   r   test_nuiscance_columns  s8   

	"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;   rS   r   r  r   Nr  r  r   r  r;   sizerS   )r  r  Tr&  )	r   r   rd   r}   r   r   r   r+   r  )r  r@   r   r   result1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 r(t|jd	 |jd
}n
t|jd |jd
}t	|| d S )Nr  r   r  r  r  r  r   )r   r   r   rc   r   )
r   r   rd   r}   r   r<   re   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)r;   r;   rS   g      ?r  r  c                 S   s   t |   S r"   )r   dropnar  )	group_colr   r   r   r]   &  rI  z&test_agg_listlike_result.<locals>.funcr;   rS   r   r  rc   r]   )
r   r   rd   rO   r   r   r   to_framer   r+   )r@   r]   r   r   r   r   r   test_agg_listlike_result"  s    
 r  zargs, kwargs)r   r      r  r(   rs   r   rM   rs   rr   c                 C   sj   ddd}t ddgddgg}| dkrtdd	g}ntd
dg}|j|| g|R i |}t|| d S )NrS   c                 S   s   |   || |  S r"   rb  )r$   rM   rs   rt   r   r   r   r  @  s   ztest_agg_args_kwargs.<locals>.fr   r;   r:   r   r  g      @r  g       @)rS   r   r   rO   r   r   )r   r)   kwargsr  r@   r   r   r   r   r   test_agg_args_kwargs2  s   
r  num_cols)r;   rS   r   c                 C   sJ   t g d}| }td|t| d}|dd  |j|jks#J d S )N)z
1950-06-30z
1952-10-24z
1953-05-29r   r  c                 S   rF   r"   r   rG   r   r   r   r&   U  rH   z,test_frequency_is_original.<locals>.<lambda>)rY   DatetimeIndexr   r   r   r   r  )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+0000rC   rD   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r  )rP   r   c                 S   r  r"   r  rG   r   r   r   r&   b  r  z.test_apply_datetime_tz_issue.<locals>.<lambda>r   r   )r   rP   )r   r   r   r   r   r   )
timestampsr@   r   r   r   r   r   test_apply_datetime_tz_issueY  s   


r  r@   rt   r   rX   method)r#  r$  rz   c                 C   sR   |dkrd}nd}t jt|d t| |  W d    d S 1 s"w   Y  d S )Nrz   z-can only concatenate str \(not "int"\) to strz4not supported between instances of 'str' and 'float'ro  )rr  rs  r  r}   )r@   r  r  r   r   r   test_mixed_column_raisesh  s   "r  rR  r   c                 C   s:   t d| ggddgd}|dd }|j}t|| d S )Nr   rM   rs   ru   c                 S   r  r"   rq   rG   r   r   r   r&   x  r  z"test_apply_dtype.<locals>.<lambda>)r   r   dtypesr   r   )rR  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|| |s1| r9t|| d S t|| d S )	Nd      r  c                 S   s*   | j }| jd  d7  < | j |usJ | S )NrM   r   )_mgrrf   )r=  mgrr   r   r   r]     s   z!test_apply_mutating.<locals>.funcrM   r   r   )r   r   r   r   r   r+   )using_array_managerusing_copy_on_writer@   df_origr]   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;   rS   r:   r   r   r   r   r   r   rM   rs   ru   c                 S   s   g S r"   r   rG   r   r   r   r&     rH   z.test_apply_empty_list_reduce.<locals>.<lambda>rp   r   r  rq   )r   r   r   rS  r   r   r   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:   r   rS   r   r.   ru   rz   c                 S   ra   r"   rb  rG   r   r   r   r&     r'   z,test_apply_no_suffix_index.<locals>.<lambda>)rW  rW  rW  )   r  r  rX   )rz   r&   r&   rc   rk   )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   aabbbc                 S   rj   r   r   rG   r   r   r   r&     r'   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr   r   r   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   )r;   r   r   )rS   r   r   )r:   r   r:   )r   r   r   )r   r   r   )att1att2att3ru   c                 S   s   |   d S r   rb  r   r   r   r   sum_div2  r   z4test_aggregation_func_column_order.<locals>.sum_div2rz   r  r#  )g      5@g      %@r  r   )g      2@g      "@r  r@  )g      1@g      !@r  r@  )rz   r  r  r#  rc   rN   )r@   r  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 )
Nr  r   r  rM   c                 S   s   | j d | j d  S )Nr   r   )r   rG   r   r   r   r&     r   z+test_apply_getitem_axis_1.<locals>.<lambda>r   r   )r   r;   r:   r   r   r   r   r   test_apply_getitem_axis_1  s   r  c                  C   sX   dd } t dg di}tjtdd || g W d    d S 1 s%w   Y  d S )Nc                 S   s   t d |  S )NHello, World!)warningswarnrz   rG   r   r   r   expected_warning  s   
zDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warningrM   r   r  ro  )r   r   assert_produces_warningUserWarningrO   )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 )NrS   r  g      ?i  r   )r!  r"  rr   rc   r   r   )
r   r~   r   r   rd   r   typer   r   r   r   r   r   r   test_apply_type  s   r$  c                  C   sL   t ddgddgd} | djdd dd}tg tjd	}t|| d S )
Nr   r;   rS   r   r  c                 S   s   t | d | d S )NrM   rs   )r$  rG   r   r   r   r&     r   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   rq   )r   headr   r   r   rw   r   r   r   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   r;   rS   r  c                 S   s   | j gS r"   rO  rG   r   r   r   r&     r'   z(test_apply_return_list.<locals>.<lambda>rk   r   r   r   r   test_apply_return_list  s   r'  ztest, constant)r   r   r   )r;   r;   r;   c                 C   s>   t | }t|ddgdd}ddd}||}t|| d S )NrM   rs   rS  )rP   r   r>   r  r  r  )r  constantdf1r   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   r;   r   )r   r;   r   )r   r   r   r  )TTFr   r   r|   )r   r   r   r   r|   r   r9  r   r   r   /test_any_apply_keyword_non_zero_axis_regression  s   r,  c                  C   s.  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 sDw   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 sw   Y  t g ddgd}t|| d S )Nr$   r   r   r   c                 S      | | | S r"   r   )r$   rM   rt   r   r   r   foo10  r   z.test_agg_mapping_func_deprecated.<locals>.foo1r;   c                 S   r-  r"   r   )r$   rs   rt   r   r   r   foo23  r   z.test_agg_mapping_func_deprecated.<locals>.foo2rS   r:   )rt   r   z+using .+ in Series.agg cannot aggregate andro  r   r   r   r.  r/  ru   )r;   rS   r:   )r   r   )r;   r   )r   rO   r   r+   r   FutureWarning)r@   r.  r/  r   r   r  r   r   r    test_agg_mapping_func_deprecated,  s&   

"r1  c                  C   s   t tdddddgd} tjtdd | tj}W d    n1 s'w   Y  t	d	d	d
t
d}t|| tjtdd | tjg}W d    n1 sTw   Y  t d	d	d
dgd}t|| d S )Nr   rS   r;   r   r.   ru   zusing DataFrame.stdro  r   rX   rq   zusing Series.stdstdrc   )r   r   r[   reshaper   r   r0  rO   r2  r   r~   r   r+   r   r   r   r   test_agg_stdK  s   r4  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  r  )r   r   r/   r   r  )r   r   rd   re   rO   r  r   r   r   r   r   r   (test_agg_dist_like_and_nonunique_columnsY  s   
r5  )r   r  numpyr   rr  pandas.core.dtypes.dtypesr   pandasrY   r   r   r   r   r   pandas._testing_testingr   pandas.tests.frame.commonr	   r    markparametrizer-   rA   rK   rQ   r\   r   r   r`   ri   rl   ry   r   r   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)  r2  r7  r8  r:  rB  rD  rL  rQ  rT  rd   r]  r`  rz   rl  rv  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r"  r$  r&  r'  r+  r,  r1  r4  r5  r   r   r   r   <module>   s   







		
			"4	*

	


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