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   h/var/www/html/visualizacion-main/env/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   pdto_timedeltar   aranger   r   r@   rA   r   r   r   test_apply_mixed_datetimelikeD   s   
rS   funcc                 C   s   t  }|| }|jsJ d S r;   )r   r   empty)rT   empty_framer   r   r   r   test_apply_emptyQ   s   
rW   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   rX   r;   rY   r<   r   r   r   r,   a   rZ   r   r   )	r   r   r   nancolumnsr   r   locr   )r   no_rowsr   r   no_colsr   r   r   test_apply_float_frameZ   s   ra   c                  C   s.   t dgd} | jdd dd}t||  d S )NrD   r[   c                 S      | d S )NrD   r   r<   r   r   r   r,   i   rZ   z/test_apply_empty_except_index.<locals>.<lambda>r   r   r   r   r   r@   rA   r   r   r   test_apply_empty_except_indexf   s   rd   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   )rV   r=   r   r   empty_with_colsr   r   r   test_apply_with_reduce_emptym   s   rq   )sumprodanyr   c                 C   sL   t g dd}|tt| }t||  }| dv r|t}t|| d S )Nrj   rm   )rr   rs   )r   r   getattrr   r*   floatr   r   )rT   r4   r   r   r   r   r   test_apply_funcs_over_empty   s   
rw   c                  C   sX   t g dd} |  }td| jd}t|| | j }tg tjd}t|| d S )Nrj   rm   r   r[   ri   )	r   nuniquer   r]   r   r   Tr   ro   r4   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/   rJ   )r.         )      	   )rD   rD   rl   r[   c                 S   rb   Nr   r   rM   r   r   r   r,      rZ   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r.   r   c                 S   rb   r   r   r   r   r   r   r,      rZ   r   )r   r   r   r   r   ry   rz   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rg   r[   )r   r   r   r   r   r   r@   r   r   r   r   r   r   test_apply_broadcast_scalars   s   r   c                    sF   | j tjddd}| jdd t fdd| jD }t|| d S )Nr   r   rf   r   c                       i | ]}| qS r   r   .0rl   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   rf   ro   )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>ro   )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 Nr|   r   r<   r   r   r   r,      rZ   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r   rf   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 )Nr|   abcr[   )r   r   r<   r   r   r   r,          z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>r   r   rf   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,      rZ   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   )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   int64ri   r[   )
r   r   r   r   r   r\   rP   Indexr   r   rz   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   rb   )Nr   r   r<   r   r   r   r,     rZ   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   r[   c                 S   rb   )Nr!   r   r<   r   r   r   r,     rZ   r   r   r   r   r   rz   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   rX   r;   rY   r<   r   r   r   r,   #  rZ   identityr   )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 )
Nrj   r
   r   T)recordf8ri   r   r   )r   r   r   warningscatch_warningssimplefilterRuntimeWarningarrayr1   r   r   _get_agg_axisr   r   )	r   rT   r   r   r4   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 )Nrj   r[   c                 S   rX   r;   rY   r<   r   r   r   r,   ;  rZ   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  rZ   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;   rY   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   rY   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   r\   ilocr]   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   r[   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>rm   r   c                 S   r   r   r   r   r   r   r   r   {  r   )r   r   randomrandnr   r   r   itemsr]   r   r@   ry   r   rz   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positionrm   c                 S   rb   )Nr  r   )rr   r   r   f  r   ztest_apply_bug.<locals>.fr   r   i  r   r   r/   r   r   rJ   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   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>r[   )r   r   r]   r   r   r   r   r   r   test_apply_attach_name  s   r  c                 C   s2   | j dd dd}t| j| jd}t|| d S )Nc                 S   r  r;   r  r<   r   r   r   r,     r  z.test_apply_attach_name_axis1.<locals>.<lambda>r   r   r[   )r   r   r   r   r   r   r   r   r   test_apply_attach_name_axis1  s   r  c                 C   sD   |  dd }tt| jt| jdf| j| jd}t|| d S )Nc                 S      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   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   rk   )rl   r   r   r   r/   rJ   r.   r}   r~   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
check_like)r   from_arraysr   r   r   r@   )r   rM   r   r   r   r   r   test_apply_multi_index  s
   $$r)  z	df, dictsr   r  spameggs)r   r   r/   rJ   c                 C   sV   dd }| j |dd}| j |dd}|  |}t|| t||  t|| d S )Nc                 S   rX   r;   )to_dictr<   r   r   r   r,   	  rZ   z!test_apply_dict.<locals>.<lambda>rh   r   re   )r   r   r   r@   )r4   dictsfnreduce_truereduce_falsereduce_noner   r   r   test_apply_dict  s   
