o
    d~                     @   sZ  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mZ d dlZd dlmZmZmZmZmZmZmZ d dlmZ d dlmZ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)IncompatibleFrequency)is_datetime64_dtypeis_datetime64tz_dtype)CategoricalIndexSeries	Timedeltabdate_range
date_rangeisna)nanopsops)expressions)NUMEXPR_INSTALLEDTi@B numexprpython)autouseparamsidsc                 c   s"    t j}| jt _| jV  |t _d S N)expr_MIN_ELEMENTSparam)requestr    r   h/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/series/test_arithmetic.pyswitch_numexpr_min_elements&   s
   
r    c                 C   s   |  tjt| S r   )takenprandom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 ]+Z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 qPejdedd ZdS )TestSeriesFlexArithmetictsc                 C      | S r   r   xr   r   r   <lambda>6       z!TestSeriesFlexArithmetic.<lambda>c                 C   s   | d S N   r   r+   r   r   r   r-   6       Fc                 C   r*   r   r   r+   r   r   r   r-   7   r.   c                 C   s   | d d d S r/   r   r+   r   r   r   r-   7   s    c                 C   r*   r   r   r+   r   r   r   r-   8   r.   c                 C   s   dS )N   r   r+   r   r   r   r-   8   r.   Tc                 C   s   t  S r   )tmmakeFloatSeriesr+   r   r   r   r-   9   r1   opname)addsubmulfloordivtruedivpowc                 C   s   t  d}|d |}|d |}|d }tt|}tt|}|||}	|||}
t |	|
 |rLttd| }|||}	|||}
t |	|
 d S d S )Nr)   r      r0   r)r3   makeTimeSeriesrenamegetattrr   operatorassert_almost_equal)selfr5   r)   tserseriesothercheck_reverseopaltresultexpectedropr   r   r   test_flex_method_equivalence3   s   





z5TestSeriesFlexArithmetic.test_flex_method_equivalencec                    sT   G  fdddt  |}tt |} g ddd}d|_||d}|jdks(J 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   )rC   MySeriesr   r   _constructorX   s   zgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeries._constructorN)__name__
__module____qualname__	_metadatapropertyrP   r   rN   r   r   rO   U   s    rO   r<   r0      testname*   r<   )r   r@   r,   )rC   all_arithmetic_operatorsr5   rH   mrJ   r   rN   r   /test_flex_method_subclass_metadata_preservationS   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   r<   rW      r2   r0   
fill_value)r   r"   nanfillnar6   r3   assert_series_equal)rC   serexpresr   r   r   test_flex_add_scalar_fill_valuec   s   z8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer<   )r6   r7   r8   r;   r:   r9   r   r=   c                 C   s   t t||| S r   )r@   rA   )r,   yrH   r   r   r   r-   r   s    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   s8  |j |j }||}||}t|}t|}g }tt|D ]f}	tjddV ||	 rK||	 r@|tj	 	 W d    q!|||||	  n-||	 rl||	 ra|tj	 	 W d    q!||||	 | n||||	 ||	  W d    n1 sw   Y  q!| |||d}
t
||}t|
| d S )Nignore)allr`   )indexunionreindexr   ranger%   r"   errstateappendrb   r   r3   rd   )methrH   abra   	exp_indexamaskbmask
exp_valuesirJ   rK   r   r   r   _check_fillx   s6   



zDTestSeriesFlexArithmetic.test_operators_combine.<locals>._check_fill      ?g       @g      @r2   rl   r<   rW         @   r`   axis)r   )r   r"   rb   aranger3   rd   )	rC   rH   equiv_opfvrz   rs   rt   rJ   rf   r   r   r   test_operators_combinev   s   
 $

