
    bMh                     (   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mZ d dl	Z
d dl	mZmZmZmZmZ d dlmZ d dlmZ d dlmZ ej        d             Z ej        d	 ej        d
ej        j                  g          d             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* Z0d+ Z1d, Z2ej                            dd dg          d-             Z3ej                            dd dg          d.             Z4d/ Z5ej                            dd dg          d0             Z6d1 Z7d2 Z8d3 Z9ej        :                    d4          ej                            d5d6d7g          ej                            dd8 d9 gd:d;g<          ej                            dddg          ej                            dd dg          d=                                                             Z;d> Z<d? Z=d@ Z>dA Z?dB Z@dC ZAdD ZBdE ZCdF ZDdG ZEdH ZFdI ZGdJ ZHdK ZIdL ZJej                            dM edNdOgdPdQgg           edNdPdRdOdQdRg          g ed dgdSdTgg           ed dSdRddTdRg          gg          dU             ZKdV ZLdW ZMdX ZNej                            dYdZd[g          d\             ZOd] ZPd^ ZQd_ ZRd` ZSej                            dadbdcdejT        g          dd             ZUde ZVej                            dfdgdh idi edgdSdTgi          fdgdj id edgdSdTgi          fdgdk idi edgdTi          fdgdl id edgdTi          fdgdmejW        dn gidi edgg doig dpq          fdgdmejW        dr gid edgg doig dpq          fdgds idi edgddgi          fdgdt id edgdi          fg          du             ZXej                            dvdgdw idgdx idgdmejW        dy gidgdz ig          d{             ZYd| ZZd} Z[d~ Z\ej                            dfd gdi eddSdTgi          fd gd eddSdTgi          fd gdi edgdTgidgq          fd gd edgdTgidgq          fdmejW        d gdi edgg doig dpq          fdmejW        d gd edgg doig dpq          fd d gdi edSdTgdTdTggdgdggddgg          fd d gd edgdSdTgiddg          fg          d             Z]ej                            dvd gd gdmejW        d gd d gg          d             Z^d Z_d Z`d Zad Zbej                            daddSg          d             Zcd Zdej                            dg dddSgg          d             Zed Zfd Zgd Zhd Zid Zjd Zkd Zlej                            dd d d gg d<          d             Zmd Znd Zod Zpd Zqd Zrd Zsd Ztd Zuej                            dd[dZg          d             Zvej                            dd[dZg          d             Zwd Zxej                            dd dg          ej                            ddi fdi fdi fdddSifdddSdfddSddfdddSdTdfg          d                         Zyej                            dg d          d             Zzd Z{ej                            d edgdgddgdŜ          g          ej                            dg dǢ          dȄ                         Z|ej                            dddddgejT        g          d˄             Z}d̄ Z~d̈́ Zd΄ Zdτ ZdЄ Zdф Zd҄ Zdӄ ZdԄ ZdՄ Zej                            dg dg dעdg ddgdfg dآg dעddSgdgdfg          dل             Zdڄ Zdۄ Zd܄ Zd݄ ZdS )    )datetimeN)is_platform_arm)CategoricalDtype)	DataFrame
MultiIndexSeries	Timestamp
date_range)
zip_frames)Versionc                      t          t          j        t          j        dd          d                              dd          dz   g d          } | S )	z
    Fixture for DataFrame of ints which are constant per column

    Columns are ['A', 'B', 'C'], with values (per column): [1, 2, 3]
       int64dtype      ABCcolumns)r   nptilearangereshape)dfs    c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/apply/test_frame_apply.pyint_frame_const_colr!      sY     

	!7+++Q//772>>B
 
 
B I    pythonnumba)marks)paramsc                 N    | j         dk    rt          j        d           | j         S )Nr$   )parampytestimportorskip)requests    r    enginer,   &   s(    }G$$$=r"   c                 
   |dk    r:t           j                            d          }|j                            |           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 )Nr$   z+numba engine not supporting numpy ufunc yetreasonignoreallr   r,   r   r   axisr,   )r)   markxfailnode
add_markerr   errstatesqrtapplytmassert_series_equalmeanindexxs)float_framer,   r+   r6   resultexpectedds          r    
test_applyrF   -   s   {  (U VV%%%		"	"	" 1 1S)**$$RWV$<<SA
vx000 ""276"::3?7;s+,,!!!!a ""276"BB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E88E<?E<r5   r   rawTFc                    |dk    rt          j        d          }t          |j                  t          d          k    r*t	                      rt          j        d|j                    t           j                            d          }|j        	                    |           | 
                    d |d||          }| d	z   }t          j        ||           d S )
Nr$   z0.61z&Segfaults on ARM platforms with numba z!numba engine doesn't support argsr.   c                     | |z   S N xys     r    <lambda>z!test_apply_args.<locals>.<lambda>M   s
    QU r"   r   )argsrG   r,   r   )r)   r*   r   __version__r   skipr6   r7   r8   r9   r<   r=   assert_frame_equal)	rB   r5   rG   r,   r+   r$   r6   rC   rD   s	            r    test_apply_argsrU   C   s     #G,,5$%%88_=N=N8KTARTTUUU{  (K LL%%%DtV   F 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   r   r   )r   r   DrW   c0c1c                 ,    |                      d          S )Ncategoryastype)tss    r    rO   z-test_apply_categorical_func.<locals>.<lambda>V   s    :!6!6 r"   )      rY   rZ   )r   r<   shape
