o
    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dd	 Zd
d Zejdddedgfeddddfgdd Zdd Zejdejejgdd Zdd Zdd Zdd Zejdg dd d! Z d"d# Z!d$d% Z"d&d' Z#d(d) Z$d*d+ Z%d,d- Z&d.d/ Z'd0d1 Z(ejd2d dgd3d4 Z)ejd2d dgd5d6 Z*d7d8 Z+ejd2d dgd9d: Z,d;d< Z-d=d> Z.d?d@ Z/ejdAdBdCgejjddDdE dFdE gdGdHgdIejdJdKdLgejd2d dgdMdN Z0dOdP Z1dQdR Z2dSdT Z3dUdV Z4dWdX Z5dYdZ Z6d[d\ Z7d]d^ Z8d_d` Z9dadb Z:dcdd Z;dedf Z<dgdh Z=didj Z>dkdl Z?ejdme	dndogdpdqggedndpdrdodqdrgge	d dgdsdtgged dsdrddtdrgggdudv Z@dwdx ZAejdyddzgd{d| ZBd}d~ ZCejddedgdeDdggdd ZEejde	 e	eFdde	eFdde	g g g dgejdeGddE gdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPejdddgdd ZQdd ZRdd ZSdd ZTdd ZUdd ZVejdydddejWgdd ZXdd ZYdd ZZdd Z[dd Z\dd Z]dd Z^dd Z_dd Z`ejdyddsgdd ZaddĄ Zbejdg dƢddsggddȄ Zcddʄ Zddd̄ Zedd΄ ZfddЄ Zgdd҄ ZhddԄ Ziddք ZjejjdddE ddE ddE gg dۢdIdd݄ Zkdd߄ Zldd Zmdd Zndd Zodd Zpdd Zqdd Zrdd Zsejdddgdd Ztejdddgdd Zudd Zvejd2d dgejddi fdi fdi fdddsifdddsdfddsddfdddsdtdfgdd Zwejdg d dd Zxdd Zyejde	ddgddgd	gejd
g ddd ZzejdddzdKdej{gdd Z|dd Z}dd Z~dd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zejd#g dƢg d$dg dƢdgdfg d%g d$ddsgdgdfg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   W/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/apply/test_frame_apply.py
test_apply   s   
"r    c                  C   s^   t g dg dd} | dd }|jdksJ t|d jts#J t|d jts-J d S )	N)r   r   Br!   )Cr"   Dr#   c0c1c                 S   
   |  dS )Ncategoryastype)tsr   r   r   <lambda>+      
 z-test_apply_categorical_func.<locals>.<lambda>)      r%   r&   )r   r   shape
isinstancedtyper   dfr   r   r   r   test_apply_categorical_func(   s
   r5   c                  C   s8   t dtdddgi} | jdd dd}t||  d S )	Nr   
