def test_api_multiple_trees(self): mrbbagging = MRBBagging(5, tree.DecisionTreeClassifier(random_state=0), random_state=0) mrbbagging.fit(X_train, y_train) y_pred = mrbbagging.predict(X_test) assert all(y_pred == y_test)
def test_api_with_feature_selection(self): mrbbagging = MRBBagging(1, tree.DecisionTreeClassifier(random_state=0), feature_selection=True, random_state=0) mrbbagging.fit(X_train, y_train) y_pred = mrbbagging.predict(X_test) assert all(y_pred == y_test)
def test_fit_with_invalid_labels(self): mrbbagging = MRBBagging(1, MagicMock()) x = [[1, 1, 1], [2, 2, 2], [3, 3, 3]] y = ["A", "B"] with self.assertRaises(AssertionError): mrbbagging.fit(x, y)