`L iddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZddlmZmZmZmZmZm Z m!Z!eZ"dZ#ejHjKd e ed d gfe ed d d gfeed d gfed ed d gfe ded d gfe ded d gfeded d gfeed d gfed ed d gfg dZ&ejHjKde efe efeefed efe defe defedefeefed efg dZ'y)N)clone) load_iris)train_test_split) MaxAbsScaler MinMaxScalerPowerTransformerQuantileTransformer RobustScalerStandardScaler maxabs_scale minmax_scalepower_transformquantile_transform robust_scalescale)assert_allcloseassert_array_equal)BSR_CONTAINERSCOO_CONTAINERSCSC_CONTAINERSCSR_CONTAINERSDIA_CONTAINERSDOK_CONTAINERSLIL_CONTAINERScR|dd|gftj|dd|fS)z"Get non NaN samples in column of XN)npisnan)Xcols m/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/preprocessing/tests/test_common.py_get_valid_samples_by_columnr!%s+ QX;1c6++ ,,z9est, func, support_sparse, strictly_positive, omit_kwargsTFclip) with_meanz yeo-johnsonzbox-cox ) n_quantiles)with_centeringc < tjjd}tjj }d}tj ||j|jd||j|jd|f<|r|tj|dzz }t|d\}} tjtj|djrJtjtj|djsJtjtj| djsJtj | dddf<tj5tj d t"|j%|j'| } dddt)tj tj| tj5tj d t"|j'|} ddd|j+} |D]} | j-| }||fi| }t)tj|tj t/|tj|| tj| |j1| }t)tj|tj| t/|tj|| tj| t3|jdD]}|j%t5||tj5tj d t"|j'| dd|gf}dddt/| dd|gftj| dd|fjr|j't5| |}t)||tj|j7|rt9|}t9|}tj5tj d t"|j%|j'| }|j1|}dddt:t<zt>zt@ztBztDztFzD]}||}|| }tj5tj d tHtj d t"|j%|j'|}dddt/jKtj5tj d tHtj d t"|j1|}dddt/jKyy#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYhxYw) N*2r)sizeg?) random_state)axiserrorignore)&rrandom RandomStateirisdatacopynanrandintshapenanminrallranywarningscatch_warnings simplefilterRuntimeWarningfit transformr get_paramspoprinverse_transformranger!squeezerrrrrrrrPendingDeprecationWarningtoarray)estfuncsupport_sparsestrictly_positive omit_kwargsrngr n_missingX_trainX_testXtXt_classkwargskwarg_Xt_funcXt_inviXt_col Xt_col_nonan est_dense est_sparseXt_dense Xt_inv_densesparse_container X_train_sp X_test_spXt_sp Xt_inv_sps r test_missing_value_handlingre*sF$ ))   #C AI  AGGAJY /QWWQZi1XX RYYq\C &qq9OGVvvbhhw'a04466 6 66"((7#! , 0 0 22 2 66"((6" + / / 11 166F1a4L  "0g~6 WWW  ' ' /0 rxx|RXXf%56  "*g~6==)*^^ F JJu 7%f%Grxx("((8*<=GRXXg../288H;M:M1NO " "2 &Frxx'&)9:FBHHV,,-vrxx7G6G/HI 1771:  R ,Wa89  $ $ & 3  ! !'> :]]6!aS&>2F 3 1qc6 +xxq!t %))+==)Efa)PQL |VRXXfnn>N5O4O-P Q R#J 3Z  $ $ & A  ! !'> : }}W-77?H$66x@L A             ? *'2J(0I((* H%%h0IJ%%g~>"z2<&88?  @ I--/ >1 ?Q00**. 3 3 A A$ H H  @ @sK;Y ,Y84Y,.A Y9AZ%AZYY),Y6 9ZZ Z z est, funcc tjd}tjdddtjtjdddgtjtjdddtjtjdggd gj }|j |d gd  }|d jd|d <|j|}|j|}t||y)Npandasr,rf)r,rirfrjrkrmrlInt16)abc)dtypecolumnsrrint) pytest importorskiprarrayr6T DataFrameastype fit_transformr)rIrJpdrX_dfX_trans X_df_transs r $test_missing_value_pandas_na_supportrs&   X &B  1bffbffaA . VVRVVQ1bffbffa 8 $   a  <</< BDS   'DI"G""4(JGZ(r")(r<numpyrrv sklearn.basersklearn.datasetsrsklearn.model_selectionrsklearn.preprocessingrrrr r r r r rrrrsklearn.utils._testingrrsklearn.utils.fixesrrrrrrrr3r!mark parametrizererr"r rs &4    G{- ? tUB7 uefX>  5%3 % (%ubA - (/5%L ) $oudBG  ,.@$rR ueR8 U +\4K  \? \?~ & &  5! % (%0 - (/: Y '    +-?@ & U +\: ")#")r"