
    UgcU                        d dl Zd dlZd dlmZ d dlmZ d dlm	Z	m
Z
 d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ ed             Zd	 Zd
 Zd ZdMdZd Zd Zd Zd Zd Zd Z d Z!d Z"ej#        $                    dg dg df ej%        ddgddgddgddgg           ej%        ddgddgddgddgg          fg          d             Z&ej#        $                    dg d          d             Z'd Z(d  Z)d! Z*d" Z+d# Z,d$ Z-d% Z.d& Z/d' Z0d( Z1d) Z2d* Z3d+ Z4d, Z5d- Z6d. Z7ej#        $                    d/g d0d1d id2fg d0d3dd gid4f ej8        g d0g d0g          d3did4fg d0d3d5id6f ej8        g d0g d0g          d3dd7gid8fgg d9:          d;             Z9d< Z:ej#        $                    d=e          d>             Z;ej#        $                    d=e          d?             Z<ej#        $                    d=e          d@             Z=ej#        $                    d=e          dA             Z>dMdBZ?dC Z@dD ZAdE ZBej#        $                    dg dFdGgdHz  f ej%        ddgddgddgddgg          dGdGggdHz  fg          dI             ZCej#        $                    dg dJ          dK             ZDej#        $                    dg d          dL             ZEdS )N    N)clone)DummyClassifierDummyRegressor)NotFittedError)assert_almost_equalassert_array_almost_equalassert_array_equalignore_warnings)CSC_CONTAINERS)_weighted_percentilec           	         |                      |          }|                     |          }t          j        |          }|j        dk    rt          j        |d          }|j        d         }t          |          }|dk    r|g}|g}t          |          D ]}||         j        d         |k    sJ ||         j        d         t          t          j	        |d d |f                             k    sJ t          ||                             d          t          j        t          |                               t          t          j        ||                   ||                    d S )N   )r   r   axis)predict_probapredict_log_probanp
