o
    i)                     @   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 Zrejdg ddd 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dd  Zwejdg ddd Zxdd Zyejde	ddgd	d
gdgejdg 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   ^/var/www/edux/Edux_v2/venv/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
   r6   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   rC   zdata, dtyper   Nr   r7   r8   r9   c                 C   s<   t | g| ggddg|d}|jdd dd}t|| d S )Nar   r3   c                 S   r;   r<   r   r=   r   r   r    r-   @   r?   z0test_agg_axis1_duplicate_index.<locals>.<lambda>r   r   r   aggr   rA   )datar3   r   r   r   r   r    test_agg_axis1_duplicate_index9   s   rJ   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   rA   rB   r   r   r    test_apply_mixed_datetimelikeD   s   
rU   funcc                 C   s   t  }|| }|jsJ d S r<   )r   r   empty)rV   empty_framer   r   r   r    test_apply_emptyQ   s   
rY   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   rZ   r<   r[   r=   r   r   r    r-   a   r\   r   r   )	r   r   r   nancolumnsr   r   locr   )r   no_rowsr   r   no_colsr   r   r    test_apply_float_frameZ   s   rc   c                  C   s.   t dgd} | jdd dd}t||  d S )NrE   r]   c                 S      | d S )NrE   r   r=   r   r   r    r-   i   r\   z/test_apply_empty_except_index.<locals>.<lambda>r   r   r   r   r   rA   rB   r   r   r    test_apply_empty_except_indexf   s   rf   c                  C   s   t  } g }| j|jddd}t||  | j|jddd}tg tjg tdt	j
d}t|| t g dd}|j|jddd}t|| |j|jddd}tg tjg tdt	j
d}t|| |g kshJ d S )	Nr   expandr   result_typereducer3   rF   rE   bcr_   )r   r   appendr   rA   r   rR   Indexobjectr   float64r   )rX   r>   r   r   empty_with_colsr   r   r    test_apply_with_reduce_emptym   s   ru   )sumprodanyr   c                 C   s:   t g dd}|tt| }t||  }t|| d S )Nrl   ro   )r   r   getattrr   r   r   )rV   r5   r   r   r   r   r    test_apply_funcs_over_empty   s   rz   c                  C   s`   t g dd} |  }td| jd}t|| | j }tg tg t	j
d}t|| d S )Nrl   ro   r   r]   rF   )r   nuniquer   r_   r   r   TrR   rq   r   rs   r5   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   r0   rL   )r/         )      	   )rE   rE   rn   r]   c                 S   rd   Nr   r   rO   r   r   r    r-      r\   z/test_apply_standard_nonunique.<locals>.<lambda>r   r   )r   r/   r   c                 S   rd   r   r   r   r   r   r    r-      r\   r   )r   r   r   r   r   r|   r}   r   r   r    test_apply_standard_nonunique   s   "r   c                 C   s4   | j tjdd}t|  g| jd}t|| d S )N	broadcastri   r]   )r   r   r   r   r   r   rA   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   rh   r   c                       i | ]}| qS r   r   .0rn   mr   r    
<dictcomp>       z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>)r   r   r   r   r_   r   rA   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   rh   rs   )r3   r   r_   )	r   r   r   r   r_   r   r   r   rA   )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>rs   )r3   r   )	r   r   r   r   r   r   r_   r   rA   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-      r\   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>r   r   rh   r   r   rA   )int_frame_const_colr5   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   rh   r   copyr   rA   r   r5   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   r2   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 Nr0   r   r=   r   r   r    r-      r\   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>T)r   r0   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      ?rQ   r   r   r   int64rk   r]   )
r   r   r   r   r   r^   rR   rq   r   r   r}   r   r   r    test_apply_mixed_dtype_corner  s   r   c                  C   sr   t dgdgd} | jdd dd}tdgdgd	}t|| | jd
d dd}tdgdgd	}t|| d S )Nr   r   rQ   c                 S   rd   )Nr   r   r=   r   r   r    r-     r\   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>r   r   r   r]   c                 S   rd   )Nr"   r   r=   r   r   r    r-     r\   r   r   r   r   r   r}   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   rZ   r<   r[   r=   r   r   r    r-   !  r\   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 )
Nrl   r   r   T)recordf8rk   r   r   )r   r   r   warningscatch_warningssimplefilterRuntimeWarningarrayr2   r   r   _get_agg_axisr   r   )	r   rV   r   r   r5   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 )Nrl   r]   c                 S   rZ   r<   r[   r=   r   r   r    r-   9  r\   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>r   r   )r   r   r2   )rb   r   r   r   r    %test_apply_empty_infer_type_broadcast7  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_somer0   r   c                 S   s   | d S r   r   r=   r   r   r    r-   B  r\   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>r   r   )r   r   r   r   r   r   r    "test_apply_with_args_kwds_add_some=     
r   c                 C   r   )	Nr   c                 S   s   |   | S r<   r[   r   r   r   r    agg_and_addG     z:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addr0   r   c                 S   s   |   d S r   r[   r=   r   r   r    r-   K      z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>r   )r   r   r   )r   r   r   r   r   r   r    %test_apply_with_args_kwds_agg_and_addF  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_divideP  r   zJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide)r0   r0   )argsr   c                 S   s   | d d S )N       @r   r=   r   r   r    r-   T  r   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>r   r   )r   r   r   r   r   r   r    -test_apply_with_args_kwds_subtract_and_divideO  s   
r   c                 C   s   |  t}t||  d S r<   )r   r   r   rA   )r   r   r   r   r    test_apply_yield_listX  s   
r   c                 C   s@   t j| d jd d d< | d}| jt jdd}t|| d S )Nr   r0   r   r   )r   r^   ilocr   r   r   r   )r   r   r   r   r   r    test_apply_reduce_Series]  s   
r   c                  C   s   t ddgddggddgddgd	} | jtd
d}tddddddg| jd}t|| | jtdd}tddddddg| jd}t|| d S )Nr   r0   rL   r/   r&   r'   i0i1r_   r   r   r   )r   r   r]   r%   )r   r   dictr   r_   r   r   r   )rI   r   r   r   r   r    test_apply_reduce_to_dictd  s   "r   c                  C   s   t tjdd} | jtjdd}t dd |  D | jd}t	
|| | jtjdd}t d	d | j D | jdj}t	
|| d S )
N   
   r   r   c                 S      i | ]	\}}||  qS r   describer   ivr   r   r    r   u      z2test_apply_differently_indexed.<locals>.<dictcomp>ro   r   c                 S   r   r   r   r   r   r   r    r   y  r   )r   r   randomrandnr   r   r   itemsr_   r   rA   r|   r   r}   r   r   r    test_apply_differently_indexedq  s    r   c               
   C   s   t g dg dg dg dg dg dgg dd} d	d
 }| j|dd}t tdddddgtdddddgtdddddgtdddddgtdddddgtdddddggg dd} | j|dd}t|| d S )N)r   ABC02   )r   YUM0r   )r   DEF0r   )r0   ABC1r   )r0   YUM1r   )r0   DEF1r   )rE   marketpositionro   c                 S   rd   )Nr  r   )rr   r   r    f  r   ztest_apply_bug.<locals>.fr   r   i  r   r   r0   r   r   rL   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   sd   t g dg dg dtjdtjdtjdd} | jdd dd	jd
d}t||  d S )N)r   r   r   r   barr  r  r  r   r   r   )oner	  r	  twor	  r	  r	  r
  r
  r
  r	  )dullr  shinyr  r  r  r  r  r  r  r     )r   r"   r#   r$   EFc                 S   r;   r<   r   r=   r   r   r    r-     r?   z,test_apply_convert_objects.<locals>.<lambda>r   r   Tr   )r   r   r   r   r   _convertr   rA   rB   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   rA   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)rE   rE   rm   )rn   r   r   r   r0   rL   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   rA   )r   rO   r   r   r   r   r    test_apply_multi_index  s
   $$r+  z	df, dictsr   r  spameggs)r   r   r0   rL   c                 C   sV   dd }| j |dd}| j |dd}|  |}t|| t||  t|| d S )Nc                 S   rZ   r<   )to_dictr=   r   r   r    r-     r\   z!test_apply_dict.<locals>.<lambda>rj   r   rg   )r   r   r   rA   )r5   dictsfnreduce_truereduce_falsereduce_noner   r   r    test_apply_dict  s   
r4  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-     r\   ztest_applymap.<locals>.<lambda>r0   c                 S   s   | | fS r<   r   r=   r   r   r    r-     r\   r   r   )applymapr   rA   typer2   tuple)r   appliedr   r   r   r    test_applymap  s
   
r9  valr   c                 C   s2   t | dgd}|dd jd }|tksJ d S )NrE   )rI   c                 S   r;   r<   r   r=   r   r   r    r-      r?   z7test_applymap_float_object_conversion.<locals>.<lambda>r   )r   r5  dtypesrr   )r:  r5   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rL   r/   )rE   rE   rE   rE   )	r   r   r   r   r_   r5  strr   rA   )r5   df2colsr   r   r   r   r    test_applymap_str$  s   

rA  zcol, valr   rK   	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   r5  r>  r`   )colr:  r5   r   r   r   r    test_applymap_datetimelike1  s   
(rD  r   ABCro   r]   r   r"   r#   c                 C   r;   r<   r   r=   r   r   r    r-   F  r?   c                 C   s   |  |}t||  d S r<   )r5  r   rA   )r   rV   r   r   r   r    test_applymap_empty=  s   
rG  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   r0   rL   r/   c                 S   r   r<   r   r>   yr   r   r    r-   O  r\   z&test_applymap_kwargs.<locals>.<lambda>rI  r   r   )r   r5  r   rA   )r   r   r   r   r    test_applymap_kwargsM  s   "rK  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-   V  r   z)test_applymap_na_ignore.<locals>.<lambda>r   r0   )sizer3   c                 S   rL  r<   rM  r=   r   r   r    r-   [  r   r   )	na_action)r5  r   r   r   randintr1   boolrR   NAr   rA   )r   strlen_framefloat_frame_with_namaskstrlen_frame_na_ignorestrlen_frame_with_nar   r   r    test_applymap_na_ignoreT  s   

