gL i3.ddlmZddlZddlZddlZddlmZddlmZ m Z ddl m Z mZmZmZmZddlmZddlZddlmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z*ddl+m,cm-Zddl.m/Z/m0Z0ddl1m,cm2Z3Gd d Z4Gd d Z5d Z6GddZ7GddZ8GddZ9GddZ:GddZ;GddZ<GddZ=ej|jdejejgdZAy))datetimeN)using_string_dtype)algos hashtable) is_bool_dtypeis_complex_dtypeis_float_dtypeis_integer_dtypeis_object_dtype)CategoricalDtype) CategoricalCategoricalIndex DataFrame DatetimeIndexIndex IntervalIndex MultiIndexNaTPeriod PeriodIndexSeries Timedelta Timestampcut date_rangetimedelta_range to_datetime to_timedelta) DatetimeArrayTimedeltaArrayc eZdZdZej j eddej jdddgdZ dZ d Z d Z d Z d Zd ZdZdZdZej jdgdgdgdfgdgdgdfgdgdgdfgdZdZdZdZdZdZd Zd!Zd"Zej jdddgd#Zej jdddgd$Zd%Zej jd&ej@gd'd()ej@gd*d+)ej@gd,d-)gd.Z!ej jd/ej@gd0d()d1fej@gd2d()d3fej@gd4d+)d5fej@gd6d+)d3fej@gd7e")d8fej@gd9e")d:fej@gd;e")dej@gd?e")ej@d@d8ge")fe$j@dAd3ejJdAgdB)e$j@dAd3gdB)fgdCdDgEdFZ&ej jdgdGej@gdHejNdI)ej@d8ejJd@ge")fd8ejJd@d8gej@gdHejNdI)ej@d8ejJd@ge")fgdJZ(ej jdgdKej@gd0ejNdI)ej@d3ejJdAgdL)fd3ejJd3dAgej@gd0ejNdI)ej@d3ejJdAgejR)fgdMZ*ej jde+e,gdNej@gdOejZ)e.d8d@gd8d@gdPQfe/e,gdNej@gdOejZ)e.d8d@gd8d@gdPQfe/e0dRdRgdSTej@d1d1gejZ)e0dRgdSTfgdUZ1dVZ2yW)X TestFactorizecgd}d}tjt|5tj|\}}dddt j gdt j}tj|t j gdt}tj|y#1swYzxYw)N)@?0factorize with argument that is not not a Seriesmatch)rr$r%dtype)?y@r&) tmassert_produces_warning FutureWarningr factorizenparrayintpassert_numpy_array_equalobject)selfr2msglabelsuniquesexpected_labelsexpected_uniquess ]/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/test_algos.pytest_factorize_complexz$TestFactorize.test_factorize_complex6s@  ' ' S A 5#ooe4OFG 5((9BGG< ##FO<88$B&Q ##G-=> 5 5s B44B=zTODO(infer_string)F)reasonstrictsortTc|}|j|\}}t}t|trtj}|j }|j tjk(r|jtj}||}t|tr;|j tk(r(|j tk(r|jt}|r|j}t|} |D cgc]} | j| } } tj | tj"} t%j&|| t%j(||dycc} w)Nr@r*Texact)r0r isinstancer from_tuplesuniquer+r1float16astypefloat32boolr5 sort_valueslistindexasarrayr3r-r4assert_index_equal) r6index_or_series_objr@obj result_codesresult_uniques constructor expected_arrr;expected_uniques_listvalexpected_codess r<test_factorizezTestFactorize.test_factorizeDs%"'*}}$}'?$ n c: &$00Kzz|    +'..rzz:L&|4 sE " &&$. V#/66v>  /;;= !%%5 6FIJs/55c:JJN"''B ##L.A n.>dK Ks>E,cptjdddtjg}t|}|j d\}}tjgdtj }t ddtjg}tj||tj||y) Nr$r%Fuse_na_sentinelrr$rr%r*?@) r1r2nanrr0r3rr-r4rP)r6valuessercodesr9rYr;s r<+test_series_factorize_use_na_sentinel_falsez9TestFactorize.test_series_factorize_use_na_sentinel_falseds1aBFF+,Vnu=w,bgg> #sBFF!34 ##E>: g'78ctjgdt}tj|\}}t j |tjgdttj|d\}}tjgdtj}t j ||tjgdt}t j ||tjdtjddd}tj|\}}tjgd tj}t j ||tjgd |j}t j ||tj|d\}}tjgd tj}t j ||tjgd |j}t j ||tjd ddd}tj|\}}tjgd tj}t j ||tjgd |j}t j ||tj|d\}}tjgd tj}t j ||tjgd |j}t j ||y)N)abrirhrhcrjrjr*rhrirjTrB)rr$r$rrr%r%r%)rr$r%)rornr%r$r@)@@r`r_)rsr_r`rrrq) r1r2r5rr0r-r4r3aranger+)r6itemsrdr9exparrs r< test_basiczTestFactorize.test_basicps:AP/w ##GRXXoV-TUT:whh/rww? ##E3/hhf5 ##GS1ii)$B$/-whhbgg6 ##E3/hhcii8 ##GS148whhbgg6 ##E3/hhcii8 ##GS1iinTrT"-whhbgg6 ##E3/hh0 B ##GS148whhbgg6 ##E3/hh0 B ##GS1rfctddtjddtjg}t j |\}}tj gdtj}tj||tdddtjg}tj||t j |d\}}tj gdtj}tj||tdtjddg}tj||y) NABgQ @)rrrmr$r%rnr*TrB)r%r%rmrnrr$) rr1rainfrr0r2r3r-r4rrP)r6xrdr9rvs r< test_mixedzTestFactorize.test_mixeds Cbffc48 9+whh*"'': ##E3/S#tRVV,- gs+6whh*"'': ##E3/T2663,- gs+rfc6td}td}t||||||g}tj|\}}t j gdtj }tj||t||g}tj||tj|d\}}t j gdtj }tj||t||g}tj||y)Nz20130101 09:00:00.0000420130101rrrr$r$rr*TrB)r$r$r$rrr$) rrrr0r1r2r3r-r4rrPr6v1v2r}rdr9rvs r<test_factorize_datetime64z'TestFactorize.test_factorize_datetime64s 0 1 z " BBB+ ,+whh)9 ##E3/RH% gs+6whh)9 ##E3/RH% gs+rfc6tdd}tdd}t||||||g}tj|\}}t j gdtj }tj||tj|t||gtj|d\}}t j gdtj }tj||tj|t||gy) N201302Mfreq201303rr*TrB) rrrr0r1r2r3r-r4rPrrs r<test_factorize_periodz#TestFactorize.test_factorize_periods H3 ' H3 ' BBB+ ,+whh)9 ##E3/ g{B8'<=6whh)9 ##E3/ g{B8'<=rfc 0td}td}t|||||||g}tj|\}}t j gdtj }tj||tj|t||gtj|d\}}t j gdtj }tj||tj|t||gy)Nz 1 day 1 min1 day)rr$rrr$r$rr*TrB)r$rr$r$rrr$) rrrr0r1r2r3r-r4rPrs r<test_factorize_timedeltaz&TestFactorize.test_factorize_timedeltas - ( ' " BBBB/ 0+whh,BGG< ##E3/ g|RH'=>6whh,BGG< ##E3/ g|RH'=>rfctjdddtjgd}tjt |}dD]}|j ||}tjddd|gtj}t t|t t|k(sJtjtj|||k(tj||y)Nr$r%Or*)rm) na_sentinelr) r1r2rahtObjectFactorizerlenr0r3setr-r4pdisna)r6keyrizerridsexpecteds r<test_factorize_nanz TestFactorize.test_factorize_nanshh1a(4##CH-# 7K//#;/?CxxAq+ 6bggFHs3x=CH $66 66  ' ' h+6M N  ' 'X 6  7rfctjgdd}tjgd}tjt |}|j ||}tjgdtj }tj||tjgdd}tj|jj|y)N)r$r%rnr$r$rint64r*)FFFFFT)mask)rr$r%rrrmr$r%rn) r1r2rInt64Factorizerrr0r3r-r4r9to_array)r6datarrresultrr;s r<test_factorizer_with_maskz'TestFactorize.test_factorizer_with_masksxx*':xxAB""3t9-D188/rww? ##FH588IW= ##EMM$:$:$<>NOrfctjddddtjg}tjt |}|j |jt}tjgdtj}tj||tjgdt}tj|jj|y)Nr$r%rn)rr$r%rrmr*r)r1r2rarrrr0rIr5r3r-r4r9r)r6rrrrr;s r<test_factorizer_object_with_nanz-TestFactorize.test_factorizer_object_with_nansxxAq!RVV,-##CI.V!4588,BGG< ##FH588IV< ##EMM$:$:$<>NOrfz&data, expected_codes, expected_uniques)r$r$r$r%rrrnonsense)rr$r%r$rn)rrrr)rrrrr)rrrr)rrrr)rr$r%r$)rrrcld}tjt|5tj|\}}dddtj t j|t jtj|t}tj |y#1swYtxYw)Nr'r(r*) r-r.r/rr0r4r1r2r3comasarray_tuplesafer5)r6rrYr;r7rdr9expected_uniques_arrays r<test_factorize_tuple_listz'TestFactorize.test_factorize_tuple_lists$A  ' ' S A 0\\$/NE7 0 ##E288N"''+RS!$!6!67Gv!V ##G-CD  0 0s B**B3c"tjtdDcgc] }t|c}t}d}t j t|5tj|dddddddycc}w#1swYyxYw)Nr*z,'[<>]' not supported between instances of .*r(rmTrB) r1r2rangecomplexr5pytestraises TypeErrorrr0)r6ix17r7s r<test_complex_sortingz"TestFactorize.test_complex_sortingsmhhE"I6q 6fE< ]]9C 0 2 OOC"ID 1 2 27 2 2sBBBcB|}tjgd|}tjgdtj}tjddg|}tj|\}}t j ||t j ||y)N)r$r%r%r$r*)rr$r$rr$r%)r1r2r3rr0r-r4)r6any_real_numpy_dtyper+rrYr;rdr9s r<test_numeric_dtype_factorizez*TestFactorize.test_numeric_dtype_factorizess$xx E2,bgg>88QF%8.w ##E>: ##G-=>rfctjgdtj}|j|tjgdtj}tjgdtj}t j |\}}tj||tj||y)N)r_חAr_:0yE>rr_r*write)rr$rr%r$r)r_rr) r1r2float64setflagsr3rr0r-r4r6writablerrYr;rdr9s r<test_float64_factorizez$TestFactorize.test_float64_factorize)sxx7rzzJ H %"4BGGD88$4BJJG.w ##E>: ##G-=>rfctjgdtj}|j|tjgdtj}tjddgtj}t j |\}}tj||tj||y)N)r$rr*rrr$rrr$) r1r2uint64rr3rr0r-r4rs r<test_uint64_factorizez#TestFactorize.test_uint64_factorize3sxx1C H %)277;88YN"))D.w ##E>: ##G-=>rfctjgdtj}|j|tjgdtj}tjddgtj}t j |\}}tj||tj||y)N)rr*rrrr) r1r2rrr3rr0r-r4rs r<test_int64_factorizez"TestFactorize.test_int64_factorize=sxx8I H %)277;88Y$9J.w ##E>: ##G-=>rfcrtjgdt}|j|tjgdtj}tjgdt}t j |\}}tj||tj||y)N)rhrjrhrirjr*r)rr$rr%r$rhrjri) r1r2r5rr3rr0r-r4rs r<test_string_factorizez#TestFactorize.test_string_factorizeGsyxx1@ H %/A88O6B.w ##E>: ##G-=>rfc tjdddtjddtdgt}|j |tjgdtj }tjgdt}tj|\}}tj||tj||y)Nrhrjrir*r)rr$rmrmrr%rmr$r) r1r2rarr5rr3rr0r-r4rs r<test_object_factorizez#TestFactorize.test_object_factorizeQsxxc4c3DFS H %"=RWWM88O6B.w ##E>: ##G-=>rfctjtjdgd}|j|tjdgtj}tjdgd}t j |\}}tj||tj||y)Nz2020-01-01T00:00:00.000M8[ns]r*rrz2020-01-01T00:00:00.000000000zdatetime64[ns]) r1r2 datetime64rr3rr0r-r4rs r<test_datetime64_factorizez'TestFactorize.test_datetime64_factorize[sxx'@AB(S H %1#RWW588 , -5E d+w ##E>: ##G-=>rfctjjtd}t j dtj |f}tj||}tj|d|dtj|d|dd|j|}tj|d|dtj|d|ddy)N r*rBrr$TrC r RangeIndex from_rangerr1rtr3rr0r-r4rP)r6r@rirrs r<test_factorize_rangeindexz'TestFactorize.test_factorize_rangeindexhs]] % %eBi 099Rrww/3$/ ##F1Ix{; fQi!DA4( ##F1Ix{; fQi!DArfc*tjjtd}t j dtj |f}|ddd}|d|f}|r|dddd|ddddf}tj||}tj|d|dtj|d|dd|j|}tj|d|dtj|d|ddy) Nrr*rmrr$rBTrCr)r6r@rrri2rs r<$test_factorize_rangeindex_decreasingz2TestFactorize.test_factorize_rangeindex_decreasingvs]] % %eBi 099Rrww/32hA;# {4R4((1+dd*;;H40 ##F1Ix{; fQi!DAD) ##F1Ix{; fQi!DArfc^tjgdtj}tjt d5t j|ddddtjd5t j|dddy#1swY=xYw#1swYyxYw)N)r$rr*zgot an unexpected keywordr(T)orderF) r1r2rrrrrr0r-r.)r6rs r<test_deprecate_orderz"TestFactorize.test_deprecate_ordersxx); ]]9,G H . OOD - .  ' ' . " OOD ! " " . . " "sB8B#B #B,rru8r*)rr$ri8)__nan__foorr5ctj|\}}|ddg}tjgdtj}t j ||t j ||y)Nrr$rr*)rr0r1r2r3r-r4)r6rrdr9r;rYs r<,test_parametrized_factorize_na_value_defaultz:TestFactorize.test_parametrized_factorize_na_value_defaultsY.wA<)277; ##E>: ##G-=>rfzdata, na_valuer^r)r$rr$r%r$)rr$rrr)r$rr$r)rhrhrirh)rhr$rrhr%r)rrrrrctj||\}}|ddg}tjgdtj}t j ||t j ||y)N)na_valuer$rn)rmrrmr$r*)rfactorize_arrayr1r2r3r-r4)r6rrrdr9r;rYs r<$test_parametrized_factorize_na_valuez2TestFactorize.test_parametrized_factorize_na_values]..thGwA<.@ ##E>: ##G-=>rfz data, uniques)rirhNririr%Int64 numpy_arrayextension_array)rctj||d\}}|r=tjgdtj}tj |}n)tjgdtj}|}t j||t|tjrt j||yt j||y)NT)r@r])r$rrmr$r*)rr$rmr) rr0r1r2r3 safe_sortr-r4rEndarrayassert_extension_array_equal)r6r@rr9rdrYr;s r<test_factorize_use_na_sentinelz,TestFactorize.test_factorize_use_na_sentinels D$Ow XXm277CN$w7 XXm277CN&  ##E>: dBJJ '  ' '1A B  + +G5E Frf)rhNrirhrr$r%rr3ctjtj|td\}}t j ||dt j ||dy)Nr*Fr\T strict_nan)rr0r1r2r5r-r4r6rrYr;rdr9s r<+test_object_factorize_use_na_sentinel_falsez9TestFactorize.test_object_factorize_use_na_sentinel_falsesM$ HHT (% w ##G-=$O ##E>dKrf)r$Nr$r%rcd}tjt|5tj|d\}}dddtj |dtj |dy#1swY:xYw)Nr'r(Fr\Tr)r-r.r/rr0r4)r6rrYr;r7rdr9s r<(test_int_factorize_use_na_sentinel_falsez6TestFactorize.test_int_factorize_use_na_sentinel_falsesn$A  ' ' S A J"__T5INE7 J ##G-=$O ##E>dK  J Js A22A;)rhrhri)rrr$category) categoriesr+2017 US/Easterntzctj|\}}tj||tj||y)N)rr0r-r4rPrs r<test_factorize_mixed_valuesz)TestFactorize.test_factorize_mixed_valuess7,.w ##E>: g'78rfctdtjdgj|}tdtjdgj|}t j ||}|j \}}|jd|dk(sJtdj|}t j ||z ||z }|j \} } | jd|dk(sJt j |jd |jd } | j \} } | jd|d k(sJy) N 2016-01-01z 2015-10-11z 2016-01-02z 2015-10-15zinterval[datetime64[z ], right]rzinterval[timedelta64[ US/Pacificz, US/Pacific], right]) rr1raas_unitr from_arraysr0r+r tz_localize)r6unitleftrightidxrdcatstsidx2codes2cats2idx3codes3cats3s r< test_factorize_interval_non_nanoz.TestFactorize.test_factorize_interval_non_nanos3lBFFLABJJ4P|RVV\BCKKDQ''e4mmo tzz3D6CCCC q\ ! !$ '((EBJ?( {{ 5dV9EEEE((   \ *E,=,=l,K ( {{ 4TF:OPPPPrfN)3__name__ __module__ __qualname__r=rmarkxfailr parametrizerZrerxr~rrrrrrrrrrrrrrrrrrr1r2rr5rrrarr+rrrrr r3rrrrr"rrfr<r"r"5s ? [[)+4HQVW [[VdE]3L4XL< 9%2N, ,$>" ? 7 PP [[0=4  <3  .|=U V  E! E2 ?????? ? [[VdE]3 B4 B [[VdE]3B4B$" [[ BHHYd + BHH,D 9 BHH2( C ?? [[ RXXl$ / 3 RXXl$ / 3 RXX0 =x H RXX) 6 : RXX) 8# > RXX2& A2 F RXX8 G R  ? ? [[VdE]3 [[.f=#s62  !Q*':!Qw/  - .  G 4 G [[0&XRXXf-=>#rvvs+6:  bffc3'XRXXf-=>#rvvs+6:  LL [[0 XRXXf-=>!RVVQs3  BFFAq!XRXXf-=>!RVVQrzz:  LL [[0k/23"''2 #sc *U  {?34"''2 #sc *U  }ff%5,GH!Qrww/vh<8  (9)(9 Qrfr"c eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZej(j+ddej.ddgefdej.dgefgdZdZdZdZej(j+dej:egdZdZdZ y )! TestUniquectjjdjddd}t j |}t |tjsJy)Nr%rd2size)r1random default_rngintegersrrGrErr6rwrs r< test_intszTestUnique.test_ints3sJii##A&//3R/@c"&"**---rfctjjdjdddj d}t j |}t|tjsJy)Nr%rr,r-r.r) r1r0r1r2rIrrGrErr3s r< test_objectszTestUnique.test_objects9sUii##A&//3R/@GGLc"&"**---rfctjgdt}tdD] }t t j |"y)N)rzr{CDEr*)r1r2r5rrrrG)r6lstrs r<test_object_refcount_bugz#TestUnique.test_object_refcount_bug?s7hh0?t #A  S! " #rfctjtjdj dtj tjddg}|j }|j|j d}tj|}|jtj||y)Nrlr%) rrr1rtrepeattilerbr@rrGr-assert_almost_equal)r6mindexrrs r<test_on_index_objectzTestUnique.test_on_index_objectDs'' YYq\  #RWWRYYq\1%= > == q!6"  vx0rfc|tjtjzvr gd}ddg}nt|r gd}ddg}nt |r gd}ddg}nwt |rE33E<cttdtdgjj|}t j |}t jdgd|d}tj||y)Nrvz2016-01-01T00:00:00.000000000zM8[]r*) rrdtrrrGr1r2r-r4)r6rrcrrs r<test_order_of_appearance_dt64z(TestUnique.test_order_of_appearance_dt64sgi +Yz-BCDGGOOPTU388<=s4&PQ]S ##FH5rfcttddtddgj|}tj|}tdgd|dd}t j ||y)Nrvr r z2016-01-01 00:00:00z datetime64[z , US/Eastern])r+r)rrrrrGr-rP)r6rrxrrs r<test_order_of_appearance_dt64tzz*TestUnique.test_order_of_appearance_dt64tzsr*6*6  '$- 3 " #[m+LSW  fh/rfz arg ,expected)1r2rrr*)rrcd}tjt|5tj|}dddtj |y#1swY xYwN-unique with argument that is not not a Seriesr(r-r.r/rrGr4)r6argrr7rs r<test_tuple_with_stringsz"TestUnique.test_tuple_with_strings+sK>  ' ' S A $YYs^F $ ##FH5 $ $ AActjddgt}tj|}tjddgt}t j ||dy)Nrr*Tr)r1r2r5rrGr-r4r6rwrrs r<test_obj_none_preservationz%TestUnique.test_obj_none_preservation9sIhht}F3388UDM8 ##FHFrfctjddg}tj|}tjdg}t j ||y)NgrsrVrWs r<test_signed_zerozTestUnique.test_signed_zeroAsA HHdC[ !188TF# ##FH5rfctjdtjddd}tjdtjddd}||k7sJ||k7sJtj||g}t j |}tjtjg}tj||y)Nd=Qr) structunpackpackr1r2rrGrar-r4)r6NAN1NAN2rhrrs r<test_different_nanszTestUnique.test_different_nansHs}}S&++d4F"GHK}}S&++d4F"GHKt||t|| HHdD\ "188RVVH% ##FH5rfel_typecd}d}tjdtjd|d}tjdtjd|d}||k7sJ||k7sJtj||g|}t j |}|jdk(sJtjdtjd|dd}||k(sJy)Nlrrrrr*r$)rrrr1r2rrGr/) r6r bits_for_nan1 bits_for_nan2rrrhrresult_nan_bitss r<test_first_nan_keptzTestUnique.test_first_nan_keptTs+ * }}S&++dM"BCAF}}S&++dM"BCAFt||t|| HHdD\ 11{{a --fkk#vay.IJ1M-///rfc||urytj||gt}tj|}|j dk(sJ|d|usJ|d|usJy)Nr*r%rr$)r1r2r5rrGr/)r6unique_nulls_fixtureunique_nulls_fixture2rhrs r<test_do_not_mangle_na_valuesz'TestUnique.test_do_not_mangle_na_valuesesj #8 8  HH*,AB& Q1{{at++++t,,,,rfctdtjdgdz|}tj|}tjdtjdg|}t j ||y)Nr$r%rnr*)rrNArGr2r-r)r6any_numeric_ea_dtypercrrs r<test_unique_maskedzTestUnique.test_unique_maskedosVa]Q&.BC388QqM1EF ''9rfN)!r#r$r%r4r6r=rCrKrQrXrZrarcrfrtryr|rrrr&r(r1r2r5rrrrrrrrrrfr<r*r*2s. . # 1#6J.>6 6 .&: 6 '0R066(6 0 [[ hbhhSz@ A xrxxv6 7 66G6 6 [[YV(<=0>0 -:rfr*c|tjjdjddd}t j |}t t j|}||k(sJy)Nr%rr)r1r0r1r2r nunique_intsrrG)index_or_series_or_arrayrbrrs r<test_nunique_intsrws[ %bii&;&;A&>&G&G2r&R SF    'F5<<'(H X  rfceZdZdZdZdZejjdgdejjdgddZ d Z ejjdgdd Z ejjdgd d Z d Z dZdZdZdZdZdZdZdZejjdgeeej2ggdZdZdZdZdZdZdZ y) TestIsincd}tjt|5tjdddddtjt|5tjddgdddtjt|5tjdgddddy#1swYxYw#1swYQxYw#1swYyxYw)NzOonly list-like objects are allowed to be passed to isin\(\), you passed a `int`r(r$)rrrrisin)r6r7s r< test_invalidzTestIsin.test_invalids " ]]9C 0  JJq!   ]]9C 0  JJq1#   ]]9C 0  JJsA          s#B4CC 4B=C  Ccd}tjt|5tjddgdg}dddt j ddg}tj|tjt j ddgdg}t j ddg}tj||tjtddgdg}t j ddg}tj||tjtddgtdg}t j ddg}tj||tjtddgdh}t j ddg}tj||tjt|5tjddgdg}dddt j ddg}tj||tjtddgtdg}t j ddg}tj||tjtddgdh}t j ddg}tj||tjt|5tjddgdg}dddt j ddg}tj||y#1swYxYw#1swYVxYw#1swYQxYw) N+isin with argument that is not not a Seriesr(r$r%TFrhri) r-r.r/rrr1r2r4r)r6r7rrs r<rxzTestIsin.test_basics;  ' ' S A -ZZA,F -88T5M* ##FH5BHHaV,qc288T5M* ##FH5FAq6NQC088T5M* ##FH5FAq6NFA3K888T5M* ##FH5FAq6NQC088T5M* ##FH5  ' ' S A 3ZZc SE2F 388T5M* ##FH5FC:.u >88T5M* ##FH5FC:.688T5M* ##FH5  ' ' S A 1ZZc QC0F 188UEN+ ##FH5K - -* 3 3 1 1s#L8M(M8MMMctddj}tj||dg}t j gd}t j||tj||dd}t j gd}t j||tj|t|dd}t j gd}t j||tddj}tj||dg}t j gd}t j||tj||dd}t j gd}t j||tj|t|dd}t j gd}t j||y) NrrnperiodsrTFFr%rEr) rrbrrr1r2r-r4rrrs r<test_i8zTestIsin.test_i8slQ/66C#a&*8801 ##FH5CQq*88/0 ##FH5CS1X/88/0 ##FH5gq188C#a&*8801 ##FH5CQq*88/0 ##FH5CS1X/88/0 ##FH5rfdtype1)r_r M8[ns, UTC]z period[D]r+)rf8rcztddj}t|jdjj|}|jdj |}t j||}tj|jt}tj||y)N 2013-01-01rnrrr*)r_valuesrviewr2rIrrr1zerosshaperKr-r4)r6r+rdtarwcompsrrs r<+test_isin_datetimelike_values_numeric_compsz4TestIsin.test_isin_datetimelike_values_numeric_compss q199SXXd^$**//7%%e,E3'88EKKt4 ##FH5rfctdddj}tj||dd}t j t |t}d|d<d|d <tj||y) N20000101irP)rrrr%r*Tr$) rrbrrr1rrrKr-r4r6rPrrs r< test_largezTestIsin.test_largesb z7 = D DAq1v&88CF$/   ##FH5rfcFtddj}t|jdjj|}t |d<t j|t g}tjgdt}tj||y)Nrrnrrrrr*) rrrrr2rrrr1rKr-r4)r6r+rrwrrs r<test_isin_datetimelike_all_natz'TestIsin.test_isin_datetimelike_all_natsuq199SXXd^$**//6AC#'880= ##FH5rf)r_rrcltddj}t|jdjj|}|Dcgc] }t |}}d}t jt|5tj||}dddjsJtj|t }t jt|5tj||} ddd jsJycc}w#1swYxYw#1swY-xYw)Nrrnrrz2The behavior of 'isin' with dtype=.* is deprecatedr(r*) rrrrr2strr-r.r/rrallr1) r6r+rrwr}valsr7resvals2res2s r<)test_isin_datetimelike_strings_deprecatedz2TestIsin.test_isin_datetimelike_strings_deprecatedsq199SXXd^$**//6 #$1A$$B  ' ' S A (**S$'C (wwyyS)  ' ' S A *::c5)D *xxzz% ( (  * *sDD'D*D'*D3ctddd}t|}t|d<tj|j tg}t jgdt}tj||y)NrrnUTCrr rrr*) rrrrrrr1r2rKr-r4)r6rxrcrrvs r<test_isin_dt64tz_with_natz"TestIsin.test_isin_dt64tz_with_nats[qU;SkAjjse,hh+48 ##C-rfc~tjgd}gd}ttdgj ||}ttdgj tjddg|}tjgd}t j ||}tj||yNrrkr$r)TTFT) r1r2rr from_codesrrr-r4)r6rrSdStrrs r<test_categorical_from_codesz$TestIsin.test_categorical_from_codes sxx % K$//d; < K$//!Q0@$G H8856B# ##Hf5rfcZtjgd}gd}tdgj||}tdgjtjddg|}tjgd}t j ||}t j||yr)r1r2r rrrr-r4)r6rrcatotherrrs r<test_categorical_isinzTestIsin.test_categorical_isinsxx %1#))$5QC ++BHHaV,.LikeNanreturncy)NFr)r6rs r<__eq__z7TestIsin.test_same_object_is_in..LikeNan.__eq__Gsrfcy)Nrr)r6s r<__hash__z9TestIsin.test_same_object_is_in..LikeNan.__hash__JsrfN)r#r$r%rKrrrrfr<LikeNanrFs t  rfrrr(TF)r-r.r/r4rrr1r2)r6rrhrir7s r<test_same_object_is_inzTestIsin.test_same_object_is_in@s   y')1;  ' ' S A Q  ' ' A3(N O  ' ' A3(O P  Q Q Qs BCC ctdg}tdg}|d|dusJtjtj||}t j tjdg|tjtj|ttj|t}t j tjdg|tjtj|tjtj|tj}t j tjdg|y)NrarTr*) floatrrr1r2r-r4rOr5r)r6rrbrs r<rzTestIsin.test_different_nansVsu,Qxvay(((BHHUOV4 ##BHHdV$4f= JJuF +RZZf-M  ##BHHdV$4f= JJuBJJ /F"**1U  ##BHHdV$4f=rfcddg}dg}tjddg}d}tjt|5t j ||}dddtj|y#1swY xYw)Nss*42Fz2isin with argument that is not not a Series, Indexr()r1r2r-r.r/rrr4rs r< test_no_castzTestIsin.test_no_castnspr 88UEN+B  ' ' S A /ZZv.F / ##Hf5 / /s A22A;emptyr*ctddg}tjddg}tj||}t j ||y)NrhriF)rr1r2rrr-r4)r6rrrrs r< test_emptyzTestIsin.test_emptyysDc3Z 88UEN+D%( ##Hf5rfcDtjdtjdztdgt}tjtdgt}tjgd}t j ||}tj||y)Nra?r*)FFT) r1r2rarr5rrr-r4)r6rrrrs r<test_different_nan_objectsz#TestIsin.test_different_nan_objectssj%"eEl;6Jxxuf58801E4( ##Hf5rfctjdtjddd}tjdtjddd}||k7sJ||k7sJtj||gtj }tj|gtj }t j||}tjddg}tj||tj|gtj }t j||}tjddg}tj||y)Nrrrrrr*T) rrrr1r2rrrr-r4)r6rrrwlookup1rrlookup2s r<test_different_nans_as_float64z'TestIsin.test_different_nans_as_float64s }}S&++d4F"GHK}}S&++d4F"GHKt||t||hhd|2::6((D64C)88T4L) ##FH5((D64C)88T4L) ##FH5rfctdddgi}|jdg}tdddgi}tj||y)zComparing df with int`s (1,2) with a string at isin() ("1") -> should not match values because int 1 is not equal str 1rbr$r%rFN)rrr-assert_frame_equalr6dfrexpected_falses r<test_isin_int_df_string_searchz'TestIsin.test_isin_int_df_string_searchsI1a&) *#"Huen#=> fn5rfctdtjdgi}|jtjdgt }tdddgi}t j||y)zComparing df with nan value (np.nan,2) with a string at isin() ("NaN") -> should not match values because np.nan is not equal str NaNrbr%NaNr*FN)rr1rarr2r5r-r r s r<test_isin_nan_df_string_searchz'TestIsin.test_isin_nan_df_string_searchsY2661+. /5'89"Huen#=> fn5rfctdddgi}|jtjdgt}tdddgi}t j ||y)zComparing df with floats (1.4245,2.32441) with a string at isin() ("1.4245") -> should not match values because float 1.4245 is not equal str 1.4245rbgn?g#Ed@z1.4245r*FN)rrr1r2r5r-r r s r< test_isin_float_df_string_searchz)TestIsin.test_isin_float_df_string_searchsV67"34 58*F;<"Huen#=> fn5rfctdgtj}|jdg}td}t j ||y)Nl Gr*l GF)rr1rrr-rr6rcrrs r<test_isin_unsigned_dtypez!TestIsin.test_isin_unsigned_dtypesA)*"))<./0%= vx0rfN)!r#r$r%rrxrrr&r(rrrrrrrrrrrrrrr5r1r2rrrrrrrrrfr<rrs. '6R66 [[X'WX [[W&89 6:Y 66 [[W&VW6X6 [[W&IJK .66 661Q,>0 6 [[Wr6+?"&NO6P666,6661rfrceZdZdZdZdZdZejjde dgdZ dZ d Z d Zd Zejjdej"e d fd ZdZdZy)TestValueCountsctjjdjd}t |d}d}t j t|5tj|}dddgd}tj|jtd}tgd|d }t jj!|j!y#1swY~xYw) Niro!pandas.value_counts is deprecatedr()gV-g}?5^Ig/$ۿgףp= ?gʡE?Trm)r$rr%r$countrNro)r1r0r1standard_normalrr-r.r/r value_countsr from_breaksrIr rr sort_index)r6rwfactorr7rbreaksrNrs r<test_value_countsz!TestValueCounts.test_value_countssii##D)99!<S!2  ' ' S A 0''/F 07))&1889IRV9WX,e'B v002H4G4G4IJ  0 0s C++C4cgd}d}tjt|5tj|d}dddt dgt jdgd }tj|tjt|5tj|d d }dddt d d gt jd dgd }tj||y#1swYxYw#1swYQxYw)N)r$r%rnrorr(r$binsro)Zd;?rqrrr%F)r&r@)r'@)r(rq) r-r.r/rrrrrFr)r6rPr7rrs r<test_value_counts_binsz&TestValueCounts.test_value_counts_binss 1  ' ' S A 3''2F 3 C}00,@w  vx0  ' ' S A ?''>F ? F++\:,FG  vx0 3 3 ? ?sC3C?3C<?Dcd}tjt|5tjt j ddg}dddtdk(sJtjt|5tjt j ddgd}dddt|dk(sJtjt|5tjtgd}dddt|dk(sJd}tjt|5tjt|5tjt j d dgt dddddddy#1swYKxYw#1swYxYw#1swYxYw#1swY7xYw#1swYyxYw) Nrr(r$r_r%)r$r_rr%z*bins argument only works with numeric datarr*) r-r.r/rrr1r2rrrrrr5)r6msg2rr7s r<test_value_counts_dtypesz(TestValueCounts.test_value_counts_dtypess{2  ' ' T B <''!S(:;F <6{a  ' ' T B D''!S(:CF D6{a  ' ' T B ?''}(=>F ?6{a: ]]9C 0 M++MF M""288S!HF#C!L M M M < < D D ? ?  M M M MsG+F<-F!F+2G3F7GFF(+F47G <GG c\ttjdtgd}t ddg}d}||fD]r}t j t|5tj|}tj|d }dddtd k(sJtd k(rrJttd d id }t j t|5tj|}dddt j|td gtjdgd } t j t|5tj|} dddt j | y#1swYxYw#1swYxYw#1swY9xYw)Nr^timedelta64[ns]r*rz 2014-01-01rr(Fdropnar$r%z2014-01-01 00:00:00rrnr) rr1 timedelta64rrr-r.r/rrrrr) r6tdrr7rcvc vc_with_naexp_dt result_dtexp_td result_tds r<test_value_counts_natz%TestValueCounts.test_value_counts_natsr R^^E*C08I J %. /18 (C++ME C'',"//EB  Cr7a< <z?a' ''  (#891=GL  ' ' S A /**2.I / y&1BNN5$9#:I  ' ' S A /**2.I / y&1 C C / /  / /s$-F "FF" F F"F+r+zM8[us]c |ttdddtdddtdddtdddtdddtdddg|}|j}ttdddtdddtdddg|}tgd|d}t j ||y) Ni r$iipr*rnr%r$rr)rrrrr-r)r6r+rcr exp_indexrvs r<&test_value_counts_datetime_outofboundsz6TestValueCounts.test_value_counts_datetime_outofbounds sq!$q!$q!$q!$q!$q!$      dAq !8D!Q#7$19M N Yig> sC(rfctttd}|j}tgdt gdd}t j ||d|jj}|j}|jj|_ t j ||dy)Naaabbcr;rkrrTcheck_index_type) rr rMrrr-rr as_orderedrNrs r<rtz TestValueCounts.test_categorical"s ;tH~. /! -o>W  vx$G EE   !!224 vx$Grfc rtttd}tj|j d<|j }tgdtgdgdd}tj||d |j d }tgd td ddtjgd}tj||d tttddgd}tj|j d<|j }tgdtgdgddd}tj||d |j d }tgd td ddtjggddd}tj||d y)N aaaaabbbccr$)rornr%rkrprrTr@Fr/)rornr%r$rhrirj)rirhrj)rkr rj) rr rMr1railocrrr-rrs r<test_categorical_nansz%TestValueCounts.test_categorical_nans1so ;tL12 3FFq ! "?O  vx$Gu-  0#sC1H IPW  vx$G  \*D_ U FFq ! "*   vx$Gu- "c3'OT   vx$Grfc tttdtdd}|j}tgdtgdtddd}t j ||d y) NbbbaacabcdTrj)rnr%r$r)rirhrjrrrr@)rr rMrr-rrs r<test_categorical_zeroesz'TestValueCounts.test_categorical_zeroes\sc ;tH~$v,PTU V! $ft   vx$Grfc ztjtgdjdtddgddgdtjtgdjdtddgddgdtjtdgd zdgdzzdgd zzjdtd dgt ddgt dtjtdgd zdgd zzdgdzzjdtgd gd dtjtgdjdtddgddgdtjtgdjdtddgddgdtjtgdjdtddgddgdtgdjd}tgdgdd}tj||y)NrETr/r%r$Frrrnrlr*)rlrnr%)TFN)皙$@rprprprL)rLrprpN)rLrLrprprpNr;)rprLN)r-rrrrr5)r6rrs r<test_value_counts_dropnaz(TestValueCounts.test_value_counts_dropnais  & ' 4 4D 4 A Aq6$W =   & ' 4 4E 4 B Aq6$W =  D6A:! +tfqj8 9 F Fd F S Aq6e}F!C' R   D6A:! +tfqj8 9 F Fe F T 9$7g F   # $ 1 1 1 > Aq6#t7 ;   # $ 1 1 1 ? Aq6#t7 ;  ) * 7 7t 7 D Aq6#t7 ; 9:GGuGU)+<7K vx0rfrctdgdzdgdzztjgdzz}|j|}|j dd}tgdttjd d g| d }t j |||j dd}tddgtd d g| d }t j ||y)Nr$r%rnrlTF) normalizer0)g?g333333?g?r`r_r* proportionrg333333?g?)rr1rarIrr-r)r6r+rPs_typedrrs r<test_value_counts_normalizedz,TestValueCounts.test_value_counts_normalizeds A37aS1W$x!|3 4((5/%%U%C "&&#s+59  vx0%%T%B #Jfc3Zu=L  vx0rfc>tjdgtj}tdgdgd}d}t j t |5tj|}dddt j|tjddgt}tddgddgd}t j t |5tj|}dddt j||y#1swYxYw#1swY,xYw) Nrr*r$rrrr(rm) r1r2rrr-r.r/rrrr5)r6rwrr7rs r<test_value_counts_uint64z(TestValueCounts.test_value_counts_uint64shhwbii01#eW7;1  ' ' S A -'',F - vx0hhE{&11a&U 'B  ' ' S A -'',F - vx0 - - - -sDDDDctjdddddtjg}t|j d}tgdt j gdd d }tj||y) Nrnr$r%ror%)r%r%r$))r'r`)r`rr)rrrqzinterval[float64, right]r*rr) r1r2rarrrrFr-r)r6rbrrs r<test_value_counts_seriesz(TestValueCounts.test_value_counts_seriessp1aAq"&&12,,!,4 ++6>X   vx0rfN)r#r$r%r#r)r,r9rr&r(r5r=rtrFrJrMr1rrRrTrVrrfr<rrs K1&M&2. [[Wvx&89):), H)HV H$1L [[Wrzz68&DE1F1$1 1rfrceZdZdZej j dejgdejdddejddddejdg ejgd ejgd e ejgd ej gd Z dZ ej j degdej ddgdZej j dgdgdfgdddgfgdgdfgdZej j dgdejgde fgdZy )!TestDuplicatedc Rtjddtjddtjgt}t j |}tjgd}t j||t j |d}tjgd}t j||t j |d}tjgd }t j||t j |d }tjgd }t j||tjd t}ttddtjtjgdzdtjdtjgdzD] \}}|||< t j |}d gd z}dgd z}tj||z}t j||t j |d}tj||z}t j||t j |d }tj||z}t j||y)Nrr$r%r*)FFFTFTfirstkeeplast)TFTFFFF)TFTTFTroT) r1r2rar5r duplicatedr-r4r enumeratezip)r6keysrrrtfalsestruess r<test_duplicated_with_nasz'TestDuplicated.test_duplicated_with_nassxxArvvq!RVV4FC!!$'88DE ##FH5!!$W588DE ##FH5!!$V488DE ##FH5!!$U388BC ##FH5xx( Arvvrvv&*Q266,BQ,F G  DAqDG  !!$'1 88FUN+ ##FH5!!$V488EFN+ ##FH5!!$U388EEM* ##FH5rfcase) r$r%r$rlrnr%ror$rlg?g@gffffff @g@gffffff@) ??@@ri@@y@@rjy@@rirky@@) rhrirherjrirrhrlfr*) r$rr$rr'r$rnctjgd}tjgd}||z}tj|d}t j ||tj|d}t j ||tj|d}t j ||t |t |dfD]z}|jd}t j |||jd}t j |||jd}t j |||t|t|dfD]} | jd}t j|t|| jd}t j|t|| jd}t j|t|y) N FFTFFTFTTF TTTTFFFFFFrZr[r]Frr*) r1r2rr_r-r4rrr) r6rg exp_firstexp_last exp_false res_firstres_last res_falserrPs r<test_numeric_object_likesz(TestDuplicated.test_numeric_object_likess4HH N 88 N ( $$T8  ##Iy9##Dv6 ##Hh7$$T6  ##Iy9$Kt:!>? >CG4I  ' ' 9 =~~6~2H  ' '( ;E2I  ' ' 9 = >,t: >? AA ' 2I  " "9fY.? @|||0H  " "8VH-= > % 0I  " "9fY.? @ Arfc dgd}gd}tj|Dcgc] }t|c}tj|Dcgc]}t|dc}tj|Dcgc]}t|dc}tj|Dcgc]}tj|c}tj|Dcgc] }t |c}g}tjgd}tjgd}||z}|D]}t j|d } tj| |t j|d } tj| |t j|d } tj| |t|t|d t|tfD]z} | jd } tj| || jd } tj| || jd } tj| ||t|t|d t|tfD]} | jd } tj| t|| jd } tj| t|| jd } tj| t|ycc}wcc}wcc}wcc}wcc}w)N) 2011-01-01 2011-01-02r|r 2011-01-03r}z 2011-01-04r|rz 2011-01-06) 1 days2 daysrrz3 daysrz4 daysrrz6 daysr r r9rrrrsrZr[r]Frr*)r1r2rrrrrr_r-r4rr5rr)r6rr2rcasesrtrurvrgrwrxryrrPs r<test_datetime_likesz"TestDuplicated.test_datetime_likes)s     HHB/qil/ 0 HHR@il3@ A HH26afQS)6 7 HH31bmmA&3 4 HHB/qil/ 0  HH N 88 N ( & ED((G>v>6++Hh?NNN6 ++IyA B t t:.t6* E LLgL6 &&y&2CD<  ' ' S A &YYu%F & ##FH5 & &rN)r#r$r%rfrr&r(r1r2rar5rrzrrrrrrrrrfr<rXrXs%6N [[ BHH3 4 BHHc3RVVS#sCM N BHH   BHHGv V BHH>bii % 2&A32&APPEd [[VeI&6  a8K%LMXNX [[Q0  AZ( 67U V  6 6 [[3?vN   6 6rfrXc peZdZejj dejeje dDcgc]}d| c}}e fejeje dDcgc]}d| c}}e fejejdejfej ejdej"fej$ejdej&fgdZejj dejeje dDcgc]}d| c}}e fejeje dDcgc]}d| c}}e fejejdejfej ejdej"fej$ejdej&fgdZycc}}wcc}}wcc}}wcc}}w) TestHashTablez htable, datar;foo_r*ct||j}|tjk(rtj |j d<n7|tjk(r$tj dtg|j dd|jddjd}|jj||jd j}|j|j}tj |||j|jd \}}tj ||||} tj | |jy) Nr*rnTfracreplacedroprrZr[)return_inverse)rr+rFloat64HashTabler1ralocPyObjectHashTablersample reset_indexrbrdrop_duplicatesrGr-r4) r6htablerrrP s_duplicatedexpected_unique result_uniqueresult_inversereconstrs r<test_hashtable_uniquez#TestHashTable.test_hashtable_uniques.$ 4tzz * R(( (AEE#J r++ + ffdC0AEE#cNxxQx5AAtAL $$8$4'66G6DKK (;(;<  ##M?C)/   )8) % ~ ##M?C 0 ##Hl.A.ABrfct||j}|tjk(rtj |j d<n7|tjk(r$tj dtg|j dd|jddjd}|jj||jj}|j|j\}}|jj!j} t#j$|| |||} |jj} t#j$| | y) Nr*rrrnTrrr)rr+rrr1rarrrrrrbrrr0r0rr-r4) r6rrrrPrna_maskrrrresult_reconstructexpected_reconstructs r<test_hashtable_factorizez&TestHashTable.test_hashtable_factorizes/$ 4tzz * R(( (AEE#J r++ + ffdC0AEE#cNxxQx5AAtAL $$8$4##%,,(.(:(:<;N;N(O% ~ '--/??AHH ##M?C+>7(+CD+224;; ##$68LMrfN)r#r$r%rr&r(rrr1r2rr5StringHashTablerrtrInt64HashTablerUInt64HashTablerrr).0rs00r<rrs [[$$eDk:D*:&I  ""eDk:D*:&I  )"))D "C D     $bhh ? @   4ryy!A B  C! C: [[$$eDk:D*:&I  ""eDk:D*:&I  )"))D "C D     $bhh ? @   4ryy!A B  N! Nq;;R;;s H 5 H& H, H2rceZdZejj dejejdddejdddejg dejdddejdddejg gdZejj dejd d Z ejj dejejgd Z d Zejjd Zy)TestRankrwrpr$r%rnrqchtjd}tj|}tj|}|j }t j|}tj||<|j|}tj||<tj||y)Nz scipy.stats) r importorskipr1r2isfinitecopylibalgosrank_1dr|rankdatarar-rA)r6rwsp_statsrrrvs r<test_scipy_compatzTestRank.test_scipy_compats&&}5hhsm C  hhj!!#&FFD $FFD  vs+rfr+ AllIntegerctjddgtj}tjddg|}|j|t |}t j |}tj||y)Nr$r%r*r,r) r1r2rrrrrankr-r4)r6rr+rvrrcrs r<rxzTestRank.test_basicsbhh1vRZZ0xxC. H %TlC ##FC0rfctjddgtj}tddg|}t j t j||y)Nr$r%r*r)r1r2rrr-r4rr)r6r+rvrPs r<rczTestRank.test_uint64_overflow%sBhh1vRZZ0 Au:U + ##EJJqM37rfctjgdgdgdgg}d}tjt|5t j |dddy#1swYyxYw)Nr)rorlrh)rpr^ z%Array with ndim > 2 are not supportedr()r1r2rrrrr)r6rwr7s r<test_too_many_ndimszTestRank.test_too_many_ndims,sMhhIy9:;5 ]]9C 0  JJsO   s AA%c(tjd}tj|dj }|dk(sJtjdj dd}tj|dj }|dk(sJy)NiT)pctr$ir%)r1rtrrmaxreshape)r6rbrs r<test_pct_max_many_rowszTestRank.test_pct_max_many_rows3sw9%F-113{{9%--i;F-113{{rfN)r#r$r%rr&r(r1rar typecodesrxrrrcr single_cpurrrfr<rrs [[ VVRVVS#sBFFAq!RVV D "&&#sCAsBFF C  , , [[Wbll<&@A1B1 [[Wrzz299&=>8?8  [[rfrceZdZdZej j dejdejdzdZ dZ ej j dejdejdzdZ dZ ej j de egd Zd Zd Zd Zd ZdZdZdZy)TestModectgtjtgt}t j tjtjg|jy)Nr*)r+rN) rr1rrintr-r4rmoder2rbr6rvs r< test_no_modezTestMode.test_no_mode@s@Rrzzr1EF ##EJJrxx|$tjt j |j |j n=tjt j |j |j tj|j |y)Nbarrr%rnr*) rrr-rrrrbr4r)r6rusing_infer_stringrvrrcs r<test_strobj_multi_charzTestMode.test_strobj_multi_charysgw{eWq[(T$S# ")  + +EJJszz,BCJJ O  ' ' 3::(> K sxxz3/rfctgdd}tgdd}tjtj|j |j tj|j |tddgd}tgdd}tjtj|j |j tj|j |y)N) 1900-05-03r~ 2013-01-02rr*)r~rrr~r)r~rrr~rrr-rrrrbrrr6rvrcs r<test_datelike_modezTestMode.test_datelike_modes?xP?xP '' 3::(> L sxxz3/lL1B R  '' 3::(> L sxxz3/rfctgdd}tgdd}tjtj|j |j tj|j |tddgd}tgdd}tjtj|j |j tj|j |y)N)-1 days0 daysrr.r*)rrr2 minrrrz-1 dayz -1 day 2 minrrrrs r<test_timedelta_modezTestMode.test_timedelta_modes4 L sxxz3/gw'/@A J#  '' 3::(> L sxxz3/rfctdgt}tgd}tjt j |j |j tj|j |y)Nrr*)r$rr)rr5r-r4rrrbrrs r<test_mixed_dtypezTestMode.test_mixed_dtypesReWF+&' ##EJJszz$:CJJG sxxz3/rfcntdgtj}tgdtj}tjt j |j|jtj|j |tddgtj}tddgtj}tjt j |j|jtj|j |y)Nrr*)r$rrr$) rr1rr-r4rrrbrrs r<rczTestMode.test_uint64_overflowseWBII.&bii8 ##EJJszz$:CJJG sxxz3/aZryy1aZryy1 ##EJJszz$:CJJG sxxz3/rfctddg}|}t|jj}t j ||tgd}tdgddg}t|jj}t j ||tgd}tddggd}t|jj}t j ||y) Nr$r%r$rhrhrhrpr$r$r%rnrnrnr)r rrrr-rq)r6rjrvrs r<rtzTestMode.test_categoricals A Qinn&& ##C-  &3%QH5Qinn&& ##C-  (1a&Y7Qinn&& ##C-rfctgd}tgdtj}t j t j||jtgd}tdgt}t j t j||jtgd}tddgtj}t j t j||jtgdd }tjtd 5t j|dddy#1swYyxYw) Nrr*rrhrr$rnrr.TimedeltaIndexr() rrr1rr-r4rrrbr5rrAttributeError)r6rrvs r< test_indexzTestMode.test_indexsIYbhh/ ##EJJsOSZZ@M"cU&) ##EJJsOSZZ@O$aV288, ##EJJsOSZZ@ J# ]]>1A B  JJsO   s 7EEctgdd}|j}tdgd}tj||y)N)r$r$rnrrnr$)rrr-rrs r<test_ser_mode_with_namez TestMode.test_ser_mode_with_names7YU+1#E* vx0rfN)r#r$r%rrr&r(r1rrrrrrr5rrrrrcrtrrrrfr<rr?sJ [[T2<< #= W@U#UV0W0$H [[T2<< #= W@U#UV0W0"0 [[TC=1 02 0 0 00 0. *1rfrceZdZejj dddgdZdZejj dddgdZy ) TestDiffr+rr_ctjdjtjj |j dd}|j jdd|dddf<tj|dd }tj|jd dz}tjdd|dddf<tjdd|dddf<tj||tj|jdd }tj||jy) N rnrornsr%r$raxisr.r*)r1rtrIrrrr+typerdiffrrr1r-r4Tr6r+rwrrs r<test_diff_datetimelike_natz#TestDiff.test_diff_datetimelike_natsiim""288,11%8@@AFIINN5$/AqD C+77399,=>Bt4At4A ##FH5CEE11- ##FHJJ7rfctdddj}d}tjt|5t j |dddddy#1swYyxYw) Nrrnrrz#cannot diff DatetimeArray on axis=1r(r$r)rrwrr ValueErrorrr)r6rr7s r<test_diff_ea_axiszTestDiff.test_diff_ea_axissNq\BHH3 ]]:S 1 ' JJsAA & ' ' 's AA!int8int16ctjgd|}tj|d}tjtjddddgd}t j ||y)N)rr$r$rrr*r$rrmrJ)r1r2rrrar-r4rs r<test_diff_low_precision_intz$TestDiff.test_diff_low_precision_intsQhhe4C#88RVVQ2q1C ##FH5rfN) r#r$r%rr&r(r r rrrfr<rrsY [[Wx&:; 8< 8' [[Wvw&78696rfropc&|gd}|gd}|gd}t|tjr-tj||}t j ||ytj||}t j||y)N)rnr$rnro)r%rnr$r$)rnrnr$r$ror%)rEr1rrunion_with_duplicatesr-r4r)rlvalsrvalsrrs r<test_union_with_duplicatesrsu | E | E$%H(BJJ',,UE: ##FH5,,UE: ''9rf)Brrnumpyr1rpandas._configr pandas._libsrrrrpandas.core.dtypes.commonrrr r r pandas.core.dtypes.dtypesr pandasrr rrrrrrrrrrrrrrrrrpandas._testing_testingr-pandas.core.algorithmscore algorithmspandas.core.arraysrr pandas.core.commoncommonrr"r*rrrrXrrrrr&r(r2rrrfr<r%s  - 7(&&! zQzQzB:B:J ~1~1B }1}1@g6g6TYNYNx66r^1^1B66@"((34 :5 :rf