atleast_1dndimreshapeshapelenrangeuniquer   sumoneslog)clfXyproba	log_proba	n_outputs	n_samplesks           W/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/sklearn/tests/test_dummy.py_check_predict_probar(      sP   a  E%%a((I
aAv{{Jq'""
IAIA~~K	9 B BQx~a I----Qx~a C	!AAAqD'(:(:$;$;;;;;!%(,,A,"6"6AHHH!"&q"2"2IaLAAAAB B    c                    t          j        dgdgdgdgg          }t          j        g d          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ t          j        ddgddgddgddgg          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ d S )Nr   r      r   r   r   r,      r   arrayr   fitpredictr   r   r    r!   esty_preds        r'   _check_behavior_2dr5   +   s    
1#sQC!%&&A
A
**CGGAqMMM[[^^F7fl"""" 	1a&1a&1a&1a&122A
**CGGAqMMM[[^^F7fl""""""r)   c                    t          j        dgdgdgdgg          }t          j        g dg dg dg dg          }t          |           }|                    ||           |                    |          }|j        |j        k    sJ d S )Nr   )r   r         r-   )r,   r   r   r,   r7   )r   r   r8   r7   r,   )r   r-   r-   r,   r   r.   r2   s        r'   _check_behavior_2d_for_constantr9   <   s    
1#sQC!%&&A
///???OOO___UVVA
**CGGAqMMM[[^^F7fl""""""r)   c                     t          t          j        | |j        d         df          |           t          t          j        | |j        d         df          |           d S )Nr   r   )r   r   tiler   )	statisticy_learny_pred_learny_testy_pred_tests        r'   _check_equality_regressorrA   F   sW    bgi'-2BA1FGGVVVbgi&,q/11EFFTTTTTr)   
   c                    t          j        d          }t          j                            |           }|                    dgg|z  dg          }|                    |          }t                                          ||          }t          |d          sJ t          |d          sJ t                                          ||          }t          |d          sJ t          |d          sJ d S )Npandasseedr   	feature_1)columnsfeature_names_in_n_features_in_)pytestimportorskipr   randomRandomState	DataFramerandr   r0   hasattrr   )global_random_seedr%   pdrandom_stater    r!   r3   s          r'   (test_feature_names_in_and_n_features_in_rU   K   s    		X	&	&B9((.@(AAL
qcUY&>>A)$$A




q!
$
$C3+,,,,,3()))))




1
%
%C3+,,,,,3()))))))r)   c            	      B   dgdgdgdgg} g d}dD ]}t          |d          }|                    | |           t          |                    |           t	          j        t          |                                t          || |           |dk    rCt          |	                    | d         g          |j
                            d                     t          |	                    | d         g          |j
                            d          dk               d S )Nr   r+   most_frequentpriorstrategyrT   rY   r   r         ?)r   r0   r	   r1   r   r   r   r(   r   r   class_prior_r   )r    r!   r[   r   s       r'   %test_most_frequent_and_prior_strategyr_   \   s+   
qcA3AA.  xa@@@13;;q>>273q66??;;;S!Q'''w%!!1Q4&))3+;+C+CG+L+L    &!!1Q4&))3+;+C+CG+L+Ls+R    r)   c                  H   dgdgdgdgg} g d}dgdgdgdgg}dD ]}t          |d          }t          |d          }|                    | |           |                    | |           t          |                    |           |                    |                      d S )Nr   r+   r   r,   rW   rZ   r   r0   r	   r1   )r    y_1dy_2dr[   clf_1dclf_2ds         r'   6test_most_frequent_and_prior_strategy_with_2d_column_yrf   p   s     qcA3A<<DC!qcA3D. A A (CCC (CCC

1d

1d6>>!,,fnnQ.?.?@@@@A Ar)   c                     dgdgdgdgg} t          j        ddgddgddgddgg          }t          |           }dD ]}t          |d          }|                    | |           t          |                    |           t          j        t          j        |df          t          j	        |df          g                     t          || |           t          |           d S )Nr   r   r,   r-   )rY   rX   rZ   )r   r/   r   r   r0   r	   r1   hstackr   zerosr(   r5   )r    r!   r%   r[   r   s        r'   1test_most_frequent_and_prior_strategy_multioutputrj      s    
qcA3A
1a&1a&1a&1a&122AAI.    xa@@@1KKNNIrw	1~..)Q0H0HIJJ	
 	
 	
 	S!Q'''3   r)   c                    dggdz  }g d}t          d|           }|                    ||           dggdz  }|                    |          }t          j        |          t          t          |                    z  }t          |d         dd	           t          |d
         dd	           t          |||           d S )Nr   r7   )r   r,   r   r   r,   
stratifiedrZ     r   333333?decimalr,   皙?	r   r0   r1   r   bincountfloatr   r   r(   rR   r    r!   r   r4   ps         r'   test_stratified_strategyrw      s    
	AA
<>P
Q
Q
QCGGAqMMM
A[[^^F
FeCFFmm+A!gq1111!gq1111a#####r)   c                 0   dggdz  }t          j        ddgddgddgddgddgg          }t          d|           }|                    ||           dggdz  }|                    |          }t          |j        d                   D ]~}t          j        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         d
d	           t          |||           t          |           d S )Nr   r7   r,   r   rl   rZ   rm   rn   ro   rq   r   r/   r   r0   r1   r   r   rs   rt   r   r   r(   r5   rR   r    r!   r   r4   r&   rv   s          r'   $test_stratified_strategy_multioutputr{      s!   
	A
1a&1a&1a&1a&1a&9::A
<>P
Q
Q
QCGGAqMMM
A[[^^F171: ( (Kqqq!t%%c!ff5AaD'15555AaD'15555S!Q''''sr)   c                    dggdz  }g d}t          d|           }|                    ||           dggdz  }|                    |          }t          j        |          t          t          |                    z  }t          |d         dd	           t          |d
         dd	           t          |||           d S )Nr   r8   r+   uniformrZ   rm   r   r]   ro   r,   rr   ru   s         r'   test_uniform_strategyr~      s    
	AA
9;M
N
N
NCGGAqMMM
A[[^^F
FeCFFmm+A!c1----!c1----a#####r)   c                 *   dggdz  }t          j        ddgddgddgddgg          }t          d|           }|                    ||           dggdz  }|                    |          }t          |j        d                   D ]~}t          j        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         dd	           t          |||           t          |           d S )
Nr   r8   r,   r   r}   rZ   rm   r]   ro   ry   rz   s          r'   !test_uniform_strategy_multioutputr      s   
	A
1a&1a&1a&1a&122A
9;M
N
N
NCGGAqMMM
A[[^^F171: ( (Kqqq!t%%c!ff5AaD#q1111AaD#q1111S!Q''''sr)   c                      dggdz  } g d}t          d          }|                    | |           t          |                    |           dgdz             d S )Nr   r7   )parisr   tokyo	amsterdamberlinrX   r[   r   ra   r    r!   r   s      r'   test_string_labelsr      sa    
	A:::A
?
3
3
3CGGAqMMMs{{1~~y1}55555r)   zy,y_test)r,   r   r   r   )r,   r,   r   r   r,   r   c                     t          d          }|                    d |            |                    d |          dk    sJ d S )NrX   r   r]   )r   r0   score)r!   r?   r   s      r'   test_classifier_score_with_Noner      sM     ?
3
3
3CGGD!99T6""c))))))r)   r[   )rl   rX   rY   r}   constantc                 >   g d}dggdz  }t          | |d          }|                    ||           |                    |          }dggdz  }t          | |d          }|                    ||           |                    |          }t          ||           d S )Nr   r,   r   r   r   r8   r[   rT   r   r   )r   r0   r1   r	   )	r[   rR   r!   X1clf1predictions1X2clf2predictions2s	            r'   +test_classifier_prediction_independent_of_Xr      s     	A#B(:Q  D 	HHROOO<<##L#B(:Q  D 	HHROOO<<##L|\22222r)   c                 R   t           j                            |           }dggdz  }|                    d          }t	                      }|                    ||           t          |                    |          t          j        |          gt          |          z             d S )NrE   r   r8   )
r   rM   rN   randnr   r0   r	   r1   meanr   rR   rT   r    r!   regs        r'   test_mean_strategy_regressorr      s    9((.@(AAL
	A1A


CGGAqMMMs{{1~~

|c!ff'<=====r)   c                 (   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }|                    dd          }|                    dd          }t                      }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )NrE   rB   r7   r   r   r\      )r   rM   rN   r   r   r   r   r0   r1   rA   r5   )
rR   rT   X_learnr=   r   X_testr?   r3   r>   r@   s
             r'   (test_mean_strategy_multioutput_regressorr     s    9((.@(AAL  R((G  Q''G77###++G44DB''FA&&F 

CGGGW;;w''L++f%%KdG\6;OOOsr)   c                      t                      } t          j        t                    5  |                     g            d d d            d S # 1 swxY w Y   d S )N)r   rK   raisesr   r1   )r   s    r'   test_regressor_exceptionsr      s    


C	~	&	&  B                 s   AAAc                 V   t           j                            |           }dggdz  }|                    d          }t	          d          }|                    ||           t          |                    |          t          j        |          gt          |          z             d S )NrE   r   r7   medianr   )
r   rM   rN   r   r   r0   r	   r1   r   r   r   s        r'   test_median_strategy_regressorr   &  s    9((.@(AAL
	A1A
(
+
+
+CGGAqMMMs{{1~~	!~A'>?????r)   c                 ,   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }|                    dd          }|                    dd          }t          d	          }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )
NrE   rB   r7   r   r   r\   r   r   r   )r   rM   rN   r   r   r   r   r0   r1   rA   r5   )
rR   rT   r   r=   r   r   r?   r3   r>   r@   s
             r'   *test_median_strategy_multioutput_regressorr   1  s   9((.@(AAL  R((G  Q''GYwQ'''//88FB''FA&&F (
+
+
+CGGGW;;w''L++f%%Kfg|V[QQQsr)   c                    t           j                            |           }dggdz  }|                    d          }t	          dd          }|                    ||           t          |                    |          t          j        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j
        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j        |          gt          |          z             t	          dd          }|                    ||           t          |                    |          t          j        |d	
          gt          |          z             d S )NrE   r   r7   quantiler]   r[   r   r   g333333?   )q)r   rM   rN   r   r   r0   r	   r1   r   r   minmax
percentiler   s        r'    test_quantile_strategy_regressorr   F  s   9((.@(AAL
	A1A
*s
;
;
;CGGAqMMMs{{1~~	!~A'>???
*q
9
9
9CGGAqMMMs{{1~~q		{SVV';<<<
*q
9
9
9CGGAqMMMs{{1~~q		{SVV';<<<
*s
;
;
;CGGAqMMMs{{1~~a2(>(>(>'?#a&&'HIIIIIr)   c                 h   t           j                            |           }|                    dd          }|                    dd          }t          j        |d                              d          }t          j        |dd                              d          }|                    d	d          }|                    d	d          }t          d
d          }|                    ||           |	                    |          }	|	                    |          }
t          |||	||
           t          |           t          d
d          }|                    ||           |	                    |          }	|	                    |          }
t          |||	||
           t          |           d S )NrE   rB   r7   r   r   r\   P   )r   r   r   r   r]   r   g?)r   rM   rN   r   r   r   r   r   r0   r1   rA   r5   )rR   rT   r   r=   r   quantile_valuesr   r?   r3   r>   r@   s              r'   ,test_quantile_strategy_multioutput_regressorr   ]  s   9((.@(AAL  R((G  Q''GYwQ'''//88FmG!r:::BB7KKOB''FA&&F *s
;
;
;CGGGW;;w''L++f%%Kfg|V[QQQs *s
;
;
;CGGGW;;w''L++f%%K,   sr)   c                      dggdz  } dgdz  }t          dd           }d}t          j        t          |          5  |                    | |           d d d            d S # 1 swxY w Y   d S )Nr   r7   r   r   zIWhen using `strategy='quantile', you have to specify the desired quantilematch)r   rK   r   
ValueErrorr0   r    r!   r3   err_msgs       r'   test_quantile_invalidr   ~  s    
	A	
aA
*t
<
<
<CS  
z	1	1	1  1                 s   A  A$'A$c                      t          dd          } t          j        t                    5  |                     g g            d d d            d S # 1 swxY w Y   d S )Nr   rq   r   )r   rK   r   
IndexErrorr0   )r3   s    r'   "test_quantile_strategy_empty_trainr     s    
*s
;
;
;C	z	"	"  B                 s   AAAc                 .   t           j                            |           }dggdz  }|                    d          }t	          ddg          }|                    ||           t          |                    |          dgt          |          z             t	          dd          }|                    ||           t          |                    |          dgt          |          z             t          |j
        t           j                  rJ d S )NrE   r   r7   r   +   r[   r   )r   rM   rN   r   r   r0   r	   r1   r   
isinstancer   ndarrayr   s        r'    test_constant_strategy_regressorr     s    9((.@(AAL
	A1A
*t
<
<
<CGGAqMMMs{{1~~tc!ff}555
*r
:
:
:CGGAqMMMs{{1~~tc!ff}555 #,
3333333r)   c                    t           j                            |           }|                    dd          }|                    dd          }|                    d          }|                    dd          }|                    dd          }t	          d|          }|                    ||           |                    |          }|                    |          }	t          |||||	           t          |           d S )NrE   rB   r7   r   r   r   )	r   rM   rN   r   r   r0   r1   rA   r9   )
rR   rT   r   r=   	constantsr   r?   r3   r>   r@   s
             r'   ,test_constant_strategy_multioutput_regressorr     s    9((.@(AAL  R((G  Q''G ""1%%IB''FA&&F *y
A
A
ACGGGW;;w''L++f%%Ki,TTT#C(((((r)   c                      dggdz  } g d}t          d          }|                    | |           |j        t          j        |          k    sJ d S )Nr   r7   r   r,   r8         r   r   )r   r0   	constant_r   r   )r    r!   r3   s      r'   test_y_mean_attribute_regressorr     sZ    
	AA
&
)
)
)CGGAqMMM=BGAJJ&&&&&&r)   c                      dggdz  } g d}t          d          }d}t          j        t          |          5  |                    | |           d d d            d S # 1 swxY w Y   d S )Nr   r7   r   r   r   z)Constant target value has to be specifiedr   )r   rK   r   	TypeErrorr0   r   s       r'   &test_constants_not_specified_regressorr     s    
	AA
*
-
-
-C9G	y	0	0	0  1                 s   AA!$A!c                 \   t           j                            |           }|                    dd          }|                    dd          }t	          dg d          }d}t          j        t          |          5  |                    ||           d d d            d S # 1 swxY w Y   d S )	NrE   rB   r7   r   )r   r,   r-   r8   r   z1Constant target value should have shape \(5, 1\).r   )	r   rM   rN   r   r   rK   r   r   r0   )rR   rT   r    r!   r3   r   s         r'   (test_constant_size_multioutput_regressorr     s    9((.@(AAL2r""A2q!!A
*|||
D
D
DCBG	z	1	1	1  1                 s   =B!!B%(B%c                     dgdgdgdgg} g d}t          ddd          }|                    | |           t          |                    |           t	          j        t          |                                t          || |           dgdgdgdgg} g d}t          ddd          }|                    | |           t          |                    |           t	          j        dgdz                       t          || |           d S )	Nr   r,   r   r,   r,   r   r   r   )twooner   r   r   r8   )	r   r0   r	   r1   r   r   r   r(   r/   r   s      r'   test_constant_strategyr     s   
qcA3AA
:A
J
J
JCGGAqMMMs{{1~~rws1vv777a###
qcA3A$$$A
:A
N
N
NCGGAqMMMs{{1~~rx!'<'<===a#####r)   c            
         dgdgdgdgg} t          j        ddgddgddgddgg          }t          |           }t          ddddg          }|                    | |           t          |                    |           t          j        t          j        |df          t          j	        |df          g                     t          || |           d S )Nr   r,   r-   r   r   r   )r   r/   r   r   r0   r	   r1   rh   r   ri   r(   )r    r!   r%   r   s       r'   "test_constant_strategy_multioutputr     s    
qcA3A
1a&1a&1a&1a&122AAI
:AA
O
O
OCGGAqMMMA	27Iq>#:#:BHiQR^<T<T"UVV   a#####r)   zy, params, err_msgr   rT   zConstant.*has to be specifiedr   zConstant.*should have shapezmy-constantz.constant=my-constant.*Possible values.*\[1, 2]unknownz2constant=\[2, 'unknown'].*Possible values.*\[1, 2])zno-constantztoo-many-constantznot-enough-outputzsingle-outputzmulti-output)idsc                     dgdgdgdgg}t          dddi|}t          j        t          |          5  |                    ||            d d d            d S # 1 swxY w Y   d S )Nr   r[   r   r    )r   rK   r   r   r0   )r!   paramsr   r    r   s        r'   !test_constant_strategy_exceptionsr     s    < qcA3A

8
8:
8
8
8C	z	1	1	1  1                 s   AAAc                      dgdgdgg} g d}g d}t          d                              | ||          }t          |j        ddg           d S )	Nr   r   )r   r   r   )皙?      ?r   rl   r   gVUUUUU?g?)r   r0   r   r^   )r    r!   sample_weightr   s       r'   !test_classification_sample_weightr     se    
qcA3A		A#OOM
<
0
0
0
4
4Q=
I
ICc.I0FGGGGGr)   csc_containerc           
         dggdz  } | t          j        ddgddgddgddgddgg                    }t          |          }t          ddddg          }|                    ||           |                    |          }t          j        |          sJ t          |	                                t          j
        t          j        |df          t          j        |df          g                     d S )Nr   r7   r   r8   r   r   )r   r/   r   r   r0   r1   spissparser	   toarrayrh   r   ri   )r   r    r!   r%   r   r4   s         r'   $test_constant_strategy_sparse_targetr   $  s    
	AbhAAAAAGHHIIAAI
:AA
O
O
OCGGAqMMM[[^^F;v")RWi^%<%<bh	ST~>V>V$WXX    r)   c           
         dggdz  } |t          j        ddgddgddgddgddgg                    }t          d|           }t          j        t
          d	          5  |                    ||           d d d            n# 1 swxY w Y   dggd
z  }|                    |          }t          |j	        d                   D ]}t          j
        |d d |f                   t          t          |                    z  }t          |d         dd           t          |d         dd           t          |d         dd           d S )Nr   r7   r,   r   r8   r}   rZ   z*the uniform strategy would not save memoryr   rm   gUUUUUU?ro   )r   r/   r   rK   warnsUserWarningr0   r1   r   r   rs   rt   r   r   rR   r   r    r!   r   r4   r&   rv   s           r'   +test_uniform_strategy_sparse_target_warningr   4  s   
	AbhAAAAAGHHIIA
9;M
N
N
NC	k)U	V	V	V  1               A[[^^F171: 4 4Kqqq!t%%c!ff5AaD%3333AaD%3333AaD%33333	4 4s    BB
Bc           
         dggdz  } |t          j        ddgddgddgddgddgg                    }t          d|           }|                    ||           dggdz  }|                    |          }t          j        |          sJ |                                }t          |j	        d                   D ]}t          j
        |d d |f                   t          t          |                    z  }t          |d         dd	           t          |d         d
d	           t          |d         d
d	           d S )Nr   r7   r8   r   rl   rZ   rm   rn   ro   g?)r   r/   r   r0   r1   r   r   r   r   r   rs   rt   r   r   r   s           r'   &test_stratified_strategy_sparse_targetr   G  sM   
	AbhAAAAAGHHIIA
<>P
Q
Q
QCGGAqMMM
A[[^^F;v^^F171: 6 6Kqqq!t%%c!ff5AaD'15555AaD'15555AaD'155555	6 6r)   c           
         dggdz  } | t          j        ddgddgddgddgddgg                    }t          |          }t          j        t          j        |df          t          j        |df          g          }dD ]v}t          |d          }|                    ||           |                    |          }t          j
        |          sJ t          |                                |           wd S )Nr   r7   r   r-   r8   rW   rZ   )r   r/   r   rh   r   ri   r   r0   r1   r   r   r	   r   )r   r    r!   r%   
y_expectedr[   r   r4   s           r'   3test_most_frequent_and_prior_strategy_sparse_targetr   [  s    
	AbhAAAAAGHHIIAAIBGYN33RXy!n5M5MNOOJ. 9 9xa@@@1Q{6"""""6>>++Z88889 9r)   c                 :   t           j                            |           }dgg|z  }|                    |          }|                    |          }t	          d                              |||          }|j        t          j        ||          k    sJ t	          d                              |||          }|j        t          ||d          k    sJ t	          dd	
                              |||          }|j        t          ||d          k    sJ d S )NrE   r   r   r   )weightsr   g      I@r   gffffff?r   g     W@)	r   rM   rN   rP   r   r0   r   averager   )rR   r%   rT   r    r!   r   r3   s          r'   "test_dummy_regressor_sample_weightr   k  s   9((.@(AAL
	A)$$A %%i00M
