a nEbu@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)Zautousescopeparamsidsccs tj}|jt_|jV|t_dSN)expr _MIN_ELEMENTSparam)requestrrc/home/ja/django-apps/lartica_env/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__ _metadatapropertyrErrCrrrDRsrDr5r+testname*r5)rr8r')r;Zall_arithmetic_operatorsr.r>mr?rrCr/test_flex_method_subclass_metadata_preservationPs  zHTestSeriesFlexArithmetic.test_flex_method_subclass_metadata_preservationcCsDtddtjdddg}|dd}|jddd}t||dS)Nrr5rLr,r+ fill_value)rrnanfillnar/r-assert_series_equal)r;serexpresrrrtest_flex_add_scalar_fill_value`sz8TestSeriesFlexArithmetic.test_flex_add_scalar_fill_valuer5)r/r0r1r4r3r2rr6cCstt|||Sr)r8r9)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)allrT) indexunionreindexr ranger!rerrstateappendrVrr-rX) methr>abrUZ exp_indexZamaskZbmaskZ exp_valuesir?r@rrr _check_fillus,    : zDTestSeriesFlexArithmetic.test_operators_combine.._check_fillg?g@g@r,r`r5rLg@rTaxis)r)rrrVaranger-rX) r;r>Zequiv_opfvrjrgrhr?rZrrrtest_operators_combiness  $   z/TestSeriesFlexArithmetic.test_operators_combineN)rFrGrHpytestmark parametrizerBrRr\rdivr9r3ZrdivrZrtruedivZpairingsr>rpr8ZlopZlequivrAZrequivrerqrrrrr$/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/2010Afreqrkr+r5z4Input has different freq=D from Period\(freq=A-DEC\)matchDend)how)pdZ period_rangerrrrandnr!rVilocr-rXr#rrraisesrZasfreq)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!rQZhellomZworldmcCs(t|}||}t|}t||dSrrr-rX)r;Z target_addZ input_valueZexpected_valuergr?r@rrrtest_string_additions z)TestSeriesArithmetic.test_string_additioncCstdddtjggdd}tdtjdtjggdd}||}t||}t|d|dt|d|d||}t||}t|d|dt|d|ddS)Nr5)rgrhcdrkr+)rgrhrer)rrrVdivmodr-rXZrdivmod)r;rgrhr?r@rrr test_divmods    z TestSeriesArithmetic.test_divmodr`N cCsXttdd}td|d}d}tjt|d||Wdn1sJ0YdS)Nr5 foorkz4not all arguments converted during string formattingr{)rrcrrr TypeError)r;r`s1s2rrrrtest_series_integer_mods  z,TestSeriesArithmetic.test_series_integer_modcCsXtddgddgd}tddgddgd}||}tddtjggdd}t||dS)Nr5r+rkr )r5r5r+)rrrVr-rX)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.3ir5r+rk)datetimerdecimalrrshiftr )r;rrrYr?result2rrrtest_add_na_handlings  $z)TestSeriesArithmetic.test_add_na_handlingcCsNtgtgtjd}||}t|s.J||}t|dksJJdS)N)r`dtyper)rrrZfloat64isnanr_copyr!)r;datetime_seriesemptyr?rrrtest_add_corner_casess  z*TestSeriesArithmetic.test_add_corner_casescCsZ|tdd}||}t|jdd|j|jdddd}t|dd|dS)Nr%r`rO)astypeintrvaluesr`r-rX)r;rZint_tsaddedr@rrrtest_add_float_plus_ints z,TestSeriesArithmetic.test_add_float_plus_intcCs>tggtjd}tddi}t||ttjgdgddS)Nrr'grk)rrZint32r-rXrV)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-1rLr}periodsrzr+rrN) rr rrVrrrZNaTr-rXto_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)rtzUTCrk)r tz_convertrr`)r;dtiZdti_utcrYser_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+rLrLrSr5rlZint64rrkr,z20130101 09:00:00rZ20130101z9 hours)rrror-rXr r )r;r`rYr=r?r@rrr$test_arithmetic_with_duplicate_index#s z9TestSeriesArithmetic.test_arithmetic_with_duplicate_index)rFrGrHrrrrsrtrrrcrrrrrrrrrrrrrrvs&      rvc @seZdZejdgdddZddZejdgdd fgd d fgdd fgd dfgddfgddfgddZejdgdd dfgdd dfgdd dfgdddfgdddfgdddfgddZ dS)TestSeriesFlexComparisonrn)rNr`cCsNttjd}ttjd}t||j||d}|||}t||dS)Nrrm)rrrrr8rFr-rX)r;rn 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{r5rm) rrrrrrr ValueErrorr8rF)r;rrrrrrrtest_comparison_bad_axisDs z1TestSeriesFlexComparison.test_comparison_bad_axisz values, op)FFTFeq)TTFTnele)FFFFlt)FTTFge)FTFFgtcCsVtgdtdd}tgdtdd}t|||}t|tdd}t||dS)Nr5rLr+abcrkr+r+r+bcdabcdrlistr8r-rX)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)NrrrkrrrTrr)r;rr>rUrrr?r@rrr#test_comparison_flex_alignment_fill^s zJdS)NrrLrkr+bool)rrcr8dtypesrr)r;r.rYconstr?r@rrrtest_ser_flex_cmp_return_dtypes~s  z4TestSeriesComparison.test_ser_flex_cmp_return_dtypescCsPtgdtdd}|jdd}d}t|||j}td}||ksLJdS)NrrLrkrr+r)rrcrr8rrr)r;r.rYrrr?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)rzrrOr5r+z US/CentralZinferry)startrrOcategory) r rr7rO tz_localizerZ DatetimeIndexrr9rrZinterval_ranger) r;rrr>rrYr?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) NrrLr^)invalidOr)FTFFFF) rrrrVr Znangtrdrr-r:rrX)r;rrr?r@srrZrrrtest_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)rgrhccZacbrgr can only compare equality or notr{)rrrr_rrrr)r;rgrhrrrfrrrrtest_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{rhF)ZorderedTz1Invalid comparison between dtype=category and strrr)TTT)rrrrrrrr-rX)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)r5r5)r5r+FT)rrr)r;rr?r@rrrtest_comparison_tuples's(             z+TestSeriesComparison.test_comparison_tuplescCsDttdgtddgg}|tdgk}tddg}t||dS)Nr5r+TF)r frozensetr-rXr;rYr?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 rrrVZdropnarbr`r9rrWrrr-rX)r;rrYvalr?r@rrr"test_comparison_operators_with_nasMs  z7TestSeriesComparison.test_comparison_operators_with_nascCsPtgdgdtd}gd}t|jdk|s4Jt|jdk|sLJdS)N)rLrSr,rlr)TTFTTr,)rfloatr-Z equalContentsr`)r;r%r@rrrtest_ne^szTestSeriesComparison.test_nez left, rightrKABCr'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{)rFrrrrrr)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)rFrGrHrrrrsrtrrrrrrrrrrrrrrrrrrrqs8    %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/Easternrkz Europe/Berlin)r rrr rrrrr`rpytzrr-rX) 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) Nrrrrrkrz0Cannot join tz-naive with tz-aware DatetimeIndexr{) r rrrrr!rrrr Exception)r;rrYrrrrr"test_series_add_aware_naive_raisess &z;TestTimeSeriesArithmetic.test_series_add_aware_naive_raisescCsHttddd}tjjdd}||}ttgd}t||dS)Nz 2012-01-01rLrrl)days)z 2011-12-26z 2011-12-27z 2011-12-28)rr roffsetsZ DateOffset to_datetimer-rX)r;r<offsetr?r@rrrtest_datetime_understoods z1TestTimeSeriesArithmetic.test_datetime_understoodcCstddd}ttjd|d}|dd}|}dd|jD|_||}||}||dd}|jd|_t ||t ||dS)Nrwrrkr,cSsg|] }|qSrr).0r'rrr r)zWTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindex..) r rrrrrr`Z _with_freqr-rX)r;rr%Zts_sliceZts2r?rr@rrr*test_align_date_objects_with_datetimeindexs   zCTestTimeSeriesArithmetic.test_align_date_objects_with_datetimeindexN)rFrGrHrr 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) NrrrNr5_)andZrandxorZrxororZrorr6z*operating as a set operation is deprecatedr{r+) rrcrFstrip startswithr8r FutureWarningr-Zassert_produces_warning isinstancerO)r;rrrrZall_binary_operatorsr>rrrOZ 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/r0r1rur3r2modr4cSsg|] }d|qS)r6r)rr>rrrrr)zGTestNamePreservation.test_binop_maybe_preserve_name..changed)rOr1rr/r8)r;rr?cprr>rYrrrtest_binop_maybe_preserve_names,  z3TestNamePreservation.test_binop_maybe_preserve_namecCs|d}|j|jksJdSr*rN)r;rr?rrrtest_scalarop_preserve_namesz0TestNamePreservation.test_scalarop_preserve_nameN)rFrGrHrrrsrtrtuplerarrayrrrr 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)Nr5rr+rLr)r;Zdtype1Zdtype2Zdtype_expectedZ dtype_mulrrr@rrrtest_series_inplace_ops!s   z-TestInplaceOperations.test_series_inplace_opsN)rFrGrHrrrsrtr,rrrrr) s  r)cCs|}t|dkrtdtj|jd<|dk}|jdr.r,r2rg)r3ii i))rgr'r5)rgr'r+)rgr]r5)rgr]r+) rrcrZ MultiIndexZ from_productrlocZ IndexSlice from_tuplesr-rX)rrr?r@rrr'test_series_varied_multiindex_alignmentjs&$r6)*rrr9numpyrrrrZpandas._libs.tslibsrZpandas.core.dtypes.commonrrZpandasrrrrr r r r Zpandas._testingZ_testingr-Z pandas.corer rZpandas.core.computationrrZfixturerr#r$rvrrrrr)r.r6rrrrs8  $   m 6&=N.