z/TestSeriesFlexArithmetic.test_operators_combineN)rQ   rR   rS   pytestmarkparametrizerM   r^   rh   r   divrA   r:   rdivr   rtruedivpairingsrH   r   r@   loplequivrL   requivrq   r   r   r   r   r   r(   2   s4    	

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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    d S 1 s_w   Y  d S )N1/1/2000z1/1/2010Afreqr|   r0   r<   z4Input has different freq=D from Period\(freq=A-DEC\)matchDend)how)pdperiod_ranger   r"   r#   randnr%   rb   ilocr3   rd   r'   r   raisesr   asfreq)rC   rngr)   rJ   rK   msgr   r   r   !test_add_series_with_period_index   s   "z6TestSeriesArithmetic.test_add_series_with_period_indexz%target_add,input_value,expected_value!helloworldzhello!zworld!r]   hellomworldmc                 C   s(   t |}|| }t |}t|| d S r   r   r3   rd   )rC   
target_addinput_valueexpected_valuers   rJ   rK   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 )Nr<   )rs   rt   cdr|   r0   )rs   rt   r   er   )r   r"   rb   divmodr3   rd   rdivmod)rC   rs   rt   rJ   rK   r   r   r   test_divmod   s   



z TestSeriesArithmetic.test_divmodrl   N	   c                 C   sZ   t tdd}t d|d}d}tjt|d ||  W d    d S 1 s&w   Y  d S )Nr<   
   foor|   z4not all arguments converted during string formattingr   )r   ro   r   r   	TypeError)rC   rl   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 )Nr<   r0   r|   r         )r<   r<   r0   )r   r"   rb   r3   rd   )rC   r   r   rJ   rK   r   r   r   test_add_with_duplicate_index   s
   z2TestSeriesArithmetic.test_add_with_duplicate_indexc                 C   sl   t tdtdgtdddtdddgd}||d }|d| }t|d s,J t|d s4J d S )Nz1.3z2.3i  r<   r0   r|   r   )r   r   r   shiftr   )rC   re   rJ   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s%J d S )N)rl   dtyper   )r   r
   r"   float64isnanrk   copyr%   )rC   datetime_seriesemptyrJ   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)   rl   rZ   )astypeintr   valuesrl   r3   rd   )rC   r   int_tsaddedrK   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 )Nr   r,   g        r|   )r   r"   int32r3   rd   rb   )rC   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-1rW   r   periodsr   r0   r   rY   )r   r   r"   rb   r   r   r   NaTr3   rd   to_frame)rC   dtdt2rK   rJ   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 s#J |j|u s*J d S )Nz
2016-01-01r   CET)r   tzUTCr|   )r   
tz_convertr   rl   )rC   dtidti_utcre   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   N503r0   r|   r   r   r_   r   r   r   )
categoriesr}      )	r	   r   r   CategoricalIndexr   r"   rb   r3   rd   )rC   catser1ser2rJ   ru   rx   rK   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)r0   r0   rW   rW   r_   r<   r~   int64r   r|   r2   z20130101 09:00:00r   20130101z9 hours)r   r"   r   r3   rd   r   r   )rC   rl   re   rF   rJ   rK   r   r   r   $test_arithmetic_with_duplicate_index1  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   floatr   r<   Int64Float64)r   r"   rb   r   NAr3   rd   rC   r   r   rJ   rK   r   r   r   'test_masked_and_non_masked_propagate_naC  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   r2   r_   r   r   r{   r   )r   r   r   r"   rb   r3   rd   r   r   r   r   +test_mask_div_propagate_na_for_non_na_dtypeK  s   z@TestSeriesArithmetic.test_mask_div_propagate_na_for_non_na_dtypez
val, dtype)rW   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)r<   NrW   r   r   r<   r0   rW   r   )rC   valr   re   rJ   rK   r   r   r   test_add_list_to_masked_arrayV  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 s+w   Y  tg ddd}t|| t| g d| }W d    n1 sTw   Y  t|| d S )Nr   )TNFbooleanr   )TNT)	nodecallspecidr   UserWarningr   r3   assert_produces_warningrd   )rC   r   warningre   rJ   rK   r   r   r   %test_add_list_to_masked_array_booleana  s    z:TestSeriesArithmetic.test_add_list_to_masked_array_boolean)rQ   rR   rS   r   r   r   r   r   r   ro   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s4    



	

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   Nrl   c                 C   sN   t tjd}t tjd}t||j||d}|||}t|| d S )Nr   r   )r   r"   r#   r   r@   rQ   r3   rd   )rC   r   comparison_opleftrightrJ   rK   r   r   r   test_comparison_flex_basicx  s
   
