U W7bu@s<ddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl Z ddl mZmZmZmZmZmZmZddlmZddlmZmZddlmZejdd dd gd d gd ddZddZ GdddZ!GdddZ"GdddZ#GdddZ$GdddZ%GdddZ&GdddZ'd d!Z(d"d#Z)dS)$) timedeltaN)IncompatibleFrequency)is_datetime64_dtypeis_datetime64tz_dtype) CategoricalIndexSeries Timedelta bdate_range date_rangeisna)nanopsops) expressionsTmodulei@BZnumexprpython)ZautouseZscopeparamsidsccs tj}|jt_|jV|t_dSN)expr _MIN_ELEMENTSparam)requestrrG/tmp/pip-unpacked-wheel-rhjceu4d/pandas/tests/series/test_arithmetic.pyswitch_numexpr_min_elements!srcCs|tjt|Sr)takenprandom permutationlen)objrrr_permute+sr"c @seZdZejddddddfdddddfddd dd fd dd dd fgejd d dddddgddZddZddZe j e j dfe j ejdfgZdD]VZdZee eZee eZee deZefddZeeeefeeeefqejdedd Zd!S)"TestSeriesFlexArithmetictscCs|Srrxrrr3z!TestSeriesFlexArithmetic.cCs|dSNrr%rrrr'3r(FcCs|Srrr%rrrr'4r(cCs|dddSr)rr%rrrr'4r(cCs|Srrr%rrrr'5r(cCsdS)Nrr%rrrr'5r(TcCstSr)tmZmakeFloatSeriesr%rrrr'6r(opnameaddsubmulfloordivtruedivpowc Cstd}|d|}|d|}|d}tt|}tt|}|||} |||} t| | |rttd|} | ||} |||} t| | dS)Nr$rr*r)r,ZmakeTimeSeriesrenamegetattrroperatorassert_almost_equal) selfr-r$ZtserseriesotherZ check_reverseopZaltresultexpectedroprrrtest_flex_method_equivalence0s         z5TestSeriesFlexArithmetic.test_flex_method_equivalencecsVGfdddt|}tt|}dddgdd}d|_||d}|jdksRtdS) Ncs"eZdZdgZefddZdS)zZTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation..MySeriesr&csSrr)r:MySeriesrr _constructorUszgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation..MySeries._constructorN)__name__ __module__ __qualname__ _metadatapropertyrDrrBrrrCRsrCr4r*testname*)rr7r&AssertionError)r:Zall_arithmetic_operatorsr-r=mr>rrBr/test_flex_method_subclass_metadata_preservationPs  zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationcCsDtddtjdddg}|dd}|jddd}t||dS)Nrr4rJr+r* fill_value)rrnanfillnar.r,assert_series_equal)r:serexpresrrrtest_flex_add_scalar_fill_value`sz8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer4)r.r/r0r3r2r1rr5cCstt|||Sr)r7r8)r&yr=rrrr'or(zop, equiv_op, fvc Csddd}ttjdddtjgtdd}ttjd tjd tjd gtd d}|||}|||}t||||||||d |||dddS)Nrc Ss|j|j}||}||}t|}t|}g}tt|D]} tjdd|| r|| r||tj W5QRqB||||| nV|| r|| r|tj W5QRqB|||| |n|||| || W5QRXqB||||d} t ||} t | | dS)Nignore)allrS) indexunionreindexr ranger rerrstateappendrUrr,rW) methr=abrTZ exp_indexZamaskZbmaskZ exp_valuesir>r?rrr _check_fillus,      $ zDTestSeriesFlexArithmetic.test_operators_combine.._check_fillg?g@g@r+r_r4rJg@rSaxis)r)rrrUaranger,rW) r:r=Zequiv_opfvrirfrgr>rYrrrtest_operators_combiness  $   z/TestSeriesFlexArithmetic.test_operators_combineN)rErFrGpytestmark parametrizerArQr[rdivr8r2ZrdivrZrtruedivZpairingsr=ror7ZlopZlequivr@Zrequivrdrprrrrr#/s4      r#c@seZdZddZejddddgddgfd ddgd d gfgd d ZddZejdde dgddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$ZdS)%TestSeriesArithmeticc Cstjdddd}ttjt||d}||ddd}||}tj|jddd<t |||t |ddd}t ||d}t j t|d ||jd d d W5QRXdS) N1/1/2000z1/1/2010Afreqrjr*r4z4Input has different freq=D from Period\(freq=A-DEC\)matchDend)how)pdZ period_rangerrrrandnr rUilocr,rWr"rqraisesrZasfreq)r:rngr$r>r?msgrrr!test_add_series_with_period_indexs  z6TestSeriesArithmetic.test_add_series_with_period_indexz%target_add,input_value,expected_value!ZhelloZworldzhello!zworld!rPZhellomZworldmcCs(t|}||}t|}t||dSrrr,rW)r:Z target_addZ input_valueZexpected_valuerfr>r?rrrtest_string_additions z)TestSeriesArithmetic.test_string_additioncCstdddtjgddddgd}tdtjdtjgddddgd}||}t||}t|d |d t|d|d||}t||}t|d |d t|d|ddS) Nr4rfrgcdrjr*er)rrrUdivmodr,rWZrdivmod)r:rfrgr>r?rrr test_divmods     z TestSeriesArithmetic.test_divmodr_N c CsDttdd}td|d}d}tjt|d||W5QRXdS)Nr4 foorjz4not all arguments converted during string formattingrz)rrbrqr TypeError)r:r_s1s2rrrrtest_series_integer_mods  z,TestSeriesArithmetic.test_series_integer_modcCsZtddgddgd}tddgddgd}||}tddtjgdddgd}t||dS)Nr4r*rjr )rrrUr,rW)r:rrr>r?rrrtest_add_with_duplicate_indexs z2TestSeriesArithmetic.test_add_with_duplicate_indexcCsddlm}ddlm}t|d|dg|ddd|dddgd }||d}|d|}t|dsptt|dstdS) Nrdate)Decimalz1.3z2.3ir4r*rj)datetimerdecimalrrshiftr rO)r:rrrXr>result2rrrtest_add_na_handlings  z)TestSeriesArithmetic.test_add_na_handlingcCsNtgtgtjd}||}t|s.t||}t|dksJtdS)N)r_dtyper) rrrZfloat64isnanr^rOcopyr )r:datetime_seriesemptyr>rrrtest_add_corner_casess  z*TestSeriesArithmetic.test_add_corner_casescCsZ|tdd}||}t|jdd|j|jdddd}t|dd|dS)Nr$r_rM)astypeintrvaluesr_r,rW)r:rZint_tsaddedr?rrrtest_add_float_plus_ints z,TestSeriesArithmetic.test_add_float_plus_intcCs>tggtjd}tddi}t||ttjgdgddS)Nrr&grj)rrZint32r,rWrU)r:rrrrrtest_mul_empty_int_corner_cases z3TestSeriesArithmetic.test_mul_empty_int_corner_casecCsttdddd}tj|jd<|ddd}ttdtdtjg}||}t ||t|dd}| | d}t ||dS) Nz2012-1-1rJr|periodsryr*rrL) rr rrUrrrZNaTr,rWto_frame)r:dtZdt2r?r>rrrtest_sub_datetimelike_aligns   z0TestSeriesArithmetic.test_sub_datetimelike_aligncCsXtdddd}|d}td|d}td|d}|||j|ksFt|j|ksTtdS)Nz 2016-01-01rZCET)rtzUTCrj)r tz_convertrr_rO)r:dtiZdti_utcrXser_utcrrrtest_alignment_doesnt_change_tzs   z4TestSeriesArithmetic.test_alignment_doesnt_change_tzcCsdddddg}ttjdddd|d}ttjd dd|d}||}tddddddgd}t||ttd d d |d}ttd d d |d}||}ttd dddddgd}t||dS)Nr*rJrRr4rkZint64rrjr+z20130101 09:00:00rZ20130101z9 hours)rrrnr,rWr r )r:r_rXr<r>r?rrr$test_arithmetic_with_duplicate_index#s z9TestSeriesArithmetic.test_arithmetic_with_duplicate_index)rErFrGrrqrrrsrrrbrrrrrrrrrrrrrrus&      ruc @seZdZejddddgddZddZejd d d d d gd fd d d d gd fd d d d gdfd d d d gdfd d d d gdfd d d d gdfgddZejdd d d d gd dfd d d d gd dfd d d d gddfd d d d gddfd d d d gddfd d d d gddfgddZ dS)TestSeriesFlexComparisonrmrNr_cCsNttjd}ttjd}t||j||d}|||}t||dS)Nrrl)rrrrr7rEr,rW)r:rm comparison_opleftrightr>r?rrrtest_comparison_flex_basic<s  z3TestSeriesFlexComparison.test_comparison_flex_basicc CsVttjd}ttjd}d}tjt|dt||j|ddW5QRXdS)NrzNo axis named 1 for object typerzr4rl) rrrrrqr ValueErrorr7rE)r:rrrrrrrtest_comparison_bad_axisDs z1TestSeriesFlexComparison.test_comparison_bad_axisz values, opFTeqneleltgegtcCsZtdddgtdd}tdddgtdd}t|||}t|tdd}t||dS)Nr4rJr*abcrjbcdabcdrlistr7r,rW)r:rr=rrr>r?rrrtest_comparison_flex_alignmentLs z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_valuer*cCs^tdddgtdd}tdddgtdd}t||||d}t|tdd}t||dS) Nr4rJr*rrjrrSrr)r:rr=rTrrr>r?rrr#test_comparison_flex_alignment_fill^s zr?rrrtest_ser_flex_cmp_return_dtypes~s  z4TestSeriesComparison.test_ser_flex_cmp_return_dtypescCsRtdddgtdd}|jdd}d}t|||j}td}||ksNtdS)Nr4rJr*rjrr)rrbrr7rrrrO)r:r-rXrrr>r?rrr%test_ser_flex_cmp_return_dtypes_emptys  z:TestSeriesComparison.test_ser_flex_cmp_return_dtypes_emptynames)NNN)rbarN)bazrrc Csn|}tddd|dd}t||d}|||}|j|dksFt|d}tj|d d }t||d}|||}|j|dkst||d}t||d}|||}|j|dkst|t j t j fkrtj dd|dd }t||d}|||}|j|dkst|t j t j fkrj| d } t| |d}||| }|j|dksjtdS) Nz1949-06-07 03:00:00Hr+r)ryrrMr4r*z US/CentralZinferrx)startrrMcategory)r rr6rMrO tz_localizerZ DatetimeIndexrr8rrZinterval_ranger) r:rrr=rrXr>ZtdiiiZcidxrrrtest_ser_cmp_result_namess0       z.TestSeriesComparison.test_ser_cmp_result_namesc Cstjd}tjd}tj|dd<t||}tjdd||kd}W5QRXtj|dd<t ||t dddg}t d d d g}t d d d g}t ||k|t ||k|dS) NrrJr])invalidOrfrgrFT) rrrrUr Znangtrcrr,r9rrW)r:rrr>r?srrYrrrtest_comparisonss    z%TestSeriesComparison.test_comparisonsc Csttddd}ttddd}tdddgdd}ttddd}ttd}ttd}|dkrjt|dk|dkkstd|krt|dkd std|kd std|kd rt||kst||krt|t|kst||ks t||kst||k||kks8t||k||kksTt||krft||krxt||krt||krt||kst||kst||krt||krt||k||kkst||k||kkst||k||kks8t||k||kksTtd }tjt|d ||kW5QRXtjt|d ||kW5QRXtjt|d ||kW5QRXtjt|d ||kW5QRXdS) NrrrobjectrfrgccZacbr can only compare equality or notrz)rrrr^rOrqrr)r:rfrgrrrfrrrrtest_categorical_comparisonssP  z1TestSeriesComparison.test_categorical_comparisonsc CsDtttd}d}tjt|d|dkW5QRXtttddd}tjt|d|dkW5QRXtttddd}d}tjt|d|d kW5QRXtjt|d|d kW5QRXtjt|dd |kW5QRXtjt|dd |kW5QRXt|d ktdddgt|d ktdddgdS) NrrrzrgF)ZorderedTz1Invalid comparison between dtype=category and strr)rrrrqrrr,rW)r:catrrrr5test_unequal_categorical_comparison_raises_type_errors&zJTestSeriesComparison.test_unequal_categorical_comparison_raises_type_errorcCstddg}|dk}tddg}t|||dk}tddg}t|||dk}tddg}t|||dk}tddg}t||tddg}|dk}tddg}t|||dk}tddg}t||dS)N)r4r4)r4r*FT)rrr)r:rr>r?rrrtest_comparison_tuples's(             z+TestSeriesComparison.test_comparison_tuplescCsDttdgtddgg}|tdgk}tddg}t||dS)Nr4r*TF)r frozensetr,rWr:rXr>r?rrrtest_comparison_frozensetFs z.TestSeriesComparison.test_comparison_frozensetcCsttdddtd}tj|ddd<|d}|||}||||j}|tj krh| d t }n| d t }t ||dS) Nrvrrrr*r+TF)rr rrrUZdropnarar_r8rrVrrr,rW)r:rrXvalr>r?rrr"test_comparison_operators_with_nasMs  z7TestSeriesComparison.test_comparison_operators_with_nascCsbtdddddgdddddgtd}dddddg}t|jdk|sFtt|jdk|s^tdS) NrJrRr+rkrTF)rfloatr,Z equalContentsr_rO)r:r$r?rrrtest_ne^s"zTestSeriesComparison.test_nez left, rightr4r*rJABCr&rZABDABCDc Csd|jd}|tk r&|}|}tjt|d||kW5QRXtjt|d||kW5QRXtjt|d||kW5QRXtjt|d||kW5QRXtjt|d||kW5QRXtjt|d||kW5QRXdS)Nz%Can only compare identically-labeled z objectsrz)rErrrqrr)r:rrZframe_or_seriesrrrrtest_comp_ops_df_compatds z,TestSeriesComparison.test_comp_ops_df_compatcCs4tdddg}|dk}tdddg}t||dS)NZ IntervalAZ IntervalBZ IntervalCTFrrrrr$test_compare_series_interval_keywordsz9TestSeriesComparison.test_compare_series_interval_keywordN)rErFrGrrqrrrsrrrrrrrrrrrrrrrrrrrqs:    %8 rc@s,eZdZddZddZddZddZd S) TestTimeSeriesArithmeticc Cstddddd}tjddd}ttjd||dd}tjddd}ttjd||d d}||}|d}|d}||}|jj t j kst t ||dS) N1/1/2011drutc)rryrZz US/Easternrjz Europe/Berlin)r rrrrrrrr_rpytzrrOr,rW) r:rpermser1ser2r>Zuts1Zuts2r?rrr+test_series_add_tz_mismatch_converts_to_utcs"    zDTestTimeSeriesArithmetic.test_series_add_tz_mismatch_converts_to_utcc Cs|tdddd}ttjt||d}|d}d}tjt |d||W5QRXtjt |d||W5QRXdS) Nrrrrrjrz0Cannot join tz-naive with tz-aware DatetimeIndexrz) r rrrrr rrqr Exception)r:rrXrrrrr"test_series_add_aware_naive_raisess z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisescCsJttddd}tjjdd}||}ttdddg}t||dS) Nz 2012-01-01rJrrk)daysz 2011-12-26z 2011-12-27z 2011-12-28)rr roffsetsZ DateOffsetZ to_datetimer,rW)r:r;offsetr>r?rrrtest_datetime_understoods z1TestTimeSeriesArithmetic.test_datetime_understoodcCstddd}ttjd|d}|dd}|}dd|jD|_||}||}||dd}|jd|_t ||t ||dS)Nrvrrjr+cSsg|] }|qSrr).0r&rrr szWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex..) r rrrrrr_Z _with_freqr,rW)r:rr$Zts_sliceZts2r>rr?rrr*test_align_date_objects_with_datetimeindexs   zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rErFrGrrrr rrrrrs  rc @sTeZdZejdeeej e e e j gejdddgddZ ddZd d Zd S) TestNamePreservationboxflexTFc Cs6|}ttd|dd}ttd|dd}|jd} | dk} | oL| d} ||}|rr| rbdSt|| |} n<| r|tkrtnd} d}tj | |d |||} W5QRX|tkr| rt | tst dSt | tst |ttfkr| j |d ks2| j |d ks2t n$| j |dks2| j |dks2t dS) NrrrLr4_)andZrandxorZrxororZrorr5z*operating as a set operation is deprecatedrzr*) rrbrEstrip startswithr7r FutureWarningr,Zassert_produces_warning isinstancerOrM)r:rrrrZall_binary_operatorsr=rrrMZ is_logicalZ is_rlogicalr>warnrrrrtest_series_ops_name_retentions,  &z3TestNamePreservation.test_series_ops_name_retentioncCs||}|j|jkst||}|j|jks2t||dd}|j|jksRt|}d|_||}|jdksvt||}|jdkstddddddd d g}|d d |D}|D]T}|}t|||}|j|jkst|}d |_t|||}|jdkstqdS)Nzsomething elser.r/r0rtr2r1modr3cSsg|] }d|qS)r5r)r r=rrrr szGTestNamePreservation.test_binop_maybe_preserve_name..changed)rMrOr0rr.r7)r:rr>cprr=rXrrrtest_binop_maybe_preserve_names,  z3TestNamePreservation.test_binop_maybe_preserve_namecCs|d}|j|jkstdSr))rMrO)r:rr>rrrtest_scalarop_preserve_namesz0TestNamePreservation.test_scalarop_preserve_nameN)rErFrGrqrrrsrtuplerarrayrrrrrrrrrrr s  & r c@s"eZdZejddddZdS)TestInplaceOperationsz)dtype1, dtype2, dtype_expected, dtype_mul))Int64r#r#r#)rrrr)r#rFloat64r$)r#r$r$r$cCstdg|d}tdg|d}||7}tdg|d}t||||8}tdg|d}t||||9}tdg|d}t||dS)Nr4rr*rJr)r:Zdtype1Zdtype2Zdtype_expectedZ dtype_mulrrr?rrrtest_series_inplace_ops!s   z-TestInplaceOperations.test_series_inplace_opsN)rErFrGrqrrrsr%rrrrr" s  r"c Csj|}t|dkrtdtj|jd<|dk}|jdrt|dk}|jdrVt|jdrftdS)Nr4z%Test doesn't make sense on empty datarzInvalid comparisonrz) r rqskiprrUrZiatrOrrrrr)Zseries_with_simple_indexr;r>rrrrtest_none_comparison<s8   r'cCsttdtjjtdtdddggdddgdd}td d tdd Dtjjtdddggddgdd}|jtjd gddddf|}td dddgtjjddddgdddgdd}t ||dS)NabZxyr4r*num)rrjcSsg|] }d|qS)r)r rhrrrr ssz;test_series_varied_multiindex_alignment..r+rfr+ii i)rfr&r4)rfr&r*)rfr\r4)rfr\r*) rrbrZ MultiIndexZ from_productrlocZ IndexSlice from_tuplesr,rW)rrr>r?rrr'test_series_varied_multiindex_alignmentjs($  r.)*rrr8ZnumpyrrqrZpandas._libs.tslibsrZpandas.core.dtypes.commonrrZpandasrrrrr r r r Zpandas._testingZ_testingr,Z pandas.corer rZpandas.core.computationrrZfixturerr"r#rurrrr r"r'r.rrrrs>  $   m 6&=N.