&
)
)
)
-
-aM
B
BC=BJq-@@@@@@@
(
+
+
+
/
/1m
D
DC=0M4HHHHHH
*t
<
<
<
@
@A}
U
UC=0M4HHHHHHHHr)   c                  *   t          j        dggdggdggg          } t          j        g d          }t          j        g d          }t                      }|                    | |           |                    |           }t          ||           d S )Nfoobarbazr,   r,   r,   )r   r/   r   r0   r1   r	   )r    r!   r   clsr4   s        r'    test_dummy_regressor_on_3D_arrayr  |  s    
E7)ugY%	233A
A)))$$J


CGGAqMMM[[^^Fvz*****r)   c                  @   t          j        dggdggdggg          } g d}g d}dgdgdgg}t          d          }|                    | |           |                    |           }|                    |           }t          ||           t          ||           d S )Nr   r   r   r   r   rl   r   )r   r/   r   r0   r1   r   r	   )r    r!   r   y_proba_expectedr   r4   y_pred_probas          r'   !test_dummy_classifier_on_3D_arrayr    s    
E7)ugY%	233A		AJaS1#
<
0
0
0CGGAqMMM[[^^F$$Q''Lvz***|%566666r)   c                  8   dggdz  } t          j        g d          }t          j        g d          }t                      }|                    | |           |                    | d          }t          |          dk    sJ t          |d         |           d S )	Nr   r-   r   )r   r   r   T)
return_stdr,   r   )r   r/   r   r0   r1   r   r	   )r    r!   y_std_expectedr   y_pred_lists        r'   test_dummy_regressor_return_stdr
    s    
	A
