
    d                     H   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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' Z/ej                            d(d)d*g          ej                            dd+ d, gd-d.g/          ej                            d0d1d2g          ej                            d d dg          d3                                                 Z0d4 Z1d5 Z2d6 Z3d7 Z4d8 Z5d9 Z6d: Z7d; Z8d< Z9d= Z:d> Z;d? Z<d@ Z=dA Z>dB Z?ej                            dC e	dDdEgdFdGgg           edDdFdHdEdGdHg          g e	d dgdIdJgg           ed dIdHddJdHg          gg          dK             Z@dL ZAej                            dMddNg          dO             ZBdP ZCej                            dQdR edS          gdT ejD        dU          gg          dV             ZEej                            dW e	             e	 eFdX          Y           e	 eFdX          Z           e	g g g d[          g          ej                            deGd\ g          d]                         ZHd^ ZId_ ZJd` ZKda ZLdb ZMdc ZNdd ZOde ZPej                            dfdgdhg          di             ZQdj ZRdk ZSdl ZTdm ZUdn ZVej                            dMdodpdejW        g          dq             ZXdr ZYds ZZdt Z[du Z\dv Z]dw Z^dx Z_dy Z`ej                            dMddIg          dz             Zad{ Zbej                            d|g d}ddIgg          d~             Zcd Zdd Zed Zfd Zgd Zhd Zid Zjej                            dd d d gg d/          d             Zkd Zld Zmd Znd Zod Zpd Zqd Zrd Zsej                            ddhdgg          d             Ztej                            ddhdgg          d             Zud Zvej                            d d dg          ej                            dd}i fdi fdi fdddIifdddIdfddIddfdddIdJdfg          d                         Zwej                            dg d          d             Zxd Zyej                            d e	ddgddgd          g          ej                            dg d          d                         Zzej                            dddNd1dej{        g          d             Z|d Z}d Z~d Zd Zd Zd Zd Zd Zd Zej                            dg d}g ddg d}dgdfg dg dddIgdgdfg          d             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       _/var/www/html/t/fyr/venv311/lib/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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   Br#   )Cr$   Dr%   c0c1c                 ,    |                      d          S )Ncategoryastype)tss    r    <lambda>z-test_apply_categorical_func.<locals>.<lambda>+   s    :!6!6     )      r'   r(   )r   r   shape
isinstancedtyper   dfr   s     r    test_apply_categorical_funcr7   (   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 N xs    r    r.   z*test_apply_axis1_with_ea.<locals>.<lambda>5       a r/   r   r   r   r   r   r   assert_frame_equalr   r   s     r    test_apply_axis1_with_earF   2   sT    #	,5 A A ABCDDH^^KKa^00F&(+++++r/   zdata, dtyper   Nr   r9   r:   r;   c                     t          | g| ggddg|          }|                    d d          }t          j        ||           d S )Na)r   r4   c                     | S r>   r?   r@   s    r    r.   z0test_agg_axis1_duplicate_index.<locals>.<lambda>@   s    A r/   r   r   r   aggr   rD   )datar4   r   r   s       r    test_agg_axis1_duplicate_indexrN   9   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   r#   c                     | S r>   r?   r@   s    r    r.   z/test_apply_mixed_datetimelike.<locals>.<lambda>M   rB   r/   r   r   )	r   r	   pdto_timedeltar   aranger   r   rD   rE   s     r    test_apply_mixed_datetimeliker[   D   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)r\   empty_framer   s      r    test_apply_emptyr`   Q   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   r@   s    r    r.   z(test_apply_float_frame.<locals>.<lambda>\       QVVXX r/   r   c                 *    |                                  S r>   rc   r@   s    r    r.   z(test_apply_float_frame.<locals>.<lambda>a   rd   r/   r   r   )	r   r   r   nancolumnsr   r   locr   )r   no_rowsr   r   no_colss        r    test_apply_float_framerl   Z   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 )NrI   re   c                     | d         S )NrI   r?   r@   s    r    r.   z/test_apply_empty_except_index.<locals>.<lambda>i   s
    af r/   r   r   r   r   r   rD   rE   s     r    test_apply_empty_except_indexrp   f   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r4   rI   bcrh   )	r   r   appendr   rD   r   r   float64r   )r_   rA   r   r   empty_with_colss        r    test_apply_with_reduce_emptyr~   m   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 )Nrw   rz   )r   r   )r   r   getattrr   r,   floatr   r   )r\   r6   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 )Nrw   rz   r   re   rv   )	r   nuniquer   rh   r   r   Tr   r|   r6   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   r1   rQ   )r0         )      	   )rI   rI   ry   re   c                     | d         S Nr   r?   rT   s    r    r.   z/test_apply_standard_nonunique.<locals>.<lambda>   
    ! r/   r   r   )r   r0   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rt   re   )r   r   r   r   r   r   rD   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   rs   r   c                     i | ]}|S r?   r?   .0ry   ms     r    