isinstancer   r   r   rC   s     r    test_apply_categorical_funcre   S   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 rJ   rK   rM   s    r    rO   z*test_apply_axis1_with_ea.<locals>.<lambda>`       a r"   r   r5   r   r	   r<   r=   rT   rD   rC   s     r    test_apply_axis1_with_earq   ]   sT    #	,5 A A ABCDDH^^KKa^00F&(+++++r"   zdata, dtyper   Nrg   rh   ri   c                     t          | g| ggddg|          }|                    d d          }t          j        ||           d S )Na)r@   r   c                     | S rJ   rK   rl   s    r    rO   z0test_agg_axis1_duplicate_index.<locals>.<lambda>k       A r"   r   rn   r   aggr=   rT   )datar   rD   rC   s       r    test_agg_axis1_duplicate_indexrz   d   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20130101r   periodss)unitr   r   c                     | S rJ   rK   rl   s    r    rO   z/test_apply_mixed_datetimelike.<locals>.<lambda>x   rm   r"   r   rn   )	r   r
   pdto_timedeltar   r   r<   r=   rT   rp   s     r    test_apply_mixed_datetimeliker   o   sy     J2221C888	
 	
 H ^^KKa^00F&(+++++r"   funcc                 b    t                      }|                    | |          }|j        sJ d S )Nr3   )r   r<   empty)r   r,   empty_framerC   s       r    test_apply_emptyr   |   s9     ++KtF33F<r"   c                 r   | 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 rJ   r?   rl   s    r    rO   z(test_apply_float_frame.<locals>.<lambda>       QVVXX r"   r3   r@   c                 *    |                                  S rJ   r   rl   s    r    rO   z(test_apply_float_frame.<locals>.<lambda>   r   r"   r   r4   )	r<   r   r   nanr   r=   r>   locr@   )rB   r,   no_rowsrC   rD   no_colss         r    test_apply_float_framer      s    "1"oG]]--f]==FbfK$7888H68,,,oaaae$G]]--Af]EEFbfK$5666H68,,,,,r"   c                     t          dg          }|                    d d|           }t          j        ||           d S )Nrt   r   c                     | d         S )Nrt   rK   rl   s    r    rO   z/test_apply_empty_except_index.<locals>.<lambda>   s
    af r"   r   r4   r   r<   r=   rT   )r,   rD   rC   s      r    test_apply_empty_except_indexr      sH    u%%%H^^,,1V^DDF&(+++++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r5   result_typereducer   rt   bcr   )	r   r<   appendr=   rT   r   r   float64r>   )r   rM   rC   rD   empty_with_colss        r    test_apply_with_reduce_emptyr      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anyr2   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>   )r   r   rC   rD   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   r   r   )	r   nuniquer   r   r=   r>   Tr   r   r   rC   rD   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   ra   r   )r`      r   )      	   )rt   rt   r   r   c                     | d         S Nr   rK   r   s    r    rO   z/test_apply_standard_nonunique.<locals>.<lambda>   
    ! r"   r   rn   )r   r`   r   c                     | d         S r   rK   r   s    r    rO   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   r   )r<   r   r?   r   r@   r=   rT   rB   rC   rD   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   rn   c                     i | ]}|S rK   rK   .0r   ms     r    
<dictcomp>z6test_apply_broadcast_scalars_axis1.<locals>.<dictcomp>   s    <<<1!Q<<<r"   )r<   r   r?   r   r   r=   rT   rB   rC   rD   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 rJ   )listrangelenr   rM   rB   s    r    rO   z4test_apply_broadcast_lists_columns.<locals>.<lambda>   s"    $uS!4556677 r"   r   r   r   r   )r   r@   r   )	r<   r   r   r   r   r   r@   r=   rT   )rB   rC   r   rD   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 rJ   )r   r   r   r@   r   s    r    rO   z2test_apply_broadcast_lists_index.<locals>.<lambda>   s"    $uS!2334455 r"   r   r   c                     i | ]}|S rK   rK   r   s     r    r   z4test_apply_broadcast_lists_index.<locals>.<dictcomp>   s    +++!A+++r"   r   )r   r@   )	r<   r   r   r   r@   r   r   r=   rT   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   rK   rl   s    r    rO   z7test_apply_broadcast_list_lambda_func.<locals>.<lambda>  
    			 r"   r   r   r   r<   r=   rT   )r!   r   rC   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   abcr   )r   r   rl   s    r    rO   z9test_apply_broadcast_series_lambda_func.<locals>.<lambda>  s    &$u++666 r"   r   r   r   )r<   copyr=   rT   r!   r   rC   rD   s       r    'test_apply_broadcast_series_lambda_funcr     sS    	BXX66   F
 wwyyH&(+++++r"   c                 r    |dk    rt          j        d           d }|                     |||d           d S )Nr$   z)numba can't handle when UDF returns None.c                 X    t          | t          j                  sJ | j        dk    sJ d S Nr   rc   r   ndarrayndimrl   s    r    _assert_rawz/test_apply_raw_float_frame.<locals>._assert_raw  -    !RZ(((((v{{{{{{r"   Tr5   r,   rG   )r)   rS   r<   )rB   r5   r,   r   s       r    test_apply_raw_float_framer     sR    ?@@@   kVFFFFFr"   c                     |                      t          j        ||d          }|                      d |          }t          j        ||           d S )NTr   c                 4    | j                                         S rJ   )valuesr?   rl   s    r    rO   z3test_apply_raw_float_frame_lambda.<locals>.<lambda>  s    18==?? r"   rn   )r<   r   r?   r=   r>   )rB   r5   r,   rC   rD   s        r    !test_apply_raw_float_frame_lambdar     sU    rwT&dKKF  !:!: FFH68,,,,,r"   c                 l    |                      d |d          }| dz  }t          j        ||           d S )Nc                     | dz  S Nra   rK   rl   s    r    rO   z9test_apply_raw_float_frame_no_reduction.<locals>.<lambda>%  s
    Q r"   T)r,   rG   ra   r   )rB   r,   rC   rD   s       r    'test_apply_raw_float_frame_no_reductionr   #  s@    v4HHFQH&(+++++r"   c           
      $   |dk    rt          j        d           d }t          dddt          j        dgdz  d	          t          j        d
gdz  d	          dt          j        d                    }|                    || |d           d S )Nr$   z(isinstance check doesn't work with numbac                 X    t          | t          j                  sJ | j        dk    sJ d S r   r   rl   s    r    r   z4test_apply_raw_mixed_type_frame.<locals>._assert_raw/  r   r"         ?ra   foo
   float32r   r   int32)rt   r   r   r   r   r   Tr   )r)   rS   r   r   arrayr   r<   )r5   r,   r   r   s       r    test_apply_raw_mixed_type_framer   *  s    >???  
 