2013-01-01UTCtzc                 S      | S Nr   xr   r   r   r,   5       z*test_apply_axis1_with_ea.<locals>.<lambda>r   r   r   r   r   r   assert_frame_equalr   r   r   r   r   test_apply_axis1_with_ea2   s   rB   zdata, dtyper   Nr   r6   r7   r8   c                 C   s<   t | g| ggddg|d}|jdd dd}t|| d S )Na)r   r2   c                 S   r:   r;   r   r<   r   r   r   r,   @   r>   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   r   aggr   r@   )datar2   r   r   r   r   r   test_agg_axis1_duplicate_index9   s   rH   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   r:   r;   r   r<   r   r   r   r,   M   r>   z/test_apply_mixed_datetimelike.<locals>.<lambda>r   r   )	r   r   pdZto_timedeltar   Zaranger   r   r@   rA   r   r   r   test_apply_mixed_datetimelikeD   s   
rQ   funcc                 C   s   t  }|| }|jsJ d S r;   )r   r   empty)rR   empty_framer   r   r   r   test_apply_emptyQ   s   
rU   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   r<   r   r   r   r,   \       z(test_apply_float_frame.<locals>.<lambda>r   c                 S   rV   r;   rW   r<   r   r   r   r,   a   rX   r   r   )	r   r   r   nancolumnsr   r   locr   )r   Zno_rowsr   r   no_colsr   r   r   test_apply_float_frameZ   s   r^   c                  C   s.   t dgd} | jdd dd}t||  d S )NrD   rY   c                 S      | d S )NrD   r   r<   r   r   r   r,   i   rX   z/test_apply_empty_except_index.<locals>.<lambda>r   r   r   r   r   r@   rA   r   r   r   test_apply_empty_except_indexf   s   ra   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r2   rD   bcr[   )	r   r   appendr   r@   r   r   float64r   )rT   r=   r   r   Zempty_with_colsr   r   r   test_apply_with_reduce_emptym   s   rm   )sumprodanyr   c                 C   sL   t g dd}|tt| }t||  }| dv r|t}t|| d S )Nrg   rj   )rn   ro   )r   r   getattrr   r*   floatr   r   )rR   r4   r   r   r   r   r   test_apply_funcs_over_empty   s   
rs   c                  C   sX   t g dd} |  }td| jd}t|| | j }tg tjd}t|| d S )Nrg   rj   r   rY   rf   )	r   Znuniquer   r[   r   r   Tr   rl   r4   r   r   r   r   r   test_nunique_empty   s   
rv   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/   rJ   )r.         )      	   )rD   rD   ri   rY   c                 S   r_   Nr   r   rM   r   r   r   r,      rX   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r.   rz   c                 S   r_   r}   r   r~   r   r   r   r,      rX   r   )r   r   r   r   r   rt   ru   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rd   rY   )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   rc   r   c                       i | ]}| qS r   r   .0ri   mr   r   
<dictcomp>       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                       t tt jS r;   )listrangelenr[   r<   r   r   r   r,      r   z4test_apply_broadcast_lists_columns.<locals>.<lambda>r   r   rc   rl   )r2   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                    r   r;   )r   r   r   r   r<   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>rl   )r2   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      g dS Nrw   r   r<   r   r   r   r,      rX   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r   rc   r   r   r@   )int_frame_const_colr4   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 )Nrw   abcrY   )r   r   r<   r   r   r   r,          z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r   rc   r   copyr   r@   r   r4   r   r   r   r   r   'test_apply_broadcast_series_lambda_func   s   r   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   r1   r   ndarrayndimr<   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   r<   r   r   r   r,      r-   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>r   )r   r   r   r   r   )r   r   r   r   r   r   r   !test_apply_raw_float_frame_lambda   s   r   c                 C   s*   | j dd dd}| d }t|| d S )Nc                 S      | d S Nr/   r   r<   r   r   r   r,      rX   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   r<   r   r   r   r      r   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      ?rO   r   r   r   int64rf   rY   )
r   r   r   r   r   rZ   rP   Indexr   r   ru   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   rO   c                 S   r_   )Nr   r   r<   r   r   r   r,     rX   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   rY   c                 S   r_   )Nr!   r   r<   r   r   r   r,     rX   r   r   r   r   r   ru   r   r   r   &test_apply_mixed_dtype_corner_indexing  s   r   axr   r[   c                 C   r:   r;   r   r<   r   r   r   r,   #  r>   r,   c                 C   rV   r;   rW   r<   r   r   r   r,   #  rX   identityr   )Zidsr   TFc           	   	   C   s   t d	i | g di}tjddb tjdd tdt |tjg dd}W d    n1 s2w   Y  t|tj	 }|j
|||d}|r\||}t|tsTJ |j|u s[J nt|t scJ W d    d S W d    d S 1 svw   Y  d S )
Nrg   r
   r   T)recordZf8rf   r   r   )r   r   r   warningscatch_warningssimplefilterRuntimeWarningarrayr1   r   r   Z_get_agg_axisr   r   )	r   rR   r   r   r4   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 )Nrg   rY   c                 S   rV   r;   rW   r<   r   r   r   r,   ;  rX   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r   r   )r   r   r1   )r]   r   r   r   r   %test_apply_empty_infer_type_broadcast9  s   r   c                 C   6   ddd}| j |dd}|  dd }t|| d S )	Nr   c                 S      | | S r;   r   r=   howmuchr   r   r   add_some@     z4test_apply_with_args_kwds_add_some.<locals>.add_somer/   r   c                 S   s   | d S r   r   r<   r   r   r   r,   D  rX   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>r   r   )r   r   r   r   r   r   r   "test_apply_with_args_kwds_add_some?     
