o iu@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@Bnumexprpython)autousescopeparamsidsccs"tj}|jt_|jV|t_dSN)expr _MIN_ELEMENTSparam)requestrr^/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/tests/series/test_arithmetic.pyswitch_numexpr_min_elements!s  rcCs|tjt|Sr)takenprandom permutationlen)objrrr_permute+sr%c @seZdZejddddddfdddddfddd dd fd dd dd fgejd gd ddZddZddZe j e j dfe j ejdfgZdD]+ZdZee eZee eZee deZefddZeeeefeeeefqPejdeddZdS)TestSeriesFlexArithmetictscC|Srrxrrr3z!TestSeriesFlexArithmetic.cCs|dSNrr)rrrr+3FcCr(rrr)rrrr+4r,cCs|dddSr-rr)rrrr+4scCr(rrr)rrrr+5r,cCsdS)Nrr)rrrr+5r,TcCstSr)tmmakeFloatSeriesr)rrrr+6r/opname)addsubmulfloordivtruedivpowc Cstd}|d|}|d|}|d}tt|}tt|}|||} |||} t| | |rLttd|} | ||} |||} t| | dSdS)Nr'rr.r)r1makeTimeSeriesrenamegetattrroperatorassert_almost_equal) selfr3r'tserseriesother check_reverseopaltresultexpectedroprrrtest_flex_method_equivalence0s         z5TestSeriesFlexArithmetic.test_flex_method_equivalencecsTGfdddt|}tt|}gddd}d|_||d}|jdks(JdS)Ncs"eZdZdgZefddZdS)zZTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation..MySeriesr*csSrr)rAMySeriesrr _constructorUszgTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservation..MySeries._constructorN)__name__ __module__ __qualname__ _metadatapropertyrNrrLrrrMRsrMr:r.testname*r:)rr>r*)rAall_arithmetic_operatorsr3rFmrHrrLr/test_flex_method_subclass_metadata_preservationPs  zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationcCsDtddtjdddg}|dd}|jddd}t||dS)Nrr:rUr0r. fill_value)rr nanfillnar4r1assert_series_equal)rAserexpresrrrtest_flex_add_scalar_fill_value`sz8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer:)r4r5r6r9r8r7rr;cCstt|||Sr)r>r?)r*yrFrrrr+oszop, equiv_op, fvc Csddd}ttjdddtjgtdd}ttjd tjd tjd gtd d}|||}|||}t||||||||d |||dddS)Nrc Ss8|j|j}||}||}t|}t|}g}tt|D]f} tjddV|| rK|| r@|tj  Wdq!||||| n-|| rl|| ra|tj  Wdq!|||| |n |||| || Wdn1swYq!||||d} t ||} t | | dS)Nignore)allr^) indexunionreindexr ranger#r errstateappendr`rr1rb) methrFabr_ exp_indexamaskbmask exp_valuesirHrIrrr _check_fillus6      zDTestSeriesFlexArithmetic.test_operators_combine.._check_fillg?g@g@r0rjr:rUg@r^axis)r)rr r`aranger1rb) rArFequiv_opfvrxrqrrrHrdrrrtest_operators_combiness  $   z/TestSeriesFlexArithmetic.test_operators_combineN)rOrPrQpytestmark parametrizerKr\rfrdivr?r8rdivrrtruedivpairingsrFrr>loplequivrJrequivrorrrrrr&/s4      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 WddS1s_wYdS) N1/1/2000z1/1/2010Afreqryr.r:z4Input has different freq=D from Period\(freq=A-DEC\)matchDend)how)pd period_rangerr r!randnr#r`ilocr1rbr%rraisesrasfreq)rArngr'rHrImsgrrr!test_add_series_with_period_indexs  "z6TestSeriesArithmetic.test_add_series_with_period_indexz%target_add,input_value,expected_value!helloworldzhello!zworld!r[hellomworldmcCs(t|}||}t|}t||dSrrr1rb)rA target_add input_valueexpected_valuerqrHrIrrrtest_string_additions z)TestSeriesArithmetic.test_string_additioncCstdddtjggdd}tdtjdtjggdd}||}t||}t|d|dt|d|d||}t||}t|d|dt|d|ddS)Nr:)rqrrcdryr.)rqrrrer)rr r`divmodr1rbrdivmod)rArqrrrHrIrrr test_divmods    z TestSeriesArithmetic.test_divmodrjN cCsZttdd}td|d}d}tjt|d ||WddS1s&wYdS)Nr: fooryz4not all arguments converted during string formattingr)rrmrr TypeError)rArjs1s2rrrrtest_series_integer_mods   "z,TestSeriesArithmetic.test_series_integer_modcCsXtddgddgd}tddgddgd}||}tddtjggdd}t||dS)Nr:r.ryr )r:r:r.)rr r`r1rb)rArrrHrIrrrtest_add_with_duplicate_indexs z2TestSeriesArithmetic.test_add_with_duplicate_indexcCsddlm}ddlm}t|d|dg|ddd|dddgd }||d}|d|}t|ds8Jt|ds@JdS) Nrdate)Decimalz1.3z2.3ir:r.ry)datetimerdecimalrrshiftr )rArrrcrHresult2rrrtest_add_na_handlings  $z)TestSeriesArithmetic.test_add_na_handlingcCsNtgtgtjd}||}t|sJ||}t|dks%JdS)N)rjdtyper)rrr float64isnanricopyr#)rAdatetime_seriesemptyrHrrrtest_add_corner_casess  z*TestSeriesArithmetic.test_add_corner_casescCsZ|tdd}||}t|jdd|j|jdddd}t|dd|dS)Nr'rjrX)astypeintrvaluesrjr1rb)rArint_tsaddedrIrrrtest_add_float_plus_ints z,TestSeriesArithmetic.test_add_float_plus_intcCs>tggtjd}tddi}t||ttjgdgddS)Nrr*gry)rr int32r1rbr`)rArrrrrtest_mul_empty_int_corner_cases "z3TestSeriesArithmetic.test_mul_empty_int_corner_casecCsttdddd}tj|jd<|ddd}ttdtdtjg}||}t ||t|dd}| | d}t ||dS) Nz2012-1-1rUrperiodsrr.rrW) rr r r`rrrNaTr1rbto_frame)rAdtdt2rIrHrrrtest_sub_datetimelike_aligns   z0TestSeriesArithmetic.test_sub_datetimelike_aligncCsXtdddd}|d}td|d}td|d}|||j|us#J|j|us*JdS)Nz 2016-01-01rCET)rtzUTCry)r tz_convertrrj)rAdtidti_utcrcser_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.rUrUr]r:rzint64rryr0z20130101 09:00:00r20130101z9 hours)rr r}r1rbr r )rArjrcrDrHrIrrr$test_arithmetic_with_duplicate_index#s z9TestSeriesArithmetic.test_arithmetic_with_duplicate_index)rOrPrQrrrrrrrmrrrrrrrrrrrrrrs(       rc @seZdZejdgdddZddZejdgdd fgd d fgdd fgd dfgddfgddfgddZejdgdd dfgdd dfgdd dfgdddfgdddfgdddfgddZ dS)TestSeriesFlexComparisonr|)rNrjcCsNttjd}ttjd}t||j||d}|||}t||dS)Nrr{)rr r!rr>rOr1rb)rAr| comparison_opleftrightrHrIrrrtest_comparison_flex_basic<s  z3TestSeriesFlexComparison.test_comparison_flex_basiccCslttjd}ttjd}d}tjt|dt||j|ddWddS1s/wYdS)NrzNo axis named 1 for object typerr:r{) rr r!rrr ValueErrorr>rO)rArrrrrrrtest_comparison_bad_axisDs "z1TestSeriesFlexComparison.test_comparison_bad_axisz values, op)FFTFeq)TTFTnele)FFFFlt)FTTFge)FTFFgtcCsVtgdtdd}tgdtdd}t|||}t|tdd}t||dS)Nr:rUr.abcryr.r.r.bcdabcdrlistr>r1rb)rArrFrrrHrIrrrtest_comparison_flex_alignmentLs  z7TestSeriesFlexComparison.test_comparison_flex_alignmentzvalues, op, fill_value)FFTTr.)TTFFr)FFFT)TTTFcCsZtgdtdd}tgdtdd}t||||d}t|tdd}t||dS)Nrrryrrr^rr)rArrFr_rrrHrIrrr#test_comparison_flex_alignment_fill^s  zdtypesr r)rAr3rcconstrHrIrrrtest_ser_flex_cmp_return_dtypes~s  z4TestSeriesComparison.test_ser_flex_cmp_return_dtypescCsPtgdtdd}|jdd}d}t|||j}td}||ks&JdS)NrrUryrr.r)rrmrr>rr r)rAr3rcrrrHrIrrr%test_ser_flex_cmp_return_dtypes_emptys  z:TestSeriesComparison.test_ser_flex_cmp_return_dtypes_emptynames))NNN)rbarN)bazrrc Csj|}tddd|dd}t||d}|||}|j|dks#J|d}tj|d d }t||d}|||}|j|dksFJ||d}t||d}|||}|j|dksdJ|tj tj fvrtj dd|dd }t||d}|||}|j|dksJ|tj tj fvr| d } t| |d}||| }|j|dksJdSdS) Nz1949-06-07 03:00:00Hr0r)rrrXr:r.z US/Centralinferr)startrrXcategory) r rr=rX tz_localizer DatetimeIndexrr?rrinterval_ranger) rArrrFrrcrHtdiiicidxrrrtest_ser_cmp_result_namess4       z.TestSeriesComparison.test_ser_cmp_result_namescCstjd}tjd}tj|dd<t||}tjdd||kd}Wdn1s1wYtj|dd<t ||t gd}t gd}t gd}t ||k|t ||k|dS) NrrUrh)invalidOr)FTFFFF) r r!rr`r nangtrnrr1r@rrb)rArrrHrIsrrdrrrtest_comparisonss       z%TestSeriesComparison.test_comparisonscCs ttddd}ttddd}tgddd}ttddd}ttd}ttd}|dkr4J|dk|dkksAJd|krIJ|dkdsQJd|kdsYJd|kdraJ||ksiJ||krqJ|t|ks{J||ksJ||ksJ||k||kksJ||k||kksJ||krJ||krJ||krJ||krJ||ksJ||ksJ||krJ||krJ||k||kksJ||k||kksJ||k||kks J||k||kksJd }tjt|d  ||kWdn 1s4wYtjt|d  ||kWdn 1sPwYtjt|d  ||kWdn 1slwYtjt|d  ||kWddS1swYdS) Nrr robject)rqrrccacbrqr can only compare equality or notr)rrrrirrr)rArqrrrrrfrrrrtest_categorical_comparisonssX      $z1TestSeriesComparison.test_categorical_comparisonscCstttd}d}tjt|d |dkWdn1s wYtttddd}tjt|d |dkWdn1sEwYtttddd}d}tjt|d |d kWdn1slwYtjt|d |d kWdn1swYtjt|d d |kWdn1swYtjt|d d |kWdn1swYt|d ktgd t|d ktgd dS) NrrrrrF)orderedTz1Invalid comparison between dtype=category and strrr)TTT)rrrrrrr1rb)rAcatrrrr5test_unequal_categorical_comparison_raises_type_errors2      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)r:r:)r:r.FT)rrr)rArrHrIrrrtest_comparison_tuples's(             z+TestSeriesComparison.test_comparison_tuplescCsDttdgtddgg}|tdgk}tddg}t||dS)Nr:r.TF)r frozensetr1rbrArcrHrIrrrtest_comparison_frozensetFs z.TestSeriesComparison.test_comparison_frozensetcCsttdddtd}tj|ddd<|d}|||}||||j}|tj ur4| d t }n| d t }t ||dS) Nrrrrr.r0TF)rr rr r`dropnarlrjr?rrarrr1rb)rArrcvalrHrIrrr"test_comparison_operators_with_nasMs  z7TestSeriesComparison.test_comparison_operators_with_nascCsPtgdgdtd}gd}t|jdk|sJt|jdk|s&JdS)N)rUr]r0rzr)TTFTTr0)rfloatr1 equalContentsrj)rAr'rIrrrtest_ne^szTestSeriesComparison.test_nez left, rightrTABCr*rrABD)r.r.r.r.ABCDcCspd|jd}|tur|}|}tjt|d ||kWdn1s)wYtjt|d ||kWdn1sDwYtjt|d ||kWdn1s_wYtjt|d ||kWdn1szwYtjt|d ||kWdn1swYtjt|d ||kWddS1swYdS)Nz%Can only compare identically-labeled z objectsr)rOrrrrr)rArrframe_or_seriesrrrrtest_comp_ops_df_compatds,      "z,TestSeriesComparison.test_comp_ops_df_compatcCs0tgd}|dk}tgd}t||dS)N) IntervalA IntervalB IntervalCr1)TFFrr#rrr$test_compare_series_interval_keywords  z9TestSeriesComparison.test_compare_series_interval_keywordN)rOrPrQrrrrrrrrrr r!r$r'r+rrr0r4rrrrrqs:    %8 rc@s,eZdZddZddZddZddZd S) TestTimeSeriesArithmeticc Cstddddd}tjddd}ttjd||dd}tjddd}ttjd||d d}||}|d}|d}||}|jj t j ksWJt ||dS) N1/1/2011drutc)rrrZz US/Easternryz Europe/Berlin)r r r!r"rrrrrjrpytzrr1rb) rArpermser1ser2rHuts1uts2rIrrr+test_series_add_tz_mismatch_converts_to_utcs  zDTestTimeSeriesArithmetic.test_series_add_tz_mismatch_converts_to_utccCstdddd}ttjt||d}|d}d}tjt |d ||Wdn1s0wYtjt |d ||WddS1sLwYdS) Nr6rrrryr8z0Cannot join tz-naive with tz-aware DatetimeIndexr) r rr r!rr#r rr Exception)rArrcrrrrr"test_series_add_aware_naive_raisess   "z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisescCsHttddd}tjjdd}||}ttgd}t||dS)Nz 2012-01-01rUrrz)days)z 2011-12-26z 2011-12-27z 2011-12-28)rr roffsets DateOffset to_datetimer1rb)rArCoffsetrHrIrrrtest_datetime_understoods z1TestTimeSeriesArithmetic.test_datetime_understoodcCstddd}ttjd|d}|dd}|}dd|jD|_||}||}||dd}|jd|_t ||t ||dS)Nrrryr0cSsg|]}|qSrr).0r*rrr zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex..) r rr r!rrrj _with_freqr1rb)rArr'ts_slicets2rHrrIrrr*test_align_date_objects_with_datetimeindexs   zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rOrPrQr@rBrHrPrrrrr5s  r5c @sTeZdZejdeeej e e e j gejdddgddZ ddZd d Zd S) TestNamePreservationboxflexTFcCsJ|}ttd|dd}ttd|dd}|jd}|dv} | o&|d} ||}|r9| r1dSt|||} n(| rA|turAtnd} d} tj | | d  |||} Wdn1s\wY|turp| rpt | tsnJdSt | tswJ|ttfvr| j |d us| j |d ksJdSdS| j |dus| j |dksJdSdS) NrrrWr:_)andrandxorrxororrorr;z*operating as a set operation is deprecatedrr.) rrmrOstrip startswithr>r FutureWarningr1assert_produces_warning isinstancerX)rArSrRrall_binary_operatorsrFrrrX is_logical is_rlogicalrHwarnrrrrtest_series_ops_name_retentions.    ((z3TestNamePreservation.test_series_ops_name_retentioncCs||}|j|jks J||}|j|jksJ||dd}|j|jks)J|}d|_||}|jdus;J||}|jdusGJgd}|dd|D}|D]*}|}t|||}|j|jkskJ|}d|_t|||}|jdusJqVdS)Nzsomething else)r4r5r6rr8r7modr9cSsg|]}d|qS)r;r)rJrFrrrrK rLzGTestNamePreservation.test_binop_maybe_preserve_name..changed)rXr6rr4r>)rArrHcprrFrcrrrtest_binop_maybe_preserve_names.  z3TestNamePreservation.test_binop_maybe_preserve_namecCs|d}|j|jks JdSr-rW)rArrHrrrtest_scalarop_preserve_namesz0TestNamePreservation.test_scalarop_preserve_nameN)rOrPrQrrrrtupler arrayrrrrdrirjrrrrrQs  $ rQc@s"eZdZejddddZdS)TestInplaceOperationsz)dtype1, dtype2, dtype_expected, dtype_mul))Int64rnrnrn)r)r)r)r))rnr)Float64ro)rnrororocCstdg|d}tdg|d}||7}tdg|d}t||||8}tdg|d}t||||9}tdg|d}t||dS)Nr:rr.rUr)rAdtype1dtype2dtype_expected dtype_mulr<r=rIrrrtest_series_inplace_opss   z-TestInplaceOperations.test_series_inplace_opsN)rOrPrQrrrrtrrrrrms  rmcCs|}t|dkr|jtjjddtj|jd<|dk}|j dr$J|j dr+J|dk}|j ds6J|j ds=Jd|k}|j drHJ|j drOJd|k}|j dsZJ|j dsaJt |j skt |j rd}tj t|d d|kWdn1swYtj t|d |dkWddS1swYdSd|k}|j drJ|j drJ|dk}|j drJ|j drJdS)Nr:z%Test doesn't make sense on empty data)reasonrzInvalid comparisonr)r#node add_markerrrxfailr r`riatrrrrr)rseries_with_simple_indexrCrHrrrrtest_none_comparison:s@     "r{cCsttdtjjtdtdddgggddd}td d tdd Dtjjtdddggdd gdd}|jtjd gddddf|}tgdtjjgdgddd}t ||dS)Nabxyr:r.)r}r~num)rrycSsg|]}d|qS)r)rJrwrrrrKsrLz;test_series_varied_multiindex_alignment..r0rrq)rii i))r*r:rq)r*r.rq)rgr:rq)rgr.rq)r~rr}) rrmr MultiIndex from_productrloc IndexSlice from_tuplesr1rb)rrrHrIrrr'test_series_varied_multiindex_alignmentjs&$r)*rrr?numpyr rr:pandas._libs.tslibsrpandas.core.dtypes.commonrrpandasrrrrr r r r pandas._testing_testingr1 pandas.corer rpandas.core.computationrrfixturerr%r&rrrr5rQrmr{rrrrrs:  $   m 6&=L 0