U _{jW@s2ddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZdd Z d d Z!d d Z"ddZ#ddZ$Gdddej%Z&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,Gd d!d!Z-Gd"d#d#Z.Gd$d%d%Z/d&d'Z0Gd(d)d)Z1dJd+d,Z2d-d.Z3ej45d/d0d1gd2d3Z6ej45d4d5e7fd6e8fd7ej9fd8ej:fd9ej:fd:ej;fd;dd?d=fd@dAd=fdBefej?ej>fej@ejAfejBejAfgdDdEZCdFdGZDdHdIZEdS)KN)sparse)LinearDiscriminantAnalysis)DecisionTreeClassifier) TempMemmap_convert_container_delete_folderassert_allcloseassert_allclose_dense_sparseassert_no_warningsassert_raise_message assert_raisesassert_raises_regexcheck_docstring_parameterscreate_memmap_backed_dataignore_warningsraisesset_random_state) deprecated) available_ifcCs2t}t}t|dt|d|jdks.tdS)N)rrrZ random_stateAssertionError)Zldatreer_/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sklearn/utils/tests/test_testing.pytest_set_random_state s   rc Cstddd}d}t|}||fD]6}tjt|dt||dW5QRXt||q(tjt ddt||W5QRXtj t ddd  }t t d }tjtd dt||W5QRXdS) N rzNot equal to tolerance matchzCan only compare two sparser)offsets)rzArrays are not equal)npZarangeZreshaper csc_matrixpytestrrr ValueErrorZdiagsonesZtocsr csr_matrix)xmsgyXABrrr!test_assert_allclose_dense_sparse)s   r.c Cs2ttdttddW5QRXW5QRXdS)Nz Hello world)r))r rr r%rrrrtest_assert_raises_msg<s r/cCsjdd}dd}ttd|dttttd|dttttd|dttttd|tttttfd|dS)NcSs t|dSN)r%)messagerrr_raise_ValueErrorCsz4test_assert_raise_message.._raise_ValueErrorcSsdSr0rrrrr _no_raiseFsz,test_assert_raise_message.._no_raisetestzsomething else)r r%r r TypeErrorAttributeError)r2r3rrrtest_assert_raise_messageBs6  r7c 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_functionlsz.test_ignore_warning.._warning_functioncSstdttddSr8r9rrrr_multiple_warning_functionos z7test_ignore_warning.._multiple_warning_functioncategorycsdSr0rrr>r=rrdecorator_no_warningsz1test_ignore_warning..decorator_no_warningcs dSr0rrr>rrdecorator_no_warning_multiplesz:test_ignore_warning..decorator_no_warning_multiplecs dSr0rrr=rr decorator_no_deprecation_warningsz=test_ignore_warning..decorator_no_deprecation_warningcs dSr0rrrErrdecorator_no_user_warningsz6test_ignore_warning..decorator_no_user_warningcs dSr0rrrCrr)decorator_no_deprecation_multiple_warningszFtest_ignore_warning..decorator_no_deprecation_multiple_warningcs dSr0rrrCrr"decorator_no_user_multiple_warningsz?test_ignore_warning..decorator_no_user_multiple_warningc st W5QRXdSr0)rrrErrcontext_manager_no_warningsz7test_ignore_warning..context_manager_no_warningc s$tttfd W5QRXdSNr?)rr< UserWarningrrCrr#context_manager_no_warning_multiplesz@test_ignore_warning..context_manager_no_warning_multiplec s ttd W5QRXdSrKrr<rrErr&context_manager_no_deprecation_warnings zCtest_ignore_warning..context_manager_no_deprecation_warningc s ttd W5QRXdSrKrrLrrErrcontext_manager_no_user_warnings z.context_manager_no_user_warningc s ttd W5QRXdSrKrNrrCrr/context_manager_no_deprecation_multiple_warnings zLtest_ignore_warning..context_manager_no_deprecation_multiple_warningc s ttd W5QRXdSrKrPrrCrr(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'rcSsdSr0rrrrrr4sz!test_ignore_warning..test) r rr<r$ZwarnsrL FutureWarningrr%)rBrDrFrGrHrIrJrMrOrQrRrSZ warning_classrZsilence_warnings_funcr4rrArtest_ignore_warningisr                  rUc@seZdZddZdS) TestWarnsc Cs@dd}ttt|W5QRXtddddks.fcSs|Sr0r)r(rrrz%TestWarns.test_warn..r!)r$rrr )selfrWrrr test_warns zTestWarns.test_warnN)__name__ __module__ __qualname__r[rrrrrVsrVcCs ||}|SzFunction f Parameters ---------- a : int Parameter a b : float Parameter b Returns ------- c : list Parameter c rabcrrrf_oksrdcCs ||}|S)zFunction f Parameters ---------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c rr`rrrf_bad_sectionssrecCs ||}|Sr_r)rbrarcrrr f_bad_order srfcCs ||}|S)zFunction f Parameters ---------- a : int Parameter a b : int Parameter b c : int Parameter c Returns ------- d : list Parameter c r)rarbdrrrf_too_many_param_docstringsrhcCs ||}|S)zFunction f Parameters ---------- a : int Parameter a Returns ------- c : list Parameter c rr`rrr f_missing2s ricCs||||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)rarbrcrgerrrf_check_param_definitionCsrkc@seZdZddZddZdS)KlasscCsdSr0rrZr+r*rrrriWszKlass.f_missingcCsdS)zFunction f Parameter --------- a : int Parameter a b : float Parameter b Results ------- c : list Parameter c NrrmrrrreZszKlass.f_bad_sectionsN)r\r]r^rirerrrrrlVsrlc@s4eZdZddZddZddZddZd d Zd S) MockEstcCsdS)Z MockEstimatorNrrZrrr__init__mszMockEst.__init__cCs|Sr0rrmrrrfitpsz MockEst.fitcCs|Sr0rrZr+rrrpredictsszMockEst.predictcCs|Sr0rrrrrr predict_probavszMockEst.predict_probacCsdS)N?rrrrrrscoreysz MockEst.scoreN)r\r]r^rprqrsrtrvrrrrrnls rnc@sheZdZddZeddddZeddedd d Zed dd d ZedddZ dS)MockMetaEstimatorcCs ||_dS)zMetaEstimator to check if doctest on delegated methods work. Parameters --------- delegate : estimator Delegated estimator. Ndelegate)rZryrrrrp~szMockMetaEstimator.__init__cCs t|jdS)NrshasattrryrorrrrXrYzMockMetaEstimator.cCs |j|S)zThis is available only if delegate has predict. Parameters ---------- y : ndarray Parameter y )ryrsrrrrrrss zMockMetaEstimator.predictcCs t|jdS)NrvrzrorrrrXrYz%Testing a deprecated delegated methodcCsdS)zThis is available only if delegate has score. Parameters --------- y : ndarray Parameter y NrrrrrrrvszMockMetaEstimator.scorecCs t|jdS)NrtrzrorrrrXrYcCs|S)zThis is available only if delegate has predict_proba. Parameters --------- X : ndarray Parameter X rrrrrrrts zMockMetaEstimator.predict_probaz-Testing deprecated function with wrong paramscCsdS)z,Incorrect docstring but should not be testedNrrmrrrrqszMockMetaEstimator.fitN) r\r]r^rprrsrrvrtrqrrrrrw}s    rwc Cstjddddtt}|gks$tttdgd}|gks>tttdgd}|gksXttjtddttW5QRXtjtd dtt jW5QRXtt }t t d }|j j}|d d d gkstdddddgddddddgdddddgdddddgdd |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|}||kstd-||fqdS).NZnumpydocz+numpydoc is required to test the docstringsz1.2.0)reasonZ minversionrb)ignorezUnknown section ResultsrzUnknown section Parameterrxzxsklearn.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'].rtz&potentially wrong underline length... z Parameters z --------- in z.scorez.fitz "%s" not in "%s")r$ importorskiprrdrrir RuntimeErrorrerlrkrwrn __class__r\ziprfrhrsrtrvrq)Z incorrectZ mock_metaZmock_meta_namemessagesr)rWrrrtest_check_docstring_parameterss             S rc@seZdZddZddZdS)RegistrationCountercCs d|_dS)Nr)nb_callsrorrrrp7szRegistrationCounter.__init__cCs |jd7_|jtkstdS)Nr!)rfuncrr)rZZto_register_funcrrr__call__:szRegistrationCounter.__call__N)r\r]r^rprrrrrr6srrcCs:t|tjst|dk}|jj|ks(ttj||dS)Nr) isinstancer"Zmemmaprflags writeableZtestingZassert_array_equal) input_arrayZ mmap_data mmap_moderrrr check_memmap?src 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_tempmemmapFs      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_folderrrr)rrz`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_data[s8        rz constructor_name, container_typelisttuplearrayrZ sparse_csrZ sparse_csc dataframecCs tdjSNZpandas)r$rZ DataFramerrrrrXrYrXseriescCs tdjSr)r$rZSeriesrrrrrXrYindexcCs tdjSr)r$rIndexrrrrrXrYslicezdtype, 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!dtype)rrrrdtypesN) rrrr"Z issubdtypetyper{rr)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)rr5Zraised_and_matchedrr$r%)cmrrr test_raisessZ              rc CsttjdgtjddttttjdgtjddW5QRXttjdgtjddttttjdgtjddW5QRXttttjdgtjddW5QRXttjdgtjddd d dS) Ng㈵?rrugX?gc^?g55?gh㈵>ggh㈵>)Zatol)rr"rfloat32r$rrfloat64rrrr"test_float32_aware_assert_allcloses " " "r)r)FrrZunittestr:numpyr"r$ZscipyrZsklearn.discriminant_analysisrZ sklearn.treerZsklearn.utils._testingrrrrr r r r r rrrrrZsklearn.utils.deprecationrZsklearn.utils.metaestimatorsrrr.r/r7rUZTestCaserVrdrerfrhrirkrlrnrwrrrrmarkZ parametrizerrrZndarrayr'r#rZint32integerZint64rZfloatingrrrrrrrrst   @   'm1   %        F