r   c                 C   r   )	Nr   c                 S   s   |   | S r;   rW   r   r   r   r   agg_and_addI     z:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addr/   r   c                 S   s   |   d S r   rW   r<   r   r   r   r,   M      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_addH  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_divideR  r   zJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide)r/   r/   )argsr   c                 S   s   | d d S )N       @r   r<   r   r   r   r,   V  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_divideQ  s   
r   c                 C   s   |  t}t||  d S r;   )r   r   r   r@   )r   r   r   r   r   test_apply_yield_listZ  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   rZ   Zilocr[   Zget_locr   r   r   r   )r   r   r   r   r   r   test_apply_reduce_Series_  s   
r   c                  C   s   t ddgddggddgddgd	} | jtd
d}tddddddg| jd}t|| | jtdd}tddddddg| jd}t|| d S )Nr   r/   rJ   r.   r%   r&   i0i1r[   r   r   r   )r   r   rY   r$   )r   r   dictr   r[   r   r   r   )rG   r   r   r   r   r   test_apply_reduce_to_dictf  s   "r   c                  C   s   t tjdd} | jtjdd}t dd |  D | jd}t	
|| | jtjdd}t d	d | j D | jdj}t	
|| d S )
N   
   r   r   c                 S      i | ]	\}}||  qS r   describer   ivr   r   r   r   w      z2test_apply_differently_indexed.<locals>.<dictcomp>rj   r   c                 S   r   r   r   r   r   r   r   r   {  r   )r   r   randomrandnr   r   r   itemsr[   r   r@   rt   r   ru   r   r   r   test_apply_differently_indexeds  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   )rD   marketpositionrj   c                 S   r_   )Nr   r   )rr   r   r   f  r   ztest_apply_bug.<locals>.fr   r   i  r   r   r/   r   r   rJ   r   r.   r   rx   r   ry   r   )r   r   r   r   r   )Z	positionsr   r   r   r   r   r   test_apply_bug  s0   r   c               	   C   s\   t g dg dg dtjdtjdtj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#   EFc                 S   r:   r;   r   r<   r   r   r   r,     r>   z,test_apply_convert_objects.<locals>.<lambda>r   r   )r   r   r   r   r   r   r@   rA   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;   namer<   r   r   r   r,         z(test_apply_attach_name.<locals>.<lambda>rY   )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   r  r;   r  r<   r   r   r   r,     r  z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   rY   )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   r<   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   r  r;   r  r<   r   r   r   r,     r   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>r   r   c                 3   s&    | ]}t |d  t jV  qdS )r   N)r   r	  r   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)rD   rD   rh   )ri   r   r   r   r/   rJ   r.   rx   ry   col1col2r
  c                 S   s   t t| t| dS )N)minmax)r   r  r  r<   r   r   r   r,         z(test_apply_multi_index.<locals>.<lambda>r  r  TZ
check_like)r   Zfrom_arraysr   r   r   r@   )r   rM   r   r   r   r   r   test_apply_multi_index  s
   $$r  z	df, dictsr   r   Zspameggs)r   r   r/   rJ   c                 C   sV   dd }| j |dd}| j |dd}|  |}t|| t||  t|| d S )Nc                 S   rV   r;   )to_dictr<   r   r   r   r,   	  rX   z!test_apply_dict.<locals>.<lambda>re   r   rb   )r   r   r   r@   )r4   ZdictsfnZreduce_trueZreduce_falseZreduce_noner   r   r   test_apply_dict  s   
r  c                 C   sP   |  dd }t|| d  |  t |  dd d d }t|ts&J d S )Nc                 S   r   r   r   r<   r   r   r   r,     rX   ztest_applymap.<locals>.<lambda>r/   c                 S   s   | | fS r;   r   r<   r   r   r   r,     rX   r   r   )applymapr   r@   typer1   tuple)r   Zappliedr   r   r   r   test_applymap  s
   
r  valr   c                 C   s2   t | dgd}|dd jd }|tksJ d S )NrD   )rG   c                 S   r:   r;   r   r<   r   r   r   r,   !  r>   z7test_applymap_float_object_conversion.<locals>.<lambda>r   )r   r  dtypesobject)r   r4   r   r   r   r   %test_applymap_float_object_conversion  s   r#  c                  C   sP   t tjd} |  }g d}|| _|t}||_| t}t|| d S )NrJ   r.   )rD   rD   rD   rD   )	r   r   r   r   r[   r  strr   r@   )r4   Zdf2colsr   r   r   r   r   test_applymap_str%  s   