AXiii((N


CGGAqMMM++aD+11K{q    {1~~66666r)   )r   r   r   r,   g      ?r8   c                     t                      }|                    d |            |                    d |          dk    sJ d S )Nr   )r   r0   r   )r!   r?   r   s      r'   test_regressor_score_with_Noner    sH     

CGGD!99T6""c))))))r)   )r   r   r   r   c                 >   g d}dggdz  }t          | dd          }|                    ||           |                    |          }dggdz  }t          | dd          }|                    ||           |                    |          }t          ||           d S )Nr   r   r8   gffffff?)r[   r   r   r   )r   r0   r1   r	   )r[   r!   r   reg1r   r   reg2r   s           r'   *test_regressor_prediction_independent_of_Xr    s    A#B8a#FFFDHHROOO<<##L#B8a#FFFDHHROOO<<##L|\22222r)   c                     g d}t          j        d          }t          | dd          }|                    ||                              |          }|j        t           j        k    sJ d S )Nr   r8   r   r   )r   ri   r   r0   r   dtypefloat64)r[   r!   r    modelprobass        r'   test_dtype_of_classifier_probasr    sh     	A
AXAJJJEYYq!__**1--F<2:%%%%%%r)   )rB   )Fnumpyr   rK   scipy.sparsesparser   sklearn.baser   sklearn.dummyr   r   sklearn.exceptionsr   sklearn.utils._testingr   r   r	   r
   sklearn.utils.fixesr   sklearn.utils.statsr   r(   r5   r9   rA   rU   r_   rf   rj   rw   r{   r~   r   r   markparametrizer/   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   	transposer   r   r   r   r   r   r   r  r  r
  r  r  r  r   r)   r'   <module>r#     sr                    9 9 9 9 9 9 9 9 - - - - - -            / . . . . . 4 4 4 4 4 4 B B B0# # #"# # #U U U
* * * *"  (A A A      "$ $ $  &$ $ $  $6 6 6 	|||$BHq!fq!fq!fq!f566BHq!fq!fq!fq!f566	
	 	* *	 	* OOO 3 3 3&> > >  *  @ @ @  *J J J.  B	 	 	  4 4 4$) ) ),' ' '    $ $ $"$ $ $ 	*,KL	
QF+-JKBL,,,566O)	
 LL'=	
 BL,,,566!Y(B	
&	 	 	+   : ; :H H H .99  :9 .994 4 :94$ .996 6 :96& .999 9 :99I I I I"+ + +
7 
7 
7
7 
7 
7 	vz"	Aq6Aq6Aq6Aq62	3	3tTl^a5GH * * * %O%O%OPP3 3 QP3 OOO & & & & &r)   