U
    _{f~                     @   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)ZautouseparamsZidsc                 c   s    t j}| jt _| jV  |t _d S N)expr_MIN_ELEMENTSparam)requestr    r   b/var/www/html/Darija-Ai-API/env/lib/python3.8/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ddddddg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 r   r   xr   r   r   <lambda>6       z!TestSeriesFlexArithmetic.<lambda>c                 C   s   | d S N   r   r(   r   r   r   r*   6   r+   Fc                 C   s   | S r   r   r(   r   r   r   r*   7   r+   c                 C   s   | d d d S r,   r   r(   r   r   r   r*   7   r+   c                 C   s   | S 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   )tmZmakeFloatSeriesr(   r   r   r   r*   9   r+   opnameaddsubmulfloordivtruedivpowc                 C   s   t  d}|d |}|d |}|d }tt|}tt|}|||}	|||}
t |	|
 |rttd| }|||}	|||}
t |	|
 d S )Nr'   r      r-   r)r/   ZmakeTimeSeriesrenamegetattrr   operatorassert_almost_equal)selfr0   r'   ZtserseriesotherZcheck_reverseopZaltresultexpectedropr   r   r   test_flex_method_equivalence3   s    





z5TestSeriesFlexArithmetic.test_flex_method_equivalencec                    sV   G  fdddt  |}tt |} dddgdd}d|_||d}|jdksRt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   )r=   MySeriesr   r   _constructorX   s    zgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation.<locals>.MySeries._constructorN)__name__
__module____qualname__	_metadatapropertyrG   r   rE   r   r   rF   U   s   rF   r7   r-      testname*   )r   r:   r)   AssertionError)r=   Zall_arithmetic_operatorsr0   r@   mrA   r   rE   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   r7   rM      r.   r-   
fill_value)r   r    nanfillnar1   r/   assert_series_equal)r=   serexpresr   r   r   test_flex_add_scalar_fill_valuec   s    z8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer7   )r1   r2   r3   r6   r5   r4   r   r8   c                 C   s   t t||| S r   )r:   r;   )r)   yr@   r   r   r   r*   r   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   s  |j |j }||}||}t|}t|}g }tt|D ]}	tjdd ||	 r||	 r||tj	 W 5 Q R  qB|||||	  nV||	 r||	 r|tj	 W 5 Q R  qB||||	 | n||||	 ||	  W 5 Q R X qB| |||d}
t
||}t|
| d S )Nignore)allrV   )indexunionreindexr   ranger#   r    errstateappendrX   r   r/   rZ   )methr@   abrW   	exp_indexZamaskZbmask
exp_valuesirA   rB   r   r   r   _check_fillx   s,    

$
zDTestSeriesFlexArithmetic.test_operators_combine.<locals>._check_fill      ?g       @g      @r.   rb   r7   rM         @   rV   axis)r   )r   r    rX   aranger/   rZ   )	r=   r@   Zequiv_opfvrn   ri   rj   rA   r\   r   r   r   test_operators_combinev   s    
 $

z/TestSeriesFlexArithmetic.test_operators_combineN)rH   rI   rJ   pytestmarkparametrizerD   rT   r^   r   divr;   r5   Zrdivr   ZrtruedivZpairingsr@   rv   r:   ZlopZlequivrC   Zrequivrg   rw   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 5 Q R X d S )N1/1/2000z1/1/2010Afreqrp   r-   r7   z4Input has different freq=D from Period\(freq=A-DEC\)matchDend)how)pdZperiod_ranger   r    r!   randnr#   rX   ilocr/   rZ   r%   rx   raisesr   Zasfreq)r=   rngr'   rA   rB   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!rS   ZhellomZworldmc                 C   s(   t |}|| }t |}t|| d S r   r   r/   rZ   )r=   Z
target_addZinput_valueZexpected_valueri   rA   rB   r   r   r   test_string_addition   s    	z)TestSeriesArithmetic.test_string_additionc                 C   s   t dddtjgddddgd}t dtjdtjgddddgd}||}t||}t|d	 |d	  t|d |d  ||}t||}t|d	 |d	  t|d |d  d S )
Nr7   ri   rj   cdrp   r-   er   )r   r    rX   divmodr/   rZ   Zrdivmod)r=   ri   rj   rA   rB   r   r   r   test_divmod   s     



