
    o[weZ                        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 Zej                            dd dg          ej                            d	d
dg          d                         Zd Zd Zej                            ddd edg          f edd          dfg          d             Zd Zej                            dej        ej        g          d             Zd Zd Zd Z ej                            dg d          d             Z!d Z"d Z#d Z$d  Z%d! Z&d" Z'd# Z(d$ Z)ej                            dd dg          d%             Z*ej                            dd dg          d&             Z+d' Z,ej                            dd dg          d(             Z-d) Z.d* Z/d+ Z0ej        1                    d,          ej                            d-d.d/g          ej                            dd0 d1 gd2d3g4          ej                            d	d
dg          ej                            dd dg          d5                                                             Z2d6 Z3d7 Z4d8 Z5d9 Z6d: Z7d; Z8d< Z9d= Z:d> Z;d? Z<d@ Z=dA Z>dB Z?dC Z@dD ZAej                            dE e	dFdGgdHdIgg           edFdHdJdGdIdJg          g e	d dgdKdLgg           ed dKdJddLdJg          gg          dM             ZBdN ZCdO ZDdP ZEej                            dQdRdSg          dT             ZFdU ZGdV ZHdW ZIdX ZJej                            dYdZd[dejK        g          d\             ZLd] ZMej                            d^d_d` ida e	d_dKdLgi          fd_db id e	d_dKdLgi          fd_dc ida ed_dLi          fd_dd id ed_dLi          fd_deejN        df gida e	d_g dgig dhi          fd_deejN        dj gid e	d_g dgig dhi          fd_dk ida e	d_ddgi          fd_dl id ed_di          fg          dm             ZOej                            dnd_do id_dp id_deejN        dq gid_dr ig          ds             ZPdt ZQdu ZRdv ZSej                            d^dw gda e	dxdKdLgi          fdy gd e	dxdKdLgi          fdz gda e	d_dLgid{gi          fd| gd e	d_dLgid{gi          fdeejN        d} gda e	d_g dgig dhi          fdeejN        d~ gd e	d_g dgig dhi          fd d gda e	dKdLgdLdLggd_d_gd{d{gg          fd d gd e	d_dKdLgid{d{g          fg          d             ZTej                            dnd gd gdeejN        d gd d gg          d             ZUd ZVd ZWd ZXd ZYej                            dYddKg          d             ZZd Z[ej                            dg dddKgg          d             Z\d Z]d Z^d Z_d Z`d Zad Zbd Zcej                            dd d d gg d4          d             Zdd Zed Zfd Zgd Zhd Zid Zjd Zkd Zlej                            ddSdRg          d             Zmej                            ddSdRg          d             Znd Zoej                            dd dg          ej                            ddi fdi fdi fdddKifdddKdfddKddfdddKdLdfg          d                         Zpej                            dg d          d             Zqd Zrej                            d e	d_dgddgd          g          ej                            dg d          d                         Zsej                            dddd
d_ejK        g          dÄ             ZtdĄ Zudń ZvdƄ ZwdǄ ZxdȄ ZydɄ Zzdʄ Z{d˄ Z|d̄ Z}d̈́ Z~ej                            dg dg dϢdg ddgdfg dТg dϢddKgdgdfg          dф             Zd҄ Zdӄ ZdԄ ZdՄ ZdS )    )datetimeN)CategoricalDtype)	DataFrame
MultiIndexSeries	Timestamp
date_range)
zip_framesc                    t          j        d          5  t          j        | d                   }|                     t           j                  d         }t	          j        ||           |                     t           j                  d         }t          j        | d                   }||k    sJ | j        d         }|                     t           j        d          }t          j        |                     |                    }||         |k    sJ |j        | j        u sJ 	 d d d            d S # 1 swxY w Y   d S )NignoreallAr      axis)	nperrstatesqrtapplytmassert_series_equalmeanindexxs)float_frameresultexpectedds       Clib/python3.11/site-packages/pandas/tests/apply/test_frame_apply.py
test_applyr!      sk   		"	"	" 1 1S)**$$RW--c2
vx000 ""27++C07;s+,,!!!!a ""27"337;>>!,,--ayH$$$$|{0000001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s   DD33D7:D7r   r   rawTFc                 n    |                      d |d|          }| dz   }t          j        ||           d S )Nc                     | |z   S N xys     r    <lambda>z!test_apply_args.<locals>.<lambda>+   s
    AE     r   )argsr"   r   r   r   assert_frame_equal)r   r   r"   r   r   s        r    test_apply_argsr0   (   sF     114dLLFQH&(+++++r+   c                      t          g dg 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   Br2   )Cr3   Dr4   c0c1c                 ,    |                      d          S )Ncategoryastype)tss    r    r*   z-test_apply_categorical_func.<locals>.<lambda>3   s    :!6!6 r+   )      r6   r7   )r   r   shape
isinstancedtyper   dfr   s     r    test_apply_categorical_funcrD   0   s    	...6J6J6JKK	L	LBXX6677F<6!!!!fTl(*:;;;;;fTl(*:;;;;;;;r+   c                      t          dt          dd          gi          } |                     d d          }t          j        ||            d S )Nr   
2013-01-01UTCtzc                     | S r%   r&   r(   s    r    r*   z*test_apply_axis1_with_ea.<locals>.<lambda>=       a r+   r   r   r   r   r   r   r/   r   r   s     r    test_apply_axis1_with_earO   :   sT    #	,5 A A ABCDDH^^KKa^00F&(+++++r+   zdata, dtyper   NrF   rG   rH   c                     t          | g| ggddg|          }|                    d d          }t          j        ||           d S )Na)r   rA   c                     | S r%   r&   rK   s    r    r*   z0test_agg_axis1_duplicate_index.<locals>.<lambda>H       A r+   r   r   r   aggr   r/   )datarA   r   r   s       r    test_agg_axis1_duplicate_indexrX   A   sU     4&4&)#s5IIIH\\++A\..F&(+++++r+   c                      t          t          dd          t          j        t	          j        d          d          d          } |                     d d	          }t          j        ||            d S )
N20130101   periodss)unitr   r2   c                     | S r%   r&   rK   s    r    r*   z/test_apply_mixed_datetimelike.<locals>.<lambda>U   rL   r+   r   r   )	r   r	   pdto_timedeltar   aranger   r   r/   rN   s     r    test_apply_mixed_datetimelikere   L   sy     J2221C888	
 	
 H ^^KKa^00F&(+++++r+   funcc                 ^    t                      }|                    |           }|j        sJ d S r%   )r   r   empty)rf   empty_framer   s      r    test_apply_emptyrj   Y   s4     ++Kt$$F<r+   c                 l   | d d         }|                     d           }t          t          j        | j                  }t          j        ||           | j        d d g f         }|                     d d          }t          t          j        | j                  }t          j        ||           d S )Nr   c                 *    |                                  S r%   r   rK   s    r    r*   z(test_apply_float_frame.<locals>.<lambda>d       QVVXX r+   r   c                 *    |                                  S r%   rm   rK   s    r    r*   z(test_apply_float_frame.<locals>.<lambda>i   rn   r+   r   r   )	r   r   r   nancolumnsr   r   locr   )r   no_rowsr   r   no_colss        r    test_apply_float_framerv   b   s    "1"oG]]--..FbfK$7888H68,,,oaaae$G]]--A]66FbfK$5666H68,,,,,r+   c                      t          dg          } |                     d d          }t          j        ||            d S )NrR   ro   c                     | d         S )NrR   r&   rK   s    r    r*   z/test_apply_empty_except_index.<locals>.<lambda>q   s
    af r+   r   r   r   r   r   r/   rN   s     r    test_apply_empty_except_indexrz   n   sF    u%%%H^^,,1^55F&(+++++r+   c                  V   t                      } g }|                     |j        dd          }t          j        ||            |                     |j        dd          }t          g t          j                  }t          j        ||           t          g d          }|                    |j        dd          }t          j        ||           |                    |j        dd          }t          g t          j                  }t          j        ||           |g k    sJ d S )Nr   expandr   result_typereducerA   rR   bcrr   )	r   r   appendr   r/   r   r   float64r   )ri   r(   r   r   empty_with_colss        r    test_apply_with_reduce_emptyr   u   s   ++K
AqxaXFFF&+...qxaXFFFb
+++H68,,,888O""18!"JJF&/222""18!"JJFb
+++H68,,, 777777r+   )sumprodanyr   c                    t          g d          }|                    t          t          |                     } t          ||                       }| dv r|                    t
                    }t          j        ||           d S )Nr   r   )r   r   )r   r   getattrr   r;   floatr   r   )rf   rC   r   r   s       r    test_apply_funcs_over_emptyr      s     