r'  zcol, valr   rI   	timedeltaz1 minc                 C   sJ   t tjd}||| < |t}|jd| f t|jd| f ks#J d S )Nr$  r   )r   r   r   r  r%  r\   )colr   r4   r   r   r   r   test_applymap_datetimelike2  s   
(r*  r   ABCrj   rY   r   r!   r"   c                 C   r:   r;   r   r<   r   r   r   r,   G  r>   c                 C   s   |  |}t||  d S r;   )r  r   r@   )r   rR   r   r   r   r   test_applymap_empty>  s   
r-  c                  C   sF   t ddgddggjdd dd} t ddgdd	gg}t| | d S )
Nr   r/   rJ   r.   c                 S   r   r;   r   r=   yr   r   r   r,   P  rX   z&test_applymap_kwargs.<locals>.<lambda>r/  rx   ry   )r   r  r   r@   )r   r   r   r   r   test_applymap_kwargsN  s   "r1  c                 C   sj   |  dd }|  }tjjdd| jtd}tj||< |j dd dd}| }tj||< t	
|| d S )	Nc                 S      t t| S r;   r   r%  r<   r   r   r   r,   W  r   z)test_applymap_na_ignore.<locals>.<lambda>r   r/   )sizer2   c                 S   r2  r;   r3  r<   r   r   r   r,   \  r   r
   )Z	na_action)r  r   r   r   randintr0   boolrP   ZNAr   r@   )r   Zstrlen_frameZfloat_frame_with_namaskZstrlen_frame_na_ignoreZstrlen_frame_with_nar   r   r   test_applymap_na_ignoreU  s   

r8  c                  C   s*   t tddd} dd }t| | d S )Nz1/1/2000r   rK   c                 S   s   | j | j| jfS r;   )hourdaymonthr<   r   r   r   rR   g     z*test_applymap_box_timestamps.<locals>.func)r   r   r   r  )ZserrR   r   r   r   test_applymap_box_timestampsc  s   r=  c               	   C   s   t tdtdgtdddtdddgtdtdgtjdddtjdddgd	} | d
d }t ddgddgddgddgd	}t|| d S )Nz
2011-01-01z
2011-01-02z
US/Easternr8   z1 daysz2 daysM)freqrD   rh   ri   r   c                 S   s
   t | jS r;   )r  __name__r<   r   r   r   r,     r-   z#test_applymap_box.<locals>.<lambda>r   	TimedeltaPeriod)r   r   rP   rB  rC  r  r   r@   ru   r   r   r   test_applymap_boxn  s(   

rD  c                     s^   ddl m  tdtdddgi} |  fdd} |  fdd} | jj}|d	ks-J d S )
Nr   BDayx1i  r   c                    
   |    S r;   r   r<   rE  r   r   r,     r-   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>c                    rH  r;   r   r<   rE  r   r   r,     r-   zM8[ns])Zpandas.tseries.offsetsrF  r   r   r  rG  r2   r3   r   rE  r   (test_frame_apply_dont_convert_datetime64  s   rI  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-01rJ   zEurope/Brussels)rL   r9   c                 S   r:   r;   r   r<   r   r   r   r,     r>   z,test_apply_non_numpy_dtype.<locals>.<lambda>c                 S   s   | t d S )NZ1day)rP   rB  r<   r   r   r   r,         z
