a j=icu@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)ZautousescopeparamsZidsccs tj}|jt_|jV|t_dSN)expr _MIN_ELEMENTSparam)requestrrt/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/series/test_arithmetic.pyswitch_numexpr_min_elements!srcCs|tjt|Sr)takenprandom permutationlen)objrrr_permute+sr"c @seZdZejddddddfdddddfddd dd fd dd dd fgejd gd ddZddZddZe j e j dfe j ejdfgZdD]VZdZee eZee eZee deZefddZeeeefeeeefqejdeddZdS)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(opname)addsubmulfloordivtruedivpowc 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_equivalencecsTGfdddt|}tt|}gddd}d|_||d}|jdksPJdS)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*r4)rr7r&)r:Zall_arithmetic_operatorsr-r=mr>rrBr/test_flex_method_subclass_metadata_preservationPs  zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationcCsDtddtjdddg}|dd}|jddd}t||dS)Nrr4rKr+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 Ss6|j|j}||}||}t|}t|}g}tt|D]} tjdd|| r|| r~|tj WdqB||||| nX|| r|| r|tj WdqB|||| |n|||| || WdqB1s0YqB||||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_r4rKg@rSaxis)r)rrrUaranger,rW) r:r=Zequiv_opfvrirfrgr>rYrrrtest_operators_combiness  $   z/TestSeriesFlexArithmetic.test_operators_combineN)rErFrGpytestmark parametrizerArQr[rdivr8r2ZrdivrZrtruedivZpairingsr=ror7lopZlequivr@Zrequivrdrprrrrr#/s2      r#c@seZdZddZejddddgddgfd ddgd d gfgd d ZddZejdde dgddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$ZdS)%TestSeriesArithmeticcCstjdddd}ttjt||d}||ddd}||}tj|jddd<t |||t |ddd}t ||d}t j t|d "||jd d d Wdn1s0YdS) 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_additioncCstdddtjggdd}tdtjdtjggdd}||}t||}t|d|dt|d|d||}t||}t|d|dt|d|ddS)Nr4)rfrgcdrjr*)rfrgrer)rrrUdivmodr,rWZrdivmod)r:rfrgr>r?rrr test_divmods    z TestSeriesArithmetic.test_divmodr_N cCsXttdd}td|d}d}tjt|d||Wdn1sJ0YdS)Nr4 foorjz4not all arguments converted during string formattingr{)rrbrqr TypeError)r:r_s1s2rrrrtest_series_integer_mods  z,TestSeriesArithmetic.test_series_integer_modcCsXtddgddgd}tddgddgd}||}tddtjggdd}t||dS)Nr4r*rjr )r4r4r*)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|dspJt|dsJdS) Nrdate)Decimalz1.3z2.3ir4r*rj)datetimerdecimalrrshiftr )r:rrrXr>result2rrrtest_add_na_handlings  $z)TestSeriesArithmetic.test_add_na_handlingcCsNtgtgtjd}||}t|s.J||}t|dksJJdS)N)r_dtyper)rrrfloat64isnanr^copyr )r:datetime_seriesemptyr>rrrtest_add_corner_casess  z*TestSeriesArithmetic.test_add_corner_casescCsZ|tdd}||}t|jdd|j|jdddd}t|dd|dS)Nr$r_rN)astypeintrvaluesr_r,rW)r:rZint_tsaddedr?rrrtest_add_float_plus_ints z,TestSeriesArithmetic.test_add_float_plus_intcCs>tggtjd}tddi}t||ttjgdgddS)Nrr&grj)rrint32r,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-1rKr}periodsrzr*rrM) rr rrUrrrZNaTr,rWto_frame)r:dtZdt2r?r>rrrtest_sub_datetimelike_aligns   z0TestSeriesArithmetic.test_sub_datetimelike_aligncCsXtdddd}|d}td|d}td|d}|||j|usFJ|j|usTJdS)Nz 2016-01-01rZCET)rtzUTCrj)r tz_convertrr_)r:dtiZdti_utcrXser_utcrrrtest_alignment_doesnt_change_tzs   z4TestSeriesArithmetic.test_alignment_doesnt_change_tzcCsgd}ttjdddd|d}ttjddd|d}||}tdgdd}t||ttddd |d}ttd dd |d}||}ttd gdd}t||dS) N)r*r*rKrKrRr4rkint64rrjr+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)rErFrGrrqrrrsrrrbrrrrrrrrrrrrrrvs&      rvc @seZdZejdgdddZddZejdgdd fgd d fgdd fgd dfgddfgddfgddZejdgdd dfgdd dfgdd dfgdddfgdddfgdddfgddZ dS)TestSeriesFlexComparisonrm)rNr_cCsNttjd}ttjd}t||j||d}|||}t||dS)Nrrl)rrrrr7rEr,rW)r:rm comparison_opleftrightr>r?rrrtest_comparison_flex_basic<s  z3TestSeriesFlexComparison.test_comparison_flex_basiccCsjttjd}ttjd}d}tjt|d$t||j|ddWdn1s\0YdS)NrzNo axis named 1 for object typer{r4rl) rrrrrqr ValueErrorr7rE)r:rrrrrrrtest_comparison_bad_axisDs z1TestSeriesFlexComparison.test_comparison_bad_axisz values, op)FFTFeq)TTFTnele)FFFFlt)FTTFge)FTFFgtcCsVtgdtdd}tgdtdd}t|||}t|tdd}t||dS)Nr4rKr*abcrjr*r*r*bcdabcdrlistr7r,rW)r:rr=rrr>r?rrrtest_comparison_flex_alignmentLs z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_value)FFTTr*)TTFFr)FFFT)TTTFcCsZtgdtdd}tgdtdd}t||||d}t|tdd}t||dS)NrrrjrrrSrr)r:rr=rTrrr>r?rrr#test_comparison_flex_alignment_fill^s zJdS)NrrKrjr*bool)rrbr7dtypesrr)r:r-rXconstr>r?rrrtest_ser_flex_cmp_return_dtypes~s  z4TestSeriesComparison.test_ser_flex_cmp_return_dtypescCsPtgdtdd}|jdd}d}t|||j}td}||ksLJdS)NrrKrjrr*r)rrbrr7rrr)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|dksFJ|d}tj|d d }t||d}|||}|j|dksJ||d}t||d}|||}|j|dksJ|tj tj fvrtj dd|dd }t||d}|||}|j|dksJ|tj tj fvrj| d } t| |d}||| }|j|dksjJdS) Nz1949-06-07 03:00:00Hr+r)rzrrNr4r*z US/CentralZinferry)startrrNcategory) r rr6rN tz_localizerZ DatetimeIndexrr8rrZinterval_ranger) r:rrr=rrXr>ZtdiiiZcidxrrrtest_ser_cmp_result_namess0       z.TestSeriesComparison.test_ser_cmp_result_namescCstjd}tjd}tj|dd<t||}tjdd||kd}Wdn1sb0Ytj|dd<t ||t gd}t gd}t gd}t ||k|t ||k|dS) NrrKr])invalidOr)FTFFFF) rrrrUr Znangtrcrr,r9rrW)r:rrr>r?srrYrrrtest_comparisonss   ,    z%TestSeriesComparison.test_comparisonscCs:ttddd}ttddd}tgddd}ttddd}ttd}ttd}|dkrhJ|dk|dkksJd|krJ|dkdsJd|kdsJd|kdrJ||ksJ||krJ|t|ksJ||ksJ||ksJ||k||kks6J||k||kksRJ||krdJ||krvJ||krJ||krJ||ksJ||ksJ||krJ||krJ||k||kksJ||k||kksJ||k||kks6J||k||kksRJd }tjt|d ||kWdn1s0Ytjt|d ||kWdn1s0Ytjt|d ||kWdn1s0Ytjt|d ||kWdn1s,0YdS) Nrrrobject)rfrgccZacbrfr can only compare equality or notr{)rrrr^rqrr)r:rfrgrrrfrrrrtest_categorical_comparisonssP  (((z1TestSeriesComparison.test_categorical_comparisonscCstttd}d}tjt|d|dkWdn1s@0Ytttddd}tjt|d|dkWdn1s0Ytttddd}d}tjt|d|d kWdn1s0Ytjt|d|d kWdn1s0Ytjt|dd |kWdn1sH0Ytjt|dd |kWdn1s0Yt|d ktgd t|d ktgd dS) Nrrr{rgF)ZorderedTz1Invalid comparison between dtype=category and strrr)TTT)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 urh| d t }n| d t }t ||dS) Nrwrrrr*r+TF)rr rrrUZdropnarar_r8rrVrrr,rW)r:rrXvalr>r?rrr"test_comparison_operators_with_nasMs  z7TestSeriesComparison.test_comparison_operators_with_nascCsPtgdgdtd}gd}t|jdk|s4Jt|jdk|sLJdS)N)rKrRr+rkr)TTFTTr+)rfloatr,Z equalContentsr_)r:r$r?rrrtest_ne^szTestSeriesComparison.test_nez left, rightrJABCr&rrZABD)r*r*r*r*ABCDcCsrd|jd}|tur&|}|}tjt|d||kWdn1sR0Ytjt|d||kWdn1s0Ytjt|d||kWdn1s0Ytjt|d||kWdn1s0Ytjt|d||kWdn1s,0Ytjt|d||kWdn1sd0YdS)Nz%Can only compare identically-labeled z objectsr{)rErrrqrr)r:rrZframe_or_seriesrrrrtest_comp_ops_df_compatds &&&&(z,TestSeriesComparison.test_comp_ops_df_compatcCs0tgd}|dk}tgd}t||dS)N) IntervalAZ IntervalBZ IntervalCr)TFFrrrrr$test_compare_series_interval_keywords  z9TestSeriesComparison.test_compare_series_interval_keywordN)rErFrGrrqrrrsrrrrrrrrrrrrrrrrrrrqs8    %8 rc@s,eZdZddZddZddZddZd S) TestTimeSeriesArithmeticc Cstddddd}tjddd}ttjd||dd}tjddd}ttjd||d d}||}|d}|d}||}|jj t j ksJt ||dS) N1/1/2011drutc)rrzrZz US/Easternrjz Europe/Berlin)r rrrrrrrr_rpytzrr,rW) r:rpermser1ser2r>Zuts1Zuts2r?rrr+test_series_add_tz_mismatch_converts_to_utcs  zDTestTimeSeriesArithmetic.test_series_add_tz_mismatch_converts_to_utccCstdddd}ttjt||d}|d}d}tjt |d||Wdn1s`0Ytjt |d||Wdn1s0YdS) Nrrrrrjrz0Cannot join tz-naive with tz-aware DatetimeIndexr{) r rrrrr rrqr Exception)r:rrXrrrrr"test_series_add_aware_naive_raisess &z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisescCsHttddd}tjjdd}||}ttgd}t||dS)Nz 2012-01-01rKrrk)days)z 2011-12-26z 2011-12-27z 2011-12-28)rr roffsetsZ DateOffset to_datetimer,rW)r:r;offsetr>r?rrrtest_datetime_understoods z1TestTimeSeriesArithmetic.test_datetime_understoodcCstddd}ttjd|d}|dd}|}dd|jD|_||}||}||dd}|jd|_t ||t ||dS)Nrwrrjr+cSsg|] }|qSrr).0r&rrr r(zWTestTimeSeriesArithmetic.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)rErFrGr r rrrrrrrs  rc @sTeZdZejdeeej e e e j gejdddgddZ ddZd d Zd S) TestNamePreservationboxflexTFcCsJ|}ttd|dd}ttd|dd}|jd}|dv} | oL|d} ||}|rr| rbdSt|||} nP| r|turtnd} d} tj | | d |||} Wdn1s0Y|tur| rt | tsJdSt | tsJ|ttfvr"| j |d usF| j |d ksFJn$| j |dusF| j |dksFJdS) NrrrMr4_)andZrandxorZrxororZrorr5z*operating as a set operation is deprecatedr{r*) rrbrEstrip startswithr7r FutureWarningr,Zassert_produces_warning isinstancerN)r:rrrZall_binary_operatorsr=rrrNZ is_logicalZ is_rlogicalr>warnrrrrtest_series_ops_name_retentions, ( &z3TestNamePreservation.test_series_ops_name_retentioncCs||}|j|jksJ||}|j|jks2J||dd}|j|jksRJ|}d|_||}|jdusvJ||}|jdusJgd}|dd|D}|D]T}|}t|||}|j|jksJ|}d|_t|||}|jdusJqdS)Nzsomething else)r.r/r0rtr2r1modr3cSsg|] }d|qS)r5r)rr=rrrr r(zGTestNamePreservation.test_binop_maybe_preserve_name..changed)rNr0rr.r7)r:rr>cprr=rXrrrtest_binop_maybe_preserve_names,  z3TestNamePreservation.test_binop_maybe_preserve_namecCs|d}|j|jksJdSr)rM)r:rr>rrrtest_scalarop_preserve_namesz0TestNamePreservation.test_scalarop_preserve_nameN)rErFrGrqrrrsrtuplerarrayrrrr#r(r)rrrrrs  $ rc@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*rKr)r:Zdtype1Zdtype2Zdtype_expectedZ dtype_mulr r r?rrrtest_series_inplace_opss   z-TestInplaceOperations.test_series_inplace_opsN)rErFrGrqrrrsr/rrrrr,s  r,cCs|}t|dkr&|jtjjddtj|jd<|dk}|j drHJ|j drVJ|dk}|j dslJ|j dszJd|k}|j drJ|j drJd|k}|j dsJ|j dsJt |j st |j rNd}tj t|dd|kWdn1s 0Ytj t|d|dkWdn1sB0YnPd|k}|j drfJ|j drvJ|dk}|j drJ|j drJdS)Nr4z%Test doesn't make sense on empty data)reasonrzInvalid comparisonr{)r nodeZ add_markerrqrrZxfailrrUrZiatrrrrr)rZseries_with_simple_indexr;r>rrrrtest_none_comparison:s<   (*r2cCsttdtjjtdtdddgggddd}td d tdd Dtjjtdddggdd gdd}|jtjd gddddf|}tgdtjjgdgddd}t ||dS)Nabxyr4r*)r4r5num)rrjcSsg|] }d|qS)r)rrhrrrrsr(z;test_series_varied_multiindex_alignment..r+r6rf)r7ii i))r&r4rf)r&r*rf)r\r4rf)r\r*rf)r5r6r4) rrbrZ MultiIndexZ from_productrlocZ IndexSlice from_tuplesr,rW)rrr>r?rrr'test_series_varied_multiindex_alignmentjs&$r:)*rrr8numpyrrqrZpandas._libs.tslibsrZpandas.core.dtypes.commonrrZpandasrrrrr r r r Zpandas._testingZ_testingr,Z pandas.corer rZpandas.core.computationrrZfixturerr"r#rvrrrrr,r2r:rrrrs8  $   m 6&=L0