{KglRddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(d Z)ejTjWd e#d Z,d Z-d Z.dZ/Gddej`Z1dZ2dZ3dZ4dZ5dZ6dZ7GddZ8GddZ9GddZ:dZ;GddZejTje"d"#d$Z@ejTjWd%d&eAfd'eBfd(ejfd)ejfgeEd*d+ge$eEd,d-ge#d.d/fd0d1fd2d3fd4eFfejTjWd5ejejfejejfejejfejejfgd6ZMd7ZNd8ZOd9ZPejTje&e%d:k\d;#ejTjWdZRd?ZSd@ZTejTje"dA#dBZUejTjWd3s ! 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 cttd5ttd5 ddddddy#1swYxYw#1swYyxYw)Nz Hello world)r9)rr3rr4r%r#test_assert_raises_msgrAGs6 ^] ; := 9 : < ; 9 9 < ;sA6A? AA cd}d}ttd|dttttd|dttttd|dttttd|ttttt fd|y)Nct|N)r4)messages r#_raise_ValueErrorz4test_assert_raise_message.._raise_ValueErrorNs !!r%cyrDr@r@r%r# _no_raisez,test_assert_raise_message.._no_raiseQ r%testzsomething else)r r4rr3 TypeErrorAttributeError)rFrHs r#test_assert_raise_messagerMMs" V->G  ."6 FIV ^$ r%cddttttttjt5tt dddtj5}tt dddtdk(sJt|djtsJt|djt sJtj5}tt dddt|dk(sJt|djtsJtttt ftfd}ttt ffd}ttfd }tt fd }ttfd }tt fd }t|t|t|tjt5|dddtjt 5|dddtjt5|dddfd }fd}fd} fd} fd} fd} t|t|t| tjt5| dddtjt 5| dddtjt5| dddt } d}tjt|5t| }|dddtjt|5t| d}dddy#1swYjxYw#1swYDxYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swY$xYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)Nc8tjdtyNzdeprecation warningwarningswarnDeprecationWarningr@r%r#_warning_functionz.test_ignore_warning.._warning_functionws +-?@r%cbtjdttjdyrPrQr@r%r#_multiple_warning_functionz7test_ignore_warning.._multiple_warning_functionzs +-?@ +,r%categoryr+rr.c"yrDr@)rWrUsr#decorator_no_warningz1test_ignore_warning..decorator_no_warnings"$r%cyrDr@rWsr#decorator_no_warning_multiplez:test_ignore_warning..decorator_no_warning_multiple "$r%cyrDr@rUsr# decorator_no_deprecation_warningz=test_ignore_warning..decorator_no_deprecation_warning r%cyrDr@rasr#decorator_no_user_warningz6test_ignore_warning..decorator_no_user_warningrcr%cyrDr@r]sr#)decorator_no_deprecation_multiple_warningzFtest_ignore_warning..decorator_no_deprecation_multiple_warningr_r%cyrDr@r]sr#"decorator_no_user_multiple_warningz?test_ignore_warning..decorator_no_user_multiple_warningr_r%cRt5dddy#1swYyxYwrD)rrasr#context_manager_no_warningz7test_ignore_warning..context_manager_no_warnings      s&cjtttf5dddy#1swYyxYwNrX)rrT UserWarningr]sr##context_manager_no_warning_multiplez@test_ignore_warning..context_manager_no_warning_multiples$ '9;&G H & (I H Hs)2c^tt5dddy#1swYyxYwrmrrTrasr#&context_manager_no_deprecation_warningzCtest_ignore_warning..context_manager_no_deprecation_warnings &8 9  : 9 9#,c^tt5dddy#1swYyxYwrmrrnrasr#context_manager_no_user_warningz.context_manager_no_user_warnings k 2  3 2 2rsc^tt5dddy#1swYyxYwrmrqr]sr#/context_manager_no_deprecation_multiple_warningzLtest_ignore_warning..context_manager_no_deprecation_multiple_warnings &8 9 & (: 9 9rsc^tt5dddy#1swYyxYwrmrur]sr#(context_manager_no_user_multiple_warningzEtest_ignore_warning..context_manager_no_user_multiple_warnings k 2 & (3 2 2rszA'obj' should be a callable.+you should use 'category=UserWarning'r)cyrDr@r@r%r#rJz!test_ignore_warning..tests r%) r rrTr2warnsrn FutureWarninglen isinstancerErr4)recordr[r^rbrergrirkrorrrvrxrz warning_classr*silence_warnings_funcrJrWrUs @@r#test_ignore_warningrtsbA- '89:'8CUVW ( )@)K@B * 6K2]KM  v;!   fQi''); << < fQi'' 55 5 6I2[IK  v;!   fQi''); << <)5G4UV %%1;?@%A%012k*+01%2%k*%+%+,4578 ( )!# * k "13 # ( )*, * )  ))12:;=> ( )') * k "79 # ( )02 * M OE z / > >?P Q 0 z /  '  (  0 /C * )  L * ) " " ) )> * ) " " ) ) 0 / 0 /sOOO)9O6"P P3PP*P78Q4QOO&)O36PP PP'*P47QQ QceZdZdZy) TestWarnscd}tjt5t|dddtdddk(sJy#1swYxYw)Nc.tjdy)Nyor)rRrSr@r%r#fzTestWarns.test_warn..fs MM$ r%c|SrDr@)r8s r#z%TestWarns.test_warn..sAr%r.)r2rr3r )selfrs r# test_warnzTestWarns.test_warns@ ]]> * q !+!+q1Q666+ *s AA N)__name__ __module__ __qualname__rr@r%r#rrs7r%rc||z}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c r@abcs r#f_okr AA Hr%c||z}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c r@rs r#f_bad_sectionsr rr%c||z}|Srr@)rrrs 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 r@)rrds r#f_too_many_param_docstringr0s" AA Hr%c||z}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c r@rs r# f_missingrEs 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 r@)rrrres r#f_check_param_definitionrVs q519q=r%ceZdZdZdZy)KlasscyrDr@rr;r:s r#rzKlass.f_missingjrIr%cy)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nr@rs r#rzKlass.f_bad_sectionsms r%N)rrrrrr@r%r#rris   r%rc*eZdZdZdZdZdZdZy)MockEstcy) MockEstimatorNr@rs r#__init__zMockEst.__init__r%c|SrDr@rs r#fitz MockEst.fitr%c|SrDr@rr;s r#predictzMockEst.predictrr%c|SrDr@rs r# predict_probazMockEst.predict_probarr%cy)N?r@rs r#scorez MockEst.scoresr%N)rrrrrrrrr@r%r#rrsr%rceZdZdZeddZededdZeddZed d Z y ) MockMetaEstimatorc||_y)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rrs r#rzMockMetaEstimator.__init__s ! r%c.t|jdS)Nrhasattrrrs r#rzMockMetaEstimator.swt}}i@r%c8|jj|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )rrrs r#rzMockMetaEstimator.predicts}}$$Q''r%c.t|jdS)Nrrrs r#rzMockMetaEstimator.swt}}g>r%z%Testing a deprecated delegated methodcy)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y Nr@rs r#rzMockMetaEstimator.scorerr%c.t|jdS)Nrrrs r#rzMockMetaEstimator.swt}}oFr%c|S)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X r@rs r#rzMockMetaEstimator.predict_probas r%z-Testing deprecated function with wrong paramscy)z,Incorrect docstring but should not be testedNr@rs r#rzMockMetaEstimator.fitrr%N) rrrrrrrrrrr@r%r#rrs!@A(B(>?78 9@ FGH?@;A;r%rc tjdddtt}|gk(sJttdg}|gk(sJttdg}|gk(sJtj t d5ttdddtj t d 5ttjdddtt}tt }|jj}|gd k(sJgd gd gdgddd|dzddddddgdd|dzdzdddgdd|dzdddgdd|dzddd d!gg}t|tt ttj|j"|j$|j&|j(gD]#\}}t|}||k(rJd"|d#|d$y#1swYKxYw#1swYxYw)%Nnumpydocz+numpydoc is required to test the docstringsz1.2.0)reason minversionr)ignorezUnknown section Resultsr)zUnknown section Parameterr)zxsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('a: int')ztsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('b:')zwsklearn.utils.tests.test_testing.f_check_param_definition There was no space between the param name and colon ('d:int'))z9In function: sklearn.utils.tests.test_testing.f_bad_orderznThere's a parameter name mismatch in function docstring w.r.t. function signature, at index 0 diff: 'b' != 'a' Full diff:z - ['b', 'a']z + ['a', 'b'])zHIn function: sklearn.utils.tests.test_testing.f_too_many_param_docstringz_Parameters in function docstring have more items w.r.t. function signature, first extra item: cr - ['a', 'b']z+ ['a', 'b', 'c']z? +++++)z7In function: sklearn.utils.tests.test_testing.f_missingzaParameters in function docstring have less items w.r.t. function signature, first missing item: brrz+ ['a'])z=In function: sklearn.utils.tests.test_testing.Klass.f_missingaParameters in function docstring have less items w.r.t. function signature, first missing item: Xr - ['X', 'y']+ []z In function: z!sklearn.utils.tests.test_testing.z.predictznThere's a parameter name mismatch in function docstring w.r.t. function signature, at index 0 diff: 'X' != 'y'rz- ['X']z? ^z+ ['y'].rz&potentially wrong underline length... z Parameters z --------- in z.scorez.fitrrrz "z " not in "")r2 importorskiprrrr RuntimeErrorrrrrr __class__rziprrrrrr) incorrect mock_metamock_meta_namemessagesr9rs r#test_check_docstring_parametersrs< < +40I ??*4>I ??*9cUCI ?? |+D E">2 F |+F G"5#7#78 H++CDI!795I((11N          1.1AJ KC       1.1AC D  5     1.1AH I 4      A.AQQUV V4    OQHf  &  OO     # # OO MM  Q/q1 iLS)!LL U F E G Gs<G/G"G"G,ceZdZdZdZy)RegistrationCountercd|_y)Nr)nb_callsrs r#rzRegistrationCounter.__init__Js  r%cV|xjdz c_|jtusJy)Nr.)rfuncr)rto_register_funcs r#__call__zRegistrationCounter.__call__Ms$  $$666r%N)rrrrrr@r%r#rrIs 7r%rct|tjsJ|dk7}|jj|usJtj j ||y)Nr)rr/memmapflags writeabletestingassert_array_equal) input_array mmap_data mmap_moders r# check_memmaprRsL i ++ +S I ?? $ $ 11 1JJ!!+y9r%ct}|jtd|tjd}t |5}t ||tjj|j}dddtjdk7r!tjjrJ|jdk(sJd}t ||5}t |||tjj|j}dddtjdk7r!tjjrJ|jdk(sJy#1swYxYw#1swY[xYw)Nregisterrntr.r+rr+)rsetattratexitr/r6rrospathdirnamefilenamenameexistsr) monkeypatchregistration_counterrdata temp_folderrs r#test_tempmemmaprYs.0 ,@A''!*K K D[$'ggoodmm4  ! ww$77>>+...  ( (A -- -I K9 5[$)<ggoodmm4  6 ww$77>>+...  ( (A -- - !  6 5s6E8E&E#&E/zmemmap not fully supported)rct}|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+rrz not-an-array) rrrr/r6rrrrrrrr) rrrrfolderr input_listmmap_data_list output_dataothers r#test_create_memmap_backed_datar nsO.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@tjdjSNpandas)r2r DataFramer@r%r#rrsf11(;EEr%seriesc@tjdjSr)r2rSeriesr@r%r#rrs6..x8??r%indexc@tjdjSr)r2rIndexr@r%r#rrs&--h7==r%slicezdtype, superdtypec*|dvr|}ddg}t|||}t||sJ|dvr%tjt |d|sJyt |dr|j |k(sJyt |dr|jd|k(sJyy) z\Check that we convert the container to the right type of array with the right data type.)rpolarsr polars_seriesrrr.dtype)r rrr$dtypesN)rrr/ issubdtypetyperr$r%)constructor_namecontainer_typer$ superdtype containercontainer_converteds r#test_convert_containerr-sDVV()AI, )> :: :55}}T"5a"89:FFF $g ."((E111 $h /"))!,555 0r%ctjdtdggddgdg}|jjddk(sJy)Nrr8rr<categorical_feature_namesrrY)r2rrr%iloc)dfs r#(test_convert_container_categories_pandasr3sH !  seu B 99>>!  ** *r%ctjd}tdggddgdg}|jd|j k(sJy)Nr!r8r<r/)r2rrschema Categorical)plr2s r#(test_convert_container_categories_polarsr8sI   X &B cUGXuQTPU VB 99S>R^^- -- -r%ctjd}tdggddgdg}t|jdj|j usJy)Npyarrowr8r<r/r)r2rrr'r5DictionaryType)par2s r#)test_convert_container_categories_pyarrowr=sS   Y 'B cUGYRUQV WB  ! !! "b&7&7 77 7r%z1.8z-sparse arrays are available as of scipy 1.8.0r(r$cddg}tjtdt5t |||dddy#1swYyxYw)zCheck that if we convert to sparse array but sparse array are not supported (scipy<1.8.0), we should raise an explicit error.rr.z&only available with scipy>=1.8.0, got r)r#N)r2rr4rr)r(r$r+s r#7test_convert_container_raise_when_sparray_not_availabler?sEAI 6zlC  9&6eD    s ?Actt5t#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*rAz Did not raiseT)r*may_pass)rrKraised_and_matchedr2r4r3)cms r# test_raisesrFs  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㈵?r#rgX?gc^?g55?gh㈵>ggh㈵>)atol)r r/rfloat32r2rr3float64r@r%r#"test_float32_aware_assert_allcloserK2sBHHj\PQr%ctj5tt|ddddy#1swYyxYw)NTrg)rRr_rrc)ras r#test_turn_warnings_into_errorsrls*  "!# $G # " "s6?)r)[rrunittestrRnumpyr/r2scipyrsklearn.discriminant_analysisr sklearn.treersklearn.utils._testingrrrr r r r r rrrrrrrrrsklearn.utils.deprecationrsklearn.utils.fixesrrrrrsklearn.utils.metaestimatorsrr$mark parametrizer>rArMrTestCaserrrrrrrrrrrrrrxfailr r rndarray csr_matrixrrint32integerint64rIfloatingrJr-r3r8r=skipifr?rFrKrQparamrZrcrjrlr@r%r#rs  D/&16".9+:+& $Nrj7!!7 & & & * "&  ,".;.;bEMP77:.*8$@A#B#<&  % "** 6$$%  l. / @  l. / @  EF  ?@  =>  % " 2:: 2:: R[[! R[[! 6#46>+. 8 -&& :+.@BT-UV288RXXrzz2::"NO EPW  ECLH$8$=>L?L. ++$$]511F%   ++$$]511F% *3+*3C0)H)JKRLR )H)JKHLHr%