z TestSeriesArithmetic.test_divmodrb   N	   c              	   C   sD   t tdd}t d|d}d}tjt|d ||  W 5 Q R X d S )Nr7   
   foorp   z4not all arguments converted during string formattingr   )r   re   rx   r   	TypeError)r=   rb   s1s2r   r   r   r   test_series_integer_mod   s
    z,TestSeriesArithmetic.test_series_integer_modc                 C   sZ   t ddgddgd}t ddgddgd}|| }t ddtjgdddgd}t|| d S )Nr7   r-   rp   r         )r   r    rX   r/   rZ   )r=   r   r   rA   rB   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 sXtt|d shtd S )Nz1.3z2.3i  r7   r-   rp   r   )r   r   r   shiftr   rR   )r=   r[   rA   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.t||  }t|dksJtd S )N)rb   dtyper   )	r   r
   r    Zfloat64isnanra   rR   copyr#   )r=   datetime_seriesemptyrA   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   rP   )astypeintr   valuesrb   r/   rZ   )r=   r   Zint_tsaddedrB   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        rp   )r   r    Zint32r/   rZ   rX   )r=   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-1rM   r   periodsr   r-   r   rO   )r   r   r    rX   r   r   r   ZNaTr/   rZ   to_frame)r=   dtZdt2rB   rA   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|ksFt|j|ksTtd S )Nz
