U md_@sddlZddlZddlZddlZddlZddlmZddlZddl m Z ddl m Z m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlm Z ddl!m"Z"d d Z#d d Z$d dZ%ddZ&ddZ'Gdddej(Z)ddZ*ddZ+ddZ,ddZ-ddZ.dd Z/Gd!d"d"Z0Gd#d$d$Z1Gd%d&d&Z2Gd'd(d(Z3ej45d)ej46d*e2e1d+e3e1d+gd,d-Z7Gd.d/d/Z8dSd1d2Z9d3d4Z:ej46d5d6d7gd8d9Z;ej46d:ejej?gd;d<Z@ej46d=d>eAfd?eBfd@ejCfdAejDfdBejDfdCejEfdDdEdFfdGdHdFfdIdJdFfdKeFfg ej46dLej>ejGfej?ejGfej._raise_ValueErrorcSsdSr3rrrrr _no_raiseJsz,test_assert_raise_message.._no_raisetestzsomething else)r r(rr TypeErrorAttributeError)r5r6rrrtest_assert_raise_messageFs6  r:c sddddtttttdttttdW5QRXttttdW5QRXttttdW5QRXttttfdtfdd}tttfdfdd }ttdfd d }ttdfd d }ttdfdd}ttdfdd}t|t|t|tt |W5QRXtt |W5QRXtt |W5QRXfdd}fdd}fdd}fdd} fdd} fdd} t|t|t|tt | W5QRXtt | W5QRXtt | W5QRXt} d} tjt| dt| }|W5QRXtjt| dt| d d!}W5QRXdS)"NcSstdtdSNzdeprecation warningwarningswarnDeprecationWarningrrrr_warning_functionpsz.test_ignore_warning.._warning_functioncSstdttddSr;r<rrrr_multiple_warning_functionss z7test_ignore_warning.._multiple_warning_functioncategorycsdSr3rrrAr@rrdecorator_no_warningsz1test_ignore_warning..decorator_no_warningcs dSr3rrrArrdecorator_no_warning_multiplesz:test_ignore_warning..decorator_no_warning_multiplecs dSr3rrr@rr decorator_no_deprecation_warningsz=test_ignore_warning..decorator_no_deprecation_warningcs dSr3rrrHrrdecorator_no_user_warningsz6test_ignore_warning..decorator_no_user_warningcs dSr3rrrFrr)decorator_no_deprecation_multiple_warningszFtest_ignore_warning..decorator_no_deprecation_multiple_warningcs dSr3rrrFrr"decorator_no_user_multiple_warningsz?test_ignore_warning..decorator_no_user_multiple_warningc st W5QRXdSr3)r rrHrrcontext_manager_no_warningsz7test_ignore_warning..context_manager_no_warningc s$tttfd W5QRXdSNrB)r r? UserWarningrrFrr#context_manager_no_warning_multiplesz@test_ignore_warning..context_manager_no_warning_multiplec s ttd W5QRXdSrNr r?rrHrr&context_manager_no_deprecation_warnings zCtest_ignore_warning..context_manager_no_deprecation_warningc s ttd W5QRXdSrNr rOrrHrrcontext_manager_no_user_warnings z.context_manager_no_user_warningc s ttd W5QRXdSrNrQrrFrr/context_manager_no_deprecation_multiple_warnings zLtest_ignore_warning..context_manager_no_deprecation_multiple_warningc s ttd W5QRXdSrNrSrrFrr(context_manager_no_user_multiple_warnings zEtest_ignore_warning..context_manager_no_user_multiple_warningzA'obj' should be a callable.+you should use 'category=UserWarning'rcSsdSr3rrrrrr7sz!test_ignore_warning..test) rr r?r'ZwarnsrO FutureWarningrr()rErGrIrJrKrLrMrPrRrTrUrVZ warning_classrZsilence_warnings_funcr7rrDrtest_ignore_warningmsr                  rXc@seZdZddZdS) TestWarnsc Cs@dd}ttt|W5QRXtddddksrrrrfs zTestWarns.test_warn..fcSs|Sr3r)r+rrrz%TestWarns.test_warn..r#)r'rrr)selfrZrrr test_warns zTestWarns.test_warnN)__name__ __module__ __qualname__r^rrrrrYsrYcCs ||}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c rabcrrrf_oksrgcCs ||}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c rrcrrrf_bad_sectionssrhcCs ||}|Srbr)rerdrfrrr f_bad_ordersricCs ||}|S)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r)rdredrrrf_too_many_param_docstring!srkcCs ||}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c rrcrrr f_missing6s rlcCs||||S)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)rdrerfrjerrrf_check_param_definitionGsrnc@seZdZddZddZdS)KlasscCsdSr3rr]r.r-rrrrl[szKlass.f_missingcCsdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c Nrrprrrrh^szKlass.f_bad_sectionsN)r_r`rarlrhrrrrroZsroc@s4eZdZddZddZddZddZd d Zd S) MockEstcCsdS)Z MockEstimatorNrr]rrr__init__qszMockEst.__init__cCs|Sr3rrprrrfittsz MockEst.fitcCs|Sr3rr]r.rrrpredictwszMockEst.predictcCs|Sr3rrurrr predict_probazszMockEst.predict_probacCsdS)N?rrurrrscore}sz MockEst.scoreN)r_r`rarsrtrvrwryrrrrrqps rqc@sheZdZddZeddddZeddedd d Zed dd d ZedddZ dS)MockMetaEstimatorcCs ||_dSzMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegater]r}rrrrsszMockMetaEstimator.__init__cCs t|jdS)Nrvhasattrr}rrrrrr[r\zMockMetaEstimator.cCs |j|SzThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y r}rvrurrrrvs zMockMetaEstimator.predictcCs t|jdS)Nryrrrrrrr[r\%Testing a deprecated delegated methodcCsdSzThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y NrrurrrryszMockMetaEstimator.scorecCs t|jdS)Nrwrrrrrrr[r\cCs|SzThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X rrurrrrws zMockMetaEstimator.predict_proba-Testing deprecated function with wrong paramscCsdSz,Incorrect docstring but should not be testedNrrprrrrtszMockMetaEstimator.fitN) r_r`rarsrrvrryrwrtrrrrrzs    rzc@sbeZdZddZeddddZeddeddd Zeddd d Zed d dZ dS)%MockMetaEstimatorDeprecatedDelegationcCs ||_dSr{r|r~rrrrssz.MockMetaEstimatorDeprecatedDelegation.__init__r}r|cCs |j|Srrrurrrrvs z-MockMetaEstimatorDeprecatedDelegation.predictrcCsdSrrrurrrrysz+MockMetaEstimatorDeprecatedDelegation.scorecCs|Srrrurrrrws z3MockMetaEstimatorDeprecatedDelegation.predict_probarcCsdSrrrprrrrtsz)MockMetaEstimatorDeprecatedDelegation.fitN) r_r`rarsrrvrryrwrtrrrrrs    rz,ignore:if_delegate_has_method was deprecated mock_metar|c Cstjddddtt}|gks$tttdgd}|gks>tttdgd}|gksXttjtddttW5QRXtjtd dtt jW5QRXtt }|j j }|d d d gkstd ddddgddddddgdddddgdddddgdd|d d!dd"d#d$d#gdd|d%d&d'd(d)gdd|d*d'd(d)gdd|d+ddddgg}t |tttt j|j|j|j|jgD],\}}t|}||ks~td,||fq~dS)-NZnumpydocz+numpydoc is required to test the docstringsz1.2.0)reasonZ minversionre)ignorezUnknown section ResultsrzUnknown section Parameterzxsklearn.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'z 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: cz - ['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: bz+ ['a']z=In function: sklearn.utils.tests.test_testing.Klass.f_missingzaParameters in function docstring have less items w.r.t. function signature, first missing item: Xz - ['X', 'y']z+ []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'z- ['X']z? ^z+ ['y'].rwz&potentially wrong underline length... z Parameters z --------- in z.scorez.fitz "%s" not in "%s")r' importorskipr rgrrlr RuntimeErrorrhrorn __class__r_ziprirkrvrwryrt)rZ incorrectZmock_meta_namemessagesr,rZrrrtest_check_docstring_parameterss              G rc@seZdZddZddZdS)RegistrationCountercCs d|_dS)Nr)nb_callsrrrrrrsaszRegistrationCounter.__init__cCs |jd7_|jtkstdS)Nr#)rfuncrr)r]Zto_register_funcrrr__call__dszRegistrationCounter.__call__N)r_r`rarsrrrrrr`srrcCs:t|tjst|dk}|jj|ks(ttj||dS)Nr) isinstancer$Zmemmaprflags writeabletestingZassert_array_equal) input_arrayZ mmap_data mmap_moderrrr check_memmapisrc Cst}|td|td}t|}t||tj |j }W5QRXtj dkrdtj |rdt |jdksrt d}t||d"}t|||dtj |j }W5QRXtj dkrtj |rt |jdkst dS)Nregisterrntr#r+)rr )rsetattratexitr$r)rrospathdirnamefilenamenameexistsrr) monkeypatchregistration_counterrdataZ temp_folderrrrrtest_tempmemmapps      ralignedFTc Cs&t}|td|td}t||d}t|||jdksBtt|d|d\}}t|||t j |j kstt|jdkstd}t|||d }t||||jdkst||d|dg}t||d}t ||D]\}}t||q|jd ksttjtd d t|d gddW5QRXdS)Nrr)rr#T)Z return_folderrr rrrz`When creating aligned memmap-backed arrays, input must be a single array or a sequence of arraysrz not-an-array)rrrr$r)rrrrrrrrrr'rr() rrrrrfolderrZ input_listZmmap_data_listrrrtest_create_memmap_backed_datas8        rdtypecCsdtd|}|jdst|jds,tt|}t|ddd}t|}|tj|ddks`td S) z2Test memory mapped array on contiguous memoryview. Z C_CONTIGUOUSZALIGNEDrTrgdy=)relN) r$r%Zastyperrrrr'Zapprox)rr+Z sum_originZx_mmapZsum_mmaprrrtest_memmap_on_contiguous_datasrz constructor_name, container_typelisttuplearrayrZ sparse_csrZ sparse_csc dataframecCs tdjSNZpandas)r'rZ DataFramerrrrr[r\r[seriescCs tdjSr)r'rZSeriesrrrrr[r\indexcCs tdjSr)r'rIndexrrrrr[r\slicezdtype, superdtypecCs|dkr|}ddg}t|||d}t||s2t|dkrTtt|d|stn6t|drn|j|kstnt|dr|jd|kstdS) z\Check that we convert the container to the right type of array with the right data type.)rrrrr#r)rrrrdtypesN) rrrr$Z issubdtypetyperrr)Zconstructor_namecontainer_typerZ superdtype containerZcontainer_convertedrrrtest_convert_containers   rc Cstt tW5QRXttdd}tdW5QRX|jsDtttddgd}tdW5QRX|jsrttjtdd"tt}tdW5QRXW5QRX|jrttjtdd$ttdd }tW5QRXW5QRX|jrttjtdd&ttdd}tdW5QRXW5QRX|jr:ttjtd d&ttd d}td W5QRXW5QRX|jrttjtdd(ttd dd }td W5QRXW5QRX|jrttjtd dtt}W5QRXW5QRX|jrtttd dd}W5QRX|jr(ttttf tW5QRXtttf tW5QRXtttttfW5QRXW5QRXdS)Nz how are yourzhello how are youz not this onezthis will be raisedzthe failure message)err_msgzthis is ignoredz,should contain one of the following patternsZhelloz Bad message)rrz Did not raiseT)rZmay_pass)rr8Zraised_and_matchedrr'r()cmrrr test_raisessZ              rc CsttjdgtjddttttjdgtjddW5QRXttjdgtjddttttjdgtjddW5QRXttttjdgtjddW5QRXttjdgtjddd d dS) Ng㈵?rrxgX?gc^?g55?gh㈵>ggh㈵>)Zatol)rr$rfloat32r'rrfloat64rrrr"test_float32_aware_assert_allclose;s " " "r)r)Lr=Zunittestrrnumpyr$Zscipyrr'Zsklearn.utils.deprecationrZsklearn.utils.metaestimatorsrrZ%sklearn.utils._readonly_array_wrapperrZsklearn.utils._testingrrr r r r r rrrrrrrZ sklearn.treerZsklearn.discriminant_analysisrrr1r2r:rXZTestCaserYrgrhrirkrlrnrorqrzrmarkfilterwarningsZ parametrizerrrrrrrZint32Zint64rrrZndarrayr*r&rintegerZfloatingrrrrrrrs   @   'm11    u  %         F