
    Q3jޢ                    .   d dl Z d dlZd dlZd dlmZ d dlZd dlm	Z	m
Z
 d dlmZmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z1m2Z2 d dl3m4Z4 d dl5m6Z6m7Z7 d dl8m9Z9m:Z: d dl;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD d dlEmFZFmGZG d dlHmIZImJZJmKZKmLZLmMZMmNZNmOZO d dlPmQZQ  ej                         ZSej                  j                  d       ZVdZWdZXeVj                  ddeW      ZZeVj                  ddeW      Z[eVj                  eXeW      e[z  eZz   Z]e]d ddf   j                  deW      Z_e]ddd f   j                  eXd      Z`e_j                         Zbe`j                         Zcd Zdd Zed Zfd Zgd Zhej                  j                  g d eh             ej                  j                  d g d!      d"               Zkej                  j                  g d eh             ej                  j                  d# e:             d$               Zld% Zmej                  j                  d&dgeKz   eLz         ej                  j                  d'd(d)g      d*               Znej                  j                  d+ e'd(,       e&d(-      g      ej                  j                  d&dgeKz   eLz         ej                  j                  d'd(d)g      ej                  j                  d.ej                  ej                  g      ej                  j                  d/g d0      d1                                    Zqej                  j                  d2g d3      ej                  j                  d4g d5      ej                  j                  d.ej                  ej                  g      ej                  j                  d&dgeKz   eLz         d6                             Zrd7 ZseDd8        Ztd9 Zud: Zvd; Zwd< Zxd= Zyej                  j                  d&eKeLz         d>        Zzej                  j                  d?d)dg      ej                  j                  d&eKeLz         d@               Z{ej                  j                  d?d)dg      dA        Z|dB Z}ej                  j                  dC e:             ej                  j                  dDeGgeFE      ej                  j                  dF e!        e!d)G       e"        e"d)G       e         e#dHI       e#dJI       e#dKI       e       g	eFE      dL                      Z~ej                  j                  dF e!        e"        e        geFE      dM        ZdN Zej                  j                  dC e:             ej                  j                  dDeGgeFE      ej                  j                  d?d)dg      dO                      ZdP ZdQ ZdR ZdS Zej                  j                  d?d)dg      ej                  j                  d&eKeLz         dT               Zej                  j                  dUd)d(g      ej                  j                  dVd)d(g      ej                  j                  d&dgeKz   eLz         dW                      ZdX Zej                  j                  d&eKeLz         dY        Zej                  j                  d&eKeLz         dZ        Zej                  j                  d&eLeKz         d[        Zej                  j                  d&eLeKz         d\        Zd] Zd^ Zej                  j                  d_d)d(g      ej                  j                  d`d)d(g      ej                  j                  daej                  j                  ddb       e	j                  ddbdcd      g      de                      Zej                  j                  dfeL      dg        Zdh Zej                  j                  dig dj      ej                  j                  dkg dl      dm               Zej                  j                  dfeL      dn        Zdo Zdp Zej                  j                  dqeK      dr        Zej                  j                  dqeK      ds        Zej                  j                  dqeK      dt        Zdu Zdv Zdw Zej                  j                  dqeK      dx        Zdy Zej                  j                  dqeK      dz        Zd{ Zd| Zej                  j                  d}d~dg      d        Zd Zej                  j                  dfeL      d        Zd Zd Zd Zd Zej                  j                  d&eKeLz         d        Zd Zej                  j                  dfeL      d        Zd Zej                  j                  dfeL      d        Zd Zej                  j                  dg d      ej                  j                  dfeL      d               Zej                  j                  dg d      ej                  j                  d&eJeKz   eMz         d               Zej                  j                  dfeL      d        Zej                  j                  dfeL      d        Zej                  j                  dejf                  egeKz   eLz         d        Zej                  j                  dC e:             d        Zd Zd Zd Zd Zd Zej                  j                  d&eJeKz   eLz         d        Zd Zej                  j                  dddg      d        Zej                  j                  dddg      ej                  j                  dd)d(g      ej                  j                  dae`e]g      d                      Zd Zd Zd Zej                  j                  dae] ej                  e]       ej                  e]        ej                  e]j                        g      d        Zej                  j                  dddg      d        Zd Zd Zej                  j                  dg d      d        Zd Zd Zej                  j                  dddg      d        Zej                  j                  dddg      ej                  j                  dd)d(g      d               Zej                  j                  dddg      ej                  j                  dd)d(g      d               Zej                  j                  dddg      ej                  j                  dd)d(g      d               Zd Zej                  j                  d eNddd       geLD  cg c]'  }  | ej                  dej                              ) c} z         d        Zej                  j                  dddg      d        Zej                  j                  dejf                  geKz   eLz         d        Zd Zd Zej                  j                  de"e!e&e'e%e$g      dÄ        Zej                  j                  de"e!e&e'e%e$e#eg      dĄ        Zdń Zej                  j                  dd)d(g      dƄ        Zej                  j                  eIdǬȫ      dɄ        Zej                  j                  eO edʫ      k  dˬȫ      d̄        Zd̈́ Zej                  j                  de$e%g      dτ        Zej                  j                  de$e%g      dЄ        Zej                  j                  dddg      edԄ               Zyc c} w )    N)sparsestats)config_contextdatasets)clone)RecordingCallbackskip_callback_test_if_wasm)NotFittedError)parse)linear_kernel)cross_val_predict)Pipeline)	BinarizerKernelCentererMaxAbsScalerMinMaxScaler
