a
    j=ic                    @   sP  d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dlZ	d dl
Z
d dlZd dlm  mZ d dlZd dlmZmZmZmZ d dlmZ d dlm  mZ d dlmZ d dl m!Z!m"Z" d dl#m$Z$m%Z% e
j&d	d
d dgddgddd Z'G dd dZ(G dd dZ)G dd dZ*G dd dZ+G dd dZ,dd Z-dd Z.e
j/0dd d! Z1G d"d# d#Z2d$d% Z3d&d' Z4d(d) Z5d*d+ Z6e
j/7d,ed-d.gd/d0gge8d1d2d3fed-d4gd/d4gge8d1d2d5fgd6d7 Z9d8d9 Z:e
j/7d:ed;d;gged;d;ged;d;gggd<d= Z;d>d? Z<d@dA Z=dBdC Z>dDdE Z?dFdG Z@dHdI ZAdS )J    )deque)datetime)EnumN)	DataFrameIndex
MultiIndexSeries)expressions)_MIN_ELEMENTSNUMEXPR_INSTALLED)_check_mixed_float_check_mixed_intTmodulei@B Znumexprpython)Zautousescopeparamsidsc                 c   s    t j}| jt _| jV  |t _d S N)exprr
   param)requestr
    r   s/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/frame/test_arithmetic.pyswitch_numexpr_min_elements"   s    r   c                   @   sZ   e Zd ZddddZdd ZedddZedd	d
ZdddZdd Z	dddZ
dS )DummyElementN)returnc                 C   s   || _ t|| _d S r   )valuenpdtype)selfr   r   r   r   r   __init__-   s    zDummyElement.__init__c                 C   s   t j| j| jdS )Nr   )r   arrayr   r   r   r   r   r   	__array__1   s    zDummyElement.__array__c                 C   s   d| j  d| j dS )NzDummyElement(z, ))r   r   r#   r   r   r   __str__4   s    zDummyElement.__str__c                 C   s   t | S r   )strr#   r   r   r   __repr__7   s    zDummyElement.__repr__Fc                 C   s
   || _ | S r   r!   )r   r   copyr   r   r   astype:   s    zDummyElement.astypec                 C   s   t | | j||S r   )typer   view)r   r   r   r   r   r,   >   s    zDummyElement.viewc                 C   s
   t | jS r   )boolr   )r   axisr   r   r   anyA   s    zDummyElement.any)F)N)__name__
__module____qualname__r    r$   r'   r&   r(   r*   r,   r/   r   r   r   r   r   ,   s   
r   c                   @   sf  e Zd Zdd Zdd Zejdej	j
dddejddd	d
ej	j
dddej	j
dddd
gej	j
dddej	j
dddd
ej	j
dddejddd	d
gejddd	ejddd	d
ej	j
dddej	j
dddd
gej	j
dddejddd	d
ejddd	ejddd	d
ggdd Zejdg ddd Zdd Zdd Zdd Zdd ZdS )TestFrameComparisonsc                 C   s^   t dg di}t dg di}|dk}t|| |d d|d< |dk}t|| d S )NA)foobarbaz)TFFr5   category)r   tmassert_frame_equalr*   )r   dfexpresr   r   r   &test_comparison_with_categorical_dtypeL   s    z;TestFrameComparisons.test_comparison_with_categorical_dtypec                 C   sX   t tjddtdd}d}tjt|d |dv  W d    n1 sJ0    Y  d S )N      ABCDcolumnsz+The truth value of a DataFrame is ambiguousmatchr   )r   r   randomrandnlistpytestraises
ValueError)r   r;   msgr   r   r   test_frame_in_list[   s    z'TestFrameComparisons.test_frame_in_listz	arg, arg2
   size20010101periodsabc                    sr  t | t | k}t  fdd jD  j jd}t||  k}t  fdd jD  j jd}t|| g d}d|}tjt|d  k W d    n1 s0    Y  tjt|d  k W d    n1 s0    Y  tjt|d  k  W d    n1 s,0    Y  tjt|d  k W d    n1 sd0    Y  d S )Nc                    s   i | ]}| | | kqS r   r   .0colxyr   r   
<dictcomp>       z@TestFrameComparisons.test_comparison_invalid.<locals>.<dictcomp>indexrC   c                    s   i | ]}| | | kqS r   r   rW   rZ   r   r   r]      r^   )z=Invalid comparison between dtype=datetime64\[ns\] and ndarrayzinvalid type promotionzbThe DTypes <class 'numpy.dtype\[.*\]'> and <class 'numpy.dtype\[.*\]'> do not have a common DType.|rD   )	r   rC   r`   r9   r:   joinrI   rJ   	TypeError)r   argZarg2resultexpectedZmsgsrL   r   rZ   r   test_comparison_invalidb   s4    0	
&&(z,TestFrameComparisons.test_comparison_invalidzleft, right))gtlt)ri   rh   )gele)rk   rj   )eqrl   )nerm   c           	   	   C   s  t tjdddtjdddtjjdddtjdtt	dd}tj
|jtjt|dkd	f< tt|}tt|}|d
v r||td}|td|}t|| nd}tjt|d  ||td W d    n1 s0    Y  tjt|d  |td| W d    n1 s(0    Y  |d
v rj||td}|td|}t|| nd}tjt|d  ||td W d    n1 s0    Y  tjt|d  |td| W d    n1 s0    Y  d S )NrQ   rN   rR   Z20010102i ʚ;rO   )Zdates1dates2ZintcolZfloatcolZ	stringcolg      ?rn   )rl   rm   Z20010109zL'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'Timestamp'rD   ZnatzJ'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'NaTType')r   pd
date_ranger   rF   randintrG   rH   r9   ZrandsNaTlocrandlengetattroperator	Timestampr:   rI   rJ   rc   )	r   leftrightr;   Zleft_fZright_frf   re   rL   r   r   r   test_timestamp_compare   s>    
	 

.0
0z+TestFrameComparisons.test_timestamp_comparec                 C   s\   t ddgddgg}t ddgddgg}||k}|  r@J ||k}|  sXJ d S )Nz
1989-08-01      rU   rV   cd)r   r/   all)r   r;   otherre   r   r   r   test_mixed_comparison   s    z*TestFrameComparisons.test_mixed_comparisonc                 C   s^   t tdd}t ddgddgddgg}|dk}t|| |ddgk}t|| d S )Nr?      r}   FTr}   r}   r}   )r   r   arangereshaper9   r:   )r   r;   rf   re   r   r   r    test_df_boolean_comparison_error   s    z5TestFrameComparisons.test_df_boolean_comparison_errorc                 C   s>   t tjddtdg dd}|d }|  r:J d S )N   r   r4   BCr_   )r   r   rF   rG   range__eq__r/   r   r;   re   r   r   r   test_df_float_none_comparison  s     
z2TestFrameComparisons.test_df_float_none_comparisonc                 C   s   t ddddddg}|jdk}t|| |jddd d f  t||  |jddd d f  |jdk}t|| |jddd d f  t||  |jddd d f  d S )Nr|   r5   rT   r}   r6   r   )r   rU   r9   r:   rs   rV   )r   r;   Zmask_aZmask_br   r   r   test_df_string_comparison  s    
"$
"z.TestFrameComparisons.test_df_string_comparisonN)r0   r1   r2   r>   rM   rI   markparametrizer   rF   rq   ro   rp   rg   r{   r   r   r   r   r   r   r   r   r3   I   sX   -
*
-r3   c                   @   s   e Zd Zejdg ddd Zejdeje	gdd Z
dd	 Zd
d Zdd Zdd Zejdg ddd Zejdg ddd Zdd ZdS )TestFrameFlexComparisonsop)rl   rm   rh   ri   rj   rk   c                 C   s^  t jdd}t jdd}t|}t|}t |jd }||j sPJ |	|j
 rdJ t||}tt|}t||||| |jdd dd f  }	||	}
|||	j|j|jd}t|
| t||j|||j t|d||d d}t|t j||t j tjt|d || W d    n1 sP0    Y  d S )	N   r   )r|   r   r|   r_   r   $Unable to coerce to Series/DataFramerD   )r   rF   rG   r   onesshaperl   valuesr   rm   r/   rv   rw   r9   r:   rs   r)   reindexr`   rC   nanrI   rJ   rK   )r   r   dataZ
other_datar;   r   ndim_5foZpart_orsZxprL   r   r   r   test_bool_flex_frame  s(    

z-TestFrameFlexComparisons.test_bool_flex_frameboxc                 C   s  t jdd}t|}|t jd}|t jd}|j|dd}||}|j|dd}||}	t||t|k t||	  t||  t||j	|j	 t||t
| t||jt|dd t||jt
|dd |j|dd}
||}|j|dd}||}t||t|k t||  t|
|  t|
|j	|j	 |j|dd}||}|j|dd}||}t||t|k t||  t||  t||j	|j	 tt jd}tt jd}d S )Nr   r   r   r.   )r   rF   rG   r   rl   rm   r9   r:   r   TrH   rh   rk   rj   ri   )r   r   r   r;   Zidx_serZcol_serZidx_eqZcol_eqZidx_neZcol_neZidx_gtZcol_gtZidx_leZcol_leZidx_geZcol_geZidx_ltZcol_ltr   r   r   test_bool_flex_series=  sB    





