a h @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_maskmaskrZ/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/utils/tests/test_fixes.pytest_object_dtype_isnan srzparams, expected_dtypemaxvalcCstfi||ksJdS)zlCheck the behaviour of `smallest_admissible_index_dtype` depending only on the `max_val` parameter. NrparamsZexpected_dtyperrr,test_smallest_admissible_index_dtype_max_vals rarraysr)rrcCstfi||ksJdS)zCheck the behaviour of `smallest_admissible_index_dtype` using the passed arrays but without checking the contents of the arrays. Nrrrrr>test_smallest_admissible_index_dtype_without_checking_contents%s.rT)rcheck_contents)rrrcCstfi||ksJdS)zyCheck the behaviour of `smallest_admissible_index_dtype` using the dtype of the arrays but as well the contents. Nrrrrr9test_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.ndarraycCs@tj||dtfi|Wdn1s20YdS)z-Check that we raise the proper error message.)matchN)pytestZraisesr)rZerr_typeerr_msgrrr*test_smallest_admissible_index_dtype_errorsr)numpyr rZsklearn.utils._testingrZsklearn.utils.fixesrrmarkZ parametrizeobjectfloatrZint32ZiinfomaxZint64rr Zint8rZuint32r ValueErrorZfloat64 TypeErrorrrrrrs     (    0