2016-01-01r   ZCET)r   tzUTCrp   )r   
tz_convertr   rb   rR   )r=   dtiZdti_utcr[   ser_utcr   r   r   test_alignment_doesnt_change_tz  s    
z4TestSeriesArithmetic.test_alignment_doesnt_change_tzc                 C   s   t ddddddg}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 )NZ3z53ZLoJGZN503r-   rp   r   rU   r   )
categoriesrq      )	r	   r   r   ZCategoricalIndexr   r    rX   r/   rZ   )r=   catser1ser2rA   rk   rl   rB   r   r   r   test_alignment_categorical#  s    
z/TestSeriesArithmetic.test_alignment_categoricalc                 C   s   dddddg}t tjdddd|d}t tjd	dd|d}|| }t ddddddgd}t|| t td
d	d|d}t tdd	d|d}|| }t tddddddgd}t|| d S )Nr-   rM   rU   r7   rr   Zint64r   rp   r.   z20130101 09:00:00r   Z20130101z9 hours)r   r    ru   r/   rZ   r   r   )r=   rb   r[   r?   rA   rB   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   r7   Int64Float64)r   r    rX   r   NAr/   rZ   r=   r   r   rA   rB   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   r.   rU   r   r   ro   r   )r   r   r   r    rX   r/   rZ   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)rM   r   )g      @r   c                 C   s`   t dd dgdd}|dd |g }t dd d| g|d}t|| dd |g| }t|| d S )Nr7   rM   r   r   r-   r   )r=   valr   r[   rA   rB   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dd dgdd}t| |dd dg }W 5 Q R X tdd dgdd}t|| t| dd dg| }W 5 Q R X t|| d S )Nr   TFbooleanr   )	nodeZcallspecidr   UserWarningr   r/   Zassert_produces_warningrZ   )r=   r   warningr[   rA   rB   r   r   r   %test_add_list_to_masked_array_booleana  s    z:TestSeriesArithmetic.test_add_list_to_masked_array_boolean)rH   rI   rJ   r   rx   ry   rz   r   r   re   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dddgdd Zdd Zejd	d
d
dd
gdfddd
dgdfd
d
dd
gdfd
d
d
d
gdfd
ddd
gdfd
dd
d
gdfgdd Zejdd
d
ddgddfddd
d
gddfd
d
ddgddfd
d
d
dgddfdddd
gddfddd
d
gddfgdd Z	dS )TestSeriesFlexComparisonrt   r   Nrb   c                 C   sN   t tjd}t tjd}t||j||d}|||}t|| d S )Nr   rs   )r   r    r!   r   r:   rH   r/   rZ   )r=   rt   comparison_opleftrightrA   rB   r   r   r   test_comparison_flex_basicx  s
    
z3TestSeriesFlexComparison.test_comparison_flex_basicc              	   C   sV   t tjd}t tjd}d}tjt|d t||j|dd W 5 Q R X d S )Nr   zNo axis named 1 for object typer   r7   rs   )	r   r    r!   r   rx   r   
ValueErrorr:   rH   )r=   r   r   r   r   r   r   r   test_comparison_bad_axis  s
    z1TestSeriesFlexComparison.test_comparison_bad_axisz
values, opFTeqneleltgegtc                 C   sZ   t dddgtdd}t dddgtdd}t|||}t |tdd}t|| d S )Nr7   rM   r-   abcrp   bcdabcdr   listr:   r/   rZ   )r=   r   r@   r   r   rA   rB   r   r   r   test_comparison_flex_alignment  s
    z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_valuer-   c                 C   s^   t dddgtdd}t dddgtdd}t||||d}t |tdd}t|| d S )	Nr7   rM   r-   r   rp   r   rV   r   r   )r=   r   r@   rW   r   r   rA   rB   r   r   r   #test_comparison_flex_alignment_fill  s
    z<TestSeriesFlexComparison.test_comparison_flex_alignment_fill)
rH   rI   rJ   rx   ry   rz   r   r   r   r   r   r   r   r   r   w  s2   

r   c                
   @   s(  e Zd Zdd Zejddddddd	gd
d Zejddddddd	gdd Zejddd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ejd"ed#d$d%ged&d'd(ed$d$d$ged)d'd(fed#d$d%ged&d'd(ed$d$d$d$ged*d'd(fgd+d, Zd-d. Zd/S )0TestSeriesComparisonc              	   C   s   t dddg}t ddg}d}tjt|d ||k  W 5 Q R X t ddg}t ddd	g}tjt|d ||k W 5 Q R X d S )
Nri   rj   r   z'only compare identically-labeled Seriesr   r7   r-   rM   rU   )r   rx   r   r   )r=   ri   rj   r   r   r   r    test_comparison_different_length  s    z5TestSeriesComparison.test_comparison_different_lengthr0   r   r   r   r   r   r   c                 C   sD   t dddgtdd}d}t|||j}td}||ks@td S )Nr7   rM   r-   rp   bool)r   re   r:   dtypesr    r   rR   )r=   r0   r[   constrA   rB   r   r   r   test_ser_flex_cmp_return_dtypes  s
    
z4TestSeriesComparison.test_ser_flex_cmp_return_dtypesc                 C   sR   t dddgtdd}|jd d }d}t|||j}td}||ksNtd S )Nr7   rM   r-   rp   r   r   )r   re   r   r:   r   r    r   rR   )r=   r0   r[   r   r   rA   rB   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sFt|d}tj|d	d
}t||d }|||}|j|d kst||d }t||d }|||}|j|d kst|t	j
t	jfkrtjdd|d d}t||d }|||}|j|d kst|t	j
t	jfkrj|d}	t|	|d }|||	}|j|d ksjtd S )Nz1949-06-07 03:00:00Hr.   r   )r   r   rP   r7   r-   z
US/CentralZinferr   )startr   rP   category)r   r   r9   rP   rR   tz_localizer   ZDatetimeIndexr   r;   r   r   Zinterval_ranger   )
r=   r   r   r@   r   r[   rA   ZtdiiiZcidxr   r   r   test_ser_cmp_result_names  s0    






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 5 Q R X t j|d d< t	|| t
dddg}t
d	d
d	g}t
d	d	d	g}t||k| t||k| d S )Nr   rM   r`   )invalidOri   rj   r   FT)r    r!   r   rX   r   Znangtrf   r   r/   r<   r   rZ   )r=   r   r   rA   rB   sr   r\   r   r   r   test_comparisons  s    z%TestSeriesComparison.test_comparisonsc              	   C   s  t tddd}t tddd}t dddgdd}t tddd}ttd}ttd}|dk rjt|dk|dk k std|k rt|dkd	 std|kd	 std|kd	 rt||k st||k rt|t|k st||k s
t||k st||k ||kk s8t||k ||kk sTt||k rft||k rxt||k rt||k rt||k st||k st||k rt||k rt||k ||kk s t||k ||kk st||k ||kk s8t||k ||kk sTtd
}tjt|d ||k  W 5 Q R X tjt|d ||k  W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X d S )Nr   r   r   objectri   rj   ccZacbr    can only compare equality or notr   )r   r   r	   ra   rR   rx   r   r   )r=   ri   rj   r   r   r   fr   r   r   r   test_categorical_comparisons  sP    z1TestSeriesComparison.test_categorical_comparisonsc              	   C   sD  t ttd}d}tjt|d |dk W 5 Q R X t ttddd}tjt|d |dk W 5 Q R X t ttddd}d}tjt|d |d	k  W 5 Q R X tjt|d |d	k W 5 Q R X tjt|d d	|k  W 5 Q R X tjt|d d	|k W 5 Q R X t|d	kt dddg t|d	kt dddg d S )
Nr   r   r   rj   F)ZorderedTz1Invalid comparison between dtype=category and strr   )r   r	   r   rx   r   r   r/   rZ   )r=   r   r   r   r   r   5test_unequal_categorical_comparison_raises_type_errorD  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)r7   r7   )r7   r-   FT)r   r   r   )r=   r   rA   rB   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 )Nr7   r-   TF)r   	frozensetr/   rZ   r=   r[   rA   rB   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	krh|
dt}n|
dt}t|| d S )	Nr}   r   r   r   r-   r.   TF)r   r   r   r    rX   Zdropnard   rb   r;   r   rY   r   r   r/   rZ   )r=   r   r[   r   rA   rB   r   r   r   "test_comparison_operators_with_nas  s    

z7TestSeriesComparison.test_comparison_operators_with_nasc                 C   sb   t dddddgdddddgtd}dddddg}t|jdk|sFtt|jdk |s^td S )	NrM   rU   r.   rr      r   TF)r   r   r/   ZequalContentsrb   rR   )r=   r'   rB   r   r   r   test_ne  s    "zTestSeriesComparison.test_nezleft, rightr7   r-   rM   ABCr)   r   ZABDZABCDc              	   C   s  |t k	r(d|j d}| }| }nd|j d}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k  W 5 Q R X tjt|d ||k  W 5 Q R X d S )Nz@Can only compare identically-labeled \(both index and columns\) z objectsz%Can only compare identically-labeled r   )r   rH   r   rx   r   r   )r=   r   r   Zframe_or_seriesr   r   r   r   test_comp_ops_df_compat  s&    
z,TestSeriesComparison.test_comp_ops_df_compatc                 C   s4   t dddg}|dk}t dddg}t|| d S )NZ	IntervalAZ	IntervalBZ	IntervalCTFr   r  r   r   r   $test_compare_series_interval_keyword  s    z9TestSeriesComparison.test_compare_series_interval_keywordN)rH   rI   rJ   r   rx   ry   rz   r   r   r   r   r  r  r  r  r  r	  r   r   r  r  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kstt|| d S )
N1/1/2011d   r   utc)r   r   r   Z   z
US/Easternrp   zEurope/Berlin)r   r    r!   r"   r   r   r   r   rb   r   r   r  rR   r/   rZ   )	r=   r   permr   r   rA   Zuts1Zuts2rB   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 5 Q R X tjt	|d ||  W 5 Q R X d S )	Nr  r   r   r   rp   r  z0Cannot join tz-naive with tz-aware DatetimeIndexr   )
r   r   r    r!   r   r#   r   rx   r   	Exception)r=   r   r[   r   r   r   r   r   "test_series_add_aware_naive_raises  s    
z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisesc                 C   sJ   t tddd}tjjdd}|| }t tdddg}t|| d S )	Nz
2012-01-01rM   r   rr   )daysz
2011-12-26z
2011-12-27z
2011-12-28)r   r   r   offsetsZ
DateOffsetto_datetimer/   rZ   )r=   r>   offsetrA   rB   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   rp   r.   c                 S   s   g | ]}|  qS r   )r   ).0r)   r   r   r   
<listcomp>  s     zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex.<locals>.<listcomp>)
r   r   r    r!   r   r   rb   Z
_with_freqr/   rZ   )r=   r   r'   Zts_sliceZts2rA   r   rB   r   r   r   *test_align_date_objects_with_datetimeindex  s    zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rH   rI   rJ   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k}	||}|rd|	rTd S t|||}
n
|||}
t|
t s|t|tt fkr|
j|d ks|
j|d kstn |
j|d ks|
j|d kstd S )Nr   r   rO   r7   _)andZrandxorZrxororZrorr-   )	r   re   rH   stripr:   
isinstancerR   r
   rP   )r=   r!  r   r   Zall_binary_operatorsr@   r   r   rP   Z