z.TestFrameFlexComparisons.test_bool_flex_seriesc                 C   s   t tjdd}tj|jd< ||}|jd r6J ||}|jd sNJ ||}|jd rfJ |	|}|jd r~J |
|}|jd rJ ||}|jd rJ d S )Nr   r   r   r   )r   r   rF   rG   r   rs   rl   rm   rh   ri   rj   rk   )r   r;   r   r   r   r   test_bool_flex_frame_nah  s    





z0TestFrameFlexComparisons.test_bool_flex_frame_nac           	      C   s  t t jddt jg}t dt jdd g}td|i}td|i}dddg}tjt|d	 || W d    n1 s~0    Y  tjt|d	" |d |d  W d    n1 s0    Y  tjt|d	 |j	|j	k W d    n1 s0    Y  |
|}|j	 sJ t dt jd g}td|i}tjt|d	 |d W d    n1 sj0    Y  tjt|d	 |d d W d    n1 s0    Y  tjt|d	 |j	dk W d    n1 s0    Y  d S )
Nr|   r?   y               @   rU   ra   z9'>' not supported between instances of '.*' and 'complex'z unorderable types: .*complex\(\)rD   )r   r"   r   r   rb   rI   rJ   rc   rh   r   rm   r   )	r   arrZarr2r;   df2rL   r   Zarr3df3r   r   r   "test_bool_flex_frame_complex_dtypey  s2    (0*
*.z;TestFrameFlexComparisons.test_bool_flex_frame_complex_dtypec                 C   sT   t ddtjdgi}t ddt dgi}||}t dg di}t|| d S )NrY   r5   r6   )FTF)r   r   r   r   nowrm   r9   r:   )r   df1r   re   r<   r   r   r   !test_bool_flex_frame_object_dtype  s
    
z:TestFrameFlexComparisons.test_bool_flex_frame_object_dtypec                 C   s   t tjg}|tjk}|jd  du s,J |tj}|jd  du sNJ |tjk}|jd  du snJ |tj}|jd  du sJ d S )Nr   FT)r   ro   rr   ilocitemrl   rm   r   r   r   r   test_flex_comparison_nat  s    