2015-01-02)r   r   r   r   r@   ru   r   r   r   test_apply_non_numpy_dtype  s   rL  c                  C   s2   t dg didd} | dd }t||  d S )NrJ  )rD   rh   ri   rD   r(   rf   c                 S   r:   r;   r   r<   r   r   r   r,     r>   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>r`   r3   r   r   r   #test_apply_non_numpy_dtype_category  s   rM  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/   rJ   rD   rj   r  r   rE   r4   r   r   r   r   r   test_apply_dup_names_multi_agg  s   rO  opr   rF   c                 C   s`   dd }t tdtdd}t|| |dd}tg dg dg dg dg}t|| d S )	Nc                 S   s"   d| d  d| d  d| d  gS )Nr/   r   r"   r!   r   rowr   r   r   
apply_list  s   "z3test_apply_nested_result_axis_1.<locals>.apply_list)r.   r.   ZABCDrj   r   r   )        rT  rT  )r   r   Zzerosr   rq   r   r   r   )rP  rS  r4   r   r   r   r   r   test_apply_nested_result_axis_1  s   rU  c                  C   sR   t dtdddgidd} | dd }t||  | d	d }t||  d S )
Nr   Z2020r7   r8   zdatetime64[ns, UTC]rf   c                 S   r:   r;   r   r<   r   r   r   r,     r>   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   rV   r;   )r   r<   r   r   r   r,     rX   r?   rA   r   r   r   %test_apply_noreduction_tzaware_object  s   rV  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 )	NrD   rw   c                    s     | j d S r;   rk   r  rQ  namesr   r   reducing_function  r<  z8test_apply_function_runs_once.<locals>.reducing_functionc                    s     | j | S r;   rW  rQ  rX  r   r   non_reducing_function  s   z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r   r   r4   rZ  r[  rR   r   rX  r   test_apply_function_runs_once  s   
r]  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 )
NrD   rw   c                         |  d S r;   extendrQ  r   r   r   rZ       z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                         |  | S r;   r_  rQ  ra  r   r   r[       
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r   r   )r   r   r   rD   to_listr\  r   ra  r   !test_apply_raw_function_runs_once  s   
rf  c                     sd   t dg di} g   fdd} fdd}||fD ]} d d = | |  | j ks/J qd S )NrD   rw   c                    r^  r;   rk   r   ra  r   r   rZ    rb  z;test_applymap_function_runs_once.<locals>.reducing_functionc                    rc  r;   rg  rh  ra  r   r   r[    rd  z?test_applymap_function_runs_once.<locals>.non_reducing_function)r   r  rD   re  r\  r   ra  r    test_applymap_function_runs_once  s   

ri  c                  C   sP   t tddgdgd} t tddgdgtd}| dd }t|| d S )Ns   abcds   efghr)  rj   )r[   r2   c                 S   r'   )Nr"  r)   r<   r   r   r   r,     r-   z-test_apply_with_byte_string.<locals>.<lambda>)r   r   r   r"  r   r   r@   rN  r   r   r   test_apply_with_byte_string  s   rj  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 )Nrk  rl  ZcderD   r(   rf   c                    s   |  kS r;   r   r<   rh  r   r   r,     rX   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t |rtjn| kqS r   )rP   Zisnullr   NaN)r   r=   rh  r   r   
<listcomp>  s   $ z1test_apply_category_equalness.<locals>.<listcomp>r  )r   rm  r   rD   r   r   r   r   )r   Z	df_valuesr4   r   r   r   rh  r   test_apply_category_equalness  s   ro  c                  C   sV   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/   )ry   r/   )r   r   r   Zrandr   Zfftr0   Zrfftr3   r   r   r   test_infer_row_shape"  s
   rp  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/   rD   rh   rj   c                 S      d| d | d  iS NrM   rD   rh   r   r<   r   r   r   r,   0  r   z,test_with_dictlike_columns.<locals>.<lambda>r   c                 S   s   g | ]}d diqS )rM   rJ   r   r  r   r   r   rn  1  r   z.test_with_dictlike_columns.<locals>.<listcomp>2017-05-01 00:00:002017-05-02 00:00:00r   c                 S   rq  rr  r   r<   r   r   r   r,   8  r   c                 S   s   d| iS )NrM   r   r<   r   r   r   r,   <  rX   rM   rJ   )r   r   r   r  r   r   r   ru   r   r   r   test_with_dictlike_columns-  s   ru  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ZZ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   r<   r   r   r   r,   J  r>   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rP   to_datetimer   r   r   r   ru   r   r   r   (test_with_dictlike_columns_with_datetimeA  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/   rD   rh   rj   c                 S   rq  rr  r   r<   r   r   r   r,   R  r   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>rb   rc   rM   rJ   rs  rt  r   c                 S   rq  rr  r   r<   r   r   r   r,   Z  r   )r   r   r   r@   r   ru   r   r   r   %test_with_dictlike_columns_with_inferO  s   r|  c                  C   s   t ttjdg dtddddd} | ddg jtd	d
}tdd | ddg  D }t	
|| | ddg jtd	d
}tdd | ddg  D }t	
|| d S )Nr.   )rD   r   Zofwordsz
2016-10-01H)rL   r?  )rD   rh   r+   rD   rh   r   r   c                 S      g | ]}|d d qS rC   r   r  r   r   r   rn  i      z.test_with_listlike_columns.<locals>.<listcomp>r+   c                 S   r  rC   r   r  r   r   r   rn  m  r  )r   r   r   r   r   r   r   r  r  r   r   ru   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 )NrD   rh   qzr  r.  )r   Zj0)r   Zj1c                    s    fdd d D S )Nc                    s   g | ]
}| d  v r|qS r0  r   )r   elrQ  r   r   rn  v  s    zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>r=   r   rQ  r   rQ  r   r,   v  r  z;test_with_listlike_columns_returning_list.<locals>.<lambda>r   r   rY   )r   r   r   from_tuplesr   r   r   r   ru   r   r   r   )test_with_listlike_columns_returning_listq  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  rQ  r   r   r   r,     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   rn    r  z3test_infer_output_shape_columns.<locals>.<listcomp>r   r   r   r   r  r   r   ru   r   r   r   test_infer_output_shape_columns{  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 )Nry   rJ   r,  rj   c                 S   r   r   r   r<   r   r   r   r,     rX   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}g d qS )rw   r   r  r   r   r   rn    r   z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S      ddgS Nr   r/   r   r<   r   r   r   r,     rX   c                 S   s   g | ]}d dgqS r   r/   r   r  r   r   r   rn    r   	r   r   r   r   r   r   r  r   r   ru   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 )NrD   rw   r   rY   c                    s
   t  S r;   r   ZonesrQ  rh  r   r   r,     r-   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>r   r   c                    s   g | ]}t  qS r   r  r  rh  r   r   rn    r  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r   r   r   r  r   r   r   )r   r4   r   r   r   rh  r   0test_infer_output_shape_listlike_columns_np_func  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|   rx   r.   rJ   )rx   rJ   r.   r/   )r   r/   rJ   r.   r@  c                 S   s   dS )Nr  r   r<   r   r   r   fun  s   zDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.funr   r   c                 S   s   g | ]}d qS )r  r   r  r   r   r   rn        zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>r  )r4   r  r   r   r   r   r   7test_infer_output_shape_listlike_columns_with_timestamp  s   r  lstrw   c                    sZ   t tjddg dd}|j fdddd}t fd	d
