`L i ddlZddlZddlmZddlmZmZejjde dge dge dgfdZ ejjdiejfd ejejjiejfd ejejjdziej fgd Zejjdd ej$dd gej iej fd ej$dd gej ej$dd gej fiej fd ej$dd gej ej$dd gej fiejfd ej$dd gej& iejfej$dd gej ejejjdzdej fgdZejjdej$gej ej$gej fddejfej$dgej ddejfej$ejejjdzgej* ddej fej$dgej dejejjdzdej fej$ejejjdzgej* dddej fgdZejjdd ejej jdziedfd ej$dd gej0 iedfd dd giedfgdZy)N)assert_array_equal)_object_dtype_isnan _smallest_admissible_index_dtypez dtype, valactj|tjgtj|gg|}tjddgddgg}t|}t ||y)NdtypeFT)nparraynanrr)r valX expected_maskmasks d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/utils/tests/test_fixes.pytest_object_dtype_isnanr sV 3-"&&#/u=AHHudmdE];test_smallest_admissible_index_dtype_without_checking_contentsr %s\ , 5f 5 GG GrT)rcheck_contents)rr!rc$tdi||k(sJy)zyCheck the behaviour of `smallest_admissible_index_dtype` using the dtype of the arrays but as well the contents. Nrrrs r9test_smallest_admissible_index_dtype_by_checking_contentsr#Vsh , 5f 5 GG Grzparams, err_type, err_msgz)is to large to be represented as np.int64z$Array dtype float64 is not supportedz#Arrays should be of type np.ndarraycrtj||5tdi|dddy#1swYyxYw)z-Check that we raise the proper error message.)matchNr)pytestraisesr)rerr_typeerr_msgs r*test_smallest_admissible_index_dtype_errorr*s2$ xw /3(262333s -6)numpyr r&sklearn.utils._testingrsklearn.utils.fixesrrmark parametrizeobjectfloatrint32iinfomaxint64rr int8r uint32r# ValueErrorfloat64 TypeErrorr*rrrr;sO 5U fc]UAJ'OP,Q, RXX HBHHRXX&** +RXX6 HBHHRXX&**Q. /:HH HBHHaV2884 5rxx@ BHHaV2884BHHaV2884  HH BHHaV2884BHHaV2884  HH  HBHHaV2773 4bhh? #"((Aq6:""((288,0014  HH  ;$'PHQ'PH $288Bbhh7"BHH9UV"&  HH   rxx2884 M HH #"((HBHHRXX$6$:$:Q$>#?ryyQ"&  HH  #"((A3bhh7"&""((288,0014  HH  #"((HBHHRXX$6$:$:Q$>#?ryyQ"&  HH  I,/`Ha/`Hxrxx)--1 2  7 xrxxAbjj9 :  2 QF Y(MN  3! 3r