<dictcomp>z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>   s    <<<1!Q<<<r/   )r   r   r   r   rh   r   rD   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lenrh   rA   r   s    r    r.   z4test_apply_broadcast_lists_columns.<locals>.<lambda>   s"    $uS!4556677 r/   r   r   rs   r|   )r4   r   rh   )	r   r   r   r   rh   r   r   r   rD   )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|   )r4   r   )	r   r   r   r   r   r   rh   r   rD   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?   r@   s    r    r.   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>   
    			 r/   r   r   rs   r   r   rD   )int_frame_const_colr6   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   abcre   )r   r   r@   s    r    r.   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>   s    &$u++666 r/   r   r   rs   r   copyr   rD   r   r6   r   r   s       r    'test_apply_broadcast_series_lambda_funcr      sS    	BXX66   F
 wwyyH&(+++++r/   r   c                 <    d }|                      ||d           d S )Nc                 X    t          | t          j                  sJ | j        dk    sJ d S Nr   r3   r   ndarrayndimr@   s    r    _assert_rawz/test_apply_raw_float_frame.<locals>._assert_raw   -    !RZ(((((v{{{{{{r/   Tr   raw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   r@   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 Nr1   r?   r@   s    r    r.   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>   s
    Q r/   T)r   r1   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   r@   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      ?rV   r   r   r   int64rv   re   )
r   r   r   r   r   rg   rX   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   rV   c                     | d         S )Nr   r?   r@   s    r    r.   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>  
    # r/   r   r   r   re   c                     | d         S )Nr#   r?   r@   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/   axr   rh   c                     | S r>   r?   r@   s    r    r.   r.   #      q r/   c                 *    |                                  S r>   rc   r@   s    r    r.   r.   #  s    AFFHH r/   identityr   )idsr   TFc                 V   t          d	i | g di}t          j        d          5  t          j        d          5  t          j        dt                      |t          j        g d                    }d d d            n# 1 swxY w Y   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 )
Nrw   r   r   T)recordf8rv   r   r?   )r   r   r   warningscatch_warningssimplefilterRuntimeWarningarrayr3   r   r   _get_agg_axisr   r   )	r   r\   r   r   r6   test_resis_reductionr   agg_axiss	            r    test_apply_empty_infer_typer   !  s    
	+	+b///*	+	+B		"	"	" 1 1$D111 	6 	6!(N;;;tBHRt44455H	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 	6 &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 1s5   D:B6DB	D	B	
BDD"%D"c                      t          g d          } |                     d d          }t          |t                     sJ d S )Nrw   re   c                 *    |                                  S r>   rc   r@   s    r    r.   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>;  rd   r/   r   r   )r   r   r3   )rk   r   s     r    %test_apply_empty_infer_type_broadcastr   9  sK    ooo...G]]--;]GGFfi(((((((r/   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                     | |z   S r>   r?   rA   howmuchs     r    add_somez4test_apply_with_args_kwds_add_some.<locals>.add_some@  s    7{r/   r1   r   c                     | dz   S r   r?   r@   s    r    r.   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>D  s
    1q5 r/   r   r   )r   r   r   r   s       r    "test_apply_with_args_kwds_add_somer  ?  s^        x33F  11H&(+++++r/   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                 0    |                                  |z   S r>   rc   r   s     r    agg_and_addz:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_addI  s    vvxx'!!r/   r1   r   c                 0    |                                  dz   S r   rc   r@   s    r    r.   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>M  s    16688a< r/   r  )r   r   r   )r   r  r   r   s       r    %test_apply_with_args_kwds_agg_and_addr  H  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?   )rA   subdivides      r    subtract_and_dividezJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divideR  s    C6!!r/   )r1   r1   )argsr  c                     | dz
  dz  S )N       @r?   r@   s    r    r.   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>V  s    AGs? r/   r   r   )r   r  r   r   s       r    -test_apply_with_args_kwds_subtract_and_divider  Q  sc    " " " " 2aHHF  !:!:;;H&(+++++r/   c                 d    |                      t                    }t          j        ||            d S r>   )r   r   r   rD   )r   r   s     r    test_apply_yield_listr  Z  s.    t$$F&+.....r/   c                     t           j        | j        d d d| j                            d          f<   |                     d          }|                     t           j        d          }t          j        ||           d S )Nr1   r   r   r   )	r   rg   ilocrh   get_locr   r   r   r   )r   r   r   s      r    test_apply_reduce_Seriesr  _  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   r1   rQ   r0   r'   r(   i0i1rh   r   r   r   )r  r  re   r&   )r   r   dictr   rh   r   r   r   )rM   r   r   s      r    test_apply_reduce_to_dictr  f  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                     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 )	N   
   r   r   c                 >    i | ]\  }}||                                 S r?   describer   ivs      r    r   z2test_apply_differently_indexed.<locals>.<dictcomp>w  s&    AAAda!QZZ\\AAAr/   rz   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randnr   r   r#  itemsrh   r   rD   r   r   r   s      r    test_apply_differently_indexedr+  s  s    	29??2r**	+	+B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  )r1   ABC1r.  )r1   YUM1r  )r1   DEF1r  )rI   marketpositionrz   c                     | d         S )Nr4  r?   )rs    r    fztest_apply_bug.<locals>.f  s    {r/   r   r   i  r-  r.  r1   r/  r  rQ   r0  r0   r1  r   r2  r   r3  )r   r   r   r   r   )	positionsr8  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            
      F   t          g dg dg dt          j                            d          t          j                            d          t          j                            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=  )dullr?  shinyr?  r?  r@  r@  r?  r@  r@  r@     )r   r#   r$   r%   EFc                     | S r>   r?   r@   s    r    r.   z,test_apply_convert_objects.<locals>.<lambda>  rB   r/   r   r   )r   r   r(  r)  r   r   rD   rE   s     r    test_apply_convert_objectsrE    s           $$$$$$U+	
 +	