rX  c                  C   s*   t tddd} dd }t| | d S )Nz1/1/2000r   rM   c                 S   s   | j | j| jfS r<   )hourdaymonthr=   r   r   r    rV   f     z*test_applymap_box_timestamps.<locals>.func)r   r	   r   r5  )serrV   r   r   r    test_applymap_box_timestampsb  s   r^  c               	   C   s   t tdtdgtdddtdddgtdtdgtjdddtjdddgd	} | d
d }t ddgddgddgddgd	}t|| d S )Nz
2011-01-01z
2011-01-02z
US/Easternr9   z1 daysz2 daysM)freqrE   rm   rn   r   c                 S   s
   t | jS r<   )r6  __name__r=   r   r   r    r-   ~  r.   z#test_applymap_box.<locals>.<lambda>r   	TimedeltaPeriod)r   r   rR   rc  rd  r5  r   rA   r}   r   r   r    test_applymap_boxm  s(   

re  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=   rf  r   r    r-     r.   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>c                    ri  r<   r   r=   rf  r   r    r-     r.   zM8[ns])pandas.tseries.offsetsrg  r   r   r5  rh  r3   r4   r   rf  r    (test_frame_apply_dont_convert_datetime64  s   rk  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-01rL   zEurope/Brussels)rN   r:   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)rR   rc  r=   r   r   r    r-         z
2015-01-02)r   r	   r   r   rA   r}   r   r   r    test_apply_non_numpy_dtype  s   ro  c                  C   s2   t dg didd} | dd }t||  d S )Nrl  )rE   rm   rn   rE   r)   rk   c                 S   r;   r<   r   r=   r   r   r    r-     r?   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>re   r4   r   r   r    #test_apply_non_numpy_dtype_category  s   rp  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   r0   rL   rE   ro   r%  r   rG   r5   r   r   r   r   r    test_apply_dup_names_multi_agg  s   rr  opr   rH   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 )Nr0   r   r#   r"   r   rowr   r   r    
apply_list  s   "z3test_apply_nested_result_axis_1.<locals>.apply_list)r/   r/   ABCDro   r   r   )        rx  rx  )r   r   zerosr   ry   r   r   r   )rs  rv  r5   r   r   r   r   r    test_apply_nested_result_axis_1  s   rz  c                  C   sR   t dtdddgidd} | dd }t||  | d	d }t||  d S )
Nr   2020r8   r9   zdatetime64[ns, UTC]rk   c                 S   r;   r<   r   r=   r   r   r    r-     r?   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>c                 S   rZ   r<   )r   r=   r   r   r    r-     r\   r@   rB   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 )	NrE   r   c                    s     | j d S r<   rp   r  rt  namesr   r    reducing_function  r\  z8test_apply_function_runs_once.<locals>.reducing_functionc                    s     | j | S r<   r}  rt  r~  r   r    non_reducing_function  s   z<test_apply_function_runs_once.<locals>.non_reducing_functionr   r   )r   r   r   r   r5   r  r  rV   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 )
NrE   r   c                         |  d S r<   extendrt  r   r   r    r       z<test_apply_raw_function_runs_once.<locals>.reducing_functionc                         |  | S r<   r  rt  r  r   r    r       
z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionTr   )r   r   )r   r   r   rE   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 )NrE   r   c                    r  r<   rp   r:  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   r5  rE   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   efghrC  ro   )r_   r3   c                 S   r(   )Nrr   r*   r=   r   r   r    r-     r.   z-test_apply_with_byte_string.<locals>.<lambda>)r   r   r   rr   r   r   rA   rq  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  cderE   r)   rk   c                    s   |  kS r<   r   r=   r  r   r    r-     r\   z/test_apply_category_equalness.<locals>.<lambda>c                    s$   g | ]}t |rtjn| kqS r   )rR   isnullr   NaN)r   r>   r  r   r    
<listcomp>  s   $ z1test_apply_category_equalness.<locals>.<listcomp>r  )r   r  r   rE   r   r   r   r   )r:  	df_valuesr5   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   r0   r   r   )r   r0   )r   r0   )r   r   r   randr   fftr1   rfftr4   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   r0   rE   rm   ro   c                 S      d| d | d  iS NrO   rE   rm   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 )rO   rL   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 )NrO   r   r=   r   r   r    r-   <  r\   rO   rL   )r   r   r   r!  r   r   r   r}   r   r   r    test_with_dictlike_columns-  s   r  c                  C   s`   t  } g d| d< g d| d< tg d| d< | jdd d	d
}ti i i g}t|| d S )N)XYZauthor)BBCNBCN24	publisher)z17-10-2010 07:15:30z13-05-2011 08:20:35z15-01-2013 09:09:09datec                 S   s   i S r<   r   r=   r   r   r    r-   I  r?   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>r   r   )r   rR   to_datetimer   r   r   r   r}   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   r0   rE   rm   ro   c                 S   r  r  r   r=   r   r   r    r-   Q  r   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>rg   rh   rO   rL   r  r  r   c                 S   r  r  r   r=   r   r   r    r-   Y  r   )r   r   r   rA   r   r}   r   r   r    %test_with_dictlike_columns_with_inferN  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/   )rE   r   ofwordsz
2016-10-01H)rN   r`  )rE   rm   r,   rE   rm   r   r   c                 S      g | ]}|d d qS rD   r   r  r   r   r    r  h      z.test_with_listlike_columns.<locals>.<listcomp>r,   c                 S   r  rD   r   r  r   r   r    r  l  r  )r   r   r   r   r   r	   r   r7  r!  r   r   r}   r   r   r    test_with_listlike_columns]  s   r  c                  C   sp   t tddgdggtdgddggd} tddg| _| jd	d
 dd}tg dgg| jd}t|| d S )NrE   rm   qzr  rH  )r   j0)r   j1c                    s    fdd d D S )Nc                    s   g | ]
}| d  v r|qS rJ  r   )r   elrt  r   r    r  u  s    zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>r>   r   rt  r   rt  r    r-   u  r'  z;test_with_listlike_columns_returning_list.<locals>.<lambda>r   r   r]   )r   r   r   from_tuplesr   r   r   r   r}   r   r   r    )test_with_listlike_columns_returning_listp  s
   *r  c                  C   s\   t ddgddgtdtdgd} | jdd	 d