???	+	+	+BXXgb$''((F wr4  ""H??5))68,,,,,r+   c                  :   t          g d          } |                                 }t          d| j                  }t	          j        ||           | j                                        }t          g t          j                  }t	          j        ||           d S )Nr   r   r   ro   r   )	r   nuniquer   rr   r   r   Tr   r   rC   r   r   s      r    test_nunique_emptyr      s    	???	+	+	+BZZ\\Farz***H68,,,T\\^^Fb
+++H68,,,,,r+   c                  $   t          g dg dg dgg d          } |                     d d          }t          g d	g d          }t          j        ||           | j                            d
 d          }t          j        ||           d S )Nr   r>   r[   )r=         )      	   )rR   rR   r   ro   c                     | d         S Nr   r&   r^   s    r    r*   z/test_apply_standard_nonunique.<locals>.<lambda>   
    ! r+   r   r   )r   r=   r   c                     | d         S r   r&   r   s    r    r*   z/test_apply_standard_nonunique.<locals>.<lambda>   s
    !A$ r+   r   )r   r   r   r   r   r   r   s      r    test_apply_standard_nonuniquer      s    	IIIyyy)))4OOO	L	L	LBXXnn1X--Fiii11H68,,,TZZQZ//F68,,,,,r+   c                     |                      t          j        d          }t          |                                 g| j                  }t          j        ||           d S )N	broadcastr~   ro   )r   r   r   r   r   r   r/   r   r   r   s      r    test_apply_broadcast_scalarsr      sY    rwK@@F+**,,-[5FGGGH&(+++++r+   c                     |                      t          j        dd          }|                     d          t          fd| j        D                       }t          j        ||           d S )Nr   r   r}   r   c                     i | ]}|S r&   r&   .0r   ms     r    
<dictcomp>z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>   s    <<<1!Q<<<r+   )r   r   r   r   rr   r   r/   r   r   r   r   s      @r    "test_apply_broadcast_scalars_axis1r      ss    rwQKHHFa  A<<<<(;<<<==H&(+++++r+   c                 *                           fddd          }t          t          t           j                                      }t          |gt           j                  z  d j         j                  }t          j        ||           d S )Nc                 `    t          t          t          j                                      S r%   )listrangelenrr   r(   r   s    r    r*   z4test_apply_broadcast_lists_columns.<locals>.<lambda>   s"    $uS!4556677 r+   r   r   r}   r   )rA   r   rr   )	r   r   r   r   rr   r   r   r   r/   )r   r   r   r   s   `   r    "test_apply_broadcast_lists_columnsr      s    7777   F
 	U3{*++,,--A	
c+#$$$#	  H &(+++++r+   c                                            fdd          }t          t          t           j                                      t          fd j        D             d j                  }t          j        ||           d S )Nc                 `    t          t          t          j                                      S r%   )r   r   r   r   r   s    r    r*   z2test_apply_broadcast_lists_index.<locals>.<lambda>   s"    $uS!2334455 r+   r   r   c                     i | ]}|S r&   r&   r   s     r    r   z4test_apply_broadcast_lists_index.<locals>.<dictcomp>   s    +++!A+++r+   r   )rA   r   )	r   r   r   r   r   r   rr   r   r/   r   s   `  @r     test_apply_broadcast_lists_indexr      s    5555;   F 	U3{())**++A++++{*+++  H
 &(+++++r+   c                 f    | }|                     d dd          }t          j        ||           d S )Nc                 
    g dS Nr   r&   rK   s    r    r*   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>   
    			 r+   r   r   r}   r.   )int_frame_const_colrC   r   s      r    %test_apply_broadcast_list_lambda_funcr      s;    	BXX)){XKKF&"%%%%%r+   c                     | }|                     d dd          }|                                }t          j        ||           d S )Nc                 B    t          g dt          d                    S )Nr   abcro   )r   r   rK   s    r    r*   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>   s    &$u++666 r+   r   r   r}   r   copyr   r/   r   rC   r   r   s       r    'test_apply_broadcast_series_lambda_funcr      sS    	BXX66   F
 wwyyH&(+++++r+   c                 <    d }|                      ||d           d S )Nc                 X    t          | t          j                  sJ | j        dk    sJ d S Nr   r@   r   ndarrayndimrK   s    r    _assert_rawz/test_apply_raw_float_frame.<locals>._assert_raw   -    !RZ(((((v{{{{{{r+   Tr   r"   r   )r   r   r   s      r    test_apply_raw_float_framer      s4       k$77777r+   c                     |                      t          j        |d          }|                      d |          }t          j        ||           d S )NTr   c                 4    | j                                         S r%   )valuesr   rK   s    r    r*   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>   s    18==?? r+   r   )r   r   r   r   r   )r   r   r   r   s       r    !test_apply_raw_float_frame_lambdar      sS    rwTt<<F  !:!: FFH68,,,,,r+   c                 j    |                      d d          }| dz  }t          j        ||           d S )Nc                     | dz  S Nr>   r&   rK   s    r    r*   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>   s
    Q r+   T)r"   r>   r.   r   s      r    'test_apply_raw_float_frame_no_reductionr      s>    D99FQH&(+++++r+   c                 <    d }|                      ||d           d S )Nc                 X    t          | t          j                  sJ | j        dk    sJ d S r   r   rK   s    r    r   z4test_apply_raw_mixed_type_frame.<locals>._assert_raw  r   r+   Tr   r   )mixed_type_framer   r   s      r    test_apply_raw_mixed_type_framer     s4      
 ;Tt<<<<<r+   c                     | j         d         }|                     t          j        d          |         }t          j        |                     |                    }||k    sJ d S )Nr   r   r   )r   r   r   r   r   )r   r   r   r   s       r    test_apply_axis1r     s`    !ArwQ//2Fw{~~a(())HXr+   c                     t          dgdgd          } | d d                             t          j        d          }t	          t          j        t          j        g d          	          }t          j	        ||           d S )
Nfoo      ?r`   r   r   r   int64r   ro   )
r   r   r   r   r   rq   rb   Indexr   r   r   s      r    test_apply_mixed_dtype_cornerr     sx    	%u--	.	.BV\\"'\**F bfBHRw$?$?$?@@@H68,,,,,r+   c                  .   t          dgdgd          } |                     d d          }t          dgdg          }t          j        ||           |                     d	 d          }t          dgdg          }t          j        ||           d S )
Nr   r   r`   c                     | d         S )Nr   r&   rK   s    r    r*   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>   
    # r+   r   r   r   ro   c                     | d         S )Nr2   r&   rK   s    r    r*   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>$  r   r+   r   r   r   r   r   r   s      r    &test_apply_mixed_dtype_corner_indexingr     s    	%u--	.	.BXX&&QX//FugaS)))H68,,,XX&&QX//FseA3'''H68,,,,,r+   zignore::RuntimeWarningaxr   rr   c                     | S r%   r&   rK   s    r    r*   r*   ,      q r+   c                 *    |                                  S r%   rm   rK   s    r    r*   r*   ,  s    AFFHH r+   identityr   )idsc                    t          di | g di}t          j        d          5   |t          j        g d                    }t	          |t          j                   }|                    |||          }|r8|                    |          }t	          |t                    sJ |j	        |u sJ nt	          |t                     sJ d d d            d S # 1 swxY w Y   d S )Nr   r   r   f8r   r   r&   )
r   r   r   arrayr@   r   r   _get_agg_axisr   r   )	r   rf   r"   r   rC   test_resis_reductionr   agg_axiss	            r    test_apply_empty_infer_typer   )  s=    
	+	+b///*	+	+B		"	"	" 
1 
14400011%h
;;;$Ts33 	1''--Hff-----<8+++++fi00000
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1 
1s   B$CCCc                      t          g d          } |                     d d          }t          |t                     sJ d S )Nr   ro   c                 *    |                                  S r%   rm   rK   s    r    r*   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>B  rn   r+   r   r   )r   r   r@   )ru   r   s     r    %test_apply_empty_infer_type_broadcastr   @  sK    ooo...G]]--;]GGFfi(((((((r+   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                     | |z   S r%   r&   r(   howmuchs     r    add_somez4test_apply_with_args_kwds_add_some.<locals>.add_someG  s    7{r+   r>   r   c                     | dz   S r   r&   rK   s    r    r*   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>K  s
    1q5 r+   r   r.   )r   r  r   r   s       r    "test_apply_with_args_kwds_add_somer  F  s^        x33F  11H&(+++++r+   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                 0    |                                  |z   S r%   rm   r   s     r    agg_and_addz:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addP  s    vvxx'!!r+   r>   r  c                 0    |                                  dz   S r   rm   rK   s    r    r*   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>T  s    16688a< r+   r  )r   r   r   )r   r  r   r   s       r    %test_apply_with_args_kwds_agg_and_addr
  O  s`    " " " " {A66F  !7!788H68,,,,,r+   c                     dd}|                      |dd          }|                      d           }t          j        ||           d S )Nr   c                     | |z
  |z  S r%   r&   )r(   subdivides      r    subtract_and_dividezJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divideY  s    C6!!r+   )r>   r>   )r-   r  c                     | dz
  dz  S )N       @r&   rK   s    r    r*   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>]  s    AGs? r+   r,   r.   )r   r  r   r   s       r    -test_apply_with_args_kwds_subtract_and_divider  X  sc    " " " " 2aHHF  !:!:;;H&(+++++r+   c                 d    |                      t                    }t          j        ||            d S r%   )r   r   r   r/   )r   r   s     r    test_apply_yield_listr  a  s.    t$$F&+.....r+   c                     t           j        | j        d d d| j                            d          f<   |                     d          }|                     t           j        d          }t          j        ||           d S )Nr>   r   r   r   )	r   rq   ilocrr   get_locr   r   r   r   )r   r   r   s      r    test_apply_reduce_Seriesr  f  ss    >@fKSSqS+-55c:::;""HrwQ//F68,,,,,r+   c                  z   t          ddgddggddgddg	          } |                     t          d
          }t          ddddddg| j                  }t          j        ||           |                     t          d          }t          ddddddg| j                  }t          j        ||           d S )Nr   r>   r[   r=   r6   r7   i0i1rr   r   r   r   )r  r  ro   r5   )r   r   dictr   rr   r   r   r   )rW   r   r   s      r    test_apply_reduce_to_dictr  m  s    q!fq!f%d|D$<PPPDZZ1Z%%Faq))!1+=+=>dlSSSH68,,,ZZ1Z%%Faq))!1+=+=>djQQQH68,,,,,r+   c                  8   t          t          j                            d                              d                    } |                     t          j        d          }t          d |                                 D             | j	                  }t          j        ||           |                     t          j        d          }t          d | j                                        D             | j                  j        }t          j        ||           d S )	Nr>   )   
   r   r   c                 >    i | ]\  }}||                                 S r&   describer   ivs      r    r   z2test_apply_differently_indexed.<locals>.<dictcomp>~  s&    AAAda!QZZ\\AAAr+   r   r   c                 >    i | ]\  }}||                                 S r&   r#  r%  s      r    r   z2test_apply_differently_indexed.<locals>.<dictcomp>  s&    CCCda!QZZ\\CCCr+   )r   r   randomdefault_rngstandard_normalr   r   r$  itemsrr   r   r/   r   r   r   s      r    test_apply_differently_indexedr-  z  s    	29((++;;HEE	F	FBXXfoAX..FAAbhhjjAAA2:VVVH&(+++XXfoAX..FCCbdjjllCCCRXVVVXH&(+++++r+   c                     t          g dg dg dg dg dg dgg d          } 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
          }t          j        ||           d S )N)r   ABC02   )r   YUM0r   )r   DEF0r   )r>   ABC1r0  )r>   YUM1r   )r>   DEF1r   )rR   marketpositionr   c                     | d         S )Nr6  r&   )rs    r    fztest_apply_bug.<locals>.f  s    {r+   r   r   i  r/  r0  r>   r1  r   r[   r2  r=   r3  r   r4  r   r5  )r   r   r   r   r   )	positionsr:  r   r   s       r    test_apply_bugr<    sN   OOOOOOOOOOOO	
 ,++
 
 
