a
    Pf                     @   sV  d dl mZmZmZ 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mZmZmZ d dlmZ d dlmZ d dlmZ d d	l m!Z! e	j"d
d dgddgddd Z#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)G dd dZ*G dd  d Z+d!d" Z,d#d$ Z-d%d& Z.dS )'    )date	timedeltatimezone)DecimalN)lib)IncompatibleFrequency)CategoricalDatetimeTZDtypeIndexSeries	Timedeltabdate_range
date_rangeisna)ops)expressions)NUMEXPR_INSTALLEDTi@B numexprpython)ZautouseparamsZidsc                 c   sB   |  &}|td| j d V  W d    n1 s40    Y  d S )NZ_MIN_ELEMENTS)contextsetattrexprparam)requestZmonkeypatchm r   \/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/series/test_arithmetic.pyswitch_numexpr_min_elements    s    
r   c                 C   s   |  tjdt| S N   )takenprandomdefault_rngpermutationlen)objr   r   r   _permute'   s    r(   c                	   @   s  e Zd Zejddd dd dfdd dd dfdd d	d d
fdd dd d
fgejdg ddd Zdd Zdd Ze	j
ejdfe	jejdfgZdD ]VZdZee	eZeeeZee	de ZefddZeeeef eeeef qejdedd ZdS )TestSeriesFlexArithmetictsc                 C   s   | S Nr   xr   r   r   <lambda>/       z!TestSeriesFlexArithmetic.<lambda>c                 C   s   | d S r   r   r,   r   r   r   r.   /   r/   Fc                 C   s   | S r+   r   r,   r   r   r   r.   0   r/   c                 C   s   | d d d S r   r   r,   r   r   r   r.   0   r/   c                 C   s   | S r+   r   r,   r   r   r   r.   1   r/   c                 C   s   dS )N   r   r,   r   r   r   r.   1   r/   Tc                 C   s   t tdtjdS N
   dtyper   ranger"   float64r,   r   r   r   r.   3   r/   c                 C   s   t tdtjdS r1   r5   r,   r   r   r   r.   4   r/   opname)addsubmulfloordivtruedivpowc                 C   s   t tjdtjdtddddd}|d |}|d |}|d	 }tt |}tt|}|||}	|||}
t|	|
 |rtt d
| }|||}	|||}
t|	|
 d S )N   r3   z
2020-01-01periodsr*   indexnamer      r    r)	r   r"   aranger7   r   getattroperatortmZassert_almost_equal)selfr8   r*   ZtserseriesotherZcheck_reverseopZaltresultexpectedropr   r   r   test_flex_method_equivalence,   s$    






z5TestSeriesFlexArithmetic.test_flex_method_equivalencec                    sT   G  fdddt  |}tt |} g ddd}d|_||d}|jdksPJ d S )Nc                       s"   e Zd ZdgZe fddZdS )zZTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeriesr-   c                    s    S r+   r   )rK   MySeriesr   r   _constructorY   s    zgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeries._constructorN)__name__
__module____qualname__	_metadatapropertyrU   r   rS   r   r   rT   V   s   rT   rE   r       testrD   *   rE   )r   rH   r-   )rK   Zall_arithmetic_operatorsr8   rN   r   rO   r   rS   r   /test_flex_method_subclass_metadata_preservationT   s    

zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationc                 C   sD   t ddtjdddg}|dd}|jddd}t|| d S )Nr   rE   r\      r0   r    
fill_value)r   r"   nanfillnar9   rJ   assert_series_equal)rK   serexpresr   r   r   test_flex_add_scalar_fill_valued   s    z8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuerE   )r9   r:   r;   r>   r=   r<   r   rF   c                 C   s   t t||| S r+   )rH   rI   )r-   yrN   r   r   r   r.   s   r/   zop, equiv_op, fvc           	      C   s   ddd}t tjdddtjgtdd}t tjd	tjd
tjdgtdd}|||}|||}t|| ||||||d |||dd d S )Nr   c              	   S   s6  |j |j }||}||}t|}t|}g }tt|D ]}	tjdd ||	 r||	 r~|tj	 W d    qB|||||	  nX||	 r||	 r|tj	 W d    qB||||	 | n||||	 ||	  W d    qB1 s0    Y  qB| |||d}
t
||}t|
| d S )Nignore)allrb   )rC   unionreindexr   r6   r&   r"   Zerrstateappendrd   r   rJ   rf   )methrN   abrc   	exp_indexZamaskZbmask
exp_valuesirO   rP   r   r   r   _check_filly   s,    

