`L i_ddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZm Z m!Z!ddl"m#Z#d Z$ejJjMd e d Z'd Z(d Z)dZ*dZ+dZ,dZ-dZ.GddZ/GddZ0GddZ1edZ2dZ3dZ4dZ5edZ6eejJjMde3e4gdgd gd!d"fe3e4gd#d$gd%d&fgd'Z7eejJjMd(ejpe3e4gd)dgid*d*d+,ejpe3e4gd-d.id*d*d/,ejpe3e4e5gd)dgid0d*d1,ejpe3e4e5gd)d gid2d*d3,ejpe4e5gd)d4gid5d*d6,ejpe3e4gd.d gd!d*d7d8,gd9Z9d:Z:d;Z;d<ZZ>Gd?d@Z?dldAZ@dBZAdCZBejJjMdDdEeCfdFeDfdGejfdHejfgeGdIdJge!eGdKdLge dMdNfdOdPfdQdRfdSdTfdUdVfdWdXfdYdZfd[eHfejJjMd\ejejfejejfejejfejejfgd]ZOd^ZPd_ZQd`ZRdaZSdbZTejJjedcddeZVejJjMdfdIdKejpdJejpdLgdgZWdhZXejJjMdiedjZYejJjMdiedkZZy)mN)sparse)LinearDiscriminantAnalysis)DecisionTreeClassifier) TempMemmap_convert_container_delete_folder_get_warnings_filters_info_listassert_allcloseassert_allclose_dense_sparseassert_docstring_consistency'assert_run_python_script_without_outputcheck_docstring_parameterscreate_memmap_backed_dataignore_warningsraisesset_random_stateskip_if_no_numpydocturn_warnings_into_errors) deprecated)_IS_WASMCSC_CONTAINERSCSR_CONTAINERS) available_ifc~t}t}t|dt|d|jdk(sJy)N)rrr random_state)ldatrees f/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_stater %s: $ &C ! #DS!T1    !! ! csr_containerctjdjdd}d}||}||fD]A}tjt |5t ||dzdddt ||Ctjtd5t ||dddtjtjdd j}|tjd }tjt d 5t ||dddy#1swYxYw#1swYxYw#1swYyxYw) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)r(zArrays are not equal) nparangereshapepytestrAssertionErrorr ValueErrorrdiagsonestocsr)r"xmsgyXABs r!test_assert_allclose_dense_sparser:.s ! Q"A #CaAV+ ]]> 5 3 (AE 2 3$Q* + z)F G+$Q*+  RWWQZ+113Abggfo&A ~-C D+$Q*++ 3 3++ ++s$D, D8 E,D5 8EE cJ ddtj5tjdtttdddt j t5ttdddt j 5}ttdddtdk(sJt|djtsJt|djtsJt j 5}ttdddt|dk(sJt|djtsJtj5tjdtttfdddtfd}tttffd }ttfd }ttfd }ttfd }ttfd }tj5tjd|||dddt j t5|dddt j t5|dddt j t5|dddfd}fd}fd} fd} fd} fd} tj5tjd||| dddt j t5| dddt j t5| dddt j t5| dddt} d}t jt|5t| }|dddt jt|5t| d}dddy#1swYxYw#1swYxYw#1swYxYw#1swYRxYw#1swYxYw#1swY xYw#1swYxYw#1swYxYw#1swYxYw#1swYtxYw#1swYXxYw#1swY._warning_functionEs +-?@r!cbtjdttjdyr=r>rBr!r_multiple_warning_functionz7test_ignore_warning.._multiple_warning_functionHs +-?@ +,r!errorcategoryr'rr*c"yNrB)rErCsrdecorator_no_warningz1test_ignore_warning..decorator_no_warninggs"$r!cyrJrBrEsrdecorator_no_warning_multiplez:test_ignore_warning..decorator_no_warning_multiplel "$r!cyrJrBrCsr decorator_no_deprecation_warningz=test_ignore_warning..decorator_no_deprecation_warningp r!cyrJrBrQsrdecorator_no_user_warningz6test_ignore_warning..decorator_no_user_warningtrSr!cyrJrBrMsr)decorator_no_deprecation_multiple_warningzFtest_ignore_warning..decorator_no_deprecation_multiple_warningxrOr!cyrJrBrMsr"decorator_no_user_multiple_warningz?test_ignore_warning..decorator_no_user_multiple_warning|rOr!cRt5dddy#1swYyxYwrJ)rrQsrcontext_manager_no_warningz7test_ignore_warning..context_manager_no_warnings$       s&cjtttf5dddy#1swYyxYwNrG)rrA UserWarningrMsr#context_manager_no_warning_multiplez@test_ignore_warning..context_manager_no_warning_multiples, '9;&G H ) & ( ) ) )s)2c^tt5dddy#1swYyxYwr]rrArQsr&context_manager_no_deprecation_warningzCtest_ignore_warning..context_manager_no_deprecation_warnings' &8 9     #,c^tt5dddy#1swYyxYwr]rr^rQsrcontext_manager_no_user_warningz.context_manager_no_user_warnings& k 2     rcc^tt5dddy#1swYyxYwr]rarMsr/context_manager_no_deprecation_multiple_warningzLtest_ignore_warning..context_manager_no_deprecation_multiple_warnings' &8 9 ) & ( ) ) )rcc^tt5dddy#1swYyxYwr]rerMsr(context_manager_no_user_multiple_warningzEtest_ignore_warning..context_manager_no_user_multiple_warnings& k 2 ) & ( ) ) )rczA'obj' should be a callable.+you should use 'category=UserWarning'r%cyrJrBrBr!rtestz!test_ignore_warning..tests r!)r?catch_warnings simplefilterrrAr.warnsr^ FutureWarninglen isinstancemessagerr0)recordrKrNrRrUrWrYr[r_rbrfrhrj warning_classr&silence_warnings_funcrlrErCs @@rtest_ignore_warningrwBslA-  "Hg&)*)4FG H ( )C@)K@BC N6K2]KMN v;!   fQi''); << < fQi'' 55 5 L6I2[IKL v;!   fQi''); << <  "Wg&)5G4UVW %%1;?@%A%012k*+01%2%k*%+%  "+g&%'(* + ( )$!#$ k "4134 ( )-*,- )  ))  "1g&"$+-.0 1 ( )*')* k ":79: ( )3023 M OE z / > >?P Q  z /  '  ( aHH CCNN LL WW>++$$44--811**::33  s2Q$0Q1#Q>$R -R+R%(R2R?:S <+SS&1S3T T  T$Q.1Q;>R RR"%R/2R<?S  SS#&S03S=T  TT"c||z}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c rBabcs rf_okr~ AA Hr!c||z}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c rBrzs rf_bad_sectionsrrr!c||z}|SryrB)r|r{r}s r f_bad_orderrrr!c||z}|S)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c rB)r{r|ds rf_too_many_param_docstringrs" AA Hr!c||z}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c rBrzs r f_missingrs AA Hr!c||z|z|zS)zFunction f Parameters ---------- a: int Parameter a b: Parameter b c : This is parsed correctly in numpydoc 1.2 d:int Parameter d e No typespec is allowed without colon rB)r{r|r}res rf_check_param_definitionr&s q519q=r!ceZdZdZdZy)KlasscyrJrBselfr7r6s rrzKlass.f_missing:s r!cy)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c NrBrs rrzKlass.f_bad_sections=s r!N)__name__ __module__ __qualname__rrrBr!rrr9s   r!rc*eZdZdZdZdZdZdZy)MockEstcy) MockEstimatorNrBrs r__init__zMockEst.__init__Pr!c|SrJrBrs rfitz MockEst.fitSr!c|SrJrBrr7s rpredictzMockEst.predictVrr!c|SrJrBrs r predict_probazMockEst.predict_probaYrr!cy)N?rBrs rscorez MockEst.score\sr!N)rrrrrrrrrBr!rrrOsr!rceZdZdZeddZededdZeddZed d Z y ) MockMetaEstimatorc||_y)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rrs rrzMockMetaEstimator.__init__as ! r!c.t|jdS)Nrhasattrrrs rzMockMetaEstimator.kswt}}i@r!c8|jj|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )rrrs rrzMockMetaEstimator.predictks}}$$Q''r!c.t|jdS)Nrrrs rrzMockMetaEstimator.vswt}}g>r!z%Testing a deprecated delegated methodcy)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y NrBrs rrzMockMetaEstimator.scorevrr!c.t|jdS)Nrrrs rrzMockMetaEstimator.swt}}oFr!c|S)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X rBrs rrzMockMetaEstimator.predict_probas r!z-Testing deprecated function with wrong paramscy)z,Incorrect docstring but should not be testedNrBrs rrzMockMetaEstimator.fitrr!N) rrrrrrrrrrrBr!rrr`s!@A(B(>?78 9@ FGH?@;A;r!rc tt}|gk(sJttdg}|gk(sJttdg}|gk(sJtjt d5tt dddtjt d5ttj dddtt}tt}|jj}|gdk(sJgdgd gd gd d |d ddddddgd |ddddgd |ddddgd |dddddgg}t|ttttj|j |j"|j$|j&gD]#\}}t|}||k(rJd|d|dy#1swYI ??*9cUCI ?? |+D E3">23 |+F G9"5#7#789++CDI!795I((11N         =^/?@  5    =^ 7 < ( ;F ; < < \\+T 2 < ( ;F ; < < %W77  < < < >+...  ( (A -- -I K9 55[$)<ggoodmm4 5 ww$77>>+...  ( (A -- -5555s6E8E&E#&E/ct}|jtd|tjd}t |}t |||jdk(sJt |d\}}t |||tjj|jk(sJ|jdk(sJd}t ||}t ||||jdk(sJ||dz|dzg}t |}t||D]\}}t |||jd k(sJt |d g\}} t ||| d k(sJy) Nrrr*T) return_folderr'r r z not-an-array) rr r r+r2rrrrrrrr) rrrrfolderr input_listmmap_data_list output_dataothers rtest_create_memmap_backed_datar!WsT.0 ,@A''!*K $[ 1Dd#  ( (A -- -,[MLD&d# RWW__T]]3 33 3  ( (A -- -I $[I FDdI.  ( (A -- -{Q a@J.z:N ^<( T[$'(  ( (A -- -2K3PQKk* N "" "r!z constructor_name, container_typelisttuplearrayr sparse_csrsparse_csr_array sparse_cscsparse_csc_array dataframec@tjdjSNpandasr. importorskip DataFramerBr!rrrsf11(;EEr!seriesc@tjdjSr+r.r.SeriesrBr!rrrs6..x8??r!indexc@tjdjSr+)r.r.IndexrBr!rrrs&--h7==r!pyarrowc@tjdjSNr7)r.r.TablerBr!rrrsF// :@@r! pyarrow_arrayc@tjdjSr9)r.r.ArrayrBr!rrrs&"5"5i"@"F"Fr!polarsc@tjdjSNr>r-rBr!rrrs6..x8BBr! polars_seriesc@tjdjSr@r2rBr!rrrs&"5"5h"?"F"Fr!slicezdtype, superdtypec4|dvr|}ddg}t|||}t||sJ|dvr%tjt |d|sJy|dvryt |dr|j |k(sJyt |d r|jd|k(sJyy) z\Check that we convert the container to the right type of array with the right data type.)r)r4r>rAr7r;r0rr*dtype)r"r#r4)r>rAr7r;NrFdtypes)rrrr+ issubdtypetyperrFrG)constructor_namecontainer_typerF superdtype containercontainer_converteds rtest_convert_containerrOusL()AI, )> :: :55}}T"5a"89:FFF T T $g ."((E111 $h /"))!,555 0r!ctjdtdggddgdg}|jjddk(sJy)Nr,r4r)r8categorical_feature_namesrrH)r.r.rrGiloc)dfs r(test_convert_container_categories_pandasrUsH !  seu B 99>>!  ** *r!ctjd}tdggddgdg}|jd|j k(sJy)Nr>r4r8rQ)r.r.rschema Categorical)plrTs r(test_convert_container_categories_polarsrZsI   X &B cUGXuQTPU VB 99S>R^^- -- -r!ctjd}tdggddgdg}t|jdj|j usJy)Nr7r4r8rQr)r.r.rrIrWDictionaryType)parTs r)test_convert_container_categories_pyarrowr^sS   Y 'B cUGYRUQV WB  ! !! "b&7&7 77 7r!ctt5t#1swYnxYwttd5}td#1swYnxYwjsJttddg5}td#1swYnxYw|jsJtjtd5tt5}t d#1swYnxYw dddn #1swYnxYw|jrJtjt d5ttd5}t #1swYnxYw dddn #1swYnxYw|jrJtjtd5ttd5}t d#1swYnxYw dddn #1swYnxYw|jrJtjt d 5ttd 5}td #1swYnxYw dddn #1swYnxYw|jrJtjt d5ttd d 5}td #1swYnxYw dddn #1swYnxYw|jrJtjt d 5tt5} dddn #1swYnxYwdddn #1swYnxYw|jrJttd d5} dddn #1swYnxYw|jrJtttf5t#1swYnxYwtttf5t #1swYnxYwtjt 5tttf5 dddn #1swYnxYwdddy#1swYyxYw)Nz how are your%zhello how are youz not this onezthis will be raisedzthe failure message)err_msgzthis is ignoredz,should contain one of the following patternshelloz Bad message)r&r`z Did not raiseT)r&may_pass)rrraised_and_matchedr.r0r/)cms r test_raisesres  k   /-2+,,--    .-!@ A-R+,,--   z)> ?4 I  4"23 3 4 4 4444$$$ $ ~-B C I'< = ,    $$$ $ z)> ?4 I%6 7 4223 3 4 4 4444$$$ $ L +IW - +M* * + + ++++ $$$ $ ~-B C+ IW6K L +PRM* * + + ++++$$$ $ ~_ = I  "    $$$ $  4 8 B    $$$ $ J' (k J' (l ~ & Y + ,     s! $ AA 1 A==B2C$ CC C$$C-E + D66D? ;E  EF5 FF( $F55F>*H< HH HH'J& I22I; 7JJ=K- K K-K! K--K6L$$L- MM'? N  N/O%O O%O O%%O.c^ttjdgtjdt j t 5ttjdgtjddddttjdgtjdt j t 5ttjdgtjddddt j t 5ttjdgtjddddttjdgtjdd y#1swYxYw#1swYxYw#1swYUxYw) Ng㈵?rErgX?gc^?g55?gh㈵>ggh㈵>)atol)r r+r$float32r.rr/float64rBr!r"test_float32_aware_assert_allcloserjsBHHj\PQr!ctj5tt|ddddy#1swYyxYw)NTr)r?rmrr)r}s rtest_turn_warnings_into_errorsrss7  "H!# $GHHHs6?)r)[r rr?numpyr+r.scipyrsklearn.discriminant_analysisr sklearn.treersklearn.utils._testingrrrr r r r r rrrrrrrsklearn.utils.deprecationrsklearn.utils.fixesrrrsklearn.utils.metaestimatorsrr mark parametrizer:rwr~rrrrrrrrrrrrrrparamrrrrrrrrrr!r"r#ndarray csr_matrixrrCint32integerint64rhfloatingrirOrUrZr^rerjxfailrprxrrrrBr!rrsG  D/"1 6".9+:+&J & & & * "&  ,".;.;b|M|M~ . 4 288 EN #u > + EN %3% @ , 8 8 " EN-u5r2,   eU^&7%>B:V E7 # u %)    E7 # u %4    G  u %#    EN#u =  ;  G*-\ 8]-^ 8      -S-S`" " J77:.*#<&  % "** 6$$%  l. / @ l. / @ EF ?@ => @A FG BC FG %!* 2:: 2:: R[[! R[[! &6+<&6R+. 8 CLH$8$=>L?L.      3 3C0)H)JKRLR )H)JKHLHr!