I   qq))HdAq!!62.dAq!!62.dAq!!62.dAq!!62.dAq!!62.dAq!!62.	
 ,++
 
 
I __QQ_''F68,,,,,r+   c            
         t          g dg dg dt          j                            d                              d          t          j                            d                              d          t          j                            d                              d          d          } |                     d d	          }t          j        ||            d S )
N)r   r   r   r   barr>  r>  r>  r   r   r   )oner?  r?  twor?  r?  r?  r@  r@  r@  r?  )dullrA  shinyrA  rA  rB  rB  rA  rB  rB  rB  r>      )r   r2   r3   r4   EFc                     | S r%   r&   rK   s    r    r*   z,test_apply_convert_objects.<locals>.<lambda>  rL   r+   r   r   )r   r   r)  r*  r+  r   r   r/   rN   s     r    test_apply_convert_objectsrG    s           &&q))99"==&&q))99"==&&q))99"==U+	
 +	
- -H^ ^^KKa^00F&(+++++r+   c                     |                      d           }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S r%   namerK   s    r    r*   z(test_apply_attach_name.<locals>.<lambda>       r+   ro   )r   r   rr   r   r   r   s      r    test_apply_attach_namerM    sJ    //00Fk)1DEEEH68,,,,,r+   c                     |                      d d          }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S r%   rJ  rK   s    r    r*   z.test_apply_attach_name_axis1.<locals>.<lambda>  rL  r+   r   r   ro   )r   r   r   r   r   r   s      r    test_apply_attach_name_axis1rP    sO    //a88Fk'{/@AAAH68,,,,,r+   c                     |                      d           }t          t          j        | j        t          | j                  df          | j        | j                  }t          j        ||           d S )Nc                 P    t          j        | j        t          |                     S r%   r   repeatrK  r   rK   s    r    r*   z6test_apply_attach_name_non_reduction.<locals>.<lambda>      163q66)B)B r+   r   r   rr   )	r   r   r   tilerr   r   r   r   r/   r   s      r    $test_apply_attach_name_non_reductionrX    sv    BBCCF
#c+*;&<&<a%@AA#  H
 &(+++++r+   c                                            d d          }t           fd                                 D                       } j        |_        t	          j        ||           d S )Nc                 P    t          j        | j        t          |                     S r%   rS  rK   s    r    r*   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>  rU  r+   r   r   c              3   r   K   | ]1}t          j        |d          t          j                            V  2dS )r   N)r   rT  r   rr   )r   tr   s     r    	<genexpr>z=test_apply_attach_name_non_reduction_axis1.<locals>.<genexpr>  sO        67	!A$K/0011     r+   )r   r   