NormalizerPowerTransformerQuantileTransformerRobustScalerStandardScaleradd_dummy_featuremaxabs_scaleminmax_scale	normalizepower_transformquantile_transformrobust_scalescale)BOUNDS_THRESHOLD_handle_zeros_in_scale)SVR)gen_batchesshuffle)move_to)yield_namespace_device_dtype_combinations)	_array_api_for_tests_convert_containerassert_allcloseassert_allclose_dense_sparseassert_almost_equalassert_array_almost_equalassert_array_equalassert_array_lessskip_if_32bit)_get_check_estimator_ids check_array_api_input_and_values)_IS_WASMCOO_CONTAINERSCSC_CONTAINERSCSR_CONTAINERSLIL_CONTAINERS_sparse_random_array
sp_version)mean_variance_axis        size
   c                 >    t        | d      r| j                         } | S )Ntoarray)hasattrrB   as    R/DATA/.local/lib/python3.12/site-packages/sklearn/preprocessing/tests/test_data.pyrB   rB   Z   s    q)IIKH    c                 F    t        j                  |       j                  d   S )Nr   )npasarrayshaperD   s    rF   _check_dim_1axisrL   `   s    ::a=q!!rG   c                 J    ||k7  r| dz   |z  |k(  sJ y | |z  ||z
  z   |k(  sJ y )Nr=    )ibatch_start
batch_stopn
chunk_sizen_samples_seens         rF   assert_correct_incrrU   d   s<    QA#~555:~k!9:nLLLrG   c                  \   ddg} ddg}t        | |      D ]  \  }}t        j                  ||      }t        j                  |      }t               }t        j                  |d      dz  }t	        j
                  t              5  |j                  |||       d d d         y # 1 sw Y   xY w)N      r=   sample_weight)ziprngrandnr   pytestraises
ValueErrorfit)
n_samplessn_featuress	n_samples
n_featuresXyscalersample_weight_notOKs           rF   9test_raises_value_error_if_sample_weights_greater_than_1drj   k   s     QJa&K!$Z!=	:IIi,IIi ! "ii	15:]]:&JJq!+>J? '& "> '&s   B""B+	c            	   #     K   g dg dgg dg dg dgddgfg dg dgg dg dg dg dgt        j                  ddg      fdt         j                  dgt         j                  t         j                  dggdt         j                  dgt         j                  t         j                  dgt         j                  t         j                  dgt         j                  t         j                  dggt        j                  ddg      ffE d {    y 7 w)	Nr=   rW   rX                   @      ?r=   r   r=   )r   r   r=   r=   rX   )rI   arraynanrN   rG   rF   _yield_xw_x_sampleweightrv   }   s      	"	9-#J	
 	"	9i8HHaV	
 ^bffbffa01BFFA###	 HHaV		
  s   DDDD)Xwrf   rZ   array_constructor)rt   
sparse_csr
sparse_cscc                 @   |j                  d       }t        ||      }t        | |      } t        j                  | j                  d         }t        |      }|j                  | ||       t        j                  |j                  d         }t        |      }|j                  ||       g dg dg}	t        |j                  |j                         t        |j                  |j                         t        |j                  |	      |j                  |	             y )Nr   r   	with_meanrY         ?      @      @r   g      @g      @)
startswithr(   rI   onesrK   r   ra   r+   mean_var_	transform)
rw   rf   rZ   rx   r}   ywscaler_wrg   rh   X_tests
             rF   "test_standard_scaler_sample_weightr      s     &00::I1/0A	B 1	2B 
!	B	2HLLR}L5 	
Ai0F
JJq!/Fhnn5X]]3((0(2D2DV2LMrG   z"namespace, device_name, dtype_namec           	         t        |||      \  }}t        j                  |      j                  |d      }t        j                  |j
                  d         j                  |d      }t        j                  |       j                  |d      } t        j                  | j
                  d         j                  |d      }	t        j                  g dg dg      j                  |d      }
t               }|j                  ||       t               }|j                  | |	|       |j                  ||      }|j                  ||      }|j                  | |      }|j                  |	|      }|j                  |
|      }|j                  ||      }t               }t        d	      5  |j                  |||       t        |j                  t        d
      }t        |j                  t        d
      }d d d        t        |j                         t        |j                         t               }t        d	      5  |j                  ||       t        |j                  t        d
      }t        |j                  t        d
      }d d d        t        |j                         t        |j                         t        ||       t        ||       t        d	      5  t        t        |j                  |      t        d
      t        |j                  |      t        d
             d d d        y # 1 sw Y   CxY w# 1 sw Y   xY w# 1 sw Y   y xY w)NFcopyr   r~   r   rY   deviceTarray_api_dispatchcpuxpr   )r'   rI   rt   astyper   rK   r   ra   rJ   r   r%   r   r   r)   r   )rw   rf   rZ   	namespacedevice_name
dtype_namer   r   rg   r   r   rh   r   X_xpy_xpXw_xpyw_xp	X_test_xpsample_weight_xpscaler_w_xpw_meanw_var	scaler_xpuw_meanuw_vars                            rF   ,test_standard_scaler_sample_weight_array_apir      s    &ijIJB
:E2A

"":E":A	"		Ze		4B	!		$	$Ze	$	<BXX89@@RW@XFF
JJq!HLLR}L5 ::a:'D::a:'DJJr&J)EJJr&J)E

6&
1Izz-z? "K	4	0u4DE**r%@((R> 
1
 HNNF+HMM5)  I	4	0dD!)//b?Bu= 
1
 FLL'*FKK( GV$FE"	4	0I''	2r%HK)))4EJ	
 
1	0+ 
1	0 
1	0 
1	0s'   1AL?AM+AM?M	MM!c                     t         t        t        t        fD ]  } t               }|j	                  |       j                  | d      }t        | t              rt        j                  |       } t        |       dk(  rt        |j                  | j                                t        |j                  t        j                  t                      t#        |j%                  d      t        j&                  t                      t#        |j)                  d      t        j&                  t                      nt        |j                  | j%                                t        |j                  | j)                                t#        |j%                  d      t        j&                  t                      t#        |j%                  d      d       t#        |j)                  d      d       |j*                  | j,                  d   k(  sJ |j/                  |      }t#        ||         t        j                  d      } t               }|j	                  |       j                  | d      }t        |j                  d       t        |j                  d       t#        |j%                  d      d       t#        |j)                  d      d       |j*                  | j,                  d   k(  sJ y 	NTr   r=   r   axis        rr   ro   r=   )X_1rowX_1colX_list_1rowr   ra   r   
isinstancelistrI   rt   rL   r+   r   ravelscale_r   re   r,   mean
zeros_likestdn_samples_seen_rK   inverse_transform)rf   rh   X_scaledX_scaled_backs       rF   test_standard_scaler_1dr      s   fk;7!::a=**14*8aAA!#aggi8rwwz/BC%hmmm&;R]]:=VW%hlll&:BMM*<UVaffh7quuw7%hmmm&;R]]:=VW%hmmm&;SA%hlll&:C@%%333 00:!-3- 82 	AFzz!}&&qt&4Hc*s+hmmm3S9hlll2C8!!QWWQZ///rG   sparse_containeradd_sample_weightFTc                    t         j                  j                  d      }d}d}| rt        j                  |      }nd }d}|!t         j                  t         j
                  g}n/t         j                  t         j
                  t         j                  g}|D ]  }|j                  ||      j                  |      }	|
 ||	      }	d}t        |      }
|
j                  |	|      j                  |	      }|	j                  |j                  k(  sJ |
j                  j                  t         j                  k(  sJ |
j                  j                  t         j                  k(  rJ  y )Nr   r@   rX   TFr|   rY   )rI   randomRandomStater   float64float32float16r]   r   r   ra   r   dtyper   r   )r   r   r\   rd   re   rZ   r}   supported_dtyper   rf   rh   r   s               rF   test_standard_scaler_dtyper     s    ))


"CIJ	*I# ::rzz2::rzz2::> IIi,33E:' #AI)4::a}:=GGJww(..(((||!!RZZ///}}""bjj000 !rG   rh   r|   with_centeringr   constant)r   rr         Y@c                 N   t        |       } t        | t              r.|r,t        j                  | j
                  j                   d       t        j                  j                  d      }d}d}|r t        |j                  |      dz        }ni }t        j                  ||f||      }	||	n ||	      }
 | j                  |
fi |j                  |
      }t        | t              r8t!        | j"                  t        j$                  |
j&                  d         d	
       t!        | j(                  t        j*                  |
j&                  d                ||
usJ t-        ||
       t        | t              r-|s*t/        |
| j0                        }||
usJ t-        ||
       y y y )Nz# does not yet support sample_weightr   d   r=   r>   rW   rY   )rK   
fill_valuer   Hz>atolr|   )r   r   r   r^   skip	__class____name__rI   r   r   dictuniformfullra   r   r   r)   r   zerosrK   r   r   r*   r   r}   )rh   r   r   r   r   r\   rd   re   
fit_paramsX_arrayrf   r   
X_scaled_2s                rF   &test_standard_scaler_constant_featuresr   /  si    6]F&,',=v''0011TUV
))


"CIJ(Ca(GH

ggY
3PUVG#+1A'1JAvzz!*z*44Q7H&.)RXXaggaj%9E FMM2771771:#671 1-&.)2C1(8(89
"""$Z3	 3D)rG   rd   )r@   r   i'  average)g|=r=       _Bc                 .   d\  }}t        j                  t        ||dz         D cg c]  }d|z  	 c}|      }|j                  d   }t        j                  | |f|      }	||z   |	d | dz  d d f<   ||z
  |	| dz  d d d f<   ||	n ||	      }
t        d      j                  |
      }t        j                  t         j                        j                  }| |z  |dz  z  | dz  |dz  z  |dz  z  z   }|dz  |k  }t        j                  |      sJ t        |j                  |   ||   k        sJ t        |j                  |   d	       |	dd d f   |	d
d d f   z
  dk7  }t        |j                  t        j                  |         d       t        |j                  t        j                  |         d       t        j                   |dz  |kD  |      }t        |j                  |   t        j"                  |j                        |          y c c}w )N)i   r=   r@   r   r   rW   Fr|   rr   r<   )rI   rt   rangerK   emptyr   ra   finfor   epsanyallr   r)   r   logical_notlogical_andsqrt)rd   r   r   r   	scale_min	scale_maxrO   scalesre   rf   r   rh   r   boundswithin_boundsrepresentable_diffcommon_masks                    rF   +test_standard_scaler_near_constant_featuresr   ]  s    #IyXXeIy1}&EF&Er1u&EFeTFaJ
)Z(6A$v-A	Q$v-Ai1n#+a1A!1DGe,009F ((2::

"
"C _vqy(9a<#q&+@7A:+MMFAI'M 66-    v{{=)VM-BBCCCFMM-0#6 1a41RU8+q0FKK/A BCQGFMM"..1C"DEqI
 ..V!35GHKFMM+.0D[0QRQ Gs   Hc                      g d} t        j                  |       }| |fD ]Y  }t        |      }t        |j	                         d       t        |j                         d       t        t        |dd      |       [ y )N)rr         @      @r   r   rr   Fr}   with_std)rI   rt   r   r,   r   r   r-   )X_listX_arrrf   r   s       rF   test_scale_1dr     s_    !FHHVEe_8!(--/37!(,,.#65eeDaH	 rG   c                     t        j                  dt        j                  d      t         j                        } t	        j
                         5  t	        j                  dt               t        |        d d d        t        t        |       t        j                  d             t        j                  dt        j                  d      t         j                        } d}t        j                  t        |      5  t        |       }d d d        t        t        j                  d             t        j                  ddt         j                        } t	        j
                         5  t	        j                  dt               t        |       }d d d        t        t        j                  d             t        j                  dd	t         j                        }d
}t        j                  t        |      5  t        |      }d d d        t        t        j                  d             t        ||       t        j                  t        |      5  t        |d      }d d d        t        t        j                  d             t        ||       y # 1 sw Y   !xY w# 1 sw Y   xY w# 1 sw Y   "xY w# 1 sw Y   xY w# 1 sw Y   hxY w)N   h㈵>r   errorr@   z:standard deviation of the data is probably very close to 0matchg0.++g}Ô%ITz$Dataset may contain too large valuesFr   )rI   r   logr   warningscatch_warningssimplefilterUserWarningr   r,   r   r^   warns)xwarning_messagex_scaledx_small_scaledx_bigx_big_scaledx_big_centereds          rF   (test_standard_scaler_numerical_stabilityr    s   
 	266$<rzz2A 
	 	 	"g{3a 
# eAh4 	BFF4L

3ARO	k	98 
:h5
F"**-A		 	 	"g{3q 
# nbhhrl; GGBRZZ0E<O	k	9U| 
:lBHHRL9lN;	k	9uu5 
:nbhhrl;nn=9 
#	" 
:	9
 
#	" 
:	9 
:	9s<   &J:J '&J-7J:KJ J*-J7:KKc                  ^   t         j                  j                  d      } d}d}| j                  ||      }d|d d df<   t	               }|j                  |      j                  |d      }t        j                  t        j                  |            rJ |j                  |k(  sJ t        |j                  d      |dgz         t        |j                  d      g d       ||usJ |j                  |      }||usJ ||usJ t        ||       t        |d	d
      }t        j                  t        j                  |            rJ t        |j                  d	      |dgz         t        |d	d      }t        j                  t        j                  |            rJ t        |j                  d	      |dgz         t        |j                  d	      |dgz         ||usJ |j                  |      j                  |d
      }t        j                  t        j                  |            rJ t        |j                  d      |dgz         t        |j                  d      g d       ||u sJ | j                  dd      }d|d d df<   t	               }|j                  |      j                  |d      }t        j                  t        j                  |            rJ t        |j                  d      |dgz         t        |j                  d      g d       ||usJ y )Nr   ro   rn   r   Tr   r   r   rr   rr   rr   rr   r=   F)r   r   rr   )rI   r   r   r]   r   ra   r   r   isnanr   r,   r   r   r   r   )r\   re   rd   rf   rh   r   r   s          rF   test_scaler_2d_arraysr    s   
))


"CJI		)Z(AAadGFzz!}&&qt&4Hvvbhhx()))!!Y...hmmm3Z3%5GHhlll24MN1 ,,X6M!!!(((mQ/QQ/Hvvbhhx()))hmmm3Y#5FGQQ.Hvvbhhx()))hmmm3Y#5FGhlll2I4EF1zz!}&&qu&5Hvvbhhx()))hmmm3Z3%5GHhlll24MNq==		!QAAadGFzz!}&&qt&4Hvvbhhx()))hmmm3Z3%5GHhlll24MN1rG   c                  ,   t         j                  j                  d      } | j                  ddddg      j	                  t         j
                        }t        j                  d      5  t               j                  |      }|j                  |      }d d d        t               j                  |j	                  t         j                              }t        j                  t        j                              sJ t        ||d	       y # 1 sw Y   xxY w)
Nr   ro   r@   i@ r=   raise)overrW   decimal)rI   r   r   r   r   r   errstater   ra   r   fit_transformr   r   isfiniter,   )r\   rf   rh   r   X_scaled_f64s        rF   test_scaler_float16_overflowr    s    
))


"C 	ArFA;'..rzz:A	'	"!%%a(##A& 
#
 "#11!((2::2FGL
 66"++h'(((
 ha@ 
#	"s   (+D

Dc                      t        j                  g d      } t        | d      }t        | t        j                  g d             t        |t        j                  g d             y )N)r   gؗҜ<r=   rW   rX   Tr   )r=   r=   r=   rW   rX   )rI   rt   r!   r)   )s1s2s     rF   test_handle_zeros_in_scaler     sD    	%	&B		.BB!456B12rG   c            
      j   t         } | j                  d   }ddd||dz   fD ]  }t               j                  |       }t               }t	        t
        |      D ]  }|j                  | |         } t        |j                  |j                         t        |j                  |j                         |j                  |j                  k(  sJ t        |j                  |j                         t        |j                  |j                         t        |j                  |j                         t        d|      }t               j                  | |         }t               j                  | |         }t        |j                  |j                         t        |j                  |j                         |j                  |j                  k(  sJ t        |j                  |j                         t        |j                  |j                         t        |j                  |j                         t               j                  |       }t               }t        t	        t
        |            D ]H  \  }}|j                  | |         }t!        ||j"                  |j$                  |||j                         J  y )Nr   r=   rW   2   *   rP   rQ   rR   rS   rT   )X_2drK   r   ra   r#   rd   partial_fitr,   	data_min_	data_max_r   data_range_r   min_slice	enumeraterU   startstoprf   rR   rS   scaler_batchscaler_incrbatchbatch0rO   s           rF   test_minmax_scaler_partial_fitr4    s+    	A	
A!RAF+
#~))!,"n J7E%11!E(;K 8 	","8"8+:O:OP!,"8"8+:O:OP++{/J/JJJJ!,":":K<S<ST!,"5"5{7I7IJ!,"3"3[5E5EF q*%#~))!F)4"n006;!,"8"8+:O:OP!,"8"8+:O:OP++{/J/JJJJ!,":":K<S<ST!,"5"5{7I7IJ!,"3"3[5E5EF $~))!,"n!+i"DEHAu%11!E(;K!KK ::%*:: F= ,rG   c            
      (   t         } | j                  d   }ddd||dz   fD ]o  }t        d      j                  |       }t        d      }t	        t
        |      D ]  }|j                  | |         } t        |j                  |j                         |j                  |j                  k(  sJ |j                  |j                  k(  sJ t        d|      }t               j                  | |         }|dk(  r{t        t        j                  t        t        j                        |j                         t        t        j                   t        t        j                        |j"                         n\t        t        j$                  | |   d	      |j                         t        t        j&                  | |   d	      |j"                         t               j                  |       }t               }t)        t	        t
        |            D ]H  \  }}|j                  | |         }t+        ||j,                  |j.                  |||j                  
       J t        |j                  |j                         |j                  |j                  k(  rpJ  y )Nr   r=   rW   r"  r#  Fr   r   r   r$  )r%  rK   r   ra   r#   rd   r&  r,   r   r   r   r+  rI   r   re   r   r   r   varr   r,  rU   r-  r.  r/  s           rF    test_standard_scaler_partial_fitr7  K  s    	A	
A!RAF+
%u599!<$e4 J7E%11!E(;K 8!,"4"4k6G6GH  K$4$4444++{/J/JJJJ q*%$&221V9=?%2::68H8H &
"**5{7I7I &bffQvYQ&?AQAQR%qyq);+=+=
 &'++A.$&!+i"DEHAu%11!E(;K!KK ::%*:: F 	","3"3[5E5EF++{/J/JJJJU ,rG   c                    t         j                  j                  d      }d}d}|j                  dd|      }|j                  dd|      }|j	                  ||      |z  |z   }t               j                  |      }t               }|D ]#  }	|j                  |	j                  d	|            }% d
}
t        |j                  |j                  |
       t        |j                  |j                  |
       t        |j                  |j                  |
       d}d} | |j                  dd|      j                  t         j                        |z        }t        d      j                  |      }t        d      }|D ]4  }	|	j                   d	k(  r|	j                  d	d      }	|j                  |	      }6 d
}
|j                  J t        |j                  |j                  |
       t        |j                  |j                  |
       y )Nr   rW   r   g  4&kg  4&kCr>   g     @@    .Ar=   gư>)rtol)r   rX   g@xDFr|   r<   )rI   r   r   r   r]   r   ra   r&  reshaper)   r   r   r   randintr   r   ndim)r   r\   re   rd   offsetsr   rf   r0  r1  chunktolr?   r   rh   s                 rF   4test_standard_scaler_partial_fit_numerical_stabilityrA  ~  s    ))


"CJIkk%Jk7G[[c
[3F		)Z(61G;A!#''*L "K!--emmAz.JK  CK%%|'9'9DK$$l&7&7cBK&&(;(;#F
 DEQ4077

CeKLA e,003F 51K::? MM!R(E!--e4  C<<###K$$fkk<K&&C@rG   rZ   c                     |t        j                  dgdgdgdgg            }| r"t        j                  |j                  d         } t        ddd      }|j                  ||       j                  |      }t        |j                         |j                                |j                  |      }t        |j                         |j                                t        |j                         |j                                y )	Nrr   r   r   r   FTr}   r   r   rY   )rI   rt   r\   randrK   r   r&  r   r-   rB   r   )rZ   r   rf   null_transformX_nullX_origs         rF   test_partial_fit_sparse_inputrH    s     	C53%#">?@A,#ee$ON'''GQQRSTFv~~'5--f5Fv~~')9:v~~'5rG   c                 
   t         d dd d f   }| r"t        j                  |j                  d         } t	               }t        t        |j                  d   d            D ]  \  }}|d |dz   d d f   }|j                         }| .t	               j                  |      }|j                  ||         }n:t	               j                  || d |dz          }|j                  ||   | |         }|j                  |      }t        ||       t        ||       |j                  |      }	t        ||	       t        j                  |j                  d         }
t        j                  t               j"                  }t%        |
|j&                  |z          t%        |
|j(                  |z          | |dz   |j*                  k(  rbJ t        j,                  | d |dz          t/        j0                  |j*                        k(  rJ  y )Nr   r   r=   rY   )r%  r\   rD  rK   r   r,  r#   r   r  r&  r   r,   r   rI   r   r   floatr   r.   r   r   r   sumr^   approx)rZ   rf   r1  rO   r2  X_sofarchunks_copyscaled_batchscaled_incrright_inputzeroepsilons               rF   /test_standard_scaler_transform_with_partial_fitrT    s    	TcT1WA, "Kk!''!*a895Iq1uIqL/lln )+99'BL%11!E(;K)+99}Wq1u'= : L &11%e(< 2 K "++G4!,<!';7!33K@!';7xx
#((5/%%$ 0 07 :;$ 2 2W <= Ek9999966-!a%01V]]++6   9 :rG   c            	          t        j                  g dg dg dg dg dg dgt         j                        } t               }|j	                  |        |j                  |        y )N)r=   r=   r=   r   r=   r   )r   r   r   r=   r   r   )r=   rn   r=   r=   r   r   )r   r=   r   r   r=   r   )r   rn   r   r=   r   r=   r   )rI   rt   int32r   ra   r   )r  rh   s     rF   .test_standard_check_array_of_inverse_transformrW    sW     		
 hh
	A F
JJqM
 QrG   z(array_namespace, device_name, dtype_namecheck)ids	estimatorclipl1norml2maxc                 J    | j                   j                  } ||| |||       y )Nr   r   )r   r   )rZ  rX  array_namespacer   r   names         rF   'test_preprocessing_array_api_compliancerf     s,    6 ''D	rG   c                 B   t        ddd      \  }}t        j                  g dg dg dgt        j                        }|j                  ||	      }t	        d
      5  | j                  |      }d d d        j                  |j                  k(  sJ y # 1 sw Y   %xY wNtorchmpsr   rc  rl   rm   )   r   r@   r   r   Tr   )r'   rI   rJ   int64r   r  r   r   )rZ  r   r   X_npr   X_outs         rF   ;test_preprocessing_integer_array_api_on_float32_only_devicero  %  s     &g5YWJB ::y)Z8ID::d6:*D	4	0''- 
1 ;;"**$$$ 
1	0s   BBc                  6   t        ddd      \  } }t        j                  g dg dg dgt        j                        }| j                  ||	      }t	        d
      5  t        |      }d d d        j                  | j                  k(  sJ y # 1 sw Y   %xY wrh  )r'   rI   rJ   rl  r   r   r   r   )r   r   rm  r   rn  s        rF   7test_normalize_integer_array_api_on_float32_only_devicerq  <  sz    %g5YWJB ::y)Z8ID::d6:*D	4	0$ 
1 ;;"**$$$ 
1	0s   BBc                 `    t               }|j                  j                  } | ||||||       y )N)r   r   check_sample_weight)r   r   r   )rX  rZ   rd  r   r   rZ  re  s          rF   )test_standard_scaler_array_api_compliancert  J  s7      I''D	)rG   c                  b   t         j                  } t               }|j                  |       }t	        |j                  d      d       t	        |j                  d      d       |j                  |      }t	        | |       t        d      }|j                  |       }t	        |j                  d      d       t	        |j                  d      d       |j                  |      }t	        | |       t        d      }|j                  |       }t	        |j                  d      d       t	        |j                  d      d	       |j                  |      }t	        | |       t        d
      }t        j                  t              5  |j                  |        d d d        y # 1 sw Y   y xY w)Nr   r   r=   r=   rW   feature_rangerW   )      333333?ry  rz  )rW   r=   )irisdatar   r  r,   minra  r   r^   r_   r`   ra   )rf   rh   X_transX_trans_invs       rF   test_min_max_scaler_irisr  c  sU   		A^F""1%Ggkkqk115gkkqk115**73Ka- /F""1%Ggkkqk115gkkqk115**73Ka- 4F""1%Ggkkqk148gkkqk137**73Ka- /F	z	"