d}tdd |  D }t|| d S )Nr   r   r   r  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                 S   s   | j | jfS r<   r  r  rt  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   r}   r   r   r    test_infer_output_shape_columnsz  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   rL   rF  ro   c                 S   r   r   r   r=   r   r   r    r-     r\   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>r   r   c                 S   s   g | ]}g d qS )r   r   r  r   r   r    r    r   z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>c                 S      ddgS Nr   r0   r   r=   r   r   r    r-     r\   c                 S   s   g | ]}d dgqS r   r0   r   r  r   r   r    r    r   	r   r   r   r   r   r   r!  r   r   r}   r   r   r    (test_infer_output_shape_listlike_columns  s   r  c                    s^   t dg ditdd}|j fdddd}t fd	d
| D |jd}t|| d S )NrE   r   r   r]   c                    s
   t  S r<   r   onesrt  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   )r:  r5   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/   rL   )r   rL   r/   r0   )r   r0   rL   r/   ra  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  )r5   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/   rL   rF  ro   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  r5   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@  rF  ro   c                 S   s   t ddgddgdS )Nr   r0   r  r  r]   r  r=   r   r   r    r-     r   )r   renamer   rA   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-     r\   z"test_result_type.<locals>.<lambda>r   rg   rh   r   r   r0   r   r   r_   r   rA   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-     r\   z/test_result_type_shorter_list.<locals>.<lambda>r   rg   rh   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-     r\   z,test_result_type_broadcast.<locals>.<lambda>r   r   rh   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  rC  r  c                       t g d dS Nr   r]   r  r=   ro   r   r    r-     r  z8test_result_type_broadcast_series_func.<locals>.<lambda>r   r   rh   r   r   r   ro   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=   ro   r   r    r-     r  z<test_result_type_series_result_other_index.<locals>.<lambda>r   r   r  r   r   ro   r    *test_result_type_series_result_other_index  s   r  boxc                 C      t | S r<   )r   r=   r   r   r    r-     r\   c                 C   r  r<   )r7  r=   r   r   r    r-     r\   c                 C   s   t j| ddS )Nr   rk   )r   r   r=   r   r   r    r-     rn  )r   r7  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   rg   rh   r   r"   r   rQ   ro   )r   r   r!  r   r   r  rA   )r  r   r5   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   rA   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   rQ   r%  r&  r   r/   r   r"   r   )r   r   rH   r   rA   r}   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   rQ   r%  r&  rv         @rx        @g      9@r   r"   )r&  r%  rv   r   )r   r   rH   r   r^   r   rA   reindex_liker}   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  countr0   r  )r   rH   r   r   r   )rI   r5   r   r   r   r   r    !test_agg_with_name_as_column_nameh  s   r  c                  C   s  t g dg dg dtdddd} t dd	gd