x
)<<<XqcBhg666	
 	
 imm	
 	
 	
B HH[tFH=====r"   c                     | j         d         }|                     t          j        d          |         }t          j        |                     |                    }||k    sJ d S )Nr   r   rn   )r@   r<   r   r?   rA   )rB   rE   rC   rD   s       r    test_apply_axis1r   A  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 )
Nr   r   r   r   r   rn   r   r   r   )
r   r<   r   r?   r   r   r   Indexr=   r>   r   s      r    test_apply_mixed_dtype_cornerr  H  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   rK   rl   s    r    rO   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>S  
    # r"   r   rn   r   r   c                     | d         S )Nr   rK   rl   s    r    rO   z8test_apply_mixed_dtype_corner_indexing.<locals>.<lambda>W  r  r"   )r   r<   r   r=   r>   r   s      r    &test_apply_mixed_dtype_corner_indexingr  Q  s    	%u--	.	.BXX&&QX//FugaS)))H68,,,XX&&QX//FseA3'''H68,,,,,r"   zignore::RuntimeWarningaxr@   r   c                     | S rJ   rK   rl   s    r    rO   rO   _      q r"   c                 *    |                                  S rJ   r   rl   s    r    rO   rO   _  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   r0   r1   f8r   r   rK   )
r   r   r:   r   rc   r   r<   _get_agg_axisr   r@   )r  r   rG   r5   r,   r+   r   test_resis_reductionrC   agg_axiss              r    test_apply_empty_infer_typer  \  s?    
	+	+b///*	+	+B		"	"	" 
1 
14400011%h
;;;$T&cBB 	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   r   c                 *    |                                  S rJ   r   rl   s    r    rO   z7test_apply_empty_infer_type_broadcast.<locals>.<lambda>u  r   r"   r   r   )r   r<   rc   )r   rC   s     r    %test_apply_empty_infer_type_broadcastr  s  sK    ooo...G]]--;]GGFfi(((((((r"   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                     | |z   S rJ   rK   rM   howmuchs     r    add_somez4test_apply_with_args_kwds_add_some.<locals>.add_somez  s    7{r"   ra   r  c                     | dz   S r   rK   rl   s    r    rO   z4test_apply_with_args_kwds_add_some.<locals>.<lambda>~  s
    1q5 r"   r   r   )rB   r  rC   rD   s       r    "test_apply_with_args_kwds_add_somer  y  s^        x33F  11H&(+++++r"   c                     dd}|                      |d          }|                      d           }t          j        ||           d S )Nr   c                 0    |                                  |z   S rJ   r   r  s     r    agg_and_addz:test_apply_with_args_kwds_agg_and_add.<locals>.agg_and_add  s    vvxx'!!r"   ra   r  c                 0    |                                  dz   S r   r   rl   s    r    rO   z7test_apply_with_args_kwds_agg_and_add.<locals>.<lambda>  s    16688a< r"   r  )r<   r=   r>   )rB   r"  rC   rD   s       r    %test_apply_with_args_kwds_agg_and_addr$    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 rJ   rK   )rM   subdivides      r    subtract_and_dividezJtest_apply_with_args_kwds_subtract_and_divide.<locals>.subtract_and_divide  s    C6!!r"   )ra   ra   )rQ   r(  c                     | dz
  dz  S )N       @rK   rl   s    r    rO   z?test_apply_with_args_kwds_subtract_and_divide.<locals>.<lambda>  s    AGs? r"   rP   r   )rB   r)  rC   rD   s       r    -test_apply_with_args_kwds_subtract_and_divider,    sc    " " " " 2aHHF  !:!:;;H&(+++++r"   c                 d    |                      t                    }t          j        ||            d S rJ   )r<   r   r=   rT   )rB   rC   s     r    test_apply_yield_listr.    s.    t$$F&+.....r"   c                     t           j        | j        d d d| j                            d          f<   |                     d          }|                     t           j        d          }t          j        ||           d S )Nra   r   r   rn   )	r   r   ilocr   get_locr?   r<   r=   r>   )rB   rD   rC   s      r    test_apply_reduce_Seriesr2    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   ra   r   r`   rY   rZ   i0i1r   r@   r   rn   )r4  r5  r   rX   )r   r<   dictr   r   r=   r>   r@   )ry   rC   rD   s      r    test_apply_reduce_to_dictr8    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 )	Nra   )   r   r   rn   c                 >    i | ]\  }}||                                 S rK   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 rK   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=  itemsr   r=   rT   r   r@   r   s      r    test_apply_differently_indexedrF    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:  )ra   ABC1rI  )ra   YUM1r:  )ra   DEF1r:  )rt   marketpositionr   c                     | d         S )NrO  rK   )rs    r    fztest_apply_bug.<locals>.f  s    {r"   r   rn   i  rH  rI  ra   rJ  r:  r   rK  r`   rL  r   rM  r   rN  )r   r<   r   r=   r>   )	positionsrS  rD   rC   s       r    test_apply_bugrU    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   barrW  rW  rW  r   r   r   )onerX  rX  tworX  rX  rX  rY  rY  rY  rX  )dullrZ  shinyrZ  rZ  r[  r[  rZ  r[  r[  r[  ra      )r   r   r   rW   EFc                     | S rJ   rK   rl   s    r    rO   z,test_apply_convert_objects.<locals>.<lambda>  rm   r"   r   rn   )r   r   rB  rC  rD  r<   r=   rT   rp   s     r    test_apply_convert_objectsr`    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 rJ   namerl   s    r    rO   z(test_apply_attach_name.<locals>.<lambda>       r"   r   )r<   r   r   r=   r>   r   s      r    test_apply_attach_namerf    sJ    //00Fk)1DEEEH68,,,,,r"   c                     |                      d d          }t          | j        | j                  }t          j        ||           d S )Nc                     | j         S rJ   rc  rl   s    r    rO   z.test_apply_attach_name_axis1.<locals>.<lambda>  re  r"   r   rn   r   )r<   r   r@   r=   r>   r   s      r    test_apply_attach_name_axis1ri    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 rJ   r   repeatrd  r   rl   s    r    rO   z6test_apply_attach_name_non_reduction.<locals>.<lambda>      163q66)B)B r"   r   r@   r   )	r<   r   r   r   r   r   r@   r=   rT   r   s      r    $test_apply_attach_name_non_reductionrp    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 rJ   rl  rl   s    r    rO   z<test_apply_attach_name_non_reduction_axis1.<locals>.<lambda>'  rn  r"   r   rn   c              3   r   K   | ]1}t          j        |d          t          j                            V  2dS )r   N)r   rm  r   r   )r   trB   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_axis1rw  &  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)rt   rt   r   )r   rE   rE   r   ra   r   r`   r   r   col1col2ro  c                 Z    t          t          |           t          |           d          S )N)minmax)r   r|  r}  rl   s    r    rO   z(test_apply_multi_index.<locals>.<lambda>2  s"    vc!ffSVV&D&DEE r"   r|  r}  T