z1TestFrameFlexComparisons.test_flex_comparison_natopnamec                 C   sP   t g dg dd}d}t|||j }t|tdgtt	gd d S )Nr|   r}   r         ?       @      @rZ   r}   r`   )
r   rv   dtypesvalue_countsr9   assert_series_equalr   r   r   r-   )r   r   r;   constre   r   r   r   &test_df_flex_cmp_constant_return_types  s    z?TestFrameFlexComparisons.test_df_flex_cmp_constant_return_typesc                 C   s^   t g dg dd}d}|jd d }t|||j }t|tdgt	t
gd d S )Nr   r   rZ   r}   r   r   )r   r   rv   r   r   r9   r   r   r   r   r-   )r   r   r;   r   emptyre   r   r   r   ,test_df_flex_cmp_constant_return_types_empty  s
    zETestFrameFlexComparisons.test_df_flex_cmp_constant_return_types_emptyc                 C   s   t jg d}t||d}tddg}|j|dd}tddgddgd}t|| tddgdd	gd
}|j|dd}t|| d S )Nr   r4   r   r   r   Fr|   r}   r4   r   r   )ro   ZIntervalIndexZfrom_breaksr   r   rl   r9   r:   )r   iir;   serr=   rf   Zser2res2r   r   r   -test_df_flex_cmp_ea_dtype_with_ndarray_series  s    zFTestFrameFlexComparisons.test_df_flex_cmp_ea_dtype_with_ndarray_seriesN)r0   r1   r2   rI   r   r   r   r   r"   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s   

*'

	r   c                   @   s>  e Zd Zdd Zejje ddejdddgdd	 Z	d
d Z
dd Zdd Zejdg ddd Zejdedddd Zdd Zejdg ddd Zdd Zejd d!d"gd#d$ Zd%d& Zd'd( Zd)d* Zejdg d+d,d- Zejd.d/d0gd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd0S );TestFrameFlexArithmeticc                    st   t d}t|td  jdd}t fdd jD }t||  jjdd}t|| d S )Nr   r   r   r   c                    s   i | ]}| |  qS r   r   rW   r;   r   r   r   r]     r^   z?TestFrameFlexArithmetic.test_floordiv_axis0.<locals>.<dictcomp>)	r   r   r   r   floordivrC   r9   r:   r   )r   r   re   rf   result2r   r   r   test_floordiv_axis0  s    
z+TestFrameFlexArithmetic.test_floordiv_axis0znumexpr not installed)reasonr   r   powc                    s   t t|ttd td d dd }t| d d<  d t  |dd}t fdd	 jD }t	|| t  |j
dd}t	|| d S )
Nd   r|   r   r   r   r   c                    s   i | ]}| | qS r   r   rW   r;   r   r   r   r   r]     r^   zLTestFrameFlexArithmetic.test_floordiv_axis0_numexpr_path.<locals>.<dictcomp>)rv   rw   r   r   r
   r   r   rC   r9   r:   r   )r   r   r   re   rf   r   r   r   r    test_floordiv_axis0_numexpr_path  s    
"z8TestFrameFlexArithmetic.test_floordiv_axis0_numexpr_pathc                 C   sf   t jddd}t jddd}t|}t||d}|j|dd}t|| || d}t|| d S )N
2016-01-01rN   rR   1r   r|   r   r   )ro   rp   Ztimedelta_ranger   r   addr9   r:   )r   dtiZtdiZtserr;   re   rf   r   r   r   test_df_add_td64_columnwise  s    z3TestFrameFlexArithmetic.test_df_add_td64_columnwisec                 C   s   t jddd}tg ddd}t||d}t||d}t jdd	 }|j||d
}ttg ddd|d d}t|| d S )Nr   r   rR   )z1 Dayrr   z2 Daystimedelta64[ns]r!   r   r|   )days
fill_value)z
2016-01-02z
2016-01-03z
2016-01-05zdatetime64[ns]r}   )	ro   rp   r   r   	TimedeltaZto_timedelta64r   r9   r:   )r   r   r   r;   r   fillre   rf   r   r   r   $test_df_add_flex_filled_mixed_dtypes  s    z<TestFrameFlexArithmetic.test_df_add_flex_filled_mixed_dtypesc                    s|   |  fdd}t | d| }||d| }t|| t | d| }||d| }t|| t|dd id d S )Nc                    s2     dr"tt dd|| S tt | |S )N__r__)
startswithrv   rw   replacerZ   r   r   r   r   '  s    
z8TestFrameFlexArithmetic.test_arith_flex_frame.<locals>.fr}   r   r!   )rv   r9   r:   r   )r   all_arithmetic_operatorsfloat_framemixed_float_framer   re   rf   r   r   r   test_arith_flex_frame!  s    z-TestFrameFlexArithmetic.test_arith_flex_framer   )__add____sub____mul__c           
      C   s   t t|}t ||d| }||d| }d }	|dv rBdd d}	n|dv rRdd i}	tjrr|dkrrd| d j|	d< t|| t||	d	 t ||d| }||d| }t|| t|dd id	 t ||d| }||d| }t|| d S )
Nr}   )r   Zuint64)r   r   )r   r   r   r   r4   r!   )	rv   rw   r   USE_NUMEXPRr   r9   r:   r   r   )
r   r   Z	int_framemixed_int_framer   r   r   re   rf   r   r   r   r   test_arith_flex_frame_mixed7  s&    	
z3TestFrameFlexArithmetic.test_arith_flex_frame_mixeddimr   r?   c                 C   sV   |}t d| }d}tjt|d t||| W d    n1 sH0    Y  d S )N)r|   r   rD   )r   r   rI   rJ   rK   rv   )r   r   r   r   r   r   rL   r   r   r   test_arith_flex_frame_raise`  s
    z3TestFrameFlexArithmetic.test_arith_flex_frame_raisec                 C   s   | d}t||d  | |d d }t||tj  |d d  |}t||tj  tjtdd$ |j |jd dd W d    n1 s0    Y  tjtdd& |j |jd ddd W d    n1 s0    Y  d S )	Nr|   r   r   rD   r   r   r`   r.   r   )	r   r9   r:   r   r   rI   rJ   NotImplementedErrorr   )r   r   Z	const_addre   r   r   r   test_arith_flex_frame_cornerk  s    
2z4TestFrameFlexArithmetic.test_arith_flex_frame_corner)r   submulmodc                 C   sb   |}| d}|d }t||}tt|}t||||| t||dd||j|j d S )NrU   twor   r   )xsrv   rw   r9   r:   r   )r   simple_framer   r;   rowrY   r   r   r   r   test_arith_flex_series_ops}  s    


z2TestFrameFlexArithmetic.test_arith_flex_series_opsc                 C   sh   |}| d}|d }t|j|d d||  t||||  t|j|dd|j| j d S )NrU   r   r   r   )r   r9   r:   r   divr   )r   r   r;   r   rY   r   r   r   test_arith_flex_series  s    
z.TestFrameFlexArithmetic.test_arith_flex_seriesr   int64float64c                 C   sX   t tdd|d}t tjtjgddgddgg}|j|d dd	}t|| d S )
Nr?   r   r!   r         ?g      ?r   r`   r   )	r   r   r   r   r   infr   r9   r:   )r   r   r;   rf   re   r   r   r   #test_arith_flex_series_broadcasting  s    z;TestFrameFlexArithmetic.test_arith_flex_series_broadcastingc                 C   s   t g td}tddgd}tddgddggddgd}tjtd	d
 |j|dd W d    n1 sh0    Y  tjtd	d
$ |j|d d dd W d    n1 s0    Y  d S )Nr!   r4   r   rB   r|   r}   r   r@   r   rD   Er   r   )r   objectr   rI   rJ   r   r   r   )r   Zser_len0Zdf_len0r;   r   r   r   test_arith_flex_zero_len_raises  s    ,z7TestFrameFlexArithmetic.test_arith_flex_zero_len_raisesc                 C   s^   t jddt jdddgdd}td|itd	d
}|dd}|jddd}t|| d S )Nr   r|   r   r@   r   floatr!   r5   r?   r   r}   r   )	r   r"   r   r   r   fillnar   r9   r:   )r   Zdatr;   r<   r=   r   r   r   test_flex_add_scalar_fill_value  s
    z7TestFrameFlexArithmetic.test_flex_add_scalar_fill_valuec                 C   sV   t g dg dd}t g dg dd}t g dg dd}||}t|| d S )N)r|   r}   r   r@   r   )r|   r}   r|   r}   r   r   r   )r   r}   r   r}   r}   )r|   r|   r}   r}   r   )r   r   r9   r:   )r   r   r   rf   re   r   r   r   'test_sub_alignment_with_duplicate_index  s
    
z?TestFrameFlexArithmetic.test_sub_alignment_with_duplicate_index)r   r   r   __truediv__c                 C   sj   t tdtjdd}t|||}ddg|_ddg|_t|||}t|| t	| |j
 d S )NrN   r   r4   )r   r   r   rF   rt   rv   rC   r9   r:   r'   r   )r   r   r;   rf   re   r   r   r   &test_arithmetic_with_duplicate_columns  s    