r2  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,     rZ   ztest_applymap.<locals>.<lambda>r/   c                 S   s   | | fS r;   r   r<   r   r   r   r,     rZ   r   r   )applymapr   r@   typer1   tuple)r   appliedr   r   r   r   test_applymap  s
   
r7  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   r3  dtypesobject)r8  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]   r3  strr   r@   )r4   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   r3  r=  r^   )colr8  r4   r   r   r   r   test_applymap_datetimelike2  s   
(rC  r   ABCrm   r[   r   r!   r"   c                 C   r:   r;   r   r<   r   r   r   r,   G  r>   c                 C   s   |  |}t||  d S r;   )r3  r   r@   )r   rT   r   r   r   r   test_applymap_empty>  s   
rF  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  rZ   z&test_applymap_kwargs.<locals>.<lambda>rH  r}   r~   )r   r3  r   r@   )r   r   r   r   r   test_applymap_kwargsN  s   "rJ  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   rK  r;   rL  r<   r   r   r   r,   \  r   r
   )	na_action)r3  r   r   r   randintr0   boolrP   NAr   r@   )r   strlen_framefloat_frame_with_namaskstrlen_frame_na_ignorestrlen_frame_with_nar   r   r   test_applymap_na_ignoreU  s   

rW  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   rT   g     z*test_applymap_box_timestamps.<locals>.func)r   r   r   r3  )serrT   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   rk   rl   r   c                 S   s
   t | jS r;   )r4  __name__r<   r   r   r   r,     r-   z#test_applymap_box.<locals>.<lambda>r   	TimedeltaPeriod)r   r   rP   rb  rc  r3  r   r@   rz   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])pandas.tseries.offsetsrf  r   r   r3  rg  r2   r3   r   re  r   (test_frame_apply_dont_convert_datetime64  s   rj  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 )N1day)rP   rb  r<   r   r   r   r,         z
2015-01-02)r   r   r   r   r@   rz   r   r   r   test_apply_non_numpy_dtype  s   rn  c                  C   s2   t dg didd} | dd }t||  d S )Nrk  )rD   rk   rl   rD   r(   ri   c                 S   r:   r;   r   r<   r   r   r   r,     r>   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>rc   r3   r   r   r   #test_apply_non_numpy_dtype_category  s   ro  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   rm   r#  r   rE   r4   r   r   r   r   r   test_apply_dup_names_multi_agg  s   rq  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.   ABCDrm   r   r   )        rw  rw  )r   r   zerosr   ru   r   r   r   )rr  ru  r4   r   r   r   r   r   test_apply_nested_result_axis_1  s   ry  c                  C   sR   t dtdddgidd} | dd }t||  | d	d }t||  d S )
Nr   2020r7   r8   zdatetime64[ns, UTC]ri   c                 S   r:   r;   r   r<   r   r   r   r,     r>   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   rX   r;   )r   r<   r   r   r   r,     rZ   r?   rA   r   r   r   %test_apply_noreduction_tzaware_object  s   r{  c                     sh   t dg di} g   fdd} fdd}||fD ]} d d = | j|dd  t| jks1J qd S )	NrD   r|   c                    s     | j d S r;   rn   r  rs  namesr   r   reducing_function  r[  z8test_apply_function_runs_once.<locals>.reducing_functionc                    s     | j | S r;   r|  rs  r}  r   r   non_reducing_function  s   z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r   r   r4   r  r  rT   r   r}  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   r|   c                         |  d S r;   extendrs  r   r   r   r       z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                         |  | S r;   r  rs  r  r   r   r       
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r   r   )r   r   r   rD   to_listr  r   r  r   !test_apply_raw_function_runs_once  s   
r  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   r|   c                    r  r;   rn   r8  r  r   r   r    r  z;test_applymap_function_runs_once.<locals>.reducing_functionc                    r  r;   r  r  r  r   r   r    r  z?test_applymap_function_runs_once.<locals>.non_reducing_function)r   r3  rD   r  r  r   r  r    test_applymap_function_runs_once  s   