check_like)r   from_arraysr   r<   r=   rT   )r@   r   rC   rD   s       r    test_apply_multi_indexr  /  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   rW  spameggs)r   r   ra   r   c                    d }|                      |d          }|                      |d          }|                      |          }t          j        ||           t          j        ||            t          j        ||           d S )Nc                 *    |                                  S rJ   )to_dictrl   s    r    rO   z!test_apply_dict.<locals>.<lambda>C  s    199;; r"   r   r   r   )r<   r=   r>   rT   )r   dictsfnreduce_truereduce_falsereduce_nones         r    test_apply_dictr  7  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~   rj   c                     | S rJ   rK   rl   s    r    rO   z,test_apply_non_numpy_dtype.<locals>.<lambda>P       r"   c                 0    | t          j        d          z   S )N1day)r   	Timedeltarl   s    r    rO   z,test_apply_non_numpy_dtype.<locals>.<lambda>S  s    BL$8$8 8 r"   z
2015-01-02)r   r
   r<   r=   rT   r   s      r    test_apply_non_numpy_dtyper  M  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  )rt   r   r   rt   r\   r   c                     | S rJ   rK   rl   s    r    rO   z5test_apply_non_numpy_dtype_category.<locals>.<lambda>\  r  r"   r   rd   s     r    #test_apply_non_numpy_dtype_categoryr  Z  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   ra   r   rt   r   r|  r6  rw   r   rD   rC   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<   rx   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 )Nra   r   r   r   rK   rows    r    
apply_listz3test_apply_nested_result_axis_1.<locals>.apply_listl  s'    CHa#c(lACL99r"   )r`   r`   ABCDr   r   rn   )        r  r  )r   r   zerosr   r   r   r=   r>   )r  r  r   rC   rD   s        r    test_apply_nested_result_axis_1r  i  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   2020rh   ri   zdatetime64[ns, UTC]r   c                     | S rJ   rK   rl   s    r    rO   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>|  rm   r"   c                 *    |                                  S rJ   )r   rl   s    r    rO   z7test_apply_noreduction_tzaware_object.<locals>.<lambda>~  s    affhh r"   ro   rp   s     r    %test_apply_noreduction_tzaware_objectr  w  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 )Nrt   r   c                 <                         | j                   d S rJ   r   rd  r  namess    r    reducing_functionz8test_apply_function_runs_once.<locals>.reducing_function  s    SXr"   c                 <                         | j                   | S rJ   r  r  s    r    non_reducing_functionz<test_apply_function_runs_once.<locals>.non_reducing_function  s    SX
r"   r   rn   )r   r<   r   r@   )r   r  r  r   r  s       @r    test_apply_function_runs_oncer    s     
C#	$	$BE         #$9: ' '!!!H
ARX&&&&&	' 'r"   c                    | dk    rt          j        d           t          dg di          }g fd}fd}||fD ]L}d d = |                    || dd	           t	          |j                                                  k    sJ Md S )
Nr$   z8appending to list outside of numba func is not supportedrt   r   c                 2                         |            d S rJ   extendr  r   s    r    r  z<test_apply_raw_function_runs_once.<locals>.reducing_function  s    cr"   c                 2                         |            | S rJ   r  r  s    r    r  z@test_apply_raw_function_runs_once.<locals>.non_reducing_function  s    c
r"   Tr   )r,   rG   r5   )r)   rS   r   r<   r   rt   to_list)r,   r   r  r  r   r   s        @r    !test_apply_raw_function_runs_oncer    s    NOOO	C#	$	$BF         #$9: . .111I
f$Q777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   )r   r   c                 ,    |                      d          S )Nobjectr]   rl   s    r    rO   z-test_apply_with_byte_string.<locals>.<lambda>  s     2 2 r"   )r   r   r   r  r<   r=   rT   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"   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  cdert   r\   r   c                     | k    S rJ   rK   )rM   r  s    r    rO   z/test_apply_category_equalness.<locals>.<lambda>  s    !s( r"   c                 \    g | ](}t          j        |          rt          j        n|k    )S rK   )r   isnullr   r   )r   rM   r  s     r    
<listcomp>z1test_apply_category_equalness.<locals>.<listcomp>  s0    AAA!29Q<<	-Q#XAAAr"   rc  )r   r   r   rt   r<   r   r=   r>   )r  	df_valuesr   rC   rD   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 )Nra   )r   ra   r   rn   )r   ra   )r   r   rB  rC  r<   fftrb   rfftrd   s     r    test_infer_row_shaper    s     
29((++227;;	<	<BXXbfjqX))/FWXXbfkX**0FVr"   zops, by_row, expectedrt   c                     | dz   S r   rK   rl   s    r    rO   rO     
    Q r"   compatc                     | dz   S r   rK   rl   s    r    rO   rO     r  r"   c                 *    |                                  S rJ   r   rl   s    r    rO   rO          r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO     r  r"   r   c                 *    |                                  S rJ   r  rl   s    r    rO   rO         AEEGG r"   )r   r   r   )r   r   <lambda>r   c                 *    |                                  S rJ   r  rl   s    r    rO   rO     r  r"   c                     dS r   rK   rl   s    r    rO   rO          r"   c                     dS r   rK   rl   s    r    rO   rO     r  r"   c                     t          dddgi          }|                    | |          }t          j        ||           d S Nrt   r   ra   by_rowr   r<   r=   assert_equalopsr  rD   r   rC   s        r    test_dictlike_lambdar    sF    . 