1 
#	"	"s   
F%%F.c                     g dg dg dg} g dg dg dg}t               }|j                  |       }g dg dg d	g}t        ||       |j                  |      }t        | |       |j	                  |      }g dg d
g dg}t        ||d       t        d      }|j                  |       }g dg dg dg}t        ||       t        |       }t        ||       t        | d      }t        ||       y )Nr   rr         ?r   rr   gr   rr   g?r   rq   r        rr   r   r   rr   r   )r   r   r  r   r   r   r   r   rr   )r  r   gsh|??)r   r   gS?rW   r  rv  rw  )rr   rr   r   rr   rr   rr   )rr   rr   rq   )r   r  r,   r   r   r   )	rf   X_newrh   r~  X_expected_0_1r  X_trans_newX_expected_0_1_newX_expected_1_2s	            rF   *test_min_max_scaler_zero_variance_featuresr    s    	+-=>A/1ABE ^F""1%G%HNg~6**73Ka-""5)K,.@BTUk+=qI /F""1%G%HNg~6 1oGg~61F3Gg~6rG   c                      t         j                  } t        | d      }t        t	        j
                  |d      d       t        t	        j                  |d      d       y )Nr=   r   r   )r{  r|  r   r,   rI   r}  ra  )rf   r~  s     rF   test_minmax_scale_axis1r    sA    		A11%GbffW15q9bffW15q9rG   c                  f   t         t        t        t        fD ]<  } t        d      }|j	                  |       j                  |       }t        | t              rt        j                  |       } t        |       dk(  rgt        |j                  d      t        j                  t                     t        |j                  d      t        j                  t                     n8t        |j                  d      d       t        |j                  d      d       |j                   | j"                  d   k(  sJ |j%                  |      }t        ||        ? t        j&                  d      } t               }|j	                  |       j                  |       }|j                         dk\  sJ |j                         dk  sJ |j                   | j"                  d   k(  sJ t         j)                         }|j                         }|j                         }t        ||z
  ||z
  z  t+        |d             y r   )r   r   r   r   ra   r   r   r   rI   rt   rL   r,   r}  r   re   ra  r   rK   r   r   r   r   )rf   rh   r   r   X_1dr*  max_s          rF   test_min_max_scaler_1dr    s   fk;74(::a=**1-aAA!#%hlll&:BHHZ<PQ%hlll&:BHHZ<PQ%hlll&:C@%hlll&:C@%%333 00:!-3# 8( 	A^Fzz!}&&q)H<<>S   <<>S   !!QWWQZ/// <<>D88:D88:D	%|Dt'DrG   c                    t         j                  j                  d      }|j                  dd      }d|d d df<    ||      }| r|j	                  |j
                  d         } t        j                  t              5  t               j                  |       d d d        t        d      j                  ||       }|j                  |d	
      }t        j                  t        j                  |            rJ t        d      j                  ||       }|j                  |d	
      }t        j                  t        j                  |j                              rJ t        |j                   |j                          t        |j"                  |j"                         t        |j$                  |j$                         t        |j&                  |j&                         | =t        |j)                  d      g dd       t        |j+                  d      g d       t-        |d      \  }	}
t        |	|j)                  d             t        |
|j/                  d             ||usJ ||usJ |j1                  |      }||usJ ||usJ t        ||       |j1                  |      }||usJ ||usJ t        |j3                         |       |t4        v rqt        ddd	      }|j7                  |      }t9        |j                  |j                         |j1                  |      }t9        |j                  |j                         y y # 1 sw Y   xY w)Nr#  rn   ro   r   r   Fr|   rY   Tr   r   r   g{GzgQ@gffffffֿg(\rW   r  rC  )rI   r   r   r]   rD  rK   r^   r_   r`   r   ra   r   r   r  r|  r,   r   r   r   r   r   r   r9   r6  r   rB   r5   r  r-   )rZ   r   r\   rf   X_sparserh   r   scaler_sparseX_sparse_scaledX_sparse_scaled_meanX_sparse_scaled_varr   X_sparse_scaled_backrE  rF  rG  s                   rF   test_scaler_without_centeringr    s    ))


#C		!QAAadG"H,	z	"X& 
# e,00-0PF-Hvvbhhx()))"U377 8 M $--hT-BOvvbhh334555fllM,?,?@fkk=+=+=>fmm]-A-ABf44m6S6ST!MMqM!#CQ	
 	"(,,A,"68QR0B?TU0V--2HMMqM4IJ18<<Q<3GH 1(***,,X6M!!!(((mQ/(::?Kx///6662::<a@>)'%%dS--h76;;611&96;;6 *U 
#	"s   <MMr}   r   c                    t        j                  g dt         j                  ddgddt         j                  gddt         j                  ggt         j                        }| ||      }t	        j
                  |      r| rt        j                  d	       t        | |
      }|j                  |       t        |j                  t        j                  g d             y )Nr   r=   rX   rp   r@   ro   rn   r   r   r   z3'with_mean=True' cannot be used with sparse matrix.r   )rX   rn   rW   )rI   rt   ru   r   r   issparser^   r   r   ra   r-   r   )r}   r   r   rf   transformers        rF   #test_scaler_n_samples_seen_with_nanr    s     		RVVQOaBFF^aBFF^DBJJ	A #QqiIJ 9xHKOOA{22BHHY4GHrG   c                     | j                   |j                   cxu rJ  J | j                  |j                  cxu rJ  J | j                  |j                  cxu rJ  J | j                  |j                  k(  sJ y N)r   r   r   r   )scaler_1scaler_2s     rF   "_check_identity_scalers_attributesr    sp    >>X^^33333==HMM11111??hoo55555##x'?'????rG   c                    t        j                  g dg dg dgt         j                        } | |      }t        dd      }|j	                  |      }t        ||       t        |      }|j	                  |      }t        ||       t        ||       |j                  |       |j                  |       t        ||       |j                  |       |j                  |       t        ||       y )Nr  )ro   rp   r   )r   r   r@   r   Fr   )rI   rt   r   r   r  r)   r   r*   r  r&  ra   )r   X_denser  transformer_denseX_trans_densetransformer_sparseX_trans_sparses          rF   test_scaler_return_identityr  &  s     hh	9j9LG(H&G%33G<MM7+01'55h?N :&'8:LM!!'*""8,&'8:LM'"8$&'8:LMrG   c                    t         j                  j                  d      }|j                  dd      }d|d d df<    | |      }t	        j
                  d      5  t        d	      j                  |      }|j                  |d
      }d d d        t        j                  t        j                              rJ t	        j
                  d      5  t        d	      j                  |      }|j                  |d
      }d d d        t        j                  t        j                  j                              rJ t        j                  j                         t        |j                  |j                         t        |j                  |j                         t        |j!                  d      g dd       t        |j#                  d      g d       t%        |j'                  t(              d      \  }}	t        ||j!                  d             t        |	|j#                  d             ||usJ ||usJ |j+                  |      }
|
|usJ |
|usJ t        |
|       |j+                  |      }||usJ ||usJ t        |j-                         |       | t.        v rt        ddd      }t	        j
                  d      5  |j1                  |      }d d d        t3        j                  |j                         |j+                  |      }t3        |j                  |j                         y y # 1 sw Y   xY w# 1 sw Y   GxY w# 1 sw Y   vxY w)Nr#     rn   ro   r>   r   TrecordFr|   r   r   )r   gX9v?gV-?g      5@gl?rW   r  rC  )rI   r   r   r<  r  r  r   ra   r   r   r  r|  r,   r   r   r   r   r   r9   r   rJ  r   rB   r5   r  r-   )r   r\   rf   r  rh   r   r  r  r  X_sparse_scaled_stdr   r  rE  rF  rG  s                  rF   test_scaler_intr  @  s    ))


