'Vfi-TddlmZddlZddlZddlmZddlmZddl m cm Z ej ejejdejfejejdejfejdejfejdejfejdejfejd ejfejd ejfejdejfejdejfejdejfejd ejfejd ejfejdejfejdejfejdejfejd ejfejd ejfejdejfejdejfejdejfejd ejfejd ejfg d ZGd dZGddZdS))datetimeN)iNaTg@y@@T)paramsc|jS)N)param)requests Ec:\PYTHON\DbComparer\venv\Lib\site-packages\pandas/tests/test_take.pydtype_fill_out_dtyper s 6 =cVeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S)TestTakec|\}}}tjdddd|}gd}t j|||}|gd|gdksJ|d|ksJ|j|ksJgd }t j|||}|gd ||ksJ|j|ksJdS) Nrrrr) fill_valuerrrrrrrrrrrrrr nprandom default_rngintegersastypealgostake_ndalldtypeselfrr&r out_dtypedataindexerresults r test_1d_fill_nonnazTestTake.test_1d_fill_nonna+s';$z9y$$Q''00Aq99@@GG--tWDDDyyy!T)))_499;;;;;ayJ&&&&|y((((,,tWDDD|||$W 5::<<<<<|u$$$$$$rc|\}}}tjdddd|}gd}t j||d|}|gdddf|gdddfksJ|dddf|ksJ|j|ksJt j||d |}|ddgdf|ddgdfksJ|dddf|ksJ|j|ksJgd }t j||d|}|gd ddf||ddfksJ|j|ksJt j||d |}|ddgd f|dd|fksJ|j|ksJdS) Nrrrraxisrrrrrrrrr's r test_2d_fill_nonnazTestTake.test_2d_fill_nonna;sM';$z9y$$Q''00Av>>EEeLL--tW1LLLyyy!!!|$YYY\(::??AAAAAq!!!t *//11111|y((((tW1LLLqqq)))|$QQQ \(::??AAAAAqqq!t *//11111|y((((,,tW1LLL|||QQQ'4 +;;@@BBBBB|u$$$$tW1LLLqqq,,,'47 +;;@@BBBBB|u$$$$$$rc|\}}}tjdddd|}gd}t j||d|}|gdddddf|gdddddfksJ|dddddf|ksJ|j|ksJt j||d |}|ddgdddf|ddgdddfksJ|dddddf|ksJ|j|ksJt j||d|}|ddddgdf|ddddgdfksJ|dddddf|ksJ|j|ksJgd }t j||d|}|gd ddddf||ddddfksJ|j|ksJt j||d |}|ddgd ddf|dd|ddfksJ|j|ksJt j||d|}|ddddgd f|dddd|fksJ|j|ksJdS) Nrr)r0rrrr1rrrrrrrr's r test_3d_fill_nonnazTestTake.test_3d_fill_nonnaSs';$z9y$$Q''00AyAAHHOO--tW1LLLyyy!!!QQQ'4 111aaa+@@EEGGGGGq!!!QQQw:-2244444|y((((tW1LLLqqq)))QQQ'4999aaa+@@EEGGGGGqqq!QQQw:-2244444|y((((tW1LLLqqq!!!YYY'4111iii+@@EEGGGGGqqq!!!Qw:-2244444|y((((,,tW1LLL|||QQQ)*d7AAAqqq=.AAFFHHHHH|u$$$$tW1LLLqqq,,,)*d111gqqq=.AAFFHHHHH|u$$$$tW1LLLqqq!!!\\\)*d111aaa=.AAFFHHHHH|u$$$$$$rcHtjddtj}gd}t j||}||}tj |d<tj ||dS)Nr rrrrr rrr standard_normalr"float32r#r$takenantmassert_almost_equalr(arrr+r,expecteds r test_1d_other_dtypeszTestTake.test_1d_other_dtypesusi##A&&66r::AA"*MM--sG,,88G$$v  vx00000rctjddtj}gd}t j||d}||d}tj |d<tj ||t j||d}||d}tj |dddf<tj ||dS)Nr)r7r0r8rr2rrr9r@s r test_2d_other_dtypeszTestTake.test_2d_other_dtypes~si##A&&66w??FFrzRR--sG!44488G!8,,v  vx000sG!44488G!8,,&B vx00000rc"tjgdt}tj|gd}|gd}t j||tj|gd}|jtj ksJdS)Nrrrr&rrrrrrr rarrayboolr#r$r<r>assert_numpy_array_equalr&object_r(rAr,rBs r test_1d_boolzTestTake.test_1d_boolshyyy---sLLL1188LLL)) #FH555sJJJ//|rz))))))rctjgdgdgdgt}tj|gd}|gdd}t j||tj|gdd}|gdd}t j||tj|gd }|jtj ksJdS) NrH)rrr)rrrrIrJrrErrKrLrQs r test_2d_boolzTestTake.test_2d_boolsh 999iii8EEEsLLL1188LLLq811 #FH555sLLLq99988LLLq811 #FH555sJJJ//|rz))))))rc tjddtj}gd}t j||d}||d}tj |ddgddf<tj ||t j||d}||d}tj |ddddgf<tj ||dS)Nr)rrrrrrrrrErrr9r@s r test_2d_float32zTestTake.test_2d_float32si##A&&66v>>EEbjQQ###sG!44488G!8,, f!Q vx000sG!44488G!8,, fQF vx00000rc tjdddddz}|d}gd}t j||d }||d }t|tj dd gddf<tj ||t j||d td d d }||d }td d d |dd gddf<tj ||t j||d }||d }t|tj dddd gf<tj ||t j||d td d d }||d }td d d |dddd gf<tj ||dS)NriiYr/lh]zdatetime64[ns]rIrVrrErirr1) rrr r!viewr#r$r<rint64r>r?rr@s r test_2d_datetime64zTestTake.test_2d_datetime64s I ! !! $ $ - -j*f M M  hh-h..###sG!44488G!8,,-1 bhA * vx000sG!qRS@T@TUUU88G!8,,&tQ22!Q vx000sG!44488G!8,,-1 bhAq6 * vx000sG!qRS@T@TUUU88G!8,,&tQ22QF vx00000rcttjddd}tj|ddg}tjgdgdg}t j||tj|ddgdd }tjgdgd g}t j||dS) N rrrrr) r7 T allow_fillr)rrr)rarangereshaper#r<rMr>rOrQs r test_take_axis_0zTestTake.test_take_axis_0simm##Aq))C!R))8YYY 455 #FH555C!RTaHHH8YYY 233 #FH55555rc0tjddd}tj|ddgd}tjddgdd gd d gd d gg}t j||tj|ddgddd}tjddgddgd dgd dgg}t j||tj td5tj|ddgdddddddS#1swxYwYdS)Nr]rrrrrrErr0r^r_T)r2rarindices are out-of-boundsmatch) rrbrcr#r<rMr>rOpytestraises IndexErrorrQs r test_take_axis_1zTestTake.test_take_axis_1s~imm##Aq))C!Rq1118aVaVaVaW=>> #FH555C!RqTaPPP8aVaVaVaV<== #FH555]:-H I I I K K JsQFt J J J J K K K K K K K K K K K K K K K K K Ks"D  DDctjgd}tjddg}tjtd5t j||ddgdddn #1swxYwYtjgdt}t j||ddg}tjd dggt}tj ||dS) Nrrrrrzfill_value must be a scalarriTr`rIr) rrMrkrl ValueErrorr#r<objectr>rOr@s r !test_take_non_hashable_fill_valuez*TestTake.test_take_non_hashable_fill_values&hyyy!!(Ar7## ]:-J K K K F F JsG! E E E E F F F F F F F F F F F F F F Fhyyy///CTqcJJJ8QHF333 #FH55555sA..A25A2N)__name__ __module__ __qualname__r-r3r5rCrFrRrTrWr[rdrnrsrr rr*s%%% %%%0 % % %D111111"*** * * *111"111> 6 6 6 K K K 6 6 6 6 6rrcreZdZdZdZejdddgdZdZ dZ d S) TestExtensionTakecltjddg}d}tjt|5t j|ddgddddn #1swxYwYd}tjt|5t j|ddgd ddddS#1swxYwYdS) NrrrhrirTraz2index 2 is out of bounds for( axis 0 with)? size 2F)rrMrkrlrmr#r<)r(rAmsgs r test_bounds_check_largez)TestExtensionTake.test_bounds_check_larges;h1v) ]:S 1 1 1 5 5 JsQFt 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5C ]:S 1 1 1 6 6 JsQFu 5 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s#AA!AB))B-0B-ctjgdtj}gd}d}tjt |5t j||ddddn #1swxYwYt j||}tjgdtj}tj ||dS) NrprI)rrz7'indices' contains values less than allowed \(-2 < -1\)riTr{)rrr) rrMrZrkrlrqr#r<r>rO)r(rAr+r|r,rBs r test_bounds_check_smallz)TestExtensionTake.test_bounds_check_small shyyy111++H ]:S 1 1 1 6 6 JsG 5 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6C))8IIIRX666 #FH55555sA((A,/A,raTFcdtjgtj}tj|g|}t j||dddg}tj t|5tj|dg|ddddS#1swxYwYdS)NrIr{|z.cannot do a non-empty take from an empty axes.rhrir) rrMrZr#r<r>rOjoinrkrlrm)r(rarAr,r|s r test_take_emptyz!TestExtensionTake.test_take_emptyshr***C ;;; #C000hh@+    ]:S 1 1 1 8 8 JsQCJ 7 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s?B%%B),B)ctjtjgddgdd}tjddg}t j||dS)NrTgr`)r#r<rrMr>rO)r(r,rBs r test_take_na_emptyz$TestExtensionTake.test_take_na_empty(sSBHRLL2r(tPSTTT8S#J'' #FH55555rcgd}d}tjt|5tj|ddg}dddn #1swxYwYt jddg}tj||dS)Nrpz0take accepting non-standard inputs is deprecatedrirr)r>assert_produces_warning FutureWarningr#r<rrMrO)r(rAr|r,rBs r test_take_coerces_listz(TestExtensionTake.test_take_coerces_list-sii@  ' S A A A - -ZaV,,F - - - - - - - - - - - - - - -8QF## #FH55555sAA  A N) rtrurvr}rrkmark parametrizerrrrwrr ryrys 6 6 6 6 6 6 [\D%=99 8 8:9 8666 66666rry)rnumpyrrk pandas._libsrpandas._testing_testingr>pandas.core.algorithmscore algorithmsr#fixtureint8int16int32float64 complex128rPbool_rrryrwrr rsL &&&&&&&&& ("(3--) ("(3--* 1bh 3 # :r}- 4$ 2rz" Q # S"*% Z/ T2:& R$ 2=) R]+  BM2 bj) BJ' 1bj! 3 # :rz* 4" 2rz"- 454Q6Q6Q6Q6Q6Q6Q6Q6h56565656565656565656r