o ?Hh @sddlZddlZddlmZddlmZmZej de dge dge dgfddZ ej d iej fd eej jiej fd eej jdiejfgd d Zej d d ejddgejdiejfd ejddgej dejddgejdfiejfd ejddgej dejddgej dfiej fd ejddgejdiej fejddgej deej jddejfgddZej d ejgejdejgejdfddej fejdgejdddej fejeej jdgejdddejfejdgej ddeej jddejfejeej jdgejddddejfgddZej dd eejjdiedfd ejddgejdiedfd ddgiedfgddZdS)N)assert_array_equal)_object_dtype_isnan _smallest_admissible_index_dtypez dtype, valacCsJtj|tjgtj|gg|d}tddgddgg}t|}t||dS)NdtypeFT)nparraynanrr)rvalX expected_maskmaskr^/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/utils/tests/test_fixes.pytest_object_dtype_isnan srzparams, expected_dtypemaxvalcCtdi||ks JdS)zlCheck the behaviour of `smallest_admissible_index_dtype` depending only on the `max_val` parameter. Nrrparamsexpected_dtyperrr,test_smallest_admissible_index_dtype_max_vals rarraysr)rrcCr)zCheck the behaviour of `smallest_admissible_index_dtype` using the passed arrays but without checking the contents of the arrays. Nrrrrrr>test_smallest_admissible_index_dtype_without_checking_contents%s.rT)rcheck_contents)rrrcCr)zyCheck the behaviour of `smallest_admissible_index_dtype` using the dtype of the arrays but as well the contents. Nrrrrrr9test_smallest_admissible_index_dtype_by_checking_contentsVs4rzparams, 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.ndarraycCsBtj||dtdi|WddS1swYdS)z-Check that we raise the proper error message.)matchNr)pytestraisesr)rerr_typeerr_msgrrr*test_smallest_admissible_index_dtype_errors"r$)numpyr r sklearn.utils._testingrsklearn.utils.fixesrrmark parametrizeobjectfloatrint32iinfomaxint64rr int8ruint32r ValueErrorfloat64 TypeErrorr$rrrrs       (     0