dgddgtdtjgdddgd}tjtdd | ddg}W d    n1 sDw   Y  t	|| tjtdd | g d ddg}W d    n1 slw   Y  |g d 
ddg}t	|| d S )Nr   r   r         @r   r  bazrK   rL   rM   r   r"   r#   r$   r   r   r         @r  	foobarbazr7   r%  rv   r]   &\['D'\] did not aggregate successfullymatch)r$   r#   r"   r   )r   r	   r   rR   NaTr   assert_produces_warningFutureWarningrH   rA   reindex)mdfr   r   r   r   r    "test_agg_multiple_mixed_no_warningx  s8   

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   r0   r   )r   r&  rv   r   rv   r]   >   r_   r   r&  )axesuniquesort_valuesrR   concatr   r&  rv   r_   r|   rH   r   rA   r   r`   r   r   )r   r   r  name1name2r   r   rV   r   r   r    test_agg_reduce  sf   



r  c                  C   sJ  t g dg dg dtdddd} | d}td	d
dtdg| jd}t|| | dg}t d	d
dtdggdg| jd}t|| tj	t
dd | d}W d    n1 s^w   Y  tg dg dd}t|| tj	t
dd | dg}W d    n1 sw   Y  t g dgdgg dd}t|| d S )Nr   r  r  rK   rL   rM   r  r%  r   r   r  r]   r  Select only validr  rv   )r   r  r  rF  r  )r   r	   rH   r   r   r_   r   r   rA   r  r  r}   r   r   r    test_nuiscance_columns  s>   

	r
  method)rH   r   	transformc              	   C   sz  t g dtdd}| dkrTd}tjt|d t|| tj W d    n1 s+w   Y  tj	t
dd t|| tjdd	 W d    d S 1 sMw   Y  d S | d
krztj	t
dd t|| tj W d    d S 1 ssw   Y  d S tj	tdd2 d}tjt|d t|| tj W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )N)r   r   r0   xyzrE   rm   rH   z$The operation <function mean.*failedr  z"got an unexpected keyword argumentT)numeric_onlyr   zCould not convertzFunction did not transform)r   r   r   r  r  ry   r   r   pytestraises	TypeError
ValueError)r  r5   msgr   r   r    test_numeric_only_warning_numpy
  s*   """r  howc                 C   s  t g ddtjdgg dd}t|| ddi}tddi}t|| t|| dd	i}tdd
i}t|| t|| dd	g}t|| dd	gdd	gdd	gd}t dd
ddd
ddd
dd}tj||dd tj||dd t|| d}| }t|| d S )NNr0   rL   r   r  r   Nr  rF  r   r  r0   rN  rL   )r  rN  Tr(  )	r   r   r^   ry   r   r   r   rA   r  )r  r5   r   r   result1result2r   r   r    test_non_callable_aggregates  s0   r  c                 C   s|   t g ddtjdgg dd}t|| d|d}|dks!|d	kr,t|jd |jd
}n
t|jd |jd
}t	|| d S )Nr  r   r  r  rF  rN  r   r   r   r]   r   )
r   r   r^   ry   r   r1   r_   r   r   r   )r  r   r5   r   r   r   r   r    test_size_as_strM  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)r0   r0   rL   g      ?r  rF  c                 S   s   t |   S r<   )r   dropnar  )	group_colr   r   r    rV   a  r\  z&test_agg_listlike_result.<locals>.funcr0   rL   r   r  r]   rV   )
r   r   r^   rH   r   r   r   to_framer|   rA   )r5   rV   r   r   r   r   r    test_agg_listlike_result]  s    
 r   zargs, kwargs)r   r      r  r   rm   r   r  rl   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 )NrL   c                 S   s   |   || |  S r<   rv   )r>   rE   rm   rn   r   r   r    r  {  s   ztest_agg_args_kwargs.<locals>.fr   r0   r/   r   r  g      @r  g       @)rL   r   r   rH   r   r   )r   r   kwargsr  r5   r   r   r   r   r    test_agg_args_kwargsm  s   
r%  num_cols)r0   rL   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>)rR   DatetimeIndexr   r   r   r   r`  )r&  r   originalr5   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+0000r8   r9   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000r  )rI   r   c                 S   r  r<   r  r=   r   r   r    r-     r  z.test_apply_datetime_tz_issue.<locals>.<lambda>r   r   )r   rI   )r   r   r   r   r   r   )
timestampsr5   r   r   r   r   r    test_apply_datetime_tz_issue  s   