:
zDTestSeriesFlexArithmetic.test_operators_combine.<locals>._check_fill      ?g       @g      @r0   rC   rE   r\         @   rb   axis)r   )r   r"   rd   rG   rJ   rf   )	rK   rN   Zequiv_opfvrw   rr   rs   rO   rh   r   r   r   test_operators_combinew   s    
 $

z/TestSeriesFlexArithmetic.test_operators_combineN)rV   rW   rX   pytestmarkparametrizerR   r`   rj   r   divrI   r=   Zrdivr   ZrtruedivZpairingsrN   r~   rH   ZlopZlequivrQ   Zrequivrp   r   r   r   r   r   r)   +   s8   

r)   c                   @   s   e Zd Zdd Zejddddgddgfd	ddgd
dgfgdd Zdd Zejdde	dgdd Z
dd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zejd+d,d-gd.d/ Zd0d1 ZdS )2TestSeriesArithmeticc                 C   s   t jdddd}ttjdt||d}||d d d  }|| }tj|j	dd d< t
|| |t|d d d  }t
|| d}tjt|d	" ||jd
dd  W d    n1 s0    Y  d S )N1/1/2000z1/1/2010Yfreqr    ry   rE   z4Input has different freq=D from Period\(freq=Y-DEC\)matchDend)how)pdZperiod_ranger   r"   r#   r$   standard_normalr&   rd   ilocrJ   rf   r(   r   raisesr   Zasfreq)rK   rngr*   rO   rP   msgr   r   r   !test_add_series_with_period_index   s    z6TestSeriesArithmetic.test_add_series_with_period_indexz%target_add,input_value,expected_value!ZhelloZworldzhello!zworld!r   ZhellomZworldmc                 C   s(   t |}|| }t |}t|| d S r+   r   rJ   rf   )rK   Z
target_addZinput_valueZexpected_valuerr   rO   rP   r   r   r   test_string_addition   s    	z)TestSeriesArithmetic.test_string_additionc                 C   s   t dddtjgg dd}t dtjdtjgg dd}||}t||}t|d |d  t|d |d  ||}t||}t|d |d  t|d |d  d S )NrE   )rr   rs   cdry   r    )rr   rs   r   er   )r   r"   rd   divmodrJ   rf   Zrdivmod)rK   rr   rs   rO   rP   r   r   r   test_divmod   s    



z TestSeriesArithmetic.test_divmodrC   N	   c                 C   s\   t tdd}t d|d}d}tjttf|d ||  W d    n1 sN0    Y  d S )NrE   r2   foory   z<not all arguments converted during string formatting|mod notr   )r   r6   r   r   	TypeErrorNotImplementedError)rK   rC   s1s2r   r   r   r   test_series_integer_mod   s
    z,TestSeriesArithmetic.test_series_integer_modc                 C   sX   t ddgddgd}t ddgddgd}|| }t ddtjgg dd}t|| d S )NrE   r    ry   r2         )rE   rE   r    )r   r"   rd   rJ   rf   )rK   r   r   rO   rP   r   r   r   test_add_with_duplicate_index   s
    z2TestSeriesArithmetic.test_add_with_duplicate_indexc                 C   sp   t tdtdgtdddtdddgd}||d }|d| }t|jd sZJ t|jd slJ d S )Nz1.3z2.3i  rE   r    ry   r   )r   r   r   shiftr   r   )rK   rg   rO   result2r   r   r   test_add_na_handling   s    $z)TestSeriesArithmetic.test_add_na_handlingc                 C   sN   t g tg tjd}|| }t| s.J ||  }t|dksJJ d S )N)rC   r4   r   )r   r
   r"   r7   isnanrm   copyr&   )rK   datetime_seriesemptyrO   r   r   r   test_add_corner_cases   s
    z*TestSeriesArithmetic.test_add_corner_casesc                 C   sZ   | td d }|| }t|jd d |j |jd d dd}t|d d | d S )Nr*   rB   )astypeintr   valuesrC   rJ   rf   )rK   r   Zint_tsaddedrP   r   r   r   test_add_float_plus_int   s    z,TestSeriesArithmetic.test_add_float_plus_intc                 C   s>   t g g tjd}t ddi}t|| t tjgdgd d S )Nr3   r-   g        ry   )r   r"   Zint32rJ   rf   rd   )rK   r   r   r   r   r   test_mul_empty_int_corner_case  s    z3TestSeriesArithmetic.test_mul_empty_int_corner_casec                 C   s   t tdddd}tj|jd< |d d d }t tdtdtjg}|| }t	|| t |dd}|
 |
  d }t	|| d S )	Nz2012-1-1r\   r   rA   r   r    r   r^   )r   r   r"   rd   r   r   r   ZNaTrJ   rf   to_frame)rK   dtZdt2rP   rO   r   r   r   test_sub_datetimelike_align  s    z0TestSeriesArithmetic.test_sub_datetimelike_alignc                 C   sX   t dddd}|d}td|d}td|d}||  |j|u sFJ |j|u sTJ d S )Nz
2016-01-01r2   ZCET)rA   tzUTCry   )r   
tz_convertr   rC   )rK   dtiZdti_utcrg   ser_utcr   r   r   test_alignment_doesnt_change_tz  s    
z4TestSeriesArithmetic.test_alignment_doesnt_change_tzc                 C   s   t g d}td|d}td|d d d}|| }dgd dgd  d	g }tj||jd
}dgd tjg }t||}t|| d S )N)3z53r   LoJGr   r   N503r    ry   r   r   ra   r   r   r   )
categoriesrz      )	r   r   r   ZCategoricalIndexr   r"   rd   rJ   rf   )rK   catser1ser2rO   rt   ru   rP   r   r   r   test_alignment_categorical$  s    
z/TestSeriesArithmetic.test_alignment_categoricalc                 C   s   g d}t tjdddd|d}t tjddd|d}|| }t dg dd}t|| t tddd	|d}t td
dd	|d}|| }t tdg dd}t|| d S )N)r    r    r\   r\   ra   rE   r{   Zint64r3   ry   r0   z20130101 09:00:00r@   Z20130101z9 hours)r   r"   rG   rJ   rf   r   r   )rK   rC   rg   rM   rO   rP   r   r   r   $test_arithmetic_with_duplicate_index2  s    z9TestSeriesArithmetic.test_arithmetic_with_duplicate_indexc                 C   sL   t dtjgdd}t ddgdd}|| }t dtjgdd}t|| d S )Nr   floatr3   rE   Int64Float64)r   r"   rd   r   NArJ   rf   rK   r   r   rO   rP   r   r   r   'test_masked_and_non_masked_propagate_naD  s
    z<TestSeriesArithmetic.test_masked_and_non_masked_propagate_nac                 C   sl   t dtjddgdd}t ddtjdg}|| }t dtjtjdgdd}t|| || }t|| d S )N   r0   ra   r   r3   rx   r   )r   r   r   r"   rd   rJ   rf   r   r   r   r   +test_mask_div_propagate_na_for_non_na_dtypeL  s    z@TestSeriesArithmetic.test_mask_div_propagate_na_for_non_na_dtypez
val, dtype)r\   r   )g      @r   c                 C   s^   t g ddd}|dd |g }t dd d| g|d}t|| dd |g| }t|| d S )N)rE   Nr\   r   r3   rE   r    r\   r   )rK   valr4   rg   rO   rP   r   r   r   test_add_list_to_masked_arrayW  s    z2TestSeriesArithmetic.test_add_list_to_masked_arrayc                 C   s   |j jjdkrtrtnd }tg ddd}t| |g d }W d    n1 sV0    Y  tg ddd}t|| t| g d| }W d    n1 s0    Y  t|| d S )Nr   )TNFbooleanr3   )TNT)	nodeZcallspecidr   UserWarningr   rJ   assert_produces_warningrf   )rK   r   warningrg   rO   rP   r   r   r   %test_add_list_to_masked_array_booleanb  s    **z:TestSeriesArithmetic.test_add_list_to_masked_array_boolean)rV   rW   rX   r   r   r   r   r   r   r6   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s2   



	

r   c                   @   s   e Zd Zejdg ddd Zdd Zejdg dd	fg d
dfg ddfg ddfg ddfg ddfgdd Zejdg dd	dfg dddfg dddfg dddfg dddfg dddfgdd Z	dS )TestSeriesFlexComparisonr}   )r   NrC   c                 C   sZ   t tjdd}t tjdd}t||j||d}|||}t|| d S )Nr    r2   r|   )	r   r"   r#   r$   r   rH   rV   rJ   rf   )rK   r}   comparison_opleftrightrO   rP   r   r   r   test_comparison_flex_basicy  s
    
z3TestSeriesFlexComparison.test_comparison_flex_basicc                 C   sv   t tjdd}t tjdd}d}tjt|d$ t||j	|dd W d    n1 sh0    Y  d S )Nr    r2   zNo axis named 1 for object typer   rE   r|   )
r   r"   r#   r$   r   r   r   
ValueErrorrH   rV   )rK   r   r   r   r   r   r   r   test_comparison_bad_axis  s
    z1TestSeriesFlexComparison.test_comparison_bad_axisz
values, op)FFTFeq)TTFTnele)FFFFlt)FTTFge)FTFFgtc                 C   sV   t g dtdd}t g dtdd}t|||}t |tdd}t|| d S )NrE   r\   r    abcry   r    r    r    bcdabcdr   listrH   rJ   rf   )rK   r   rN   r   r   rO   rP   r   r   r   test_comparison_flex_alignment  s
    z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_value)FFTTr    )TTFFr   )FFFT)TTTFc                 C   sZ   t g dtdd}t g dtdd}t||||d}t |tdd}t|| d S )Nr   r   ry   r   r   rb   r   r   )rK   r   rN   rc   r   r   rO   rP   r   r   r   #test_comparison_flex_alignment_fill  s
    z<TestSeriesFlexComparison.test_comparison_flex_alignment_fillN)
rV   rW   rX   r   r   r   r   r   r   r   r   r   r   r   r   x  s2   







r   c                
   @   s  e Zd Zdd Zejdg ddd Zejdg 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dd Zejdeg dedddeg d ed!ddfeg dedddeg d"ed#ddfgd$d% Zd&d' Zd(S ))TestSeriesComparisonc                 C   s   t g d}t ddg}d}tjt|d ||k  W d    n1 sH0    Y  t ddg}t g d}tjt|d ||k W d    n1 s0    Y  d S )	Nrr   rs   r   rs   rr   z'only compare identically-labeled Seriesr   rE   r    )r    r\   ra   )r   r   r   r   )rK   rr   rs   r   r   r   r    test_comparison_different_length  s    &z5TestSeriesComparison.test_comparison_different_lengthr8   )r   r   r   r   r   r   c                 C   sB   t g dtdd}d}t|||j}td}||ks>J d S )Nr   r\   ry   r    bool)r   r6   rH   dtypesr"   r4   )rK   r8   rg   constrO   rP   r   r   r   test_ser_flex_cmp_return_dtypes  s
    
z4TestSeriesComparison.test_ser_flex_cmp_return_dtypesc                 C   sP   t g dtdd}|jd d }d}t|||j}td}||ksLJ d S )Nr   r\   ry   r   r    r   )r   r6   r   rH   r   r"   r4   )rK   r8   rg   r   r   rO   rP   r   r   r   %test_ser_flex_cmp_return_dtypes_empty  s    
z:TestSeriesComparison.test_ser_flex_cmp_return_dtypes_emptynames))NNN)r   barN)bazr   r   c           
      C   sn  |}t ddd|d d}t||d }|||}|j|d ksFJ |d}tj|d	d
}t||d }|||}|j|d ksJ ||d }t||d }|||}|j|d ksJ |tj	tj
fv rtjdd|d d}t||d }|||}|j|d ksJ |tj	tj
fv rj|d}	t|	|d }|||	}|j|d ksjJ d S )Nz1949-06-07 03:00:00hr0   r   )r   rA   rD   rE   r    z
US/CentralZinferr   )startrA   rD   category)r   r   renamerD   tz_localizer   ZDatetimeIndexr   rI   r   r   Zinterval_ranger   )
rK   r   r   rN   r   rg   rO   ZtdiiiZcidxr   r   r   test_ser_cmp_result_names  s0    






z.TestSeriesComparison.test_ser_cmp_result_namesc              	   C   s   t g d}t g d}t g d}|rdd l}d}tj|jj|d ||k W d    n1 sd0    Y  tjtddd	J tj|jj|d ||k W d    n1 s0    Y  W d    q1 s0    Y  n t	||k| t	||k| d S )
Nr   )FTFFFFr   zhas no kernelr   Z
comparisonF)r   Zcheck_stacklevel)
r   Zpyarrowr   r   r   ZArrowNotImplementedErrorrJ   r   DeprecationWarningrf   )rK   Zusing_infer_stringsr   rh   par   r   r   r   test_comparisons  s    &Fz%TestSeriesComparison.test_comparisonsc                 C   s:  t tddd}t tddd}t g ddd}t tddd}ttd}ttd}|dk rhJ |dk|dk k sJ d|k rJ |dkd sJ d|kd sJ d|kd rJ ||k sJ ||k rJ |t|k sJ ||k sJ ||k sJ ||k ||kk s6J ||k ||kk sRJ ||k rdJ ||k rvJ ||k rJ ||k rJ ||k sJ ||k sJ ||k rJ ||k rJ ||k ||kk sJ ||k ||kk sJ ||k ||kk s6J ||k ||kk sRJ 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 s0    Y  tjt|d
 ||k W d    n1 s,0    Y  d S )Nr   r   r3   object)rr   rs   ccZacbrr   r    can only compare equality or notr   )r   r   r   rm   r   r   r   )rK   rr   rs   r   r   r   fr   r   r   r   test_categorical_comparisons  sP    (((z1TestSeriesComparison.test_categorical_comparisonsc                 C   s  t ttd}d}tjt|d |dk W d    n1 s@0    Y  t ttddd}tjt|d |dk W d    n1 s0    Y  t ttddd}d}tjt|d |d	k  W d    n1 s0    Y  tjt|d |d	k W d    n1 s0    Y  tjt|d d	|k  W d    n1 sH0    Y  tjt|d d	|k W d    n1 s0    Y  t|d	kt g d
 t|d	kt g d d S )Nr   r  r   rs   F)orderedTz1Invalid comparison between dtype=category and strr   r  )TTT)r   r   r   r   r   r   rJ   rf   )rK   r   r   r   r   r   5test_unequal_categorical_comparison_raises_type_errorG  s&    &&&(((zJTestSeriesComparison.test_unequal_categorical_comparison_raises_type_errorc                 C   s   t ddg}|dk}t ddg}t|| |dk}t ddg}t|| |dk}t ddg}t|| |dk}t ddg}t|| t ddg}|dk}t ddg}t|| |dk}t ddg}t|| d S )N)rE   rE   )rE   r    FT)r   r   r   )rK   r  rO   rP   r   r   r   test_comparison_tuplesf  s(    z+TestSeriesComparison.test_comparison_tuplesc                 C   sD   t tdgtddgg}|tdgk}t ddg}t|| d S )NrE   r    TF)r   	frozensetrJ   rf   rK   rg   rO   rP   r   r   r   test_comparison_frozenset  s    z.TestSeriesComparison.test_comparison_frozensetc                 C   s   t tdddtd}tj|d d d< |d }|||}|| ||j}d}tj	t
|d< |tju r||d	t}n|d
t}W d    n1 s0    Y  t|| d S )Nr   r2   r@   r3   r    r0   zDowncasting object dtype arraysr   TF)r   r   r	  r"   rd   Zdropnaro   rC   rJ   r   FutureWarningrI   r   re   r   r   rf   )rK   r   rg   r   rO   rP   r   r   r   r   "test_comparison_operators_with_nas  s    

.z7TestSeriesComparison.test_comparison_operators_with_nasc                 C   sN   t g dg dtd}tg d}t|jdk| t|jdk | d S )N)r\   ra   r0   r{      r3   )TTFTTr0   )r   r   r"   arrayrJ   Zassert_numpy_array_equalrC   )rK   r*   rP   r   r   r   test_ne  s    zTestSeriesComparison.test_nezleft, rightr[   ABCr-   rB   r   ZABD)r    r    r    r    ZABCDc                 C   s  |t ur(d|j d}| }| }nd|j d}tjt|d ||k W d    n1 sb0    Y  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 s>0    Y  tjt|d ||k  W d    n1 sv0    Y  d S )Nz@Can only compare identically-labeled \(both index and columns\) z objectsz%Can only compare identically-labeled r   )r   rV   r   r   r   r   )rK   r   r   Zframe_or_seriesr   r   r   r   test_comp_ops_df_compat  s*    
&&&((z,TestSeriesComparison.test_comp_ops_df_compatc                 C   s0   t g d}|dk}t g d}t|| d S )N)	IntervalAZ	IntervalBZ	IntervalCr  )TFFr   r  r   r   r   $test_compare_series_interval_keyword  s    z9TestSeriesComparison.test_compare_series_interval_keywordN)rV   rW   rX   r   r   r   r   r   r   r  r  r  r  r  r  r  r  r   r   r  r  r   r   r   r   r     s8   

	
%8
r   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestTimeSeriesArithmeticc           	      C   s   t ddddd}tjddd d }ttjdd||dd	}tjddd d }ttjdd||d
d	}|| }|d}|d}|| }|	 }|j
jtju sJ t|| d S )N1/1/2011d   r   utc)rA   r   r   r    Z   z
US/Easternry   zEurope/Berlin)r   r"   r#   r$   r%   r   r   r!   r   Z
sort_indexrC   r   r   r   rJ   rf   )	rK   r   permr   r   rO   Zuts1Zuts2rP   r   r   r   +test_series_add_tz_mismatch_converts_to_utc  s$    

zDTestTimeSeriesArithmetic.test_series_add_tz_mismatch_converts_to_utcc                 C   s   t dddd}ttjdt||d}|d}d}tj	t
|d	 ||  W d    n1 sf0    Y  tj	t
|d	 ||  W d    n1 s0    Y  d S )
Nr  r2   r   r   r    ry   r   z0Cannot join tz-naive with tz-aware DatetimeIndexr   )r   r   r"   r#   r$   r   r&   r  r   r   	Exception)rK   r   rg   r   r   r   r   r   "test_series_add_aware_naive_raises  s    
&z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisesc                 C   sT   t tdd|d}tjjdd}|| }tg d|}t |}t|| d S )Nz
2012-01-01r\   )rA   unitr{   )days)z
2011-12-26z
2011-12-27z
2011-12-28)	r   r   r   offsetsZ
DateOffsetto_datetimeZas_unitrJ   rf   )rK   r&  rL   offsetrO   Zexp_dtirP   r   r   r   test_datetime_understood	  s    z1TestTimeSeriesArithmetic.test_datetime_understoodc                 C   s   t ddd}ttjdd|d}|dd  }| }dd |jD |_|| }|| }||dd   }|jd |_t	
|| t	
|| d S )	Nr   r?   r@   r    ry   r0   c                 S   s   g | ]}|  qS r   )r   ).0r-   r   r   r   
<listcomp>  r/   zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex.<locals>.<listcomp>)r   r   r"   r#   r$   r   r   rC   Z
_with_freqrJ   rf   )rK   r   r*   Zts_sliceZts2rO   r   rP   r   r   r   *test_align_date_objects_with_datetimeindex  s    zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rV   rW   rX   r#  r%  r+  r.  r   r   r   r   r    s   r  c                	   @   sT   e Zd Zejdeeej	e
eej	gejdddgdd Zdd Zd	d
 ZdS )TestNamePreservationboxflexTFc                 C   s  |}t td|d d}t td|d d}|jd}|dv }	d}
d }|ttfv r\|	r\t}||}|r|	rpd S t|||}n8tj	||
d |||}W d    n1 s0    Y  t
|t sJ |tt fv r|j|d	 u s|j|d	 ksJ n$|j|d u s|j|d ksJ d S )
Nr2   r   r^   rE   _)andZrandxorZrxororZrorzLLogical ops \(and, or, xor\) between Pandas objects and dtype-less sequencesr   r    )r   r6   rV   stripr   tupler  rH   rJ   r   
isinstancer
   rD   )rK   r1  r0  r   Zall_binary_operatorsrN   r   r   rD   Z
is_logicalr   warnrO   r   r   r   test_series_ops_name_retention&  s*    ("z3TestNamePreservation.test_series_ops_name_retentionc                 C   s  || }|j |j ksJ ||}|j |j ks2J ||d d  }|j |j ksRJ | }d|_ || }|j d u svJ ||}|j d u sJ g d}|dd |D  }|D ]T}| }t|||}|j |j ksJ | }d|_ t|||}|j d u sJ qd S )Nzsomething else)r9   r:   r;   r   r=   r<   modr>   c                 S   s   g | ]}d | qS )rF   r   )r,  rN   r   r   r   r-  ^  r/   zGTestNamePreservation.test_binop_maybe_preserve_name.<locals>.<listcomp>changed)rD   r;   r   r9   rH   )rK   r   rO   cpr   rN   rg   r   r   r   test_binop_maybe_preserve_nameK  s,    

z3TestNamePreservation.test_binop_maybe_preserve_namec                 C   s   |d }|j |j ksJ d S r   r^   )rK   r   rO   r   r   r   test_scalarop_preserve_namek  s    z0TestNamePreservation.test_scalarop_preserve_nameN)rV   rW   rX   r   r   r   r   r7  r"   r  r
   r   r   r:  r?  r@  r   r   r   r   r/  %  s
   # r/  c                   @   s"   e Zd Zejdddd ZdS )TestInplaceOperationsz)dtype1, dtype2, dtype_expected, dtype_mul))r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   c                 C   s   t dg|d}t dg|d}||7 }t dg|d}t|| ||8 }t dg|d}t|| ||9 }t dg|d}t|| d S )NrE   r3   r    r\   r   )rK   Zdtype1Zdtype2Zdtype_expectedZ	dtype_mulr   r   rP   r   r   r   test_series_inplace_opsq  s    z-TestInplaceOperations.test_series_inplace_opsN)rV   rW   rX   r   r   r   rB  r   r   r   r   rA  p  s
   	rA  c                 C   s  |}t |dk r$| tjjdd tj|jd< |d k}|jd rFJ |jd rTJ |d k}|jd sjJ |jd sxJ d |k}|jd rJ |jd rJ d |k}|jd sJ |jd sJ t	
|jdst|jtrRd}tjt|d d |k W d    n1 s0    Y  tjt|d |d k W d    n1 sF0    Y  nPd |k}|jd rjJ |jd rzJ |d k }|jd rJ |jd rJ d S )NrE   z%Test doesn't make sense on empty data)reasonr   MzInvalid comparisonr   )r&   Zapplymarkerr   r   Zxfailr"   rd   r   Ziatr   Zis_np_dtyper4   r8  r	   r   r   )r   Zseries_with_simple_indexrL   rO   r   r   r   r   test_none_comparison  s<    (*rE  c                  C   s   t tdtjjtdtdddggg ddd} t d	d
 tddD tjjtdddggddgdd}| jtjdgd d d d f  | }t g dtjjg dg ddd}t	
|| d S )N   abxyrE   r    )rG  rH  num)r   ry   c                 S   s   g | ]}d | qS )  r   )r,  rv   r   r   r   r-    r/   z;test_series_varied_multiindex_alignment.<locals>.<listcomp>r0   rI  rr   )rJ  i  i  i  ))rr   r-   rE   )rr   r-   r    )rr   rk   rE   )rr   rk   r    )r   r6   r   Z
MultiIndexZfrom_productr   locZ
IndexSlicefrom_tuplesrJ   rf   )r   r   rO   rP   r   r   r   'test_series_varied_multiindex_alignment  s&    $rM  c                  C   s2   t dgd d} t dgd }t| | d S )Nr    i'  r   rE   )r   ZrmodrJ   rf   )rO   rP   r   r   r   !test_rmod_consistent_large_series  s    rN  )/datetimer   r   r   decimalr   rI   numpyr"   r   Zpandas._libsr   Zpandas._libs.tslibsr   Zpandasr   r   r	   r
   r   r   r   r   r   Zpandas._testingZ_testingrJ   Zpandas.corer   Zpandas.core.computationr   r   Zpandas.core.computation.checkr   Zfixturer   r(   r)   r   r   r   r  r/  rA  rE  rM  rN  r   r   r   r   <module>   s8   (

u Y6  3FK0