C!Q=	!	!BXXc&X))FOFH%%%%%r"   r  c                     | dz   S r   rK   rl   s    r    rO   rO     s
    A r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO     s     r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO     s     r"   c                     dS r   rK   rl   s    r    rO   rO     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 Nrt   r   ra   zby_row=True not allowedmatchTr  r   r)   raises
ValueErrorr<   r  r   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   ra   rt   r   r   c                 (    d| d         | d         z   iS Nr   rt   r   rK   rl   s    r    rO   z,test_with_dictlike_columns.<locals>.<lambda>      afqvo 6 r"   rn   c                     g | ]}d diS )r   r   rK   r   rt  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  rK   rl   s    r    rO   z,test_with_dictlike_columns.<locals>.<lambda>
  r  r"   c                 
    d| iS )Nr   rK   rl   s    r    rO   z,test_with_dictlike_columns.<locals>.<lambda>  s
    #q r"   r   r   )r   r<   r   rv  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 rJ   rK   rl   s    r    rO   z:test_with_dictlike_columns_with_datetime.<locals>.<lambda>       r"   r   rn   )r   r   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   ra   rt   r   r   c                 (    d| d         | d         z   iS r  rK   rl   s    r    rO   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  rK   rl   s    r    rO   z7test_with_dictlike_columns_with_infer.<locals>.<lambda>,  r  r"   )r   r<   r=   rT   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   rK   rl   s    r    rO   rO   3  
    AE r"   )rt   r  c                     | dz   S r   rK   rl   s    r    rO   rO   4  r  r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO   5      AEEGG r"   r  c                 *    |                                  S rJ   r  rl   s    r    rO   rO   6  r  r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO   8      aeegg r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO   =  r  r"   c                     | dz   S r   rK   rl   s    r    rO   rO   B  s
    q1u r"   c                     dS Nr   rK   rl   s    r    rO   rO   B  s     r"   r   c                     dS r   rK   rl   s    r    rO   rO   G  r	  r"   c                     dS r  rK   rl   s    r    rO   rO   G  rv   r"   c                     t          dddgi          }|                    | |          }t          j        ||           d S r  r  r  s        r    test_listlike_lambdar#  0  sF    > 
C!Q=	!	!BXXc&X))FOFH%%%%%r"   c                     | dz   S r   rK   rl   s    r    rO   rO   W  
    1q5 r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO   X  s    15577 r"   c                 *    |                                  S rJ   r  rl   s    r    rO   rO   Y  s    !%%'' r"   c                     | dz   S r   rK   rl   s    r    rO   rO   Z  r%  r"   c                     dS r  rK   rl   s    r    rO   rO   Z  rv   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+  T  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 )Nra   r`   )rt   r   ofwordsz