- -H^ ^^KKa^00F&(+++++r/   c                     |                      d           }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S r>   namer@   s    r    r.   z(test_apply_attach_name.<locals>.<lambda>       r/   re   )r   r   rh   r   r   r   s      r    test_apply_attach_namerK    sJ    //00Fk)1DEEEH68,,,,,r/   c                     |                      d d          }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S r>   rH  r@   s    r    r.   z.test_apply_attach_name_axis1.<locals>.<lambda>  rJ  r/   r   r   re   )r   r   r   r   r   r   s      r    test_apply_attach_name_axis1rN    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   repeatrI  r   r@   s    r    r.   z6test_apply_attach_name_non_reduction.<locals>.<lambda>      163q66)B)B r/   r   r   rh   )	r   r   r   tilerh   r   r   r   rD   r   s      r    $test_apply_attach_name_non_reductionrV    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>   rQ  r@   s    r    r.   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>  rS  r/   r   r   c              3   r   K   | ]1}t          j        |d          t          j                            V  2dS )r   N)r   rR  r   rh   )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)rI   rI   rx   )ry   r   r   r   r1   rQ   r0   r   r   col1col2rT  c                 Z    t          t          |           t          |           d          S )N)minmax)r   rb  rc  r@   s    r    r.   z(test_apply_multi_index.<locals>.<lambda>  s"    vc!ffSVV&D&DEE r/   rb  rc  T
check_like)r   from_arraysr   r   r   rD   )r   rT   r   r   s       r    test_apply_multi_indexrg    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   r1   rQ   c                    d }|                      |d          }|                      |d          }|                      |          }t          j        ||           t          j        ||            t          j        ||           d S )Nc                 *    |                                  S r>   )to_dictr@   s    r    r.   z!test_apply_dict.<locals>.<lambda>	  s    199;; r/   ru   r   rr   )r   r   r   rD   )r6   dictsfnreduce_truereduce_falsereduce_nones         r    test_apply_dictrr    s     
	B((28(44K88BH855L((2,,K;...,+++;.....r/   c                    |                      d           }t          j        || dz             |                      t                     |                      d           d         d         }t	          |t
                    sJ d S )Nc                     | dz  S r   r?   r@   s    r    r.   ztest_applymap.<locals>.<lambda>  s
    QU r/   r1   c                 
    | | fS r>   r?   r@   s    r    r.   ztest_applymap.<locals>.<lambda>  s
    QF r/   r   r   )applymapr   rD   typer3   tuple)r   appliedr   s      r    test_applymaprz    s    ""??33G';?333 !!"2"233C8;Ffe$$$$$$$r/   valr   c                     t          | dg          }|                    d           j        d         }|t          k    sJ d S )NrI   )rM   c                     | S r>   r?   r@   s    r    r.   z7test_applymap_float_object_conversion.<locals>.<lambda>!  s    1 r/   r   )r   rv  dtypesobject)r{  r6   r   s      r    %test_applymap_float_object_conversionr    sJ     
c
	#	#	#B[[%%,Q/FVr/   c                  <   t          t          j                            d                    } |                                 }g d}|| _        |                    t                    }||_        |                     t                    }t          j        ||           d S )NrQ   r0   )rI   rI   rI   rI   )	r   r   r(  r   rh   rv  strr   rD   )r6   df2colsr   r   s        r    test_applymap_strr  %  s    	29##F++	,	,B
