\c@sYddlZddljZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd lmZdd lmZmZmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl"Z"defdYZ#defdYZ$defdYZ%defdYZ&defdYZ'de(fd YZ)d!efd"YZ*d#Z+d$Z,d%Z-d&Z.d'Z/d(Z0d)Z1d*Z2d+Z3d,Z4d-Z5d.Z6d/Z7d0Z8d1Z9d2Z:d3efd4YZ;d5Z<d6Z=d7efd8YZ>d9Z?d:Z@d;e(fd<YZAd=eeAfd>YZBd?ZCd@ZDdAefdBYZEedCeFdDZGdS(EiN(tassert_array_equal(t assert_true(t assert_false(t assert_equal(tassert_not_equal(t assert_raises(tassert_no_warnings(tassert_warns_message(tassert_dict_equal(tignore_warnings(t BaseEstimatortclonet is_classifier(tSVC(tPipeline(t GridSearchCV(tDecisionTreeClassifier(tDecisionTreeRegressor(tdatasets(tTransformerMixin(t MockDataFramet MyEstimatorcBseZdddZRS(icCs||_||_dS(N(tl1tempty(tselfRR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt__init__%s N(t__name__t __module__tNoneR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR#stKcBseZdddZRS(cCs||_||_dS(N(tctd(RRR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR+s N(RRRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR*stTcBseZdddZRS(cCs||_||_dS(N(tatb(RR!R"((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR1s N(RRRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR 0stModifyInitParamscBs&eZdZejdgdZRS(s_Deprecated behavior. Equal parameters but with a type cast. Doesn't fulfill a is a icCs|j|_dS(N(tcopyR!(RR!((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR;s(RRt__doc__tnptarrayR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR#6stBuggycBseZdZddZRS(s; A buggy estimator that does not set its parameters right. cCs d|_dS(Ni(R!(RR!((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRBsN(RRR%RR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR(?st NoEstimatorcBs,eZdZdddZddZRS(cCsdS(N((R((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRGscCs|S(N((RtXty((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytfitJscCsdS(N(R(RR*((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytpredictMsN(RRRRR,R-(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR)Fs t VargEstimatorcBseZdZdZRS(s-scikit-learn estimators shouldn't have vargs.cGsdS(N((Rtvargs((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRSs(RRR%R(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR.QscCsddlm}m}||dd}t|}||k sFtt|j|j||dtjd}t|}||k stdS(Ni(t SelectFprt f_classiftalphag?i i(i i( tsklearn.feature_selectionR0R1R tAssertionErrorRt get_paramsR&tzeros(R0R1tselectort new_selector((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt test_cloneZs  cCsTddlm}m}||dd}d|_t|}tt|ddS(Ni(R0R1R2g?ttestt own_attribute(R3R0R1R;R Rthasattr(R0R1R7R8((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt test_clone_2ls   cCsqt}d|_ttt|t}ttt|t}ttt|t}ttt|dS(Ni( R(R!Rt RuntimeErrorR R)t TypeErrorR.R#(tbuggyt no_estimatortvarg_esttest((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_clone_buggyzs     cCstdtjg}t|}t|j|jtdtjtjdgg}t|}t|jj|jjdS(NRi( RR&R'R RRtspt csr_matrixtdata(tclftclf2((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_clone_empty_arrays  ' cCs:tdtj}t|}|j|jks6tdS(NR(RR&tnanR RR4(RHRI((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_clone_nans cCsgttD]$}|jdr tt|^q }x{|D]s}|tjd}td|}t|}|jj |jj kst t |jj |jj q>WdS(Nt_matrixiR( tdirREtendswithtgetattrR&teyeRR Rt __class__R4Rttoarray(tnametsparse_matrix_classestclst sparse_matrixRHt clf_cloned((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_clone_sparse_matricess4  cCsnt}t|ttt}tt|dtddgd}ttt|ddS(Ns+T(a=K(c=None, d=None), b=K(c=None, d=None))R!t long_paramsii(RtreprR RRtlen(t my_estimatorR:tsome_est((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt test_reprs   cCst}t|dS(N(Rtstr(R]((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_strs cCsttt}d|jdtks3td|jdtksQt|jdd|jjdksytt t |jdddS(Nta__dtdeepita__a( R RR5tTrueR4tFalset set_paramsR!RRt ValueError(R:((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_get_paramss c Cst}t|sttt|iddgd6sCtttd|fgsdttttdt|iddgd6fgdS(Ng?itCtsvctsvc_cv(R R R4RRR(Rk((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_is_classifiers  (! cCsHtdtfg}tt|jdttt|jdtdS(NRktsvc__stupid_paramtsvm__stupid_param(RR RRhRgRe(RH((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_set_paramsscsdtffdYidd6dd6xHtdfgtigD]}|jddd dq^WdS( NtTestDecisionTreecseZfdZRS(cs,t|j||ks(t|S(N(tsuperRgR4(Rtkwargs(Rqtexpected_kwargs(s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRgs(RRRg((RqRt(s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRqsit max_depthitmin_samples_leaft estimatortestimator__max_depthtestimator__min_samples_leaf(RRRRg(RC((RqRts6lib/python2.7/site-packages/sklearn/tests/test_base.pyt%test_set_params_passes_all_parameterss cCsGtti}|jdtdd|jjdksCtdS(NRwt estimator__CgE@(RRRgR RwRjR4(tgscv((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt$test_set_params_updates_valid_paramsscCstjjd}tddtddg}tjtjg}xt||D]\}}|j |j |j |j dddt |j }t|j|j |j |j|j |j d|dd qXWdS( NiRuiii tsizet sample_weighttmsgs5Unweighted and weighted scores are unexpectedly equal(R&trandomt RandomStateRRRt load_irist load_bostontzipR,RGttargettrandintR\Rtscore(trngt estimatorstsetsRCtdsR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_score_sample_weights  ! cCsdttfdY}tjd}t|}||dd}t|}|j|jkjjsst t |j |j dS(NtDummyEstimatorcBs2eZdZdddZddZdZRS(s,This is a dummy class for generating numerical features This feature extractor extracts numerical features from pandas data frame. Parameters ---------- df: pandas data frame The pandas data frame parameter. Notes ----- icSs||_||_dS(N(tdft scalar_param(RRR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR s cSsdS(N((RR*R+((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyR,$scSsdS(N((RR*((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt transform'sN(RRR%RRR,R(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRs i Ri( R RR&tarangeRR RtvaluestallR4RR(RRRtetcloned_e((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyttest_clone_pandas_dataframes  !cCstj}tj|j|j}tj|}d|ksHtt tj |}|j |j|j}|j |j|j}t ||dS(Ntversion( RRRR,RGRtpickletdumpsR4RtloadsRR(tiristtreet tree_picklet tree_restoredtscore_of_originaltscore_of_restored((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt?test_pickle_version_warning_is_not_raised_with_matching_version5s tTreeBadVersioncBseZdZRS(cCst|jjddS(Nt_sklearn_versiont something(tdictt__dict__titems(R((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt __getstate__Cs(RRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRBssTrying to unpickle estimator {estimator} from version {old_version} when using version {current_version}. This might lead to breaking code or invalid results. Use at your own risk.cCsqtj}tj|j|j}tj|}tj dddddt j }t t |tj|dS(NRwRt old_versionRtcurrent_version(RRRR,RGRRRtpickle_error_messagetformattsklearnt __version__Rt UserWarningR(RRttree_pickle_othertmessage((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt<test_pickle_version_warning_is_issued_upon_different_versionOs  t TreeNoVersioncBseZdZRS(cCs|jS(N(R(R((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRZs(RRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRYscCstj}tj|j|j}tj|}td|kt j dddddt j }t t|tj|dS(NRRwRRspre-0.18R(RRRR,RGRRRRRRRRRRR(RRttree_pickle_noversionR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytDtest_pickle_version_warning_is_issued_when_no_version_info_in_pickle^s  cCsmtj}tj|j|j}tj|}z&tj}dt_t tj |Wd|t_XdS(Nt notsklearn( RRRR,RGRRRRRR(RRRt module_backup((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytCtest_pickle_version_no_warning_is_issued_with_non_sklearn_estimatorms   tDontPickleAttributeMixincBseZdZdZRS(cCs|jj}d|d<|S(Nt_attribute_not_pickled(RR$R(RRG((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRzs cCst|d<|jj|dS(Nt _restored(ReRtupdate(Rtstate((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt __setstate__s (RRRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRys tMultiInheritanceEstimatorcBseZddZRS(icCs||_d|_dS(N(tattribute_pickledRR(RR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRs (RRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRscCsct}d|_tj|}tj|}t|jdt|jd|js_t dS(Ns$this attribute should not be pickledi( RRRRRRRRRR4(Rwt serializedtestimator_restored((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyt3test_pickling_when_getstate_is_overwritten_by_mixins  cCszt}d}||_t|j}dt|_|j}t|idd6dd6d|d<|j|t|j d|j st Wd|t|_XdS(Ns$this attribute should not be pickledRRiRi( RRttypeRRRRRRRRR4(Rwttexttold_modR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytFtest_pickling_when_getstate_is_overwritten_by_mixin_outside_of_sklearns      tSingleInheritanceEstimatorcBseZddZdZRS(icCs||_d|_dS(N(RRR(RR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRs cCs|jj}d|d<|S(NR(RR$R(RRG((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRs (RRRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pyRs tcategorycCsTt}d|_tj|}tj|}t|jdt|jddS(Ns$this attribute should not be pickledi(RRRRRRRR(RwRR((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytCtest_pickling_works_when_getstate_is_overwritten_in_the_child_classs   (HtnumpyR&t scipy.sparsetsparseRERtsklearn.utils.testingRRRRRRRRRR t sklearn.baseR R R t sklearn.svmR tsklearn.pipelineRtsklearn.model_selectionRt sklearn.treeRRRRtsklearn.utils.mockingRRRRR R#R(tobjectR)R.R9R=RDRJRLRYR_RaRiRmRpRzR}RRRRRRRRRRRRRRRR(((s6lib/python2.7/site-packages/sklearn/tests/test_base.pytsr               &