a < b<@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..indexrAcsi|]}|||kqSrrrUrXrrr[r\)z=Invalid comparison between dtype=datetime64\[ns\] and ndarrayzinvalid type promotionzbThe DTypes and do not have a common DType.|rB) rrAr^r7r8joinrGrH TypeError)rargZarg2resultexpectedZmsgsrJrrXrtest_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)NrOrLrPZ20010102iʚ;rM)Zdates1dates2ZintcolZfloatcolZ stringcolg?rfltgtlegeeqne)rhrgrjrirkrl)rkrlZ20010109zL'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'Timestamp'rBZnatzJ'(<|>)=?' not supported between instances of 'numpy.ndarray' and 'NaTType')rpd date_rangerrDrandintrErFr7ZrandsNaTlocrandlenitemsgetattroperator Timestampr8rGrHra) rr9opsleftrightZleft_fZright_frdrcrJrrrtest_timestamp_comparesB       00 0z+TestFrameComparisons.test_timestamp_comparecCs\tddgddgg}tddgddgg}||k}|r@J||k}|sXJdS)Nz 1989-08-01rSrTcd)rr-all)rr9otherrcrrrtest_mixed_comparisons z*TestFrameComparisons.test_mixed_comparisoncCs^ttdd}tddgddgddgg}|dk}t|||ddgk}t||dS)Nr=r}FTr}r}r})rrarangereshaper7r8)rr9rdrcrrr test_df_boolean_comparison_errors   z5TestFrameComparisons.test_df_boolean_comparison_errorcCs>ttjddtdgdd}|d}|r:JdS)Nrr2BCr])rrrDrErange__eq__r-rr9rcrrrtest_df_float_none_comparisons  z2TestFrameComparisons.test_df_float_none_comparisoncCstddddddg}|jdk}t|||jddddft|| |jddddf|jdk}t|||jddddft|| |jddddfdS)Nr|r3rRr}r4r)rrSr7r8rqrT)rr9Zmask_aZmask_brrrtest_df_string_comparison s "$ "z.TestFrameComparisons.test_df_string_comparisonN)r.r/r0r<rKrGmark parametrizerrDrormrnrer{rrrrrrrrr1HsP         - *0 r1c@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])rqr'reindexr^rAr7r8)methopr9rZpart_orsZxprrr_check_unaligned_frameszMTestFrameFlexComparisons.test_bool_flex_frame.._check_unaligned_framerkrlrhrgrjrir$Unable to coerce to Series/DataFramerBcSs|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,) rkrlr7r8rTrFrhrirjrg)r9idx_sercol_serZidx_eqZcol_eqZidx_neZcol_neZidx_gtZcol_gtZidx_leZcol_leZidx_geZcol_geZidx_ltZcol_ltrrr _test_seq8s6      z@TestFrameFlexComparisons.test_bool_flex_frame.._test_seqrr)rrDrEronesshaperkvaluesrrlr-rurvr7r8nanrGrHrIrrqrhrgrjri)rdataZ other_datar9rndim_5rrforJrrrrrrrtest_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@rSr_z9'>' not supported between instances of '.*' and 'complex'z unorderable types: .*complex\(\)rB) rrrrr`rGrHrarhrrlr) rarrZarr2r9df2rJrZarr3df3rrr"test_bool_flex_frame_complex_dtypens2  (0*  *.z;TestFrameFlexComparisons.test_bool_flex_frame_complex_dtypecCsTtddtjdgi}tddtdgi}||}tdgdi}t||dS)NrWr3r4)FTF)rrrrnowrlr7r8)rdf1rrcr: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)rrmrpilocitemrkrlrrrrtest_flex_comparison_nats     z1TestFrameFlexComparisons.test_flex_comparison_natopnamercCsPtgdgdd}d}t|||j}t|tdgtt gddS)Nr|r}r?@@rXr}r^) rrudtypes value_countsr7assert_series_equalrrrr+)rrr9constrcrrr&test_df_flex_cmp_constant_return_typessz?TestFrameFlexComparisons.test_df_flex_cmp_constant_return_typescCs^tgdgdd}d}|jdd}t|||j}t|tdgt t gddS)NrrrXr}rr) rrrurrr7rrrrr+)rrr9remptyrcrrr,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) Nrr2rrrFr|r}r2rr)rmZ IntervalIndexZ from_breaksrrrkr7r8)riir9serr;rdZser2res2rrr-test_df_flex_cmp_ea_dtype_with_ndarray_seriess  zFTestFrameFlexComparisons.test_df_flex_cmp_ea_dtype_with_ndarray_seriesN) r.r/r0rrrrrGrrrrrrrrrrsX'  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|]}||qSrrrUr9rrrr[r\z?TestFrameFlexArithmetic.test_floordiv_axis0..) rrrrfloordivrAr7r8r)rrrcrdresult2rrrtest_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|]}||qSrrrUr9rrrrr[r\zLTestFrameFlexArithmetic.test_floordiv_axis0_numexpr_path..) rurvrrr rrrAr7r8r)rrrrcrdrrrr test_floordiv_axis0_numexpr_paths " z8TestFrameFlexArithmetic.test_floordiv_axis0_numexpr_pathcCsftjddd}tjddd}t|}t||d}|j|dd}t||||d}t||dS)N 2016-01-01rLrP1rr|rr)rmrnZtimedelta_rangerraddr7r8)rdtiZtdiZtserr9rcrdrrrtest_df_add_td64_columnwisesz3TestFrameFlexArithmetic.test_df_add_td64_columnwisecCstjddd}tgddd}t||d}t||d}tjdd }|j||d }ttgd d d|d d}t||dS)NrrrP)z1 Dayrpz2 Daystimedelta64[ns]rrr|)days fill_value)z 2016-01-02z 2016-01-03z 2016-01-05zdatetime64[ns]r}) rmrnrr TimedeltaZto_timedelta64rr7r8)rrrr9rfillrcrdrrr$test_df_add_flex_filled_mixed_dtypessz.fr}rr)rur7r8r )rall_arithmetic_operators float_framemixed_float_framerrcrdrrrtest_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)rrrrr2r) rurvr USE_NUMEXPRrr7r8r r ) rrZ int_framemixed_int_framerrrrcrdrrrrtest_arith_flex_frame_mixed,s&     z3TestFrameFlexArithmetic.test_arith_flex_frame_mixedc Csf|}tddD]R}td|}d}tjt|dt|||Wdq1sV0YqdS)Nrr=)r|rrB)rrrrGrHrIru)rrrrZdimrrJrrrtest_arith_flex_frame_raiseUs z3TestFrameFlexArithmetic.test_arith_flex_frame_raisecCs|d}t||d||dd}t||tj|dd|}t||tjtjtdd$|j|jdddWdn1s0Ytjtdd&|j|jddddWdn1s0YdS) Nr|rrrBrrr^r,r) rr7r8rrrGrHNotImplementedErrorr)rrZ const_addrcrrrtest_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)NrStwo)rsubmulmodrrr=rint64rr?g?r^float64)xsrurvr7r8rrdivrrrrrinf) r simple_framer9rowrWrxrrrdrcrrrtest_arith_flex_seriesrs(     z.TestFrameFlexArithmetic.test_arith_flex_seriescCstgtd}tddgd}tddgddggddgd}tjtd d |j|d d Wdn1sh0Ytjtd d $|j|dddd Wdn1s0YdS)Nrr2rr@r|r}rr>rrBErr)robjectrrGrHrrr)rZser_len0Zdf_len0r9rrrtest_arith_flex_zero_len_raisess ,z7TestFrameFlexArithmetic.test_arith_flex_zero_len_raisescCs^tjddtjdddgdd}td|itd d }|dd }|jd dd }t||dS) Nrr|rr>rfloatrr3r=rr}r) rrrrrfillnarr7r8)rZdatr9r: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)rrr7r8)rrrrdrcrrr'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)NrLrr2) rrrrDrrrurAr7r8r%r)rrr9rdrcrrr&test_arithmetic_with_duplicate_columnss   z>TestFrameFlexArithmetic.test_arithmetic_with_duplicate_columnslevelrNcCstgdgdd}|jd|_tgdgdd}|jddg|_|j||d}tgdgdd}|jddg|_t||dS) Nrr|r}rrZL1rrr))r2r)r2DZL2)r )rrAZ set_namesrr7r8)rr rrrcrdrrrtest_broadcast_multiindexsz1TestFrameFlexArithmetic.test_broadcast_multiindex)r.r/r0rrGrZskipifr rrrrrrrrrrrrr 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) Nrprrr|r}rr>rrrr7r8)rrr9rcrdrrrtest_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}TFrrAr^r|r>rrAr^r rrrrrrAr^rr7r8)rrr9rowlikerdrcrrr'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 rLrr)rrr9collikerdrcrrr'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|r2rr)td&mark_array_manager_not_yet_implementedrrrrrrurqsqueezerAr^r7r8) rrrusing_array_managerrrr9rexvalsrdrcrrr)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)rrVrYrrr 4r\zPTestFrameArithmetic.test_df_arith_2d_array_collike_broadcasts..r)rrrrrrrrurZ common_typerrAr^r7r8) rrrr rrr9rr!rrdrcrrr)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)rr9rckindsrrrtest_df_bool_mul_int;sz(TestFrameArithmetic.test_df_bool_mul_intcCsDtgdgdd}||}tgdgdd}t||dS)NrSrTr~rr)ZaaZbbccr}r>r=rr7r8)rryrcrdrrrtest_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)rAZ is_uniquerIr7r)r9rrcrWrrr_test_opUs   z@TestFrameArithmetic.test_arith_getitem_commute.._test_opcSs||SrrrXrrrr)er\z@TestFrameArithmetic.test_arith_getitem_commute..cSs||SrrrXrrrr)fr\cSs||SrrrXrrrr)gr\cSs||SrrrXrrrr)hr\cSs||SrrrXrrrr)ir\cSs||SrrrXrrrr)kr\cSs||SrrrXrrrr)lr\cSs||SrrrXrrrr)mr\cSs||SrrrXrrrr)nr\cSs||SrrrXrrrr)or\)rrvrrrtruedivrr)rr9r3rrrtest_arith_getitem_commuteRs$      z.TestFrameArithmetic.test_arith_getitem_commuterr|r}r|r}rcCsDtddgddgd}tddgddgd}||}t||dS)Nr|rr}rr1)rrr9rdrcrrr&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|rLf8rrZonerthreer.rrSr]rrrr^)rrrrrrr^rAr7r8rrrFrDrrr)rr9Zval1addedZval2Zval3rrrtest_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}r@rLcsg|]}t|qSr)ru)rVnnumrrr r\zVTestFrameArithmetic.test_operations_with_interval_categories_index..)rmZCategoricalIndexZinterval_rangerrur7r8)rrindrr9rcrdrrAr.test_operations_with_interval_categories_indexszBTestFrameArithmetic.test_operations_with_interval_categories_indexcCsttdtdgtdtdgdddgd}|dg}||}ttd td gtjtjgdddgd}t||dS) NZ2019Z2020Z2018Z2021)r3r4r3r4r@r)rrmrwrrrr7r8)rr9rrcrdrrrtest_frame_with_frame_reindexs  z1TestFrameArithmetic.test_frame_with_frame_reindexz value, dtype)r|i8)rr8)r8)y? complex128)rHrI)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+r2rrMrLrIz3ufunc 'remainder' not supported for the input typesz-numpy boolean subtract, the `-` operator, is zcannot perform __z7__ with this index type: (DatetimeArray|TimedeltaArray)rBrz+operator '.*' not implemented for .* dtypes)rvr4rrrrrrrrr UserWarningr.rGrHrar7assert_produces_warningrrrrr) rrrrrskipelemr9invalidwarnrJrcrdrrrtest_binop_othersd     P .. ,z$TestFrameArithmetic.test_binop_otherN) r.r/r0rrrr"r%r-r2r5rGrrrrrrr7r<rErFZ timedelta64Z datetime64rvrrrr4rrrTrrrrrsN  *    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}r2rr@rFr]zM8[ns])rrrDrErrrrrrAr7r8rO FutureWarningr^r*)r9rrcrdrrrr,test_frame_with_zero_len_series_corner_casess    &  &rVcCsBtddgtjd}tddgddgd}||}|}t||dS)Nr2rrArr|r}r)rrrrr7r8)r9rrcrdrrr,test_zero_len_frame_with_series_corner_cases*s rXz(ignore:.*Select only valid:FutureWarningcCsDdtddtjgi}t|}|jdd}tgd}t||dS)NZOner2333333?r|r)r2rYr)rrrrsumr7r)rr9rcrdrrr+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/2011rLH US/Eastern)rQfreqtzrSr]z Europe/Moscow) rmrnrrrDrErs tz_convertr^r`pytzutc)rrngr9Z df_moscowrcrrr*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/2010r2)r_rrr}r|) rmZ period_rangerrrDrErsrrr7r8ZtakeZ permutation)rrdtsrcrdZhalfrrrtest_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(comZisnar7r8)rrr9Zfilledrcrdrrrtest_operators_none_as_na_s       z5TestFrameArithmeticUnsorted.test_operators_none_as_nazop,res)rF)__ne__Tz ignore:elementwise:FutureWarningcCs*t||d}t||us&JdS)Nr3)rur+r)rrr;rrcrrr%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)'Nabcr9r)firstsecondthirdnamesQr)Zvalue1Zvalue2Zvalue3r])rrrrr4)r$@gY@rqr)r r,cs:g|]2\}}jdddd|fddf|qSrrqrVr*vr9idxZoparrrCr\zETestFrameArithmeticUnsorted.test_binary_ops_align..rrvrr:rpcs4g|],\}}jdd|fddf|qSrrwrxrzrrrCr\r2rrSrT)r}r>rrr@r|r}rRZlvl0Zlvl1)r,r )r|r}r|r})r from_productrFrrrrZ sort_indexrmZ IndexSlicerurvrconcatrtr7r8Z reindex_likerr'rArsr^namerr)rr^rrYrcrdZmidxsrs2Zres1rZres3Zres4Zres5Zres6r:r;rrzrtest_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)r`z Asia/Tokyor}r^r|r2rr)rmZ DatetimeIndexrarrrr7r8)rbaseZidx1Zidx2rrr:rrr test_add_with_dti_mismatched_tzssz.test_comp) r7ZmakeTimeDataFramerrrrrvrkrlrgrhrjri)rrrrrrrtest_comparisonsss      z,TestFrameArithmeticUnsorted.test_comparisonscCs\tdddD}tt|}d}tjt|d||dWdn1sN0YdS)NcSsi|]}|ddddqS)r3r4r5)rYrZzrr#rrrr[r\zZTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raises..r.z;'[<>]=?' not supported between instances of 'str' and 'int'rBr)rrurvrGrHra)rZcompare_operators_no_eq_ner9rrJrrr*test_strings_to_numbers_comparisons_raisess  zFTestFrameArithmeticUnsorted.test_strings_to_numbers_comparisons_raisescCst}tj|j|jddf<tjdd|jdk}Wdn1sJ0Ytjdd|dkj}Wdn1s0Yt||dS)Nrr2ignore)rRraise) r7Z makeDataFramerrrqr^Zerrstaterr)rZ missing_dfrdrcrrr'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 shapesrBZABrnr)rrrrrZ atleast_2drtupler7r8rrrGrHrIrrFr^rA) rr9rTZb_rZb_clsttuprdrcZmsg1dZmsg2dZmsg2dbrrrtest_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) NZabcdefgrLrrrr|ZbedcfZbcdefr@) rFrrrrsrrrr'rr7r8) rrAZX_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) NrrrrLrMrr|rr3rr2) rrrrDrorr'r7rZ_mgrr8) rZs_origZdf_origrrr9rrrdrrrtest_inplace_ops_identity,sX           z5TestFrameArithmeticUnsorted.test_inplace_ops_identity) randrrrrorrrr4xorcCs|dkr dStgdgdd}d}|dvr:gd|d<|}d |d }d |d }t|||t|||}t||t|}t||ksJdS) NrrrrRr})rrr)TFTrSZ__ir)rr'rur7r8id)rrr9operandZdf_copyZioprdrrrtest_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)rrrrAz5Unable to coerce to Series, length must be 3: given 2r}r6rBr>rr=)rrrrzGUnable to coerce to Series/DataFrame, dimension must be <= 2: (3, 3, 3))rrrDrErmcorerxZalign_method_FRAMErrrr^r7r8rGrHrIrAZzerosrr)rr^rAr9ZalignvalrdrJrrrtest_alignment_non_pandassX $,.,, ,z5TestFrameArithmeticUnsorted.test_alignment_non_pandascCsZtddgddgd}|d}tdt|||Wdn1sL0YdS)Nr=rr)rr7rOrurrr9rTrrrtest_no_warnings z+TestFrameArithmeticUnsorted.test_no_warningcCs`tddgddgd}|d}tjtdd t|||dWdn1sR0YdS)Nr=rrztakes 2 positional argumentsrBr)rrGrHrarurrrrtest_dunder_methods_binarysz6TestFrameArithmeticUnsorted.test_dunder_methods_binarycCsjtjddddd}tjdtd}t|}||d<|t}|| }|| }t ||dS)NrrrrL)rLr|z0.X) rrrrintrrr(rmeanr7r8)rrrrrrcrdrrrtest_align_int_fill_bugs    z3TestFrameArithmeticUnsorted.test_align_int_fill_bugN)!r.r/r0rergrGrrrvrrrr4rkfilterwarningsrmrrrrrrrrrrrrrrrrrrrrrr\Fs<     F <).%  O1: ;r\cCsLtdgdi}tgdd}||}tdtjdtjgi}t||dS)Nr2r rrrrrr7r8)ryrzrcrdrrrtest_pow_with_realignments rcCsptdtjtjtjgi}tdgdi}tdgdi}||}t|||d|d}t||ddS)Nr2r )rrr)rrrr7r8r)ryrzrdrcrrrtest_pow_nan_with_zeros rcCsttjdddgdd}tgdgdd}|d|dd d }t||jd d }|d }||}t ||||d }t ||dS) Nrr)rLrr.r@rrrrrZInt64rW) rrrDrorZto_numpyrrAr(r7r8)r9rrdZdf_earcrrr&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)rrL)r|rrrr}r>rcsi|]}|||qSrr)rVr*ryrzrrr[r\z6test_dataframe_blockwise_slicelike..) rrDrorr'rrrrAr7r8)rrrrZdf4Zdf5r;rdrrr"test_dataframe_blockwise_slicelike srz df, col_dtyperrg@rhabr@rrTrcCsTtdtjgdtjggtdd}|d|i}|tdgtdd}t||dS) Nr=rrr@rTgrSr)rrrrFr(rr7r8)r9Z col_dtyperdrcrrr/test_dataframe_operation_with_non_numeric_types!s rcCsZtdggdgd}tgdggdd}||}ttjddgggdd}t||dS)Nrrp)rrAr )rorprpr@r)rrrcrdrrr"test_arith_reindex_with_duplicates0s rto_addr|cCstddgddgd}dt|dd}tjt|d||Wdn1sV0Ytjt|d||Wdn1s0YdS)Nr|r}rXzUnable to coerce list of rz to Series/DataFramerB)rr)rGrHrI)rr9rJrrr"test_arith_list_of_arraylike_raise9s &rcCsTtdgdi}|d}|j}|d7}|j|us4Jtdgdi}t||dS)Nr2rr|)r}rr>)rZ_valuesr7r8)r9rvalsrdrrr%test_inplace_arithmetic_series_updateEsrcCstdggdgtjdgdggddgdd}tdggdgtdgddd}tdggdgtjdgdggddgdd}||}t||dS) zP Regression test for: https://github.com/pandas-dev/pandas/issues/33765 r|rSrrTrrr])r~N)rrr|rr7r8)rrrdrcrrr!test_arithemetic_multiindex_alignRsrcCsFtdtdtd}|d}ttdtdtd}t||dS)NTrZcdrr)rrFrrr7r8)r9rcrdrrrtest_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.r/r0propertyrrrrrrrrpsrcs>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.r/r0 _metadatarrrr __classcell__rrrrrys   rZ some_datarrrr}r0)rrL r)Zsdfrcrdrrr+test_frame_op_subclass_nonclass_constructorms  r)> collectionsrrrrvrZnumpyrrGrbZpandas.util._test_decoratorsutilZ_test_decoratorsrZpandasrmrrrrZpandas._testingZ_testingr7Zpandas.core.commonrcommonrjZpandas.core.computationrrZ#pandas.core.computation.expressionsr r Zpandas.tests.frame.commonr r Zfixturerrr1rrrrVrXrrr[r\rrrrrrFrrrrrrrrrrrsr     MBwH  #    .