z>TestFrameFlexArithmetic.test_arithmetic_with_duplicate_columnslevelr   Nc                 C   s   t g dg dd}|jd|_t g dg dd}|jddg|_|j||d}t g dg dd}|jddg|_t|| d S )	Nr   r|   r}   r   r   ZL1r   r   r   ))r4   r   )r4   DZL2)r  )r   rC   Z	set_namesr   r9   r:   )r   r  r   r   re   rf   r   r   r   test_broadcast_multiindex  s    z1TestFrameFlexArithmetic.test_broadcast_multiindexc                 C   s   t g dg ddtjdgdgg dgg ddd	}td
gtjdgdggddgdd	}t g dg ddtjdgdgg dgg ddd	}|j|dd}t|| d S )Nr   r   r@   r   i  i  rU   rV   r  scenr   idnamesr   皙?r   r  )ffffff?333333@333333@)r  皙@g@r   r   r   r   from_productr   r   r9   r:   r   r;   seriesrf   re   r   r   r    test_frame_multiindex_operations  s$    z8TestFrameFlexArithmetic.test_frame_multiindex_operationsc                 C   s   t dgdgdtjdgdggddgdd	}tg d
tjdgdgg dgg ddd	}t g dg ddtjdgdgg dgg ddd	}|j|dd}t|| d S )Nr|   r   r  rU   rV   r  r   r  r   )      $@g      4@g      >@r  r  )g      &@   g      ?@)g      *@g      7@g     @@r   r   r  r  r   r   r   <test_frame_multiindex_operations_series_index_to_frame_index  s$    zTTestFrameFlexArithmetic.test_frame_multiindex_operations_series_index_to_frame_indexc              	   C   s   t g dg ddtjdgdgg dgg ddd	}td
gtjdgdggddgdd	}t tjtjdtjdddddtjfgg ddd	}|j|dd}t	|| d S )Nr   r  r  rU   rV   r  r  r  r   r  r~   r   r  rU   rV   r   rU   rV   r|   )rU   rV   r}   r   r   )
r   r   r  r   r   r   from_tuplesr   r9   r:   r  r   r   r   )test_frame_multiindex_operations_no_align  s.    
zATestFrameFlexArithmetic.test_frame_multiindex_operations_no_alignc                 C   s   t g dg ddtjg dg ddd}tdgtjd	gd
ggddgdd}t ddtjgddtjgdtjg dg ddd}|j|dd}t	|| d S )Nr   r  r  )r$  r%  )rU   r~   r}   r  r  r   r  rV   rU   r   r  r  r  r  r  r   r   )
r   r   r&  r   r  r   r   r   r9   r:   r  r   r   r   +test_frame_multiindex_operations_part_align#  s(    zCTestFrameFlexArithmetic.test_frame_multiindex_operations_part_align)r0   r1   r2   r   rI   r   Zskipifr   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     s8   
(



	

r   c                   @   s  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	e
jdddgdd Ze
jdddgdeddgeddeddggdd Zdd Zdd Zdd  Ze
jd!d"d#d$d%d&d'ed(d)d*fed(d)d+fge
jjd,ejejejejejejgd-d. d/d0d1 Zd2S )3TestFrameArithmeticc                 C   sJ   t ddgdd}tddgddgg}|| }t||d}t|| d S )	Nrr   r   r!   r|   r}   r   r@   r   r   r   r9   r:   )r   r   r;   re   rf   r   r   r   test_td64_op_nat_castingF  s
    z,TestFrameArithmetic.test_td64_op_nat_castingc                 C   s   t ddd}t|ddgg dd}|dgd d f }|jd|jd fksRJ tdd	gd	dgdd
gg|j|j|jd}|| }t	|| || }t	|| d S )Nr?   r   r}   TFr   rC   r`   r|   r@   r   rC   r`   r   
r   r   r   r   r   rC   r`   r   r9   r:   )r   r   r;   rowlikerf   re   r   r   r   'test_df_add_2d_array_rowlike_broadcastsP  s    z;TestFrameArithmetic.test_df_add_2d_array_rowlike_broadcastsc                 C   s   t ddd}t|ddgg dd}|d d dgf }|j|jd	 dfksRJ tddgd
dgddgg|j|j|jd}|| }t	|| || }t	|| d S )Nr?   r   r}   TFr   r,  r|   r   r   	   rN   r-  r.  )r   r   r;   collikerf   re   r   r   r   'test_df_add_2d_array_collike_broadcastse  s    z;TestFrameArithmetic.test_df_add_2d_array_collike_broadcastsc                 C   s   |}|r|dv rt | tddd}t|ddgg dd}|d	gd d f }|jd	|jd	 fkslJ t|jd
 ||	 t|jd ||	 t|jd ||	 g}t||j
|jd}	t|||}
t|
|	 d S )N__rmod____rfloordiv__r?   r   r}   TFr   r,  r|   r4   r   r   )td&mark_array_manager_not_yet_implementedr   r   r   r   r   rv   rs   squeezerC   r`   r9   r:   )r   r   r   using_array_managerr   r   r;   r/  exvalsrf   re   r   r   r   )test_df_arith_2d_array_rowlike_broadcastsz  s    
z=TestFrameArithmetic.test_df_arith_2d_array_rowlike_broadcastsc                 C   s   |}|r|dv rt | tddd}t|ddgg dd}|d d d	gf }|j|jd
 d	fkslJ t|d || t|d || d}d }	|dv rtj	dd |
 D  }	t||j|j|	d}