#CBV$AAadG"H		 	 	-%044Q7##AD#1 
. vvbhhx()))		 	 	-&7;;HE'11(1F 
. vvbhh334555fllM,?,?@fkk=+=+=>fmm]-A-AB1? hlll24MN0Bu%q1-- 2HMMqM4IJ18<<Q<3GH 1(***,,X6M!!!(((mQ/(::?Kx///6662::<a@>)'%%dS$$D1#11(;F 26;;611&96;;6 *O 
.	-
 
.	-H 21s$   /L8/MM8MMMc                    t         j                  j                  d      }|j                  dd      }d|d d df<    | |      }|j	                         }t        d      j                  |       t        ||       |j	                         }t        dd      j                  |       t        |j                         |j                                y )	Nr#  rn   ro   r   r   Fr   )r}   r   )	rI   r   r   r]   r   r   ra   r-   rB   )r   r\   rf   r  X_copyX_sparse_copys         rF   test_scaler_without_copyr  y  s     ))


#C		!QAAadG"HVVXF""1%q&!MMOMU/33H=x'')=+@+@+BCrG   c                    t         j                  j                  d      }|j                  dd      } | |      }t	        j
                  t              5  t        |d       d d d        t	        j
                  t              5  t        d      j                  |       d d d        t        d      j                  |      }t	        j
                  t              5  |j                  |       d d d         | |j                  |            }t	        j
                  t              5  |j                  |       d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   lxY w# 1 sw Y   y xY w)Nr#  rn   ro   Tr|   )rI   r   r   r]   r^   r_   r`   r   r   ra   r   r   )r   r\   rf   r  rh   X_transformed_sparses         rF   +test_scale_sparse_with_mean_raise_exceptionr    s   
))


#C		!QA"H 
z	"h$' 
#	z	"&**84 
# d+//2F	z	"" 
# ,F,<,<Q,?@	z	"  !56 
#	" 
#	"	"	"
 
#	" 
#	"s0   D?EE$E#?EEE #E,c                      t         j                  ddddgg} t        j                  t        d      5  t        |        d d d        y # 1 sw Y   y xY w)Nro   rp   rk  r   z8Input X contains infinity or a value too large for dtyper   )rI   infr^   r_   r`   r   rf   s    rF   &test_scale_input_finiteness_validationr    sD    
&&!Q1	A	U
 	a
 
 
s   AAc                      t        j                  dd      } t        d      }d}t        j                  t
        |      5  |j                  |        d d d        y # 1 sw Y   y xY w)Nr;   r@   Tr   zCannot center sparse matricesr   )r   rD  r   r^   r_   r`   ra   )r  rh   err_msgs      rF   test_robust_scaler_error_sparser    sG    {{4$H.F-G	z	1

8 
2	1	1s    AA$r   with_scalingrf   rX   r  densityc                 r   |r*t        j                  |       rt        j                  d       t	        ||      }|j                  |        |r&t        |j                  t        j                        sJ |j                  J |r't        |j                  t        j                        sJ y |j                  J y )Nz(RobustScaler cannot center sparse matrix)r   r  )r   r  r^   r   r   ra   r   center_rI   ndarrayr   )rf   r   r  rh   s       rF   test_robust_scaler_attributesr    s    
 &//!,>?lSF
JJqM&.."**555~~%%%&--444}}$$$rG   csr_containerc                    t         j                  j                  dd      }d|d d df<    | |      }t        d      }|j	                  |       |j
                  d   t        j                  d      k(  sJ |j                  |      }t        |d d dgf   j                         |d d dgf   j                                y )Nr@   ro   r   Fr   r=   )rI   r   r]   r   ra   r   r^   rL  r   r)   rB   )r  rf   rh   r~  s       rF   "test_robust_scaler_col_zero_sparser    s     			AAAadGaA/F
JJqM==v}}Q////q!GAa!fI%%'QC)@)@)BCrG   c                  T   t         j                  j                  d      } | j                  dd      }d|d d df<   t	               }|j                  |      j                  |      }t        t        j                  |d      ddgz         t        |j                  d      d   d       y )Nr   rn   ro   r   r   )
rI   r   r   r]   r   ra   r   r,   medianr   )r\   rf   rh   r   s       rF   test_robust_scaler_2d_arraysr    s    
))


"C		!QAAadG^Fzz!}&&q)Hbiiq91u9Ehlll215q9rG   r  )r   g?皙?r  r=   strictly_signed)positivenegativer   Nc                 T   t        j                  dd|       j                         }|dk(  r%t        j                  |j
                        |_        nn|dk(  r&t        j                  |j
                         |_        nC|dk(  r>t        j                  |j
                  j                  t        j                        |_        |j                         }t        d	      }t        d	      }|j                  |       |j                  |       t        |j                  |j                         y )
Nr;   ro   r  r  r  r   r   Fr   )r   rD  tocscrI   absr|  r   rK   r   rB   r   ra   r)   r   )r  r  r  r  r  scaler_denses         rF   +test_robust_scaler_equivalence_dense_sparser    s     {{4G4::<H*$x}}-	J	&..	G	#!4!4BJJG G 6Mu5LhWM((,*=*=>rG   c                    t         j                  j                  d      }|j                  dd      }t        j                  g dg      }t        d      }|j                  |      }|j                   | |            }||j                  z  }t        |j                         |       |j                  |      }t        ||j                                y )Nr   rn   ro   )r  rr   rq   r   r  Fr   )rI   r   r   r]   rt   r   ra   r   r   r,   rB   r   )r  r\   rf   
single_rowrh   	row_transrow_expectedrow_scaled_backs           rF   (test_robust_scaler_transform_one_row_csrr    s     ))


"C		!QA567J/FZZ]F  z!:;I-Li//1<@..y9Oj/*A*A*CDrG   c                  4   t         j                  } t               }|j                  |       }t	        t        j                  |d      d       |j                  |      }t	        | |       t        j                  |dd      }|d   |d   z
  }t	        |d       y )Nr   r      K   qr   r=   	r{  r|  r   r  r,   rI   r  r   
percentile)rf   rh   r~  r  r  iqrs         rF   test_robust_scaler_irisr    s~    		A^F""1%Gbiia8!<**73Ka-
g2A
A$1+Cc1%rG   c                  8   t         j                  } t        d      }|j                  |       }t	        t        j                  |d      d       |j                  |      }t	        | |       t        j                  |dd      }|d   |d   z
  }t	        |d       y )N)r@   Z   quantile_ranger   r   r  r=   r  )rf   rh   r~  r  r  q_ranges         rF   !test_robust_scaler_iris_quantilesr    s    		A2F""1%Gbiia8!<**73Ka-