''))CDBJ||C  HH[[F&(+++++r/   zcol, valr   rP   	timedeltaz1 minc                     t          t          j                            d                    }||| <   |                    t                    }|j        d| f         t	          |j        d| f                   k    sJ d S )Nr  r   )r   r   r(  rv  r  ri   )colr{  r6   r   s       r    test_applymap_datetimeliker  2  so     
29##F++	,	,BBsG[[F:afRVAsF^!4!4444444r/   r   ABCrz   re   r   r#   r$   c                     | S r>   r?   r@   s    r    r.   r.   G  s    1 r/   c                 Z    |                      |          }t          j        ||            d S r>   )rv  r   rD   )r   r\   r   s      r    test_applymap_emptyr  >  s0     t$$F&(+++++r/   c                      t          ddgddgg                              d d          } t          ddgddgg          }t          j        | |           d S )	Nr   r1   rQ   r0   c                     | |z   S r>   r?   rA   ys     r    r.   z&test_applymap_kwargs.<locals>.<lambda>P  s
    q1u r/   r  r   r   )r   rv  r   rD   )r   r   s     r    test_applymap_kwargsr  N  sf    AA'((112D2D1JJF1a&1a&)**H&(+++++r/   c                 r   |                      d           }|                                 }t          j                            dd| j        t                    }t          j        ||<   |                     d d          }|                                }t          j        ||<   t          j
        ||           d S )Nc                 :    t          t          |                     S r>   r   r  r@   s    r    r.   z)test_applymap_na_ignore.<locals>.<lambda>W  s    #c!ff++ r/   r   r1   )sizer4   c                 :    t          t          |                     S r>   r  r@   s    r    r.   z)test_applymap_na_ignore.<locals>.<lambda>\  s    #c!ff++ r/   r   )	na_action)rv  r   r   r(  randintr2   boolrX   NAr   rD   )r   strlen_framefloat_frame_with_namaskstrlen_frame_na_ignorestrlen_frame_with_nas         r    test_applymap_na_ignorer  U  s    ''(=(=>>L%**,,9Q(9FFD "099 :   (,,..!#02FGGGGGr/   c                      t          t          dd                    } d }t          |                               |           d S )Nz1/1/2000r   rR   c                 *    | j         | j        | j        fS r>   )hourdaymonthr@   s    r    r\   z*test_applymap_box_timestamps.<locals>.funcg  s    qw''r/   )r   r	   r   rv  )serr\   s     r    test_applymap_box_timestampsr  c  sM    
J333
4
4C( ( ( cNND!!!!!r/   c            
         t          t          d          t          d          gt          dd          t          dd          gt          j        d          t          j        d          gt          j        dd          t          j        dd          gd	          } |                     d
           }t          ddgddgddgddgd	          }t          j        ||           d S )Nz