t|||}t||
 d S )Nr4  r?   r   r}   TFr   r,  r|   r   )TFc                 s   s   | ]}|j V  qd S r   )r   rX   r[   r   r   r   	<genexpr>  r^   zPTestFrameArithmetic.test_df_arith_2d_array_collike_broadcasts.<locals>.<genexpr>r-  )r7  r8  r   r   r   r   r   rv   r9  Zcommon_typer   rC   r`   r9   r:   )r   r   r   r:  r   r   r;   r2  r;  r   rf   re   r   r   r   )test_df_arith_2d_array_collike_broadcasts  s     
z=TestFrameArithmetic.test_df_arith_2d_array_collike_broadcastsc                 C   sh   t ddgddgg}|d }|jdd }|dk s<J d| }|jdd }|dk sdJ d S )NFTr|   c                 S   s   | j S r   kindr[   r   r   r   <lambda>  r^   z:TestFrameArithmetic.test_df_bool_mul_int.<locals>.<lambda>ic                 S   s   | j S r   r@  rB  r   r   r   rC    r^   )r   r   applyr   )r   r;   re   kindsr   r   r   test_df_bool_mul_int  s    z(TestFrameArithmetic.test_df_bool_mul_intc                 C   sD   t g dg dd}|| }t g dg dd}t|| d S )NrU   rV   r~   r   r   )ZaaZbbccr}   r@   r?   r   r9   r:   )r   ry   re   rf   r   r   r   test_arith_mixed  s    z$TestFrameArithmetic.test_arith_mixedrY   r4   r   c                 C   sB   t ddgddgd}||d| }||| d}t|| d S )Ng?gffffff
@g      @g333333r   r|   )r   r9   r   )r   Zall_arithmetic_functionsrY   r;   re   rf   r   r   r   test_arith_getitem_commute  s    z.TestFrameArithmetic.test_arith_getitem_commuter   r|   r}   r|   r}   r   c                 C   sD   t ddgddgd}t ddgddgd}|| }t|| d S )Nr|   r   r}   r   rK  )r   r   r;   rf   re   r   r   r   &test_arith_alignment_non_pandas_object  s    z:TestFrameArithmetic.test_arith_alignment_non_pandas_objectc                 C   s2  t tjddddddg dg dd}|d	j}t |j| |j|jd
}t	|| | t |jj
| j
|j|jd
}t	|j|dd| t|d }t |j| |j|jd
}t	|| | t |jj
| j
|j|jd
}t	|j|dd| tjj|j }t |j| |j|jd
}t	||| d S )Nr|   rN   f8r!   r   oner   threerH  r,  rU   r_   r   r   r   r`   )r   r   r   r   r   r   r`   rC   r9   r:   r   r   rH   rF   rt   r   )r   r;   Zval1addedZval2Zval3r   r   r   test_arith_non_pandas_object  s$    z0TestFrameArithmetic.test_arith_non_pandas_objectc                    sn   |t t jddd}ddg}t|g|d}d t| }t fdd	|D g|d}t|| d S )
N        r   )startendr|   r}   rB   rN   c                    s   g | ]}t | qS r   )rv   )rX   nnumr   r   r   
<listcomp>  r^   zVTestFrameArithmetic.test_operations_with_interval_categories_index.<locals>.<listcomp>)ro   ZCategoricalIndexZinterval_ranger   rv   r9   r:   )r   r   indr   r;   re   rf   r   rZ  r   .test_operations_with_interval_categories_index  s    zBTestFrameArithmetic.test_operations_with_interval_categories_indexc                 C   s   t tdtdgtdtdgdddgd}|dg }|| }t td	td	gtjtjgdddgd}t|| d S )
NZ2019Z2020Z2018Z2021)r5   r6   r5   r6   rB   r   )r   ro   rx   r   r   r   r9   r:   )r   r;   r   re   rf   r   r   r   test_frame_with_frame_reindex  s    
 z1TestFrameArithmetic.test_frame_with_frame_reindexzvalue, dtype)r|   i8)r   rP  )            rP  )y              ?
complex128)ra  rb  )Tr-      ns<m8[ns]<M8[ns]r   c                 C   s   | j S r   )r0   rB  r   r   r   rC  ,  r^   zTestFrameArithmetic.<lambda>)r   c              
   C   s~  t jdft jdft jdft jdfh}t||}td|j|jgi|jd}t jdft j	dft jdft jdft jdft jdft jdft j
dft j	dfh	}||f|v rpd }	|dkr|t jks|dkr|t jkrd }
n4|dkrd}
t}	n"|t j
u rd}
t}	nd	|j d
}
tjt|
dH t|	 |||j W d    n1 sB0    Y  W d    n1 sb0    Y  n
||f|v r(|t jt jfv rtjr|dkrt}	nd }	t|	 |||j W d    n1 s0    Y  n@d}
tjt|
d |||j W d    n1 s0    Y  nRtd * |||jj}|||j}W d    n1 sd0    Y  t|| d S )Nr-   r4   r!   rf  re  rb  z3ufunc 'remainder' not supported for the input typesz-numpy boolean subtract, the `-` operator, is zcannot perform __z7__ with this index type: (DatetimeArray|TimedeltaArray)rD   r   z+operator '.*' not implemented for .* dtypes)rw   truedivr   r   r   r   r   r   r   r   r   UserWarningr0   rI   rJ   rc   r9   assert_produces_warningr   r   r   r   r   )r   r   r   r   r   skipelemr;   invalidwarnrL   re   rf   r   r   r   test_binop_other  sd    

P..,z$TestFrameArithmetic.test_binop_otherN) r0   r1   r2   r+  r0  r3  r<  r?  rG  rL  rI   r   r   rM  r   r"   r   r   rO  rU  r^  r_  Ztimedelta64Z
datetime64rw   r   r   r   rg  r   r   rn  r   r   r   r   r)  E  sP   
 
*
r)  c                  C   s  t tjdddddgd} ttjd}| | }t | jtj | j	d}t
|| t
t | |k}W d    n1 s~0    Y  t d| j| j	d	}t
|| t | jd
| j	d}t
t ||k}W d    n1 s0    Y  t d| j| j	d	}t
|| d S )Nr?   r   r}   r4   r   rB   r!   Fr_   zM8[ns])r   r   rF   rG   r   r   r   r   r   rC   r9   r:   ri  FutureWarningr`   r,   )r;   r   re   rf   r   r   r   r   ,test_frame_with_zero_len_series_corner_casess  s     &&rp  c                  C   sB   t ddgtjd} tddgddgd}| | }| }t|| d S )Nr4   r   rC   r   r|   r}   r   )r   r   r   r   r9   r:   )r;   r   re   rf   r   r   r   ,test_zero_len_frame_with_series_corner_cases  s
    rr  z(ignore:.*Select only valid:FutureWarningc                  C   sD   dt ddtjgi} t| }|jdd}t g d}t|| d S )NZOner4   333333?r|   r   )r4   rs  r   )r   r   r   r   sumr9   r   )r   r;   re   rf   r   r   r   +test_frame_single_columns_object_sum_axis_1  s    ru  c                	   @   s  e Zd Zdd Zdd Zejdej	ej
ejejgdd Zejdd	d
gejddd Zejdg ddd Zdd Zdd Zdd Zdd Zdd Zdd Zejdejejejejejejgdd Zd d! Zd"d# Zd$d% Z d&d' Z!d(d) Z"ejdg d*d+d, Z#ejd-g d.d.e$j%g d.e$j&d/e'd0d1gd2d3 Z(ejd-d0d4gd5e$%d0d4ge'd0d6gd7d8 Z)d9d: Z*d;d< Z+d=d> Z,d?d@ Z-dAS )BTestFrameArithmeticUnsortedc                 C   sp   t jddddd}ttjt||dgd}|d}|| }|jj	t
ju sRJ || }|jj	t
ju slJ d S )	Nz1/1/2011rN   H
US/Eastern)rS   freqtzrU   r_   zEurope/Moscow)ro   rp   r   r   rF   rG   ru   
tz_convertr`   rz  pytzutc)r   rngr;   Z	df_moscowre   r   r   r   *test_frame_add_tz_mismatch_converts_to_utc  s    
zFTestFrameArithmeticUnsorted.test_frame_add_tz_mismatch_converts_to_utcc                 C   s   t jdddd}ttjt|d|d}||d d d  }|| }tj|jdd d< t	
|| |d d d }||tjt| }t	
|| d S )	Nz1/1/2000z1/1/2010r4   )ry  r   r   r}   r|   )ro   Zperiod_ranger   r   rF   rG   ru   r   r   r9   r:   takeZpermutation)r   r~  tsre   rf   Zhalfr   r   r   test_align_frame  s    z,TestFrameArithmeticUnsorted.test_align_framer   c                 C   s   t g dg ddtd}|tj}||d}||dt}d |t|< t	|| |||}|||t}d |t|< t	|| |||d}t	|| ||d|}tj	||dd d S )	N)r}         @{   N)r|   r}   r   r@   )col1Zcol2r!   r   r   F)Zcheck_dtype)
r   r  r  r   r   r*   comZisnar9   r:   )r   r   r;   Zfilledre   rf   r   r   r   test_operators_none_as_na  s     

z5TestFrameArithmeticUnsorted.test_operators_none_as_nazop,res)r   F)__ne__Tz ignore:elementwise:FutureWarningc                 C   s*   t ||d}t|  |u s&J d S )Nr5   )rv   r-   r   )r   r   r=   r   re   r   r   r   %test_logical_typeerror_with_non_valid  s    zATestFrameArithmeticUnsorted.test_logical_typeerror_with_non_valid)r   r   r   r   rg  c                    s  t jtdg dg dgg dd}ttddd|g d	d
  tj	t
t|d d u rfd S tg dg d}t
 ||ddd}t fdd| D  }t|| tddgddg}t
 ||ddd}t fdd| D   }t|| d S )NabcrQ  r   )firstsecondthirdr  Q      r   )Zvalue1value2Zvalue3r_   )r   r!  g      Y@r  r   )r  r.   c                    s:   g | ]2\}} j d d d d |f d d f |qS r   rs   rX   rD  vr;   idxopar   r   r\    r^   zETestFrameArithmeticUnsorted.test_binary_ops_align.<locals>.<listcomp>r   r!  r   rS  r  c                    s4   g | ],\}} j d d |f d d f |qS r   r  r  r  r   r   r\    r^   )r   r  rH   r   r   r   r   Z
sort_indexro   Z
IndexSlicerv   rw   r   concatitemsr9   r:   Zreindex_like)r   r   r`   r[   re   rf   r   r  r   test_binary_ops_align  s4    
 z1TestFrameArithmeticUnsorted.test_binary_ops_alignc                 C   s*  t ddgddgg}ttjddd|d}td	d