2016-10-01h)r~   freq)rt   r   r_   rt   r   r   rn   c                 "    g | ]}|d d         S rr   rK   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>o  s     BBBquBBBr"   r_   c                 "    g | ]}|d d         S rr   rK   r  s     r    r  z.test_with_listlike_columns.<locals>.<listcomp>s  s     CCCquCCCr"   )r   r   r   rB  rC  rD  r
   r<   tuplerv  r=   r>   r   s      r    test_with_listlike_columnsr4  d  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 )Nrt   r   qzrt  rL   )r4  j0)r5  j1c                 ,      fd d         D             S )Nc                 (    g | ]}|d          v |S )rN   rK   )r   elr  s     r    r  zOtest_with_listlike_columns_returning_list.<locals>.<lambda>.<locals>.<listcomp>|  s"    "K"K"K"B#c(NN2NNNr"   rM   rK   r  s   `r    rO   z;test_with_listlike_columns_returning_list.<locals>.<lambda>|  s     "K"K"K"KC"K"K"K r"   r   rn   r   )r   r   r   from_tuplesr@   r<   r=   r>   r   s      r    )test_with_listlike_columns_returning_listr>  w  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   rW  z2017-11-29 03:30:00z2017-11-29 03:45:00)numberstringr   c                     | j         | j        fS rJ   r@  rA  r  s    r    rO   z1test_infer_output_shape_columns.<locals>.<lambda>  s    3:sz": r"   r   rn   c                 *    g | ]}|j         |j        fS rK   rC  r  s     r    r  z3test_infer_output_shape_columns.<locals>.<listcomp>  s!    EEE!(+EEEr"   r   r	   r<   r   rv  r=   r>   r   s      r    test_infer_output_shape_columnsrF    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 )Nra   )r   r   r   r   c                 
    g dS r   rK   rl   s    r    rO   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  r   r"   r   rn   c                     g | ]}g d S )r   rK   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    :::Qyyy:::r"   c                 
    ddgS Nr   ra   rK   rl   s    r    rO   z:test_infer_output_shape_listlike_columns.<locals>.<lambda>  
    A r"   c                     g | ]}d dgS r   ra   rK   r  s     r    r  z<test_infer_output_shape_listlike_columns.<locals>.<listcomp>  s    777!1v777r"   
r   r   rB  rC  rD  r<   r   rv  r=   r>   r   s      r    (test_infer_output_shape_listlike_columnsrP    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 )	Nrt   r   r   r   c                 ,    t          j                  S rJ   r   ones)r  r  s    r    rO   zBtest_infer_output_shape_listlike_columns_np_func.<locals>.<lambda>  s    "'#,, r"   r   rn   c                 8    g | ]}t          j                  S rK   rS  )r   rt  r  s     r    r  zDtest_infer_output_shape_listlike_columns_np_func.<locals>.<listcomp>  s!    ===rws||===r"   )r   r   r<   r   rv  r@   r=   r>   )r  r   rC   rD   s   `   r    0test_infer_output_shape_listlike_columns_np_funcrV    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`   ra   )r   ra   r   r`   )rt   r   r   rE   c                     dS )NrN  rK   rl   s    r    funzDtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.fun  s    vr"   r   rn   c                     g | ]}d S )rN  rK   r  s     r    r  zKtest_infer_output_shape_listlike_columns_with_timestamp.<locals>.<listcomp>  s    777!v777r"   rE  )r   rY  rC   rD   s       r    7test_infer_output_shape_listlike_columns_with_timestampr[    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 )	Nra   )r`   r   r   r   c                     S rJ   rK   )rM   r\  s    r    rO   z3test_consistent_coerce_for_shapes.<locals>.<lambda>  s     r"   r   rn   c                     g | ]}S rK   rK   )r   rt  r\  s     r    r  z5test_consistent_coerce_for_shapes.<locals>.<listcomp>  s    444qs444r"   rO  )r\  r   rC   rD   s   `   r    !test_consistent_coerce_for_shapesr`    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colsr   r   rl   s    r    rO   z'test_consistent_names.<locals>.<lambda>  s    &*C*C*CDDD r"   r   rn   rc  rd  re  r   r   c                 ,    t          ddgddg          S )Nr   ra   rc  rd  r   rf  rl   s    r    rO   z'test_consistent_names.<locals>.<lambda>  s    1vfg5F G G G r"   )r<   renamer=   rT   r   s       r    test_consistent_namesri    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   rK   rl   s    r    rO   z"test_result_type.<locals>.<lambda>  r   r"   r   r   r   r   r   ra   r<   r   r   r=   rT   r   s       r    test_result_typern    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 rK  rK   rl   s    r    rO   z/test_result_type_shorter_list.<locals>.<lambda>  rL  r"   r   r   r   r   r   r   rm  r   s       r    test_result_type_shorter_listrq    sb     
BXX&&QHXEEF3*~""$$H1vH&(+++++r"   c                    |dk    r:t           j                            d          }|j                            |           | }|                    d dd|          }|                                }t          j        ||           d S )Nr$   z(numba engine doesn't support list returnr.   c                 
    g dS r   rK   rl   s    r    rO   z,test_result_type_broadcast.<locals>.<lambda>  s
    ))) r"   r   r   r5   r   r,   	r)   r6   r7   r8   r9   r<   r   r=   rT   )r!   r+   r,   r6   r   rC   rD   s          r    test_result_type_broadcastrv    s     {  (R SS%%%	BXX!V   F wwyyH&(+++++r"   c                    |dk    r:t           j                            d          }|j                            |           | }g d|                    fddd|          }|                                }t          j        ||           d S )	Nr$   <numba Series constructor only support ndarrays not list datar.   rd  r  r  c                 *    t          g d          S Nr   r   rf  rM   r   s    r    rO   z8test_result_type_broadcast_series_func.<locals>.<lambda>  s    &'222 r"   r   r   rt  ru  r!   r,   r+   r6   r   rC   rD   r   s          @r    &test_result_type_broadcast_series_funcr~    s     {  Q ! 
 
 	%%%	B'''GXX2222	   F wwyyH&(+++++r"   c                    |dk    r:t           j                            d          }|j                            |           | }|                    d d|          }|                                }t          j        ||           d S )Nr$   rx  r.   c                 2    t          g d| j                  S r{  )r   r@   rl   s    r    rO   z0test_result_type_series_result.<locals>.<lambda>%  s    yyy @ @ @ r"   r   r4   ru  )r!   r,   r+   r6   r   rC   rD   s          r    test_result_type_series_resultr    s     {  Q ! 
 
 	%%%	BXX@@qQWXXXFwwyyH&(+++++r"   c                 *   |dk    r:t           j                            d          }|j                            |           | }g d|                    fdd|          }|                                }|_        t          j	        ||           d S )Nr$   z:no support in numba Series constructor for list of columnsr.   ry  c                 *    t          g d          S r{  rf  r|  s    r    rO   z<test_result_type_series_result_other_index.<locals>.<lambda>6  s    yyy @ @ @ r"   r   r4   )
r)   r6   r7   r8   r9   r<   r   r   r=   rT   r}  s          @r    *test_result_type_series_result_other_indexr  *  s     {  O ! 
 
 	%%%	B'''GXX@@@@qQWXXXFwwyyHH&(+++++r"   boxc                      t          |           S rJ   )r   rl   s    r    rO   rO   >  s    tAww r"   c                      t          |           S rJ   )r3  rl   s    r    rO   rO   >  s    %(( r"   c                 .    t          j        | d          S )Nr   r   )r   r   rl   s    r    rO   rO   >  s    bhq6P6P6P r"   )r   r3  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 rK  rK   rM   r  s    r    rO   z,test_consistency_for_boxed.<locals>.<lambda>E      QF r"   r   rn   c                 *    g | ]} d dg          S rN  rK   )r   rt  r  s     r    r  z.test_consistency_for_boxed.<locals>.<listcomp>F  s%    <<<qssAq6{{<<<r"   c                      ddg          S rK  rK   r  s    r    rO   z,test_consistency_for_boxed.<locals>.<lambda>I  r  r"   r   r   r   r   r   r   r   )r<   r   rv  r=   r>   rh  rT   )r  r!   r   rC   rD   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   r0   r1   rn   r;   absolute)r   r:   absr;   r   r<   r=   rT   r   from_productr   r@   r   )r5   rB   
other_axisf_absf_sqrtrD   rC   s          r    test_agg_transformr  N  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   r|  r}  r   r`   r   r   r6  )r   r   rx   r=   rT   r   s      r    	test_demor  s  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   r|  r}  r         @r        @g      9@r   r   )r}  r|  r   r6  )r   r   rx   r   r   r=   rT   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 )Nrd  r   rW  countra   rc  )r   rx   r   r=   r>   )ry   r   rC   rD   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   rW  bazr   r   r   r         @rW  	foobarbazr|  r   r   )r   r   r   )r   rx   r=   rT   reindex)mdfrD   rC   s      r    test_agg_multiple_mixedr    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  r|   r   r}   r   r   r   rW   does not support reductionr  r|  r   )rW   r   r   r   )r   r
   r)   r  	TypeErrorrx   )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   ra   rn   )r?   r}  r   r?   r   r   >   r   r   r}  )axesuniquesort_valuesr   concatr?   r}  r   r   r   rx   r=   rT   r   r   r>   r   )r5   rB   r  name1name2rD   rC   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                              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  r|   r   r}   r  r|  r   r   rW  r   nsro  r  r  r   r   )r   r  r  )r   r
   rx   r   r	   r   r=   r>   as_unitrT   r)   r  r  )r   rC   rD   r  s       r    test_nuiscance_columnsr    sO   	 &&&J222		
 	

 