2011-01-01z
2011-01-02z
US/Easternr;   z1 daysz2 daysM)freqrI   rx   ry   r   c                 *    t          |           j        S r>   )rw  __name__r@   s    r    r.   z#test_applymap_box.<locals>.<lambda>  s    477#3 r/   r   	TimedeltaPeriod)r   r   rX   r  r  rv  r   rD   r   s      r    test_applymap_boxr  n  s
   	L))9\+B+BC,<888,<888 ,x((",x*@*@A	,S111	,S111	
 	

 
B [[3344F{+{+{+H%		
 	
 H &(+++++r/   c                      ddl m t          dt          ddd          gi          } |                     fd          } |                     fd          } | j        j        }|dk    sJ d S )	Nr   )BDayx1i  r   c                     |              z   S r>   r?   rA   r  s    r    r.   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>      q4466z r/   c                     |              z   S r>   r?   r  s    r    r.   z:test_frame_apply_dont_convert_datetime64.<locals>.<lambda>  r  r/   zM8[ns])pandas.tseries.offsetsr  r   r   rv  r  r4   )r6   r   r  s     @r    (test_frame_apply_dont_convert_datetime64r    s    ++++++	D8D!Q//01	2	2B	))))	*	*B	))))	*	*BU[FX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-01rQ   zEurope/Brussels)rS   r<   c                     | S r>   r?   r@   s    r    r.   z,test_apply_non_numpy_dtype.<locals>.<lambda>       r/   c                 0    | t          j        d          z   S )N1day)rX   r  r@   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   rD   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 )Nr  )rI   rx   ry   rI   r*   rv   c                     | S r>   r?   r@   s    r    r.   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>  r  r/   ro   r5   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   r1   rQ   rI   rz   rb  r  rK   r6   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   rL   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 )Nr1   r   r$   r#   r?   rows    r    
apply_listz3test_apply_nested_result_axis_1.<locals>.apply_list  s'    CHa#c(lACL99r/   )r0   r0   ABCDrz   r   r   )        r  r  )r   r   zerosr   r   r   r   r   )r  r  r6   r   r   s        r    test_apply_nested_result_axis_1r    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   2020r:   r;   zdatetime64[ns, UTC]rv   c                     | S r>   r?   r@   s    r    r.   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>  rB   r/   c                 *    |                                  S r>   )r   r@   s    r    r.   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>  s    affhh r/   rC   rE   s     r    %test_apply_noreduction_tzaware_objectr    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 )NrI   r   c                 <                         | j                   d S r>   r{   rI  r  namess    r    reducing_functionz8test_apply_function_runs_once.<locals>.reducing_function  s    SXr/   c                 <                         | j                   | S r>   r  r  s    r    non_reducing_functionz<test_apply_function_runs_once.<locals>.non_reducing_function  s    SX
r/   r   r   )r   r   r   r   )r6   r  r  r\   r  s       @r    test_apply_function_runs_oncer    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 )NrI   r   c                 2                         |            d S r>   extendr  r   s    r    r  z<test_apply_raw_function_runs_once.<locals>.reducing_function      cr/   c                 2                         |            | S r>   r  r  s    r    r  z@test_apply_raw_function_runs_once.<locals>.non_reducing_function      c
r/   Tr   )r   r   )r   r   r   rI   to_listr6   r  r  r\   r   s       @r    !test_apply_raw_function_runs_oncer    s     
C#	$	$BF         #$9: . .111I
4a(((bdllnn-------	. .r/   c                      t          dg di          } g fd}fd}||fD ];}d d = |                     |           | j                                        k    sJ <d S )NrI   r   c                 2                         |            d S r>   r{   r{  r   s    r    r  z;test_applymap_function_runs_once.<locals>.reducing_function  r  r/   c                 2                         |            | S r>   r  r  s    r    r  z?test_applymap_function_runs_once.<locals>.non_reducing_function  r  r/   )r   rv  rI   r  r  s       @r     test_applymap_function_runs_oncer    s    	C#	$	$BF         #$9: ( (111I
D'''''	( (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   efghr  rz   )rh   r4   c                 ,    |                      d          S )Nr  r+   r@   s    r    r.   z-test_apply_with_byte_string.<locals>.<lambda>  s     2 2 r/   )r   r   r   r  r   r   rD   r  s      r    test_apply_with_byte_stringr    sz    	28Wg.//%	A	A	AB7G"455wfUUUH XX2233F&(+++++r/   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  cderI   r*   rv   c                     | k    S r>   r?   )rA   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?   )rX   isnullr   NaN)r   rA   r{  s     r    
<listcomp>z1test_apply_category_equalness.<locals>.<listcomp>  s0    AAA!29Q<<	-Q#XAAAr/   rH  )r   r  r   rI   r   r   r   r   )r{  	df_valuesr6   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                  ,   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   r1   r   r   )r   r1   )r   r1   )r   r   r(  randr   fftr2   rfftr5   s     r    test_infer_row_shaper  "  s     
29>>"a((	)	)BXXbfjqX))/FWXXbfkX**0FVr/   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   r1   rI   rx   rz   c                 (    d| d         | d         z   iS NrT   rI   rx   r?   r@   s    r    r.   z,test_with_dictlike_columns.<locals>.<lambda>0      afqvo 6 r/   r   c                     g | ]}d diS )rT   rQ   r?   r   rZ  s     r    r  z.test_with_dictlike_columns.<locals>.<listcomp>1  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?   r@   s    r    r.   z,test_with_dictlike_columns.<locals>.<lambda>8  r
  r/   c                 
    d| iS )NrT   r?   r@   s    r    r.   z,test_with_dictlike_columns.<locals>.<lambda><  s
    #q r/   rT   rQ   )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?   r@   s    r    r.   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>J       r/   r   r   )r   rX   to_datetimer   r   r   r   r   s      r    (test_with_dictlike_columns_with_datetimer   A  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   r1   rI   rx   rz   c                 (    d| d         | d         z   iS r	  r?   r@   s    r    r.   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>R  r
  r/   rr   rs   rT   rQ   r  r  r   c                 (    d| d         | d         z   iS r	  r?   r@   s    r    r.   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>Z  r
  r/   )r   r   r   rD   r   r   s      r    %test_with_dictlike_columns_with_inferr$  O  s    	QFQF#c3Z	8	8	8BXX66QHXUUF#1v''H&(+++ 	'(('((BtH XX66QHXUUF&(+++++r/   c            	      T   t          t          t          j                            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 )Nr0   )rI   r   ofwordsz
2016-10-01H)rS   r  )rI   rx   r-   rI   rx   r   r   c                 "    g | ]}|d d         S rG   r?   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>i  s     BBBquBBBr/   r-   c                 "    g | ]}|d d         S rG   r?   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>m  s     CCCquCCCr/   )r   r   r   r(  r)  r	   r   rx  r\  r   r   r   s      r    test_with_listlike_columnsr+  ^  s   		**++---\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 )NrI   rx   qzrZ  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>v  s"    "K"K"K"B#c(NN2NNNr/   rA   r?   r  s   `r    r.   z;test_with_listlike_columns_returning_list.<locals>.<lambda>v  s     "K"K"K"KC"K"K"K r/   r   r   re   )r   r   r   from_tuplesr   r   r   r   r   s      r    )test_with_listlike_columns_returning_listr5  q  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>   r7  r8  r  s    r    r.   z1test_infer_output_shape_columns.<locals>.<lambda>  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>  s!    EEE!(+EEEr/   r   r   r   r   r\  r   r   r   s      r    test_infer_output_shape_columnsr=  {  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   rQ   r  rz   c                 
    g dS r   r?   r@   s    r    r.   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  r   r/   r   r   c                     g | ]}g d S )r   r?   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    :::Qyyy:::r/   c                 
    ddgS Nr   r1   r?   r@   s    r    r.   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  
    A r/   c                     g | ]}d dgS r   r1   r?   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    777!1v777r/   	r   r   r(  r)  r   r   r\  r   r   r   s      r    (test_infer_output_shape_listlike_columnsrG    s     
29??1a((///	B	B	BB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 )	NrI   r   r   re   c                 ,    t          j                  S r>   r   ones)r  r{  s    r    r.   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>  s    "'#,, r/   r   r   c                 8    g | ]}t          j                  S r?   rJ  )r   rZ  r{  s     r    r  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>  s!    ===rws||===r/   )r   r   r   r   r\  r   r   r   )r{  r6   r   r   s   `   r    0test_infer_output_shape_listlike_columns_np_funcrM    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   r0   rQ   )r   rQ   r0   r1   )r   r1   rQ   r0   r  c                     dS )NrE  r?   r@   s    r    funzDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    vr/   r   r   c                     g | ]}d S )rE  r?   r  s     r    r  zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>  s    777!v777r/   r<  )r6   rP  r   r   s       r    7test_infer_output_shape_listlike_columns_with_timestamprR    s    	 ,'','','',''	 
	
 
	

 
B   XXcX""F77r}}77788H68,,,,,r/   lstr   c                 "    t          t          j                            dd          g d          }|                     fdd          }t           fd|                                D                       }t          j        ||           d S )	Nr0   rQ   r  rz   c                     S r>   r?   )rA   rS  s    r    r.   z3test_consistent_coerce_for_shapes.<locals>.<lambda>  s     r/   r   r   c                     g | ]}S r?   r?   )r   rZ  rS  s     r    r  z5test_consistent_coerce_for_shapes.<locals>.<listcomp>  s    444qs444r/   rF  )rS  r6   r   r   s   `   r    !test_consistent_coerce_for_shapesrW    s     
29??1a((///	B	B	BB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otherr  re   r   r@   s    r    r.   z'test_consistent_names.<locals>.<lambda>  s    &*C*C*CDDD r/   r   r   rZ  r[  r  r  rz   c                 ,    t          ddgddg          S )Nr   r1   rZ  r[  re   r\  r@   s    r    r.   z'test_consistent_names.<locals>.<lambda>  s    1vfg5F G G G r/   )r   renamer   rD   r   s       r    test_consistent_namesr_    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?   r@   s    r    r.   z"test_result_type.<locals>.<lambda>  r   r/   r   rr   rs   r   r   r1   r   r   rh   r   rD   r   s       r    test_result_typerd    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 rB  r?   r@   s    r    r.   z/test_result_type_shorter_list.<locals>.<lambda>  rC  r/   r   rr   rs   r   r#   r   rc  r   s       r    test_result_type_shorter_listrg    sb     
BXX&&QHXEEF3*~""$$H1vH&(+++++r/   c                     | }|                     d dd          }|                                }t          j        ||           d S )Nc                 
    g dS r   r?   r@   s    r    r.   z,test_result_type_broadcast.<locals>.<lambda>  r   r/   r   r   rs   r   r   s       r    test_result_type_broadcastrj    sJ     
BXX)){XKKFwwyyH&(+++++r/   c                     | }g d|                     fddd          }|                                }t          j        ||           d S )Nr[  r  r  c                 *    t          g d          S Nr   re   r\  rA   rh   s    r    r.   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &'222 r/   r   r   rs   r   r   r6   r   r   rh   s       @r    &test_result_type_broadcast_series_funcrq    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 rn  )r   r   r@   s    r    r.   z0test_result_type_series_result.<locals>.<lambda>	  s    yyy @ @ @ r/   r   r   r   r   s       r    test_result_type_series_resultrt    sH     
BXX@@qXIIFwwyyH&(+++++r/   c                     | }g d|                     fdd          }|                                }|_        t          j        ||           d S )Nrl  c                 *    t          g d          S rn  r\  ro  s    r    r.   z<test_result_type_series_result_other_index.<locals>.<lambda>  s    yyy @ @ @ r/   r   r   rc  rp  s       @r    *test_result_type_series_result_other_indexrw    sb     
B'''GXX@@@@qXIIFwwyyHH&(+++++r/   boxc                      t          |           S r>   )r   r@   s    r    r.   r.     s    tAww r/   c                      t          |           S r>   )rx  r@   s    r    r.   r.     s    %(( r/   c                 .    t          j        | d          S )Nr   rv   )r   r   r@   s    r    r.   r.     s    bhq6P6P6P r/   )r   rx  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 rB  r?   rA   rx  s    r    r.   z,test_consistency_for_boxed.<locals>.<lambda>#      QF r/   r   r   c                 *    g | ]} d dg          S rE  r?   )r   rZ  rx  s     r    r  z.test_consistency_for_boxed.<locals>.<listcomp>$  s%    <<<qssAq6{{<<<r/   c                      ddg          S rB  r?   r~  s    r    r.   z,test_consistency_for_boxed.<locals>.<lambda>'  r  r/   rr   rs   r   r#   r   rV   rz   )r   r   r\  r   r   r^  rD   )rx  r   r6   r   r   s   `    r    test_consistency_for_boxedr    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   rD   r   from_productrh   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   rV   rb  rc  r   r0   r   r#   r  )r   r   rL   r   rD   r   s      r    	test_demor  Q  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   rV   rb  rc  r         @r        @g      9@r   r#   )rc  rb  r   r  )r   r   rL   r   rg   r   rD   reindex_liker   s      r    test_demo_dict_aggr  \  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 )NrI  r   r<  countr1   rH  )r   rL   r   r   r   )rM   r6   r   r   s       r    !test_agg_with_name_as_column_namer  h  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<  bazr  r   r   r         @r<  	foobarbazrb  r   re   )r$   r#   r   )r   rL   r   rD   reindex)mdfr   r   s      r    test_agg_multiple_mixedr  x  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  rP   rQ   rR   r   r#   r$   r%   does not support reductionmatchrb  r   )r%   r$   r#   r   )r   r	   pytestraises	TypeErrorrL   )r  msgs     r    test_agg_multiple_mixed_raisesr    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 )Nr  r   r   r1   r   )r   rc  r   r   r   re   >   rh   r   rc  )axesuniquesort_valuesrX   concatr   rc  r   rh   r   rL   r   rD   r   ri   r   r   )r   r   r  name1name2r   r   r\   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  rP   rQ   rR   r  rb  r   r   r<  re   rT  r  r  r   r  )r   r  r  )r   r	   rL   r   r   rh   r   r   rD   r  r  r  )r6   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Nr1   rQ   r   r  r   Nr<  r  r   r  r1   r  rQ   )r  r  Trd  )	r   r   rg   r   r   r   r   rD   r  )r  r6   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  r  r  r   )r   r   r   re   r   )
r   r   rg   r   r   r2   rh   r   r   r   )r  r   r6   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)r1   r1   rQ   g      ?r  r  c                 h    t          |                                                                           S r>   )r   dropnar  )	group_cols    r    r\   z&test_agg_listlike_result.<locals>.funcR  s(    I$$&&--//000r/   r1   rQ   r   r<  re   r\   )
r   r   rg   rL   r   r   r   to_framer   rD   )r6   r\   r   r   s       r    test_agg_listlike_resultr  N  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      rE  r  rx   r?   rI   rx   rw   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 )NrQ   c                 <    |                                  ||z   |z  z   S r>   r   )rA   rI   rx   ry   s       r    r8  ztest_agg_args_kwargs.<locals>.fl  s    uuww!a%1$$r/   r   r1   r0   r   r  g      @r  g       @)rQ   r   r   rL   r   r   )r   r  kwargsr8  r6   r   r   s          r    test_agg_args_kwargsr  ^  s    % % % % 
QFQF#	$	$Bqyy3*%%3*%%RVAt-d---f--F68,,,,,r/   num_cols)r1   rQ   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   rT  c                     | S r>   r?   r@   s    r    r.   z,test_frequency_is_original.<locals>.<lambda>  r   r/   )rX   DatetimeIndexr   r   r   r   r  )r  r   originalr6   s       r    test_frequency_is_originalr  {  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+0000r:   r;   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000rb  )rM   r   c                     | j         S r>   rH  r@   s    r    r.   z.test_apply_datetime_tz_issue.<locals>.<lambda>  s     r/   r   r   )r   rM   )r   r   r   r   r   r   )
timestampsr6   r   r   s       r    test_apply_datetime_tz_issuer    s     	3>>>3>>>3>>>J
 
				4	4	4BXX&&QX//FJZ888H68,,,,,r/   r6   rI   ry   r   rV   method)rb  rc  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   )r6   r  r  s      r    test_mixed_column_raisesr    s     >D	y	,	,	,  F                 s   AAAr  c                     t          d| ggddg          }|                    d           }|j        }t          j        ||           d S )Nr   rI   rx   rz   c                     | j         S r>   rv   r@   s    r    r.   z"test_apply_dtype.<locals>.<lambda>  s     r/   )r   r   r~  r   r   )r  r6   r   r   s       r    test_apply_dtyper    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 )NrI   r   )_mgrri   )r  mgrs     r    r\   z!test_apply_mutating.<locals>.func  s9    hxs""""