is_logicalrA   r   r   r   test_series_ops_name_retention  s    
"z3TestNamePreservation.test_series_ops_name_retentionc                 C   s  || }|j |j kst||}|j |j ks2t||d d  }|j |j ksRt| }d|_ || }|j d ksvt||}|j d kstddddddd	d
g}|dd |D  }|D ]T}| }t|||}|j |j kst| }d|_ t|||}|j d kstqd S )Nzsomething elser1   r2   r3   r{   r5   r4   modr6   c                 S   s   g | ]}d | qS )r8   r   )r  r@   r   r   r   r  G  s     zGTestNamePreservation.test_binop_maybe_preserve_name.<locals>.<listcomp>changed)rP   rR   r3   r   r1   r:   )r=   r   rA   cpr   r@   r[   r   r   r   test_binop_maybe_preserve_name4  s,    

z3TestNamePreservation.test_binop_maybe_preserve_namec                 C   s   |d }|j |j kstd S r,   )rP   rR   )r=   r   rA   r   r   r   test_scalarop_preserve_nameT  s    z0TestNamePreservation.test_scalarop_preserve_nameN)rH   rI   rJ   rx   ry   rz   r   tupler    arrayr
   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 )Nr7   r   r-   rM   r   )r=   Zdtype1Zdtype2Zdtype_expectedZ	dtype_mulr   r   rB   r   r   r   test_series_inplace_opsZ  s    z-TestInplaceOperations.test_series_inplace_opsN)rH   rI   rJ   rx   ry   rz   r2  r   r   r   r   r1  Y  s
   	r1  c              	   C   sv  |}t |dk r&| jtjjdd tj|jd< |d k}|j	d rHt