B VVE]]Fq#ui
&;&;<BJOOOH68,,,VVUG__F
S%:..66t<<	=>g
  H
 &(+++
&C	y	,	,	,  
u                $$U++F+++???CCCH68,,,
&C	y	,	,	,  
w                 s$    D""D&)D&F55F9<F9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Nra   r   r   r  r   NrW  r   r   r  ra   sizer   )r  r  Tr~  )	r   r   r   r   r   r=   r>   rT   r  )r  r   rC   rD   result1result2s         r    test_non_callable_aggregatesr  3  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  rn   )r   r@   r   r   r   )
r   r   r   r   r   rb   r   r@   r=   r>   )r  r5   r   rC   rD   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)ra   ra   r   g      ?r  r   c                 h    t          |                                                                           S rJ   )r   dropnar  )	group_cols    r    r   z&test_agg_listlike_result.<locals>.funct  s(    I$$&&--//000r"   ra   r   r   rW  r   r   )
r   r   r   rx   r   r=   r>   to_framer   rT   )r   r   rC   rD   s       r    test_agg_listlike_resultr  p  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      rN  rP   r   rK   rt   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 rJ   r  )rM   rt   r   r   s       r    rS  ztest_agg_args_kwargs.<locals>.f  s    uuww!a%1$$r"   r   ra   r`   r   r  g      @r  g       @)r   r   r   rx   r=   r>   )r5   rQ   kwargsrS  r   rD   rC   s          r    test_agg_args_kwargsr    s    % % % % 
QFQF#	$	$Bqyy3*%%3*%%RVAt-d---f--F68,,,,,r"   num_cols)ra   r   r   c                 l   |dk    r:t           j                            d          }|j                            |           t          j        g d          }|                                }t          d|t          |                     }|
                    d |           |j        |j        k    sJ d S )	Nr$   z*numba engine only supports numeric indicesr.   )z
1950-06-30z
1952-10-24z
1953-05-29r   ro  c                     | S rJ   rK   rl   s    r    rO   z,test_frequency_is_original.<locals>.<lambda>  r	  r"   r3   )r)   r6   r7   r8   r9   r   DatetimeIndexr   r   r   r<   r0  )r  r,   r+   r6   r@   originalr   s          r    test_frequency_is_originalr    s     {  (T UU%%%GGGHHEzz||H	1E5??	;	;	;BHH[[H(((:&&&&&&r"   c                    | dk    r:t           j                            d          }|j                            |           t          dd          t          dd          t          dd          g}t          g d	|
          }|                    d d|           }t          ||          }t          j
        ||           d S )Nr$   z0numba engine doesn't support non-numeric indexesr.   z2019-03-15 12:34:31.909000+0000rh   ri   z2019-03-15 12:34:34.359000+0000z2019-03-15 12:34:34.660000+0000rl  )ry   r@   c                     | j         S rJ   rc  rl   s    r    rO   z.test_apply_datetime_tz_issue.<locals>.<lambda>  s     r"   r   r4   )r@   ry   )r)   r6   r7   r8   r9   r	   r   r<   r   r=   r>   )r,   r+   r6   
timestampsr   rC   rD   s          r    test_apply_datetime_tz_issuer    s     {  E ! 
 
 	%%% 	3>>>3>>>3>>>J
 
				4	4	4BXX&&QvX>>FJZ888H68,,,,,r"   r   r   rE   r   method)r|  r}  r   c                     |dk    rd}nd}|sNt          j        t          |          5   t          | |                       d d d            d S # 1 swxY w Y   d S  t          | |                       d S )Nr   z>can only concatenate str \(not "int"\) to str|does not supportz4not supported between instances of 'str' and 'float'r  )r)   r  r  r   )r   r  using_infer_stringr  s       r    test_mixed_column_raisesr    s     OD ]9C000 	" 	"GB!!!	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	" 	Fs   AAAr  r   c                     t          d| ggddg          }|                    d           }|j        }t          j        ||           d S )Nr   rt   r   r   c                     | j         S rJ   r   rl   s    r    rO   z"test_apply_dtype.<locals>.<lambda>  s     r"   )r   r<   dtypesr=   r>   )r  r   rC   rD   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<   t          j        |          5  |                    |d          }d d d            n# 1 swxY w Y   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 )Nrt   r   )_mgrr   )r  mgrs     r    r   z!test_apply_mutating.<locals>.func  s9    hxs""""