r/   rI   r   r   )r   r   r   r   r   rD   )using_array_managerusing_copy_on_writer6   df_origr\   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   r1   rQ   r0   r   r   r   r   r   r   rI   rx   rz   c                     g S r>   r?   r@   s    r    r.   z.test_apply_empty_list_reduce.<locals>.<lambda>  r  r/   ru   r   r  rv   )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 )Nr0   r   rQ   r   r#   rz   r   c                 *    |                                  S r>   r  r@   s    r    r.   z,test_apply_no_suffix_index.<locals>.<lambda>  s     r/   c                 *    |                                  S r>   r  r@   s    r    r.   z,test_apply_no_suffix_index.<locals>.<lambda>  s    AEEGG r/   )r  r  r  )   r  r  rV   )r   <lambda>r  re   ro   )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?   r@   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   )r1   r   r   )rQ   r   r   )r0   r   r0   )r   r   r   )r   r   r   )att1att2att3rz   c                 0    |                                  dz  S r   r  r   s    r    sum_div2z4test_aggregation_func_column_order.<locals>.sum_div2  s    uuww{r/   r   r  rb  )g      5@g      %@r  r   )g      2@g      "@r  r  )g      1@g      !@r  r  )r   r  r  rb  re   rK   )r6   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 )	Nrb  r   r  rI   c                 $    | d         | d         z   S )Nr   r   r?   r@   s    r    r.   z+test_apply_getitem_axis_1.<locals>.<lambda>  s    AaD1Q4K r/   r   r   )r   r1   r0   r   r   s      r    test_apply_getitem_axis_1r    sm    	33	4	4Bc
^!!"7"7a!@@F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!)r   warnr   r@   s    r    expected_warningzDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warning  s    o&&&uuwwr/   rI   r   r  r  )r   r   assert_produces_warningUserWarningrL   )r  r6   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                    }t          t          t          t          gt          t          t          gdg d          }t          j
        ||           |                     t          d	
          }t          t          t          d          }t          j        ||           |                     t          d
          }t          t          t          t          d          }t          j        ||           d S )NrQ   r8  g      ?i  r   )r_  r`  rw   re   r   r   )r   r   r   r   rg   rv  rw  intr  r   rD   r   r   r   r   s      r    test_apply_typer	    s&   	Xu%htQ6J6JBF/STToo
 
 
B [[FsD!E8U+CDDoo  H &(+++ 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   r1   rQ   r   r  c                 :    t          | d         | d                   S )NrI   rx   )rc  r@   s    r    r.   z/test_apply_on_empty_dataframe.<locals>.<lambda>5  s    AcFAcF(;(; r/   r   rv   )r   headr   r   r   r|   r   r   r   s      r    test_apply_on_empty_dataframer  2  sr    	!Qq!f--	.	.BWWQZZ;;!DDFb
+++H68,,,,,r/   ztest, constant)r   r   r   )r1   r1   r1   c                     t          |           }t          |ddgd          }ddd}|                    |          }t          j        ||           d S )NrI   rx   r  )rM   r   r4   r  r  r  )rZ  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   r1   r   )r   r1   r   )r   r   r   r  )TTFr   r   r   )r   r   r   r   r   r   r  s      r    /test_any_apply_keyword_non_zero_axis_regressionr  L  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}t          j        t          |	          5  |                     ||gdd
d
d           d d d            n# 1 swxY w Y   |                     ||gdd
d          }t          ddgddgddggt          j        ddg                    }t          j        ||           d S )NrA   r   r   r   c                     | |z   |z   S r>   r?   )rA   rI   ry   s      r    foo1z/test_agg_list_like_func_with_args.<locals>.foo1]      1uqyr/   r1   c                     | |z   |z   S r>   r?   )rA   rx   ry   s      r    foo2z/test_agg_list_like_func_with_args.<locals>.foo2`  r  r/   z/foo1\(\) got an unexpected keyword argument 'b'r  rQ   r0   )rx   ry   )ry   r   r   r   )rA   r  )rA   r  rz   )r   r   )r1   r   )	r   r  r  r  rL   r   r4  r   rD   )r6   r  r  r  r   r   s         r    !test_agg_list_like_func_with_argsr  Y  sH   	C#	$	$B       =C	y	,	,	, - -
d|QQ!,,,- - - - - - - - - - - - - - - VVT4L!Q!V,,F
Q!Q"b"&}'EFF  H &(+++++s   A""A&)A&)r   r   numpyr   r  pandas.core.dtypes.dtypesr   pandasrX   r   r   r   r   r	   pandas._testing_testingr   pandas.tests.frame.commonr
   r!   r7   rF   markparametrizerN   r[   r   r   r`   rl   rp   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:  rE  rK  rN  rV  r]  rg  rr  rz  r  r  r  r  r   roundr  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r$  r+  r5  r=  rG  rM  rR  rW  r_  rd  rg  rj  rq  rt  rw  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rg   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&< < <, , , $$aS))*YY|-N-N-NPT,UV , ,	 ,
, 
, 
, "'27!344  54	- 	- 	-, , ,  , !>!>!>??- - @?-
- 
- 
-- - -, , ,, , ,, , ,"
, 
, 
,& & &, , , !Q((8 8 )(8 !Q((- - )(-, , , !Q((= = )(=  - - -- - - 344
[[,,-J3G    u..!Q((1 1 )( /.  54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	 		/ 	/	 		/% % % C))  *)
, 
, 
, ))J''(;W8M8M*NO 5 5	 5 		$$u++&&&	U$$$	"--..	  %!566, , 76 ,, , ,H H H" " ", , ,:	 	 	
, 
, 
,& & &, , , /00
- 
- 10
-, , ,' ' '(. . .(( ( ($, , , D"& 9::	- 	- ;:	-"  - - -(- - -, , ,- - -&- - -- - -$- - - 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- - -, , ,- - -, , ,>- - -# # #"- - -4- - - yyyyy	)	)!+E+EFyyyyy	)	)!A3+?+?@ - - -
- 
- 
-, , , , ,r/   