d}| }ddg|j_| }d|j	_
|j|d	d	d}|j|d	d	d}|j|d	d	d}|j|d	d	d}	|j|d	dd}
|j|d	dd}ttjg dg dgdd|d}||fD ]}t|| qddg|j_||	|
|fD ]}t|| qd S )Nr4   r   rU   rV   )r}   r@   r   r!   rB   r|   r}   rT   Zlvl0Zlvl1)r.   r  )r|   r}   r|   r}   )r   r  r   r   r   r   r)   rC   r  r`   namer   r"   r9   r:   )r   Zmidxr;   sr   s2Zres1r   Zres3Zres4Zres5Zres6r<   r=   r   r   r   &test_binary_ops_align_series_dataframe  s*    zBTestFrameArithmeticUnsorted.test_binary_ops_align_series_dataframec                 C   s   t jg ddd}|dd d }|ddd  }tdddgi|d	}tdddgi|d	}tdtjd
tjgi|d	}t|| | d S )N)z
2011-01-01z
2011-01-02z
2011-01-03UTC)rz  z
Asia/Tokyor}   rx  r|   r4   r   r   )ro   ZDatetimeIndexr{  r   r   r   r9   r:   )r   baseZidx1Zidx2r   r   r<   r   r   r    test_add_with_dti_mismatched_tzs+  s    z<TestFrameArithmeticUnsorted.test_add_with_dti_mismatched_tzsc                 C   s  | |jd d d }|d= tj|jd |jd df< || }|d  j}|d d  }t|d  |j|  tj|j|j	| < t|d |j|d j  t
|d  |jd d  sJ t
|d  sJ || }t|j|j || }	t
|d  sJ t
|	d  s4J |t  }
t
|
j sTJ t | }t
|j stJ t t  }|jsJ |j |jd d d d}t|| |d  || }t|d	d
 || }t|d	d
 || }t|dd id
 || }t|d	d
 d S )Nr}   r  r@   r   r4   r   r   rB   r   r!   )r   r`   r   r   rs   Zdropnar)   r9   r   isinisnanr   assert_index_equalr   r   r   rC   r:   r   )r   r   r   r   Z
frame_copyrT  Zindexerr<   Z
self_addedZ	added_revZ
plus_emptyZ
empty_plusZempty_emptyreverser   r   r   test_combineFrame5  s@    &

z-TestFrameArithmeticUnsorted.test_combineFramec           
      C   s\  | |jd }|| }| D ] \}}t||| ||   q | }d|d< t|}|| }	| D ] \}}t|	| |||   qjd|	v sJ t|	d 	 sJ || }t	|j
|jksJ ||d }t|dd id ||d }t|dd id |d| d	 }t|d	d
d	d	dd |d| d }t|dd
dd	dd d S )Nr   r|   r  float32r   r!   float16r   r   r   )r4   r   r   r  int32)r   r`   r  r9   r   to_dictr   r   r  r   r   r   r*   r   r   )
r   r   r   r   r  rT  keyr  Zlarger_seriesZlarger_addedr   r   r   test_combine_seriesr  s4    z/TestFrameArithmeticUnsorted.test_combine_seriesc                 C   sx  |d }|j |dd}| D ]`\}}|| }tj|| |dd || j|ksTJ |j|jkrp|jdks~J q|jd u sJ q|d d }|j |dd}t|j|j |d d }	|j |	dd}
t||
 |j |d d dd}tt	j
|j|jd}t|| |d d j |dd}tt	j
|j|jd}t|| |d d	 jg d
}|j|dd}t|t|kstJ d S )Nr4   r`   r   F)Zcheck_namesr   r_   r|   rB   )r   r  r9   r   r  r  r`   r:   r   r   r   rC   r   r   ru   )r   Zdatetime_framer  rT  r  rY   re   Zsmaller_frameZsmaller_addedZ
smaller_tsZsmaller_added2rf   framer   r   r   test_combine_timeseries  s8    z3TestFrameArithmeticUnsorted.test_combine_timeseriesc                 C   s   |d }t |j|jd  |d }| D ] \}}t |j|| jd  q,t|dd id t d }|jt js|J t|j	dksJ d S )Nr}   r   r!   r   )
r9   assert_numpy_array_equalr   r  r   r   r`   equalsru   rC   )r   r   r   re   r~   r  r   r   r   test_combineFunc  s    
z,TestFrameArithmeticUnsorted.test_combineFuncfuncc                 C   s  t  }t  }|d}t|jd }|||}t |j||j|j d}	tj	t
t|	d ||| W d    n1 s0    Y  |||}
t |
j||j|j ||d}t |j||jd d}	tj	t
|	d" |||d d  W d    n1 s0    Y  d S )NrU   r|   r|   r|   zNUnable to coerce to Series/DataFrame, dimension must be <= 2: (30, 4, 1, 1, 1)rD   r   z.Can only compare identically-labeled DataFramer}   )r9   ZmakeTimeDataFramer   r   r   r   r  r   rI   rJ   rK   reescape)r   r   r   r  r   r   r   r   re   rL   r   result3r   r   r   test_comparisons  s&    

(

z,TestFrameArithmeticUnsorted.test_comparisonsc                 C   s\   t dd dD }tt|}d}tjt|d ||d W d    n1 sN0    Y  d S )Nc                 S   s   i | ]}|d dddqS )r5   r6   r7   )r[   r\   zr   r=  r   r   r   r]     r^   zZTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raises.<locals>.<dictcomp>rH  z;'[<>]=?' not supported between instances of 'str' and 'int'rD   r   )r   rv   rw   rI   rJ   rc   )r   Zcompare_operators_no_eq_ner;   r   rL   r   r   r   *test_strings_to_numbers_comparisons_raises  s    
zFTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raisesc                 C   s   t  }tj|j|jd df< tjdd |jdk }W d    n1 sJ0    Y  tjdd |dk j}W d    n1 s0    Y  t || d S )Nr   r4   ignore)rl  raise)	r9   ZmakeDataFramer   r   rs   r`   Zerrstater   r  )r   Z
missing_dfrf   re   r   r   r   'test_comparison_protected_from_errstate  s    ((zCTestFrameArithmeticUnsorted.test_comparison_protected_from_errstatec                 C   s^  t tdd}tddg}tddg}|j}g d}t|}t ddgddgddgg}||k}t	|| |j
|k}t||j
 d}	d}
