ó
    ¦ñ:icU  ã                   ó”  • S SK rS SKrS SKJr  S SKJr  S SKJ	r	J
r
  S SKJr  S SKJrJrJrJr  S SKJr  S SKJr  \S 5       rS	 rS
 rS rSLS jrS rS rS rS rS rS r S r!S r"\RF                  RI                  S/ SQ/ SQ4\RJ                  " SS/SS/SS/SS//5      \RJ                  " SS/SS/SS/SS//5      4/5      S 5       r&\RF                  RI                  S/ SQ5      S 5       r'S r(S r)S  r*S! r+S" r,S# r-S$ r.S% r/S& r0S' r1S( r2S) r3S* r4S+ r5S, r6S- r7\RF                  RI                  S./ S/QS0S 0S14/ S/QS2SS /0S34\Rp                  " / S/Q/ S/Q/5      S2S0S34/ S/QS2S40S54\Rp                  " / S/Q/ S/Q/5      S2SS6/0S74// S8QS99S: 5       r9S; r:\RF                  RI                  S<\5      S= 5       r;\RF                  RI                  S<\5      S> 5       r<\RF                  RI                  S<\5      S? 5       r=\RF                  RI                  S<\5      S@ 5       r>SLSA jr?SB r@SC rASD rB\RF                  RI                  S/ SEQSF/SG-  4\RJ                  " SS/SS/SS/SS//5      SFSF//SG-  4/5      SH 5       rC\RF                  RI                  S/ SIQ5      SJ 5       rD\RF                  RI                  S/ SQ5      SK 5       rEg)Mé    N)Úclone)ÚDummyClassifierÚDummyRegressor)ÚNotFittedError)Úassert_almost_equalÚassert_array_almost_equalÚassert_array_equalÚignore_warnings)ÚCSC_CONTAINERS)Ú_weighted_percentilec           	      óŽ  • U R                  U5      nU R                  U5      n[        R                  " U5      nUR                  S:X  a  [        R
                  " US5      nUR                  S   n[        U5      nUS:X  a  U/nU/n[        U5       H±  nX7   R                  S   U:X  d   eX7   R                  S   [        [        R                  " US S 2U4   5      5      :X  d   e[        X7   R                  SS9[        R                  " [        U5      5      5        [        [        R                  " X7   5      XG   5        M³     g )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           Ú[/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/sklearn/tests/test_dummy.pyÚ_check_predict_probar(      s  € à×Ñ˜aÓ €Eà×%Ñ% aÓ(€Iä
�Š�aÓ€AØ‡v�v�ƒ{Ü�JŠJ�q˜'Ó"ˆà—‘˜‘
€IÜ�A“€Ià�Aƒ~Ø�ˆØ�Kˆ	ä�9ÖˆØ‰x�~‰~˜aÑ  IÓ-Ð-Ð-Ø‰x�~‰~˜aÑ ¤C¬¯	ª	°!²A°q°D±'Ó(:Ó$;Ó;Ð;Ð;Ü! %¡(§,¡,°A ,Ð"6¼¿ºÄÀAÃ»ÔHä!¤"§&¢&¨©Ó"2°I±LÖAò ó    c                 óØ  • [         R                  " S/S/S/S//5      n[         R                  " / SQ5      n[        U 5      nUR                  X5        UR	                  U5      nUR
                  UR
                  :X  d   e[         R                  " SS/SS/SS/SS//5      n[        U 5      nUR                  X5        UR	                  U5      nUR
                  UR
                  :X  d   eg )Nr   ©r   é   r   r   r   r,   é   ©r   Úarrayr   ÚfitÚpredictr   ©r   r    r!   ÚestÚy_preds        r'   Ú_check_behavior_2dr5   +   sÆ   € ä
�Š�1�#˜�s˜Q˜C ! Ð%Ó&€AÜ
�Š’Ó€AÜ
�‹*€CØ‡G�GˆA„MØ�[‰[˜‹^€FØ�7‰7�f—l‘lÓ"Ð"Ð"ô 	�Š�1�a�&˜1˜a˜& 1 a &¨1¨a¨&Ð1Ó2€AÜ
�‹*€CØ‡G�GˆA„MØ�[‰[˜‹^€FØ�7‰7�f—l‘lÓ"Ð"Ñ"r)   c                 ó  • [         R                  " S/S/S/S//5      n[         R                  " / SQ/ SQ/ SQ/ SQ/5      n[        U 5      nUR                  X5        UR	                  U5      nUR
                  UR
                  :X  d   eg )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   <   so   € ä
�Š�1�#˜�s˜Q˜C ! Ð%Ó&€AÜ
�Š’/¢?²OÂ_ÐUÓV€AÜ
�‹*€CØ‡G�GˆA„MØ�[‰[˜‹^€FØ�7‰7�f—l‘lÓ"Ð"Ñ"r)   c                 óÀ   • [        [        R                  " XR                  S   S45      U5        [        [        R                  " XR                  S   S45      U5        g )Nr   r   )r   r   Útiler   )Ú	statisticÚy_learnÚy_pred_learnÚy_testÚy_pred_tests        r'   Ú_check_equality_regressorrA   F   sC   € ÜœbŸgšg i·-±-ÀÑ2BÀAÐ1FÓGÈÔVÜœbŸgšg i·,±,¸q±/À1Ð1EÓFÈÕTr)   c                 ó´  • [         R                  " S5      n[        R                  R	                  U S9nUR                  S//U-  S/S9nUR                  U5      n[        5       R                  XE5      n[        US5      (       d   e[        US5      (       d   e[        5       R                  XE5      n[        US5      (       d   e[        US5      (       d   eg )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_rT   K   sÍ   € Ü	×	Ò	˜XÓ	&€Bä—9‘9×(Ñ(Ð.@Ð(ÐA€Là
�‰�q�c�U˜YÑ&°°ˆÐ>€AØ×Ñ˜)Ó$€Aä
Ó
×
Ñ
˜qÓ
$€CÜ�3Ð+×,Ñ,Ð,Ð,Ü�3Ð(×)Ñ)Ð)Ð)ä
Ó
×
Ñ
 Ó
%€CÜ�3Ð+×,Ñ,Ð,Ð,Ü�3Ð(×)Ñ)Ð)Ñ)r)   c            	      óæ  • S/S/S/S//n / SQnS HÝ  n[        USS9nUR                  X5        [        UR                  U 5      [        R
                  " [        U 5      5      5        [        X0U5        US:X  a:  [        UR                  U S   /5      UR                  R                  S5      5        M¢  [        UR                  U S   /5      UR                  R                  S5      S:„  5        Mß     g )Nr   r+   ©Úmost_frequentÚprior©ÚstrategyrS   rX   ©r   r   ç      à?)r   r0   r	   r1   r   r   r   r(   r   r   Úclass_prior_r   )r    r!   rZ   r   s       r'   Ú%test_most_frequent_and_prior_strategyr^   \   sÓ   € Ø
ˆˆqˆc�A�3˜˜Ð€AÚ€Aã.ˆÜ x¸aÑ@ˆØ�‰�ŒÜ˜3Ÿ;™; q›>¬2¯7ª7´3°q³6«?Ô;Ü˜S QÔ'à�wÓÜ%Ø×!Ñ! 1 Q¡4 &Ó)¨3×+;Ñ+;×+CÑ+CÀGÓ+Löô &Ø×!Ñ! 1 Q¡4 &Ó)¨3×+;Ñ+;×+CÑ+CÀGÓ+LÈsÑ+Röò /r)   c                  ó  • S/S/S/S//n / SQnS/S/S/S//nS Hc  n[        USS9n[        USS9nUR                  X5        UR                  X5        [        UR                  U 5      UR                  U 5      5        Me     g )Nr   r+   r   r,   rV   rY   ©r   r0   r	   r1   )r    Úy_1dÚy_2drZ   Úclf_1dÚclf_2ds         r'   Ú6test_most_frequent_and_prior_strategy_with_2d_column_yre   p   s‹   € ð ˆˆqˆc�A�3˜˜Ð€AÚ€DØˆC�!��q�c˜A˜3Ð€Dã.ˆÜ ¨(ÀÑCˆÜ ¨(ÀÑCˆà�
‰
�1ÔØ�
‰
�1ÔÜ˜6Ÿ>™>¨!Ó,¨f¯n©n¸QÓ.?Ö@ò /r)   c                  ó¢  • S/S/S/S//n [         R                  " SS/SS/SS/SS//5      n[        U 5      nS H’  n[        USS9nUR	                  X5        [        UR                  U 5      [         R                  " [         R                  " US45      [         R                  " US45      /5      5        [        X@U5        [        U5        M”     g )Nr   r   r,   r-   )rX   rW   rY   )r   r/   r   r   r0   r	   r1   Úhstackr   Úzerosr(   r5   )r    r!   r%   rZ   r   s        r'   Ú1test_most_frequent_and_prior_strategy_multioutputri   €   s»   € Ø
ˆˆqˆc�A�3˜˜Ð€AÜ
�Š�1�a�&˜1˜a˜& 1 a &¨1¨a¨&Ð1Ó2€Aä�A“€Iã.ˆÜ x¸aÑ@ˆØ�‰�ŒÜØ�K‰K˜‹NÜ�IŠI”r—w’w 	¨1˜~Ó.´·²¸)ÀQ¸Ó0HÐIÓJô	
ô 	˜S QÔ'Ü˜3Öò /r)   c                 ó&  • S//S-  n/ SQn[        SU S9nUR                  X5        S//S-  nUR                  U5      n[        R                  " U5      [        [        U5      5      -  n[        US   SSS	9  [        US
   SSS	9  [        X1U5        g )Nr   r7   )r   r,   r   r   r,   Ú
stratifiedrY   éô  r   ç333333ã?©Údecimalr,   çš™™™™™Ù?©	r   r0   r1   r   ÚbincountÚfloatr   r   r(   ©rQ   r    r!   r   r4   Úps         r'   Útest_stratified_strategyrv   ‘   s‹   € Ø
ˆˆ�‰	€AÚ€AÜ
 <Ð>PÑ
Q€CØ‡G�GˆA„Mà
ˆˆ�‰€AØ�[‰[˜‹^€FÜ
�Š�FÓœe¤C¨£F›mÑ+€AÜ˜˜!™˜g¨qÒ1Ü˜˜!™˜g¨qÒ1Ü˜ Õ#r)   c                 óÊ  • S//S-  n[         R                  " SS/SS/SS/SS/SS//5      n[        SU S9nUR                  X5        S//S-  nUR	                  U5      n[        UR                  S   5       H]  n[         R                  " US S 2U4   5      [        [        U5      5      -  n[        US   SSS	9  [        US   S
SS	9  [        X1U5        M_     [        U5        g )Nr   r7   r,   r   rk   rY   rl   rm   rn   rp   ©r   r/   r   r0   r1   r   r   rr   rs   r   r   r(   r5   ©rQ   r    r!   r   r4   r&   ru   s          r'   Ú$test_stratified_strategy_multioutputrz   Ÿ   sÜ   € Ø
ˆˆ�‰	€AÜ
�Š�1�a�&˜1˜a˜& 1 a &¨1¨a¨&°1°a°&Ð9Ó:€Aä
 <Ð>PÑ
Q€CØ‡G�GˆA„Mà
ˆˆ�‰€AØ�[‰[˜‹^€Fä�1—7‘7˜1‘:ÖˆÜ�KŠK˜šq !˜t™Ó%¬¬c°!«f«Ñ5ˆÜ˜A˜a™D '°1Ò5Ü˜A˜a™D '°1Ò5Ü˜S QÖ'ñ	 ô �sÕr)   c                 ó&  • S//S-  n/ SQn[        SU S9nUR                  X5        S//S-  nUR                  U5      n[        R                  " U5      [        [        U5      5      -  n[        US   SSS	9  [        US
   SSS	9  [        X1U5        g )Nr   r8   r+   ÚuniformrY   rl   r   r\   rn   r,   rq   rt   s         r'   Útest_uniform_strategyr}   ²   s‹   € Ø
ˆˆ�‰	€AÚ€AÜ
 9Ð;MÑ
N€CØ‡G�GˆA„Mà
ˆˆ�‰€AØ�[‰[˜‹^€FÜ
�Š�FÓœe¤C¨£F›mÑ+€AÜ˜˜!™˜c¨1Ò-Ü˜˜!™˜c¨1Ò-Ü˜ Õ#r)   c                 óÄ  • S//S-  n[         R                  " SS/SS/SS/SS//5      n[        SU S9nUR                  X5        S//S-  nUR	                  U5      n[        UR                  S   5       H]  n[         R                  " US S 2U4   5      [        [        U5      5      -  n[        US   SSS	9  [        US   SSS	9  [        X1U5        M_     [        U5        g )
Nr   r8   r,   r   r|   rY   rl   r\   rn   rx   ry   s          r'   Ú!test_uniform_strategy_multioutputr   À   sÖ   € Ø
ˆˆ�‰	€AÜ
�Š�1�a�&˜1˜a˜& 1 a &¨1¨a¨&Ð1Ó2€AÜ
 9Ð;MÑ
N€CØ‡G�GˆA„Mà
ˆˆ�‰€AØ�[‰[˜‹^€Fä�1—7‘7˜1‘:ÖˆÜ�KŠK˜šq !˜t™Ó%¬¬c°!«f«Ñ5ˆÜ˜A˜a™D #¨qÒ1Ü˜A˜a™D #¨qÒ1Ü˜S QÖ'ñ	 ô �sÕr)   c                  óŒ   • S//S-  n / SQn[        SS9nUR                  X5        [        UR                  U 5      S/S-  5        g )Nr   r7   )Úparisr�   ÚtokyoÚ	amsterdamÚberlinrW   ©rZ   r�   r`   ©r    r!   r   s      r'   Útest_string_labelsr‡   Ò   sA   € Ø
ˆˆ�‰	€AÚ:€AÜ
 ?Ñ
3€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¨ y°1¡}Õ5r)   zy,y_test)r,   r   r   r   )r,   r,   r   r   r,   r   c                 ój   • [        SS9nUR                  S U 5        UR                  S U5      S:X  d   eg )NrW   r…   r\   )r   r0   Úscore)r!   r?   r   s      r'   Útest_classifier_score_with_NonerŠ   Ú   s5   € ô  ?Ñ
3€CØ‡G�GˆD�!ÔØ�9‰9�T˜6Ó" cÓ)Ð)Ñ)r)   rZ   )rk   rW   rX   r|   Úconstantc                 óî   • / SQnS//S-  n[        XSS9nUR                  X25        UR                  U5      nS//S-  n[        XSS9nUR                  Xb5        UR                  U5      n[        XX5        g )N©r   r,   r   r   r   r8   ©rZ   rS   r‹   r   )r   r0   r1   r	   )	rZ   rQ   r!   ÚX1Úclf1Úpredictions1ÚX2Úclf2Úpredictions2s	            r'   Ú+test_classifier_prediction_independent_of_Xr•   ê   s   € ò 	€AØˆ#ˆ�‰€BÜØÀQñ€Dð 	‡H�HˆR„OØ—<‘< Ó#€Làˆ#ˆ�‰€BÜØÀQñ€Dð 	‡H�HˆR„OØ—<‘< Ó#€Lä�|Õ2r)   c                 ó  • [         R                  R                  U S9nS//S-  nUR                  S5      n[	        5       nUR                  X#5        [        UR                  U5      [         R                  " U5      /[        U5      -  5        g )NrD   r   r8   )
r   rL   rM   Úrandnr   r0   r	   r1   Úmeanr   ©rQ   rS   r    r!   Úregs        r'   Útest_mean_strategy_regressorr›      so   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là
ˆˆ�‰	€AØ×Ñ˜1Ó€Aä
Ó
€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯ª°«
 |´c¸!³fÑ'<Õ=r)   c                 óÀ  • [         R                  R                  U S9nUR                  SS5      nUR                  SS5      n[         R                  " USS9R                  S5      nUR                  SS5      nUR                  SS5      n[        5       nUR                  X#5        UR                  U5      nUR                  U5      n	[        XCX†U	5        [        U5        g )NrD   é
   r7   r   r   r[   é   )r   rL   rM   r—   r˜   r   r   r0   r1   rA   r5   )
rQ   rS   ÚX_learnr=   r˜   ÚX_testr?   r3   r>   r@   s
             r'   Ú(test_mean_strategy_multioutput_regressorr¡     sÃ   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là× Ñ   RÓ(€GØ× Ñ   QÓ'€Gä�7Š7�7 Ñ#×+Ñ+¨GÓ4€Dà×Ñ  BÓ'€FØ×Ñ  AÓ&€Fô Ó
€CØ‡G�GˆGÔØ—;‘;˜wÓ'€LØ—+‘+˜fÓ%€Kä˜d¨\À;ÔOÜ�sÕr)   c                  ó¢   • [        5       n [        R                  " [        5         U R	                  / 5        S S S 5        g ! , (       d  f       g = f)N)r   rJ   Úraisesr   r1   )rš   s    r'   Útest_regressor_exceptionsr¤      s-   € Ü
Ó
€CÜ	�Š”~Õ	&Ø�‰�BŒ÷ 
'×	&Ö	&ús   ¥A Á 
Ac                 ó  • [         R                  R                  U S9nS//S-  nUR                  S5      n[	        SS9nUR                  X#5        [        UR                  U5      [         R                  " U5      /[        U5      -  5        g )NrD   r   r7   Úmedianr…   )
r   rL   rM   r—   r   r0   r	   r1   r¦   r   r™   s        r'   Útest_median_strategy_regressorr§   &  sq   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là
ˆˆ�‰	€AØ×Ñ˜1Ó€Aä
 (Ñ
+€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯	ª	°!« ~¼¸A»Ñ'>Õ?r)   c                 ó¾  • [         R                  R                  U S9nUR                  SS5      nUR                  SS5      n[         R                  " USS9R                  S5      nUR                  SS5      nUR                  SS5      n[        SS	9nUR                  X#5        UR                  U5      nUR                  U5      n	[        XCX†U	5        [        U5        g )
NrD   r�   r7   r   r   r[   rž   r¦   r…   )r   rL   rM   r—   r¦   r   r   r0   r1   rA   r5   )
rQ   rS   rŸ   r=   r¦   r    r?   r3   r>   r@   s
             r'   Ú*test_median_strategy_multioutput_regressorr©   1  sÅ   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là× Ñ   RÓ(€GØ× Ñ   QÓ'€Gä�YŠY�w QÑ'×/Ñ/°Ó8€Fà×Ñ  BÓ'€FØ×Ñ  AÓ&€Fô  (Ñ
+€CØ‡G�GˆGÔØ—;‘;˜wÓ'€LØ—+‘+˜fÓ%€Kä˜f¨|À[ÔQÜ�sÕr)   c                 ó$  • [         R                  R                  U S9nS//S-  nUR                  S5      n[	        SSS9nUR                  X#5        [        UR                  U5      [         R                  " U5      /[        U5      -  5        [	        SSS9nUR                  X#5        [        UR                  U5      [         R                  " U5      /[        U5      -  5        [	        SSS9nUR                  X#5        [        UR                  U5      [         R                  " U5      /[        U5      -  5        [	        SSS9nUR                  X#5        [        UR                  U5      [         R                  " US	S
9/[        U5      -  5        g )NrD   r   r7   Úquantiler\   ©rZ   r«   r   g333333Ó?é   )Úq)r   rL   rM   r—   r   r0   r	   r1   r¦   r   ÚminÚmaxÚ
percentiler™   s        r'   Ú test_quantile_strategy_regressorr²   F  s+  € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là
ˆˆ�‰	€AØ×Ñ˜1Ó€Aä
 *°sÑ
;€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯	ª	°!« ~¼¸A»Ñ'>Ô?ä
 *°qÑ
9€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯ª¨q«	 {´S¸³VÑ';Ô<ä
 *°qÑ
9€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯ª¨q«	 {´S¸³VÑ';Ô<ä
 *°sÑ
;€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¬¯ª°a¸2Ñ(>Ð'?Ä#ÀaÃ&Ñ'HÕIr)   c                 ó´  • [         R                  R                  U S9nUR                  SS5      nUR                  SS5      n[         R                  " USS9R                  S5      n[         R                  " USSS9R                  S5      nUR                  S	S5      nUR                  S	S5      n[        S
SS9nUR                  X#5        UR                  U5      n	UR                  U5      n
[        XCX—U
5        [        U5        [        S
SS9nUR                  X#5        UR                  U5      n	UR                  U5      n
[        XSX—U
5        [        U5        g )NrD   r�   r7   r   r   r[   éP   )r   r®   rž   r«   r\   r¬   gš™™™™™é?)r   rL   rM   r—   r¦   r   r±   r   r0   r1   rA   r5   )rQ   rS   rŸ   r=   r¦   Úquantile_valuesr    r?   r3   r>   r@   s              r'   Ú,test_quantile_strategy_multioutput_regressorr¶   ]  s3  € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là× Ñ   RÓ(€GØ× Ñ   QÓ'€Gä�YŠY�w QÑ'×/Ñ/°Ó8€FÜ—m’m G°!°rÑ:×BÑBÀ7ÓK€Oà×Ñ  BÓ'€FØ×Ñ  AÓ&€Fô  *°sÑ
;€CØ‡G�GˆGÔØ—;‘;˜wÓ'€LØ—+‘+˜fÓ%€Kä˜f¨|À[ÔQÜ�sÔô  *°sÑ
;€CØ‡G�GˆGÔØ—;‘;˜wÓ'€LØ—+‘+˜fÓ%€KäØ ,¸ôô �sÕr)   c                  ó¾   • S//S-  n S/S-  n[        SS S9nSn[        R                  " [        US9   UR	                  X5        S S S 5        g ! , (       d  f       g = f)Nr   r7   r«   r¬   zIWhen using `strategy='quantile', you have to specify the desired quantile©Úmatch)r   rJ   r£   Ú
ValueErrorr0   ©r    r!   r3   Úerr_msgs       r'   Útest_quantile_invalidr½   ~  sU   € Ø
ˆˆ�‰	€AØ	
ˆˆa‰€Aä
 *°tÑ
<€CàSð ô 
�Š”z¨Ó	1Ø�‰�Œ÷ 
2×	1Ö	1ús   ³AÁ
Ac                  ó¤   • [        SSS9n [        R                  " [        5         U R	                  / / 5        S S S 5        g ! , (       d  f       g = f)Nr«   rp   r¬   )r   rJ   r£   Ú
IndexErrorr0   )r3   s    r'   Ú"test_quantile_strategy_empty_trainrÀ   Š  s3   € Ü
 *°sÑ
;€CÜ	�Š”zÕ	"Ø�‰��BŒ÷ 
#×	"Ö	"ús   ¥AÁ
Ac                 óÒ  • [         R                  R                  U S9nS//S-  nUR                  S5      n[	        SS/S9nUR                  X#5        [        UR                  U5      S/[        U5      -  5        [	        SSS9nUR                  X#5        [        UR                  U5      S/[        U5      -  5        [        UR                  [         R                  5      (       a   eg )NrD   r   r7   r‹   é+   ©rZ   r‹   )r   rL   rM   r—   r   r0   r	   r1   r   Ú
isinstancer‹   Úndarrayr™   s        r'   Ú test_constant_strategy_regressorrÆ   �  s¿   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là
ˆˆ�‰	€AØ×Ñ˜1Ó€Aä
 *¸°tÑ
<€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¨ t¬c°!«f¡}Ô5ä
 *°rÑ
:€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¨ t¬c°!«f¡}Ô5ô ˜#Ÿ,™,¬¯
©
×3Ñ3Ð3Ð3Ð3r)   c                 óš  • [         R                  R                  U S9nUR                  SS5      nUR                  SS5      nUR                  S5      nUR                  SS5      nUR                  SS5      n[	        SUS9nUR                  X#5        UR                  U5      nUR                  U5      n	[        XCX†U	5        [        U5        g )NrD   r�   r7   rž   r‹   rÃ   )	r   rL   rM   r—   r   r0   r1   rA   r9   )
rQ   rS   rŸ   r=   Ú	constantsr    r?   r3   r>   r@   s
             r'   Ú,test_constant_strategy_multioutput_regressorrÉ   ¢  s¾   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là× Ñ   RÓ(€GØ× Ñ   QÓ'€Gð ×"Ñ" 1Ó%€Ià×Ñ  BÓ'€FØ×Ñ  AÓ&€Fô  *°yÑ
A€CØ‡G�GˆGÔØ—;‘;˜wÓ'€LØ—+‘+˜fÓ%€Kä˜i°,ÈÔTÜ# CÕ(r)   c                  óš   • S//S-  n / SQn[        SS9nUR                  X5        UR                  [        R                  " U5      :X  d   eg )Nr   r7   ©r   r,   r8   é   é   r˜   r…   )r   r0   Ú	constant_r   r˜   )r    r!   r3   s      r'   Útest_y_mean_attribute_regressorrÏ   ¸  sC   € Ø
ˆˆ�‰	€AÚ€Aä
 &Ñ
)€CØ‡G�GˆA„Mà�=‰=œBŸGšG A›JÓ&Ð&Ñ&r)   c                  ó¸   • S//S-  n / SQn[        SS9nSn[        R                  " [        US9   UR	                  X5        S S S 5        g ! , (       d  f       g = f)Nr   r7   rË   r‹   r…   z)Constant target value has to be specifiedr¸   )r   rJ   r£   Ú	TypeErrorr0   r»   s       r'   Ú&test_constants_not_specified_regressorrÒ   Â  sH   € Ø
ˆˆ�‰	€AÚ€Aä
 *Ñ
-€CØ9€GÜ	�Š”y¨Ó	0Ø�‰�Œ÷ 
1×	0Ö	0ús   °AÁ
Ac                 ó*  • [         R                  R                  U S9nUR                  SS5      nUR                  SS5      n[	        S/ SQS9nSn[
        R                  " [        US9   UR                  X#5        S S S 5        g ! , (       d  f       g = f)	NrD   r�   r7   r‹   )r   r,   r-   r8   rÃ   z1Constant target value should have shape \(5, 1\).r¸   )	r   rL   rM   r—   r   rJ   r£   rº   r0   )rQ   rS   r    r!   r3   r¼   s         r'   Ú(test_constant_size_multioutput_regressorrÔ   Ì  st   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€LØ×Ñ˜2˜rÓ"€AØ×Ñ˜2˜qÓ!€Aä
 *²|Ñ
D€CØB€GÜ	�Š”z¨Ó	1Ø�‰�Œ÷ 
2×	1Ö	1ús   Á)BÂ
Bc                  ó²  • S/S/S/S//n / SQn[        SSSS9nUR                  X5        [        UR                  U 5      [        R
                  " [        U 5      5      5        [        X U5        S/S/S/S//n / SQn[        SSSS9nUR                  X5        [        UR                  U 5      [        R                  " S/S-  5      5        [        X U5        g )	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½   € Ø
ˆˆqˆc�A�3˜˜Ð€AÚ€Aä
 :¸AÈÑ
J€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¤r§w¢w¬s°1«v£Ô7Ü˜ Ô#à
ˆˆqˆc�A�3˜˜Ð€AÚ$€AÜ
 :¸AÈÑ
N€CØ‡G�GˆA„MÜ�s—{‘{ 1“~¤r§x¢x°°¸!±Ó'<Ô=Ü˜ Õ#r)   c            
      ó€  • S/S/S/S//n [         R                  " SS/SS/SS/SS//5      n[        U 5      n[        SSSS/S9nUR	                  X5        [        UR                  U 5      [         R                  " [         R                  " US45      [         R                  " US45      /5      5        [        X0U5        g )Nr   r,   r-   r   r‹   rŽ   )r   r/   r   r   r0   r	   r1   rg   r   rh   r(   )r    r!   r%   r   s       r'   Ú"test_constant_strategy_multioutputrÛ   è  s¯   € Ø
ˆˆqˆc�A�3˜˜Ð€AÜ
�Š�1�a�&˜1˜a˜& 1 a &¨1¨a¨&Ð1Ó2€Aä�A“€Iä
 :¸AÈÈAÈÑ
O€CØ‡G�GˆA„MÜØ�‰�A‹œŸ	š	¤2§7¢7¨I°q¨>Ó#:¼B¿HºHÀiÐQRÀ^Ó<TÐ"UÓVôô ˜ Õ#r)   zy, params, err_msgrÖ   rS   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                 óº   • S/S/S/S//n[        SSS0UD6n[        R                  " [        US9   UR	                  X05        S S S 5        g ! , (       d  f       g = f)Nr   rZ   r‹   r¸   © )r   rJ   r£   rº   r0   )r!   Úparamsr¼   r    r   s        r'   Ú!test_constant_strategy_exceptionsrá   ö  sQ   € ð< ˆˆqˆc�A�3˜˜Ð€Aä
Ñ
8 :Ð
8°Ñ
8€CÜ	�Š”z¨Ó	1Ø�‰�Œ÷ 
2×	1Ö	1ús   ±AÁ
Ac                  ó†   • S/S/S//n / SQn/ SQn[        SS9R                  XU5      n[        UR                  SS/5        g )	Nr   r   )r   r   r   )çš™™™™™¹?ç      ð?rã   rk   r…   gVUUUUUÅ?g«ªªªªªê?)r   r0   r   r]   )r    r!   Úsample_weightr   s       r'   Ú!test_classification_sample_weightræ     sI   € Ø
ˆˆqˆc�A�3ˆ€AÚ€AÚ#€Mä
 <Ñ
0×
4Ñ
4°Q¸=Ó
I€CÜ˜c×.Ñ.°¸IÐ0FÕGr)   Úcsc_containerc           
      óÎ  • S//S-  nU " [         R                  " SS/SS/SS/SS/SS//5      5      n[        U5      n[        SSSS/S9nUR	                  X5        UR                  U5      n[        R                  " U5      (       d   e[        UR                  5       [         R                  " [         R                  " US45      [         R                  " US45      /5      5        g )Nr   r7   r   r8   r‹   rŽ   )r   r/   r   r   r0   r1   ÚspÚissparser	   Útoarrayrg   r   rh   )rç   r    r!   r%   r   r4   s         r'   Ú$test_constant_strategy_sparse_targetrì   $  sÆ   € à
ˆˆ�‰	€AÙ”b—h’h  A ¨¨A¨°°A°¸¸A¸ÀÀAÀÐGÓHÓI€Aä�A“€Iä
 :¸AÈÈAÈÑ
O€CØ‡G�GˆA„MØ�[‰[˜‹^€FÜ�;Š;�v×ÑÐÐÜØ�‰Óœ"Ÿ)š)¤R§W¢W¨i¸¨^Ó%<¼b¿hºhÈ	ÐSTÀ~Ó>VÐ$WÓXõr)   c           
      ó*  • S//S-  nU" [         R                  " SS/SS/SS/SS/SS//5      5      n[        SU S9n[        R                  " [
        SS	9   UR                  X#5        S S S 5        S//S
-  nUR                  U5      n[        UR                  S   5       H_  n[         R                  " US S 2U4   5      [        [        U5      5      -  n[        US   SSS9  [        US   SSS9  [        US   SSS9  Ma     g ! , (       d  f       N¢= f)Nr   r7   r,   r   r8   r|   rY   z*the uniform strategy would not save memoryr¸   rl   gUUUUUUÕ?rn   )r   r/   r   rJ   ÚwarnsÚUserWarningr0   r1   r   r   rr   rs   r   r   ©rQ   rç   r    r!   r   r4   r&   ru   s           r'   Ú+test_uniform_strategy_sparse_target_warningrñ   4  sþ   € à
ˆˆ�‰	€AÙ”b—h’h  A ¨¨A¨°°A°¸¸A¸ÀÀAÀÐGÓHÓI€Aä
 9Ð;MÑ
N€CÜ	�Š”kÐ)UÓ	VØ�‰�Œ÷ 
Wð ˆˆ�‰€AØ�[‰[˜‹^€Fä�1—7‘7˜1‘:ÖˆÜ�KŠK˜šq !˜t™Ó%¬¬c°!«f«Ñ5ˆÜ˜A˜a™D %°Ò3Ü˜A˜a™D %°Ò3Ü˜A˜a™D %°Ô3ò	 ÷ 
WÕ	Vús   ÁDÄ
Dc           
      ó  • S//S-  nU" [         R                  " SS/SS/SS/SS/SS//5      5      n[        SU S9nUR                  X#5        S//S-  nUR	                  U5      n[
        R                  " U5      (       d   eUR                  5       n[        UR                  S   5       H_  n[         R                  " US S 2U4   5      [        [        U5      5      -  n[        US   SSS	9  [        US   S
SS	9  [        US   S
SS	9  Ma     g )Nr   r7   r8   r   rk   rY   rl   rm   rn   gš™™™™™É?)r   r/   r   r0   r1   ré   rê   rë   r   r   rr   rs   r   r   rð   s           r'   Ú&test_stratified_strategy_sparse_targetró   G  sþ   € à
ˆˆ�‰	€AÙ”b—h’h  A ¨¨A¨°°A°¸¸A¸ÀÀAÀÐGÓHÓI€Aä
 <Ð>PÑ
Q€CØ‡G�GˆA„Mà
ˆˆ�‰€AØ�[‰[˜‹^€FÜ�;Š;�v×ÑÐÐØ�^‰^Ó€Fä�1—7‘7˜1‘:ÖˆÜ�KŠK˜šq !˜t™Ó%¬¬c°!«f«Ñ5ˆÜ˜A˜a™D '°1Ò5Ü˜A˜a™D '°1Ò5Ü˜A˜a™D '°1Ô5ò	 r)   c           
      óÞ  • S//S-  nU " [         R                  " SS/SS/SS/SS/SS//5      5      n[        U5      n[         R                  " [         R                  " US45      [         R
                  " US45      /5      nS Hf  n[        USS9nUR                  X5        UR                  U5      n[        R                  " U5      (       d   e[        UR                  5       U5        Mh     g )Nr   r7   r   r-   r8   rV   rY   )r   r/   r   rg   r   rh   r   r0   r1   ré   rê   r	   rë   )rç   r    r!   r%   Ú
y_expectedrZ   r   r4   s           r'   Ú3test_most_frequent_and_prior_strategy_sparse_targetrö   [  sË   € à
ˆˆ�‰	€AÙ”b—h’h  A ¨¨A¨°°A°¸¸A¸ÀÀAÀÐGÓHÓI€Aä�A“€IÜ—’œBŸGšG Y° NÓ3´R·X²X¸yÈ!¸nÓ5MÐNÓO€JÛ.ˆÜ x¸aÑ@ˆØ�‰�Œà—‘˜Q“ˆÜ�{Š{˜6×"Ñ"Ð"Ð"Ü˜6Ÿ>™>Ó+¨ZÖ8ò /r)   c                 óà  • [         R                  R                  U S9nS//U-  nUR                  U5      nUR                  U5      n[	        SS9R                  X4U5      nUR                  [         R                  " XES9:X  d   e[	        SS9R                  X4U5      nUR                  [        XES5      :X  d   e[	        SS	S
9R                  X4U5      nUR                  [        XES5      :X  d   eg )NrD   r   r˜   r…   )Úweightsr¦   g      I@r«   gffffffî?r¬   g     ÀW@)	r   rL   rM   rO   r   r0   rÎ   Úaverager   )rQ   r%   rS   r    r!   rå   r3   s          r'   Ú"test_dummy_regressor_sample_weightrú   k  sÞ   € Ü—9‘9×(Ñ(Ð.@Ð(ÐA€Là
ˆˆ�	Ñ€AØ×Ñ˜)Ó$€AØ ×%Ñ% iÓ0€Mä
 &Ñ
)×
-Ñ
-¨a°MÓ
B€CØ�=‰=œBŸJšJ qÑ@Ó@Ð@Ð@ä
 (Ñ
+×
/Ñ
/°°mÓ
D€CØ�=‰=Ô0°À4ÓHÓHÐHÐHä
 *°tÑ
<×
@Ñ
@ÀÀ}Ó
U€CØ�=‰=Ô0°À4ÓHÓHÐHÑHr)   c                  ó  • [         R                  " S//S//S///5      n [         R                  " / SQ5      n[         R                  " / SQ5      n[        5       nUR                  X5        UR	                  U 5      n[        XB5        g )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  |  sg   € Ü
�Š�E�7�)˜u˜g˜Y¨%¨¨	Ð2Ó3€AÜ
�Š’Ó€AÜ—’š)Ó$€JÜ
Ó
€CØ‡G�GˆA„MØ�[‰[˜‹^€FÜ�vÕ*r)   c                  ó  • [         R                  " S//S//S///5      n / SQn/ SQnS/S/S//n[        SS9nUR                  X5        UR	                  U 5      nUR                  U 5      n[        XR5        [        Xc5        g )Nrü   rý   rþ   rÿ   r   rk   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�   € Ü
�Š�E�7�)˜u˜g˜Y¨%¨¨	Ð2Ó3€AÚ€AÚ€JØ˜˜a˜S 1 #�ÐÜ
 <Ñ
0€CØ‡G�GˆA„MØ�[‰[˜‹^€FØ×$Ñ$ QÓ'€LÜ�vÔ*Ü�|Õ6r)   c                  ó  • S//S-  n [         R                  " / SQ5      n[         R                  " / SQ5      n[        5       nUR                  X5        UR	                  U SS9n[        U5      S:X  d   e[        US   U5        g )	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
  “  sq   € Ø
ˆˆ�‰	€AÜ
�Š’Ó€AÜ—X’XšiÓ(€NÜ
Ó
€CØ‡G�GˆA„MØ—+‘+˜a¨D�+Ð1€Käˆ{Ó˜qÓ Ð Ð ä�{ 1‘~ ~Õ6r)   )r   r   r   r,   g      ô?r8   c                 ól   • [        5       nUR                  S U 5        UR                  S U5      S:X  d   eg )Nrä   )r   r0   r‰   )r!   r?   rš   s      r'   Útest_regressor_score_with_Noner     s3   € ô Ó
€CØ‡G�GˆD�!ÔØ�9‰9�T˜6Ó" cÓ)Ð)Ñ)r)   )r˜   r¦   r«   r‹   c                 óò   • / SQnS//S-  n[        U SSS9nUR                  X!5        UR                  U5      nS//S-  n[        U SSS9nUR                  XQ5        UR                  U5      n[        XG5        g )Nr�   r   r8   gffffffæ?)rZ   r‹   r«   r   )r   r0   r1   r	   )rZ   r!   r�   Úreg1r‘   r’   Úreg2r”   s           r'   Ú*test_regressor_prediction_independent_of_Xr  ­  sw   € â€AØˆ#ˆ�‰€BÜ 8°aÀ#ÑF€DØ‡H�HˆR„OØ—<‘< Ó#€Làˆ#ˆ�‰€BÜ 8°aÀ#ÑF€DØ‡H�HˆR„OØ—<‘< Ó#€Lä�|Õ2r)   c                 óÎ   • / SQn[         R                  " S5      n[        U SSS9nUR                  X!5      R	                  U5      nUR
                  [         R                  :X  d   eg )Nr�   r8   r   rŽ   )r   rh   r   r0   r   ÚdtypeÚfloat64)rZ   r!   r    ÚmodelÚprobass        r'   Útest_dtype_of_classifier_probasr  ½  sR   € ò 	€AÜ
�Š�‹€AÜ X¸AÈÑJ€EØ�Y‰Y�q‹_×*Ñ*¨1Ó-€Fà�<‰<œ2Ÿ:™:Ó%Ð%Ñ%r)   )r�   )FÚnumpyr   rJ   Ú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   rT   r^   re   ri   rv   rz   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#     s�  ðÛ Û Ý å ß 9Ý -÷ó õ /Ý 4ð ñBó ðBò0#ò"#òUô
*ò"ò(Aò  ò"$òò&$òò$6ð ‡�×ÑØâ	’|Ð$à�HŠH�q˜!�f˜q !˜f q¨! f¨q°!¨fÐ5Ó6Ø�HŠH�q˜!�f˜q !˜f q¨! f¨q°!¨fÐ5Ó6ð	
ðó	ñ*ó	ð*ð ‡�×ÑØÒOóñ3óð3ò&>òò*ò@òò*Jò.òB	òò4ò$)ò,'òòò$ò"$ð ‡�×ÑØâ	˜¨Ð*Ð,KÐLÚ	˜
 Q¨ FÐ+Ð-JÐKà�LŠLš,ªÐ5Ó6Ø˜ˆOØ)ð	
ò Ø˜Ð'Ø=ð	
ð �LŠLš,ªÐ5Ó6Ø˜!˜Y˜Ð(ØBð	
ðò&	ð+ ð ñ:ó;ð:òHð ‡�×Ñ˜¨.Ó9ñó :ðð ‡�×Ñ˜¨.Ó9ñ4ó :ð4ð$ ‡�×Ñ˜¨.Ó9ñ6ó :ð6ð& ‡�×Ñ˜¨.Ó9ñ9ó :ð9ôIò"+ò
7ò
7ð ‡�×ÑØâ	˜�v ‘zÐ"Ø	�Š�A�q�6˜A˜q˜6 A q 6¨A¨q¨6Ð2Ó	3°t¸T°l°^ÀaÑ5GÐHðóñ*óð*ð ‡�×Ñ˜Ò%OÓPñ3ó Qð3ð ‡�×ÑØÒOóñ&óñ&r)   