Tests for new ensembles.

w4k2 · Nov 21, 2019 · 8bcee6d · 8bcee6d
1 parent 2776b7e
commit 8bcee6d
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/strlearn/tests/test_ensembles.py b/strlearn/tests/test_ensembles.py
@@ -26,6 +26,29 @@ def test_WAE():
     evaluator.process(stream, clf)
 
 
+def test_OOB():
+    """Bare WAE."""
+    stream = get_stream()
+    clf = sl.ensembles.OOB()
+    evaluator = sl.evaluators.TestThenTrainEvaluator()
+    evaluator.process(stream, clf)
+
+def test_OB():
+    """Bare WAE."""
+    stream = get_stream()
+    clf = sl.ensembles.OnlineBagging()
+    evaluator = sl.evaluators.TestThenTrainEvaluator()
+    evaluator.process(stream, clf)
+
+def test_UOB():
+    """Bare WAE."""
+    stream = get_stream()
+    clf = sl.ensembles.UOB()
+    evaluator = sl.evaluators.TestThenTrainEvaluator()
+    evaluator.process(stream, clf)
+
+
+
 def test_pp_WAE():
     """Post pruned WAE."""
     stream = get_stream()