From cebb7daee8f59cd9857ff80703e29e22f6c66b5f Mon Sep 17 00:00:00 2001 From: Gilles Louppe Date: Mon, 26 Aug 2013 09:43:30 +0200 Subject: [PATCH] TST: test_base in sklearn.ensemble --- sklearn/ensemble/tests/test_base.py | 32 +++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 sklearn/ensemble/tests/test_base.py diff --git a/sklearn/ensemble/tests/test_base.py b/sklearn/ensemble/tests/test_base.py new file mode 100644 index 00000000000..8747fdf9d9d --- /dev/null +++ b/sklearn/ensemble/tests/test_base.py @@ -0,0 +1,32 @@ +""" +Testing for the base module (sklearn.ensemble.base). +""" + +# Authors: Gilles Louppe +# License: BSD 3 clause + +from numpy.testing import assert_equal +from nose.tools import assert_raises, assert_true + +from sklearn.datasets import load_iris +from sklearn.ensemble import BaggingClassifier +from sklearn.linear_model import Perceptron + + +def test_base(): + """Check BaseEnsemble methods.""" + ensemble = BaggingClassifier(base_estimator=Perceptron(), n_estimators=3) + + iris = load_iris() + ensemble.fit(iris.data, iris.target) + ensemble.estimators_ = [] # empty the list and create estimators manually + + ensemble._make_estimator() + ensemble._make_estimator() + ensemble._make_estimator() + ensemble._make_estimator(append=False) + + assert_equal(3, len(ensemble)) + assert_equal(3, len(ensemble.estimators_)) + + assert_true(isinstance(ensemble[0], Perceptron))