g2AdQqTkGgq)rG   csc_containerc                    t         j                  }t        d      }|j                  |      }|j	                  |      }t        ||       t        dd      }|j                  |      }|j	                  |      }t        ||        | |      }|j                  |      }|j	                  |      }t        |j                         |j                                y )Nr:   n_quantilesnormal)r  output_distribution)r{  r|  r   r  r   r,   rB   )r  rf   r  r~  r  r  X_sparse_tranX_sparse_tran_invs           rF   test_quantile_transform_irisr     s    		A%"5K''*G//8Ka-%"(SK''*G//8Ka- QH--h7M#55mDh..02C2K2K2MNrG   c                    t        j                  g dg dg dg      } | |      }t        j                  g dg dg dg      } | |      }d}t        j                  t        |      5  t        d      j                  |       d d d        t        d	      }d
}t        j                  t        |      5  |j                  |       d d d        |j                  |       d
}t        j                  t        |      5  |j                  |       d d d        t        j                  g dg dg      }d}t        j                  t        |      5  |j                  |       d d d        t        d	      j                  |      }t        j                  t        d      5  |j                  d       d d d        t        d	      }d}t        j                  t        |      5 }|j                  |       d d d        t              dk(  sJ |j                  |j                  d   k(  sJ y # 1 sw Y   xY w# 1 sw Y   sxY w# 1 sw Y   8xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   wxY w)N)
r   r  r"  r   r   r   r  r   r   r   )
rW   rn   r   r   rp   r   r   r@   r   r   )
r   r   @ffffff@r   r   ffffff@r         #@r  )
rn   r   r   rp   r   r   r@   r   r   zmThe number of quantiles cannot be greater than the number of samples used. Got 1000 quantiles and 10 samples.r   r@   )	subsampler  z>QuantileTransformer only accepts non-negative sparse matrices.zKX has 2 features, but QuantileTransformer is expecting 3 features as input.z+Expected 2D array, got scalar array insteadr   zn_quantiles is set to n_samplesr=   r   )rI   	transposer^   r_   r`   r   ra   r   r   r  r  lenn_quantiles_rK   )r  rf   X_negr  r  
X_bad_featwarn_msgr  s           rF   #test_quantile_transform_check_errorr
  5  s   
/+4	
	A 	aALL/,4	
E % E	 
 
z	1b)--a0 
2 &"5KNG	z	1 
2OOANG	z	1e$ 
2 	,.VWJ 	V  
z	1%%j1 
2 &"599!<K	z)V	Wb! 
X &#6K0H	k	2f 
3v;!##qwwqz111? 
2	1
 
2	1 
2	1 
2	1
 
X	W
 
3	2sH   *H-7H:?II!I $I,-H7:III I),I5c                    t        j                  ddgddgddgddgddgg      } | |      }t        dd      }d}t        j                  t
        |      5  |j                  |       d d d        t        j                  ddgddgddgddgddgg      }|j                  |      }t        ||j                                t        j                  g d	      }t        j                  g d
      }t        j                  g d      }	 | ||	|ff      }|j                  |      }t        j                  ddgddgddgddgddgddgddgddgddgg	      }t        ||j                                t        dd      }t        j                  g d      }t        j                  g d      }t        j                  g d      }	 | ||	|ff      }|j                  |      }t        j                  ddgddgddgddgddgddgddgg      }t        ||j                                t        |j                         |j                  |      j                                t        dddd      }|j                  |      }t        ||j                                t        |j                         |j                  |      j                                y # 1 sw Y   axY w)Nr   r=   rW   Tro   )ignore_implicit_zerosr  z['ignore_implicit_zeros' takes effect only with sparse matrix. This parameter has no effect.r   )r   r   r=   r   rW   rW   r=   r   r=   rW   r   )r   r   r=   r=   r=   r=   r=   r=   r=   r=   r=   )r   rn   r   r=   rW   rX   rn   ro   rp   rk  r   r   r  rr   )	r<   r<   r=   r   r   r   r=   r<   r=   )	r   r   r=   r=   r=   r=   r=   r=   r=   )	r   rn   r   r=   rW   rX   rn   ro   rp   g      ?r   )r  r  r  random_state)rI   rt   r   r^   r  r  ra   r  r+   rB   r   )
r  rf   r  r  r  
X_expectedr~  X_dataX_colX_rows
             rF   +test_quantile_transform_sparse_ignore_zerosr  o  s   
1a&1a&1a&1a&1a&9:AQH%DaPK	 
 
k	9 
: Aq6Aq6Aq6Aq6Aq6BCJ''1G
GOO$56 XX78FHH67EHH67Efuen56H''1G#J#J#J#J#J#J#J#J#J
	
J 
GOO$56%DaPKXX45FHH01EHH01Efuen56H''1G
Q!UaZ!UaVaVaVLJ 
GOO$56K99'BJJL
 &"QQK ''1G
GOO$56K99'BJJLc 
:	9s   KKc                  2   t        j                  g dg dg dg dg dg      } t        d      }|j                  |        |j	                  |       }t        j
                  t        j                  dd	d
      d      j                  }t        t        j                  |d      |       t        j                  g dg dg      }t        j                  g dg dg      }t        |j                  |      |       |j                  |      }t        | |       y )N)r   rW   r  )r  rn   r  )r"  rp   r   )r  r   r  )r   r@   r  ro   r  r   r=   )num)rX   r=   r   )r<   r=   r   )e      r@   )r   r   r   r=   r=   r=   )rI   rt   r   ra   r  tilelinspaceTr+   sortr,   r   r   )rf   r  r~  r  r   r  s         rF   !test_quantile_transform_dense_toyr    s    
	lL,O	A &!4KOOA ''*GQq16:<<Ja0*=XX	
F 	
J k33F;ZH//8Ka-rG   c                     d} d}t        j                  t         j                  j                  | df      d      }d}g }t	        |      D ]  }t        ||| dz        }|j                  |       t        j                  dd|      t        j                  |j                        z
  }t        j                  t        j                  |            }|d	k  sJ |j                  |        t        t        j                  |            t        |      k(  sJ t        j                   | dd
dd      }g }t	        |      D ]  }t        ||| dz        }|j                  |       t        j                  dd|      t        j                  |j                        z
  }t        j                  t        j                  |            }|dk  sJ |j                  |        t        t        j                  |            t        |      k(  sJ y )N@B r;   r=   r   r   ro   r@   )r  r  r  {Gz?gGz?csc)r  formatr  r  )rI   r  r   sampler   r   ra   r  r   
quantiles_ra  r  appendr  uniquer   rD  )	rd   r  rf   ROUNDinf_norm_arrr  r  diffinf_norms	            rF   #test_quantile_transform_subsamplingr*    s    IK
		  )Q0q9AELe)%#2o

 	{{1a-9O9O0PP66"&&,'$H% % ryy&'3|+<<<< 	Iq$u1MALe)%#2o

 	{{1a-9O9O0PP66"&&,'$H% % ryy&'3|+<<<<rG   c                     t         j                  j                  d      j                  d      } d}t	        |d      j                  |       }t        j                  dd|      }t        |j                  |       t        j                  | j                         |      }t        |j                  j                         |       y)zCCheck the behaviour of `QuantileTransformer` when `subsample=None`.r   )   r=   r>   ro   N)r  r  r=   )rI   r   r   r  r   ra   r  r)   references_quantiler   r#  )rf   r  r  expected_referencesexpected_quantiless        rF   ,test_quantile_transform_subsampling_disabledr1     s    
		a ''X'6AK%+NRRSTUK++aK8K++-@AQWWY0CDK**0024FGrG   c                    t        j                  g dg dg dg dg dg dg dg dg d	g d
g
      } | |      }t        d      }|j                  |       |j	                  |      }t        t        j                  |j                         d      d       t        t        j                  |j                         d      d       |j                  |      }t        |j                         |j                                t        d      j                  |j                               }|j                  |      }t        t        j                  |j                         d      d       t        t        j                  |j                         d      d       |j                  |      }t        |j                         |j                                y )N)r   rq   r   )g      9@      @r   )g      I@r   r  )r   r   r  )r         @r   )r   g       @r   )g     R@r   r   )r   g      $@r   )r   r   r  )r   r   r  r@   r  r   r   r   rr   )rI   rt   r   ra   r  r,   r}  rB   ra  r   r   )r  rf   r  r~  r  r  s         rF   "test_quantile_transform_sparse_toyr5    sS   
	
	A 	aA%"5KOOA''*GbffW__%6Q?EbffW__%6Q?E//8Kaiik;+>+>+@A+;??		L))!,GbffW__%6Q?EbffW__%6Q?E#55g>Kaiik;+>+>+@ArG   c                      t        j                  g dg dg dg      } t        | j                  dd      }t        | dd      }t	        ||j                         y )N)r   r  r"  r  r   )rW   rn   rp   r   r@   )r  r  r   r  r  r   ro   )r   r  r=   )rI   rt   r   r  r,   )rf   
X_trans_a0
X_trans_a1s      rF   test_quantile_transform_axis1r9  4  sH    
&(8:STUA#ACCaQ?J#AA1=Jj*,,7rG   c                    t        j                  ddgddgddgg      } | |      }t        dd      j                  |      }t	        ||       t        dd      j                  |      }t	        |j                         |       t	        ||j                                t        j                  ddgddgddgg      }t        j                  ddgddgddgg      }t        d      j                  |      }|j                  |      }t	        ||       t         j                  j                  d      }t               }|j                  |       |j                  d	gg      |j                  t        j                  |      gg      k(  sJ |j                  d
gg      |j                  t        j                  |      gg      k(  sJ |j                  d	gg      |j                  t        j                  |j                        gg      k(  sJ |j                  d
gg      |j                  t        j                  |j                        gg      k(  sJ y )Nr   r=   rX   r  r  r  r  r  )r;   r=   r@   )rI   rt   r   r  r,   rB   ra   r   r   r}  ra  r   r-  )r  r  r  r~  
X_trans_sprf   X1r  s           rF   test_quantile_transform_boundsr?  <  s'    hhAAA/0GW%H "aa@NNwWGgw/$CQQJ j002G<gz'9'9';< 	1a&1c(QF+,A	As8aX3x0	1B%!488;K##B'Ggr* 			#A%'KOOA  3%)[-B-BRVVAYK=-QQQQ  2$(K,A,ABFF1I;-,PPPP((3%1[5R5R
&&((
)	*+6    ((2$0K4Q4Q
&&((
)	*+5   rG   c            	         t         j                  } t        j                  dgt        dz  gdgdgdgdgdgg      }| |fD ]?  }t        dd	      }|j                  |      }|j                  |      }t        ||d
       A y )Nr   r@   r   rW   rX   rn   r;   r   r;  	   r  )	r{  r|  rI   rt   r    r   r  r   r,   )X_1X_2rf   r  r~  r  s         rF   #test_quantile_transform_and_inverserD  b  s    
))C
((SE,r12SEA3aS1#N
OC3Z)dK++A.!33G<!![!<	 rG   c                     t        j                  t         j                  dddgt         j                  t         j                  ddgt         j                  dddgg      } t        dd      }|j	                  |        t        j
                  |j                  d d df         j                         sJ t        j
                  |j                  d d dd f         j                         rJ y )Nr   r=   r  r@   r#  r;  )	rI   rt   ru   r   r  r  r#  r   r   )rf   r  s     rF   test_quantile_transform_nanrF  l  s    
2661a#bffbffa%=1a?PQRA%"2FKa  88K**1a40155777xx..q!"u56::<<<<rG   
array_typert   r   c                 @   t        j                  g ddz        }d|j                  dd      z  }t        ||       }d}t	        |      j                  |      }|j                  d d df   }t        |      dk(  sJ t        t        j                  |      dk\        sJ y )	N)r   r=   r=   rW   rW   rX   rX   rn   ro   ro   r=   r=   rA  rA  rA  r   r   rk  r@   r  r<   r=   r   r  r   )
rI   rt   r;  r(   r   ra   r#  r  r   r(  )rG  rf   r  qt	quantiless        rF   *test_quantile_transformer_sorted_quantilesrK  x  s     	G"LMAaiiAA1j)AK		5	9	9!	<B ad#Iy>S   rwwy!Q&'''rG   c                      dD ]Q  } t        |       }t        j                  t        d      5  |j	                  t
        j                         d d d        S y # 1 sw Y   ^xY w)N))r<   r  )r  )r@   r  )g      Y@r  )r  r"  r  zInvalid quantile range: \(r   )r   r^   r_   r`   ra   r{  r|  )range_rh   s     rF    test_robust_scaler_invalid_rangerO    sL     V4]]:-JKJJtyy! LK LKs    AA!	c                    t         j                  j                  d      }|j                  dd      }d|d d df<    | |      }t	        |d      }t        j
                  t        j                  |            rJ t	        |d      }t        j
                  t        j                  |j                              rJ t	        |j                         d      }t        ||j                                t        j                  t              5  t	        |dd	       d d d        t        |j                  d
      g dd       t        |j                  d
      g d       ||usJ t!        |d      \  }}t        ||j                  d
             t        ||j                  d
             t	        |ddd      }t        |j                         |j                                y # 1 sw Y   xY w)Nr#  rn   ro   r   r   Fr|   r=   )r}   r   r   r  rW   r  TrC  )rI   r   r   r]   r   r   r  r|  r  r,   rB   r^   r_   r`   r   r   r9   )	r  r\   rf   X_csrr   X_csr_scaledX_csc_scaledX_csr_scaled_meanX_csr_scaled_stds	            rF   %test_scale_function_without_centeringrV    s   
))


#C		!QAAadG!EQ%(Hvvbhhx()))%0Lvvbhh|001222 %8Lh(<(<(>? 
z	"eu1- 
# 1? hlll24MN1*<\1*M''/A1FG.!0DE %%dKLemmo|/C/C/EF! 
#	"s   	G""G+c                      t         j                  } t        | d      }t        t	        j
                  |d      d       t	        j                  |dd      }|d   |d   z
  }t        |d       y )Nr=   r   r   r  r  r{  r|  r   r,   rI   r  r  rf   r~  r  r  s       rF   test_robust_scale_axis1rZ    sY    		A11%Gbiia8!<
g2A
A$1+Cc1%rG   c                      t         j                  d d df   } t        |       }t        t	        j
                  |      d       t	        j                  |d      }|d   |d   z
  }t        |d       y )Nr=   r   r  )r  rX  rY  s       rF   test_robust_scale_1d_arrayr\    sZ    		!Q$A1oGbii0!4
g*A
A$1+Cc1%rG   c                  "   g dg dg dg} t               }|j                  |       }g dg dg dg}t        ||       |j                  |      }t        | |       g dg dg d	g}|j	                  |      }g d
g dg dg}t        ||d       y )Nr  r  r  r  )r   r   r  r  r  r  r  )r   rr   r   )r  r   g_)r   r   gіs)?rX   r  )r   r  r,   r   r   )rf   rh   r~  r  r  r  r  X_expected_news           rF   )test_robust_scaler_zero_variance_featuresr_    s    	+-=>A^F""1%G #$46FGJgz2**73Ka- /1ABE""5)K')>@UVNk>1ErG   c                  8   t         j                  j                  d      } | j                  dd      }t        j                  |t        j
                  d      dz  t        j
                  d      dz  g      }d}t        |d	      j                  |      }|j                  |      }|j                  t        j                  d
d      k(  sJ |j                  t        j                  dd      k(  sJ |j                         t        j                  dd      k(  sJ y )Nr#  r  r=   )r   r=   r   i)r=   c   T)r  unit_variancer   MbP?)r  r  )rI   r   r   r]   vstackr   r   ra   r   r  r^   rL  r   r   )r\   rf   X_with_outliersr  robust_scalerr~  s         rF    test_robust_scaler_unit_variancerg    s     ))


#C		'1AiiBGGH$5$;RWWX=NQU=U VWON dSWWM %%a(G  FMM!$>>>>6==#====;;=FMM!6666rG   c                 4   g dg dg dg dg}t               }|j                  |      }g dg dg dg dg}t        ||       |j                  |      }t        ||       g dg d	g dg}|j	                  |      }g d
g d	g dg}t        ||d       t        |      }t        ||        | |      }	|j                  |	      }
g dg dg dg dg}t        |
j                         |       |j                  |
      }t        ||j                                y )Nr  )r   rr   333333ӿr  r  )r   rr   UUUUUU?)r   rr   gɿ)r   rr   rr   r  r  )r   rq   rj  rW   r  )r   r  r,   r   r   r   rB   )r   rf   rh   r~  r  r  r  r  r^  r  r  X_trans_sparse_invs               rF   )test_maxabs_scaler_zero_variance_featuresrl    s    
+-=?OPA^F""1%G	J gz2**73Ka- /1ABE""5)K,.>@PQNk>1E 1oGgz2  "H))(3N	J n446
C11.Aa!3!;!;!=>rG   c                      g dg dg dg dg} t               }|j                  |       }g dg dg dg dg}t        ||       y )	N)r   rr   r  r  )r   rr   ri  ry  )r   rr   g      Yr   )r   r   r   g       )r   rr   g{Gzt?ry  )r   rr   g~jthg      п)r   rr   r  r   )r   r   r   r  )r   r  r,   )rf   rh   r~  r  s       rF   'test_maxabs_scaler_large_negative_valuern  &  sO     			A ^F""1%G!	J gz2rG   c                 J    | g dg      }t               }|j                  |      }|j                  |      } | g dg      }t        |j	                         |j	                                |j                  |      }t        |j	                         |j	                                y )N)r  rr   rr   r  )r   ra   r   r,   rB   r   )r  rf   rh   r~  r  r   s         rF   (test_maxabs_scaler_transform_one_row_csrrp  :  s     	'(A^FZZ]Fq!G01Jgoo/1C1C1EF,,W5Maiik=+@+@+BCrG   c                     t         t        t        t        fD ]  } t        d      }|j	                  |       j                  |       }t        | t              rt        j                  |       } t        |       dk(  rGt        t        j                  |j                  d            t        j                  t                     n/t        t        j                  |j                  d            d       |j                   | j"                  d   k(  sJ |j%                  |      }t        ||         t        j                  d      } t               }|j	                  |       j                  |       }t        t        j                  |j                  d            d       |j                   | j"                  d   k(  sJ t         j'                         }t        j                  |      j                         }t        ||z  t)        |d             y )NTr   r=   r   r   rr   r   )r   r   r   r   ra   r   r   r   rI   rt   rL   r,   r  ra  r   re   r   rK   r   r   r   )rf   rh   r   r   r  max_abss         rF   test_maxabs_scaler_1drs  G  sx   fk;74(::a=**1-aAA!#%bffX\\q\-A&BBGGJDWX%bffX\\q\-A&BCH%%333 00:!-3 8$ 	A^Fzz!}&&q)HbffX\\q\%9:C@!!QWWQZ/// <<>DffTl GdWnl4d.KLrG   c           
         t         d dd d f   }|j                  d   }ddd||dz   fD ]  }t               j                  |      }t               }t               }t               }t	        ||      D ]N  }|j                  ||         } | ||         }	|j                  |	      } | ||         }
