a hW@s<ddlZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*ej+,d e-gd gd gd d Z.d dZ/e%ddZ0e%ddZ1ej+,dddgddZ2ej+,de ej+,dddddgdfdddddgfdd gddgd fgd!ddd"d#gfd$d%gddgd&fgd'ddd(d)gfddgddgd*fgd+ddd,d-gfgd gd gddgdfgd.gd/gddddgfd0ddd gd1fddd d2d3gfdd gdd gd fgd!dd d4d5gfd$d%gdd gd6fgd'dd d7d8gfddgdd gd9fgd+dd d5d:gfgd gd gdd gd;fgd.gd/gdd dd?Z3ej+,d@e d dAdBdCZ4ej+,d@e ddAej+,dDdddge5dEfdddgge6e5fdFfdgdGe6dFfddHdge7dIffdJdKZ8dLdMZ9e%dNdOZ:e%ej+,dPgdQej+,dRddej;gedfddSej; gedSfeje%ej+,dPdddegdfdgZ?dhdiZ@Gdjdkdke ZAe%ej+,dldedmdnfddodnfdddpdnfgdqdrZBe%dsdtZCdudvZDdwdxZEej+,dae dydzZFej+,dae d{d|ZGej+,d}e ej+,d~dd gej+,ddd gej+,dgdej+,dgdddZHddZIej+,de ej+,de(ej+,dgdej+,dgdddZJej+,de ej+,ddd gddZKej+,de(ej+,ddd gddZLdS)N)partial)assert_allclose)config_context) BaseEstimator)_asarray_with_order_atol_for_type_average_convert_to_numpy_count_nonzero _estimator_with_converted_arrays_fill_or_add_to_diagonal_is_numpy_namespace_isin_max_precision_float_dtype_nanmax_nanmean_nanmin_NumPyAPIWrapper_raveldevice get_namespaceget_namespace_and_deviceindexing_dtypesupported_float_dtypes)yield_namespace_device_dtype_combinations)SkipTest_array_api_for_testsassert_array_equal'skip_if_array_api_compat_not_configured) _IS_32BITCSR_CONTAINERS np_version parse_versionXcCs&t|\}}t|tsJ|r"JdS)z.Check that get_namespace returns NumPy wrapperN)r isinstancer)r#xp_outis_array_api_compliantr+^/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/utils/tests/test_array_api.py"test_get_namespace_ndarray_default*s r-cCsztgd}t|\}}|jdddd}t|dgdtjtdd|jddd Wd n1sl0Yd S) z;Check expected behavior with device and creation functions.r$ @cpu)Z fill_valuerzUnsupported devicematchcudarN) numpyasarrayrfullrpytestraises ValueErrorzeros)r#r)_Z full_arrayr+r+r,*test_get_namespace_ndarray_creation_device2s  r=cCstd}t|jtdkr(tjddtgdg}tdd2t|\}}|sXJ||jusfJWdn1sz0YdS) z%Test get_namespace on NumPy ndarrays.array_api_compatz1.9zZarray_api_compat was temporarily reporting NumPy as API compliant and this test would fail)reasonr$TZarray_api_dispatchN) r8 importorskipr" __version__skipr5r6rr)r>X_npr)r*r+r+r,(test_get_namespace_ndarray_with_dispatch>s   rEc Cstd}tgdg}||}tddt|\}}|sDJttt||\}}Wdn1sr0Ydd}|d|t j d d ksJtjt d d t|Wdn1s0YWdn1s0YdS) z'Test get_namespace for ArrayAPI arrays.array_api_strictr$Tr@NcSs|dkr dSdS)NSCIPY_ARRAY_API0r+)keyr+r+r, mock_getenvasz1test_get_namespace_array_api..mock_getenvzos.environ.getrG1z#scipy's own support is not enabled.r1) r8rAr5r6rrr9 TypeErrorsetattrosenvironget RuntimeError)Z monkeypatchxprDZX_xpr)r*rJr+r+r,test_get_namespace_array_apiSs      , rS array_apir5rFcCsBt|}|gd}t|d|d}t|}|jds>JdS)z=Test _asarray_with_order passes along order for NumPy arrays.)g333333?g333333 @gffffff@F)orderrRZ F_CONTIGUOUSN)r8rAr6rr5flags)rTrRr#ZX_newZX_new_npr+r+r,test_asarray_with_orderns   rXz$array_namespace, device_, dtype_namez"weights, axis, normalize, expected)NNT @T)g@rY@r%r&F)TTF?rZ皙?g?)g?g@g @)r]皙?r^g?g@r'r[)r%r%r&g@g@)r%r&r%)r&r&r&)NNF)r[ r'rc)皙??g?gffffff?gffffff@)rc rera)r[c Cst||}tjgdgdg|d}|j||d}|durVtj||d}|j||d}tdd t||||d} Wdn1s0Yttdksttd krt|t| ksJt| |} t | |t |d dS) Nr$r`r[rddtyper4Tr@)axisweights normalize2.0.02.1.0)Zatol) rr5r6rrr!r"rr rr) array_namespacedevice_ dtype_namerqrprrexpectedrRarray_inresultr+r+r, test_averagezs#  . r{z#array_namespace, device, dtype_name)Zinclude_numpy_namespacesc Cst||}tjddg|ddtjddg|d}|jj}t||sXt|d||j||d}d }td d Ftj t |d t |Wdn1s0YWdn1s0YdS) Nr&rrny?r`r'z does not support r4z;Complex floating point values are not supported by average.Tr@r1) rr5r6ronamehasattrr8rCrr9NotImplementedErrorr)rurrwrRryZcomplex_type_nameerr_msgr+r+r,$test_average_raises_with_wrong_dtypes    rzaxis, weights, error, error_msgzAxis must be specifiedrq)r%r&r'r`z(Weights sum to zero, can't be normalizedc Cst||}tjgdgdg|d}|j||d}tj||d}|j||d}tddLtj||dt|||dWdn1s0YWdn1s0YdS) Nr$rmrnr4Tr@r1)rprq)rr5r6rr8r9r) rurrwrprqerror error_msgrRryr+r+r,+test_average_raises_with_invalid_parameterss rcCs$tdusJtdddus JdS)Nr|r4r+r+r+r,test_device_none_if_no_inputsrc s4GdddGfddd}ttt|djWdn1sN0Yt|d|ddusrJd}td d tjt|d "t|d|dWdn1s0Y|d}|d}|jt|ksJ|jt||ksJ|jt|||ksJWdn1s&0YdS) Nc@s,eZdZddZddZddZddZd S) z&test_device_inspection..DevicecSs ||_dSNr|)selfr|r+r+r,__init__sz/test_device_inspection..Device.__init__cSs |j|jkSrr)rrr+r+r,__eq__sz-test_device_inspection..Device.__eq__cSs tddS)NzDevice object is not hashable)rLrr+r+r,__hash__sz/test_device_inspection..Device.__hash__cSs|jSrrrr+r+r,__str__sz.test_device_inspection..Device.__str__N)__name__ __module__ __qualname__rrrrr+r+r+r,DevicesrcseZdZfddZdS)z%test_device_inspection..Arraycs||_dSrr4)rZ device_namerr+r,r sz.test_device_inspection..Array.__init__N)rrrrr+rr+r,Array srrr0Zmygpuz.Input arrays use different devices: cpu, mygpuTr@r1)r8r9rLhashrrr:)rrZarray1Zarray2r+rr,test_device_inspections  , 0rlibrary)r5rFtorchzX,reduction,expected)r`r[@)rp?r/@r@)r@rrrgr/rcCsZt|}tdd|||}Wdn1s80Yt||}t||dS)z-Check NaN reductions like _nanmin and _nanmaxTr@N)r8rArr6r r)rr#Z reductionrxrRrzr+r+r,test_nan_reductions.s 0  , rznamespace, _device, _dtypecCst||}gdgdgdgdg}|j||d}tddt|}Wdn1s\0Yt||}tj|dd }t||t|rt|j d sJdS) Nr$rm)rbrkrc)r. rhr4Tr@C)rVZ C_CONTIGUOUS) rr6rrr r5Zravelrr rW) namespace_device_dtyperRarrayarray_xprzrxr+r+r, test_ravelgs  &  rZcupyrcCsrt|}|dkr<|jjs(td|jgddd}n|gd}t||d}tgd}t ||dS)z0Check convert_to_numpy for GPU backed libraries.rztest requires cudarr3r4rRN) r8rAbackendsr3Zis_builtrCr6r r5r)rrRZX_gpuX_cpuexpected_outputr+r+r,test_convert_to_numpy_gpu{s    rcCsDtd}|jgddd}t||d}tgd}t||dS)z.Check convert_to_numpy for PyTorch CPU arrays.rrr0r4rN)r8rAr6r r5r)rZX_torchrrr+r+r,test_convert_to_numpy_cpus   rc@seZdZdddZdS)SimpleEstimatorNcCs||_|jd|_|S)Nr)X_shapeZ n_features_)rr#yr+r+r,fits zSimpleEstimator.fit)N)rrrrr+r+r+r,rsrzarray_namespace, convertercCs |Sr)r0r5rr+r+r,rcCs t|Sr)r5r6rr+r+r,rrcCs|Sr)rPrr+r+r,rrcCsFt|}|ddgg}t|}t||}t|jtj sBJdS)z(Convert estimator attributes to ndarray.rgrZN) r8rAr6rrr r(rr5Zndarray)ru converterrRr#estnew_estr+r+r,!test_convert_estimator_to_ndarrays   rcsLtdtddgg}t|}t|fdd}t|jdsHJdS)z0Convert estimator attributes to ArrayAPI arrays.rFrgrZcs |Sr)r6rrr+r,rrz5test_convert_estimator_to_array_api..Z__array_namespace__N) r8rAr5r6rrr r}r)rDrrr+rr,#test_convert_estimator_to_array_apis   rcCst}|gdgdg}|j|ddd}|j|us:J|j|ddd}|j|jusZJtjtdd ||d Wd n1s0Yd S) zDCheck reshape behavior with copy and is strict with non-tuple shape.r$r_)rF)copy)rdr%Tzshape must be a tupler1rN)rr6reshapebaser8r9rL)rRr#Z X_no_copyZX_copyr+r+r,test_reshape_behaviorsrcCst}||j|jsJ||jds,J||jds>J||jdrPJt|D]}||dsXJqX||jdsJ||jdrJ||jdsJ||jdrJ||j dsJ||j drJ||j dsJ||jdsJ||jdsJ||jdr&J||j dr:J||j dsNJ||j dsbJtjtdd$||jd sJWd n1s0Yd S) z2Test isdtype implementation from _NumPyAPIWrapper.z real floatingboolzsigned integerzunsigned integernumericzcomplex floatingzUnrecognized data typer1unknownN)rZisdtypefloat32float64int32rrint16Zuint32Zuint16int64Zint8Z complex64Z complex128r8r9r:)rRror+r+r,$test_get_namespace_array_api_isdtypes, rcCs8t||}tr"t||jks4Jnt||jks4JdSr)rrrrr)rrrrRr+r+r,test_indexing_dtypes rcCs4t||}|dkr|jn|j}t|||ks0JdS)NZmps)rrrr)rrrrRZexpected_dtyper+r+r,test_max_precision_float_dtypes rzarray_namespace, device, _invert assume_unique element_size)rdr.ri int_dtype)rrrZuint8cCst||}|d}dt||df|} tjtd|d} |j| |d} |j| |d} tj| | ||d} tdd"t | | |||d}Wdn1s0Yt t ||d | dS) Nr&rirnr4)element test_elementsrrTr@)rrrRrrr) rr5arangerZastyperr6isinrrrr )rurr<rrrrrRrrrZ element_xpZtest_elements_xprxrzr+r+r, test_isins*  $rcCstd}td}|jddd}td}t|\}}}|t|dusNJ|rVJ|dusbJtdd@t|\}}}||usJ|sJ||jksJWdn1s0YdS) Nrzarray_api_compat.torchr'r0r4rTr@)r8rArr5rrrr)rZxp_torchZsome_torch_tensorZsome_numpy_arrayrZ is_array_apirr+r+r,test_get_namespace_and_device's      r csr_containerrp)rr%Nrrsample_weight_type)Nintfloatc Csddlm}t||}tgdgdgdgdgdg}|dkrVtgd } n |d krrtjgd |d } nd} ||||| d } |j||d} tdd"t| || ||d} Wdn1s0Ytt | |d| t t dkst t dkrt | t | ksJdS)Nr) count_nonzero)rr'r)r&rr)rrr)rcrkrb)r`rr[r)r%r&r&r'r%r)g?r\rfg @g333333@rn)rp sample_weightr4Tr@)rprrRrrrsrt) Zsklearn.utils.sparsefuncsrrr5rr6rr rr r!r"r) rurvrwrrprZsparse_count_nonzerorRrrrxrrzr+r+r,test_count_nonzero@s&  (   $rwrapcCs\t||}tjdtjd}||}t|d|d|dtj|d|dtt||d|dS)N)r[r`rnr%F)valuerRZ add_valuer)valrr) rr5r;rr6r Z fill_diagonalrr )rurvrwrrRZarray_nprr+r+r,test_fill_or_add_to_diagonales   rdispatchcCs|tdgg|tdgg}}zt|dpt||dusFJt|tdgdus`Jt||ddusvJt|tdgddusJWdn1s0YWntytdYn0dS)Nr%r&r@z!array_api_compat is not installed)r5rrrr ImportErrorr)rrabr+r+r,test_sparse_devicers& @ r)MrN functoolsrr5r8Z numpy.testingrZsklearn._configrZ sklearn.baserZsklearn.utils._array_apirrrr r r r r rrrrrrrrrrrrrZsklearn.utils._testingrrrrZsklearn.utils.fixesrr r!r"markZ parametrizer6r-r=rErSrXr{rrLr:ZeroDivisionErrorrrrnaninfrrrrrrrrrrrrrrrrr+r+r+r,s^    \         2         ,         #