o tBh|d@sddlZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl m Z mZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"ddl#m$Z$d d Z%d d Z&d dZ'ddZ(ddZ)Gdddej*Z+ddZ,ddZ-ddZ.ddZ/ddZ0dd Z1Gd!d"d"Z2Gd#d$d$Z3Gd%d&d&Z4Gd'd(d(Z5e j67d)e j68d*e4e3d+e5e3d+gd,d-Z9Gd.d/d/Z:dSd1d2Z;d3d4Zej?ej@ejAgd;d<ZBe j68d=d>eCfd?eDfd@ejEfdAejFfdBejFfdCejGfdDdEdFfdGdHdFfdIdJdFfdKeHfg e j68dLej@ejIfejAejIfej>ejJfej?ejJfgdMdNZKdOdPZLdQdRZMdS)TN)sparse) deprecated) available_ifif_delegate_has_method) _test_sum) assert_raises assert_warnsassert_no_warningsset_random_stateassert_raise_messageignore_warningscheck_docstring_parametersassert_allclose_dense_sparseassert_raises_regex TempMemmapcreate_memmap_backed_data_delete_folder_convert_containerraisesassert_allclose)DecisionTreeClassifier)LinearDiscriminantAnalysiscCs2t}t}t|dt|d|jdksJdS)N)rrr random_state)ldatreerw/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_state&s   rc Cstddd}d}t|}||fD]%}tjt|dt||dWdn1s/wYt||qtjt dd t||Wdn1sQwYtj t ddd  }t t d }tjtd dt||WddS1swYdS) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)r#zArrays are not equal)nparangereshaper csc_matrixpytestrAssertionErrorr ValueErrordiagsonestocsr csr_matrix)xmsgyXABrrr!test_assert_allclose_dense_sparse/s      "r7c Cslttd'ttddWdn1swYWddSWddS1s/wYdS)Nz Hello world)r2)rr+rr,rrrrtest_assert_raises_msgBs "r8cCsjdd}dd}ttd|dttttd|dttttd|dttttd|tttttfd|dS)NcSst|N)r,)messagerrr_raise_ValueErrorIsz4test_assert_raise_message.._raise_ValueErrorcSdSr9rrrrr _no_raiseLz,test_assert_raise_message.._no_raisetestzsomething else)r r,rr+ TypeErrorAttributeError)r;r=rrrtest_assert_raise_messageHs6  rBcsddddtttttdttttdWdn1s-wYttttdWdn1sIwYttttdWdn1sewYttttfdtfdd}tttfdfdd }ttdfd d }ttdfd d }ttdfdd}ttdfdd}t|t|t|tt |Wdn1swYtt |Wdn1swYtt |Wdn 1swYfdd}fdd}fdd}fdd} fdd} fdd} t|t|t|tt | Wdn 1sOwYtt | Wdn 1shwYtt | Wdn 1swYt} d} tjt| dt| }|Wdn 1swYtjt| dt| d d!}WddS1swYdS)"NcStdtdSNzdeprecation warningwarningswarnDeprecationWarningrrrr_warning_functionrz.test_ignore_warning.._warning_functioncSstdttddSrDrErrrr_multiple_warning_functionus z7test_ignore_warning.._multiple_warning_functioncategorycsdSr9rrrKrIrrdecorator_no_warnings z1test_ignore_warning..decorator_no_warningc dSr9rrrKrrdecorator_no_warning_multiple z:test_ignore_warning..decorator_no_warning_multiplecrPr9rrrIrr decorator_no_deprecation_warningrSz=test_ignore_warning..decorator_no_deprecation_warningcrPr9rrrTrrdecorator_no_user_warningrSz6test_ignore_warning..decorator_no_user_warningcrPr9rrrQrr)decorator_no_deprecation_multiple_warningrSzFtest_ignore_warning..decorator_no_deprecation_multiple_warningcrPr9rrrQrr"decorator_no_user_multiple_warningrSz?test_ignore_warning..decorator_no_user_multiple_warningcs2t WddS1swYdSr9)r rrTrrcontext_manager_no_warnings"z7test_ignore_warning..context_manager_no_warningcs:tttfd WddS1swYdSNrL)r rH UserWarningrrQrr#context_manager_no_warning_multiples"z@test_ignore_warning..context_manager_no_warning_multiplec6ttd WddS1swYdSrZr rHrrTrr&context_manager_no_deprecation_warning "zCtest_ignore_warning..context_manager_no_deprecation_warningcr]rZr r[rrTrrcontext_manager_no_user_warningr`z.context_manager_no_user_warningcr]rZr^rrQrr/context_manager_no_deprecation_multiple_warningr`zLtest_ignore_warning..context_manager_no_deprecation_multiple_warningcr]rZrarrQrr(context_manager_no_user_multiple_warningr`zEtest_ignore_warning..context_manager_no_user_multiple_warningzA'obj' should be a callable.+you should use 'category=UserWarning'r cSr<r9rrrrrr?sz!test_ignore_warning..test) r r rHr*warnsr[ FutureWarningrr,)rOrRrUrVrWrXrYr\r_rbrcrd warning_classr!silence_warnings_funcr?rrNrtest_ignore_warningos                   $ric@s&eZdZddZeedddZdS) TestWarnsc Csdd}t:tdttjdd}tttt|dks%JWdn1s/wYtj|ks;JWdn1sEwYt t  t |Wdn1s^wYt ddddksnJdS)NcSstddS)Nyor)rFrGrrrrfs zTestWarns.test_warn..fignorercSs|Sr9r)r1rrrz%TestWarns.test_warn..r%) rFcatch_warnings simplefilterr[filtersr*rerfrrr+r )selfrl filters_origrrr test_warns    zTestWarns.test_warnrLc Csvdd}d}tjdjdd}zz tt|d}Wn ty#YnwW|tjd_n|tjd_w|r9tddS)NcSrC)Nrk)rFrGrfrrrrrlrJz,TestWarns.test_warn_wrong_warning..fFrFTz#wrong warning caught by assert_warn)sysmodulesrrrr[r+)rsrlfailedrrrrrtest_warn_wrong_warnings  z!TestWarns.test_warn_wrong_warningN)__name__ __module__ __qualname__rur rfryrrrrrjsrjcC ||}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c rabcrrrf_ok rcCr})zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c rrrrrf_bad_sections rrcCs ||}|Sr~r)rrrrrr f_bad_order3rrcCr})zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r)rrdrrrf_too_many_param_docstringFsrcCr})zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c rrrrr f_missing[s rcCs||||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)rrrrerrrf_check_param_definitionlsrc@eZdZddZddZdS)KlasscCr<r9rrsr4r3rrrrr>zKlass.f_missingcCdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c NrrrrrrszKlass.f_bad_sectionsN)rzr{r|rrrrrrr rc@s4eZdZddZddZddZddZd d Zd S) MockEstcCr) MockEstimatorNrrsrrr__init__rozMockEst.__init__cC|Sr9rrrrrfitr>z MockEst.fitcCrr9rrsr4rrrpredictr>zMockEst.predictcCrr9rrrrr predict_probar>zMockEst.predict_probacCr)N?rrrrrscorer>z MockEst.scoreN)rzr{r|rrrrrrrrrrs  rc@sheZdZddZeddddZeddedd d Zed dd d ZedddZ dS)MockMetaEstimatorcC ||_dSzMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegatersrrrrr zMockMetaEstimator.__init__cC t|jdS)Nrhasattrrrrrrrn zMockMetaEstimator.cC |j|SzThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y rrrrrrr zMockMetaEstimator.predictcCr)Nrrrrrrrnr%Testing a deprecated delegated methodcCrzThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y NrrrrrrrozMockMetaEstimator.scorecCr)NrrrrrrrnrcCrzThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X rrrrrr zMockMetaEstimator.predict_proba-Testing deprecated function with wrong paramscCrz,Incorrect docstring but should not be testedNrrrrrrrozMockMetaEstimator.fitN) rzr{r|rrrrrrrrrrrrs     rc@sbeZdZddZeddddZeddeddd Zeddd d Zed d dZ dS)%MockMetaEstimatorDeprecatedDelegationcCrrrrrrrrrz.MockMetaEstimatorDeprecatedDelegation.__init__rrcCrrrrrrrrrz-MockMetaEstimatorDeprecatedDelegation.predictrcCrrrrrrrrroz+MockMetaEstimatorDeprecatedDelegation.scorecCrrrrrrrrrz3MockMetaEstimatorDeprecatedDelegation.predict_probarcCrrrrrrrrroz)MockMetaEstimatorDeprecatedDelegation.fitN) rzr{r|rrrrrrrrrrrrs     rz,ignore:if_delegate_has_method was deprecated mock_metarc Cstjddddtt}|gksJttdgd}|gksJttdgd}|gks,Jtjtdd ttWdn1sBwYtjtd d ttjWdn1s^wYtt }|j j }|gd kssJgd gd gd gddd|dddddddgdd|dddddgdd|ddddgdd|ddddd gg}t |t tttj|j|j|j|jgD]\}}t|}||ksJd!||fqdS)"Nnumpydocz+numpydoc is required to test the docstringsz1.2.0)reason minversionr)rmzUnknown section Resultsr zUnknown section Parameter)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 "%s" not in "%s")r* importorskipr rrr RuntimeErrorrrr __class__rzziprrrrrr)r incorrectmock_meta_namemessagesr2rlrrrtest_check_docstring_parameterss           G  rc@r)RegistrationCountercCs d|_dS)Nr)nb_callsrrrrrs zRegistrationCounter.__init__cCs |jd7_|jtusJdS)Nr%)rfuncr)rsto_register_funcrrr__call__szRegistrationCounter.__call__N)rzr{r|rrrrrrrrrrcCs:t|tjsJ|dk}|jj|usJtj||dS)Nr) isinstancer&memmapflags writeabletestingassert_array_equal) input_array mmap_data mmap_moderrrr check_memmapsrcCst}|td|td}t|}t||tj |j }Wdn1s*wYtj dkrggh㈵>)atol)rr&rfloat32r*rr+float64rrrr"test_float32_aware_assert_allclose]s    r)r)NrFunittestrvrrnumpyr&scipyrr*sklearn.utils.deprecationrsklearn.utils.metaestimatorsrr%sklearn.utils._readonly_array_wrapperrsklearn.utils._testingrrr r r r r rrrrrrrr sklearn.treersklearn.discriminant_analysisrrr7r8rBriTestCaserjrrrrrrrrrrmarkfilterwarnings parametrizerrrrrrrint32int64rrrndarrayr0r)rintegerfloatingrrrrrrrs   D   'm11 1   u  "          F