add extra tree's tests to decision tree tester

openscilab · Sep 20, 2023 · a11be71 · a11be71
1 parent 4237507
commit a11be71
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_decision_trees/test_decision_trees.py b/tests/test_decision_trees/test_decision_trees.py
@@ -3,9 +3,12 @@
 
 from decision_tree.decision_tree_regression import decision_tree_regression
 from decision_tree.decision_tree_classification import decision_tree_classification
+from extra_tree.extra_tree_regression import extra_tree_regression
+from extra_tree.extra_tree_classification import extra_tree_classification
+
 DECISION_TREES = {
     "DECISION_TREE": [decision_tree_regression, decision_tree_classification],
-
+    "EXTRA TREE": [extra_tree_regression, extra_tree_classification],
 }
 
 @pytest.fixture(scope="session", autouse=True)