| D }t|| d S )Nr.   rJ   r,  rj   c                    s    S r;   r   r<   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   rn    r  z5test_consistent_coerce_for_shapes.<locals>.<listcomp>r  )r  r4   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 )Nrw   )testotherr&  rY   r   r<   r   r   r   r,     r   z'test_consistent_names.<locals>.<lambda>r   r   r  r  r&  r,  rj   c                 S   s   t ddgddgdS )Nr   r/   r  r  rY   r  r<   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   r<   r   r   r   r,     rX   z"test_result_type.<locals>.<lambda>r   rb   rc   r   r   r/   r   r   r[   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   r<   r   r   r   r,     rX   z/test_result_type_shorter_list.<locals>.<lambda>r   rb   rc   r   r!   r   r  r   r   r   r   test_result_type_shorter_list  s
   
r  c                 C   r   )Nc                 S   r   r   r   r<   r   r   r   r,     rX   z,test_result_type_broadcast.<locals>.<lambda>r   r   rc   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  r)  rY  c                       t g d dS Nrw   rY   r  r<   rj   r   r   r,     r  z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r   rc   r   r   r   rj   r   &test_result_type_broadcast_series_func  s   r  c                 C   s.   | }|j dd dd}| }t|| d S )Nc                 S   s   t g d| jdS r  )r   r   r<   r   r   r   r,   	  r   z0test_result_type_series_result.<locals>.<lambda>r   r   r   r   r   r   r   test_result_type_series_result  s   r  c                    s@   | }g d |j  fdddd}| } |_t|| d S )Nr  c                    r  r  r  r<   rj   r   r   r,     r  z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   r  r   r   rj   r   *test_result_type_series_result_other_index  s   r  boxc                 C      t | S r;   )r   r<   r   r   r   r,     rX   c                 C   r  r;   )r  r<   r   r   r   r,     rX   c                 C   s   t j| ddS )Nr   rf   )r   r   r<   r   r   r   r,     rK  )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   r<   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   rn  $  r  z.test_consistency_for_boxed.<locals>.<listcomp>c                    r  r  r   r<   r  r   r   r,   '  r   rb   rc   r   r!   r   rO   rj   )r   r   r  r   r   r  r@   )r  r   r4   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   Zfrom_productr[   r   r	   )r   r   
other_axisZf_absZ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 )
Nrx   rO   r  r  r   r.   r   r!   r   )r   r   rF   r   r@   ru   r   r   r   	test_demoQ  s   r  c                  C   sl   t tddd} | ddgddgd}t ddtjgdtjd	gdd
dgg dd}t||| d S )Nrx   rO   r  r  rn         @rT        @g      9@r   r!   )r  r  rn   r   )r   r   rF   r   rZ   r   r@   Zreindex_likeru   r   r   r   test_demo_dict_agg\  s   r  c                  C   sl   dddgi} t | }|ddi}tddi}t|| |d ddi}tddidd}t|| d S )Nr  r   r   countr/   r  )r   rF   r   r   r   )rG   r4   r   r   r   r   r   !test_agg_with_name_as_column_nameh  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 )Nrw   r   r         @r   r   Zbazr,  r   ry   r         @r   	foobarbazr  rn   rY   )r"   r!   r   )r   rF   r   r@   Zreindex)mdfr   r   r   r   r   test_agg_multiple_mixedx  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 )Nrw   r  r  rI   rJ   rK   r   r!   r"   r#   does not support reductionmatchr  rn   )r#   r"   r!   r   )r   r   pytestraises	TypeErrorrF   )r  msgr   r   r   test_agg_multiple_mixed_raises  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  rn   r   rn   rY   >   r   r[   r  )ZaxesuniqueZsort_valuesrP   concatr   r  rn   r[   rt   rF   r   r@   r   r\   r   r   )r   r   r  Zname1Zname2r   r   rR   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 )Nrw   r  r  rI   rJ   rK   r  r  r   r   r   rY   r
  r  r  rn   r,  )ry   r  r  )r   r   rF   r   r   r[   r   r   r@   r  r  r  )r4   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/   rJ   r   r  r   Nr   r,  r   r  r/   r4  rJ   )r  r4  Tr  )	r   r   rZ   rq   r   r   r   r@   r  )r  r4   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 r(t|jd	 |jd
}n
t|jd |jd
}t	|| d S )Nr  r   r  r  r,  r4  r   )r   r   r   rY   r   )
r   r   rZ   rq   r   r0   r[   r   r   r   )r  r   r4   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/   rJ   g      ?r  r,  c                 S   s   t |   S r;   )r   Zdropnar  )Z	group_colr   r   r   rR   R  r<  z&test_agg_listlike_result.<locals>.funcr/   rJ   r   r   rY   rR   )
r   r   rZ   rF   r   r   r   Zto_framert   r@   )r4   rR   r   r   r   r   r   test_agg_listlike_resultN  s    
 r  zargs, kwargs)r{   rz      r  r   rh   r   rD   rh   rg   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 )NrJ   c                 S   s   |   || |  S r;   rn   )r=   rD   rh   ri   r   r   r   r   l  s   ztest_agg_args_kwargs.<locals>.fr   r/   r.   r   r  g      @r  g       @)rJ   r   r   rF   r   r   )r   r   kwargsr   r4   r   r   r   r   r   test_agg_args_kwargs^  s   