itertuplesr   r   r   r   s   `  r    *test_apply_attach_name_non_reduction_axis1r_    s    BBKKF    ;F;Q;Q;S;S    H !&HN68,,,,,r+   c                  
   t          j        g dg dg          } t          ddgddgddgg| d	d
g          }|                    d d          }t          ddgddgddgg| ddg          }t	          j        ||d           d S )N)rR   rR   r   )r   r   r   r   r>   r[   r=   r   r   col1col2rV  c                 Z    t          t          |           t          |           d          S )N)minmax)r   rd  re  rK   s    r    r*   z(test_apply_multi_index.<locals>.<lambda>  s"    vc!ffSVV&D&DEE r+   rd  re  T
check_like)r   from_arraysr   r   r   r/   )r   r^   r   r   s       r    test_apply_multi_indexri    s    "OOO___#EFFEAq6Aq6Aq6*%&&AQRRRAWWEEqIIF1a&1a&1a&1PUWWWH&(t<<<<<<r+   z	df, dictsr   r>  spameggs)r   r   r>   r[   c                    d }|                      |d          }|                      |d          }|                      |          }t          j        ||           t          j        ||            t          j        ||           d S )Nc                 *    |                                  S r%   )to_dictrK   s    r    r*   z!test_apply_dict.<locals>.<lambda>  s    199;; r+   r   r   r|   )r   r   r   r/   )rC   dictsfnreduce_truereduce_falsereduce_nones         r    test_apply_dictrt    s     
	B((28(44K88BH855L((2,,K;...,+++;.....r+   c                  6   t          dt          ddd          i          } |                     d           }t          j        ||            |                     d           }t          dt          ddd          i          }t          j        ||           d S )	Ndtz
2015-01-01r[   zEurope/Brussels)r]   rI   c                     | S r%   r&   rK   s    r    r*   z,test_apply_non_numpy_dtype.<locals>.<lambda>       r+   c                 0    | t          j        d          z   S )N1day)rb   	TimedeltarK   s    r    r*   z,test_apply_non_numpy_dtype.<locals>.<lambda>   s    BL$8$8 8 r+   z
2015-01-02)r   r	   r   r   r/   r   s      r    test_apply_non_numpy_dtyper|    s    	D*\1ARSSST	U	UBXXkk""F&"%%%XX8899F	z,6GHHHI H &(+++++r+   c                      t          dg did          } |                     d           }t          j        ||            d S )Nrv  )rR   r   r   rR   r9   r   c                     | S r%   r&   rK   s    r    r*   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>)  rx  r+   ry   rB   s     r    #test_apply_non_numpy_dtype_categoryr  '  sL    	D.../z	B	B	BBXXkk""F&"%%%%%r+   c                      t          ddgddggddg          } t          ddggddgdg          }|                     dg          }t          j        ||           d S )	Nr   r   r>   r[   rR   r   rd  r  rU   rC   r   r   s      r    test_apply_dup_names_multi_aggr  -  so    	QFQF#c3Z	8	8	8B1a&C:eWEEEHVVUG__F&(+++++r+   opr   rV   c                    d }t          t          j        d          t          d                    } t	          ||           |d          }t          g dg dg dg dg          }t          j        ||           d S )Nc                 B    d| d         z  d| d         z  d| d         z  gS )Nr>   r   r3   r2   r&   rows    r    
apply_listz3test_apply_nested_result_axis_1.<locals>.apply_list9  s'    CHa#c(lACL99r+   )r=   r=   ABCDr   r   r   )        r  r  )r   r   zerosr   r   r   r   r   )r  r  rC   r   r   s        r    test_apply_nested_result_axis_1r  6  s    : : : 
28F##T&\\	:	:	:BWR__Za000F	///???OOOL H 68,,,,,r+   c                      t          dt          dd          gid          } |                     d           }t          j        ||            |                     d           }t          j        ||            d S )	Nr   2020rG   rH   zdatetime64[ns, UTC]r   c                     | S r%   r&   rK   s    r    r*   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>I  rL   r+   c                 *    |                                  S r%   )r   rK   s    r    r*   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>K  s    affhh r+   rM   rN   s     r    %test_apply_noreduction_tzaware_objectr  D  s    	6e,,,-.6K  H ^^KK((F&(+++^^..//F&(+++++r+   c                      t          dg di          } g fd}fd}||fD ]8}d d = |                     |d           t          | j                  k    sJ 9d S )NrR   r   c                 <                         | j                   d S r%   r   rK  r  namess    r    reducing_functionz8test_apply_function_runs_once.<locals>.reducing_functionU  s    SXr+   c                 <                         | j                   | S r%   r  r  s    r    non_reducing_functionz<test_apply_function_runs_once.<locals>.non_reducing_functionX  s    SX
r+   r   r   )r   r   r   r   )rC   r  r  rf   r  s       @r    test_apply_function_runs_oncer  O  s     
C#	$	$BE         #$9: ' '!!!H
ARX&&&&&	' 'r+   c                      t          dg di          } g fd}fd}||fD ]K}d d = |                     |dd           t          | j                                                  k    sJ Ld S )NrR   r   c                 2                         |            d S r%   extendr  r   s    r    r  z<test_apply_raw_function_runs_once.<locals>.reducing_functioni  s    cr+   c                 2                         |            | S r%   r  r  s    r    r  z@test_apply_raw_function_runs_once.<locals>.non_reducing_functionl  s    c
r+   Tr   )r"   r   )r   r   r   rR   to_list)rC   r  r  rf   r   s       @r    !test_apply_raw_function_runs_oncer  c  s     
C#	$	$BF         #$9: . .111I
4a(((bdllnn-------	. .r+   c                      t          t          j        ddg          dg          } t          t          j        ddg          dgt                    }|                     d           }t          j        ||           d S )Ns   abcds   efghcolr   )rr   rA   c                 ,    |                      d          S )Nobjectr:   rK   s    r    r*   z-test_apply_with_byte_string.<locals>.<lambda>}  s     2 2 r+   )r   r   r   r  r   r   r/   r  s      r    test_apply_with_byte_stringr  w  sz    	28Wg.//%	A	A	AB7G"455wfUUUH XX2233F&(+++++r+   valasd   c                      dd dddt           j        g}t          d|id          }|j                             fd          }t           fd|D             d	          }t          j        ||           d S )
Nr  r  cderR   r9   r   c                     | k    S r%   r&   )r(   r  s    r    r*   z/test_apply_category_equalness.<locals>.<lambda>  s    !s( r+   c                 \    g | ](}t          j        |          rt          j        n|k    )S r&   )rb   isnullr   rq   )r   r(   r  s     r    
<listcomp>z1test_apply_category_equalness.<locals>.<listcomp>  s0    AAA!29Q<<	-Q#XAAAr+   rJ  )r   rq   r   rR   r   r   r   r   )r  	df_valuesrC   r   r   s   `    r    test_apply_category_equalnessr    s     b%7I	C#:	6	6	6BTZZ****++FAAAAyAAA  H 68,,,,,r+   c                  P   t          t          j                            d                              d                    } |                     t          j        j        d          j        }|dk    sJ |                     t          j        j        d          j        }|dk    sJ d S )Nr>   )r!  r>   r   r   )r   r>   )r   r   r)  r*  r   fftr?   rfftrB   s     r    test_infer_row_shaper    s     
29((++227;;	<	<BXXbfjqX))/FWXXbfkX**0FVr+   zops, by_row, expectedrR   c                     | dz   S r   r&   rK   s    r    r*   r*     
    Q r+   compatc                     | dz   S r   r&   rK   s    r    r*   r*     r  r+   c                 *    |                                  S r%   r   rK   s    r    r*   r*          r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*     r  r+   r   c                 *    |                                  S r%   r  rK   s    r    r*   r*         AEEGG r+   )r[   r[   r[   )r   r   <lambda>ro   c                 *    |                                  S r%   r  rK   s    r    r*   r*     r  r+   c                     dS r   r&   rK   s    r    r*   r*          r+   c                     dS r   r&   rK   s    r    r*   r*     r  r+   c                     t          dddgi          }|                    | |          }t          j        ||           d S NrR   r   r>   by_rowr   r   r   assert_equalopsr  r   rC   r   s        r    test_dictlike_lambdar    sF    . 
C!Q=	!	!BXXc&X))FOFH%%%%%r+   r  c                     | dz   S r   r&   rK   s    r    r*   r*     s
    A r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*     s     r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*     s     r+   c                     dS r   r&   rK   s    r    r*   r*     s     r+   c                     t          dddgi          }t          j        t          d          5  |                    | d           d d d            d S # 1 swxY w Y   d S NrR   r   r>   zby_row=True not allowedmatchTr  r   pytestraises