|j	d rVt
|d k}|j	d slt
|j	d szt
d |k}|j	d rt
|j	d rt
d |k}|j	d st
|j	d st
t|jst|jr"d}tjt|d d |k W 5 Q R X tjt|d |d k W 5 Q R X nPd |k}|j	d r:t
|j	d rJt
|d k }|j	d rbt
|j	d rrt
d S )Nr7   z%Test doesn't make sense on empty data)reasonr   zInvalid comparisonr   )r#   r   Z
add_markerrx   ry   Zxfailr    rX   r   ZiatrR   r   r   r   r   r   )r   Zseries_with_simple_indexr>   rA   r   r   r   r   test_none_comparisonu  s<    r4  c                  C   s   t tdtjjtdtdddggdddg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 ddddgtjjddddgdddgdd}t	
|| d S )N   abZxyr7   r-   num)r   rp   c                 S   s   g | ]}d | qS )  r   )r  rm   r   r   r   r    s     z;test_series_varied_multiindex_alignment.<locals>.<listcomp>r.   ri   r8  i  i  i  )r)   r7   ri   )r)   r-   ri   )r_   r7   ri   )r_   r-   ri   )r   re   r   Z
MultiIndexZfrom_productr   locZ
IndexSlicefrom_tuplesr/   rZ   )r   r   rA   rB   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 )Nr-   i'  r   r7   )r   Zrmodr/   rZ   )rA   rB   r   r   r   !test_rmod_consistent_large_series  s    r<  )0datetimer   r   r   decimalr   r;   numpyr    rx   Zpandas._libs.tslibsr   Zpandas.core.dtypes.commonr   r   Zpandasr   r	   r
   r   r   r   r   r   Zpandas._testingZ_testingr/   Zpandas.corer   r   Zpandas.core.computationr   r   Zpandas.core.computation.checkr   Zfixturer   r%   r&   r|   r   r   r  r  r1  r4  r;  r<  r   r   r   r   <module>   s8   $	
m Y6  /=B0