r  num_cols)r/   rJ   rx   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   r:   r;   r   r<   r   r   r   r,     r>   z,test_frequency_is_original.<locals>.<lambda>)rP   ZDatetimeIndexr   r   r   r   r?  )r  r   originalr4   r   r   r   test_frequency_is_original{  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+0000r7   r8   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r  )rG   r   c                 S   r  r;   r  r<   r   r   r   r,     r  z.test_apply_datetime_tz_issue.<locals>.<lambda>r   r   )r   rG   )r   r   r   r   r   r   )Z
timestampsr4   r   r   r   r   r   test_apply_datetime_tz_issue  s   


r  r4   rD   ri   r   rO   method)r  r  rn   c                 C   sR   |dkrd}nd}t jt|d t| |  W d    d S 1 s"w   Y  d S )Nrn   z-can only concatenate str \(not "int"\) to strz4not supported between instances of 'str' and 'float'r  )r  r  r  rq   )r4   r  r  r   r   r   test_mixed_column_raises  s   "r  r)  c                 C   s:   t d| ggddgd}|dd }|j}t|| d S )Nr   rD   rh   rj   c                 S   r  r;   rf   r<   r   r   r   r,     r  z"test_apply_dtype.<locals>.<lambda>)r   r   r!  r   r   )r)  r4   r   r   r   r   r   test_apply_dtype  s   r  c                 C   s   t tdtddd}| }dd }| }|d  d7  < |j|dd}t|| |s1| r9t|| d S t|| d S )	Nd      r  c                 S   s*   | j }| jd  d7  < | j |usJ | S )NrD   r   )Z_mgrr\   )rR  Zmgrr   r   r   rR     s   z!test_apply_mutating.<locals>.funcrD   r   r   )r   r   r   r   r   r@   )Zusing_array_managerZusing_copy_on_writer4   Zdf_origrR   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/   rJ   r.   rx   ry   rz   r{   r|   r   rD   rh   rj   c                 S   s   g S r;   r   r<   r   r   r   r,     r>   z.test_apply_empty_list_reduce.<locals>.<lambda>re   r   r  rf   )r   r   r   r"  r   r   ru   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|   rJ   r   r!   rj   rn   c                 S   rV   r;   r  r<   r   r   r   r,     rX   z,test_apply_no_suffix_index.<locals>.<lambda>)rl  rl  rl  )   r  r  rO   )rn   r,   r,   rY   r`   )Zpdfr   r   r   r   r   test_apply_no_suffix_index  s   r  c                  C   s@   t dddgi} | jdd ddd}tddg}t|| d S )	Nr   Zaabbbc                 S   r_   r}   r   r<   r   r   r   r,     rX   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr   r   ru   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   )rJ   r   r   )r.   rx   r.   )rx   ry   ry   )ry   rz   rz   )Zatt1Zatt2Zatt3rj   c                 S   s   |   d S r   r  r~   r   r   r   sum_div2  r   z4test_aggregation_func_column_order.<locals>.sum_div2rn   r  r  )g      5@g      %@r  r   )g      2@g      "@r  rT  )g      1@g      !@r  rT  )rn   r  r  r  rY   rE   )r4   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 )
Nr  rw   r  rD   c                 S   s   | d | d  S )Nr   r   r   r<   r   r   r   r,     r  z+test_apply_getitem_axis_1.<locals>.<lambda>r   r   )r   r/   r.   r   ru   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!)r   warnrn   r<   r   r   r   expected_warning  s   
zDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warningrD   rw   r  r  )r   r   Zassert_produces_warningUserWarningrF   )r  r4   r   r   r   *test_nuisance_depr_passes_through_warnings  s
   "r  c                  C   s   t ddtgdtdddtjgdg dd} | t}t tttgtttgdg dd}t	
|| | jtd	d
}tttd}t	|| | jtdd
}ttttd}t	|| d S )NrJ   r  g      ?i  r   )r  r  rg   rY   r   r   )r   rr   r   r   rZ   r  r  intr%  r   r@   r   r   r   ru   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/   rJ   r   r  c                 S   s   t | d | d S )NrD   rh   )r  r<   r   r   r   r,   5  r   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   rf   )r   headr   r   r   rl   r   r   ru   r   r   r   test_apply_on_empty_dataframe2  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 )NrD   rh   r"  )rG   r   r2   r  r  r  )r  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   r/   r   )r   r/   r   )r   r   r   r,  )TTFr   r   rp   )r   r   r   r   rp   r   rN  r   r   r   /test_any_apply_keyword_non_zero_axis_regressionL  s   r  c                  C   s   t dg di} ddd}ddd	}d
}tjt|d | j||gddddd W d    n1 s2w   Y  | j||gdddd}t ddgddgddggtddgd}t|| d S )Nr=   rw   r   r   c                 S      | | | S r;   r   )r=   rD   ri   r   r   r   foo1]  r   z/test_agg_list_like_func_with_args.<locals>.foo1r/   c                 S   r  r;   r   )r=   rh   ri   r   r   r   foo2`  r   z/test_agg_list_like_func_with_args.<locals>.foo2z/foo1\(\) got an unexpected keyword argument 'b'r  rJ   r.   )rh   ri   )ri   r{   r|   r   )r=   r  )r=   r  rj   )r   r   )r/   r   )	r   r  r  r  rF   r   r  r   r@   )r4   r  r  r  r   r   r   r   r   !test_agg_list_like_func_with_argsY  s   

r  )r   r   numpyr   r  Zpandas.core.dtypes.dtypesr   ZpandasrP   r   r   r   r   r   Zpandas._testingZ_testingr   Zpandas.tests.frame.commonr	   r    r5   rB   markZparametrizerH   rQ   r   r   rU   r^   ra   rm   rs   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'  rB  r*  r   roundr-  r1  r8  r=  rD  rI  rL  rM  rO  rU  rV  r]  rf  ri  rj  rm  ro  rp  ru  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  rZ   r  r  r  r  r  r  r  r  r  r  r  r  r  r   r   r   r   <module>   sx   






					"4	*




		


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