\c@sOdZddlZddlZddlZddlZddlZddlZddlZddl Zddlm Z ddl m Z y$ddl mZddl mZWn1ek rddlmZddlmZnXddlZddlZddljZddlZddlZyeWnek r>dZnXddlmZddlmZdd lm Z dd lm!Z!dd lm"Z"dd lm#Z#ddl$Z%ddl&Z&dd l'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6gZ7y9ddl8m9Z:dZ;e.e;e:Z9e7j<dWnek r}nXy9ddl8m=Z>dZ;e.e;e>Z=e7j<dWnek rnXddddddddd d!d"d#d$d%d&d'd(gZ?e?j@e7e4d)ZAeAjBZCeAjDZEeAjFZGeAjHZIeAjJZKejLjMZMeAjNZOeAjPZQeAjRZSeAjTZUeAjVZWeAjXZYeAjZZ[eAj\Z]e]Z^d*Z_d+Z`d,Zad-Zbdecd.Zdd/eefd0YZfd1Zgd2d3d4d5Zhe.d6dd7Zie.d6d8eefd9YZjd:Zkd;Zld<d=d>d?d@dAdBdCdDdEdFg ZmdGdHdIdJdKdLgZndMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdigZoepepdepdjZqdkdlZrdmZsyddltZtetjujve0dndoZwetjujvejxjydpdqkdndrZzetjuj{e/de|dndsZ}etjujve2j~j dndtZetjujvejdukdndvZWnek rnXdwZdxZepdyZdzeefd{YZd|epd}Zepd~ZddZddddZdS(sTesting utilities.iN(twraps(t itemgetter(turlopen(t HTTPError(tassert_allclose(tassert_almost_equal(tassert_approx_equal(tassert_array_equal(tassert_array_almost_equal(tassert_array_less(t BaseEstimatortClassifierMixint ClusterMixintRegressorMixintTransformerMixin(t deprecatedtIS_PYPYt _IS_32BIT(tjoblib(tTestCase(t signature(traisesssklearn.utils.testing.raises has been deprecated in version 0.20 and will be removed in 0.22. Please use sklearn.utils.testing.assert_raises instead.R(t with_setupssklearn.utils.testing.with_setup has been deprecated in version 0.20 and will be removed in 0.22.If your code relies on with_setup, please use nose.tools.with_setup instead.Rt assert_equaltassert_not_equalt assert_raisestassert_raises_regexpt assert_truet assert_falseRRRR t assert_lesstassert_less_equaltassert_greatertassert_greater_equalRRtSkipTestt__init__c sttjdt}tjd|||}ttdrug|D]}|jtjk rN|^qN}nt |dkst d|j nt fd|D}|st d|j |fnWdQX|S( sTest that a certain warning occurs. Parameters ---------- warning_class : the warning class The class to test for, e.g. UserWarning. func : callable Callable object to trigger warnings. *args : the positional arguments to `func`. **kw : the keyword arguments to `func` Returns ------- result : the return value of `func` trecordtalwaystVisibleDeprecationWarningis!No warning raised when calling %sc3s|]}|jkVqdS(N(tcategory(t.0twarning(t warning_class(s4lib/python2.7/site-packages/sklearn/utils/testing.pys ss#%s did not give warning: %s( is %s)N( tclean_warning_registrytwarningstcatch_warningstTruet simplefilterthasattrtnpR&R%tlentAssertionErrort__name__tany(R)tfunctargstkwtwtresulttetfound((R)s4lib/python2.7/site-packages/sklearn/utils/testing.pyt assert_warnsts  !c sttjdt}tjdttdrNtjdtjn|||}t|dkst d|j ng|D]}t |j |^q}t |st d|j |fnt} xgt|D]\} } | r| ^qD]u} || j} tt| dr6| jdn| } trT}nfd }|| rt} PqqW| st d |j | fnWd QX|S( s{Test that a certain warning occurs and with a certain message. Parameters ---------- warning_class : the warning class The class to test for, e.g. UserWarning. message : str | callable The message or a substring of the message to test for. If callable, it takes a string as the argument and will trigger an AssertionError if the callable returns `False`. func : callable Callable object to trigger warnings. *args : the positional arguments to `func`. **kw : the keyword arguments to `func`. Returns ------- result : the return value of `func` R#R$R%tignoreis!No warning raised when calling %ss&No warning raised for %s with class %sR6cs |kS(N((tmsg(tmessage(s4lib/python2.7/site-packages/sklearn/utils/testing.pytssCDid not receive the message you expected ('%s') for <%s>, got: '%s'N(R*R+R,R-R.R/R0R%R1R2R3t issubclassR&R4tFalset enumerateR?tstrR6tcallable(R)R?R5R6R7R8R9R(R;t message_foundtitxtindexR>tcheck_in_message((R?s4lib/python2.7/site-packages/sklearn/utils/testing.pytassert_warns_messages6 % 2 (   c Osztjdddd\y#tttjdtjdWntk rY|||SXttd|||SWdQXdS(sAssume that numpy's warning for divide by zero is raised Handles the case of platforms that do not support warning on divide by zero Parameters ---------- func *args **kw tdividetwarntinvalidisinvalid value encounteredN(R0terrstateR<tRuntimeWarningRLtzerosR2RK(R5R6R7((s4lib/python2.7/site-packages/sklearn/utils/testing.pytassert_warns_div0s # c Osttjdt}tjd|||}ttdrug|D]}|jtjk rN|^qN}nt |dkrt d|j dj d|DfnWdQX|S( s? Parameters ---------- func *args **kw R#R$R%is"Got warnings when calling %s: [%s]s, css|]}t|VqdS(N(RD(R'R(((s4lib/python2.7/site-packages/sklearn/utils/testing.pys sN( R*R+R,R-R.R/R0R&R%R1R2R3tjoin(R5R6R7R8R9R:((s4lib/python2.7/site-packages/sklearn/utils/testing.pytassert_no_warningss   !)cCsrt|trBt|trB|j}tdjd|n,t|ratd||Std|SdS(sContext manager and decorator to ignore warnings. Note: Using this (in both variants) will clear all warnings from all python modules loaded. In case you need to test cross-module-warning-logging, this is not your tool of choice. Parameters ---------- obj : callable or None callable where you want to ignore the warnings. category : warning class, defaults to Warning. The category to filter. If Warning, all categories will be muted. Examples -------- >>> with ignore_warnings(): ... warnings.warn('buhuhuhu') >>> def nasty_warn(): ... warnings.warn('buhuhuhu') ... print(42) >>> ignore_warnings(nasty_warn)() 42 s'obj' should be a callable where you want to ignore warnings. You passed a warning class instead: 'obj={warning_name}'. If you want to pass a warning class to ignore_warnings, you should use 'category={warning_name}'t warning_nameR&N( t isinstancettypeRAtWarningR3t ValueErrortformatREt_IgnoreWarnings(tobjR&RU((s4lib/python2.7/site-packages/sklearn/utils/testing.pytignore_warningss   R[cBs;eZdZdZdZdZdZdZRS(swImproved and simplified Python warnings context manager and decorator. This class allows the user to ignore the warnings raised by a function. Copied from Python 2.7.5 and modified as required. Parameters ---------- category : tuple of warning class, default to Warning The category to filter. By default, all the categories will be muted. cCs8t|_tjd|_t|_g|_||_dS(NR+( R-t_recordtsystmodulest_moduleRBt_enteredtlogR&(tselfR&((s4lib/python2.7/site-packages/sklearn/utils/testing.pyR"Ns    cs"tfd}|S(s<Decorator to catch and hide warnings without visual nesting.cs>ttj%tjdj||SWdQXdS(NR=(R*R+R,R.R&(R6tkwargs(tfnRd(s4lib/python2.7/site-packages/sklearn/utils/testing.pytwrapperWs (R(RdRfRg((RfRds4lib/python2.7/site-packages/sklearn/utils/testing.pyt__call__UscCsrg}|jr|jdn|jtjdk rL|jd|jnt|j}d|dj|fS(Ns record=TrueR+s module=%rs%s(%s)s, (R^tappendRaR_R`RWR3RS(RdR6tname((s4lib/python2.7/site-packages/sklearn/utils/testing.pyt__repr__`s cCsq|jrtd|nt|_|jj|_|j|j_|jj|_tt j d|j dS(NsCannot enter %r twiceR=( Rbt RuntimeErrorR-Ratfilterst_filterst showwarningt _showwarningR*R+R.R&(Rd((s4lib/python2.7/site-packages/sklearn/utils/testing.pyt __enter__is  cGsO|jstd|n|j|j_|j|j_g|j(tdS(Ns%Cannot exit %r without entering first( RbRlRnRaRmRpRoRcR*(Rdtexc_info((s4lib/python2.7/site-packages/sklearn/utils/testing.pyt__exit__ss   (R3t __module__t__doc__R"RhRkRqRs(((s4lib/python2.7/site-packages/sklearn/utils/testing.pyR[As    cOsy|||WnD|k rW}t|}||krtd||fqnNXt|trdjd|D}n |j}td||jfdS(sHelper function to test the message raised in an exception. Given an exception, a callable to raise the exception, and a message string, tests that the correct exception is raised and that the message is a substring of the error thrown. Used to test that the specific message thrown during an exception is correct. Parameters ---------- exceptions : exception or tuple of exception An Exception object. message : str The error message or a substring of the error message. function : callable Callable object to raise error. *args : the positional arguments to `function`. **kwargs : the keyword arguments to `function`. sRError message does not include the expected string: %r. Observed error message: %rs or css|]}|jVqdS(N(R3(R'R:((s4lib/python2.7/site-packages/sklearn/utils/testing.pys ss%s not raised by %sN(RDR2RVttupleRSR3(t exceptionsR?tfunctionR6ReR:t error_messagetnames((s4lib/python2.7/site-packages/sklearn/utils/testing.pytassert_raise_message|s   gHz>g& .>tc Cstjj|rtjj|r|j}|j}|j|jt|j|jd|t|j|jd|t|j |j d|d|d|nTtjj| rtjj| rt||d|d|d|n t ddS(s}Assert allclose for sparse and dense data. Both x and y need to be either sparse or dense, they can't be mixed. Parameters ---------- x : array-like or sparse matrix First array to compare. y : array-like or sparse matrix Second array to compare. rtol : float, optional relative tolerance; see numpy.allclose atol : float, optional absolute tolerance; see numpy.allclose. Note that the default here is more tolerant than the default for numpy.testing.assert_allclose, where atol=0. err_msg : string, default='' Error message to raise. terr_msgtrtoltatolsGCan only compare two sparse matrices, not a sparse matrix and an array.N( tsptsparsetissparsettocsrtsum_duplicatesRtindicestindptrRtdataRY(RHtyR~RR}((s4lib/python2.7/site-packages/sklearn/utils/testing.pytassert_allclose_dense_sparses$    (&"s8deprecated in version 0.20 to be removed in version 0.22cCst|}x|D]}||j||R:tunittestR*t NameErrorRt numpy.testingRRRRRR tnumpyR0Rt sklearn.baseR R R R Rt sklearn.utilsRRRtsklearn.utils._joblibRt sklearn.utils._unittest_backportRtsklearn.utils.fixesRtadditional_names_in_allt nose.toolsRt _nose_raisestdeprecation_messageRiRt _with_setupt__all__Rt_dummyt assertEqualRtassertNotEqualRt assertTrueRt assertFalseRt assertRaisesRtcaseR!tassertDictEqualtassert_dict_equaltassertInt assert_int assertNotInt assert_not_int assertLessRt assertGreaterRtassertLessEqualRtassertGreaterEqualR tassertRaisesRegextassert_raises_regexRR<RKRRRTRXR]RR[R{RRRRRRRRRBRR Rtpytesttmarktskipift skip_if_32bitR$R%t skip_travistxfailtNotImplementedErrort fails_if_pypytparalleltmptskip_if_no_paralleltplatformt!if_safe_multiprocessing_with_blasR*R&R,R-R1RKRQRo(((s4lib/python2.7/site-packages/sklearn/utils/testing.pyts                 (                     , E  *; *) + 0          f    !