a oEb<@s4ddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl m m Z ddlZddlmZmZmZmZddlmZddlmmZddlmZddlmZm Z ddl!m"Z"m#Z#ej$dd dd gd d gd ddZ%GdddZ&GdddZ'GdddZ(GdddZ)GdddZ*ddZ+ddZ,ej-.ddd Z/Gd!d"d"Z0d#d$Z1d%d&Z2d'd(Z3d)d*Z4ej-5d+ed,d-gd.d/gge6d0d1d2fed,d3gd.d3gge6d0d1d4fgd5d6Z7d7d8Z8ej-5d9ed:d:gged:d:ged:d:gggd;d<Z9d=d>Z:d?d@Z;dAdBZz@TestFrameComparisons.test_comparison_invalid..indexrBcsi|]}|||kqSrrrVrYrrr\r])z=Invalid comparison between dtype=datetime64\[ns\] and ndarrayzinvalid type promotionzbThe DTypes and do not have a common DType.|rC) rrBr_r8r9joinrHrI TypeError)rargZarg2resultexpectedZmsgsrKrrYrtest_comparison_invalidas40   &&(z,TestFrameComparisons.test_comparison_invalidc Csttjdddtjdddtjjdddtjdtt dd}tj |j tj t |dkd f<d d d d ddd}|D]\}}tt|}tt|}|dvr||td}|td|}t||nd} tjt| d ||tdWdn1s0Ytjt| d |td|Wdn1sN0Y|dvr||td}|td|}t||qd} tjt| d ||tdWdn1s0Ytjt| d |td|Wdq1s 0YqdS)NrPrMrQZ20010102iʚ;rN)Zdates1dates2ZintcolZfloatcolZ stringcolg?rgltgtlegeeqne)rirhrkrjrlrm)rlrmZ20010109zL'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'Timestamp'rCZnatzJ'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'NaTType')rpd date_rangerrErandintrFrGr8ZrandsNaTlocrandlenitemsgetattroperator Timestampr9rHrIrb) rr:opsleftrightZleft_fZright_frerdrKrrrtest_timestamp_comparesB       00 0z+TestFrameComparisons.test_timestamp_comparecCs\tddgddgg}tddgddgg}||k}|r@J||k}|sXJdS)Nz 1989-08-01rTrUcd)rr.all)rr:otherrdrrrtest_mixed_comparisons z*TestFrameComparisons.test_mixed_comparisoncCs^ttdd}tddgddgddgg}|dk}t|||ddgk}t||dS)Nr>r~FTr~r~r~)rrarangereshaper8r9)rr:rerdrrr test_df_boolean_comparison_errors   z5TestFrameComparisons.test_df_boolean_comparison_errorcCs>ttjddtdgdd}|d}|r:JdS)Nrr3BCr^)rrrErFrange__eq__r.rr:rdrrrtest_df_float_none_comparisons  z2TestFrameComparisons.test_df_float_none_comparisoncCstddddddg}|jdk}t|||jddddft|| |jddddf|jdk}t|||jddddft|| |jddddfdS)Nr}r4rSr~r5r)rrTr8r9rrrU)rr:Zmask_aZmask_brrrtest_df_string_comparison s "$ "z.TestFrameComparisons.test_df_string_comparisonN)r/r0r1r=rLrHmark parametrizerrErprnrorfr|rrrrrrrrr2HsP         - *0 r2c@sheZdZddZddZddZddZej d gd d d Z ej d gd d dZ ddZ dS)TestFrameFlexComparisonsc Cs$tjdd}tjdd}t|}t|}t|jd}dd}||jsXJ| |j rlJdD]}t ||}t t |} t ||| ||||| ||t ||j| ||jt |d| |dd} t |tj| |tjtjt| d ||Wdqp1s(0Yqpd d } ttjd} ttjd} | || | | || j| jtj|jd <||}|jd rJ| |}|jd sJ||}|jd rJ||}|jd rJ||}|jd rJ||}|jd r JdS) Nr)r}rcSsJ|jddddf}||}|||j|j|jd}t||dS)Nrr}r^)rrr(reindexr_rBr8r9)methopr:rZpart_orsZxprrr_check_unaligned_frameszMTestFrameFlexComparisons.test_bool_flex_frame.._check_unaligned_framerlrmrirhrkrjr$Unable to coerce to Series/DataFramerCcSs|j|dd}||}|j|dd}||}t||t|kt|| t|| t||j|jt||t|t||jt|ddt||jt|dd|j|dd}||}|j|dd} ||} t||t|kt|| t|| t||j|j|j |dd} | |} |j |dd} | |}t| |t|kt| | t| | t| |j |jdS)Nrr-) rlrmr8r9rTrGrirjrkrh)r:idx_sercol_serZidx_eqZcol_eqZidx_neZcol_neZidx_gtZcol_gtZidx_leZcol_leZidx_geZcol_geZidx_ltZcol_ltrrr _test_seq8s6      z@TestFrameFlexComparisons.test_bool_flex_frame.._test_seqrr)rrErFronesshaperlvaluesrrmr.rvrwr8r9nanrHrIrJrrrrirhrkrj)rdataZ other_datar:rndim_5rrforKrrrrrrrtest_bool_flex_framesJ  *         z-TestFrameFlexComparisons.test_bool_flex_framec Csttjddtjg}tdtjddg}td|i}td|i}dddg}tjt|d ||Wdn1s~0Ytjt|d "|d|dWdn1s0Ytjt|d |j |j kWdn1s0Y| |}|j sJtdtjdg}td|i}tjt|d |dWdn1sj0Ytjt|d |ddWdn1s0Ytjt|d |j dkWdn1s0YdS) Nr}r>y@rTr`z9'>' not supported between instances of '.*' and 'complex'z unorderable types: .*complex\(\)rC) rr rrrarHrIrbrirrmr) rarrZarr2r:df2rKrZarr3df3rrr"test_bool_flex_frame_complex_dtypens2  (0*  *.z;TestFrameFlexComparisons.test_bool_flex_frame_complex_dtypecCsTtddtjdgi}tddtdgi}||}tdgdi}t||dS)NrXr4r5)FTF)rrrrnowrmr8r9)rdf1rrdr;rrr!test_bool_flex_frame_object_dtypes  z:TestFrameFlexComparisons.test_bool_flex_frame_object_dtypecCsttjg}|tjk}|jddus,J|tj}|jddusNJ|tjk}|jddusnJ|tj}|jddusJdS)NrFT)rrnrqilocitemrlrmrrrrtest_flex_comparison_nats     z1TestFrameFlexComparisons.test_flex_comparison_natopnamercCsPtgdgdd}d}t|||j}t|tdgtt gddS)Nr}r~r?@@rYr~r_) rrvdtypes value_countsr8assert_series_equalrrrr,)rrr:constrdrrr&test_df_flex_cmp_constant_return_typessz?TestFrameFlexComparisons.test_df_flex_cmp_constant_return_typescCs^tgdgdd}d}|jdd}t|||j}t|tdgt t gddS)NrrrYr~rr) rrrvrrr8rrrrr,)rrr:remptyrdrrr,test_df_flex_cmp_constant_return_types_emptys zETestFrameFlexComparisons.test_df_flex_cmp_constant_return_types_emptycCstjgd}t||d}tddg}|j|dd}tddgddgd}t||tddgdd gd }|j|dd}t||dS) Nrr3rrrFr}r~r3rr)rnZ IntervalIndexZ from_breaksrrrlr8r9)riir:serr<reZser2res2rrr-test_df_flex_cmp_ea_dtype_with_ndarray_seriess  zFTestFrameFlexComparisons.test_df_flex_cmp_ea_dtype_with_ndarray_seriesN) r/r0r1rrrrrHrrrrrrrrrrsX'  rc@seZdZddZejje ddejdddgdd Z d d Z d d Z ddZ ejdgdddZ ddZddZddZddZddZddZejdgd d!d"Zejd#d$d%gd&d'Zd%S)(TestFrameFlexArithmeticcsttd}t|tdjdd}tfddjD}t||jjdd}t||dS)Nrrrrcsi|]}||qSrrrVr:rrrr\r]z?TestFrameFlexArithmetic.test_floordiv_axis0..) rrrrfloordivrBr8r9r)rrrdreresult2rrrtest_floordiv_axis0s  z+TestFrameFlexArithmetic.test_floordiv_axis0znumexpr not installed)reasonrrpowcstt|ttdtdddd}t|dd<dt|dd}tfdd jD}t ||t|j dd}t ||dS) Ndr}rrrrcsi|]}||qSrrrVr:rrrrr\r]zLTestFrameFlexArithmetic.test_floordiv_axis0_numexpr_path..) rvrwrrr rrrBr8r9r)rrrrdrerrrr test_floordiv_axis0_numexpr_paths " z8TestFrameFlexArithmetic.test_floordiv_axis0_numexpr_pathcCsftjddd}tjddd}t|}t||d}|j|dd}t||||d}t||dS)N 2016-01-01rMrQ1rr}rr)rnroZtimedelta_rangerraddr8r9)rdtiZtdiZtserr:rdrerrrtest_df_add_td64_columnwisesz3TestFrameFlexArithmetic.test_df_add_td64_columnwisecCstjddd}tgddd}t||d}t||d}tjdd }|j||d }ttgd d d|d d}t||dS)NrrrQ)z1 Dayrqz2 Daystimedelta64[ns]rrr})days fill_value)z 2016-01-02z 2016-01-03z 2016-01-05zdatetime64[ns]r~) rnrorr TimedeltaZto_timedelta64rr8r9)rrrr:rfillrdrerrr$test_df_add_flex_filled_mixed_dtypessz.fr~rr)rvr8r9r )rall_arithmetic_operators float_framemixed_float_framerrdrerrrtest_arith_flex_frames   z-TestFrameFlexArithmetic.test_arith_flex_framer)__add____sub____mul__c Cstt|}t||d|}||d|}d} |dvrBddd} n|dvrRddi} tjrr|dkrrd|dj| d<t||t|| d t||d|}||d|}t||t|ddid t||d|}||d|}t||dS) Nr~)rZuint64)rr)rrrrr3r) rvrwr USE_NUMEXPRrr8r9r r ) rrZ int_framemixed_int_framerrrrdrerrrrtest_arith_flex_frame_mixed,s&     z3TestFrameFlexArithmetic.test_arith_flex_frame_mixedc Csf|}tddD]R}td|}d}tjt|dt|||Wdq1sV0YqdS)Nrr>)r}rrC)rrrrHrIrJrv)rrrrdimrrKrrrtest_arith_flex_frame_raiseUs z3TestFrameFlexArithmetic.test_arith_flex_frame_raisecCs|d}t||d||dd}t||tj|dd|}t||tjtjtdd$|j|jdddWdn1s0Ytjtdd&|j|jddddWdn1s0YdS) Nr}rrrCrrr_r-r) rr8r9rrrHrINotImplementedErrorr)rrZ const_addrdrrrtest_arith_flex_frame_corner`s 2z4TestFrameFlexArithmetic.test_arith_flex_frame_cornerc Csj|}|d}|d}gd}|D]L}t||}tt|}t|||||t||dd||j|jq"t|j|dd||t||||t|j|dd|j|jtt d ddd }tt j t j gd d gd d gg}|j|dd d} t| |tt d ddd }tt j t j gd d gd d gg}|j|dd d} t| |dS)NrTtwo)rsubmulmodrrr>rint64rr?g?r_float64)xsrvrwr8r9rrdivrrrrrinf) r simple_framer:rowrXryrrrerdrrrtest_arith_flex_seriesrs(     z.TestFrameFlexArithmetic.test_arith_flex_seriescCstgtd}tddgd}tddgddggddgd}tjtd d |j|d d Wdn1sh0Ytjtd d $|j|dddd Wdn1s0YdS)Nrr3rrAr}r~rr?rrCErr)robjectrrHrIrrr)rZser_len0Zdf_len0r:rrrtest_arith_flex_zero_len_raisess ,z7TestFrameFlexArithmetic.test_arith_flex_zero_len_raisescCs^tjddtjdddgdd}td|itd d }|dd }|jd dd }t||dS) Nrr}rr?rfloatrr4r>rr~r) rr rrrfillnarr8r9)rZdatr:r;r<rrrtest_flex_add_scalar_fill_values z7TestFrameFlexArithmetic.test_flex_add_scalar_fill_valuecCsVtgdgdd}tgdgdd}tgdgdd}||}t||dS)N)r}r~rr?r)r}r~r}r~rrr)rr~rr~r~)r}r}r~r~r)rrr8r9)rrrrerdrrr'test_sub_alignment_with_duplicate_indexs  z?TestFrameFlexArithmetic.test_sub_alignment_with_duplicate_index)rrr __truediv__cCsjttdtjdd}t|||}ddg|_ddg|_t|||}t||t ||j dS)NrMrr3) rrrrErsrvrBr8r9r&r)rrr:rerdrrr&test_arithmetic_with_duplicate_columnss   z>TestFrameFlexArithmetic.test_arithmetic_with_duplicate_columnslevelrNcCstgdgdd}|jd|_tgdgdd}|jddg|_|j||d}tgdgdd}|jddg|_t||dS) Nrr}r~rrZL1rrr))r3r)r3DZL2)r )rrBZ set_namesrr8r9)rr rrrdrerrrtest_broadcast_multiindexsz1TestFrameFlexArithmetic.test_broadcast_multiindex)r/r0r1rrHrZskipifr rrrrrrrrrrr r r rrrrrrs&   (    rc@s eZdZddZddZddZddZd d Zd d Zd dZ ddZ e j dddgdeddgeddeddggddZddZddZddZe j ddd d!d"d#d$ed%d&d'fed%d&d(fge j j d)ejejejejejejgd*d+d,d-d.Zd/S)0TestFrameArithmeticcCsJtddgdd}tddgddgg}||}t||d}t||dS) Nrqrrr}r~rr?rrrr8r9)rrr:rdrerrrtest_td64_op_nat_castings z,TestFrameArithmetic.test_td64_op_nat_castingcCstddd}t|ddggdd}|dgddf}|jd|jdfksRJtdd gd dgdd gg|j|j|jd }||}t ||||}t ||dS) Nr>rr~TFrrBr_r}r?rrBr_r rrrrrrBr_rr8r9)rrr:rowlikererdrrr'test_df_add_2d_array_rowlike_broadcastss z;TestFrameArithmetic.test_df_add_2d_array_rowlike_broadcastscCstddd}t|ddggdd}|dddgf}|j|jd dfksRJtddgd dgd d gg|j|j|jd }||}t ||||}t ||dS)Nr>rr~TFrrr}rr rMrr)rrr:collikererdrrr'test_df_add_2d_array_collike_broadcastss z;TestFrameArithmetic.test_df_add_2d_array_collike_broadcastsc Cs|}|r|dvrt|tddd}t|ddggdd}|d gddf}|jd |jd fkslJt|jd || t|jd || t|jd || g}t||j |j d} t|||} t | | dS) N__rmod__ __rfloordiv__r>rr~TFrrr}r3rr)td&mark_array_manager_not_yet_implementedrrrrrrvrrsqueezerBr_r8r9) rrrusing_array_managerrrr:rexvalsrerdrrr)test_df_arith_2d_array_rowlike_broadcastss  z=TestFrameArithmetic.test_df_arith_2d_array_rowlike_broadcastsc Cs|}|r|dvrt|tddd}t|ddggdd}|ddd gf}|j|jd d fkslJt|d||t|d||d }d} |dvrtj d d | D} t||j |j | d} t|||} t | | dS)Nrr>rr~TFrrr}r)TFcss|] }|jVqdSr)rrWrZrrr 4r]zPTestFrameArithmetic.test_df_arith_2d_array_collike_broadcasts..r)r r!rrrrrrvr"Z common_typerrBr_r8r9) rrrr#rrr:rr$rrerdrrr)test_df_arith_2d_array_collike_broadcastss   z=TestFrameArithmetic.test_df_arith_2d_array_collike_broadcastscCshtddgddgg}|d}|jdd}|dksCr]z:TestFrameArithmetic.test_df_bool_mul_int..icSs|jSrr)r+rrrr,Gr])rrapplyr)rr:rdkindsrrrtest_df_bool_mul_int;sz(TestFrameArithmetic.test_df_bool_mul_intcCsDtgdgdd}||}tgdgdd}t||dS)NrTrUrrr)ZaaZbbccr~r?r>rr8r9)rrzrdrerrrtest_arith_mixedJsz$TestFrameArithmetic.test_arith_mixedcCstddgddgd}dd}||tj||tj||tj||tj||tj||tj||dd ||d d ||d d ||d d ||d d ||dd ||dd ||dd ||dd ||dd dS)Ng?gffffff @g@g333333rcSsD||d}|jjstd|jD]}t|||||dq dS)Nr}z*Only unique columns supported by this test)rBZ is_uniquerJr8r)r:rrdrXrrr_test_opUs   z@TestFrameArithmetic.test_arith_getitem_commute.._test_opcSs||SrrrYrrrr,er]z@TestFrameArithmetic.test_arith_getitem_commute..cSs||SrrrYrrrr,fr]cSs||SrrrYrrrr,gr]cSs||SrrrYrrrr,hr]cSs||SrrrYrrrr,ir]cSs||SrrrYrrrr,kr]cSs||SrrrYrrrr,lr]cSs||SrrrYrrrr,mr]cSs||SrrrYrrrr,nr]cSs||SrrrYrrrr,or])rrwrrrtruedivrr)rr:r6rrrtest_arith_getitem_commuteRs$      z.TestFrameArithmetic.test_arith_getitem_commuterr}r~r}r~rcCsDtddgddgd}tddgddgd}||}t||dS)Nr}rr~rr4)rrr:rerdrrr&test_arith_alignment_non_pandas_objectqsz:TestFrameArithmetic.test_arith_alignment_non_pandas_objectcCs2ttjddddddgdgdd}|d j}t|j||j|jd }t |||t|jj |j |j|jd }t |j |d d |t |d }t|j||j|jd }t |||t|jj |j |j|jd }t |j |dd |tj j|j}t|j||j|jd }t | ||dS)Nr}rMf8rronerthreer1rrTr^rrrr_)rrrrrrr_rBr8r9rrrGrErsr)rr:Zval1addedZval2Zval3rrrtest_arith_non_pandas_object{s$  z0TestFrameArithmetic.test_arith_non_pandas_objectcsn|ttjddd}ddg}t|g|d}dt|}tfdd |Dg|d}t||dS) Nr)startendr}r~rArMcsg|]}t|qSr)rv)rWnnumrrr r]zVTestFrameArithmetic.test_operations_with_interval_categories_index..)rnZCategoricalIndexZinterval_rangerrvr8r9)rrindrr:rdrerrEr.test_operations_with_interval_categories_indexszBTestFrameArithmetic.test_operations_with_interval_categories_indexcCsttdtdgtdtdgdddgd}|dg}||}ttd td gtjtjgdddgd}t||dS) NZ2019Z2020Z2018Z2021)r4r5r4r5rAr)rrnrxrrrr8r9)rr:rrdrerrrtest_frame_with_frame_reindexs  z1TestFrameArithmetic.test_frame_with_frame_reindexz value, dtype)r}i8)rr;)r;)y? complex128)rLrM)Tr,ns)rc Cs~tjdftjdftjdftjdfh}t||}td|j|jgi|jd}tjdftj dftjdftjdftjdftjdftjdftj dftj dfh }||f|vrpd} |dkr|tjks|dkr|tjkrd} n4|dkrd} t } n"|tj urd} t } nd |j d } t jt| d Ht| |||jWdn1sB0YWdn1sb0Yn ||f|vr(|tjtjfvrtjr|d krt } nd} t| |||jWdn1s0Yn@d } t jt| d |||jWdn1s0YnRtd*|||jj} |||j} Wdn1sd0Yt| | dS)Nr,r3rrQrPrMz3ufunc 'remainder' not supported for the input typesz-numpy boolean subtract, the `-` operator, is zcannot perform __z7__ with this index type: (DatetimeArray|TimedeltaArray)rCrz+operator '.*' not implemented for .* dtypes)rwr7rrrrrrrrr UserWarningr/rHrIrbr8assert_produces_warningrrrrr) rrrrrskipelemr:invalidwarnrKrdrerrrtest_binop_othersd     P .. ,z$TestFrameArithmetic.test_binop_otherN) r/r0r1rrrr%r(r0r5r8rHrrrr rrr:r@rIrJZ timedelta64Z datetime64rwrrrr7rrrXrrrrrsN  *    rcCsttjdddddgd}ttjd}||}t|jtj|j d}t ||t t ||k}Wdn1s~0Ytd|j|j d }t ||t|jd |j d}t t ||k}Wdn1s0Ytd|j|j d }t ||dS) Nr>rr~r3rrArFr^zM8[ns])rrrErFrrrrrrBr8r9rS FutureWarningr_r+)r:rrdrerrrr,test_frame_with_zero_len_series_corner_casess    &  &rZcCsBtddgtjd}tddgddgd}||}|}t||dS)Nr3rrBrr}r~r)rrrrr8r9)r:rrdrerrr,test_zero_len_frame_with_series_corner_cases*s r\z(ignore:.*Select only valid:FutureWarningcCsDdtddtjgi}t|}|jdd}tgd}t||dS)NZOner3333333?r}r)r3r]r)rrrrsumr8r)rr:rdrerrr+test_frame_single_columns_object_sum_axis_14s   r_c@seZdZddZddZejdej ej ej ej gddZ ejdd d gejd d d ZddZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zejdgd&d'd(Zd)d*Zd+d,Zd-d.Zd/d0Z d1S)2TestFrameArithmeticUnsortedcCsptjddddd}ttjt||dgd}|d}||}|jj t j usRJ||}|jj t j uslJdS) Nz1/1/2011rMH US/Eastern)rRfreqtzrTr^z Europe/Moscow) rnrorrrErFrt tz_convertr_rdpytzutc)rrngr:Z df_moscowrdrrr*test_frame_add_tz_mismatch_converts_to_utcGs zFTestFrameArithmeticUnsorted.test_frame_add_tz_mismatch_converts_to_utccCstjdddd}ttjt|d|d}||ddd}||}tj|jddd<t |||ddd}|| tj t|}t ||dS) Nz1/1/2000z1/1/2010r3)rcrrr~r}) rnZ period_rangerrrErFrtrrr8r9ZtakeZ permutation)rrhtsrdreZhalfrrrtest_align_frameRs z,TestFrameArithmeticUnsorted.test_align_framercCstgdgddtd}|tj}||d}||dt}d|t|<t |||||}|||t}d|t|<t |||||d}t ||||d|}tj ||dddS) N)r~@{N)r}r~rr?)Zcol1Zcol2rrrF)Z check_dtype) rrrrrr)comZisnar8r9)rrr:Zfilledrdrerrrtest_operators_none_as_na_s       z5TestFrameArithmeticUnsorted.test_operators_none_as_nazop,res)rF)__ne__Tz ignore:elementwise:FutureWarningcCs*t||d}t||us&JdS)Nr4)rvr,r)rrr<rrdrrr%test_logical_typeerror_with_non_validzszATestFrameArithmeticUnsorted.test_logical_typeerror_with_non_validcsLtjtdgdgdggdd}ttddd|gd d tj d D]}t t |ddurlqRt gd gd}t ||d dd}t fdd|D}t||t ddgddg}t ||ddd}t fdd|D}t||qRtddgddgg}ttjddd|dt d d!d"}}d#d$g|j_|} d$| j_j|d d d%} j| d d d%} |j|d d d%} |j| d d d%} |j|d d$d%}|j| d d$d%}ttjgd&gd&gdd|d}| | fD]}t||qd#d$g|j_| | ||fD]}t||q4dS)'Nabcr<r)firstsecondthirdnamesQr)Zvalue1Zvalue2Zvalue3r^)rrrrr7)r$@gY@rur)r r-cs:g|]2\}}jdddd|fddf|qSrrrrWr-vr:idxoparrrGr]zETestFrameArithmeticUnsorted.test_binary_ops_align..rrzrr>rtcs4g|],\}}jdd|fddf|qSrr{r|r~rrrGr]r3rrTrU)r~r?rrrAr}r~rSZlvl0Zlvl1)r-r )r}r~r}r~)r from_productrGrrrrZ sort_indexrnZ IndexSlicervrwrconcatrur8r9Z reindex_likerr(rBrwr_namerr )rr_rrZrdreZmidxsrs2Zres1rZres3Zres4Zres5Zres6r;r<rr~rtest_binary_ops_aligns`       z1TestFrameArithmeticUnsorted.test_binary_ops_aligncCstjgddd}|ddd}|ddd}tdddgi|d }tdddgi|d }tdtjd tjgi|d }t|||dS) N)z 2011-01-01z 2011-01-02z 2011-01-03UTC)rdz Asia/Tokyor~rbr}r3rr)rnZ DatetimeIndexrerrrr8r9)rbaseZidx1Zidx2rrr;rrr test_add_with_dti_mismatched_tzssz.test_comp) r8ZmakeTimeDataFramerrrrrwrlrmrhrirkrj)rrrrrrrtest_comparisonsss      z,TestFrameArithmeticUnsorted.test_comparisonscCs\tdddD}tt|}d}tjt|d||dWdn1sN0YdS)NcSsi|]}|ddddqS)r4r5r6)rZr[zrr&rrrr\r]zZTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raises..r1z;'[<>]=?' not supported between instances of 'str' and 'int'rCr)rrvrwrHrIrb)rZcompare_operators_no_eq_ner:rrKrrr*test_strings_to_numbers_comparisons_raisess  zFTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raisescCst}tj|j|jddf<tjdd|jdk}Wdn1sJ0Ytjdd|dkj}Wdn1s0Yt||dS)Nrr3ignore)rVraise) r8Z makeDataFramerrrrr_Zerrstaterr)rZ missing_dfrerdrrr'test_comparison_protected_from_errstates((zCTestFrameArithmeticUnsorted.test_comparison_protected_from_errstatec Cs^ttdd}tddg}tddg}|j}gd}t|}tddgddgddgg}||k}t |||j |k}t ||j d} d} d } t j t| d ||kWdn1s0Yt j t| d ||k}Wdn1s0Y||k}t |||j |k}t ||j t j t| d ||kWdn1s^0Yt j t| d |j |kWdn1s0Ytddgddgddgg}||k}t ||t j t| d ||k}Wdn1s0Yt j t| d ||k}Wdn1s60Y||k}t |||j |k}t ||j t j t| d ||kWdn1s0Y|j j|jksJttddtd td d }|j|_|j|_t j t| d ||k}Wdn1s0Yt j t| d ||k}Wdn1sP0YdS)Nr>rr~r~r~r~FTz5Unable to coerce to Series, length must be 2: given 3,Unable to coerce to DataFrame, shape must bez4operands could not be broadcast together with shapesrCZABrrr)rrrrr Z atleast_2drtupler8r9rrrHrIrJrrGr_rB) rr:rUZb_rZb_clsttuprerdZmsg1dZmsg2dZmsg2dbrrrtest_boolean_comparisonsb  &&  (* ((  ((z3TestFrameArithmeticUnsorted.test_boolean_comparisonc Cstd}ttdt|dt||tdd}d|jddddf}td}td}|}|||j |d }|||7<|j |d }|}||||j |d } ||||7<|j |d } t ||t || t || |}|||j |d }|||8<|j |d }|}||||j |d } ||||8<|j |d } t ||t || t || dS) NZabcdefgrMrrrr}ZbedcfZbcdefrA) rGrrrrtrrrr(rr8r9) rrBZX_origZZblock1subsXZresult1rrZresult4rrrtest_inplace_ops_alignments>         z6TestFrameArithmeticUnsorted.test_inplace_ops_alignmentc Cs"tgd}ttjjdddddd}|}|}|d7}t||t|d|||usfJ|j |j usvJ|}|}|d7}t ||t |d|||usJ|j |j usJ|}|}|d7}t||t|d||}|}|d7}t ||t |d|||us0J|j |j usBJtjjdddd}t|d d }|}|}|d d7<t|dd d }t ||t |||j |j usJ|}|}|d d7<t|dd d }t ||t |||j |j usJdS) NrrrrMrNrr}rr4rr3) rrrrErprr(r8rZ_mgrr9) rZs_origZdf_origrrr:rrrerrrtest_inplace_ops_identity,sX           z5TestFrameArithmeticUnsorted.test_inplace_ops_identity) randrrrrorrrr7xorcCs|dkr dStgdgdd}d}|dvr:gd|d<|}d |d }d |d }t|||t|||}t||t|}t||ksJdS) NrrrrSr~)rrr)TFTrTZ__ir)rr(rvr8r9id)rrr:operandZdf_copyZioprerrrtest_inplace_ops_identity2fs    z6TestFrameArithmeticUnsorted.test_inplace_ops_identity2c Csgd}gd}ttjdd||d}tjjj}gddtjgdtj dt ddfD]l}t|||d|j d }t |||d d|tgd gd gd d|j d }t |||dd|qXd}ddgdtddgt ddfD]|}tjt|d|||d Wdn1s 0Ytjt|d|||dWdq1s\0Yqtgdgdgdg}t |||d dt||j |jdt |||ddt||j |jdd}tgdgdg}tjt|d|||d Wdn1s0Ytjt|d|||dWdn1sX0Ytd }td}tjt|d|||d Wdn1s0Ytjt|d|||dWdn1s0YdS)Nr)rYrrr^rrr}r?rr_rr)rrrrBz5Unable to coerce to Series, length must be 3: given 2r~r9rCr?rr>)rrrrzGUnable to coerce to Series/DataFrame, dimension must be <= 2: (3, 3, 3))rrrErFrncoreryZalign_method_FRAMEr rrr_r8r9rHrIrJrBZzerosrr)rr_rBr:alignvalrerKrrrtest_alignment_non_pandassX $,.,, ,z5TestFrameArithmeticUnsorted.test_alignment_non_pandascCsZtddgddgd}|d}tdt|||Wdn1sL0YdS)NrArr)rr8rSrvrrr:rUrrrtest_no_warnings z+TestFrameArithmeticUnsorted.test_no_warningcCs`tddgddgd}|d}tjtdd t|||dWdn1sR0YdS)NrArrztakes 2 positional argumentsrCr)rrHrIrbrvrrrrtest_dunder_methods_binarysz6TestFrameArithmeticUnsorted.test_dunder_methods_binarycCsjtjddddd}tjdtd}t|}||d<|t}|| }|| }t ||dS)NrrrrM)rMr}z0.X) rrrrintrr"r)rmeanr8r9)rrrrrrdrerrrtest_align_int_fill_bugs    z3TestFrameArithmeticUnsorted.test_align_int_fill_bugN)!r/r0r1rirkrHrrrwrrrr7rofilterwarningsrqrrrrrrrrrrrrrrrrrrrrrr`Fs<     F <).%  O1: ;r`cCsLtdgdi}tgdd}||}tdtjdtjgi}t||dS)Nr3rrrrrrr8r9)rzr{rdrerrrtest_pow_with_realignments rcCsptdtjtjtjgi}tdgdi}tdgdi}||}t|||d|d}t||ddS)Nr3r)rrr)rrrr8r9r)rzr{rerdrrrtest_pow_nan_with_zeros rcCsttjdddgdd}tgdgdd}|d|dd d }t||jd d }|d }||}t ||||d }t ||dS) Nrr)rMrr1rArrrrrZInt64r[) rrrErprZto_numpyrrBr)r8r9)r:rreZdf_eardrrr&test_dataframe_series_extension_dtypess  rcstjddd}t|}|}tj|jdgdf<|}tj|jddgf<|}tj|jdtddf<|}tj|jdtddf<||f||f||ffD]6\}tfd d jD}t ||qdS) Nri)rrM)r}rrrr~r?rcsi|]}|||qSrr)rWr-rzr{rrr\r]z6test_dataframe_blockwise_slicelike..) rrErprr(rrrrBr8r9)rrrrZdf4Zdf5r<rerrr"test_dataframe_blockwise_slicelike srz df, col_dtyperrg@rlabrArrUrcCsTtdtjgdtjggtdd}|d|i}|tdgtdd}t||dS) NrArrrArUgrTr)rrrrGr)rr8r9)r:Z col_dtypererdrrr/test_dataframe_operation_with_non_numeric_types!s rcCsZtdggdgd}tgdggdd}||}ttjddgggdd}t||dS)Nrrt)rrBr)rsrtrtrAr)rrrdrerrr"test_arith_reindex_with_duplicates0s rto_addr}cCstddgddgd}dt|dd}tjt|d||Wdn1sV0Ytjt|d||Wdn1s0YdS)Nr}r~rYzUnable to coerce list of rz to Series/DataFramerC)rr*rHrIrJ)rr:rKrrr"test_arith_list_of_arraylike_raise9s &rcCsTtdgdi}|d}|j}|d7}|j|us4Jtdgdi}t||dS)Nr3rr})r~rr?)rZ_valuesr8r9)r:rvalsrerrr%test_inplace_arithmetic_series_updateEsrcCstdggdgtjdgdggddgdd}tdggdgtdgddd}tdggdgtjdgdggddgdd}||}t||dS) zP Regression test for: https://github.com/pandas-dev/pandas/issues/33765 r}rTrrUrvr^)rN)rrrrr8r9)rrrerdrrr!test_arithemetic_multiindex_alignRsrcCsFtdtdtd}|d}ttdtdtd}t||dS)NTrcdrr)rrGrrr8r9)r:rdrerrrtest_bool_frame_mult_floatesrcsGfdddtGfdddtdgdgdd}|d }dgd gd d}t||||}t||dS) Ncs,eZdZefddZefddZdS)zEtest_frame_op_subclass_nonclass_constructor..SubclassedSeriescsSrrr!SubclassedSeriesrr _constructorqszRtest_frame_op_subclass_nonclass_constructor..SubclassedSeries._constructorcsSrrr!)SubclassedDataFramerr_constructor_expanddimusz\test_frame_op_subclass_nonclass_constructor..SubclassedSeries._constructor_expanddimN)r/r0r1propertyrrrrrrrrpsrcs>eZdZdgZfddZeddZefddZZS)zHtest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame my_extra_datacs||_tj|i|dSr)rsuperr)rrargskwargs __class__rrr|szQtest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame.__init__cSstt||jSr) functoolspartialr*rr!rrrrszUtest_frame_op_subclass_nonclass_constructor..SubclassedDataFrame._constructorcsSrrr!rrr_constructor_slicedsz\test_frame_op_subclass_nonclass_constructor..SubclassedDataFrame._constructor_sliced) r/r0r1 _metadatarrrr __classcell__rrrrrys   rZ some_datarrrr~r3)rrM r)Zsdfrdrerrr+test_frame_op_subclass_nonclass_constructorms  r)> collectionsrrrrwrnumpyrrHrfZpandas.util._test_decoratorsutilZ_test_decoratorsr ZpandasrnrrrrZpandas._testingZ_testingr8Zpandas.core.commonrcommonrnZpandas.core.computationrrZ#pandas.core.computation.expressionsr r Zpandas.tests.frame.commonr r Zfixturerrr2rrrrZr\rrr_r`rrrrrrGrrrrrrrrrrrsr     MBwH  #    .