d	}tjt|	d
 ||k W d    n1 s0    Y  tjt|	d
 ||k W d    n1 s0    Y  ||k}t	|| |j
|k}t||j
 tjt|
d
 ||k W d    n1 s^0    Y  tjt|d
 |j
|k W d    n1 s0    Y  t ddgddgddgg}||k}t	|| tjt|	d
 ||k W d    n1 s0    Y  tjt|	d
 ||k W d    n1 s60    Y  ||k}t	|| |j
|k}t||j
 tjt|
d
 ||k W d    n1 s0    Y  |j
j|jksJ t tddtdtdd}|j|_|j|_tjt|	d
 ||k W d    n1 s0    Y  tjt|	d
 ||k W d    n1 sP0    Y  d S )Nr?   r   r}   r}   r}   r}   FTz5Unable to coerce to Series, length must be 2: given 3,Unable to coerce to DataFrame, shape must bez4operands could not be broadcast together with shapesrD   ZABr  r,  )r   r   r   r   r"   Z
atleast_2dr   tupler9   r:   r   r  rI   rJ   rK   r   rH   r`   rC   )r   r;   rV   Zb_rZb_clsttuprf   re   Zmsg1dZmsg2dZmsg2dbr   r   r   test_boolean_comparison  sb    
&&
(*((
((z3TestFrameArithmeticUnsorted.test_boolean_comparisonc                 C   s  t d}ttdt| dt||tdd}d|jd d ddf   }t d}t d}| }|| | j	|d	}||  |7  < |j	|d	}| }|| ||  j	|d	}	||  || 7  < |j	|d	}
t
|| t
||	 t
||
 | }|| | j	|d	}||  |8  < |j	|d	}| }|| ||  j	|d	}	||  || 8  < |j	|d	}
t
|| t
||	 t
||
 d S )
NZabcdefgrN   r   r,  r   r|   ZbedcfZbcdefrB   )rH   r   r   r   ru   r   r   r   r)   r   r9   r:   )r   rC   ZX_origZZblock1subsXZresult1r   r  Zresult4r   r   r   test_inplace_ops_alignment[  s>    z6TestFrameArithmeticUnsorted.test_inplace_ops_alignmentc           	      C   s"  t g d}ttjjdddddd}| }|}|d7 }t|| t|d | ||u sfJ |j	|j	u svJ | }|}|d7 }t
|| t
|d | ||u sJ |j	|j	u sJ | }|}|d7 }t|| t|d | | }|}|d7 }t
|| t
|d | ||u s0J |j	|j	u sBJ tjjdddd}t| d	d
}| }|}|d  d7  < t| d d	d
}t
|| t
|| |j	|j	u sJ | }|}|d  d7  < t| d d	d
}t
|| t
|| |j	|j	u sJ d S )Nr   r   r   rN   rO   r   r|   r   r5   r   r4   )r   r   r   rF   rq   r   r)   r9   r   Z_mgrr:   )	r   Zs_origZdf_origr  r  r;   r   r   rf   r   r   r   test_inplace_ops_identity  sX    z5TestFrameArithmeticUnsorted.test_inplace_ops_identity)r   andr   r   r   r   orr   r   rg  xorc                 C   s   |dkrd S t g dg dd}d}|dv r:g d|d< | }d	| d
}d
| d
}t||| t|||}t|| t|}t||ksJ d S )Nr   r   r   rT   r}   )r  r  r  )TFTrU   Z__ir   )r   r)   rv   r9   r:   r  )r   r   r;   operandZdf_copyZioprf   r   r   r   test_inplace_ops_identity2  s    z6TestFrameArithmeticUnsorted.test_inplace_ops_identity2valr   r!   r|   r@   c                 C   s   g d}g d}t tjdd||d}tjjj}t |||d|jd}t	
|||dd | t g dg d	g d
d|jd}t	
|||dd | d S )Nr   r  Yr  r   r_   r   r`   r|   r  r  r   r   r   rC   )r   r   rF   rG   ro   coreopsalign_method_FRAMEr`   r9   r:   )r   r  r`   rC   r;   alignrf   r   r   r   test_alignment_non_pandas  s    

z5TestFrameArithmeticUnsorted.test_alignment_non_pandasr}   rN  r   c                 C   s   g d}g d}t tjdd||d}tjjj}d}tj	t
|d |||d W d    n1 sf0    Y  tj	t
|d |||d W d    n1 s0    Y  d S )	Nr   r  r   r_   z5Unable to coerce to Series, length must be 3: given 2rD   r`   rC   )r   r   rF   rG   ro   r  r  r  rI   rJ   rK   )r   r  r`   rC   r;   r  rL   r   r   r   )test_alignment_non_pandas_length_mismatch  s    
*zETestFrameArithmeticUnsorted.test_alignment_non_pandas_length_mismatchc                 C   s  g d}g d}t tjdd||d}tjjj}tg dg dg dg}t	
|||dd	 t ||j|jd t	
|||d
d	 t ||j|jd d}tg dg dg}tjt|d |||d W d    n1 s0    Y  tjt|d |||d
 W d    n1 s 0    Y  td}td}tjt|d |||d W d    n1 sp0    Y  tjt|d |||d
 W d    n1 s0    Y  d S )Nr   r  r   r_   r   r@   r   r?   )r   r   r1  r`   r|   rC   r  rD   r  zGUnable to coerce to Series/DataFrame, dimension must be <= 2: (3, 3, 3))r   r   rF   rG   ro   r  r  r  r"   r9   r:   r`   rC   rI   rJ   rK   Zzerosr  r  )r   r`   rC   r;   r  r  rL   r   r   r   'test_alignment_non_pandas_index_columns  s6    
*,
,zCTestFrameArithmeticUnsorted.test_alignment_non_pandas_index_columnsc                 C   sZ   t ddgdd gd}|d }td  t||| W d    n1 sL0    Y  d S )NrV  r   r   )r   r9   ri  rv   r   r   r;   rV   r   r   r   test_no_warning6  s    z+TestFrameArithmeticUnsorted.test_no_warningc                 C   s`   t ddgdd gd}|d }tjtdd  t|||d W d    n1 sR0    Y  d S )NrV  r   r   ztakes 2 positional argumentsrD   r   )r   rI   rJ   rc   rv   r  r   r   r   test_dunder_methods_binary<  s    z6TestFrameArithmeticUnsorted.test_dunder_methods_binaryc                 C   sj   t jddddd}t jdtd}t|}| |d< |t}||	  }||	  }t
|| d S )Nr   r   r!   rN   )rN   r|   z0.X)r   r   r   r   intr   r9  r*   r  meanr9   r:   )r   r  r  r   r   re   rf   r   r   r   test_align_int_fill_bugC  s    
z3TestFrameArithmeticUnsorted.test_align_int_fill_bugN).r0   r1   r2   r  r  rI   r   r   rw   r   r   r   rg  r  filterwarningsr  r  r  r  r  r  r  r  rl   rm   ri   rh   rj   rk   r  r  r  r  r  r  r  r   r"   r   r   r  r  r  r  r  r  r   r   r   r   rv    s^   


