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 u/Volumes/fast/ai/experiments/voice-extract-mac/.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 (AE 26$Q*  z)F G$Q* H  RWWQZ+113Abggfo&A ~-C D$Q* E D6 5 H G E Ds$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 & (I H Hs)2c^tt5dddy#1swYyxYwr]rrArQsr&context_manager_no_deprecation_warningzCtest_ignore_warning..context_manager_no_deprecation_warnings &8 9  : 9 9#,c^tt5dddy#1swYyxYwr]rr^rQsrcontext_manager_no_user_warningz.context_manager_no_user_warnings k 2  3 2 2rcc^tt5dddy#1swYyxYwr]rarMsr/context_manager_no_deprecation_multiple_warningzLtest_ignore_warning..context_manager_no_deprecation_multiple_warnings &8 9 & (: 9 9rcc^tt5dddy#1swYyxYwr]rerMsr(context_manager_no_user_multiple_warningzEtest_ignore_warning..context_manager_no_user_multiple_warnings k 2 & (3 2 2rczA'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_warningrwBsA-  "g&)*)4FG # ( )@)K@B * 6K2]KM  v;!   fQi''); << < fQi'' 55 5 6I2[IK  v;!   fQi''); << <  "g&)5G4UV # %%1;?@%A%012k*+01%2%k*%+%  "g&%'(* # ( )!# * k "13 # ( )*, * )  ))  "g&"$+-.0 # ( )') * k "79 # ( )02 * M OE z / > >?P Q 0 z /  '  (  0 /a # " * )   # "> # " * ) " " ) )8 # " * ) " " ) ) 0 / 0 /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 E">2 F |+F G"5#7#78 H++CDI!795I((11N         =^/?@  5    =^ 7 ( ;F ;8 7 \\+T 2 ( ;F ;3 2 %W77 8 73 2s A4 B4A=B cy)zFunction four. Parameters ---------- labels : array-like, default=None The set of labels to include when `average != 'binary'`, and their order if `average is None`. Labels present in the data can be excluded. NrBlabelss rf_fourr r!cy)aFunction five. Parameters ---------- labels : array-like, default=None The set of labels to include when `average != 'binary'`, and their order if `average is None`. This is an extra line. Labels present in the data can be excluded. NrBrs rf_fivers r!cy)zFunction six. Parameters ---------- labels : array-like, default=None The group of labels to add when `average != 'binary'`, and the order if `average is None`. Labels present on them datas can be excluded. NrBrs rf_sixrrr!cd}tjt|5ttt t gddddy#1swYyxYw)z8Check `assert_docstring_consistency` difference message.aThe description of Parameter 'labels' is inconsistent between \['f_four'\] and \['f_five'\] and \['f_six'\]: \*\*\* \['f_four'\] --- \['f_five'\] \*\*\*\*\*\*\*\*\*\*\*\*\*\*\* \*\*\* 10,25 \*\*\*\* --- 10,30 ---- 'binary'`, and their order if `average is None`. \+ This is an extra line. Labels present in the data can be excluded. \*\*\* \['f_four'\] --- \['f_six'\] \*\*\*\*\*\*\*\*\*\*\*\*\*\*\* \*\*\* 1,25 \*\*\*\* The ! set of labels to ! include when `average != 'binary'`, and ! their order if `average is None`. Labels present ! in the data can be excluded. --- 1,25 ---- The ! group of labels to ! add when `average != 'binary'`, and ! the order if `average is None`. Labels present ! on them datas can be excluded.r%T)rN)r.rr/r rrr)r5s r+test_assert_docstring_consistency_error_msgrs7( CT ~S 1$ffe%>+...  ( (A -- -I K9 5[$)<ggoodmm4  6 ww$77>>+...  ( (A -- - !  6 5s6E8E&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!WsO.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+,, 0 /    .-!@ AR+,, B A   z)> ? I "23 3   @ ? ?$$$ $ ~-B C I'< =, > = = D C C$$$ $ z)> ? I%6 7223 38 7 7 @ ? ?$$$ $ L IW -M* *. - -    $$$ $ ~-B C IW6K LPRM* *M L L D C C$$$ $ ~_ = I "    > = =$$$ $  4 8B  9 8 8$$$ $ 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_errorsrss*  "!# $G # " "s6?)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!