SAMOA-34: Check for ensemble consistency
diff --git a/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java b/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
index 6c88d94..a49065b 100644
--- a/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
+++ b/samoa-api/src/main/java/org/apache/samoa/learners/classifiers/ensemble/BaggingDistributorProcessor.java
@@ -34,6 +34,8 @@
import org.apache.samoa.moa.core.MiscUtils;
import org.apache.samoa.topology.Stream;
+import com.google.common.base.Preconditions;
+
/**
* The Class BaggingDistributorPE.
*/
@@ -58,6 +60,8 @@
* @return true, if successful
*/
public boolean process(ContentEvent event) {
+ Preconditions.checkState(ensembleSize == ensembleStreams.length, String.format(
+ "Ensemble size ({}) and number of enseble streams ({}) do not match.", ensembleSize, ensembleStreams.length));
InstanceContentEvent inEvent = (InstanceContentEvent) event;
if (inEvent.getInstanceIndex() < 0) {