r+  r5   rE   rn   r   rQ   )r%  r&  rv   c                 C   sf   t jtdd t| ddg | }t| ddg | }W d    n1 s&w   Y  t || d S )Nr	  r  r   r"   )r   r  r  ry   r   )r5   r  none_in_first_column_resultnone_in_second_column_resultr   r   r    =test_consistency_of_aggregates_of_columns_with_missing_values  s
   r.  rC  c                 C   s:   t d| ggddgd}|dd }|j}t|| d S )Nr   rE   rm   ro   c                 S   r  r<   rk   r=   r   r   r    r-     r  z"test_apply_dtype.<locals>.<lambda>)r   r   r;  r   r   )rC  r5   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 )NrE   r   )_mgrr`   )ru  mgrr   r   r    rV     s   z!test_apply_mutating.<locals>.funcrE   r   r   )r   r   r   r   r   rA   )using_array_managerusing_copy_on_writer5   df_origrV   r   r   r   r   r    test_apply_mutating  s   r7  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   r0   rL   r/   r   r   r   r   r   r   rE   rm   ro   c                 S   s   g S r<   r   r=   r   r   r    r-     r?   z.test_apply_empty_list_reduce.<locals>.<lambda>rj   r   r  rk   )r   r   r   rr   r   r   r}   r   r   r    test_apply_empty_list_reduce  s   .r8  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   rL   r   r"   ro   rv   c                 S   rZ   r<   r"  r=   r   r   r    r-     r\   z,test_apply_no_suffix_index.<locals>.<lambda>)r  r  r  )   r9  r9  rQ   )rv   r-   r-   r]   re   )pdfr   r   r   r   r    test_apply_no_suffix_index  s   r;  c                  C   s@   t dddgi} | jdd ddd}tddg}t|| d S )	Nr   aabbbc                 S   rd   r   r   r=   r   r   r    r-     r\   z/test_apply_raw_returns_string.<locals>.<lambda>r   Tr   r   r}   r   r   r    test_apply_raw_returns_string  s   r>  c                  C   s   t g ddd} dd }d|ddg}tjtd	d
 | |}W d    n1 s)w   Y  t dtjddgg dg dg ddg dd}t|| d S )N))1r   r   r   )2r0   r   r   )3rL   r   r   )4r/   r   r/   )5r   r   r   )6r   r   r   )itematt1att2att3ro   c                 S   s   |   d S r   r"  r   r   r   r    r     r   z/test_aggregation_func_column_order.<locals>.foorv   r  r%  z)\['item'\] did not aggregate successfullyr  123456r   r?  )g      5@g      %@r  r   )g      2@g      "@r  rx  )g      1@g      !@r  rx  )rv   r   r  r%  r]   )r   r   r  r  rH   r   r^   rA   )r5   r   aggsr   r   r   r   r    "test_aggregation_func_column_order  s(   	rK  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  rE   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   r0   r/   r   r}   r   r   r    test_apply_getitem_axis_1  s   rL  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   warnrv   r=   r   r   r    r   !  s   