z3TestSeriesFlexComparison.test_comparison_flex_basicc                 C   sl   t tjd}t tjd}d}tjt|d t||j|dd W d    d S 1 s/w   Y  d S )Nr   zNo axis named 1 for object typer   r<   r   )	r   r"   r#   r   r   r   
ValueErrorr@   rQ   )rC   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r<   rW   r0   abcr|   r0   r0   r0   bcdabcdr   listr@   r3   rd   )rC   r   rH   r  r  rJ   rK   r   r   r   test_comparison_flex_alignment  s
   z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_value)FFTTr0   )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  r|   r  r  r`   r  r  )rC   r   rH   ra   r  r  rJ   rK   r   r   r   #test_comparison_flex_alignment_fill  s
   z<TestSeriesFlexComparison.test_comparison_flex_alignment_fillN)
rQ   rR   rS   r   r   r   r  r  r  r  r   r   r   r   r  w  s4    







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 s$w   Y  t ddg}t g d}tjt|d ||k W d    d S 1 sLw   Y  d S )	Nrs   rt   r   rt   rs   z'only compare identically-labeled Seriesr   r<   r0   )r0   rW   r_   )r   r   r   r  )rC   rs   rt   r   r   r   r    test_comparison_different_length  s   

"z5TestSeriesComparison.test_comparison_different_lengthr5   )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  rW   r|   r0   bool)r   ro   r@   dtypesr"   r   )rC   r5   re   constrJ   rK   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s&J d S )Nr  rW   r|   r   r0   r  )r   ro   r   r@   r  r"   r   )rC   r5   re   r   r  rJ   rK   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   sj  |}t ddd|d d}t||d }|||}|j|d ks#J |d}tj|d	d
}t||d }|||}|j|d ksFJ ||d }t||d }|||}|j|d ksdJ |tj	tj
fv rtjdd|d d}t||d }|||}|j|d ksJ |tj	tj
fv r|d}	t|	|d }|||	}|j|d ksJ d S d S )Nz1949-06-07 03:00:00Hr2   r   )r   r   rZ   r<   r0   z
US/Centralinferr   )startr   rZ   category)r   r   r?   rZ   tz_localizer   DatetimeIndexr   rA   r  r	  interval_ranger   )
rC   r  r  rH   r   re   rJ   tdiiicidxr   r   r   test_ser_cmp_result_names  s4   






z.TestSeriesComparison.test_ser_cmp_result_namesc                 C   s   t jd}t jd}t j|d d< t||}t jdd ||kd}W d    n1 s1w   Y  t j|d d< t	|| t
g d}t
g d}t
g d}t||k| t||k| d S )	Nr   rW   rj   )invalidOr  )FTFFFF)r"   r#   r   rb   r   nangtrp   r   r3   rB   r   rd   )rC   r  r  rJ   rK   sr   rf   r   r   r   test_comparisons  s   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 r4J |dk|dk k sAJ d|k rIJ |dkd sQJ d|kd sYJ d|kd raJ ||k siJ ||k rqJ |t|k s{J ||k sJ ||k sJ ||k ||kk sJ ||k ||kk sJ ||k rJ ||k rJ ||k rJ ||k rJ ||k sJ ||k sJ ||k rJ ||k rJ ||k ||kk sJ ||k ||kk sJ ||k ||kk sJ ||k ||kk sJ d	}tjt|d
 ||k  W d    n	1 s4w   Y  tjt|d
 ||k  W d    n	1 sPw   Y  tjt|d
 ||k W d    n	1 slw   Y  tjt|d
 ||k W d    d S 1 sw   Y  d S )Nr  r%  r   object)rs   rt   ccacbrs   r    can only compare equality or notr   )r   r  r	   rk   r   r   r   )rC   rs   rt   r   r   r   fr   r   r   r   test_categorical_comparisons  sX   



$z1TestSeriesComparison.test_categorical_comparisonsc                 C   s  t ttd}d}tjt|d |dk W d    n1 s w   Y  t ttddd}tjt|d |dk W d    n1 sEw   Y  t ttddd}d}tjt|d |d	k  W d    n1 slw   Y  tjt|d |d	k W d    n1 sw   Y  tjt|d d	|k  W d    n1 sw   Y  tjt|d d	|k W d    n1 sw   Y  t|d	kt g d
 t|d	kt g d d S )Nr  r6  r   rt   F)orderedTz1Invalid comparison between dtype=category and strr   r/  )TTT)r   r	   r  r   r   r   r3   rd   )rC   r   r   r   r   r   5test_unequal_categorical_comparison_raises_type_errorD  s2   





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)r<   r<   )r<   r0   FT)r   r   r   )rC   r1  rJ   rK   r   r   r   test_comparison_tuplesc  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 )Nr<   r0   TF)r   	frozensetr3   rd   rC   re   rJ   rK   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}|tj	u r4|
dt}n|
dt}t|| d S )	Nr   r   r   r   r0   r2   TF)r   r   r3  r"   rb   dropnarn   rl   rA   r	  rc   r   r  r3   rd   )rC   r  re   r   rJ   rK   r   r   r   "test_comparison_operators_with_nas  s   

z7TestSeriesComparison.test_comparison_operators_with_nasc                 C   sP   t g dg dtd}g d}t|jdk|sJ t|jdk |s&J d S )N)rW   r_   r2   r~      r   )TTFTTr2   )r   r   r3   equalContentsrl   )rC   r)   rK   r   r   r   test_ne  s   zTestSeriesComparison.test_nezleft, rightrV   ABCr,   r   r  ABD)r0   r0   r0   r0   ABCDc                 C   s  |t urd|j d}| }| }nd|j d}tjt|d ||k W d    n1 s1w   Y  tjt|d ||k W d    n1 sLw   Y  tjt|d ||k W d    n1 sgw   Y  tjt|d ||k W d    n1 sw   Y  tjt|d ||k  W d    n1 sw   Y  tjt|d ||k  W d    d S 1 sw   Y  d S )Nz@Can only compare identically-labeled \(both index and columns\) z objectsz%Can only compare identically-labeled r   )r   rQ   r   r   r   r  )rC   r  r  frame_or_seriesr   r   r   r   test_comp_ops_df_compat  s6   






"z,TestSeriesComparison.test_comp_ops_df_compatc                 C   s0   t g d}|dk}t g d}t|| d S )N)	IntervalA	IntervalB	IntervalCrI  )TFFr   r=  r   r   r   $test_compare_series_interval_keyword  s   z9TestSeriesComparison.test_compare_series_interval_keywordN)rQ   rR   rS   r  r   r   r   r  r  r,  r2  r8  r:  r;  r>  r@  rC  r   r  rH  rL  r   r   r   r   r    s:    

	
%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 }ttjd||dd}tjdd d }ttjd||d	d}|| }|d}|d}|| }|jj	t
ju sWJ t|| d S )
N1/1/2011d   r"  utc)r   r   r   Z   z
US/Easternr|   zEurope/Berlin)r   r"   r#   r$   r   r   r!   r   rl   r   r   rP  r3   rd   )	rC   r   permr   r   rJ   uts1uts2rK   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t||d}|d}d}tjt	|d ||  W d    n1 s0w   Y  tjt	|d ||  W d    d S 1 sLw   Y  d S )	NrN  r   r"  r   r|   rP  z0Cannot join tz-naive with tz-aware DatetimeIndexr   )
r   r   r"   r#   r   r%   r&  r   r   	Exception)rC   r   re   r   r   r   r   r   "test_series_add_aware_naive_raises  s   


"z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisesc                 C   sH   t tddd}tjjdd}|| }t tg d}t|| d S )Nz
2012-01-01rW   r   r~   )days)z
2011-12-26z
2011-12-27z
2011-12-28)r   r   r   offsets
DateOffsetto_datetimer3   rd   )rC   rE   offsetrJ   rK   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 |jD |_|| }|| }||dd   }|jd |_t	|| t	|| d S )Nr      r   r|   r2   c                 S   s   g | ]}|  qS r   )r   ).0r,   r   r   r   
<listcomp>      zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex.<locals>.<listcomp>)
r   r   r"   r#   r   r   rl   
_with_freqr3   rd   )rC   r   r)   ts_slicets2rJ   r   rK   r   r   r   *test_align_date_objects_with_datetimeindex  s   zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rQ   rR   rS   rU  rW  r]  re  r   r   r   r   rM    s
    	rM  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 }	||}|r2|	r*d S t|||}
n|||}
t|
t s>J |tt fv rX|
j|d u sT|
j|d ksVJ d S d S |
j|d u sh|
j|d ksjJ d S d S )Nr   r   rY   r<   _)andrandxorrxororrorr0   )r   ro   rQ   stripr@   
isinstancer
   rZ   )rC   rh  rg  r  all_binary_operatorsrH   r  r  rZ   
is_logicalrJ   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sJ ||d d  }|j |j ks)J | }d|_ || }|j d u s;J ||}|j d u sGJ g d}|dd |D  }|D ]*}| }t|||}|j |j kskJ | }d|_ t|||}|j d u sJ qVd S )Nzsomething else)r6   r7   r8   r   r:   r9   modr;   c                 S   s   g | ]}d | qS )r=   r   )r_  rH   r   r   r   r`  G  ra  zGTestNamePreservation.test_binop_maybe_preserve_name.<locals>.<listcomp>changed)rZ   r8   r   r6   r@   )rC   r   rJ   cpr   rH   re   r   r   r   test_binop_maybe_preserve_name4  s.   