'
=).
	O1:
	
(
"rv  c                  C   sL   t dg di} t g dd}| | }t dtjdtjgi}t|| d S )Nr4   r  r   r   r   r   r   r9   r:   )ry   rz   re   rf   r   r   r   test_pow_with_realignmentR  s
    r  c                  C   sp   t dtjtjtjgi} t dg di}t dg di}| | }t|| | d |d  }t||d  d S )Nr4   r  )r   r   r   )r   r   r   r9   r:   r   )ry   rz   rf   re   r   r   r   test_pow_nan_with_zero]  s    r  c                  C   s   t tjdddg dd} tg dg dd}| d|dd	d
 }t || jdd}| d}|| }t	
|| ||d }t	
|| d S )Nr   r   )rN   r   rH  rB   r   r   r   r   r   ZInt64rq  )r   r   rF   rq   r   Zto_numpyr   rC   r*   r9   r:   )r;   r   rf   Zdf_eare   r   r   r   &test_dataframe_series_extension_dtypesj  s    
r  c                     s   t jddd} t| }| }t j|jdg df< | }t j|jddgf< | }t j|jdt ddf< | }t j|jdt ddf< ||f||f||ffD ]6\   }t fd	d
 jD }t	
|| qd S )Nr   i  )r   rN   )r|   r   r   r   r}   r@   r   c                    s   i | ]}| | |  qS r   r   )rX   rD  ry   rz   r   r   r]     r^   z6test_dataframe_blockwise_slicelike.<locals>.<dictcomp>)r   rF   rq   r   r)   r   r   r   rC   r9   r:   )r   r   r   r   Zdf4Zdf5r=   rf   r   r  r   "test_dataframe_blockwise_slicelikey  s    r  zdf, col_dtyper   r   g      @r  abrB   r   rV   r  c                 C   sT   t dtjgdtjggtdd}|d|i}| tdgtdd }t|| d S )	NrV  r   r  rB   rV   g      rU   r   )r   r   r   rH   r*   r   r9   r:   )r;   Z	col_dtyperf   re   r   r   r   /test_dataframe_operation_with_non_numeric_types  s    	 r  c                  C   sZ   t dggdgd} t g dgg dd}| | }t tjddggg dd}t|| d S )Nr   r  )r   rC   r  )r  r  r  rB   r  )r   r   re   rf   r   r   r   "test_arith_reindex_with_duplicates  s
    r  to_addr|   c                 C   s   t ddgddgd}dt| d  d}tjt|d ||   W d    n1 sV0    Y  tjt|d | |  W d    n1 s0    Y  d S )Nr|   r}   rZ   zUnable to coerce list of r   z to Series/DataFramerD   )r   r+   rI   rJ   rK   )r  r;   rL   r   r   r   "test_arith_list_of_arraylike_raise  s    &r  c                  C   sT   t dg di} | d }|j}|d7 }|j|u s4J t dg di}t| | d S )Nr4   r   r|   )r}   r   r@   )r   Z_valuesr9   r:   )r;   r  valsrf   r   r   r   %test_inplace_arithmetic_series_update  s    r  c                  C   s   t dggdgtjdgdggddgdd} t dggdgtdgddd}t dggdgtjdgdggddgdd}| | }t|| dS )	zP
    Regression test for: https://github.com/pandas-dev/pandas/issues/33765
    r|   rU   r   rV   r  r_   r  N)r   r   r  r   r9   r:   )r   r   rf   re   r   r   r   !test_arithemetic_multiindex_align  s    r  c                  C   sF   t dtdtd} | d }t tdtdtd}t|| d S )NTr  cdr   r   )r   rH   r   r   r9   r:   )r;   re   rf   r   r   r   test_bool_frame_mult_float  s    r  c                 C   sP   t g d| d}t g d| d}tg d| d}| |  }t|| d S )N)r|   r}   Nr!   r   )r   r   N)r   r   Zto_framer9   r:   )Zany_int_ea_dtypeZseries1Zseries2rf   re   r   r   r   test_frame_sub_nullable_int  s
    r  c                     s   G  fdddt G fdddt  dg dg dd} | d	 } dg d
g dd}t|| | |  }t|| d S )Nc                       s,   e Zd ZefddZe fddZdS )zEtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeriesc                    s    S r   r   r#   SubclassedSeriesr   r   _constructor  s    zRtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeries._constructorc                    s    S r   r   r#   )SubclassedDataFramer   r   _constructor_expanddim  s    z\test_frame_op_subclass_nonclass_constructor.<locals>.SubclassedSeries._constructor_expanddimN)r0   r1   r2   propertyr  r  r   r  r  r   r   r    s   r  c                       s>   e Zd ZdgZ fddZedd ZefddZ  ZS )zHtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFramemy_extra_datac                    s   || _ t j|i | d S r   )r  superr    )r   r  argskwargs	__class__r   r   r      s    zQtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame.__init__c                 S   s   t t| | jS r   )	functoolspartialr+   r  r#   r   r   r   r    s    zUtest_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame._constructorc                    s    S r   r   r#   r  r   r   _constructor_sliced  s    z\test_frame_op_subclass_nonclass_constructor.<locals>.SubclassedDataFrame._constructor_sliced)	r0   r1   r2   	_metadatar    r  r  r  __classcell__r   r  r  r   r    s   
r  Z	some_datar   r  r   r}   rJ  )r   rN      r*  )Zsdfre   rf   r   r  r   +test_frame_op_subclass_nonclass_constructor  s    	r  c                  C   sd   t dd} t| jg di}t| jg di}|| j || j k}tg d| jd}t|| d S )NColsz	col1 col2r   )TTTr  )r   r   r  r   r9   r   )r  q1q2re   rf   r   r   r   test_enum_column_equality  s    
r	  )Bcollectionsr   r   enumr   r  rw   r  numpyr   rI   r|  Zpandas.util._test_decoratorsutilZ_test_decoratorsr7  Zpandasro   r   r   r   r   Zpandas._testingZ_testingr9   Zpandas.core.commonr  commonr  Zpandas.core.computationr	   r   Z#pandas.core.computation.expressionsr
   r   Zpandas.tests.frame.commonr   r   Zfixturer   r   r3   r   r   r)  rp  rr  r   r  ru  rv  r  r  r  r  r   rH   r  r  r  r  r  r  r  r  r	  r   r   r   r   <module>   sz   
 U D  g  0


     /
	.
	$