o i @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)rvalXZ expected_maskmaskrm/var/www/html/eduruby.in/lip-sync/lip-sync-env/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. NrrparamsZexpected_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)pytestZraisesr)rZerr_typeerr_msgrrr*test_smallest_admissible_index_dtype_errors"r )numpyr rZsklearn.utils._testingrZsklearn.utils.fixesrrmarkZ parametrizeobjectfloatrZint32ZiinfomaxZint64rr Zint8rZuint32r ValueErrorZfloat64 TypeErrorr rrrrs       (     0