|j                  |
      }P t        |j                  |j                         t        |j                  |j                         t        |j                  |j                         |j                  |j                  k(  sJ |j                  |j                  k(  sJ |j                  |j                  k(  sJ t        |j                  |j                         t        |j                  |j                         t        |j                  |j                         t        |j                  |      |j                  |             t        d|      }t               j                  ||         }t               j                  ||         }t        |j                  |j                         |j                  |j                  k(  sJ t        |j                  |j                         t        |j                  |      |j                  |             t               j                  |      }t               }t        t	        ||            D ]H  \  }}|j                  ||         }t        ||j                  |j                  |||j                         J ! y )Nr   r   r=   rW   r"  r#  r$  )r%  rK   r   ra   r#   r&  r,   max_abs_r   r   r   r+  r,  rU   r-  r.  )r  rf   rR   rS   r0  r1  scaler_incr_csrscaler_incr_cscr2  rQ  X_cscr3  rO   s                rF   test_maxabs_scaler_partial_fitry  g  s    	TcT1WA	
A!RAF+
#~))!,"n&.&. J/E%11!E(;K!!E(+E-99%@O!!E(+E-99%@O 0 	","7"79M9MN!,"7"79Q9QR!,"7"79Q9QR++{/J/JJJJ++/N/NNNN++/N/NNNN!,"5"5{7I7IJ!,"5"57M7MN!,"5"57M7MN!,"8"8";[=R=RST=UV q*%#~))!F)4"n006;!,"7"79M9MN++{/J/JJJJ!,"5"5{7I7IJ!,"8"8";[=R=RST=UV $~))!,"n!+a"<=HAu%11!E(;K!KK ::%*:: >M ,rG   c                    | dk(  rTt        j                  |      j                  d      }t        d      D ]  }t	        ||   d        t	        |d   d       y	| dk(  rUt        d      D ]$  }t	        t        j                  ||         d       & t	        t        j                  |d         d       y	| dk(  rJt        |      j                  d      }t        d      D ]  }t	        ||   d        t	        |d   d       y	y	)
zr
    Convenient checking function for `test_normalizer_l1_l2_max` and
    `test_normalizer_l1_l2_max_non_csr`
    r]  r=   r   rX   rr   r   r`  ra  N)rI   r  rK  r   r+   lar_  ra  )r_  X_normrow_sumsrO   row_maxss        rF   check_normalizerr    s    
 t|66&>%%1%-qAS1 HQK-	qAq	 2C8 BGGF1I.4	v;???*qAS1 HQK-	 
rG   r_  r]  r`  ra  c                    t         j                  j                  d      }|j                  dd      } ||      }d|dd d f<   |j                  d   }|j                  d   }d|j
                  ||  ||      }|||fD ]u  }t        | d      }	|	j                  |      }
|
|usJ t        |
      }
t        | d      }	|	j                  |      }||u sJ t        |      }|
|fD ]  }t        | |        w y )	Nr   rn   ro   r   rX   Tr_  r   F)
rI   r   r   r]   indptrr|  r   r   rB   r  )r_  r  r\   r  X_sparse_unprunedindptr_3indptr_4X_sparse_prunedrf   
normalizerX_norm1X_norm2r|  s                rF   test_normalizer_l1_l2_maxr    s    ))


"Cii1oG%g. GAqDM !''*H ''*H038H- $G,O (9:T5
&&q)a'"T6
&&q)!||'"(FT6* ) ;rG   c                 F   t         j                  j                  d      }|j                  dd      }d|dd d f<    ||      }t	        | d      j                  |      }||usJ t        j                  |      r|j                  dk(  sJ t        |      }t        | |       y )	Nr   rn   ro   r   rX   Fr  csr)rI   r   r   r]   r   r   r   r  r!  rB   r  )r_  r   r\   r  rf   r|  s         rF   !test_normalizer_l1_l2_max_non_csrr    s    
 ))


"Cii1oG GAqDM!AT.88;F????6"v}}'===V_FT6"rG   c           	         t         j                  j                  d      }|j                  dd      }d|dd d f<   |dt	        |dd d f         j                         fxx   dz  cc<   t        j                  |       } | |      }|||fD ]k  }t        d	      }|j                  |      }||usJ t        |      }t        t        j                  |      t        j                  t        |                   m y )
Nr   rn   ro   r   rX   rW   r<   ra  r^  )rI   r   r   r]   r  argmaxr   r   rB   r-   sign)r  r\   r  	X_all_negX_all_neg_sparserf   r  r|  s           rF   test_normalizer_max_signr    s     ))