r  c                  C   sP   t tddgdgd} t tddgdgtd}| dd }t|| d S )Ns   abcds   efghrB  rm   )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@   rp  r   r   r   test_apply_with_byte_string  s   r  asd   c                    s`   dd dddt jg}td|idd}|j fdd}t fd	d
|D dd}t|| d S )Nr  r  cderD   r(   ri   c                    s   |  kS r;   r   r<   r  r   r   r,     rZ   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t |rtjn| kqS r   )rP   isnullr   NaN)r   r=   r  r   r   
<listcomp>  s   $ z1test_apply_category_equalness.<locals>.<listcomp>r  )r   r  r   rD   r   r   r   r   )r8  	df_valuesr4   r   r   r   r  r   test_apply_category_equalness  s   r  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/   )r~   r/   )r   r   r   randr   fftr0   rfftr3   r   r   r   test_infer_row_shape"  s
   r  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   rk   rm   c                 S      d| d | d  iS NrM   rD   rk   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   r  1  r   z.test_with_dictlike_columns.<locals>.<listcomp>2017-05-01 00:00:002017-05-02 00:00:00r   c                 S   r  r  r   r<   r   r   r   r,   8  r   c                 S   s   d| iS )NrM   r   r<   r   r   r   r,   <  rZ   rM   rJ   )r   r   r   r  r   r   r   rz   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   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   rz   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   rk   rm   c                 S   r  r  r   r<   r   r   r   r,   R  r   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>re   rf   rM   rJ   r  r  r   c                 S   r  r  r   r<   r   r   r   r,   Z  r   )r   r   r   r@   r   rz   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   ofwordsz
2016-10-01H)rL   r_  )rD   rk   r+   rD   rk   r   r   c                 S      g | ]}|d d qS rC   r   r  r   r   r   r  i      z.test_with_listlike_columns.<locals>.<listcomp>r+   c                 S   r  rC   r   r  r   r   r   r  m  r  )r   r   r   r   r   r   r   r5  r  r   r   rz   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   rk   qzr  rG  )r   j0)r   j1c                    s    fdd d D S )Nc                    s   g | ]
}| d  v r|qS rI  r   )r   elrs  r   r   r  v  s    zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>r=   r   rs  r   rs  r   r,   v  r%  z;test_with_listlike_columns_returning_list.<locals>.<lambda>r   r   r[   )r   r   r   from_tuplesr   r   r   r   rz   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  rs  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   r    r  z3test_infer_output_shape_columns.<locals>.<listcomp>r   r   r   r   r  r   r   rz   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 )Nr~   rJ   rE  rm   c                 S   r   r   r   r<   r   r   r   r,     rZ   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    r   z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S      ddgS Nr   r/   r   r<   r   r   r   r,     rZ   c                 S   s   g | ]}d dgqS r   r/   r   r  r   r   r   r    r   	r   r   r   r   r   r   r  r   r   rz   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   r|   r   r[   c                    s
   t  S r;   r   onesrs  r  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  r  r   r   r    r%  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>)r   r   r   r   r  r   r   r   )r8  r4   r   r   r   r  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   r}   r.   rJ   )r}   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   r        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  lstr|   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   rE  rm   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   r    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 )Nr|   )testotherr?  r[   r   r<   r   r   r   r,     r   z'test_consistent_names.<locals>.<lambda>r   r   r  r  r?  rE  rm   c                 S   s   t ddgddgdS )Nr   r/   r  r  r[   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,     rZ   z"test_result_type.<locals>.<lambda>r   re   rf   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,     rZ   z/test_result_type_shorter_list.<locals>.<lambda>r   re   rf   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,     rZ   z,test_result_type_broadcast.<locals>.<lambda>r   r   rf   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  rB  r~  c                       t g d dS Nr|   r[   r  r<   rm   r   r   r,     r  z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r   rf   r   r   r   rm   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<   rm   r   r   r,     r  z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   r  r   r   rm   r   *test_result_type_series_result_other_index  s   r  boxc                 C      t | S r;   )r   r<   r   r   r   r,     rZ   c                 C   r  r;   )r5  r<   r   r   r   r,     rZ   c                 C   s   t j| ddS )Nr   ri   )r   r   r<   r   r   r   r,     rm  )r   r5  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   r  $  r  z.test_consistency_for_boxed.<locals>.<listcomp>c                    r  r  r   r<   r  r   r   r,   '  r   re   rf   r   r!   r   rO   rm   )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   from_productr]   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}   rO   r#  r$  r   r.   r   r!   r   )r   r   rF   r   r@   rz   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 )Nr}   rO   r#  r$  rr         @rw        @g      9@r   r!   )r$  r#  rr   r   )r   r   rF   r   r\   r   r@   reindex_likerz   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 )Nr|   r   r         @r   r  bazrE  r   r~   r         @r  	foobarbazr#  rr   r[   )r"   r!   r   )r   rF   r   r@   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 )Nr|   r  r  rI   rJ   rK   r   r!   r"   r#   does not support reductionmatchr#  rr   )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$  rr   r   rr   r[   >   r]   r   r$  )axesuniquesort_valuesrP   concatr   r$  rr   r]   ry   rF   r   r@   r   r^   r   r   )r   r   r  name1name2r   r   rT   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  rI   rJ   rK   r  r#  r   r   r  r[   r  r  r  rr   rE  )r~   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  rE  r   r  r/   rM  rJ   )r  rM  Tr&  )	r   r   r\   ru   r   r   r   r@   r  )r  r4   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  rE  rM  r   )r   r   r   r[   r   )
r   r   r\   ru   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  rE  c                 S   s   t |   S r;   )r   dropnar  )	group_colr   r   r   rT   R  r[  z&test_agg_listlike_result.<locals>.funcr/   rJ   r   r  r[   rT   )
r   r   r\   rF   r   r   r   to_framery   r@   )r4   rT   r   r   r   r   r   test_agg_listlike_resultN  s    
 r  zargs, kwargs)r   r      r  r   rk   r   rD   rk   rj   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;   rr   )r=   rD   rk   rl   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   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   r:   r;   r   r<   r   r   r   r,     r>   z,test_frequency_is_original.<locals>.<lambda>)rP   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   )
timestampsr4   r   r   r   r   r   test_apply_datetime_tz_issue  s   


r"  r4   rD   rl   r   rO   method)r#  r$  rr   c                 C   sR   |dkrd}nd}t jt|d t| |  W d    d S 1 s"w   Y  d S )Nrr   z-can only concatenate str \(not "int"\) to strz4not supported between instances of 'str' and 'float'r  )r  r   r  ru   )r4   r#  r  r   r   r   test_mixed_column_raises  s   "r$  rB  c                 C   s:   t d| ggddgd}|dd }|j}t|| d S )Nr   rD   rk   rm   c                 S   r  r;   ri   r<   r   r   r   r,     r  z"test_apply_dtype.<locals>.<lambda>)r   r   r9  r   r   )rB  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   )_mgrr^   )rt  mgrr   r   r   rT     s   z!test_apply_mutating.<locals>.funcrD   r   r   )r   r   r   r   r   r@   )using_array_managerusing_copy_on_writer4   df_origrT   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.   r}   r~   r   r   r   r   rD   rk   rm   c                 S   s   g S r;   r   r<   r   r   r   r,     r>   z.test_apply_empty_list_reduce.<locals>.<lambda>rh   r   r  ri   )r   r   r   r:  r   r   rz   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!   rm   rr   c                 S   rX   r;   r  r<   r   r   r   r,     rZ   z,test_apply_no_suffix_index.<locals>.<lambda>)r  r  r  )   r/  r/  rO   )rr   r,   r,   r[   rc   )pdfr   r   r   r   r   test_apply_no_suffix_index  s   r1  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   rb   r   r   r<   r   r   r   r,     rZ   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr   r   rz   r   r   r   test_apply_raw_returns_string  s   r4  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.   r}   r.   )r}   r~   r~   )r~   r   r   )att1att2att3rm   c                 S   s   |   d S r   r  r   r   r   r   sum_div2  r   z4test_aggregation_func_column_order.<locals>.sum_div2rr   r  r#  )g      5@g      %@r  r   )g      2@g      "@r  rw  )g      1@g      !@r  rw  )rr   r8  r  r#  r[   rE   )r4   r8  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  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   rz   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   warnrr   r<   r   r   r   expected_warning  s   
zDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warningrD   r|   r<  r  )r   r   assert_produces_warningUserWarningrF   )r>  r4   r   r   r   *test_nuisance_depr_passes_through_warnings  s
   "rA  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"  rj   r[   r   r   )r   rv   r   r   r\   r3  r4  intr=  r   r@   r   r   r   rz   r   r   r   test_apply_type  s    
rC  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   rk   )r$  r<   r   r   r   r,   5  r   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   ri   )r   headr   r   r   ro   r   r   rz   r   r   r   test_apply_on_empty_dataframe2  s   rE  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   rk   r:  )rG   r   r2   r  r  r  )r  constantdf1r   aggregationr   r   r   r   test_unique_agg_type_is_series:  s
   	

rI  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   rE  )TTFr   r   rt   )r   r   r   r   rt   r   rp  r   r   r   /test_any_apply_keyword_non_zero_axis_regressionL  s   rJ  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=   r|   r   r   c                 S      | | | S r;   r   )r=   rD   rl   r   r   r   foo1]  r   z/test_agg_list_like_func_with_args.<locals>.foo1r/   c                 S   rK  r;   r   )r=   rk   rl   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.   )rk   rl   )rl   r   r   r   )r=   rL  )r=   rM  rm   )r   r   )r/   r   )	r   r  r   r  rF   r   r  r   r@   )r4   rL  rM  r  r   r   r   r   r   !test_agg_list_like_func_with_argsY  s   

rN  )r   r   numpyr   r  pandas.core.dtypes.dtypesr   pandasrP   r   r   r   r   r   pandas._testing_testingr   pandas.tests.frame.commonr	   r    r5   rB   markparametrizerH   rS   r   r   rW   ra   rd   rq   rw   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)  r2  r7  r;  r@  rb  rC  r   roundrF  rJ  rW  r]  rd  rj  rn  ro  rq  ry  r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r   r"  r$  r\   r%  r-  r.  r1  r4  r:  r;  rA  rC  rE  rI  rJ  rN  r   r   r   r   <module>   sx   






					"4	*



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

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