z3TestNamePreservation.test_binop_maybe_preserve_namec                 C   s   |d }|j |j ksJ d S r/   rY   )rC   r   rJ   r   r   r   test_scalarop_preserve_nameT  s   z0TestNamePreservation.test_scalarop_preserve_nameN)rQ   rR   rS   r   r   r   r  tupler"   arrayr
   r   r   rt  ry  rz  r   r   r   r   rf    s     rf  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 )Nr<   r   r0   rW   r   )rC   dtype1dtype2dtype_expected	dtype_mulr   r   rK   r   r   r   test_series_inplace_opsZ  s   z-TestInplaceOperations.test_series_inplace_opsN)rQ   rR   rS   r   r   r   r  r   r   r   r   r}  Y  s    	r}  c                 C   s  |}t |dk r| jtjjdd tj|jd< |d k}|j	d r$J |j	d r+J |d k}|j	d s6J |j	d s=J d |k}|j	d rHJ |j	d rOJ d |k}|j	d sZJ |j	d saJ t
|jskt|jrd}tjt|d d |k W d    n1 sw   Y  tjt|d |d k W d    d S 1 sw   Y  d S d |k}|j	d rJ |j	d rJ |d k }|j	d rJ |j	d rJ d S )Nr<   z%Test doesn't make sense on empty data)reasonr   zInvalid comparisonr   )r%   r   
add_markerr   r   xfailr"   rb   r   iatr   r   r   r   r   )r   series_with_simple_indexrE   rJ   r   r   r   r   test_none_comparisonu  s@   

"r  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xyr<   r0   )r  r  num)r  r|   c                 S   s   g | ]}d | qS )  r   )r_  ry   r   r   r   r`    ra  z;test_series_varied_multiindex_alignment.<locals>.<listcomp>r2   r  rs   )r  i  i  i  ))r,   r<   rs   )r,   r0   rs   )ri   r<   rs   )ri   r0   rs   )r  r  r  )r   ro   r   
MultiIndexfrom_productr  loc
IndexSlicefrom_tuplesr3   rd   )r   r   rJ   rK   r   r   r   'test_series_varied_multiindex_alignment  s&   $r  c                  C   s2   t dgd d} t dgd }t| | d S )Nr0   i'  r   r<   )r   rmodr3   rd   )rJ   rK   r   r   r   !test_rmod_consistent_large_series  s   r  )0datetimer   r   r   decimalr   rA   numpyr"   r   pandas._libs.tslibsr   pandas.core.dtypes.commonr   r   pandasr   r	   r
   r   r   r   r   r   pandas._testing_testingr3   pandas.corer   r   pandas.core.computationr   r   pandas.core.computation.checkr   fixturer    r'   r(   r   r  r  rM  rf  r}  r  r  r  r   r   r   r   <module>   s:    $	
m Y6  /=B0