z7test_nuisance_depr_passes_through_warnings.<locals>.foorE   r   rM  r  )r   r   r  UserWarningrH   )r   r5   r   r   r    *test_nuisance_depr_passes_through_warnings  s
   "rP  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 )NrL   r  g      ?i  r   )r#  r$  rl   r]   r   r   )r   floatr   r   r^   r5  r6  intr>  r   rA   r   r   r   r}   r   r   r    test_apply_type*  s    
rS  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   r0   rL   r   r  c                 S   s   t | d | d S )NrE   rm   )r&  r=   r   r   r    r-   G  r   z/test_apply_on_empty_dataframe.<locals>.<lambda>r   rk   )r   headr   r   r   rs   r   r   r}   r   r   r    test_apply_on_empty_dataframeD  s   rU  ztest, constant)r   r   r   )r0   r0   r0   c                 C   s>   t | }t|ddgdd}ddd}||}t|| d S )NrE   rm   rr   )rI   r   r3   r  r  r#  )r  constantdf1r   aggregationr   r   r   r    test_unique_agg_type_is_seriesL  s
   	

rY  c                  C   s   t ddgddgddgd} d}tjt|d | dd }W d    n1 s(w   Y  tddd	d}t|| tg d
}d}tjt|d |d}W d    n1 sZw   Y  d	}t|| d S )Nr   r0   r   rF  zaIn a future version of pandas all arguments of DataFrame.any and Series.any will be keyword-only.r  r   TF)FFF)r   r   r  r  rx   r   r   assert_equal)r5   r  r   r   rO   r   r   r     test_any_non_keyword_deprecation^  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   r0   r   )r   r0   r   )r   r   r   rF  )TTFr   r   rx   )r   r   r   r   rx   r   rq  r   r   r    /test_any_apply_keyword_non_zero_axis_regressiont  s   r\  )r   r   numpyr   r  pandas.core.dtypes.dtypesr   pandasrR   r   r   r   r   r	   pandas._testing_testingr   pandas.tests.frame.commonr
   r!   r6   rC   markparametrizerJ   rU   r   r   rY   rc   rf   ru   rz   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+  r4  r9  r<  rA  rc  rD  r   roundrG  rK  rX  r^  re  rk  ro  rp  rr  rz  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/  r7  r8  r;  r>  rK  rL  rP  rS  rU  rY  r[  r\  r   r   r   r    <module>   sz   



	


					#4	*



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



	





&&E'

-

	*			#""