ValueErrorr   r  rC   s     r    test_dictlike_lambda_raisesr         
C!Q=	!	!B	z)B	C	C	C # #
T"""# # # # # # # # # # # # # # # # # #   AAAc                  "   t          ddgddggddg          } |                     d d          }t          d |                                 D                       }t	          j        ||           t          d	          t          d
          g| d<   |                     d d          }t	          j        ||           | d         | d         z                       d           }t          ddiddig          }t	          j        ||           d S )Nr   r>   rR   r   r   c                 (    d| d         | d         z   iS Nr^   rR   r   r&   rK   s    r    r*   z,test_with_dictlike_columns.<locals>.<lambda>      afqvo 6 r+   r   c                     g | ]}d diS )r^   r[   r&   r   r\  s     r    r  z.test_with_dictlike_columns.<locals>.<listcomp>  s    999AQx999r+   2017-05-01 00:00:002017-05-02 00:00:00r   c                 (    d| d         | d         z   iS r  r&   rK   s    r    r*   z,test_with_dictlike_columns.<locals>.<lambda>  r  r+   c                 
    d| iS )Nr^   r&   rK   s    r    r*   z,test_with_dictlike_columns.<locals>.<lambda>  s
    #q r+   r^   r[   )r   r   r   r^  r   r   r   r   s      r    test_with_dictlike_columnsr    s!   	QFQF#c3Z	8	8	8BXX66QX??F99999::H68,,, 	'(('((BtH XX66QX??F68,,, g3&&'9'9::FQx#q*++H68,,,,,r+   c                      t                      } g d| d<   g d| d<   t          j        g dd          | d<   |                     d	 d
          }t	          i i i g          }t          j        ||           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                     i S r%   r&   rK   s    r    r*   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>       r+   r   r   )r   rb   to_datetimer   r   r   r   r   s      r    (test_with_dictlike_columns_with_datetimer    s    	B"??BxL+++B{OMMM  BvJ XXllX++Fr2rl##H68,,,,,r+   c                  X   t          ddgddggddg          } |                     d dd          }t          d	d
d
gi          }t          j        ||           t	          d          t	          d          g| d<   |                     d dd          }t          j        ||           d S )Nr   r>   rR   r   r   c                 (    d| d         | d         z   iS r  r&   rK   s    r    r*   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>  r  r+   r|   r}   r^   r[   r  r  r   c                 (    d| d         | d         z   iS r  r&   rK   s    r    r*   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>  r  r+   )r   r   r   r/   r   r   s      r    %test_with_dictlike_columns_with_inferr    s    	QFQF#c3Z	8	8	8BXX66QHXUUF#1v''H&(+++ 	'(('((BtH XX66QHXUUF&(+++++r+   c                     | dz   S r   r&   rK   s    r    r*   r*     
    AE r+   )rR   r  c                     | dz   S r   r&   rK   s    r    r*   r*     r  r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*          AEEGG r+   r  c                 *    |                                  S r%   r  rK   s    r    r*   r*     r  r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*         aeegg r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*     r  r+   c                     | dz   S r   r&   rK   s    r    r*   r*     s
    q1u r+   c                     dS Nr[   r&   rK   s    r    r*   r*     s     r+   r   c                     dS r   r&   rK   s    r    r*   r*     r   r+   c                     dS r  r&   rK   s    r    r*   r*     rT   r+   c                     t          dddgi          }|                    | |          }t          j        ||           d S r  r  r  s        r    test_listlike_lambdar    sF    > 
C!Q=	!	!BXXc&X))FOFH%%%%%r+   c                     | dz   S r   r&   rK   s    r    r*   r*   "  
    1q5 r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*   #  s    15577 r+   c                 *    |                                  S r%   r  rK   s    r    r*   r*   $  s    !%%'' r+   c                     | dz   S r   r&   rK   s    r    r*   r*   %  r  r+   c                     dS r  r&   rK   s    r    r*   r*   %  rT   r+   c                     t          dddgi          }t          j        t          d          5  |                    | d           d d d            d S # 1 swxY w Y   d S r  r  r  s     r    test_listlike_lambda_raisesr    r  r  c            	      z   t          t          t          j                            d                              d                    g dt          ddd          d          } | dd	g                             t          d
          }t          d | dd	g         	                                D                       }t          j        ||           | ddg                             t          d
          }t          d | ddg         	                                D                       }t          j        ||           d S )Nr>   r=   )rR   r   ofwordsz
2016-10-01H)r]   freq)rR   r   r<   rR   r   r   r   c                 "    g | ]}|d d         S rP   r&   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>:  s     BBBquBBBr+   r<   c                 "    g | ]}|d d         S rP   r&   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>>  s     CCCquCCCr+   )r   r   r   r)  r*  r+  r	   r   tupler^  r   r   r   s      r    test_with_listlike_columnsr  /  s-   		--a00@@CCDD---\13???	
 	

 
B c
^!!%a!00FBBb#sn&?&?&A&ABBBCCH68,,,d_""5q"11FCCb#to&@&@&B&BCCCDDH68,,,,,r+   c                  4   t          t          ddgdgg          t          dgddgg          d          } t          j        ddg          | _        |                     d	 d
          }t          g dgg| j                  }t          j        ||           d S )NrR   r   qzr\  r'   )r  j0)r  j1c                 ,      fd d         D             S )Nc                 (    g | ]}|d          v |S )r)   r&   )r   elr  s     r    r  zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>G  s"    "K"K"K"B#c(NN2NNNr+   r(   r&   r  s   `r    r*   z;test_with_listlike_columns_returning_list.<locals>.<lambda>G  s     "K"K"K"KC"K"K"K r+   r   r   ro   )r   r   r   from_tuplesr   r   r   r   r   s      r    )test_with_listlike_columns_returning_listr'  B  s    	#scU 3446C53PS*BU;V;VWW	X	XB%|\&BCCBHXXKKRSXTTFrC5k222H68,,,,,r+   c                     t          ddgddgt          d          t          d          gd          } |                     d d	
          }t          d |                                 D                       }t          j        ||           d S )Nr   r  r   r>  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                     | j         | j        fS r%   r)  r*  r  s    r    r*   z1test_infer_output_shape_columns.<locals>.<lambda>Y  s    3:sz": r+   r   r   c                 *    g | ]}|j         |j        fS r&   r,  r  s     r    r  z3test_infer_output_shape_columns.<locals>.<listcomp>Z  s!    EEE!(+EEEr+   r   r   r   r   r^  r   r   r   s      r    test_infer_output_shape_columnsr/  L  s     
Cjen/00/00	
 	
	
 	
B XX::XCCFEER]]__EEEFFH68,,,,,r+   c                     t          t          j                            d                              d          g d          } |                     d d          }t          d |                                 D                       }t          j	        ||           |                     d	 d          }t          d
 |                                 D                       }t          j	        ||           d S )Nr>   )r   r[   r   r2   r3   r   c                 
    g dS r   r&   rK   s    r    r*   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>e  r   r+   r   r   c                     g | ]}g d S )r   r&   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>f  s    :::Qyyy:::r+   c                 
    ddgS Nr   r>   r&   rK   s    r    r*   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>i  
    A r+   c                     g | ]}d dgS r   r>   r&   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>j  s    777!1v777r+   
r   r   r)  r*  r+  r   r   r^  r   r   r   s      r    (test_infer_output_shape_listlike_columnsr:  ^  s     

	a  0088///
 
 
B XX))X22F::"--//:::;;H68,,,XX&&QX//F77r}}77788H68,,,,,r+   c                     t          dg dit          d                    }|                     fdd          }t           fd|                                D             |j                  }t          j        ||           d S )	NrR   r   r   ro   c                 ,    t          j                  S r%   r   ones)r  r  s    r    r*   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>s  s    "'#,, r+   r   r   c                 8    g | ]}t          j                  S r&   r=  )r   r\  r  s     r    r  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>t  s!    ===rws||===r+   )r   r   r   r   r^  r   r   r   )r  rC   r   r   s   `   r    0test_infer_output_shape_listlike_columns_np_funcr@  n  s     
C#4;;	7	7	7BXX....QX77F====R]]__===RXNNNH68,,,,,r+   c                  ^   t          t          d          t          d          t          d          t          d          gg dg dg dd          } d	 }|                     |d
          }t          d |                                 D                       }t          j        ||           d S )Nz
2010-02-01z
2010-02-04z
2010-02-05z
2010-02-06)r   r   r=   r[   )r   r[   r=   r>   )r   r>   r[   r=   )rR   r   r   r   c                     dS )Nr8  r&   rK   s    r    funzDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    vr+   r   r   c                     g | ]}d S )r8  r&   r  s     r    r  zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>  s    777!v777r+   r.  )rC   rC  r   r   s       r    7test_infer_output_shape_listlike_columns_with_timestamprE  x  s    	 ,'','','',''	 
	
 
	

 
B   XXcX""F77r}}77788H68,,,,,r+   lstr   c                 F    t          t          j                            d                              d          g d          }|                     fdd          }t           fd|                                D                       }t          j	        ||           d S )	Nr>   )r=   r[   r1  r   c                     S r%   r&   )r(   rF  s    r    r*   z3test_consistent_coerce_for_shapes.<locals>.<lambda>  s     r+   r   r   c                     g | ]}S r&   r&   )r   r\  rF  s     r    r  z5test_consistent_coerce_for_shapes.<locals>.<listcomp>  s    444qs444r+   r9  )rF  rC   r   r   s   `   r    !test_consistent_coerce_for_shapesrJ    s     

	a  0088///
 
 
B XXmmmm!X,,F4444BMMOO44455H68,,,,,r+   c                    | }|                     d d          }|                     dddd          }t          j        ||           |                     d	 d          }|ddg         }t          j        ||           d S )
Nc                 ,    t          g dg d          S )Nr   )testothercolsro   r   rK   s    r    r*   z'test_consistent_names.<locals>.<lambda>  s    &*C*C*CDDD r+   r   r   rM  rN  rO  r1  r   c                 ,    t          ddgddg          S )Nr   r>   rM  rN  ro   rP  rK   s    r    r*   z'test_consistent_names.<locals>.<lambda>  s    1vfg5F G G G r+   )r   renamer   r/   r   s       r    test_consistent_namesrS    s    	BXXDD1   F #))788 *  H &(+++XXGGaXPPF)*H&(+++++r+   c                     | }|                     d dd          }|                                }g d|_        t          j        ||           d S )Nc                 
    g dS r   r&   rK   s    r    r*   z"test_result_type.<locals>.<lambda>  r   r+   r   r|   r}   r   r   r>   r   r   rr   r   r/   r   s       r    test_result_typerX    sV     
BXX))xXHHFwwyyH yyH&(+++++r+   c                     | }|                     d dd          }|ddg                                         }ddg|_        t          j        ||           d S )Nc                 
    ddgS r5  r&   rK   s    r    r*   z/test_result_type_shorter_list.<locals>.<lambda>  r6  r+   r   r|   r}   r   r2   r   rW  r   s       r    test_result_type_shorter_listr[    sb     
BXX&&QHXEEF3*~""$$H1vH&(+++++r+   c                     | }|                     d dd          }|                                }t          j        ||           d S )Nc                 
    g dS r   r&   rK   s    r    r*   z,test_result_type_broadcast.<locals>.<lambda>  r   r+   r   r   r}   r   r   s       r    test_result_type_broadcastr^    sJ     
BXX)){XKKFwwyyH&(+++++r+   c                     | }g d|                     fddd          }|                                }t          j        ||           d S )NrN  r  r  c                 *    t          g d          S Nr   ro   rP  r(   rr   s    r    r*   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &'222 r+   r   r   r}   r   r   rC   r   r   rr   s       @r    &test_result_type_broadcast_series_funcre    se     
B'''GXX2222   F wwyyH&(+++++r+   c                     | }|                     d d          }|                                }t          j        ||           d S )Nc                 2    t          g d| j                  S rb  )r   r   rK   s    r    r*   z0test_result_type_series_result.<locals>.<lambda>  s    yyy @ @ @ r+   r   r   r   r   s       r    test_result_type_series_resultrh    sH     
BXX@@qXIIFwwyyH&(+++++r+   c                     | }g d|                     fdd          }|                                }|_        t          j        ||           d S )Nr`  c                 *    t          g d          S rb  rP  rc  s    r    r*   z<test_result_type_series_result_other_index.<locals>.<lambda>  s    yyy @ @ @ r+   r   r   rW  rd  s       @r    *test_result_type_series_result_other_indexrk    sb     
B'''GXX@@@@qXIIFwwyyHH&(+++++r+   boxc                      t          |           S r%   )r   rK   s    r    r*   r*     s    tAww r+   c                      t          |           S r%   )r  rK   s    r    r*   r*     s    %(( r+   c                 .    t          j        | d          S )Nr   r   )r   r   rK   s    r    r*   r*     s    bhq6P6P6P r+   )r   r  r   c                 f    |}|                      fdd          }t           fd|                                D                       }t          j        ||           |                      fddd          }|dd	g                             d
dd          }t          j        ||           d S )Nc                      ddg          S r5  r&   r(   rl  s    r    r*   z,test_consistency_for_boxed.<locals>.<lambda>      QF r+   r   r   c                 *    g | ]} d dg          S r8  r&   )r   r\  rl  s     r    r  z.test_consistency_for_boxed.<locals>.<listcomp>  s%    <<<qssAq6{{<<<r+   c                      ddg          S r5  r&   rr  s    r    r*   z,test_consistency_for_boxed.<locals>.<lambda>  rs  r+   r|   r}   r   r2   r   r`   r   )r   r   r^  r   r   rR  r/   )rl  r   rC   r   r   s   `    r    test_consistency_for_boxedrv    s     
BXX++++!X44F<<<<BMMOO<<<==H68,,,XX++++!XJJF"C:.55AA>N>N5OOH&(+++++r+   c                    | dv rdnd}t          j        d          5  t          j        |          }t          j        |          }|                                }|                    t           j        |           }t          j        ||           |                    t           j        g|           }|                                }| dv r"t          j	        |j
        dgg          |_
        n!t          j	        |j        dgg          |_        t          j        ||           |                    t           j        t           j        g|           }t          ||g|          }| dv r#t          j	        |j
        ddgg          |_
        n"t          j	        |j        ddgg          |_        t          j        ||           d d d            d S # 1 swxY w Y   d S )	N   r   r   r   r   r   r   r   r   absolute)r   r   absr   r   r   r   r/   r   from_productrr   r   r
   )r   r   
other_axisf_absf_sqrtr   r   s          r    test_agg_transformr    s   l**J		"	"	" 0 0{##%% ;;==""27"66
fh/// ""BG94"88;;==<)68KfX7VWWH'4k6G&5RSSHN
fh///
 ""BFBG#44"@@ufoJ???<)6$z6&:;   H (4"Z$89 HN 	fh///?0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s   F#GGGc                      t          t          d          dd          } |                     ddg          }t          ddgddgdddgddg	          }t          j        ||           d S )
Nr   r`   rd  re  r   r=   r   r2   r  )r   r   rV   r   r/   r   s      r    	test_demor  &  s    	q**	+	+BVVUEN##F!fAq6""S#Juen  H &(+++++r+   c                  2   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          }t          j        |                    |          |           d S )Nr   r`   rd  re  r         @r        @g      9@r   r2   )re  rd  r   r  )r   r   rV   r   rq   r   r/   reindex_liker   s      r    test_demo_dict_aggr  1  s    	q**	+	+BVV5%.u~>>??FC RVT':;;c
###  H
 &--h77BBBBBr+   c                  4   dddgi} t          |           }|                    ddi          }t          ddi          }t          j        ||           |d                             ddi          }t          ddid          }t          j        ||           d S )NrK  r   r>  countr>   rJ  )r   rV   r   r   r   )rW   rC   r   r   s       r    !test_agg_with_name_as_column_namer  =  s    UEN#D	4B VVVW%&&Fvqk""H68,,, Z^^VW-..Fvqk///H68,,,,,r+   c                  p   t          g dg dg dd          } t          ddgddgd	d
gdddg          }|                     ddg          }t          j        ||           | g d                             ddg          }|g d                             ddg          }t          j        ||           d S )Nr   r   r        @r   r>  bazr1  r   r   r         @r>  	foobarbazrd  r   ro   )r3   r2   r   )r   rV   r   r/   reindex)mdfr   r   s      r    test_agg_multiple_mixedr  M  s    
 &&&	
 	
 C Qs%	
 	

 en  H WWeU^$$F&(+++!%%uen55F (00%@@H&(+++++r+   c            	         t          g dg dg dt          dd          d          } d}t          j        t          |	          5  |                     d
dg           d d d            n# 1 swxY w Y   t          j        t          |	          5  | g d                             dd
g           d d d            d S # 1 swxY w Y   d S )Nr   r  r  rZ   r[   r\   r   r2   r3   r4   does not support reductionr  rd  r   )r4   r3   r2   r   )r   r	   r  r  	TypeErrorrV   )r  msgs     r    test_agg_multiple_mixed_raisesr  i  sh   
 &&&J222		
 	
 C 'C	y	,	,	,                                  
y	,	,	, 6 6   !%%uen5556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s$   A++A/2A/ B>>CCc                    | dv rdnd}|j         |                                         d d                                         \  }}t          j        |                    |           |                    |           |                    |           gd          }g d|_        | dv r|j	        n|}|
                    g d|           }t          j        ||           |d|di}|
                    ||           }t          |                    |          |                                         |                    |          |                                         g||g	          }t          j        ||           |dg|dgi}|
                    ||           }t!          |t          |                    |          |                                         gdg	          |t          |                    |          |                                         gdg	          i          }| d
v r|j	        n|}t          j        ||           |ddg|ddgi}|
                    ||           }t          j        |t          |                    |          |                                         |                    |          |                                         gddg	          |t          |                    |          |                                         |                    |          |                                         gddg	          id          }| d
v r|j	        n|}t          j        ||           d S )Nrx  r   r   r>   r   )r   re  r   r   r   ro   >   rr   r   re  )axesuniquesort_valuesrb   concatr   re  r   rr   r   rV   r   r/   r   rs   r   r   )r   r   r|  name1name2r   r   rf   s           r    test_agg_reducer  }  s   l**J#J/6688!<HHJJLE5 y$''OOO&&OOO&&	

   H .--H!\11xzzxH__333$_??F&(+++ 65%(D__T_--FOOJ''.3355OOJ''.2244	
 en  H 68,,, F8UUG,D__T_--F6;??:66u=BBDDEfXVVV6;??:66u=AACCDUGTTT	
 H "^33xzzH&(+++ FE?EE5>:D__T_--Fy6OOJ//6;;==OOJ//6::<< uo   6OOJ//6::<<OOJ//6::<< en  	
  #  H& "^33xzzH&(+++++r+   c            	      ^   t          g dg dg dt          dd          d          } |                     d          }t          d	d
dt	          d          g| j                  }t          j        ||           |                     dg          }t          d	d
dt	          d          ggdg| j                  }t          j        ||           d}t          j
        t          |          5  |                     d           d d d            n# 1 swxY w Y   | g d                             d          }t          g dg d          }t          j        ||           d}t          j
        t          |          5  |                     dg           d d d            d S # 1 swxY w Y   d S )Nr   r  r  rZ   r[   r\   r  rd  r   r   r>  ro   rV  r  r  r   r1  )r   r  r  )r   r	   rV   r   r   rr   r   r   r/   r  r  r  )rC   r   r   r  s       r    test_nuiscance_columnsr    sA   	 &&&J222		
 	

 
B VVE]]Fq#ui
&;&;<BJOOOH68,,,VVUG__F
S%:..	/0g
  H
 &(+++
&C	y	,	,	,  
u                $$U++F+++???CCCH68,,,
&C	y	,	,	,  
w                 s$   -DDD>F""F&)F&howc                    t          g ddt          j        dgg dd          } t          ||           ddi          }t	          ddi          }t          j        ||            t          ||           dd	i          }t	          dd
i          }t          j        ||            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           t          j        ||d            t          ||           d          }|                                }t          j        ||           d S )NNr>   r[   r   r  r   Nr>  r1  r   r  r>   sizer[   )r  r  Trf  )	r   r   rq   r   r   r   r   r/   r  )r  rC   r   r   result1result2s         r    test_non_callable_aggregatesr    s    
ll#rvs!3:N:N:NOO
 
B
 WRsGn--FsAhH68,,, WRsFm,,FsAhH68,,, gb#011Ggb#w&7w>OPP G a((a((a((	
 	
 H '7t<<<<'8==== WRg&&FxxzzH68,,,,,r+   c                 <   t          g ddt          j        dgg dd          } t          ||           d|          }|dv r"t	          |j        d	         |j        
          }n!t	          |j        d         |j        
          }t          j	        ||           d S )Nr  r   r  r  r1  r  r   )r   r   r   ro   r   )
r   r   rq   r   r   r?   rr   r   r   r   )r  r   rC   r   r   s        r    test_size_as_strr    s     
ll#rvs!3:N:N:NOO
 
B
 WRf4000F|"(1+RZ888"(1+RX66668,,,,,r+   c                  h   t          g ddt          j        dgg dd          } d }|                     |          }t	          ddgdgdd	ggg d
          }t          j        ||           |                     |g          }|                    d          j        }t          j	        ||           d S )N)r>   r>   r[   g      ?r  r1  c                 h    t          |                                                                           S r%   )r   dropnar  )	group_cols    r    rf   z&test_agg_listlike_result.<locals>.func'  s(    I$$&&--//000r+   r>   r[   r   r>  ro   rf   )
r   r   rq   rV   r   r   r   to_framer   r/   )rC   rf   r   r   s       r    test_agg_listlike_resultr  #  s    	#rvs);BVBVBVWW	X	XB1 1 1 VVD\\F1vuuen5___MMMH68,,,VVTF^^F  ((*H&(+++++r+   zargs, kwargs)r   r      r8  r,   r   r&   rR   r   r   c                     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          j        ||           d S )Nr[   c                 <    |                                  ||z   |z  z   S r%   r  )r(   rR   r   r   s       r    r:  ztest_agg_args_kwargs.<locals>.fA  s    uuww!a%1$$r+   r   r>   r=   r   r  g      @r  g       @)r[   r   r   rV   r   r   )r   r-   kwargsr:  rC   r   r   s          r    test_agg_args_kwargsr  3  s    % % % % 
QFQF#	$	$Bqyy3*%%3*%%RVAt-d---f--F68,,,,,r+   num_cols)r>   r[   r   c                     t          j        g d          }|                                }t          d|t	          |                     }|                    d            |j        |j        k    sJ d S )N)z
1950-06-30z
1952-10-24z
1953-05-29r   rV  c                     | S r%   r&   rK   s    r    r*   z,test_frequency_is_original.<locals>.<lambda>V  r   r+   )rb   DatetimeIndexr   r   r   r   r  )r  r   originalrC   s       r    test_frequency_is_originalr  P  sq     GGGHHEzz||H	1E5??	;	;	;BHH[[:&&&&&&r+   c                     t          dd          t          dd          t          dd          g} t          g d|           }|                    d d	
          }t          | |           }t	          j        ||           d S )Nz2019-03-15 12:34:31.909000+0000rG   rH   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000rV  )rW   r   c                     | j         S r%   rJ  rK   s    r    r*   z.test_apply_datetime_tz_issue.<locals>.<lambda>c  s     r+   r   r   )r   rW   )r   r   r   r   r   r   )
timestampsrC   r   r   s       r    test_apply_datetime_tz_issuer  Z  s     	3>>>3>>>3>>>J
 
				4	4	4BXX&&QX//FJZ888H68,,,,,r+   rC   r   r   r`   method)rd  re  r   c                     |dk    rd}nd}t          j        t          |          5   t          | |                       d d d            d S # 1 swxY w Y   d S )Nr   z-can only concatenate str \(not "int"\) to strz4not supported between instances of 'str' and 'float'r  )r  r  r  r   )rC   r  r  s      r    test_mixed_column_raisesr  i  s     >D	y	,	,	,  F                 s   AAAr  r   c                     t          d| ggddg          }|                    d           }|j        }t          j        ||           d S )Nr   rR   r   r   c                     | j         S r%   r   rK   s    r    r*   z"test_apply_dtype.<locals>.<lambda>y  s     r+   )r   r   dtypesr   r   )r  rC   r   r   s       r    test_apply_dtyper  u  sU     
S#J<#s	4	4	4BXX''((FyH68,,,,,r+   c                    t          t          d          t          dd          d          }|                                }d }|                                }|dxx         dz  cc<   |                    |d          }t	          j        ||           |s| rt	          j        ||           d S t	          j        ||           d S )Nd      r  c                 T    | j         }| j        dxx         dz  cc<   | j         |usJ | S )NrR   r   )_mgrrs   )r  mgrs     r    rf   z!test_apply_mutating.<locals>.func  s9    hxs""""
r+   rR   r   r   )r   r   r   r   r   r/   )using_array_managerusing_copy_on_writerC   df_origrf   r   r   s          r    test_apply_mutatingr    s    	s%S//::	;	;BggiiG   wwyyHSMMMQMMMXXdX##F&(+++ *1 * 	b'*****
b&)))))r+   c                      t          ddgddgddgddgd	d
ggddg          } |                     d d          }t          g g dt                    }t	          j        ||           d S )Nr   r>   r[   r=   r   r   r   r   r   r!  rR   r   r   c                     g S r%   r&   rK   s    r    r*   z.test_apply_empty_list_reduce.<locals>.<lambda>  r  r+   r   r   r  r   )r   r   r   r  r   r   r   s      r    test_apply_empty_list_reducer    s    	QFQFQFQFQG<sCj	Q	Q	QBXXllX99FBR((777H68,,,,,r+   c                      t          ddggdz  ddg          } |                     dd d	 g          }t          g d
g ddg d          }t          j        ||           d S )Nr=   r   r[   r   r2   r   r   c                 *    |                                  S r%   r  rK   s    r    r*   z,test_apply_no_suffix_index.<locals>.<lambda>  s     r+   c                 *    |                                  S r%   r  rK   s    r    r*   z,test_apply_no_suffix_index.<locals>.<lambda>  s    AEEGG r+   )r  r  r  )   r  r  r`   )r   r  r  ro   ry   )pdfr   r   s      r    test_apply_no_suffix_indexr    s    
aVHqL3*
5
5
5CYY002C2CDEEFll..6U6U6U  H &(+++++r+   c                      t          dddgi          } |                     d dd          }t          ddg          }t          j        ||           d S )Nr   aabbbc                     | d         S r   r&   rK   s    r    r*   z/test_apply_raw_returns_string.<locals>.<lambda>  r   r+   r   Tr   r   r   s      r    test_apply_raw_returns_stringr    s[    	C$'	(	(BXXnn1$X77FtUm$$H68,,,,,r+   c                      t          g dd          } d }d|ddg}|                     |          }t          g dg d	g d
dg d          }t          j        ||           d S )N))r   r   r   )r>   r   r   )r[   r   r   )r=   r   r=   )r   r   r   )r   r   r   )att1att2att3r   c                 0    |                                  dz  S r   r  r   s    r    sum_div2z4test_aggregation_func_column_order.<locals>.sum_div2  s    uuww{r+   r   r  rd  )g      5@g      %@r  r   )g      2@g      "@r  r  )g      1@g      !@r  r  )r   r  r  rd  ro   rU   )rC   r  aggsr   r   s        r    "test_aggregation_func_column_orderr    s     
	
 	
 	
 )

 

 

B   8We,DVVD\\F***))))))	
 	

 211  H &(+++++r+   c                      t          g dg dd          } | ddg                             d d          }t          g d          }t          j        ||           d S )	NrV  r   r  rR   c                 8    | j         d         | j         d         z   S )Nr   r   )r  rK   s    r    r*   z+test_apply_getitem_axis_1.<locals>.<lambda>  s    AF1Iq	,A r+   r   r   )r   r>   r=   r   r   s      r    test_apply_getitem_axis_1r    sm    	33	4	4Bc
^!!"A"A!JJFiii  H68,,,,,r+   c                      d } t          dg di          }t          j        t          d          5  |                    | g           d d d            d S # 1 swxY w Y   d S )Nc                 R    t          j        d           |                                 S )NHello, World!)warningswarnr   rK   s    r    expected_warningzDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warning  s    o&&&uuwwr+   rR   r   r  r  )r   r   assert_produces_warningUserWarningrV   )r  rC   s     r    *test_nuisance_depr_passes_through_warningsr    s       
C#	$	$B		#K	G	G	G # #
 !"""# # # # # # # # # # # # # # # # # #s   AAAc            	         t          ddt          gdt          ddd          t          j        gdg d          } |                     t          d	
          }t          t          t          d          }t          j	        ||           |                     t          d
          }t          t          t          t          d          }t          j	        ||           d S )Nr[   r*  g      ?i  r   )ra  rb  r   ro   r   r   )
r   r   r   r   rq   r   typer   r   r   r   s      r    test_apply_typer    s    	Xu%htQ6J6JBF/STToo
 
 
B XXdX##Fvv6677H68,,, XXdX##FFf==>>H68,,,,,r+   c                      t          ddgddgd          } |                     d                              d d          }t          g t          j                  }t          j        ||           d S )	Nr   r>   r[   r   r  c                 :    t          | d         | d                   S )NrR   r   )re  rK   s    r    r*   z/test_apply_on_empty_dataframe.<locals>.<lambda>  s    AcFAcF(;(; r+   r   r   )r   headr   r   r   r   r   r   r   s      r    test_apply_on_empty_dataframer    sr    	!Qq!f--	.	.BWWQZZ;;!DDFb
+++H68,,,,,r+   c                      t          ddgddgd          } |                     d           }t          ddggddggd          }t          j        ||           d S )Nr   r>   r[   r  c                     | j         gS r%   )r   rK   s    r    r*   z(test_apply_return_list.<locals>.<lambda>	  s
    
 r+   ry   r   s      r    test_apply_return_listr    sn    	!Qq!f--	.	.BXX**++FAx1vh7788H&(+++++r+   ztest, constant)r   r   r   )r>   r>   r>   c                     t          |           }t          |ddgd          }ddd}|                    |          }t          j        ||           d S )NrR   r   r  )rW   r   rA   r  r  r  )rM  constantdf1r   aggregationr   s         r    test_unique_agg_type_is_seriesr    s^     D//C8C:XFFFH x00KWW[!!F68,,,,,r+   c                  Z   t          g dg dg dd          } t          g d          }t          j        |                     d          |           |                     dd          }t          j        ||           |                     dd          }t          j        ||           d S )	N)r   r>   r   )r   r>   r   )r   r   r   r1  )TTFr   r   r   )r   r   r   r   r   r   r  s      r    /test_any_apply_keyword_non_zero_axis_regressionr     s    	CC	D	DB)))**H266q6>>8444XXe!X$$F68,,,XXeQF68,,,,,r+   c                     t          dg di          } dd}dd}|                     |ddd	
          }| dz   }t          j        ||           d}t          j        t
          |          5  |                     ||gddd	
          }d d d            n# 1 swxY w Y   t          ddgddgddggddgddgg          }t          j        ||           t          j        t
          |          5  |                     d|iddd	
          }d d d            n# 1 swxY w Y   t          g ddg          }t          j        ||           d S )Nr(   r   r   r   c                     | |z   |z   S r%   r&   )r(   rR   r   s      r    foo1z.test_agg_mapping_func_deprecated.<locals>.foo11      1uqyr+   r>   c                     | |z   |z   S r%   r&   )r(   r   r   s      r    foo2z.test_agg_mapping_func_deprecated.<locals>.foo24  r  r+   r[   r=   )r   r   z+using .+ in Series.agg cannot aggregate andr  r   r   r!  r  r  r   )r>   r[   r=   )r   r   )r>   r   )r   rV   r   r/   r  FutureWarning)rC   r  r  r   r   r  s         r     test_agg_mapping_func_deprecatedr  -  s   	C#	$	$B       VVD!Q!V$$FAvH&(+++
7C		#M	=	=	= 1 1taa001 1 1 1 1 1 1 1 1 1 1 1 1 1 1
Q!Q"b"c3Z&&9I,J  H &(+++ 
	#M	=	=	= 0 0dQQ//0 0 0 0 0 0 0 0 0 0 0 0 0 0 0SE222H&(+++++s$   ,BBB/DDDc                  r   t          t          j        d                              dd          ddg          } t	          j        t          d          5  |                     t          j                  }d d d            n# 1 swxY w Y   t          d	d	d
t                    }t	          j        ||           t	          j        t          d          5  |                     t          j        g          }d d d            n# 1 swxY w Y   t          d	d	d
dg          }t	          j        ||           d S )Nr   r[   r>   r   r2   r   zusing DataFrame.stdr  r  r`   r   zusing Series.stdstdro   )r   r   rd   reshaper   r  r  rV   r  r   r   r   r/   r   s      r    test_agg_stdr  L  s   	29Q<<''1--Sz	B	B	BB		#M9N	O	O	O                                 Cc**%888H68,,,		#M9K	L	L	L " "!!" " " " " " " " " " " " " " "s--eW===H&(+++++s$    BBB!DD	Dc                      t          g ddt          j        dgg dd          } g d| _        |                     ddi          }| d                                         }t          j        ||           d S )	Nr  r   r  r  r1  )r   r   r3   r   r  )r   r   rq   rr   rV   r  r   r   r   s      r    (test_agg_dist_like_and_nonunique_columnsr  Z  s~    	ll#rvs!3:N:N:NOO
 