"Cii1oGGAqDM As71a4=!((**+r1+ I$Y/y"23U+
%%a(Q2776?BGGGAJ,?@ 4rG   c                 *   t         j                  j                  d      j                  dd      }t	        t        |d      t        |j                  dd      j                         t         j                  j                  d      }|j                  dd	      } | |      }t        j                  d      }||fD ]  }t         j                  t         j                  fD ]  }d
D ]  }|j                  |      }t        ||      }|j                  |k(  sJ t        |      }|dk(  r&t        j                  |      j                  d      }	n|dz  }
|
j                  d      }	t        |	|          t        j                   g dg dg dg      }dD ]  }t        ||d      \  }}|dk(  r"t        |t        j                   g d             ;|dk(  r"t        |t        j                   g d             bt        |t        j                   g d               | |      }d
D ]2  }t#        j$                  t&              5  t        ||d       d d d        4 t        |dd      \  }}t        |t        j                   g d             y # 1 sw Y   qxY w)N%   rX   rW   Fr   r   )r   r   r@   ro   )r]  r`  r^  r]  r=   r   )r   r   r3  )rr   r   r   )rq   r   r   r  T)r_  return_norm)      @rr   r   r`  )r   rr   g1C+@)r3  rr   r   ra  )rI   r   r   r]   r-   r   r  r   r   r   r   r   rB   r  rK  r,   rt   r^   r_   NotImplementedError)r  rf   rsr  r  r   r   r_  r|  r}  X_norm_squared_normss                rF   test_normalizer    s    			b!''1-Ay/133QU1S1U1UV			q	!Bhhr1oGW%H77B=Dx jj"**-E$HHUO"140||u,,, 4<!vvf~11q19H%+QYN-11q19H)(D9 % . !" hhHIG#W4TB54<%eRXXo-FGT\%eRXX6L-MN%eRXXo-FG $ W%H]]./hTt< 0/  4@HAueRXXo%>? 0/s   =J		J	constructorc                    t        j                  g dg dg      } | |j                               }t        dd      }t	        |j                  |            }t        j                  |dk(        dk(  sJ t        j                  |dk(        d	k(  sJ |j                  |      }t        j                  |      t        j                  |      k(  sJ t        d
      j                  |      }t	        |j                  |            }||usJ t        j                  |dk(        d	k(  sJ t        j                  |dk(        dk(  sJ t        d
      }|j                  |      }||usJ t	        |      }t        j                  |dk(        d	k(  sJ t        j                  |dk(        dk(  sJ t        d
      }|j                  |      }| t        ur||u sJ t        d
      }t        j                  g dg dgt         j                        }|j                  |      }| t        ur||u sJ t	        |      }t        j                  |dk(        d	k(  sJ t        j                  |dk(        dk(  sJ t        dd      }| t         j                  t        fv r{ | |j                               }t	        |j                  |            }t        j                  |dk(        dk(  sJ t        j                  |dk(        dk(  sJ |j                  |      }| t        v r:t        j                  t              5  |j                   | |             d d d        y y # 1 sw Y   y xY w)N)r=   r   ro   )rW   rX   r<   rq   T)	thresholdr   r   rn   r=   rW   r   Fr   ry  ro   )rI   rt   r   r   rB   r   rK  r   r  ra   r   r   r4   r^   r_   r`   )r  X_rf   	binarizerX_binX_floats         rF   test_binarizerr  /  s    
9j)	*BBGGIACd3II''*+E66%1*"""66%1*""""E??1!7777t$((+II''*+E>>66%1*"""66%1*"""t$I"E>>ENE66%1*"""66%1*"""u%I"E$zzu%Ihh	:.bjjAG(E$ENE66%1*"""66%1*"""Dt4Irxx&&	"	++A./vveqj!Q&&&vveqj!Q&&&##A& n$]]:&A/ '& %&&s   5MM c                    t        | ||      \  }}|ddfD ]  }t        j                  t        j                  g d|      d      }|j                  ||      }t	        d      j                  |      }t        d	
      5  t	        d      j                  |      }	d d d        t        t        	t        d      |        y # 1 sw Y   (xY w)NrV  rl  )r   r=   rW   rX   rn   r   )r<   r=   r   r   )r  Tr   r   r   )	r'   rI   r;  rJ   r   r  r   r-   r%   )
rd  r   r   r   r   dtype_name_rm  r   binarized_npbinarized_xps
             rF   test_binarizer_array_api_intr  h  s     &o{JOJB"GW5zz"**_KH'Rzz$vz. 3/==dCt4$s3AA$GL 57<BuE|T 6 54s    CC	c                     t         j                  j                  d      } | j                  d      }t	        d      }|j                  |       |j                  |      }t        j                  ||j                        }t               }t        j                  ||j                        }|j                  |      }t        ||       | j                  d      }t        j                  ||j                        }	|j                  |      }
t        j                  |
|j                        }|j                  |	      }t        ||       t        j                  |      |j                  d   z  }|||z  z
  ||z  z
  ||z  |z  z   }t        ||       t        j                  |	      |j                  d   z  }|	||z  z
  |	|z  z
  ||z  |z  z   }t        ||       y )Nr   ro   rn   Fr   )rW   rn   )rI   r   r   random_sampler   ra   r   dotr  r   r  r,   	ones_likerK   r)   )r\   X_fitrh   X_fit_centeredK_fitcentererK_fit_centeredK_fit_centered2X_predK_predX_pred_centeredK_pred_centeredK_pred_centered2ones_MK_fit_centered3ones_prime_MK_pred_centered3s                    rF   test_center_kernelr  x  s    ))


"Cf%EU+F
JJu%%e,NFF5%''"E HVVNN,<,<=N,,U3Ono> v&FVVFEGG$F&&v.Off_n.>.>?O))&1o/?@ \\% 5;;q>1Ffun,uv~=QW@WWONO4 <<'%++a.8L%%7,:NQW:WW  O%56rG   c                     t         j                  j                  d      } | j                  dd      | j                  dd      }}d } ||      } ||      }t	        d      }|j                  |      }|j                  |      }||j                  z  }	||j                  z  }
||j                  z  }||j                  z  }t               }|j                  |	       t        |j                  |	      |       t        |j                  |
      |       t        j                  |	      |	j                  d   z  }|	||	z  z
  |	|z  z
  ||	z  |z  z   }t        |j                  |	      |       t        j                  |
      |	j                  d   z  }|
||	z  z
  |
|z  z
  ||	z  |z  z   }t        |j                  |
      |       y)	z-Check kernel centering for non-linear kernel.r   r   r"  r  c                     t        j                  t        j                  | dd      t        j                  | dd       g      S )zOur mapping function phi.r   N)a_mina_max)rI   rd  r\  r  s    rF   phiz2test_kernelcenterer_non_linear_kernel.<locals>.phi  s;    yy$/$a00
 	
rG   Fr   N)rI   r   r   r]   r   r  r   r  r   ra   r)   r  rK   )r\   rf   r   r  phi_X
phi_X_testrh   phi_X_centerphi_X_test_centerKK_testK_centerK_test_centerkernel_centererr  
K_centeredr  K_test_centereds                     rF   %test_kernelcenterer_non_linear_kernelr    s   
))


"C		#r"CIIb"$5vA
 FEVJ U+F''.L((4 	A%''!Flnn,H%6M$&OO--a0(;O--f5}E \\!_qwwqz)FVaZ!f*,vzF/BBJO--a0*= <<'!''!*4L!!FVO3lQ6F6OO  O--f5GrG   c                  n   t        j                  g dg dg dg dg      } t        j                  d      }| j                  | j                        }t               }t        d|fdt               fg      }|j                         j                  j                  sJ t        |||d	      }t        ||       y )
N)rX   r   r   )r   rX   r   )r   r   rX   r  )rn   r  svrrW   )cv)rI   rt   r   r  r  r   r   r"   __sklearn_tags__
input_tagspairwiser   r,   )rf   y_truer  kcentpipeliney_preds         rF   test_cv_pipeline_precomputedr    s     	)Y	9=>AWWT]F	acc
AE+U3eSU^DEH $$&11::::
 xFq9Fff-rG   c                  "   t         j                  j                  d      } | j                  d      }t	               t               t               fD ]?  }|j                  |      j                  |      }|j                  |      }t        ||       A y )Nr   r  )rI   r   r   r  r   r   r   ra   r   r  r-   )r\   rf   objX_transformedX_transformed2s        rF   test_fit_transformr    sp    
))


"C&!A *,	<
,,Q/**1-=.9 =rG   c                  Z    ddgddgddgg} t        |       } t        | g dg dg dg       y Nr=   r   )r=   r=   r   rs   )r   r-   r  s    rF   test_add_dummy_featurer    s5    
Q!Q!Q A!Aq9i;<rG   c                      | ddgddgddgg      }|j                   }t        |      }t        j                  |      r|j                   |k(  sJ |       t	        |j                         g dg dg dg       y r  )r!  r   r   r  r-   rB   )r   rf   desired_formats      rF   test_add_dummy_feature_sparser    sm     	1a&1a&1a&12AXXN!A??1!((n"<?a?<qyy{Y	9$EFrG   c                      t         j                  } | d d d df   }t        dd      t               t	               g}|D ]$  }|j                  |        |j                  |       & y )NrW   Fr   )r{  r|  r   r   r   r  )rf   r%  scalersrh   s       rF   test_fit_cold_startr  	  sc    		AQU8D 	7G Q 	T"	 rG   methodbox-coxyeo-johnsonc                 J   t        |       }t        j                  t              }t	        j
                  t              5  |j                  |       d d d        t	        j
                  t              5  |j                  |       d d d        y # 1 sw Y   =xY w# 1 sw Y   y xY w)Nr  )	r   rI   r  r   r^   r_   r
   r   r   )r  ptrf   s      rF    test_power_transformer_notfittedr  	  sg    		(B
vA	~	&
Q 
'	~	&
Q 
'	& 
'	&	&	&s   B2BBB"standardizec                     | dk(  rt        j                  |      n|}t        | |      }|j                  |      }t	        ||j                  |             y )Nr  r  r  )rI   r  r   r  r+   r   )r  r  rf   r  r~  s        rF   test_power_transformer_inverser  $	  sK     y(q	aA	[	ABq!G2//89rG   c                  x   t        j                  t              } dD ]  }t        d|      }|j	                  |       }t        | d|      }t        j                  | j                               \  }}|rt        |      }t        |j                  dd      |       t        |j                  dd      |       t        | |j                  |             t        ||j                  d          t        |j                        | j                  d   k(  sJ t!        |j                  t         j"                        rJ  y )NTFr  r  r<   r=   r   )rI   r  r   r   r  r   r   boxcoxflattenr   r+   r;  r   lambdas_r  rK   r   r  )rf   r  r  r~  X_trans_funcr  lambda_expecteds          rF   test_power_transformer_1dr  0	  s    
vA$YKH""1%&qT&+ll199;&?#
Oz*JJ..r15w?J..r15|DAr33G<=OR[[^<2;;1771:---"++rzz222% %rG   c                     t        j                  t              } dD ]%  }t        d|      }|j	                  |       }t        | d|      }||fD ]  }t        |j                  d         D ]h  }t        j                  | d d |f   j                               \  }}|rt        |      }t        |d d |f   |       t        ||j                  |          j |j                  |      }	t        |	|         t!        |j                        | j                  d   k(  sJ t#        |j                  t         j$                        r&J  y )Nr  r  r  r=   )rI   r  r%  r   r  r   r   rK   r   r  r  r   r+   r  r   r,   r  r   r  )
rf   r  r  X_trans_classr  r~  jr  lmbdaX_invs
             rF   test_power_transformer_2dr  H	  s   
tA$YKH((+&qT%|4G7==+,$)LL1a41B$C!
E!&z!2J#GAqDM:>#E2;;q>: - ((1E%eQ/ 5 2;;1771:---"++rzz222+ %rG   c                  D   t        d      } | j                  t        j                  t                     t        }d}t        j                  t        |      5  | j                  |       d d d        t        j                  t        |      5  | j                  |       d d d        t        j                  t        |      5  t        |d       d d d        t        j                  t        |      5  | j                  t        j                  t        j                               d d d        t        j                  t        |      5  | j                  t        j                  t        j                               d d d        t        j                  t        |      5  t        t        j                  t        j                        d       d d d        y # 1 sw Y   oxY w# 1 sw Y   GxY w# 1 sw Y   #xY w# 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   y xY w)Nr  r  zstrictly positiver   )r   ra   rI   r  r%  r^   r_   r`   r   r   r   rK   )r  X_with_negativesnot_positive_messages      rF   9test_power_transformer_boxcox_strictly_positive_exceptionr  c	  sW    
	+BFF266$<.	z)=	>
%& 
? 
z)=	>
  
? 
z)=	>(; 
? 
z)=	>
RXXdjj)* 
? 
z)=	>
rxx

#$ 
? 
z)=	>,Y? 
?	> 
?	> 
?	> 
?	> 
?	> 
?	> 
?	>sH   GG$G133G>	3H
/HG!$G.1G;>H
HHc                     t        | d       y )Nr  r  )r   r  s    rF   +test_power_transformer_yeojohnson_any_inputr	  	  s     Am,rG   c                    t        |       }t        j                  t              }|j	                  |       d}t        j                  t        |      5  |j                  |d d ddf          d d d        t        j                  t        |      5  |j                  |d d ddf          d d d        y # 1 sw Y   HxY w# 1 sw Y   y xY w)Nr  zBX has \d+ features, but PowerTransformer is expecting \d+ featuresr   r   r=   )
r   rI   r  r%  ra   r^   r_   r`   r   r   )r  r  rf   wrong_shape_messages       rF   &test_power_transformer_shape_exceptionr  	  s    		(B
tAFF1I
 	N  
z)<	=
Qq!A#vY 
> 
z)<	=
Qq!A#vY' 
>	= 
>	= 
>	=s   B6C6B?Cc                     t        j                  t              d d ddf   } t        dd      j	                  |       }t        j
                  dg      |_        |j                  |       }t        |j                  |      |        y )Nr   r=   r  Fr  )
rI   r  r%  r   ra   rt   r  r   r,   r   )rf   r  r~  s      rF   "test_power_transformer_lambda_zeror  	  sh    
tQ!VA		>	B	B1	EB ((A3-BKll1oGb227;Q?rG   c                      t        dd      } t        j                  t              d d ddf   }t        j                  dg      | _        | j                  |      }t        ||       y )Nr  Fr  r   r=   )r   rI   r  r%  rt   r  r   r,   )r  rf   r~  s      rF   !test_power_transformer_lambda_oner  	  sP    	E	BB
tQ!VA((A3-BKll1oGgq)rG   zmethod, lmbda))r  r  )r  r  )r  r  )r  r  )r  rr   c                 T   t         j                  j                  d      }d}|j                  dd|df      }| dk(  rt        j                  |d|z  dz   d       }t        | d	      j                  t        j                  |            }|g|_        |j                  |      }t        | d	      }|j                  |      }t        dt         j                  j                  ||z
        |z  d
       t        d|j                         d       t        d|j                         d       y )Nr   i N  r=   )locr   r?   r  r<   r   Fr  rW   r  )rI   r   r   r  r\  r   ra   r  r  r   r  r+   linalgr_  r   r   )r  r  r\   rd   rf   r  r  X_inv_transs           rF   #test_optimization_power_transformerr  	  s    " ))


"CI

qA
7A GGArEzD($/	U	;	?	?q		JB'BK  #E	U	;B""5)K299>>!k/:YFPQR;++-q9;??,a8rG   c                      t        dd      j                  dgdgg      } dg| _        | j                  dgg      }t	        j
                  |      sJ y )Nr  Fr  rr   rq   r  g )r   ra   r  r   rI   r  )r  r  s     rF   test_invserse_box_coxr  	  sN    		>	B	BSEC5>	RB%BK  4&*E88E??rG   c                      g d} t        j                  |       j                  dd      } t        d      j	                  |       j
                  }t        j                  |dd      sJ y )	N)gffffff@g rr   rq   gffffff?g333333@r   gffffff@g?g@r  r   g"@g      @g      r<   r=   r  r  gzG?rc  r   )rI   rt   r;  r   ra   r  allclose)rf   r  s     rF   test_yeo_johnson_darwin_exampler  	  sU     	VA
B"AM266q9BBE;;ue$///rG   c                    t        j                  t              }t        |       }|j	                  |       |j
                  d   }t        j                  |t        j                  |t         j                        g      }t        |d      }|j	                  |       |j
                  d   }t        ||d       |j                  |      }t        t        j                  |      t        j                  |             y )Nr  r   )r  ro   r  )rI   r  r   r   ra   r  concatenate	full_likeru   r$   r+   r   r-   r  )r  rf   r  lmbda_no_nans
lmbda_nansr~  s         rF   test_power_transformer_nansr   	  s    
 	vA		(BFF1IKKNM 	2<<266234A"AFF1IQJz1=ll1oGrxx("((1+6rG   c                     t         }| dk(  rt        j                  |      }t        | |      }t	        |j                  |      j                  |      |j                  |             y )Nr  )r  )r   rI   r  r   r,   ra   r   r  )r  r  rf   r  s       rF   $test_power_transformer_fit_transformr"  	  sR     	AFF1I	&k	:BbffQi11!4b6F6Fq6IJrG   c                    t         }| dk(  rt        j                  |      }|j                         }||usJ t	        ||       t        | |d      }|j                  |       t	        ||       |j                  |      }||usJ |j                  |      }t	        ||       ||usJ |j                  |      }||usJ y )Nr  Tr  r   )
r   rI   r  r   r,   r   ra   r   r  r   r  r  rf   
X_originalr  r~  r  s          rF    test_power_transformer_copy_Truer'  
  s    
 	AFF1IJJa,	&k	EBFF1Ia,ll1oG!q!Ga,!&&w/K+%%%rG   c                    t         j                         }| dk(  rt        j                  |      }|j                         }||usJ t	        ||       t        | |d      }|j                  |       t	        ||       |j                  |      }||u sJ | dk(  rt        j                  |      }|j                  |      }||u sJ |j                  |      }||u sJ y )Nr  Fr$  )
r   r   rI   r  r,   r   ra   r   r  r   r%  s          rF   !test_power_transformer_copy_Falser)  "
  s    
 	AFF1IJJa,	&k	FBFF1Ia,ll1oGa<<FF1Iq!Ga<<&&w/Kk!!!rG   c                      t         j                  d      } t        j                  | dddf<   d}t	        d      }t        j                  t        |      5  |j                  |        ddd       y# 1 sw Y   yxY w)zqCheck that box-cox raises informative when a column contains all nans.

    Non-regression test for gh-26303
    r  Nr   zColumn must not be all nan.r  r  r   )	r\   r  rI   ru   r   r^   r_   r`   r  )rf   r  r  s      rF   1test_power_transformer_box_cox_raise_all_nans_colr+  @
  s\    
 	&!AffAadG+G		+B	z	1
 
2	1	1s   A11A:rC  )r@   r=   皙?)r  r\   )r   c                     t        dd      }t        d      }|j                  |      j                  |        t	        j
                  |j                  d         sJ y )Nr   r,  r  Fr|   r   )r7   r   ra   r&  rI   r  r   )rC  rB  rh   s      rF   7test_standard_scaler_sparse_partial_fit_finite_variancer.  O
  sJ     vs
3Ce,F
JJsO$;;v{{1~&&&rG   rx  )r   r=   )r<  r@   c                 X   t         j                  }t        | d      j                  |      }t	        j
                  |d      t	        j                  |d      }}t        j                  |d d dz
  |dd  dz   f   g}|j                  |      }t        || d   | d   | d   | d   gg       y )NT)rx  r\  r   r   rW   r@   r=   )
r{  r|  r   ra   rI   r}  ra  r_r   r)   )rx  rf   rh   X_minX_maxr   r  s          rF   test_minmax_scaler_clipr3  `
  s     			ADAEEaHF 66!!$bffQQ&75EeeE"1INE!"IN234F$$V,M

M!,mA.>a@P	QRrG   data_constructorc                     | t         j                        }t        j                  |      }t	        d      j                  |      }t        j                  t        j                  |      d      }|r|j                  n|} | t        j                  |d d dz   |dd   dz
  f      j                  dd            }|j                  |      }t        | | g d	g             y )
NTr[  r   r   rW   r@   r=   r<   )r=   r=   r<   r<   )r{  r|  r   r  r   ra   rI   ra  r  hstackr;  r   r*   )r4  rf   	is_sparserh   rr  r   r  s          rF   test_maxabs_scaler_clipr8  p
  s    
 	#A"It$((+F ffRVVAYQ'G'gllWG
		72A;#gabk\B%678@@BGF $$V,M 0@.AQ0RSrG   c                      t               j                  t              } d}t        j                  t
        |      5  | j                  t        dddf          ddd       y# 1 sw Y   yxY w)zCheck that `inverse_transform` from `StandardScaler` raises an error
    with 1D array.
    Non-regression test for:
    https://github.com/scikit-learn/scikit-learn/issues/19518
    z'Expected 2D array, got 1D array insteadr   Nr   )r   ra   r%  r^   r_   r`   r   )rh   r  s     rF   -test_standard_scaler_raise_error_for_1d_inputr:  
  sL     !!$'F7G	z	1  ad, 
2	1	1s   A!!A*c                     dt        j                  g ddz  g dz   t         j                        j                  dd      z  } t	               }t        j                         5  t        j                  dt               |j                  |       }d	d	d	       t        j                  t        j                              rJ |j                         t        j                  d
      k(  sJ |j                         t        j                  d      k(  sJ |j!                         dkD  sJ |j#                         dk  sJ y	# 1 sw Y   xY w)a  Check that significantly non-Gaussian data before transforms correctly.

    For some explored lambdas, the transformed data may be constant and will
    be rejected. Non-regression test for
    https://github.com/scikit-learn/scikit-learn/issues/14959
    r9  )rz  rq   r   r3  rn   )r     r<        r  U   r  r   r<   r=   r   Nr   rr   r  rW   )rI   rt   r   r;  r   r  r  r  RuntimeWarningr  r   r  r   r^   rL  r   r}  ra  )X_non_gaussianr  r~  s      rF   1test_power_transformer_significantly_non_gaussianrB  
  s     288q #CC2::gb!nN 
	B		 	 	"g~6"">2 
# vvbhhw'(((<<>V]]3////;;=FMM#....;;=2;;=1 
#	"s   !,D::ETransformerc                      |        j                  t        j                        }|j                  t        j                        }t        |t        j                         y)9Check one-to-one transformers give correct feature names.N)ra   r{  r|  get_feature_names_outfeature_namesr-   )rC  tr	names_outs      rF   test_one_to_one_featuresrJ  
  s@     
		499	%B((););<Iy$"4"45rG   c                 @   t        j                  d      }|j                  t        j                  t        j
                        } |        j                  |      }|j                         }t        |t        j
                         |j                  t        j
                        }t        |t        j
                         t        j                  d      }t        j                  t        |      5  t        d      }|j                  |       ddd       y# 1 sw Y   yxY w)rE  pandas)columnsz0input_features is not equal to feature_names_in_r   abcdN)r^   importorskip	DataFramer{  r|  rG  ra   rF  r-   reescaper_   r`   r   )rC  pddfrH  names_out_df_defaultnames_out_df_valid_inmsginvalid_namess           rF   test_one_to_one_features_pandasrY  
  s     
		X	&B	dii););	<B			2	B335+T-?-?@44T5G5GH,d.@.@A
))F
GC	z	-V
  / 
.	-	-s   .DDc                  @   t         j                  j                  d      } | j                  d      }t	        |      }t               j                  |      }|j                         }|j                  d   }t        |t        |      D cg c]  }d| 	 c}       yc c}w )z.Test that kernel centerer `feature_names_out`.r   )rp   rn   r=   kernelcentererN)rI   r   r   r  r   r   ra   rF  rK   r-   r   )r\   rf   
X_pairwiser  rI  samples_out2rO   s          rF   &test_kernel_centerer_feature_names_outr^  
  s     ))


"C&!Aq!J##J/H..0I##A&Ly|AT"UATA^A3#7AT"UV"Us   B
c                 8   g dg dg dg}t        d|       j                  |      }t        |j                  g d       |j	                  |      }|j                  |      }||fD ]2  }| r"t        |t        j                  |      d       't        ||       4 y)z?Check that PowerTransformer leaves constant features unchanged.)r  r   rW   r  r  r  g+=r   N)r   ra   r)   r  r  r   rI   r   )r  rf   r  XftXtXt_s         rF   'test_power_transformer_constant_featurerc  
  s     
Z,A	K	H	L	LQ	OBBKK+


1
C	aBRyCq!1>C#	 rG   zcno floating point exceptions, see https://github.com/numpy/numpy/pull/21895#issuecomment-1311525881)reasonc                     t        d      } t        j                  g d      j                  dd      }| j	                  |       | j
                  d   }|dk  sJ t        j                  dg      j                  dd      }t        j                  t        d	      5  | j                  |      j                         }d
d
d
       t        j                        sJ y
# 1 sw Y   !xY w)zCheck if a warning is triggered when the inverse transformations of the
    Box-Cox and Yeo-Johnson transformers return NaN values.r  r  )r=   r=   r   r<   r=   r   r@   zSome values in columnr   N)r   rI   rt   r;  ra   r  r^   r  r  r   itemr  )transr  r  psix_invs        rF   *test_yeo_johnson_inverse_transform_warningrj  
  s     M2E
&&r1-A	IIaLNN1E199 ((B4.
 
 Q
'C	k)@	A'',113 
B 88E?? 
B	As    CC%z1.12z2scipy version 1.12 required for stable yeo-johnsonc                      t        j                  g d      } d } || j                  dd              || dd j                  dd             y)a-  Verify that PowerTransformer operates without raising any warnings on valid data.

    This test addresses numerical issues with floating point numbers (mostly
    overflows) with the Yeo-Johnson transform, see
    https://github.com/scikit-learn/scikit-learn/issues/23319#issuecomment-1464933635
    )
g     L@g     x@g     4@g     @@     d@rl  g     @g     <@g     \@g     @c                     t        j                  d      5 }t        j                  d       t        dd      j	                  |        ddd       r J ddj                  d	 |D              z          y# 1 sw Y   ,xY w)
z0Internal helper to test for unexpected warnings.Tr  alwaysr  r  Nz!Unexpected warnings were raised:

c              3   F   K   | ]  }t        |j                          y wr  )strmessage).0ws     rF   	<genexpr>zPtest_power_transformer_no_warnings.<locals>._test_no_warnings.<locals>.<genexpr>2  s      U
$3qC		NOs   !)r  r  r  r   r  join)r|  caught_warningss     rF   _test_no_warningsz=test_power_transformer_no_warnings.<locals>._test_no_warnings,  s{    $$D1_!!(+MtDRRSWX 2 # 	
$H499 U
$3U
 L
 %
 	
"?	 21s   2A44A=r<   r=   Nro   )rI   rt   r;  )r  rx  s     rF   "test_power_transformer_no_warningsry    sL     		
	A
 aiiA&' aemmB*+rG   c                      t        d      j                  t              } | j                  d   t	        j
                  dd      k(   y)zFCheck that the results are consistent across different SciPy versions.r  r  r   gd8?r   )relN)r   ra   r   r  r^   rL  )r  s    rF   +test_yeojohnson_for_different_scipy_versionr|  =  s2    		/	3	3F	;BKKNfmmJD99rG   TransformerClassc                 H   t        j                  d      }|j                  g dg dd      } |        }|j                  |       t	        j
                         5  t	        j                  d       |j                  |j                                ddd       y# 1 sw Y   yxY w)zCheck that inverse_transform does not raise a warning about feature
    names when fitted on a DataFrame and transforming a NumPy array.

    Non-regression test for issue #31947.
    rL  )rr   rq   r   )r3  r   r4  )rE   br   N)	r^   rO  rP  ra   r  r  r  r   to_numpy)r}  rS  X_dfr  s       rF   8test_transformer_inverse_transform_feature_names_warningr  C  su     
		X	&B<<oODED"$KOOD		 	 	"g&%%dmmo6 
#	"	"s   5BB!c                 D   t        j                  ddgddgddgg      } |        j                  |      }t        j                  dgdgdgg      }d| j                   d}t	        j
                  t        |	      5  |j                  |       d
d
d
       y
# 1 sw Y   y
xY w)zLCheck that an informative error is raised when the input shape is incorrect.rr   rq   r   r3  r   r4  zX has 1 features, but z is expecting 2 featuresr   N)rI   rt   ra   r   r^   r_   r`   r   )r}  rf   r  X_wrongrW  s        rF   .test_transformer_inverse_transform_shape_errorr  U  s     	3*sCj3*56A"$((+Khhuse,-G"#3#<#<"==U
VC	z	-%%g. 
.	-	-s   ;BB
fit_methodra   r&  c                 8   t         j                  j                  d      j                  d      }t	               }t               j                  |      } t        ||       |       |j                  |      }|j                  d      dk(  sJ |j                  d      dk(  sJ |j                  d      dk(  sJ |j                  d      dk(  sJ |j                  D cg c]  }|d   dk(  s| c}d   }|d	   d
   }t        |j                  |      |       yc c}w )z>Check that the reconstruction attributes are correctly passed.r   )r@   rW   setupr=   teardownon_fit_task_beginon_fit_task_endre  kwargsfitted_estimatorN)rI   r   r   r  r   r   set_callbacksgetattrr   count_hooksr  r)   )r  rf   cbrh   ra  rectask_end_recordfitted_scalers           rF   %test_standard_scaler_callback_supportr  a  s    			a ..w7A		B++B/FGFJ"			!	B >>'"a'''>>*%***>>-.!333>>+,111&(iiTis3v;BS3SsiTUVWO#H-.@AMM++A.3 Us   D)D)rQ  r  numpyrI   numpy.linalgr  r{  r^   scipyr   r   sklearnr   r   sklearn.baser   sklearn.callback.tests._utilsr   r	   sklearn.exceptionsr
   $sklearn.externals._packaging.versionr   parse_versionsklearn.metrics.pairwiser   sklearn.model_selectionr   sklearn.pipeliner   sklearn.preprocessingr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   sklearn.preprocessing._datar    r!   sklearn.svmr"   sklearn.utilsr#   r$   sklearn.utils._array_apir%   r&   sklearn.utils._testingr'   r(   r)   r*   r+   r,   r-   r.   r/   sklearn.utils.estimator_checksr0   r1   sklearn.utils.fixesr2   r3   r4   r5   r6   r7   r8   sklearn.utils.sparsefuncsr9   	load_irisr{  r   r   r\   re   rd   r   r>  r   r]   r%  r;  r   r   tolistr   X_list_1colrB   rL   rU   rj   rv   markparametrizer   r   r   r   r   r   r   r   r   r  r  r  r   r4  r7  rA  rH  rT  rW  rf  ro  rq  rt  r  r  r  r  r  r  r  r  r  r  r  r  r  rD  r  r  r  r  r  r  r  r  r
  r  r  r*  r1  r5  r9  r?  rD  rF  rK  rO  rV  rZ  r\  r_  rg  rl  rn  rp  rs  ry  r  r  r  r  r  rt   r   r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   rK   r	  r  r  r  r  r  r  r   r"  r'  r)  r+  r   ru   r.  r3  r8  r:  rB  rJ  rY  r^  rc  xfailrj  skipifry  r|  r  r  r  )r  s   0rF   <module>r     s   
      ,  . G 2 5 %    & Q  .
 
 
   9x 	iiA
	
++b!*+
-	Q	,
yyJ'&07:	ad		Az	*	ad		Iq	)mmommo"M@$2 57O7QR,.STN U SN, 57O7QR(-/5
	 S
5
p"0J +dVn-D~-UV,udm<1 = W1: 'E* +dVn-D~-UV,udm<2::rzz":;_5 4 6 < = W 4F &78$452::rzz":;+dVn-D~-UV/S W < 6 9/Sd	I #> #>L1hA23-`0Kf +^n-LM+A N+A\ 4,7+^n-LM6 N 86 4,7& 8&R 0 .-/ %&   
 $$
 	!  
	0
 
 	!  %%% .-/ %&   
 4,7 8	@7>:#L 4,7+^n-LM87 N 887v tUm4dE]3+dVn-D~-UVI W 4 5I @ +^n-LMN NN2 +^n-LM57 N57p +^n-LMD ND  +^n-LM7 N7* )D%=9$7ryyr15{v{{2qRU7VWX% Y 8 :%$ .9D :D
: $:;*,ST? U <?( .9E :E	&	* .9O :O( .962 :62r .9> :>B.@,=^
H .9#B :#BL8 .9" :"J=	= ':;( <(&" .9!G :!GH&&F27$ +^n-LM&? N&?R3( .9	D :	DM@ .95 :5p.* !45.9+ : 6+@ !457.H# 6#" .9A :A( .9*@ :*@Z BHHd#n4~E3030l .-/	U		U)7X3Hl.(:= 7.HGG#$ I}#=>  ?  I}#=>u6vtn-: . 7 ?:3036@8 tVRVVD\FBFF4L=("((4::BVWX- Y-
 I}#=>( ?($@* 	9	980 I}#=>7 ?7, I}#=>u6K 7 ?K I}#=>u6& 7 ?&4 I}#=>u6" 7 ?"8 	'3A67 ,+M 	bggg"&&g9:+'' 69*=> ? 
^3nDTT 	-0 
6
6 	00&
W u6$ 7$$ 	M  $ v&&?  $,	$,N: +.>@S-TU7 V7" +.>@S-TU/ V/ }'=>4  ?4cs   :,~