
    -ir                        S SK rS SKrS SKJrJr  S SKJrJrJ	r	J
r
Jr  \" \R                  " 5       5      rS r\R                   R#                  S\	" S SSS5      \	" S SSS5      \	" S SSS5      \	" S SSS5      \	" \R$                  * \R$                  SS5      \	" \R$                  * \R$                  SS5      \	" \R$                  * \R$                  SS5      \	" \R$                  * \R$                  SS5      \	" S	S
SS5      \	" S	S
SS5      \	" S	S
SS5      \	" S	S
SS5      /5      S 5       r\R                   R#                  S\5      S 5       r\R                   R#                  S\5      S 5       rg)    N)assert_allcloseassert_array_equal)_LINKSHalfLogitLinkIntervalMultinomialLogit_inclusive_low_highc                      [         R                  " [        SS9   [        SSSS5        SSS5        g! , (       d  f       g= f)z5Test that interval with low > high raises ValueError.z-One must have low <= high; got low=1, high=0.)match   r   FN)pytestraises
ValueErrorr        P/var/www/html/venv/lib/python3.13/site-packages/sklearn/_loss/tests/test_link.pytest_interval_raisesr      s4    	I
 	Aue$
 
 
s	   2
A intervalr   FTc                 "   [        U 5      u  p[        R                  " XSS9nU R                  U5      (       d   eU R                  [        R                  X0R
                  4   5      U R                  :X  d   eU R                  [        R                  X0R                  4   5      U R                  :X  d   eU R                  [        R                  X0R
                  U R                  4   5      U R                  =(       a    U R                  :X  d   eg )N
   )num)	r	   nplinspaceincludesr_lowlow_inclusivehighhigh_inclusive)r   r   r    xs       r   test_is_in_ranger#      s    & $H-IC
C2&AQ RUU1ll?348N8NNNN RUU1mm#3459P9PPPP RUU1llHMM#ABC:8#:#:  r   linkc                 *   [         R                  R                  U5      nU " 5       n Su  p4U R                  (       a;  SnUR	                  SSX44S9n[        U [        5      (       a  U R                  U5      nO8[        U [        5      (       a  UR	                  SSUS9nOUR	                  SSUS9n[        U R                  U R                  U5      5      U5        U R                  U5      n[        U R                  U R                  U5      5      U5        g )Nd   Nr   i   r   r    sizer   )r   randomRandomStateis_multiclassuniform
isinstancer   symmetrize_raw_predictionr   r   r$   inverse)r$   global_random_seedrng	n_samples	n_classesraw_predictiony_preds          r   test_link_inverse_identityr8   <   s     ))

 2
3C6D$I 	2Y<RSd,--!;;NKN	D-	(	(2YH2YHDIIdll>:;^L\\.)FDLL6!23V<r   c                    [         R                  R                  S5      nU " 5       n Su  p#U R                  (       a;  SnUR	                  SSX#4S9n[        U [        5      (       a  U R                  U5      nOUR                  SSUS9nU R                  US S9n[         R                  " U5      nU R                  XFS9n[        XV5        [        Xg5        [         R                  " Xg5      (       d   e[         R                  " U5      nU R                  XVS9n[        XF5        [        Xh5        [         R                  " Xh5      (       d   eg )	N*   r&   r   r   )locscaler*   r   r)   )out)r   r+   r,   r-   normalr/   r   r0   r.   r1   
empty_liker   r   shares_memoryr$   )	r$   r3   r4   r5   r6   r7   r=   y_pred_2raw_prediction_2s	            r   test_link_out_argumentrC   T   s    ))


#C6D$I	9:PQd,--!;;NKN 2YH\\.d\3F
--
'C||N|4HF s%C****
--
Cyyy1N(s-C2222r   )numpyr   r   numpy.testingr   r   sklearn._loss.linkr   r   r   r   r	   listvaluesLINK_FUNCTIONSr   markparametrizeinfr#   r8   rC   r   r   r   <module>rM      st     =  fmmo&% Aue$Aud#AtU#AtT""&&"&&%/"&&"&&%."&&"&&$."&&"&&$-b%'b%&b$&b$%"#"& 0= 1=. 03 13r   