r"   rt   r   rn   )r   r   r   r=   assert_cow_warningr<   rT   )using_array_managerusing_copy_on_writewarn_copy_on_writer   df_origr   rD   rC   s           r    test_apply_mutatingr    s;   	s%S//::	;	;BggiiG   wwyyHSMMMQMMM		1	2	2 ( ($Q''( ( ( ( ( ( ( ( ( ( ( ( ( ( ( &(+++ *1 * 	b'*****
b&)))))s   =B!!B%(B%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   ra   r   r`   r   r   r   r   r   r   rt   r   r   c                     g S rJ   rK   rl   s    r    rO   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                 N   | dk    r:t           j                            d          }|j                            |           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$   z;numba engine doesn't support list-likes/dict-like callablesr.   r`   r   r   r   r   r   r   c                 *    |                                  S rJ   r  rl   s    r    rO   z,test_apply_no_suffix_index.<locals>.<lambda>  s     r"   c                 *    |                                  S rJ   r  rl   s    r    rO   z,test_apply_no_suffix_index.<locals>.<lambda>  s    AEEGG r"   r3   )r  r  r  )   r  r  r   )r   r  r  r   )	r)   r6   r7   r8   r9   r   r<   r=   rT   )r,   r+   r6   pdfrC   rD   s         r    test_apply_no_suffix_indexr    s    {  P ! 
 
 	%%%
aVHqL3*
5
5
5CYY002C2CDVYTTFll..6U6U6U  H &(+++++r"   c                     | dk    rt          j        d           t          dddgi          }|                    d | dd	          }t	          ddg          }t          j        ||           d S )
Nr$   z No object dtype support in numbar   aabbbc                     | d         S r   rK   rl   s    r    rO   z/test_apply_raw_returns_string.<locals>.<lambda>  r   r"   r   T)r,   r5   rG   )r)   rS   r   r<   r   r=   r>   r,   r   rC   rD   s       r    test_apply_raw_returns_stringr    sy    6777	C$'	(	(BXXnnV!XFFF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   )ra   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  r|  )g      5@g      %@r  r   )g      2@g      "@r  r  )g      1@g      !@r  r  )r   r  r  r|  r   rw   )r   r  aggsrC   rD   s        r    "test_aggregation_func_column_orderr
    s     
	
 	
 	
 )

 

 

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

 211  H &(+++++r"   c                 @   | dk    r:t           j                            d          }|j                            |           t          g dg dd          }|ddg                             d d	| 
          }t          g d          }t          j	        ||           d S )Nr$   z2numba engine not supporting duplicate index valuesr.   rl  r   r  rt   c                 8    | j         d         | j         d         z   S )Nr   r   )r0  rl   s    r    rO   z+test_apply_getitem_axis_1.<locals>.<lambda>>  s    !&)afQi' r"   r   r4   )r   ra   r`   )
r)   r6   r7   r8   r9   r   r<   r   r=   r>   )r,   r+   r6   r   rC   rD   s         r    test_apply_getitem_axis_1r  5  s    {  G ! 
 
 	%%%	33	4	4Bc
^!!''a "  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!)warningswarnr   rl   s    r    expected_warningzDtest_nuisance_depr_passes_through_warnings.<locals>.expected_warningL  s    o&&&uuwwr"   rt   r   r  r  )r   r=   assert_produces_warningUserWarningrx   )r  r   s     r    *test_nuisance_depr_passes_through_warningsr  D  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   rA  g      ?i  r   )ry  rz  r   r   r   rn   )
r   r   r   r   r   r<   typer   r=   r>   r   s      r    test_apply_typer  U  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   ra   r   r   r  c                 :    t          | d         | d                   S )Nrt   r   )r}  rl   s    r    rO   z/test_apply_on_empty_dataframe.<locals>.<lambda>j  s    AcFAcF(;(; r"   r4   r   )r   headr<   r   r   r   r=   r>   r  s       r    test_apply_on_empty_dataframer  g  st    	!Qq!f--	.	.BWWQZZ;;!FSSF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   ra   r   r  c                     | j         gS rJ   )r   rl   s    r    rO   z(test_apply_return_list.<locals>.<lambda>q  s
    
 r"   r   r   s      r    test_apply_return_listr   o  sn    	!Qq!f--	.	.BXX**++FAx1vh7788H&(+++++r"   ztest, constant)r   r   r   )ra   ra   ra   c                     t          |           }t          |ddgd          }ddd}|                    |          }t          j        ||           d S )Nrt   r   r  )ry   r@   r   r  r  r  )rc  constantdf1rD   aggregationrC   s         r    test_unique_agg_type_is_seriesr%  v  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   ra   r   )r   ra   r   )r   r   r   r   )TTFr   rn   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 )NrM   r   r   r   c                     | |z   |z   S rJ   rK   )rM   rt   r   s      r    foo1z.test_agg_mapping_func_deprecated.<locals>.foo1      1uqyr"   ra   c                     | |z   |z   S rJ   rK   )rM   r   r   s      r    foo2z.test_agg_mapping_func_deprecated.<locals>.foo2  r+  r"   r   r`   )r   r   z+using .+ in Series.agg cannot aggregate andr  r   r   r   r*  r-  r   )ra   r   r`   )r   r   )ra   r   )r   rx   r=   rT   r  FutureWarning)r   r*  r-  rC   rD   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   ra   r   r   r   zusing DataFrame.stdr  r+  r   r   zusing Series.stdstdr   )r   r   r   r   r=   r  r.  rx   r1  r   r   r>   rT   r   s      r    test_agg_stdr2    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  r   )r   r   r   r   r  )r   r   r   r   rx   r  r=   r>   r   s      r    (test_agg_dist_like_and_nonunique_columnsr4    s~    	ll#rvs!3:N:N:NOO
 
B !BJVVS'N##F#w}}H68,,,,,r"   )r   r  numpyr   r)   pandas.compatr   pandas.core.dtypes.dtypesr   pandasr   r   r   r   r	   r
   pandas._testing_testingr=   pandas.tests.frame.commonr   pandas.util.versionr   fixturer!   r(   r6   
single_cpur,   rF   parametrizerU   re   rq   rz   r   r;   r?   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  filterwarningsr  r  r  r$  r,  r.  r2  r8  rF  rU  r`  rf  ri  rp  rw  r  r  r  r  r  r  r  r  r  r  r   r  r  r   r  r  r   r  r  r#  r+  r4  r>  rF  rP  rV  r[  r`  ri  rn  rq  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/  r2  r4  rK   r"   r    <module>rA     s               ) ) ) ) ) ) 6 6 6 6 6 6                        0 0 0 0 0 0 ' ' ' ' ' ' 
 
 
 ,&,wfk>T"U"U"UVWWW  XW1 1 1, !Q((u.., , /. )(,< < <, , , $$aS))*YY|-N-N-NPT,UV , ,	 ,
, 
, 
, "'27!344  54	- 	- 	-, , ,  , !>!>!>??- - @?-
- 
- 
-- - -, , ,, , ,, , ,"
, 
, 
,& & &, , , !Q((G G )(G !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 ;<<- - =<-* * *6- - -, , , - - -, , ,>- - -# # #"- - -$- - -, , , yyyyy	)	)!+E+EFyyyyy	)	)!A3+?+?@ - - -
- 
- 
-, , ,>, , ,	- 	- 	- 	- 	-r"   