B !BJVVS'N##F#w}}H68,,,,,r+   )r   r  numpyr   r  pandas.core.dtypes.dtypesr   pandasrb   r   r   r   r   r	   pandas._testing_testingr   pandas.tests.frame.commonr
   r!   markparametrizer0   rD   rO   rX   re   r   r   rj   rv   rz   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   filterwarningsr   r   r  r
  r  r  r  r  r-  r<  rG  rM  rP  rX  r_  ri  rt  r|  r  r  r  r  r  r  r  rq   r  r  r   r  r  r  r  r  r  r  r  r'  r/  r:  r@  rE  rJ  rS  rX  r[  r^  re  rh  rk  rv  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r&   r+   r    <module>r      s               6 6 6 6 6 6                        0 0 0 0 0 01 1 1& !Q((u.., , /. )(,< < <, , , $$aS))*YY|-N-N-NPT,UV , ,	 ,
, 
, 
, "'27!344  54	- 	- 	-, , ,  , !>!>!>??- - @?-
- 
- 
-- - -, , ,, , ,, , ,"
, 
, 
,& & &, , , !Q((8 8 )(8 !Q((- - )(-, , , !Q((= = )(=  - - -- - - 455344
[[,,-J3G    u..!Q((1 1 )( /.  54 651 ) ) ), , ,- - -, , ,/ / /
- - -
- 
- 
-	, 	, 	,- - -D1, 1, 1,h- - -- - -, , ,- - -= = =  Iu~'7899F&))u+@+@ABB	
 
QFQF#	$	$ff!llaLL-I&J&JK	 		/ 	/	 		/
, 
, 
,& & &, , , /00
- 
- 10
-, , ,' ' '(. . .(, , , D"& 9::	- 	- ;:	-"   
	99cAq6]+C+CD
			3A-(@(@A
  	!8VVS!H-=-=>
  	!5&&#q*:*:;5"&"3"345IsIII&.H.H.HIII	
 5"&"3"345IsIII&.H.H.HIII	

 {{	Xyy#1v'?'?@
{{	UFFC8$4$45! *& &+ *& 		oo	 	ubf//01	kk	 # # #- - -(- - -, , , 
/	Hii1BQF0K&L&LM
/	E99.?!Q-H#I#IJ

	h		3*ZL(Q(Q(QR

	eYYaSz*%N%N%NOBF--.IsIII&.H.H.HIII	
 BF--.IsIII&.H.H.HIII	
 _kk*I1v1v&#sj*=U0VWWW	
 [++&IsQFmj*%=>>	
) :& &; :& 					))*	++&	 # # #- - -&- - -- - -$- - -  A''- - ('-- - -0 QF 344	- 	- 54	-, , ,", , ,, , ,, , ,	, 	, 	,, , ,	, 	, 	, 	**,P,PQ"""   

, 
, 

,"0 "0 "0J, , ,	C 	C 	C- - - , , ,86 6 6(B, B, B,J! ! !H  011)- )- 21)-X  011- - 21-, , ,  !Q((	B	R		Qx	11	11	111%%& - -  )(-  YYY//' ' 0/'- - - 		d3**M*M N NOPP#8#8#899  :9 QP CsBF ;<<- - =<-* * *4- - -, , ,- - -, , ,>- - -# # #"- - -$- - -, , , yyyyy	)	)!+E+EFyyyyy	)	)!A3+?+?@ - - -
- 
- 
-, , ,>, , ,	- 	- 	- 	- 	-r+   